setup structure for graph coloring

include/opt/confgraph.h

@@ -0,0 +1,21 @@
+#pragma once
+#include "IR/IR_basic.h"
+#include "cfg.h"
+class ConfGraphNode {
+ public:
+  std::string var_name;
+  size_t color;
+  bool is_move_related;
+  bool move_related_but_frozen;
+  bool is_binded_with_physical_reg;
+  std::vector<ConfGraphNode *> neighbors;
+  ConfGraphNode(std::string var_name) : var_name(var_name) {}
+};
+class ConfGraph {
+ public:
+  std::unordered_map<std::string, ConfGraphNode *> name_to_node;
+  std::vector<std::shared_ptr<ConfGraphNode>> nodes;
+};
+
+ConfGraph BuildConfGraph(CFGType &cfg);
+bool TryColoring(std::shared_ptr<FunctionDefItem> src, CFGType &cfg, ConfGraph& confgraph);

include/opt/regalloc.h

@@ -1,5 +1,6 @@
 #pragma once
 #include "liveanalysis.h"
 #include "phieliminate.h"
+#include "confgraph.h"
 
 std::shared_ptr<ModuleItem> RegAlloc(std::shared_ptr<ModuleItem> src);

src/opt/confgraph.cpp

@@ -0,0 +1,6 @@
+#include "confgraph.h"
+#include <unordered_map>
+#include "cfg.h"
+
+ConfGraph BuildConfGraph(CFGType &cfg) { ; }
+bool TryColoring(std::shared_ptr<FunctionDefItem> src, CFGType &cfg, ConfGraph& confgraph) { return false; }

src/opt/regalloc.cpp

@@ -1,5 +1,7 @@
 #include "regalloc.h"
 #include "IR/IR_basic.h"
+#include "cfg.h"
+#include "confgraph.h"
 #include "liveanalysis.h"
 #include "phieliminate.h"
 #include "tools.h"
@@ -125,17 +127,22 @@ void EnforcePhysicalRegs(CFGType &cfg) {
     }
   }
 }
-void ConductRegAllocForFunction([[maybe_unused]] std::shared_ptr<FunctionDefItem> func, CFGType &cfg) {
+void ConductRegAllocForFunction(std::shared_ptr<FunctionDefItem> func) {
-  EnforcePhysicalRegs(cfg);
+  CFGType cfg;
-  LiveAnalysis(cfg);
+  ConfGraph confgraph;
+  do {
+    cfg = BuildCFGForFunction(func);
+    EnforcePhysicalRegs(cfg);
+    LiveAnalysis(cfg);
+    confgraph = BuildConfGraph(cfg);
+  } while (TryColoring(func, cfg, confgraph));
 }
 
 std::shared_ptr<ModuleItem> RegAlloc(std::shared_ptr<ModuleItem> src) {
   auto res = src;
   for (auto &func : res->function_defs) {
     // func = std::make_shared<FunctionDefItem>(*func);
-    auto cfg = BuildCFGForFunction(func);
+    ConductRegAllocForFunction(func);
-    ConductRegAllocForFunction(func, cfg);
   }
   return res;
 }