Add random module

windelbouwman · windelbouwman · commit f227ce04986e · 2018-11-25T23:20:10.000+01:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/vm/Cargo.toml b/vm/Cargo.toml
@@ -8,6 +8,7 @@ bitflags = "1.0.4"
 num-complex = "0.2"
 num-bigint = "0.2.1"
 num-traits = "0.2"
+rand = "0.5"
 log = "0.3"
 rustpython_parser = {path = "../parser"}
 serde = "1.0.66"
diff --git a/vm/src/stdlib/mod.rs b/vm/src/stdlib/mod.rs
@@ -4,6 +4,7 @@ mod json;
 mod keyword;
 mod math;
 mod pystruct;
+mod random;
 mod re;
 mod time_module;
 mod tokenize;
@@ -23,6 +24,7 @@ pub fn get_module_inits() -> HashMap<String, StdlibInitFunc> {
     modules.insert("keyword".to_string(), keyword::mk_module as StdlibInitFunc);
     modules.insert("math".to_string(), math::mk_module as StdlibInitFunc);
     modules.insert("re".to_string(), re::mk_module as StdlibInitFunc);
+    modules.insert("random".to_string(), random::mk_module as StdlibInitFunc);
     modules.insert("struct".to_string(), pystruct::mk_module as StdlibInitFunc);
     modules.insert("time".to_string(), time_module::mk_module as StdlibInitFunc);
     modules.insert(
diff --git a/vm/src/stdlib/random.rs b/vm/src/stdlib/random.rs
@@ -0,0 +1,62 @@
+//! Random module.
+
+extern crate rand;
+
+use super::super::obj::{objfloat, objtype};
+use super::super::pyobject::{
+    DictProtocol, PyContext, PyFuncArgs, PyObjectRef, PyResult, TypeProtocol,
+};
+use super::super::VirtualMachine;
+use stdlib::random::rand::distributions::{Distribution, Normal};
+
+pub fn mk_module(ctx: &PyContext) -> PyObjectRef {
+    let py_mod = ctx.new_module(&"random".to_string(), ctx.new_scope(None));
+    py_mod.set_item("gauss", ctx.new_rustfunc(random_gauss));
+    py_mod.set_item("normalvariate", ctx.new_rustfunc(random_normalvariate));
+    py_mod.set_item("random", ctx.new_rustfunc(random_random));
+    // py_mod.set_item("weibull", ctx.new_rustfunc(random_weibullvariate));
+    // TODO: implement more random functions.
+    py_mod
+}
+
+fn random_gauss(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    // TODO: is this the same?
+    random_normalvariate(vm, args)
+}
+
+fn random_normalvariate(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(
+        vm,
+        args,
+        required = [
+            (mu, Some(vm.ctx.float_type())),
+            (sigma, Some(vm.ctx.float_type()))
+        ]
+    );
+    let mu = objfloat::get_value(mu);
+    let sigma = objfloat::get_value(sigma);
+    let normal = Normal::new(mu, sigma);
+    let value = normal.sample(&mut rand::thread_rng());
+    let py_value = vm.ctx.new_float(value);
+    Ok(py_value)
+}
+
+fn random_random(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(vm, args);
+    let value = rand::random::<f64>();
+    let py_value = vm.ctx.new_float(value);
+    Ok(py_value)
+}
+
+/*
+ * TODO: enable this function:
+fn random_weibullvariate(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(vm, args, required = [(alpha, Some(vm.ctx.float_type())), (beta, Some(vm.ctx.float_type()))]);
+    let alpha = objfloat::get_value(alpha);
+    let beta = objfloat::get_value(beta);
+    let weibull = Weibull::new(alpha, beta);
+    let value = weibull.sample(&mut rand::thread_rng());
+    let py_value = vm.ctx.new_float(value);
+    Ok(py_value)
+}
+*/
