Add snappy compression/decompression code.

[apitrace] / trace_file.hpp

diff --git a/trace_file.hpp b/trace_file.hpp
index 9ea6bb8dd893b067754af82070f29cb1047787bd..5a460f79b0c13e1027970d89c6b281757878f389 100644 (file)
--- a/trace_file.hpp
+++ b/trace_file.hpp
@@ -2,6 +2,7 @@
 #define TRACE_FILE_HPP
 
 #include <string>
+#include <fstream>
 
 namespace Trace {
 
@@ -58,6 +59,45 @@ private:
     void *m_gzFile;
 };
 
+namespace snappy {
+    class File;
+}
+
+#define SNAPPY_CHUNK_SIZE (1 * 1024 * 1024)
+class SnappyFile : public File {
+public:
+    SnappyFile(const std::string &filename = std::string(),
+               File::Mode mode = File::Read);
+    virtual ~SnappyFile();
+
+protected:
+    virtual bool rawOpen(const std::string &filename, File::Mode mode);
+    virtual bool rawWrite(const void *buffer, int length);
+    virtual bool rawRead(void *buffer, int length);
+    virtual char rawGetc();
+    virtual void rawClose();
+    virtual void rawFlush();
+
+private:
+    inline int freeCacheSize() const
+    {
+        if (m_cacheSize > 0)
+            return m_cacheSize - (m_cachePtr - m_cache);
+        else
+            return 0;
+    }
+    void flushCache();
+    void createCache(size_t size);
+private:
+    std::fstream m_stream;
+    char *m_cache;
+    char *m_cachePtr;
+    size_t m_cacheSize;
+
+    char *m_compressedCache;
+};
+
+
 }
 
 #endif