Generate a call to _divide_by_zero_error on RiscV

1 files changed, 22 insertions, 11 deletions

diff --git a/lib/Language/Elna/RiscV/CodeGenerator.hs b/lib/Language/Elna/RiscV/CodeGenerator.hs
index 6e5a92f..4be7dfd 100644
--- a/lib/Language/Elna/RiscV/CodeGenerator.hs
+++ b/lib/Language/Elna/RiscV/CodeGenerator.hs
@@ -234,20 +234,31 @@ quadruple (DivisionQuadruple operand1 operand2 (Store register))
     , IntOperand immediateOperand2 <- operand2 =
         let statements2 = lui immediateOperand2 register
             Store operandRegister1 = variableOperand1
-        in pure $ Vector.snoc statements2
-            $ Instruction
-            $ RiscV.BaseInstruction RiscV.Op
-            $ RiscV.R register RiscV.DIV operandRegister1 register
-            $ RiscV.Funct7 0b0000001
+            operationInstruction
+                | immediateOperand2 == 0 =
+                    RiscV.CallInstruction "_divide_by_zero_error"
+                | otherwise = RiscV.BaseInstruction RiscV.Op
+                    $ RiscV.R register RiscV.DIV operandRegister1 register
+                    $ RiscV.Funct7 0b0000001
+         in pure $ Vector.snoc statements2
+            $ Instruction operationInstruction
     | IntOperand immediateOperand1 <- operand1
-    , VariableOperand variableOperand2 <- operand2 =
+    , VariableOperand variableOperand2 <- operand2 = do
         let statements1 = lui immediateOperand1 register
             Store operandRegister2 = variableOperand2
-        in pure $ Vector.snoc statements1
-            $ Instruction
-            $ RiscV.BaseInstruction RiscV.Op
-            $ RiscV.R register RiscV.DIV register operandRegister2
-            $ RiscV.Funct7 0b0000001
+            divisionInstruction = Instruction
+                $ RiscV.BaseInstruction RiscV.Op
+                $ RiscV.R register RiscV.DIV register operandRegister2 (RiscV.Funct7 0b0000001)
+        branchLabel <- createLabel
+        let branchInstruction = Instruction
+                $ RiscV.RelocatableInstruction RiscV.Branch
+                $ RiscV.RBranch branchLabel RiscV.BNE RiscV.Zero operandRegister2
+        pure $ mappend statements1 $ Vector.fromList 
+            [ branchInstruction
+            , Instruction (RiscV.CallInstruction "_divide_by_zero_error")
+            , JumpLabel branchLabel []
+            , divisionInstruction
+            ]
 quadruple (LabelQuadruple (Label label)) = pure $ Vector.singleton $ JumpLabel label mempty
 quadruple (GoToQuadruple label) = pure $ Vector.singleton $ unconditionalJal label
 quadruple (EqualQuadruple operand1 operand2 goToLabel)