Only emit include for IOSystem.h when needed

souffle-lang · Dec 16, 2022 · 251e8fd · 251e8fd
1 parent 09b7a26
commit 251e8fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/synthesiser/Synthesiser.cpp b/src/synthesiser/Synthesiser.cpp
@@ -334,6 +334,8 @@ void Synthesiser::emitCode(std::ostream& out, const Statement& stmt) {
         void visit_(type_identity<IO>, const IO& io, std::ostream& out) override {
             PRINT_BEGIN_COMMENT(out);
 
+            synthesiser.currentClass->addInclude("\"souffle/io/IOSystem.h\"", true);
+
             // print directives as C++ initializers
             auto printDirectives = [&](const std::map<std::string, std::string>& registry) {
                 auto cur = registry.begin();
@@ -2648,7 +2650,6 @@ void Synthesiser::generateCode(GenDb& db, const std::string& id, bool& withShare
 
         gen.addInclude("\"souffle/SouffleInterface.h\"");
         gen.addInclude("\"souffle/SignalHandler.h\"");
-        gen.addInclude("\"souffle/io/IOSystem.h\"", true);
 
         GenFunction& constructor = gen.addConstructor(Visibility::Public);