perf: Add DivideRI and DivideRR instructions

mrunix00 · Apr 24, 2024 · f509e02 · f509e02
1 parent 0b0bc06
commit f509e02
Show file tree

Hide file tree

Showing 4 changed files with 55 additions and 0 deletions.
diff --git a/src/bytecode/builtin_functions/DivideFunction.cpp b/src/bytecode/builtin_functions/DivideFunction.cpp
@@ -23,6 +23,30 @@ namespace Bytecode::BuiltinFunctions {
             instructions.push_back(new (Instruction){Instruction::Divide});
             return;
         }
+        if (args[0]->type == SyntaxTreeNode::TokenNode &&
+            args[1]->type == SyntaxTreeNode::TokenNode &&
+            segment->find_variable(((TokenNode *) args[0])->getName()) != -1) {
+            if (((TokenNode *) args[0])->token.type == Token::Symbol &&
+                ((TokenNode *) args[1])->token.type == Token::Number) {
+                instructions.push_back(new (Instruction){
+                        Instruction::DivideRI,
+                        {.ri_params = {
+                                 segment->find_variable(((TokenNode *) args[0])->getName()),
+                                 StackObject(((TokenNode *) args[1])->token)}},
+                });
+                return;
+            }
+            if (((TokenNode *) args[0])->token.type == Token::Symbol &&
+                ((TokenNode *) args[1])->token.type == Token::Symbol) {
+                instructions.push_back(new (Instruction){
+                        Instruction::DivideRR,
+                        {.rr_params = {
+                                 segment->find_variable(((TokenNode *) args[0])->getName()),
+                                 segment->find_variable(((TokenNode *) args[1])->getName())}},
+                });
+                return;
+            }
+        }
         args[0]->compile(segment, program, instructions);
         args[1]->compile(segment, program, instructions);
         instructions.push_back(new (Instruction){Instruction::Divide});

diff --git a/src/bytecode/compiler/Instruction.h b/src/bytecode/compiler/Instruction.h
@@ -20,6 +20,8 @@ namespace Bytecode {
             MultiplyRI,
             MultiplyRR,
             Divide,
+            DivideRI,
+            DivideRR,
             Increment,
             Decrement,
             DecrementR,

diff --git a/src/bytecode/vm/Interpreter.cpp b/src/bytecode/vm/Interpreter.cpp
@@ -132,6 +132,29 @@ void Bytecode::Interpreter::execute(const Program &program) {
                 }
                 vm.program_stack.push(object1.data.number / object2.data.number);
             } break;
+            case Instruction::DivideRI: {
+                const auto object2 = currentInstruction->params.ri_params.intermediate;
+                const auto object1 = vm.call_stack.getLocal(currentInstruction->params.ri_params.reg);
+                if (object1.type != ObjectType::Number) {
+                    throw SyntaxError("Invalid argument type for function \"/\", Expected number, got string");
+                }
+                if (object2.data.number == 0) {
+                    throw SyntaxError("Division by zero", 0, 0);
+                }
+                vm.program_stack.push(object1.data.number / object2.data.number);
+            } break;
+            case Instruction::DivideRR: {
+                const auto object1 = vm.call_stack.getLocal(currentInstruction->params.rr_params.reg1);
+                const auto object2 = vm.call_stack.getLocal(currentInstruction->params.rr_params.reg2);
+                if (object1.type != ObjectType::Number ||
+                    object2.type != ObjectType::Number) {
+                    throw SyntaxError("Invalid argument type for function \"/\", Expected number, got string");
+                }
+                if (object2.data.number == 0) {
+                    throw SyntaxError("Division by zero", 0, 0);
+                }
+                vm.program_stack.push(object1.data.number / object2.data.number);
+            } break;
             case Instruction::Equals: {
                 const auto object2 = vm.program_stack.pop();
                 const auto object1 = vm.program_stack.pop();

diff --git a/src/utils/dump_bytecode.cpp b/src/utils/dump_bytecode.cpp
@@ -47,6 +47,12 @@ void print_instruction(const Bytecode::Instruction &instruction) {
         case Bytecode::Instruction::Divide:
             std::cout << "Divide\n";
             break;
+        case Bytecode::Instruction::DivideRI:
+            std::cout << "DivideRI $r" << instruction.params.ri_params.reg << " " << instruction.params.ri_params.intermediate.toString() << '\n';
+            break;
+        case Bytecode::Instruction::DivideRR:
+            std::cout << "DivideRR $r" << instruction.params.rr_params.reg1 << " $r" << instruction.params.rr_params.reg2 << '\n';
+            break;
         case Bytecode::Instruction::Increment:
             std::cout << "Increment\n";
             break;