blitzfortranio.h

Go to the documentation of this file.

// include guard
#ifndef TF_BLITZFORTRANIO_H_VERSION

#define TF_BLITZFORTRANIO_H_VERSION \
  "TF_BLITZFORTRANIO_H   V1.0   "

#include<fstream>
#include<blitz/array.h>
#include<tfxx/fortranio.h>
#include<tfxx/misc.h>
#include<tfxx/error.h>

namespace tfxx {
namespace fortranio {

template<class T, int N>
struct blitz_magic {
}; // blitz_magic

template<> struct blitz_magic<float, 1> { static const char magic[]="ZZf1"; };
#ifndef DOXYGEN_MUST_SKIP_THIS
template<> struct blitz_magic<float, 2> { static const char magic[]="ZZf2"; };
template<> struct blitz_magic<float, 3> { static const char magic[]="ZZf3"; };
template<> struct blitz_magic<float, 4> { static const char magic[]="ZZf4"; };
template<> struct blitz_magic<float, 5> { static const char magic[]="ZZf5"; };
template<> struct blitz_magic<double, 1> { static const char magic[]="ZZd1"; };
template<> struct blitz_magic<double, 2> { static const char magic[]="ZZd2"; };
template<> struct blitz_magic<double, 3> { static const char magic[]="ZZd3"; };
template<> struct blitz_magic<double, 4> { static const char magic[]="ZZd4"; };
template<> struct blitz_magic<double, 5> { static const char magic[]="ZZd5"; };
template<> struct blitz_magic<int, 1> { static const char magic[]="ZZi1"; };
template<> struct blitz_magic<int, 2> { static const char magic[]="ZZi2"; };
template<> struct blitz_magic<int, 3> { static const char magic[]="ZZi3"; };
template<> struct blitz_magic<int, 4> { static const char magic[]="ZZi4"; };
template<> struct blitz_magic<int, 5> { static const char magic[]="ZZi5"; };
#ifdef TF_COMPLEX_ARRAY
#include<complex>
template<> struct blitz_magic<std::complex<float>, 1> 
{ static const char magic[]="ZZc1"; };
template<> struct blitz_magic<std::complex<float>, 2> 
{ static const char magic[]="ZZc2"; };
template<> struct blitz_magic<std::complex<float>, 3> 
{ static const char magic[]="ZZc3"; };
template<> struct blitz_magic<std::complex<float>, 4> 
{ static const char magic[]="ZZc4"; };
template<> struct blitz_magic<std::complex<float>, 5> 
{ static const char magic[]="ZZc5"; };
template<> struct blitz_magic<std::complex<double>, 1> 
{ static const char magic[]="ZZz1"; };
template<> struct blitz_magic<std::complex<double>, 2> 
{ static const char magic[]="ZZz2"; };
template<> struct blitz_magic<std::complex<double>, 3> 
{ static const char magic[]="ZZz3"; };
template<> struct blitz_magic<std::complex<double>, 4> 
{ static const char magic[]="ZZz4"; };
template<> struct blitz_magic<std::complex<double>, 5> 
{ static const char magic[]="ZZz5"; };
#endif
#endif

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

template<class T, int N>
  tfxx::fortranio::FortranBinOutput& 
    operator << (tfxx::fortranio::FortranBinOutput& fo,
                 const blitz::Array<T, N>& a)
{
  fo.write_magic(blitz_magic<T,N>::magic);
  fo.put(N);
  fo.end_block();
  for (int i=0; i<N; i++) fo.put(a.lbound(i));
  fo.end_block();
  for (int i=0; i<N; i++) fo.put(a.extent(i));
  fo.end_block();
  for (blitz::Array<T, N>::const_iterator i=a.begin(); i!=a.end(); ++i)
  { fo.put(*i); }
  fo.end_block();
  return(fo);
}

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

template<class T, int N>
  tfxx::fortranio::FortranBinInput& 
    operator >> (tfxx::fortranio::FortranBinInput& fi, 
                 blitz::Array<T, N>& a)
{
  TFXX_assert((fi.match_magic(blitz_magic<T,N>::magic)),
    "ERROR (blitz array operator>>): input has no matching magic number!");
  int n;
  fi.get(n);
  TFXX_assert((N==n), "ERROR (blitz array operator>>): illegal rank!");
  blitz::TinyVector<int, N> lbound, extent;
  for (int i=0; i<N; i++) fi.get(lbound[i]);
  for (int i=0; i<N; i++) fi.get(extent[i]);
  blitz::Array<T, N> result(lbound, extent, blitz::fortranArray);
  for (blitz::Array<T, N>::iterator i=result.begin(); i!=result.end(); ++i)
  { fi.get(*i); }
  a.reference(result);
  return(fi);
}

  
} // namespace fortranio
} // namespace tfxx

#endif // TF_BLITZFORTRANIO_H_VERSION (includeguard)

/* ----- END OF blitzfortranio.h ----- */