mech_interpreter/stdlib/assign/
record.rs

1#[macro_use]
2use crate::stdlib::*;
3use self::assign::*;
4
5// x.a = 1 --------------------------------------------------------------------
6
7// Record Set -----------------------------------------------------------------
8
9#[derive(Debug)]
10pub struct RecordAssign<T> {
11  pub sink: Ref<T>,
12  pub source: Ref<T>,
13}
14impl<T> MechFunctionImpl for RecordAssign<T> 
15  where
16  T: Debug + Clone + Sync + Send + PartialEq + 'static,
17  Ref<T>: ToValue
18{
19  fn solve(&self) {
20    let source_ptr = self.source.as_ptr();
21    let sink_ptr = self.sink.as_mut_ptr();
22    unsafe {
23      *sink_ptr = (*source_ptr).clone();
24    }
25  }
26  fn out(&self) -> Value { self.sink.to_value() }
27  fn to_string(&self) -> String { format!("{:#?}", self) }
28}
29#[cfg(feature = "compiler")]
30impl<T> MechFunctionCompiler for RecordAssign<T> 
31where
32  T: CompileConst + ConstElem + AsValueKind,
33{
34  fn compile(&self, ctx: &mut CompileCtx) -> MResult<Register> {
35    let name = format!("RecordAssign<{}>", T::as_value_kind());
36    compile_unop!(name, self.sink, self.source, ctx, FeatureFlag::Builtin(FeatureKind::Assign) );
37  }
38}
39
40fn impl_set_record_column_fxn(sink: Value, source: Value, key: Value) -> MResult<Box<dyn MechFunction>> {
41  match (&sink, &source, &key) {
42    (Value::Record(rcrd),source,Value::Id(k)) => {
43      let rcrd_brrw = rcrd.borrow();
44      match (rcrd_brrw.data.get(k),source) {
45        #[cfg(all(feature = "bool", feature = "record"))]
46        (Some(Value::Bool(sink)), Value::Bool(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
47        #[cfg(all(feature = "i8", feature = "record"))]
48        (Some(Value::I8(sink)), Value::I8(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
49        #[cfg(all(feature = "i16", feature = "record"))]
50        (Some(Value::I16(sink)), Value::I16(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
51        #[cfg(all(feature = "i32", feature = "record"))]
52        (Some(Value::I32(sink)), Value::I32(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
53        #[cfg(all(feature = "i64", feature = "record"))]
54        (Some(Value::I64(sink)), Value::I64(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
55        #[cfg(all(feature = "i128", feature = "record"))]
56        (Some(Value::I128(sink)), Value::I128(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
57        #[cfg(all(feature = "u8", feature = "record"))]
58        (Some(Value::U8(sink)), Value::U8(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
59        #[cfg(all(feature = "u16", feature = "record"))]
60        (Some(Value::U16(sink)), Value::U16(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
61        #[cfg(all(feature = "u32", feature = "record"))]
62        (Some(Value::U32(sink)), Value::U32(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
63        #[cfg(all(feature = "u64", feature = "record"))]
64        (Some(Value::U64(sink)), Value::U64(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
65        #[cfg(all(feature = "u128", feature = "record"))]
66        (Some(Value::U128(sink)), Value::U128(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
67        #[cfg(all(feature = "f32", feature = "record"))]
68        (Some(Value::F32(sink)), Value::F32(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
69        #[cfg(all(feature = "f64", feature = "record"))]
70        (Some(Value::F64(sink)), Value::F64(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
71        #[cfg(all(feature = "string", feature = "record"))]
72        (Some(Value::String(sink)), Value::String(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
73        #[cfg(all(feature = "complex", feature = "record"))]
74        (Some(Value::C64(sink)), Value::C64(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
75        #[cfg(all(feature = "rational", feature = "record"))]
76        (Some(Value::R64(sink)), Value::R64(source)) => return Ok(Box::new(RecordAssign{sink: sink.clone(), source: source.clone()})),
77        _ => return Err(MechError2::new(
78            UndefinedRecordFieldError { id: k.clone() },
79            None
80          ).with_compiler_loc()
81        ),
82      }
83    }
84    _ => return Err(MechError2::new(
85      UnhandledFunctionArgumentKind3 { arg: (sink.kind(), source.kind(), key.kind()), fxn_name: "record/assign-column".to_string() },
86      None
87    ).with_compiler_loc()
88    ),
89  }
90}
91
92pub struct AssignRecordColumn {}
93impl NativeFunctionCompiler for AssignRecordColumn {
94  fn compile(&self, arguments: &Vec<Value>) -> MResult<Box<dyn MechFunction>> {
95    if arguments.len() < 3 {
96      return Err(MechError2::new(IncorrectNumberOfArguments { expected: 1, found: arguments.len() }, None).with_compiler_loc());
97    }
98    let sink = arguments[0].clone();
99    let source = arguments[1].clone();
100    let key = arguments[2].clone();
101    match impl_set_record_column_fxn(sink.clone(), source.clone(), key.clone()) {
102      Ok(fxn) => Ok(fxn),
103      Err(_) => {
104        match (&sink,&source,&key) {
105          (Value::MutableReference(sink),_,_) => { impl_set_record_column_fxn(sink.borrow().clone(), source.clone(), key.clone()) }
106          x => Err(MechError2::new(
107              UnhandledFunctionArgumentKind3 { arg: (arguments[0].kind(), arguments[1].kind(), arguments[2].kind()), fxn_name: "record/assign-column".to_string() },
108              None
109            ).with_compiler_loc()
110          ),
111        }
112      }
113    }
114  }
115}
116
117
118#[derive(Debug, Clone)]
119pub struct UndefinedRecordFieldError {
120  pub id: u64,
121}
122impl MechErrorKind2 for UndefinedRecordFieldError {
123  fn name(&self) -> &str { "UndefinedRecordField" }
124  fn message(&self) -> String {
125      format!("Field {:?} is not defined in this record.", self.id)
126  }
127}
mech_interpreter/stdlib/assign/record.rs

mech_interpreter/stdlib/assign/
record.rs
