fortranio.h

Go to the documentation of this file.

// include guard
#ifndef TF_FORTRANIO_H_VERSION

#define TF_FORTRANIO_H_VERSION \
  "TF_   V1.5   "

#include<tfxx/bytesex.h>
#include<fstream>
#include<sstream>

namespace tfxx {

namespace fortranio {

class FortranBinInput {
  public:
    FortranBinInput(std::istream& is, const bool& swap=false):
      Mistream(is), Mswap(swap), Mnbytes(0), Mnremain(0) { }
    ~FortranBinInput() { this->finish_block(); }
    bool more() const 
    { return((Mnbytes>0) || Mistream.good()); }
    template<typename T> void get(T& value);
    bool match_magic(const char* cmagic);

  private:
    char extract_next_char();
    void extract_chars(char* buf, const int& n);
    void read_block_size();
    void finish_block();

    std::istream& Mistream;
    bool Mswap;
    long int Mnbytes;
    long int Mnremain;
};

class FortranBinOutput {
  public:
    FortranBinOutput(std::ostream& os):
      Mostream(os), Mbuffer(std::string()) { }
    ~FortranBinOutput() { this->end_block(); }
    template<typename T>
      void put(const T& value);
    void end_block();
    void write_magic(const char* cmagic);

  private:
    void put_char(const char& c);
    void put_chars(const char* buf, const int& n);

    std::ostream& Mostream;
    std::ostringstream Mbuffer;
};

/*----------------------------------------------------------------------*/
// definition of member function templates
template<typename T> 
  void FortranBinOutput::put(const T& value)
{
  tfxx::ioswap::IOUnion<T> buf;
  buf.value=value;
  put_chars(buf.bytes, sizeof(T));
}

template<typename T> 
  void FortranBinInput::get(T& value)
{
  tfxx::ioswap::IOUnion<T> buf;
  extract_chars(buf.bytes, sizeof(T));
  value=buf.value;
  if (Mswap) { value=tfxx::ioswap::swap(buf.value); }
}

/*----------------------------------------------------------------------*/

template<typename T>
  tfxx::fortranio::FortranBinOutput& 
    operator << (tfxx::fortranio::FortranBinOutput& fo, const T& value)
{
  fo.put(value);
  return(fo);
}

template<typename T>
  tfxx::fortranio::FortranBinInput& 
    operator >> (tfxx::fortranio::FortranBinInput& fi, T& value)
{
  fi.get(value);
  return(fi);
}

} // namespace fortranio

} // namespace tfxx

#endif // TF_FORTRANIO_H_VERSION (includeguard)

/* ----- END OF fortranio.h ----- */