From 45ed2cd201c79fae1c61a4f6f2982a9f5cfbceca Mon Sep 17 00:00:00 2001 From: troido Date: Thu, 6 Feb 2020 21:20:09 +0100 Subject: grass now also loaded from json --- src/componentparameter.rs | 27 +++++++++++++++++++++++++-- 1 file changed, 25 insertions(+), 2 deletions(-) (limited to 'src/componentparameter.rs') diff --git a/src/componentparameter.rs b/src/componentparameter.rs index a2f7b70..491486a 100644 --- a/src/componentparameter.rs +++ b/src/componentparameter.rs @@ -1,12 +1,14 @@ use std::collections::HashMap; +use rand::Rng; use serde_json::Value; use crate::parameter::{Parameter, ParameterType}; #[derive(Debug, PartialEq)] pub enum ComponentParameter { Constant(Parameter), - Argument(String) + Argument(String), + Random(Vec) } impl ComponentParameter { @@ -18,6 +20,10 @@ impl ComponentParameter { Self::Argument(argname) => { Some(arguments.get(argname.as_str())?.clone()) } + Self::Random(options) => { + let r = rand::thread_rng().gen_range(0, options.len()); + options[r].evaluate(arguments) + } } } @@ -32,6 +38,14 @@ impl ComponentParameter { let argname = paramvalue.as_str().ok_or("argument parameter not a string")?.to_string(); Ok(Self::Argument(argname)) }, + "random" => { + let optionvalues = paramvalue.as_array().ok_or("random argument not a a string")?; + let mut options = Vec::new(); + for option in optionvalues { + options.push(Self::from_json(option)?) + } + Ok(Self::Random(options)) + }, _ => Err("unknown compparam type") } } @@ -40,7 +54,16 @@ impl ComponentParameter { pub fn get_type(&self, arguments: &Vec<(String, ParameterType, Option)>) -> Result{ Ok(match self { Self::Constant(param) => param.paramtype(), - Self::Argument(argname) => arguments.iter().find(|(n, _t, _d)| n == argname).ok_or("unknown argument name")?.1 + Self::Argument(argname) => arguments.iter().find(|(n, _t, _d)| n == argname).ok_or("unknown argument name")?.1, + Self::Random(options) => { + let typ: ParameterType = options.get(0).ok_or("random has no options")?.get_type(arguments)?; + for param in options { + if param.get_type(arguments)? != typ { + return Err("inconsistent parameter types"); + } + } + typ + } }) } } -- cgit