CompStream.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
/* Local Variables:  */
/* mode: c++         */
/* End:              */
 
#ifndef COMPSTREAM_H
#define COMPSTREAM_H
 
 
#include <istream>
#include <fstream>
#include <ostream>
#include <string>
#include <boost/iostreams/filtering_streambuf.hpp>
 
namespace o2
{
namespace io
{
 
enum class CompressionMethod {
  None,
  Gzip,
  Zlib,
  Bzip2,
  Lzma
};
 
class icomp_stream : public std::istream
{
 public:
  using streambuffer_t = boost::iostreams::filtering_streambuf<boost::iostreams::input>;
 
  icomp_stream(std::string filename, CompressionMethod method = CompressionMethod::None);
 
  icomp_stream(std::istream& backend, CompressionMethod method = CompressionMethod::None);
 
  icomp_stream(std::string filename, std::string method);
 
  icomp_stream(std::istream& backend, std::string method);
 
 private:
  streambuffer_t mStreamBuffer;
  std::ifstream mInputFile;
};
 
class ocomp_stream : public std::ostream
{
 public:
  using streambuffer_t = boost::iostreams::filtering_streambuf<boost::iostreams::output>;
 
  ocomp_stream(std::string filename, CompressionMethod method = CompressionMethod::None);
 
  ocomp_stream(std::ostream& backend, CompressionMethod method = CompressionMethod::None);
 
  ocomp_stream(std::string filename, std::string method);
 
  ocomp_stream(std::ostream& backend, std::string method);
 
 private:
  streambuffer_t mStreamBuffer;
};
 
} // namespace io
} // namespace o2
#endif // COMPSTREAM_H