hack_blitztogrepg.cc

Go to the documentation of this file.

#define TF_HACK_BLITZTOGREPG_CC_VERSION \
  "TF_HACK_BLITZTOGREPG_CC   V1.0   "

#include <fstream>
#include <tfxx/misc.h>
#include <tfxx/fortranio.h>
#include <tfxx/complexio.h>
#include <tfxx/blitzutil.h>
#include <tfxx/hack_blitztogrepg.h>

namespace tfxx {
namespace hacks {

void write_grepg_c(const blitz::Array<std::complex<float>, 2>& array,
                   const char* filename,
                   const double& dx=1., const double& dy=1.)
{
  std::ofstream os(filename);
  tfxx::ioswap::file_magic_write(os, "1234", true);
  tfxx::fortranio::FortranBinOutput fos(os);
  int nom=array.extent(blitz::firstDim);
  int fom=array.lbound(blitz::firstDim);
  int nslo=array.extent(blitz::secondDim);
  int fslo=array.lbound(blitz::secondDim);
  fos << nom << nslo;
  fos.end_block();
  for (int i=0; i<nom; i++) 
  { fos << float(3.14159265358979311*2.*(i+fom-1)*dx); }
  for (int i=0; i<nslo; i++) { fos << float(1.e-3*(i+fslo-1)*dy); }
  fos.end_block();
  for (int i=0; i<nslo; i++)
  {
    for (int j=0; j<nom; j++)
    {
      fos << array(j+fom,i+fslo);
    }
  }
  fos.end_block();
}

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

void write_grepg_f(const blitz::Array<float, 2>& array,
                   const char* filename,
                   const double& dx=1., const double& dy=1.)
{
  // create intermediate complex data
  blitz::Array<std::complex<float>, 2> tmp(blitz::fortranArray);
  tfxx::blitzutil::setToSameDomain(tmp, array.domain());
  tmp=blitz::cast<std::complex<float> >(array);
  write_grepg_c(tmp, filename, dx, dy);
}

} // namespace hacks
} // namespace tfxx

/* ----- END OF hack_blitztogrepg.cc ----- */