Class: Lisp::PrimMath

Inherits:
Object show all
Defined in:
lib/rubylisp/prim_math.rb

Class Method Summary collapse

Class Method Details

.abs_impl(args, env) ⇒ Object 



315
316
317
318
319
 
# File 'lib/rubylisp/prim_math.rb', line 315

def self.abs_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("abs needs a numeric argument, but received #{arg}", env) unless arg.number?
  Number.with_value(arg.value.abs)
end

.add_impl(args, env) ⇒ Object 



130
131
132
133
134
135
136
137
138
139
140
141
 
# File 'lib/rubylisp/prim_math.rb', line 130

def self.add_impl(args, env)
  acc = 0
  c = args
  while !c.nil?
    n = c.car
    return Lisp::Debug.process_error("add needs number arguments but was given a #{n.type}: #{n}", env) unless n.type == :number
    acc += n.value
    c = c.cdr
  end

  Number.with_value(acc)
end

.bind(name, number) ⇒ Object 



126
127
128
 
# File 'lib/rubylisp/prim_math.rb', line 126

def self.bind(name, number)
  EnvironmentFrame.global.bind(Symbol.named(name), Number.with_value(number))
end

.ceiling_impl(args, env) ⇒ Object 



235
236
237
238
239
 
# File 'lib/rubylisp/prim_math.rb', line 235

def self.ceiling_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("ceiling needs a number argument, but received #{arg}", env) unless arg.number?
  Number.with_value(arg.value.ceil)
end

.cos_impl(args, env) ⇒ Object 



368
369
370
371
372
 
# File 'lib/rubylisp/prim_math.rb', line 368

def self.cos_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("cos needs a numeric argument, but received #{arg}", env) unless arg.number?
  Number.with_value(::Math.cos(arg.value).round(5))
end

.even_impl(args, env) ⇒ Object 



249
250
251
252
253
 
# File 'lib/rubylisp/prim_math.rb', line 249

def self.even_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("even? needs a number argument, but received #{arg}", env) unless arg.number?
  Boolean.with_value(arg.value.even?)
end

.float_impl(args, env) ⇒ Object 



301
302
303
304
305
 
# File 'lib/rubylisp/prim_math.rb', line 301

def self.float_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("float needs a numeric or string argument, but received #{arg}", env) unless arg.number? || arg.string?
  Number.with_value(arg.value.to_f)
end

.floor_impl(args, env) ⇒ Object 



242
243
244
245
246
 
# File 'lib/rubylisp/prim_math.rb', line 242

def self.floor_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("floor needs a number argument, but received #{arg}", env) unless arg.number?
  Number.with_value(arg.value.floor)
end

.integer_impl(args, env) ⇒ Object 



308
309
310
311
312
 
# File 'lib/rubylisp/prim_math.rb', line 308

def self.integer_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("integer needs a numeric or string argument, but received #{arg}", env) unless arg.number? || arg.string?
  Number.with_value(arg.value.to_i)
end

.interval_impl(args, env) ⇒ Object 



284
285
286
287
288
289
290
291
 
# File 'lib/rubylisp/prim_math.rb', line 284

def self.interval_impl(args, env)
  initial = args.car
  return Lisp::Debug.process_error("interval needs number arguments, but received #{initial}", env) unless initial.number?
  final = args.cadr
  return Lisp::Debug.process_error("interval needs number arguments, but received #{final}", env) unless final.number?
  return Lisp::Debug.process_error("interval's arguments need to be in natural order", env) unless initial.value <= final.value
  Lisp::ConsCell.array_to_list((initial.value..final.value).to_a.map {|n| Number.with_value(n)})
end

.max_impl(args, env) ⇒ Object 



345
346
347
348
349
350
351
352
353
354
355
356
357
358
 
# File 'lib/rubylisp/prim_math.rb', line 345

def self.max_impl(args, env)
  initial = args.car
  return Lisp::Debug.process_error("max requires numeric arguments, but received #{initial}", env) unless initial.number?
  acc = initial.value
  c = args.cdr
  while !c.nil?
    n = c.car
    return Lisp::Debug.process_error("max needs number arguments, but received #{n}", env) unless n.number?
    acc = n.value if n.value > acc
    c = c.cdr
  end

  Number.with_value(acc)
end

.min_impl(args, env) ⇒ Object 



