huffman.h

Go to the documentation of this file.

// MIT License
 
// Copyright (c) 2022 ZooK
 
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
 
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.
#pragma once
 
#include <memory>
#include <b8/type.h>
#include <pipe.h>
 
namespace Huffman {
constexpr u8  Signature = 0x77;
 
class CHuffmanEncoder {
  std::shared_ptr<Pipe::CPipe> _pipe_in = std::make_shared<Pipe::CNullPipe>();
  std::shared_ptr<Pipe::CPipe> _pipe_out = std::make_shared<Pipe::CNullPipe>();
  void _Output(std::shared_ptr<Pipe::CPipe> pipe_mem_huf_packed_);
 
public:
  void SetIn(std::shared_ptr<Pipe::CPipe> pipe_in_);
 
  void SetOut(std::shared_ptr<Pipe::CPipe> pipe_out_);
 
  void Encode();
};
 
class CHuffmanDecoder {
  std::shared_ptr< Pipe::CPipe > _pipe_in  = std::make_shared< Pipe::CNullPipe >();
  std::shared_ptr< Pipe::CPipe > _pipe_out = std::make_shared< Pipe::CNullPipe >();
public:
  enum DecodeResult{
    DECODE_OK,
    DECODE_ERR_INVALID_SIGNATURE,
    DECODE_INVALID_DATA,
  };
private:
  CHuffmanDecoder::DecodeResult  _DoDecodeFlat();
 
public:
  void  SetIn ( std::shared_ptr< Pipe::CPipe > pipe_in_  );
 
  void  SetOut( std::shared_ptr< Pipe::CPipe > pipe_out_ );
 
  CHuffmanDecoder::DecodeResult  Decode();
};
 
} // namespace Huffman