diff --git a/params.toml b/params.toml index 4488db3..3a2c575 100644 --- a/params.toml +++ b/params.toml @@ -15,8 +15,8 @@ [statements.gen_decl] gen_boolean = 0.0 gen_character = 0.0 - gen_integer = 0.0 - gen_real = 0.0 + gen_integer = 0.5 + gen_real = 0.5 gen_tuple = 0.0 qual_const = 0.0 qual_var = 0.0 diff --git a/src/ast_builder.rs b/src/ast_builder.rs index b017fec..17db0bf 100644 --- a/src/ast_builder.rs +++ b/src/ast_builder.rs @@ -53,7 +53,16 @@ impl AstBuilder { } fn gen_decl(&mut self) -> Box { - let t = BaseType::Real; + let p: f64 = self.rng.gen(); + let ps = &self.params.statements.gen_decl; + + let t = if p < ps.gen_integer { + BaseType::Int + } else { + BaseType::Real + }; + + let v = self.gen_variable_quantified(t, Quantifier::Const); Box::new(