Better overload validation

agui2/src/Algorithm/Algorithm.cpp

@@ -15,6 +15,10 @@ Algorithm::Algorithm(std::string fullName)
 }
 uint8_t Algorithm::getInputCount() const {
+    if (this->overloads.empty()) {
+        throw std::runtime_error { std::string("Input count has not yet been determined for algorithm ") + this->fullName };
+    }
+
     return static_cast<uint8_t>(this->overloads[0].parameterTypes.size());
 }
@@ -26,6 +30,24 @@ void Algorithm::addOverload(Algorithm::Overload overload) {
     this->overloads.push_back(overload);
 }
+bool Algorithm::canAddOverload(const Algorithm::Overload& overload) {
+    for (const auto& paramType: overload.parameterTypes) {
+        if (paramType != "automaton::NFA<object::Object, object::Object>" && paramType != "automaton::DFA<object::Object, object::Object>") {
+            return false;
+        }
+    }
+
+    if (overload.returnType != "automaton::NFA<object::Object, object::Object>" && overload.returnType != "automaton::DFA<object::Object, object::Object>") {
+        return false;
+    }
+
+    if (this->overloads.empty()) {
+        return true;
+    }
+
+    return this->overloads.back().parameterTypes.size() == overload.parameterTypes.size();
+}
+
 std::string Algorithm::getPrettyName() const {
     std::string eps = u8"\u03B5";
     static std::map<std::string, std::string> prettyNames = {

agui2/src/Algorithm/Algorithm.hpp

@@ -22,6 +22,7 @@ public:
 private:
     void addOverload(Overload overload);
+    bool canAddOverload(const Overload& overload);
     std::string name;
     std::string fullName;

agui2/src/Algorithm/Registry.cpp

@@ -18,9 +18,18 @@ void Registry::initialize() {
                     paramNames.push_back(std::get<0>(param));
                 }
-                algorithm->addOverload({ resultType, paramNames });
+                Algorithm::Overload _overload { resultType, paramNames };
+
+                if (!algorithm->canAddOverload(_overload)) {
+                    continue;
+                }
+
+                algorithm->addOverload(_overload);
             }
+            if (algorithm->overloads.empty())
+                continue;
+
             in.algorithms[name] = std::move(algorithm);
         }
     }