329
330
331
332
333
334
335
336
337
338
339
340
341
342
 
# File 'lib/rubylisp/prim_math.rb', line 329

def self.min_impl(args, env)
  initial = args.car
  return Lisp::Debug.process_error("min requires numeric arguments, but received #{initial}", env) unless initial.number?
  acc = initial.value
  c = args.cdr
  while !c.nil?
    n = c.car
    return Lisp::Debug.process_error("min needs number arguments, but received #{n}", env) unless n.number?
    acc = n.value if n.value < acc
    c = c.cdr
  end

  Number.with_value(acc)
end

.multiply_impl(args, env) ⇒ Object 



162
163
164
165
166
167
168
169
170
171
172
173
 
# File 'lib/rubylisp/prim_math.rb', line 162

def self.multiply_impl(args, env)
  acc = 1
  c = args
  while !c.nil?
    n = c.car
    return Lisp::Debug.process_error("multiply needs number arguments, but received #{n}", env) unless n.number?
    acc *= n.value
    c = c.cdr
  end

  Number.with_value(acc)
end

.negative_impl(args, env) ⇒ Object 



277
278
279
280
281
 
# File 'lib/rubylisp/prim_math.rb', line 277

def self.negative_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("negative? needs a number argument, but received #{arg}", env) unless arg.number?
  Boolean.with_value(arg.value < 0)
end

.odd_impl(args, env) ⇒ Object 



256
257
258
259
260
 
# File 'lib/rubylisp/prim_math.rb', line 256

def self.odd_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("odd? needs a number argument, but received #{arg}", env) unless arg.number?
  Boolean.with_value(arg.value.odd?)
end

.positive_impl(args, env) ⇒ Object 



270
271
272
273
274
 
# File 'lib/rubylisp/prim_math.rb', line 270

def self.positive_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("positive? needs a number argument, but received #{arg}", env) unless arg.number?
  Boolean.with_value(arg.value > 0)
end

.pred_impl(args, env) ⇒ Object 



382
383
384
385
386
 
# File 'lib/rubylisp/prim_math.rb', line 382

def self.pred_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("pred needs a numeric argument, but received #{arg}", env) unless arg.number?
  Number.with_value(arg.value - 1)
end

.quotient_impl(args, env) ⇒ Object 



176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
 
# File 'lib/rubylisp/prim_math.rb', line 176

def self.quotient_impl(args, env)
  first = args.car
  return Lisp::Debug.process_error("quotient needs number arguments, but received #{first}", env) unless first.number?
  return first if args.length == 1
  acc = first.value
  c = args.cdr
  while !c.nil?
    n = c.car
    return Lisp::Debug.process_error("quotient needs number arguments, but received #{n}", env) unless n.number?
    return Lisp::Debug.process_error("divide by zero", env) if n.value == 0
    acc /= n.value
    c = c.cdr
  end

  Number.with_value(acc)
end

.random_impl(args, env) ⇒ Object 



294
295
296
297
298
 
# File 'lib/rubylisp/prim_math.rb', line 294

def self.random_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("random needs a number argument, but received #{arg}", env) unless arg.nil? || arg.number?
  Number.with_value(arg.nil? ? rand() : rand(arg.value))
end

.register ⇒ Object 



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
 
# File 'lib/rubylisp/prim_math.rb', line 5

