Extract ordering functionality into definition group.

code/compiler/12/definition.cpp

@@ -5,6 +5,7 @@
 #include "llvm_context.hpp"
 #include "type.hpp"
 #include "type_env.hpp"
+#include "graph.hpp"
 #include <llvm/IR/DerivedTypes.h>
 #include <llvm/IR/Function.h>
 #include <llvm/IR/Type.h>
@@ -107,3 +108,58 @@ void definition_data::generate_llvm(llvm_context& ctx) {
         ctx.builder.CreateRetVoid();
     }
 }
+
+void definition_group::find_free(type_mgr& mgr, type_env_ptr& env, std::set<std::string>& into) {
+    this->env = type_scope(env);
+
+    for(auto& def_pair : defs_defn) {
+        def_pair.second->find_free(mgr, env);
+        std::set<std::string> local_dependencies;
+        for(auto& free_var : def_pair.second->free_variables) {
+            if(defs_defn.find(free_var) == defs_defn.end()) {
+                into.insert(free_var);
+            } else {
+                local_dependencies.insert(free_var);
+            }
+        }
+        std::swap(def_pair.second->free_variables, local_dependencies);
+    }
+}
+
+void definition_group::typecheck(type_mgr& mgr) {
+    for(auto& def_data : defs_data) {
+        def_data.second->insert_types(env);
+    }
+    for(auto& def_data : defs_data) {
+        def_data.second->insert_constructors();
+    }
+
+    function_graph dependency_graph;
+
+    for(auto& def_defn : defs_defn) {
+        def_defn.second->find_free(mgr, env);
+        dependency_graph.add_function(def_defn.second->name);
+
+        for(auto& dependency : def_defn.second->free_variables) {
+            if(defs_defn.find(dependency) == defs_defn.end())
+                throw 0;
+            dependency_graph.add_edge(def_defn.second->name, dependency);
+        }
+    }
+
+    std::vector<group_ptr> groups = dependency_graph.compute_order();
+    for(auto it = groups.rbegin(); it != groups.rend(); it++) {
+        auto& group = *it;
+        for(auto& def_defnn_name : group->members) {
+            auto& def_defn = defs_defn.find(def_defnn_name)->second;
+            def_defn->insert_types(mgr);
+        }
+        for(auto& def_defnn_name : group->members) {
+            auto& def_defn = defs_defn.find(def_defnn_name)->second;
+            def_defn->typecheck(mgr);
+        }
+        for(auto& def_defnn_name : group->members) {
+            env->generalize(def_defnn_name, mgr);
+        }
+    }
+}