toLLVM en visitor, todo : réparer la Declaration

2025-04-05 20:11:52 +02:00
parent 3f0ca04b19
commit b9b609f1e2
6 changed files with 159 additions and 116 deletions
--- a/src/main/java/TP2/asd/toLLVM_Visitor.java
+++ b/src/main/java/TP2/asd/toLLVM_Visitor.java
@@ -0,0 +1,118 @@
+package TP2.asd;
+
+import java.util.ArrayList;
+
+import TP2.asd.Interface.*;
+import TP2.asd.Program.*;
+import TP2.asd.SymTable.*;
+import TP2.llvm.Interface.*;
+import TP2.llvm.ProgramLLVM;
+import TP2.llvm.ProgramLLVM.*;
+
+public class toLLVM_Visitor implements  ProgramVisitor<SymTable,ProgramLLVMImpl>,
+                                        FunctionVisitor<SymTable,DefineLLVM>,
+                                        InstrVisitor<SymTable,ArrayList<InstructionLLVM>>,
+                                        ExprVisitor<SymTable,ArrayList<AssignLVMImp>>,
+                                        TypeVisitor<SymTable,TypeLLVM>
+                                        {
+
+
+    //PROGRAM
+
+    @Override
+    public ProgramLLVMImpl visitProgram(ProgramImp prog, SymTable h) {
+        ArrayList<DefineLLVM> fonctionLLVM = new ArrayList<>();
+        for(int i = 0; i<prog.fonctions().size(); i++){
+            fonctionLLVM.add(prog.fonctions().get(i).accept(this, h));
+        }
+        return new ProgramLLVMImpl(new ArrayList<>(),fonctionLLVM);
+    }
+
+    //FUNCTION
+
+    @Override
+    public DefineLLVM visitFunction(FunctionImp fun, SymTable h) {
+        ArrayList<InstructionLLVM> instrLLVM = new ArrayList<>();
+        for(int i = 0; i<fun.instructions().size(); i++){
+            instrLLVM.addAll(fun.instructions().get(i).accept(this, h));
+        }
+        return new DefineLLVMImpl(fun.nom(), fun.type().accept(this, h), instrLLVM);
+    }
+
+    //INSTRUCTION 
+
+    @Override
+    public ArrayList<InstructionLLVM> visitReturn(Return_instrImp instr, SymTable h) {
+        ArrayList<AssignLVMImp> list = instr.e().accept(this,h);
+        InstructionLLVM r = new ReturnLLVMImp(new IntLLVMImpl()/*TODO*/,list.getLast().getVar());
+        ArrayList<InstructionLLVM> result = new ArrayList<>();
+        result.addAll(list);
+        result.add(r);
+
+        return result;
+    }
+
+    @Override
+    public ArrayList<InstructionLLVM> visitAssign(AssignImp instr, SymTable h) {
+        ArrayList<AssignLVMImp> list = instr.e().accept(this,h);
+        InstructionLLVM r = new AssignLVMImp(new VarLLVMImpl(new IntLLVMImpl()/*TODO*/,instr.t()),list.getLast().getVar());
+        ArrayList<InstructionLLVM> result = new ArrayList<>();
+        result.addAll(list);
+        result.add(r);
+        return result;
+    }
+
+    @Override
+    public ArrayList<InstructionLLVM> visitDeclaration(DeclarationImp instr, SymTable h) {
+        ArrayList<InstructionLLVM> list = new ArrayList<>();
+        for(int i = 0; i<instr.s().size();i++){
+            TypeLLVM t2 = instr.t().accept(this,h);
+            String name = instr.s().get(i);
+            list.add(new AssignLVMImp(new VarLLVMImpl(t2, name),new allocaLLVMImpl(t2)));
+        }
+        return list;
+    }
+
+    @Override
+    public ArrayList<AssignLVMImp> visitConst(ConstImp e, SymTable h) {
+        throw new UnsupportedOperationException("Can't call toLLVM on Expression Const");
+    }
+
+    @Override
+    public ArrayList<AssignLVMImp> visitBinOp(BinopExpressionImp e, SymTable h) {
+        ArrayList<AssignLVMImp> list = new ArrayList<>();
+            ValLLVM val1 = null;
+            ValLLVM val2 = null;
+            if(e.e1() instanceof Val){
+                val1 = ((Val) e.e1()).getValLLVM();
+            }
+            else {
+                ArrayList<AssignLVMImp> eLLVM1 = e.e1().accept(this,h);
+                list.addAll(eLLVM1);
+                val1 = eLLVM1.getLast().getVar();
+            }
+            if(e.e2() instanceof Val){
+                val2 = ((Val) e.e2()).getValLLVM();
+            }
+            else {
+                ArrayList<AssignLVMImp> eLLVM2 = e.e2().accept(this,h);
+                list.addAll(eLLVM2);
+                val2 = eLLVM2.getLast().getVar();
+            }
+
+            list.add(new AssignLVMImp(new VarLLVMImpl(new IntLLVMImpl()/*TODO*/,"todo"), new BinOpLLVMImp(new IntLLVMImpl()/*TODO*/,e.op(),val1,val2)));
+
+            return list;
+    }
+
+    @Override
+    public TypeLLVM visitInt(Type_intImp t, SymTable h) {
+        return new IntLLVMImpl();
+    }
+
+    @Override
+    public TypeLLVM visitVoid(Type_voidImp t, SymTable h) {
+        return new VoidLLVMImpl();
+    }
+
+}