particlesampler.cpp

Go to the documentation of this file.

#include <particlesampler.h>

particlesampler::particlesampler
(
  uintc & state_,
  uintc samplecountermax_,
  uintc filecountermax_,
  histogram<double> const hist_,
  string const filename_,
  particle const * & pi_,
  uintc & numParticles_
  ) : 
  state(state_), 
  samplecounter(0), 
  filecounter(0),
  hist(hist_),
  filename(filename_),
  pi(pi_), 
  numParticles(numParticles_), 
  samplecountermax(samplecountermax_), 
  filecountermax(filecountermax_)
{
  assert(state<4);
}

void particlesampler::writefile()
{
  ofstream f(filename.c_str());
  //assert(!f);
  hist.printfrequencydistpair(f);
  //hist.frequencydistprintxy(f);
}

void particlesampler::sample()
{
  if (!state)
    return;

  samplecounter=0;
  for(uint i=0; i<numParticles; ++i)
  {
    double x(pi[i].vel[0]);
    double y(pi[i].vel[1]);
    switch(state)
    {
      case 1: 
      hist.eval( sqrt( x*x+y*y ) );
      break;

      case 2:
      hist.eval( x );
      break;

      case 3:
      hist.eval( y );
      break;
    };
  };

  ++filecounter;
  if ((filecounter%filecountermax)==0)
    writefile();
}


void particlesampler::reset()
{
  assert(!filename.empty());

  samplecounter=0;

  assert(samplecountermax!=0);
}

void particlesampler::operator ++ ()
{ 
  ++samplecounter; 
  if ((samplecounter%samplecountermax)==0) 
    sample(); 
}