TAO/tao/Compression/bzip2/Bzip2Compressor.cpp

   1 #include "Bzip2Compressor.h"
   2 #include "bzlib.h"
   3
   4 TAO_BEGIN_VERSIONED_NAMESPACE_DECL
   5
   6 namespace TAO
   7 {
   8 Bzip2Compressor::Bzip2Compressor (
   9   ::Compression::CompressorFactory_ptr compressor_factory,
  10   ::Compression::CompressionLevel compression_level) :
  11     BaseCompressor (compressor_factory, compression_level)
  12 {
  13 }
  14
  15 void
  16 Bzip2Compressor::compress (
  17     const ::Compression::Buffer & source,
  18     ::Compression::Buffer & target
  19   )
  20 {
  21   // Ensure maximum is at least a bit bigger than input length.
  22   target.length (static_cast <CORBA::ULong> ((source.length () * 1.01) + 600));
  23
  24   unsigned int max_length = static_cast <unsigned int> ( target.maximum () );
  25
  26   int const retval = ::BZ2_bzBuffToBuffCompress (reinterpret_cast <char*>(target.get_buffer ()),
  27                                   &max_length,
  28                                   reinterpret_cast <char*>(const_cast<CORBA::Octet*>(source.get_buffer ())),
  29                                   source.length (),
  30                                   9,
  31                                   1,
  32                                   this->compression_level () * 25);
  33
  34   if (retval != BZ_OK)
  35     {
  36       throw ::Compression::CompressionException (retval, "");
  37     }
  38   else
  39     {
  40       target.length (static_cast <CORBA::ULong> (max_length));
  41     }
  42
  43   // Update statistics for this compressor
  44   this->update_stats (source.length (), target.length ());
  45 }
  46
  47 void
  48 Bzip2Compressor::decompress (
  49   const ::Compression::Buffer & source,
  50   ::Compression::Buffer & target)
  51 {
  52   unsigned int max_length = static_cast <unsigned int> (target.maximum ());
  53   // todo, check 0,1 values
  54   int const retval = ::BZ2_bzBuffToBuffDecompress (reinterpret_cast <char*>(target.get_buffer ()),
  55                                  &max_length,
  56                                  reinterpret_cast <char*>(const_cast<CORBA::Octet*>(source.get_buffer ())),
  57                                  source.length (),
  58                                  0,
  59                                  1);
  60
  61   if (retval != BZ_OK)
  62     {
  63       throw ::Compression::CompressionException (retval, "");
  64     }
  65   else
  66     {
  67       target.length (static_cast  <CORBA::ULong> (max_length));
  68     }
  69 }
  70 }
  71
  72 TAO_END_VERSIONED_NAMESPACE_DECL