lib/standard/math.nit

   1 # This file is part of NIT ( http://www.nitlanguage.org ).
   2 #
   3 # Copyright 2004-2008 Jean Privat <jean@pryen.org>
   4 #
   5 # This file is free software, which comes along with NIT.  This software is
   6 # distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
   7 # without  even  the implied warranty of  MERCHANTABILITY or  FITNESS FOR A
   8 # PARTICULAR PURPOSE.  You can modify it is you want,  provided this header
   9 # is kept unaltered, and a notification of the changes is added.
  10 # You  are  allowed  to  redistribute it and sell it, alone or is a part of
  11 # another product.
  12
  13 # Mathematical operations
  14 module math
  15
  16 import kernel
  17 import collection
  18
  19 in "C header" `{
  20 #include <math.h>
  21 `}
  22
  23 redef class Int
  24         # Returns a random `Int` in `[0 .. self[`.
  25         fun rand: Int is extern "kernel_Int_Int_rand_0"
  26         fun bin_and(i: Int): Int is extern "kernel_Int_Int_binand_0"
  27         fun bin_or(i: Int): Int is extern "kernel_Int_Int_binor_0"
  28         fun bin_xor(i: Int): Int is extern "kernel_Int_Int_binxor_0"
  29         fun sqrt : Int is extern `{ return sqrtl(recv); `}
  30         fun sin : Int is extern `{ return sinl(recv); `}
  31         fun cos : Int is extern `{ return cosl(recv); `}
  32 end
  33
  34 redef class Float
  35         fun sqrt: Float is extern "kernel_Float_Float_sqrt_0"
  36         fun cos: Float is extern "kernel_Float_Float_cos_0"
  37         fun sin: Float is extern "kernel_Float_Float_sin_0"
  38         fun tan: Float is extern "kernel_Float_Float_tan_0"
  39         fun acos: Float is extern "kernel_Float_Float_acos_0"
  40         fun asin: Float is extern "kernel_Float_Float_asin_0"
  41         fun atan: Float is extern "kernel_Float_Float_atan_0"
  42         fun abs: Float `{ return abs(recv); `}
  43
  44         fun pow(e: Float): Float is extern "kernel_Float_Float_pow_1"
  45         fun log: Float is extern "kernel_Float_Float_log_0"
  46         fun exp: Float is extern "kernel_Float_Float_exp_0"
  47
  48         # Returns a random `Float` in `[0.0 .. self[`.
  49         fun rand: Float is extern "kernel_Float_Float_rand_0"
  50         fun hypot_with( b : Float ) : Float is extern "hypotf"
  51 end
  52
  53 redef class Collection[ E ]
  54         # Return a random element form the collection
  55         # There must be at least one element in the collection
  56         fun rand: E
  57         do
  58                 if is_empty then abort
  59                 var rand_index = length.rand
  60
  61                 for e in self do
  62                         if rand_index == 0 then return e
  63                         rand_index -= 1
  64                 end
  65                 abort
  66         end
  67 end
  68
  69 fun atan2(x: Float, y: Float): Float is extern "kernel_Any_Any_atan2_2"
  70 fun pi: Float is extern "kernel_Any_Any_pi_0"
  71 fun srand_from(x: Int) is extern "kernel_Any_Any_srand_from_1"
  72 fun srand is extern "kernel_Any_Any_srand_0"