bz_compress.cpp

Go to the documentation of this file.

/*
 * Compresses data in Bz2 format
 *
 * without parameters reads data from standard input and writes compressed data to standard output
 * with 1 parameter reads data from the file given and writes compressed data to standard output
 * with 2 parameters reads data from the file in the 1st argument and writes compressed data to the file in the 2nd argument
 *
 */

#include <xstream/bz.h>

#include <iostream>
#include <fstream>

using namespace std;;
using namespace xstream;;


int main(int argc, char* argv[]){

   const size_t len = 4*1024;
   char buf[len];

   istream* readfrom;
   ostream* writeto;

   try{
      /* Binary flags mandatory on win32 systems, otherwise files will be corrupt */
      if(1<argc){
         readfrom = new ifstream(argv[1], ios::binary);
         if(2<argc){
            writeto = new ofstream(argv[2], ios::binary);
         }else{
            writeto = &cout;
         }
      }else{
         readfrom = &cin;
         writeto  = &cout;
      }

      bz::ostreambuf bz_o(writeto->rdbuf());
      ostream bzout(&bz_o);

      //raise exceptions
      bzout.exceptions(ios::badbit);

      while(readfrom->good()){
         readfrom->read(buf,len);
         bzout.write(buf,readfrom->gcount());
      }

      bzout.flush();

   }
   catch(exception& e){
      cerr<<"Error: "<<e.what()<<endl;
   }

   //pointers not being freed

   return 0;
}