Add comment field for model, resolves hack for printing sep logic models.

1 files changed, 9 insertions, 0 deletions

diff --git a/src/theory/theory_engine.cpp b/src/theory/theory_engine.cpp
index f6894e07b..913a6800b 100644
--- a/src/theory/theory_engine.cpp
+++ b/src/theory/theory_engine.cpp
@@ -796,6 +796,15 @@ void TheoryEngine::collectModelInfo( theory::TheoryModel* m, bool fullModel ){
   }
 }
 
+void TheoryEngine::collectModelComments( theory::TheoryModel* m ){
+  for(TheoryId theoryId = theory::THEORY_FIRST; theoryId < theory::THEORY_LAST; ++theoryId) {
+    if(d_logicInfo.isTheoryEnabled(theoryId)) {
+      Trace("model-builder-debug") << "  CollectModelComments on theory: " << theoryId << endl;
+      d_theoryTable[theoryId]->collectModelComments( m );
+    }
+  }
+}
+
 /* get model */
 TheoryModel* TheoryEngine::getModel() {
   return d_curr_model;