llvmir some opt

midend/llvm/IrModule.java

@@ -0,0 +1,100 @@
+package midend.llvm;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+
+import midend.llvm.constant.IrConstantStr;
+import midend.llvm.value.IrFuncValue;
+import midend.llvm.value.IrGlobalValue;
+import midend.llvm.instr.GetIntInstr;
+import midend.llvm.instr.PutChInstr;
+import midend.llvm.instr.PutIntInstr;
+import midend.llvm.instr.PutStrInstr;
+
+public class IrModule {
+    private ArrayList<String> decls;
+    private HashMap<String, IrConstantStr> strs;
+    private ArrayList<IrGlobalValue> globalVars;
+    private ArrayList<IrFuncValue> funcs;
+
+    public IrModule() {
+        decls = new ArrayList<>();
+        strs = new HashMap<>();
+        globalVars = new ArrayList<>();
+        funcs = new ArrayList<>();
+        decls.add(GetIntInstr.getIntDecl());
+        decls.add(PutChInstr.putChDecl());
+        decls.add(PutIntInstr.putIntDecl());
+        decls.add(PutStrInstr.putStrDecl());
+    }
+
+    public ArrayList<String> getDecls() {
+        return decls;
+    }
+
+    public HashMap<String, IrConstantStr> getStrs() {
+        return strs;
+    }
+
+    public ArrayList<IrGlobalValue> getGlobalVars() {
+        return globalVars;
+    }
+
+    public ArrayList<IrFuncValue> getFuncs() {
+        return funcs;
+    }
+
+    public void addFunc(IrFuncValue func) {
+        funcs.add(func);
+    }
+
+    public void addStr(IrConstantStr str) {
+        if (strs.containsKey(str.getValue())) {
+            return;
+        }
+        strs.put(str.getValue(), str);
+    }
+
+    public boolean containStr(String str) {
+        return strs.containsKey(str);
+    }
+
+    public IrConstantStr getStr(String str) {
+        if (containStr(str)) {
+            return strs.get(str);
+        }
+        return null;
+    }
+
+    public void addGlobalVar(IrGlobalValue globalVar) {
+        globalVars.add(globalVar);
+    }
+
+    public IrFuncValue getMainFunc() {
+        for (IrFuncValue func : funcs) {
+            if (func.isMain()) {
+                return func;
+            }
+        }
+        return null;
+    }
+
+    //TODO: toString()方法编写
+    public String toString() {
+        StringBuilder sb = new StringBuilder();
+        for (String decl : decls) {
+            sb.append(decl).append("\n");
+        }
+        for (IrConstantStr str : strs.values()) {
+            sb.append(str.toString()).append("\n");
+        }
+        for (IrGlobalValue globalVar : globalVars) {
+            sb.append(globalVar.toString()).append("\n");
+        }
+        for (IrFuncValue func : funcs) {
+            sb.append(func.toString()).append("\n");
+        }
+        // System.out.println(funcs.size());
+        return sb.toString();
+    }
+}