Make debug bitstreams seekable

Author: flanglet

src/bitstream/DebugInputBitStream.hpp

@@ -19,12 +19,16 @@ limitations under the License.
 
 #include "../InputBitStream.hpp"
 #include "../OutputStream.hpp"
+#include "../Seekable.hpp"
 
 
-namespace kanzi
-{
+namespace kanzi {
 
-   class DebugInputBitStream : public InputBitStream
+#if defined(WIN32) || defined(_WIN32)
+   class DebugInputBitStream FINAL : public InputBitStream
+#else
+   class DebugInputBitStream FINAL : public InputBitStream, public Seekable
+#endif
    {
    private:
        InputBitStream& _delegate;
@@ -75,8 +79,47 @@ namespace kanzi
        void setMark(bool mark) { _mark = mark; }
 
        bool mark() const { return _mark; }
+
+#if !defined(WIN32) && !defined(_WIN32)
+       int64 tell();
+
+       bool seek(int64 pos);
+#endif
    };
 
+#if !defined(WIN32) && !defined(_WIN32)
+   // Return a position at the byte boundary
+   inline int64 DebugInputBitStream::tell()
+   {
+#ifdef NO_RTTI
+       throw std::ios_base::failure("Not supported");
+#else     
+       Seekable* seekable = dynamic_cast<Seekable*>(&_delegate);
+
+       if (seekable == nullptr)
+           throw std::ios_base::failure("The stream is not seekable");
+
+       return seekable->tell();
+#endif
+   }
+
+   // Only support a new position at the byte boundary (pos & 7 == 0)
+#ifdef NO_RTTI
+   inline bool DebugInputBitStream::seek(int64)
+   {
+       throw std::ios_base::failure("Not supported");
+#else     
+   inline bool DebugInputBitStream::seek(int64 pos)
+   {
+       Seekable* seekable = dynamic_cast<Seekable*>(&_delegate);
+
+       if (seekable == nullptr)
+           throw std::ios_base::failure("The stream is not seekable");
+
+       return seekable->seek(pos);
+#endif
+    }
+#endif
 }
 #endif
 

src/bitstream/DebugOutputBitStream.hpp

@@ -19,12 +19,16 @@ limitations under the License.
 
 #include "../OutputBitStream.hpp"
 #include "../OutputStream.hpp"
-
+#include "../Seekable.hpp"
 
 namespace kanzi
 {
 
-   class DebugOutputBitStream : public OutputBitStream
+#if defined(WIN32) || defined(_WIN32)
+   class DebugOutputBitStream FINAL : public OutputBitStream
+#else
+   class DebugOutputBitStream FINAL : public OutputBitStream, public Seekable
+#endif
    {
    private:
        OutputBitStream& _delegate;
@@ -71,8 +75,47 @@ namespace kanzi
        void setMark(bool mark) { _mark = mark; }
 
        bool mark() const { return _mark; }
+
+#if !defined(WIN32) && !defined(_WIN32)
+       int64 tell();
+
+       bool seek(int64 pos);
+#endif
    };
 
+#if !defined(WIN32) && !defined(_WIN32)                                                                                                                // Return a position at the byte boundary
+   inline int64 DebugOutputBitStream::tell()
+   {
+#ifdef NO_RTTI
+       throw std::ios_base::failure("Not supported");
+#else
+       Seekable* seekable = dynamic_cast<Seekable*>(&_delegate);
+
+       if (seekable == nullptr)
+           throw std::ios_base::failure("The stream is not seekable");
+
+       return seekable->tell();
+#endif
+   }
+
+   // Only support a new position at the byte boundary (pos & 7 == 0)
+#ifdef NO_RTTI
+   inline bool DebugOutputBitStream::seek(int64)
+   {
+       throw std::ios_base::failure("Not supported");
+#else
+   inline bool DebugOutputBitStream::seek(int64 pos)
+   {
+       Seekable* seekable = dynamic_cast<Seekable*>(&_delegate);
+
+       if (seekable == nullptr)
+           throw std::ios_base::failure("The stream is not seekable");
+
+       return seekable->seek(pos);
+#endif
+    }
+#endif
+
 }
 #endif
 

src/bitstream/DefaultInputBitStream.hpp

@@ -30,7 +30,7 @@ namespace kanzi {
 #if defined(WIN32) || defined(_WIN32)
    class DefaultInputBitStream FINAL : public InputBitStream
 #else
-   class DefaultInputBitStream FINAL : public InputBitStream, Seekable
+   class DefaultInputBitStream FINAL : public InputBitStream, public Seekable
 #endif
    {
    private:

src/bitstream/DefaultOutputBitStream.hpp

@@ -31,7 +31,7 @@ namespace kanzi
 #if defined(WIN32) || defined(_WIN32)
    class DefaultOutputBitStream FINAL : public OutputBitStream
 #else
-   class DefaultOutputBitStream FINAL : public OutputBitStream, Seekable
+   class DefaultOutputBitStream FINAL : public OutputBitStream, public Seekable
 #endif
    {
    private: