mosesdecoder/moses/TranslationModel/UG/generic/file_io/ug_stream.h

// -*- c++ -*-
// (c) 2006,2007,2008 Ulrich Germann
// makes opening files a little more convenient

#ifndef __UG_STREAM_HH
#define __UG_STREAM_HH
#include <boost/iostreams/device/file.hpp>
#include <boost/iostreams/categories.hpp> // input_filter_tag
#include <boost/iostreams/operations.hpp> // get, WOULD_BLOCK
#include <boost/iostreams/copy.hpp>       // get, WOULD_BLOCK
#include <boost/iostreams/filtering_stream.hpp>
#include <boost/iostreams/filter/bzip2.hpp>
#include <boost/iostreams/filter/gzip.hpp>

#include <iostream>
#include <fstream>
#include <string>

namespace ugdiss
{

  /** open input file that is possibly compressed
   *  decompression filters are automatically added based on the file name
   *  gzip for .gz; bzip2 for bz2.
   */
  boost::iostreams::filtering_istream* 
  open_input_stream(std::string fname);

  void open_input_stream(std::string fname, 
			 boost::iostreams::filtering_istream& in);

  boost::iostreams::filtering_ostream* 
  open_output_stream(std::string fname);

  void open_output_stream(std::string fname, 
			  boost::iostreams::filtering_ostream& in);

}
#endif