From c4231c6faf4e1b4650b075c641b0bb8c053739e4 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Patrick=20Sch=C3=B6nberger?= <patrick.schoenberger@posteo.de>
Date: Thu, 12 Aug 2021 09:26:17 +0200
Subject: [PATCH] add generic.h (how did i miss that...)

---
 src/generic.h | 101 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 src/generic.h

diff --git a/src/generic.h b/src/generic.h
new file mode 100644
index 0000000..df203b3
--- /dev/null
+++ b/src/generic.h
@@ -0,0 +1,101 @@
+#pragma once
+
+#include "repr.h"
+#include "typeInfo.h"
+#include "visit.h"
+
+void addGenericInstantiation(
+  std::vector<std::vector<Type>> & insts,
+  const std::vector<Type> & newInst)
+{
+  if (insts.empty())
+  {
+    insts.push_back(newInst);
+    return;
+  }
+  for (auto inst : insts)
+  {
+    for (int i = 0; i < inst.size(); i++)
+    {
+      if (inst[i] != newInst[i])
+      {
+        insts.push_back(newInst);
+        return;
+      }
+    }
+  }
+}
+
+Program instantiateGenerics(const Program & p)
+{
+  Program result = p;
+
+  // Find generic instantiations
+
+  Visitor findGenericInstantiations;
+  findGenericInstantiations.onExpr =
+  [&](const Expr & e, const std::shared_ptr<Context> ctx)
+  {
+    if (e.type == ExprType::Func && !e._func.genericInstantiation.empty())
+    {
+      auto f = findFunctionPtr(e._func.functionName, e._func.namespacePrefixes, ctx);
+      if (f.has_value())
+      {
+        if (std::get<0>(*f)->genericTypeNames.empty())
+          throw "Trying to instantiate non-generic function";
+        if (e._func.genericInstantiation.size() != std::get<0>(*f)->genericTypeNames.size())
+          throw "Trying to instantiate function with wrong number of types";
+        addGenericInstantiation(std::get<0>(*f)->genericInstantiations, e._func.genericInstantiation);
+      }
+    }
+    // TODO: generic methods
+  };
+  findGenericInstantiations.onType =
+  [&](const Type & t, const std::shared_ptr<Context> ctx)
+  {
+    if (!t.genericInstantiation.empty())
+    {
+      auto s = findStructPtr(t.name, t.namespacePrefixes, ctx);
+      if (s.has_value())
+      {
+        if (std::get<0>(*s)->genericTypeNames.empty())
+          throw "Trying to instantiate non-generic struct";
+        if (t.genericInstantiation.size() != std::get<0>(*s)->genericTypeNames.size())
+          throw "Trying to instantiate struct with wrong number of types";
+        addGenericInstantiation(std::get<0>(*s)->genericInstantiations, t.genericInstantiation);
+      }
+    }
+  };
+  Visit v(findGenericInstantiations);
+  v.visit(result);
+
+  return result;
+}
+
+std::string genericAppendix(const std::vector<Type> & ts)
+{
+  std::stringstream sstr;
+  for (auto t : ts)
+  {
+    sstr << "_";
+    sstr << t.name;
+    for (auto m : t.modifiers)
+    {
+      if (m.type == TypeModifierType::Array)
+      {
+        sstr << "_arr";
+        if (m._staticArray)
+          sstr << m._arraySize;
+      }
+      else if (m.type == TypeModifierType::Pointer)
+      {
+        sstr << "_ptr";
+      }
+    }
+    if (!t.genericInstantiation.empty())
+    {
+      sstr << genericAppendix(t.genericInstantiation);
+    }
+  }
+  return sstr.str();
+}
\ No newline at end of file
-- 
2.50.1