Optimize performance by using FILE* instead of ofstream for outputgen

1 files changed, 16 insertions, 0 deletions

diff --git a/src/textstream.h b/src/textstream.h
index cee4cb7..a9c4a31 100644
--- a/src/textstream.h
+++ b/src/textstream.h
@@ -69,6 +69,12 @@ class TextStream final
       m_s = s;
     }
 
+    void setFile(FILE *f)
+    {
+      flush();
+      m_f = f;
+    }
+
     /** Returns the attached std::ostream object.
      *  @see setStream()
      */
@@ -77,6 +83,11 @@ class TextStream final
       return m_s;
     }
 
+    FILE *file() const
+    {
+      return m_f;
+    }
+
     /** Adds a character to the stream */
     TextStream &operator<<( char c)
     {
@@ -165,6 +176,10 @@ class TextStream final
       {
         m_s->write(m_buffer.c_str(),m_buffer.length());
       }
+      else if (m_f)
+      {
+        fwrite(m_buffer.c_str(),1,m_buffer.length(),m_f);
+      }
       m_buffer.clear();
     }
 
@@ -230,6 +245,7 @@ class TextStream final
     }
     std::string m_buffer;
     std::ostream *m_s = nullptr;
+    FILE *m_f = nullptr;
 };
 
 #endif