add internal params representations

souffle-lang · Jul 22, 2020 · 16756bb · 16756bb
1 parent f943254
commit 16756bb
Showing 1 changed file with 61 additions and 0 deletions.
diff --git a/src/ast/transform/IOAttributesTransformer.h b/src/ast/transform/IOAttributesTransformer.h
@@ -46,10 +46,40 @@ class IOAttributesTransformer : public AstTransformer {
 
         changed |= setAttributeNames(translationUnit);
         changed |= setAttributeTypes(translationUnit);
+        changed |= setAttributeParams(translationUnit);
 
         return changed;
     }
 
+    bool setAttributeParams(AstTranslationUnit& translationUnit) {
+        bool changed = false;
+        AstProgram* program = translationUnit.getProgram();
+        auto auxArityAnalysis = translationUnit.getAnalysis<AuxiliaryArity>();
+
+        for (AstIO* io : program->getIOs()) {
+            AstRelation* rel = getRelation(*translationUnit.getProgram(), io->getQualifiedName());
+            // Prepare type system information.
+            std::vector<std::string> attributesParams;
+
+            for (const auto* attribute : rel->getAttributes()) {
+                attributesParams.push_back(attribute->getName());
+            }
+
+            // Casting due to json11.h type requirements.
+            long long arity{static_cast<long long>(rel->getArity() - auxArityAnalysis->getArity(rel))};
+            long long auxArity{static_cast<long long>(auxArityAnalysis->getArity(rel))};
+
+            Json relJson = Json::object{{"arity", arity}, {"auxArity", auxArity},
+                    {"params", Json::array(attributesParams.begin(), attributesParams.end())}};
+
+            Json params = Json::object{{"relation", relJson}, {"records", getRecordsParams(translationUnit)}};
+
+            io->addDirective("params", params.dump());
+            changed = true;
+        }
+        return changed;
+    }
+
     bool setAttributeNames(AstTranslationUnit& translationUnit) {
         bool changed = false;
         AstProgram* program = translationUnit.getProgram();
@@ -148,6 +178,37 @@ class IOAttributesTransformer : public AstTransformer {
         ramRecordTypes = Json(records);
         return ramRecordTypes;
     }
+
+    Json getRecordsParams(AstTranslationUnit& translationUnit) const {
+        static Json ramRecordParams;
+        // Check if the types where already constructed
+        if (!ramRecordParams.is_null()) {
+            return ramRecordParams;
+        }
+
+        AstProgram* program = translationUnit.getProgram();
+        std::vector<std::string> elementParams;
+        std::map<std::string, Json> records;
+
+        // Iterate over all record types in the program populating the records map.
+        for (auto* astType : program->getTypes()) {
+            if (isA<AstRecordType>(astType)) {
+                elementParams.clear();
+
+                for (const auto field : as<AstRecordType>(astType)->getFields()) {
+                    elementParams.push_back(field->getName());
+                }
+                const size_t recordArity = elementParams.size();
+                Json recordInfo = Json::object{
+                        {"types", std::move(elementParams)}, {"arity", static_cast<long long>(recordArity)}};
+                records.emplace(astType->getQualifiedName().toString(), std::move(recordInfo));
+            }
+        }
+
+        ramRecordParams = Json(records);
+        return ramRecordParams;
+    }
+
 };
 
 }  // namespace souffle