def self.register
  self.bind("PI", ::Math::PI)
  self.bind("E", ::Math::E)
  
  Primitive.register("+", ">=1", "(+ number...)\n\nAdds a series of numbers.") do |args, env|
    Lisp::PrimMath.add_impl(args, env)
  end
  
  Primitive.register("-", ">=1", "(- number...)\n\nSequentially subtracts a sequence of numbers.\nAs expected, a unary form of - is available as well.") do |args, env|
    Lisp::PrimMath.subtract_impl(args, env)
  end

  Primitive.register("*", ">=1", "(* number...)\n\nMultiplies a series of numbers.") do |args, env|
    Lisp::PrimMath.multiply_impl(args, env)
  end

  Primitive.register("/", ">=1", "(/ number...)\n\nSequentially divides a sequence of numbers.") do |args, env|
    Lisp::PrimMath.quotient_impl(args, env)
  end

  Primitive.register("%", "1|2", "(% number number)\n\nReturns the remainder of the division of two numbers. NOTE: modulus only works for integers.") do |args, env|
    Lisp::PrimMath.remainder_impl(args, env)
  end

  Primitive.register("modulo", "1|2", "(modulo number number)\n\nReturns the remainder of the division of two numbers. NOTE: modulus only works for integers.") do |args, env|
    Lisp::PrimMath.remainder_impl(args, env)
  end

  Primitive.register("even?", "1", "(even? number)\n\nReturns whether the argument is even.") do |args, env|
    Lisp::PrimMath.even_impl(args, env)
  end

  Primitive.register("odd?", "1", "(odd? number)\n\nReturns whether the argument is odd.") do |args, env|
    Lisp::PrimMath.odd_impl(args, env)
  end

  Primitive.register("zero?", "1", "(zero? number)\n\nReturns whether the argument is zero.") do |args, env|
    Lisp::PrimMath.zero_impl(args, env)
  end

  Primitive.register("positive?", "1", "(positive? number)\n\nReturns whether the argument is positive.") do |args, env|
    Lisp::PrimMath.positive_impl(args, env)
  end

  Primitive.register("negative?", "1", "(negative? number)\n\nReturns whether the argument is negative.") do |args, env|
    Lisp::PrimMath.negative_impl(args, env)
  end

  Primitive.register("interval", "2", "(interval lo hi)\n\nCreates a list of numbers from `lo` to `hi`, inclusive.") do |args, env|
    Lisp::PrimMath.interval_impl(args, env)
  end

  Primitive.register("truncate", "1", "(truncate number)\n\nReturns the integer value of number. If it is an integer, it is simply returned. However, if it is a float the integer part is returned.") do |args, env|
    Lisp::PrimMath.truncate_impl(args, env)
  end

  Primitive.register("round", "1", "(round number)\n\nIf number is an integer, it is simply returned. However, if it is a float the closest integer is returned.") do |args, env|
    Lisp::PrimMath.round_impl(args, env)
  end

  Primitive.register("ceiling", "1", "(ceiling number)\n\nIf `number` is an integer, it is simply returned. However, if it is a float the smallest integer greater than or equal to `number` is returned.") do |args, env|
    Lisp::PrimMath.ceiling_impl(args, env)
  end

  Primitive.register("floor", "1", "(floor number)\n\nIf `number` is an integer, it is simply returned. However, if it is a float the
largest integer less than or equal to `number` is returned.") do |args, env|
    Lisp::PrimMath.floor_impl(args, env)
  end

  Primitive.register("random", "0|1", "(random)\n\nReturns a pseudo-random floating point number between 0.0 and 1.0, including 0.0 and excluding 1.0.\n\n(random n)\n\nReturns a pseudo-random integer greater than or equal to 0 and less than n.") do |args, env|
    Lisp::PrimMath.random_impl(args, env)
  end

  Primitive.register("float", "1", "(float number)\n\nReturns the floating point value of number. If it is a float, it is simply returned. However, if it is an integer it is converted to float and returned.") do |args, env|
    Lisp::PrimMath.float_impl(args, env)
  end

  Primitive.register("integer", "1", "(integer number)\n\nReturns the integer value of number. If it is an integer, it is simply returned. However, if it is a float the integer part is returned. This is the same as tuncate.") do |args, env|
    Lisp::PrimMath.integer_impl(args, env)
  end

  Primitive.register("sqrt", "1", "(sqrt number)\n\nReturns the square root of `number'.") do |args, env|
    Lisp::PrimMath.sqrt_impl(args, env)
  end

  Primitive.register("min", ">=1", "(min number...)\n\nReturns the smallest of all the `number` arguments.") do |args, env|
    Lisp::PrimMath.min_impl(args, env)
  end

  Primitive.register("max", ">=1", "(max number...)\n\nReturns the largest of all the `number` arguments.") do |args, env|
    Lisp::PrimMath.max_impl(args, env)
  end

  Primitive.register("abs", "1", "(abs number)\n\nReturns the absolute value of `number'.") do |args, env|
    Lisp::PrimMath.abs_impl(args, env)
  end

  Primitive.register("sin", "1", "(sin number)\n\nReturns the sine of `number'.") do |args, env|
    Lisp::PrimMath.sin_impl(args, env)
  end

  Primitive.register("cos", "1", "(cos number)\n\nReturns the cosine of `number'.") do |args, env|
    Lisp::PrimMath.cos_impl(args, env)
  end

  Primitive.register("tan", "1", "(tan number)\n\nReturns the tangent of `number'.") do |args, env|
    Lisp::PrimMath.tan_impl(args, env)
  end

  Primitive.register("pred", "1", "(pred number)\n\nReturns one less than `number'.") do |args, env|
    Lisp::PrimMath.pred_impl(args, env)
  end
  
  Primitive.register("succ", "1", "(succ number)\n\nReturns one more than `number'.") do |args, env|
    Lisp::PrimMath.succ_impl(args, env)
  end

  
end

.remainder_impl(args, env) ⇒ Object 



194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
 
# File 'lib/rubylisp/prim_math.rb', line 194

def self.remainder_impl(args, env)
  first = args.car
  return Lisp::Debug.process_error("remainder needs number arguments, but received #{first}", env) unless first.number?
  return first if args.length == 1
  acc = first.value
  c = args.cdr
  while !c.nil?
    n = c.car
    return Lisp::Debug.process_error("remainder needs number arguments, but received #{n}", env) unless n.number?
    return Lisp::Debug.process_error("divide by zero", env) if n.value == 0
    acc %= n.value
    c = c.cdr
  end

  Number.with_value(acc)
end

.round_impl(args, env) ⇒ Object 



218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
 
# File 'lib/rubylisp/prim_math.rb', line 218

def self.round_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("round needs a number argument, but received #{arg}", env) unless arg.number?
  num = arg.value
  int = num.to_i
  Number.with_value(if (num - int).abs == 0.5
                      if int.even?
                        int
                      else
                        int + (int < 0 ? -1 : 1)
                      end
                    else
                      arg.value.round
                    end)
end

.sin_impl(args, env) ⇒ Object 



361
362
363
364
365
 
# File 'lib/rubylisp/prim_math.rb', line 361

def self.sin_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("sin needs a numeric argument, but received #{arg}", env) unless arg.number?
  Number.with_value(::Math.sin(arg.value).round(5))
end

.sqrt_impl(args, env) ⇒ Object 



322
323
324
325
326
 
# File 'lib/rubylisp/prim_math.rb', line 322

def self.sqrt_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("sqrt needs a numeric argument, but received #{arg}", env) unless arg.number?
  Number.with_value(::Math.sqrt(arg.value).round(5))
end

.subtract_impl(args, env) ⇒ Object 



144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
 
# File 'lib/rubylisp/prim_math.rb', line 144

def self.subtract_impl(args, env)
  return Number.with_value(-1 * args.car.value) if args.length == 1

  first = args.car
  return Lisp::Debug.process_error("subtract needs number arguments, but received #{first}", env)  unless first.number?
  acc = first.value
  c = args.cdr
  while !c.nil?
    n = c.car
    return Lisp::Debug.process_error("subtract needs number arguments, but received #{n}", env) unless n.number?
    acc -= n.value
    c = c.cdr
  end

  Number.with_value(acc)
end

.succ_impl(args, env) ⇒ Object 



389
390
391
392
393
 
# File 'lib/rubylisp/prim_math.rb', line 389

def self.succ_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("succ needs a numeric argument, but received #{arg}", env) unless arg.number?
  Number.with_value(arg.value + 1)
end

.tan_impl(args, env) ⇒ Object 



375
376
377
378
379
 
# File 'lib/rubylisp/prim_math.rb', line 375

def self.tan_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("tan needs a numeric argument, but received #{arg}", env) unless arg.number?
  Number.with_value(::Math.tan(arg.value).round(5))
end

.truncate_impl(args, env) ⇒ Object 



211
212
213
214
215
 
# File 'lib/rubylisp/prim_math.rb', line 211

def self.truncate_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("truncate needs a number argument, but received #{arg}", env) unless arg.number?
  Number.with_value(arg.value.truncate)
end

.zero_impl(args, env) ⇒ Object 



263
264
265
266
267
 
# File 'lib/rubylisp/prim_math.rb', line 263

def self.zero_impl(args, env)
  arg = args.car
  return Lisp::Debug.process_error("zero? needs a number argument, but received #{arg}", env) unless arg.number?
  Boolean.with_value(arg.value.zero?)
end