compressor.h

/*
 *      Copyright (C) 2016 Jean-Luc Barriere
 *
 *  This library is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published
 *  by the Free Software Foundation; either version 3, or (at your option)
 *  any later version.
 *
 *  This library is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 *  GNU Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with this library; see the file COPYING.  If not, write to
 *  the Free Software Foundation, 51 Franklin Street, Fifth Floor, Boston,
 *  MA 02110-1301 USA
 *  http://www.gnu.org/copyleft/gpl.html
 *
 */
 
#ifndef COMPRESSOR_H
#define COMPRESSOR_H
 
#include "local_config.h"
 
#include <cstddef> // for size_t
#include <cstring> // for memcpy
 
namespace NSROOT
{
 
  class Compressor
  {
  public:
    typedef int(*STREAM_READER)(void *handle, void *buf, int sz);
 
    Compressor(STREAM_READER reader, void *handle, bool gzip, int level = -1);
    Compressor(const char *input, size_t len, bool gzip, int level = -1);
    virtual ~Compressor();

    bool HasOutputData() { return !m_stop; }

    bool IsCompleted();

    bool HasBufferError();

    bool HasStreamError();

    size_t ReadOutput(char *buf, size_t len);

    size_t FetchOutput(const char **data);
 
  private:
    int m_status;
    int m_flush;
    bool m_stop;
    size_t m_chunk_size;
 
    const enum
    {
      MEM_BUFFER,
      FCB_READER
    } m_type_in;
 
    size_t m_input_len;
    const char *m_input;
 
    STREAM_READER m_rstream;
    void *m_rstream_hdl;
    char *m_rstream_buf;
 
    char *m_output;
    size_t m_output_pos;
    size_t m_output_len;
 
    void *_opaque;
 
    static int _init(void *zp, void *out, size_t len, int wbits, int level);
    size_t NextChunk();
  };
 
 
  class Decompressor
  {
  public:
    typedef int(*STREAM_READER)(void *handle, void *buf, int sz);
 
    Decompressor(STREAM_READER reader, void *handle, bool gzip);
    Decompressor(const char *input, size_t len, bool gzip);
    virtual ~Decompressor();

    bool HasOutputData() { return !m_stop; }

    bool IsCompleted();

    bool HasBufferError();

    bool HasStreamError();

    size_t ReadOutput(char *buf, size_t len);

    size_t FetchOutput(const char **data);
 
  private:
    int m_status;
    bool m_stop;
    size_t m_chunk_size;
 
    const enum
    {
      MEM_BUFFER,
      FCB_READER
    } m_type_in;
 
    size_t m_input_len;
    const char *m_input;
 
    STREAM_READER m_rstream;
    void *m_rstream_hdl;
    char *m_rstream_buf;
 
    char *m_output;
    size_t m_output_pos;
    size_t m_output_len;
 
    void *_opaque;
 
    static int _init(void *zp, void *out, size_t len, int wbits);
    size_t NextChunk();
  };
 
}
 
#endif /* COMPRESSOR_H */