test_FftwReal2dDftData.cxx File Reference

#include <string>
#include <iostream>
#include <iomanip>
#include <fstream>
#include <vector>
#include "dunecore/DuneInterface/Data/FftwReal2dDftData.h"
#include <cassert>

Go to the source code of this file.

Typedefs
using	DftData = FftwDouble2dDftData
using	Index = DftData::Index
using	IndexArray = DftData::IndexArray
using	IndexArrayVector = DftData::IndexArrayVector
using	Complex = DftData::Complex
using	Float = DftData::Float

Functions
Index	realval (Index idat)
Index	imagval (Index idat)
int	test_FftwReal2dDftData (Index nsam0, Index nsam1)
int	main (int argc, char *argv[])

Typedef Documentation

using Complex = DftData::Complex

Definition at line 28 of file test_FftwReal2dDftData.cxx.

using DftData = FftwDouble2dDftData

Definition at line 24 of file test_FftwReal2dDftData.cxx.

using Float = DftData::Float

Definition at line 29 of file test_FftwReal2dDftData.cxx.

using Index = DftData::Index

Definition at line 25 of file test_FftwReal2dDftData.cxx.

using IndexArray = DftData::IndexArray

Definition at line 26 of file test_FftwReal2dDftData.cxx.

using IndexArrayVector = DftData::IndexArrayVector

Definition at line 27 of file test_FftwReal2dDftData.cxx.

Function Documentation

Index imagval

(

Index

idat

)

Definition at line 32 of file test_FftwReal2dDftData.cxx.

{ return 2*idat + 2; }

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

Definition at line 155 of file test_FftwReal2dDftData.cxx.

                                 {
  if ( argc > 1 ) {
    cout << "Usage: " << argv[0] << " [ARG]" << endl;
    return 0;
  }
  Index stat = 0;
  stat += test_FftwReal2dDftData(3, 4);
  stat += test_FftwReal2dDftData(4, 3);
  return stat;
}

Index realval

(

Index

idat

)

Definition at line 31 of file test_FftwReal2dDftData.cxx.

{ return 2*idat + 1; }

int test_FftwReal2dDftData	(	Index	nsam0,
		Index	nsam1
	)

Definition at line 36 of file test_FftwReal2dDftData.cxx.

                                                     {
  const string myname = "test_FftwReal2dDftData: ";
#ifdef NDEBUG
  cout << myname << "NDEBUG must be off." << endl;
  abort();
#endif
  string line = myname + "-----------------------------";

  cout << line << endl;
  cout << myname << "Sample sizes: " << nsam0 << ", " << nsam1 << endl;

  cout << line << endl;
  cout << myname << "Checking index array." << endl;
  IndexArray nsams({nsam0, nsam1});
  assert( nsams[0] == nsam0 );
  assert( nsams[1] == nsam1 );
  Index nComplexExp = nsam0*(nsam1/2 + 1);
  Index ndftComplex = DftData::dftComplexDataSize(nsams);
  cout << myname << "  Complex size: " << ndftComplex << endl;
  Index ndftFloat = DftData::dftFloatDataSize(nsams);
  cout << myname << "    Float size: " << ndftFloat << endl;
  assert( ndftComplex == nComplexExp );
  assert( ndftFloat == 2*nComplexExp );

  cout << line << endl;
  cout << myname << "Construct DFT object." << endl;
  DftData::Norm norm(22);
  DftData dft(norm, nsams);
  assert( dft.rank() == 2 );
  assert( dft.normalization() == norm );
  assert( dft.size() == ndftComplex );
  assert( dft.nSamples(0) == nsam0 );
  assert( dft.size(0) == nsam0 );
  assert( dft.nSamples(1) == nsam1 );
  assert( dft.size(1) == nsam1 );

  cout << myname << "Check indexing." << endl;
  Index cntij = 0;
  for ( Index idat=0; idat<nComplexExp; ++idat ) {
    IndexArrayVector idxas = dft.indexArrays(idat);
    cout << myname << setw(3) << idat << ": ";
    Index nerr = idxas.size() == 0;
    bool first = true;
    for ( IndexArray idxa : idxas ) {
      Index jdat = dft.globalIndex(idxa);
      if ( first ) first = false;
      else cout << ", ";
      cout << "[" << setw(3) << idxa[0] << "," << setw(3) << idxa[1] << "]";
      if ( jdat != idat ) cout << " --> " << jdat << " (!)";
      if ( jdat != idat ) ++nerr;
      ++cntij;
    }
    cout << endl;
    assert( nerr == 0 );
  }
  assert( cntij == nsam0*nsam1 );

  cout << line << endl;
  cout << myname << "Create test data." << endl;
  vector<Complex> datc(nComplexExp);
  vector<Float> datf(2*nComplexExp);
  Index iflt = 0;
  Index idat;
  for ( idat=0; idat<nComplexExp; ++idat ) {
    IndexArrayVector arrs = dft.indexArrays(idat);
    Float re = realval(idat);
    Float im = imagval(idat);
    datf[iflt++] = re;
    datf[iflt++] = im;
    Complex cval(re, im);
    datc[idat] = cval;
  }
  assert( idat == datc.size() );
  assert( iflt == datf.size() );

  cout << line << endl;
  cout << myname << "Copy data in." << endl;
  assert( dft.copyIn(nsams, datc) == 0 );
  for ( Index idat=0; idat<nComplexExp; ++idat ) {
    IndexArrayVector idxas = dft.indexArrays(idat);
    assert( idxas.size() );
    Complex cval = dft.value(idat);
    Index ire = cval.real() + 0.1;
    Index iim = cval.imag() + 0.1;
    cout << myname << setw(3) << idat << ": ["
         << setw(3) << idxas[0][0] << "," << setw(3) << idxas[0][1] << "]: "
         << "(" << setw(3) << ire << "," << setw(3) << iim << ")"
         << endl;
  }

  cout << line << endl;
  cout << myname << "Copy object." << endl;
  DftData dft2(dft);
  assert( dft2.rank() == 2 );
  assert( dft2.normalization() == norm );
  assert( dft2.size() == ndftComplex );
  assert( dft2.nSamples(0) == nsam0 );
  assert( dft2.size(0) == nsam0 );
  assert( dft2.nSamples(1) == nsam1 );
  assert( dft2.size(1) == nsam1 );
  for ( Index idat=0; idat<nComplexExp; ++idat ) {
    IndexArrayVector idxas = dft.indexArrays(idat);
    assert( idxas.size() );
    Complex cval = dft2.value(idat);
    Index ire = cval.real() + 0.1;
    Index iim = cval.imag() + 0.1;
    cout << myname << setw(3) << idat << ": ["
         << setw(3) << idxas[0][0] << "," << setw(3) << idxas[0][1] << "]: "
         << "(" << setw(3) << ire << "," << setw(3) << iim << ")"
         << endl;
  }

  cout << line << endl;
  cout << myname << "All tests passed." << endl;
  return 0;
}