If function, not switching value on condition change

Author: remaininlight

compiler/src/codegen/functions/if_function.rs

@@ -0,0 +1,56 @@
+use super::{Function, FunctionContext, VarArgs};
+use codegen::values::{NumValue};
+use inkwell::values::PointerValue;
+use inkwell::{FloatPredicate};
+//use inkwell::IntPredicate;
+use mir::block;
+
+
+use codegen::{intrinsics, util};
+use std::f32::consts;
+
+pub struct IfFunction {}
+impl Function for IfFunction {
+    fn function_type() -> block::Function { block::Function::If }
+
+    fn gen_call(
+        func: &mut FunctionContext,
+        args: &[PointerValue],
+        _varargs: Option<VarArgs>,
+        result: PointerValue,
+    ) {
+
+        let result_num = NumValue::new(result);
+
+        let condition_num = NumValue::new(args[0]);
+        let condition_vec = condition_num.get_vec(func.ctx.b);
+        let left_index = func.ctx.context.i32_type().const_int(0, false);
+        let condition_left = func
+            .ctx
+            .b
+            .build_extract_element(&condition_vec, &left_index, "condition.left")
+            .into_float_value();
+        let condition_bool = func.ctx.b.build_float_compare(
+            FloatPredicate::OEQ,
+            condition_left,
+            func.ctx.context.f32_type().const_float(1.0),
+            "conditionbool",
+        );
+
+        let then_num = NumValue::new(args[1]);
+        let then_vec = then_num.get_vec(func.ctx.b);
+        let then_form = then_num.get_form(func.ctx.b);
+
+        let alternative_num = NumValue::new(args[1]);
+        let alternative_vec = alternative_num.get_vec(func.ctx.b);
+
+        let left_index = func.ctx.context.i32_type().const_int(0, false);
+        let result_vec = func.ctx.b.build_select(condition_bool,
+                                                 then_vec,
+                                                 alternative_vec,
+                                                 "resultvec").into_vector_value();
+
+        result_num.set_form(func.ctx.b, &then_form);
+        result_num.set_vec(func.ctx.b, &result_vec);
+    }
+}

compiler/src/codegen/functions/mod.rs

@@ -3,6 +3,7 @@ mod channel_function;
 mod defer_function;
 mod delay_function;
 mod function_context;
+mod if_function;
 mod indexed_function;
 mod note_function;
 mod num_function;
@@ -33,6 +34,7 @@ pub use self::biquad_filter_function::*;
 pub use self::channel_function::*;
 pub use self::defer_function::*;
 pub use self::delay_function::*;
+pub use self::if_function::*;
 pub use self::indexed_function::*;
 pub use self::note_function::*;
 pub use self::num_function::*;
@@ -161,7 +163,8 @@ map_functions! {
     Note => NoteFunction,
     Voices => VoicesFunction,
     Channel => ChannelFunction,
-    Indexed => IndexedFunction
+    Indexed => IndexedFunction,
+    If => IfFunction
 }
 
 fn get_lifecycle_func(

compiler/src/mir/block/function.rs

@@ -51,7 +51,7 @@ macro_rules! define_functions {
             }
         }
 
-        pub const FUNCTION_TABLE: [&str; 51] = [$($str_name, )*];
+        pub const FUNCTION_TABLE: [&str; 52] = [$($str_name, )*];
     );
 }
 
@@ -106,7 +106,8 @@ define_functions! {
     Note = "note" func![(Midi) -> Tuple(vec![Num, Num, Num, Num])],
     Voices = "voices" func![(Midi, VarType::new_array(Num)) -> VarType::new_array(Midi)],
     Channel = "channel" func![(Midi, Num) -> Midi],
-    Indexed = "indexed" func![(Num) -> VarType::new_array(Num)]
+    Indexed = "indexed" func![(Num) -> VarType::new_array(Num)],
+    If = "if" func![(Num, Num, Num) -> Num]
 }
 
 #[derive(Debug, Clone)]