feat: added mod operator

Author: Zffu

compiler/astoir_mir/src/builder.rs

@@ -121,6 +121,16 @@ pub fn build_int_div(ctx: &mut MIRContext, left: MIRIntValue, right: MIRIntValue
 	return res.as_int();
 }
 
+pub fn build_int_mod(ctx: &mut MIRContext, left: MIRIntValue, right: MIRIntValue, signed: bool) -> DiagnosticResult<MIRIntValue> {
+	if left.size != right.size {
+		unsure_panic!("Tried using imod on different sized integers");
+	}
+
+	let res = ctx.append_inst(MIRInstruction::IntegerMod { signed, left, right }).get()?;
+
+	return res.as_int();
+}
+
 pub fn build_int_shift_left(ctx: &mut MIRContext, left: MIRIntValue, shift: MIRIntValue) -> DiagnosticResult<MIRIntValue> {
 	let res = ctx.append_inst(MIRInstruction::ShiftLeft { a: left, shift }).get()?;
 
@@ -181,6 +191,15 @@ pub fn build_float_div(ctx: &mut MIRContext, left: MIRFloatValue, right: MIRFloa
 	return res.as_float();
 }
 
+pub fn build_float_mod(ctx: &mut MIRContext, left: MIRFloatValue, right: MIRFloatValue, signed: bool) -> DiagnosticResult<MIRFloatValue> {
+	if left.size != right.size {
+		unsure_panic!("Tried using fmod on different sized integers");
+	}
+
+	let res = ctx.append_inst(MIRInstruction::FloatMod { signed, left, right }).get()?;
+
+	return res.as_float();
+}
 
 pub fn build_float_neg(ctx: &mut MIRContext, val: MIRFloatValue) -> DiagnosticResult<MIRFloatValue> {
 	let res = ctx.append_inst(MIRInstruction::FloatNeg { val }).get()?;

compiler/astoir_mir/src/insts/mod.rs

@@ -34,6 +34,7 @@ pub enum MIRInstruction {
 	FloatSub { signed: bool, left: MIRFloatValue, right: MIRFloatValue }, 
 	FloatMul { signed: bool, left: MIRFloatValue, right: MIRFloatValue }, 
 	FloatDiv { signed: bool, left: MIRFloatValue, right: MIRFloatValue },
+	FloatMod { signed: bool, left: MIRFloatValue, right: MIRFloatValue },
 	FloatNeg { val: MIRFloatValue }, 
 
 	// Bitwise (int typed)
@@ -230,6 +231,7 @@ impl Display for MIRInstruction {
 			Self::FloatSub { signed, left, right } => writeln!(f, "fsub s{} {} {}", signed, left, right)?,
 			Self::FloatMul { signed, left, right } => writeln!(f, "fmul s{} {} {}", signed, left, right)?,
 			Self::FloatDiv { signed, left, right } => writeln!(f, "fdiv s{} {} {}", signed, left, right)?,
+			Self::FloatMod { signed, left, right } => writeln!(f, "fmod s{} {} {}", signed, left, right)?,
 			Self::FloatNeg { val } => writeln!(f, "fneg {}", val)?,
 
 			Self::BitwiseAnd { a, b } => writeln!(f, "and {} {}", a, b)?,

compiler/astoir_mir_lowering/src/math.rs

@@ -1,5 +1,5 @@
 use astoir_hir::nodes::{HIRNode, HIRNodeKind};
-use astoir_mir::{blocks::refer::MIRBlockReference, builder::{build_float_add, build_float_div, build_float_mul, build_float_sub, build_int_add, build_int_div, build_int_mul, build_int_sub, build_shift_left, build_shift_right}, vals::base::BaseMIRValue};
+use astoir_mir::{blocks::refer::MIRBlockReference, builder::{build_float_add, build_float_div, build_float_mod, build_float_mul, build_float_sub, build_int_add, build_int_div, build_int_mod, build_int_mul, build_int_sub, build_shift_left, build_shift_right}, vals::base::BaseMIRValue};
 use compiler_typing::raw::RawType;
 use compiler_utils::operators::{MathOperator, MathOperatorType};
 use diagnostics::{DiagnosticResult, builders::{make_math_operation_req_assign, make_req_type_kind}, unsure_panic};
@@ -55,7 +55,8 @@ pub fn lower_hir_math_operation_int(left: BaseMIRValue, right: BaseMIRValue, ope
 		MathOperatorType::Multiply => build_int_mul(&mut ctx.mir_ctx, left, right, signed)?,
 		MathOperatorType::Divide => build_int_div(&mut ctx.mir_ctx, left, right, signed)?,
 		MathOperatorType::ShiftLeft => build_shift_left(&mut ctx.mir_ctx, left, right)?,
-		MathOperatorType::ShiftRight => build_shift_right(&mut ctx.mir_ctx, left, right)?
+		MathOperatorType::ShiftRight => build_shift_right(&mut ctx.mir_ctx, left, right)?,
+		MathOperatorType::Modulo => build_int_mod(&mut ctx.mir_ctx, left, right, signed)?
 	};
 
 	return Ok(res.into());
@@ -72,6 +73,7 @@ pub fn lower_hir_math_operation_float(left: BaseMIRValue, right: BaseMIRValue, o
 		MathOperatorType::Subtract => build_float_sub(&mut ctx.mir_ctx, left, right, signed)?,
 		MathOperatorType::Multiply => build_float_mul(&mut ctx.mir_ctx, left, right, signed)?,
 		MathOperatorType::Divide => build_float_div(&mut ctx.mir_ctx, left, right, signed)?,
+		MathOperatorType::Modulo => build_float_mod(&mut ctx.mir_ctx, left, right, signed)?,
 
 		_ => return Err(make_req_type_kind(node, &"integer".to_string()).into())
 	};