integration< D, F > Class Template Reference

#include <integration.h>

Collaboration diagram for integration< D, F >:

List of all members.

Public Member Functions
void	LJcalc ()
	integration ()
void	reset (int argc, char **argv)
void	draw ()
void	bruteforce ()
void	bruteforce (uintc *p, uintc sz)
void	togglestartstop ()
void	step ()
void	steplarge ()
void	uniformgridtest ()
Public Attributes
long	stepcount
gltextmsg	stepcountmsg
uint	numParticles
bool	uniform
bool	printstate
float	h
bool	displayEk
bool	LJ
	Lennard-Jones potential is enabled with brute force only.
double	LJpsi
double	LJsigma
double	histlow
double	histhigh
uint	histn
particlesampler	sampler
uint	samplerstate
particle *	pi
uint *	pu
framerate	fr
bool	run

---

Detailed Description

template<typename D, typename F>
class integration< D, F >

Definition at line 21 of file integration.h.

---

Constructor & Destructor Documentation

template<typename D , typename F >

integration< D, F >::integration

(

)

[inline]

Definition at line 489 of file integration.h.

References integration< D, F >::displayEk, integration< D, F >::h, integration< D, F >::LJ, integration< D, F >::LJpsi, integration< D, F >::LJsigma, integration< D, F >::numParticles, integration< D, F >::pi, integration< D, F >::printstate, integration< D, F >::run, and integration< D, F >::uniform.

  : pd(gluNewQuadric(),10,3),
  distribution 
  (
    0.02,  //radius
    5.0,  //vmax
    boxcollision(0.0,10.0,0.0,10.0)  //box dimensions
  ),
  fn
  (
    distribution.box
  ),
  stepcount(0),
  stepcountmsg("0",0.5,-0.8,0.0,0.0,1.0),
  histlow(0.0),
  histhigh(7.0),
  histn(81),
  sampler
  (
    (uintc &)samplerstate,
    (uintc)20,     // Sample every x steps all particle speeds
    (uintc)5,      // Inner loop: sample every x2 steps
    histogram<double>
    (
      histlow,   // Lowest speed
      histhigh, // Highest speed
      histn      // Intervals
    ),
    string("data.txt"),
    (particle const * &)pi,
    (uintc & )numParticles
  )
/*
  (
    samplerstate,
    (uintc)20,     // Sample every x steps all particle speeds
    (uintc)5,      // Inner loop: sample every x2 steps
    histogram<double>
    (
      histlow,   // Lowest speed
      histhigh, // Highest speed
      histn      // Intervals
    ),
    string("data.txt"),
    (particle const *)pi,
    numParticles
  )
*/
{
  uniform=false;
  printstate=false;
  pi=0;
  numParticles=10;
  h=0.01;
  //zpr=true;
  run=false;
  displayEk=false;
  LJ=false;
  LJsigma=1.0;
  LJpsi=1.0;
}

---

Member Function Documentation

template<typename D , typename F >

void integration< D, F >::bruteforce	(	uintc *	p,
		uintc	sz
	)			[inline]

Definition at line 308 of file integration.h.

References boxcollision::walls().

{
  boxcollision const & box(distribution.box);

//cout << SHOW(sz) "  ";
//print(p,p+sz);


  uint a;
  uint b;
  for (uint i=0; i+1<sz; ++i)
  {
    for (uint k=i+1; k<sz; ++k)
    {
      a = x[i];
      b = x[k];

      if (pi[a].isIntersecting(pi[b]))
      {
        pi[a].step(-h);
        pi[b].step(-h);

        pi[a].collisionDynamics(pi[b]);

        pi[a].step(h);
        pi[b].step(h);

        box.walls(pi[a],h);
        box.walls(pi[b],h);
      }
    }
  }
}

template<typename D , typename F >

void integration< D, F >::bruteforce

(

)

[inline]

Definition at line 347 of file integration.h.

References particle::collisionDynamics(), integration< D, F >::h, integration< D, F >::LJ, integration< D, F >::LJcalc(), integration< D, F >::numParticles, integration< D, F >::pi, particle::step(), and boxcollision::walls().

Referenced by integration< D, F >::step(), test02(), and integration< D, F >::uniformgridtest().

{
  boxcollision const & box(distribution.box);
  for (uint i=0; i<numParticles; ++i)
  {
    pi[i].step(h);
    box.walls(pi[i],h);
  }

  if (LJ)
    LJcalc();
  
  for (uint i=0; i+1<numParticles; ++i)
  {
    for (uint k=i+1; k<numParticles; ++k)
    {

      if (pi[i].isIntersecting(pi[k]))
      {
        pi[i].step(-h);
        pi[k].step(-h);

        pi[i].collisionDynamics(pi[k]);

        pi[i].step(h);
        pi[k].step(h);

if (!LJ)
{
        box.walls(pi[i],h);
        box.walls(pi[k],h);
}
      }
    }
  }

    

#ifdef DEBUG_PART
if (false)
{
cout << endl;
for (uint i=0; i<numParticles; ++i)
  cout << pi[i] << endl;
}
#endif

}

template<typename D , typename F >

void integration< D, F >::draw

(

)

[inline]

Definition at line 398 of file integration.h.

References gltextmsg::display(), framerate< period >::display(), integration< D, F >::displayEk, particledisp::draw(), integration< D, F >::fr, integration< D, F >::numParticles, integration< D, F >::pi, integration< D, F >::stepcountmsg, and particle::vel.

Referenced by display().

{
  // Display the particles

  glPushAttrib(GL_CURRENT_BIT);
  glPushAttrib(GL_LIGHTING_BIT);
  glColor3f(1.0,0.0,0.0);
  pd.draw(pi[0]);
  glPopAttrib();
  glPopAttrib();

  for (uint i=1; i+1<numParticles; ++i)
  {
    if ((i%10)!=0)
      pd.draw(pi[i]);
    else
    {
      glPushAttrib(GL_CURRENT_BIT);
      glPushAttrib(GL_LIGHTING_BIT);
      glColor3f(0.1,0.2,1.0);
      pd.draw(pi[i]);
      glPopAttrib();
      glPopAttrib();
    }
  }

  glPushAttrib(GL_CURRENT_BIT);
  glPushAttrib(GL_LIGHTING_BIT);
  glColor3f(0.0,1.0,0.0);
  pd.draw(pi[numParticles-1]);
  glPopAttrib();
  glPopAttrib();


  // Verify energy in system.
  if (displayEk)
  {

    static int ekcounter;
    ++ekcounter;
    if ((ekcounter%10)==0)
    {

      double sum(0.0);
      double energy;
      for (uint i=0; i<numParticles; ++i)
      {
        particle const & p(pi[i]);
        energy = p.vel[0] * p.vel[0] + p.vel[1] * p.vel[1];
     
        sum += energy;
      }
      cout << "Ek*2/m=" << sum << endl;
    }
  }

  // Draw Box
  glPushAttrib(GL_CURRENT_BIT);
  glPushAttrib(GL_LIGHTING_BIT);
  glDisable(GL_LIGHTING);
  glColor3f(.6,.03,.03);
  glBegin(GL_LINES);

  doublec x0 = distribution.box.x0;
  doublec x1 = distribution.box.x1;
  doublec y0 = distribution.box.y0;
  doublec y1 = distribution.box.y1;

  glVertex3f(x0,y0,0.0);
  glVertex3f(x1,y0,0.0);
  glVertex3f(x1,y0,0.0);
  glVertex3f(x1,y1,0.0);
  glVertex3f(x1,y1,0.0);
  glVertex3f(x0,y1,0.0);
  glVertex3f(x0,y1,0.0);
  glVertex3f(x0,y0,0.0);

  glEnd();
  glPopAttrib();
  glPopAttrib();

  // Update Counters

  fr.display();
  stepcountmsg.display();
}

template<typename D , typename F >

void integration< D, F >::LJcalc

(

)

[inline]

Definition at line 244 of file integration.h.

References particle::acc, integration< D, F >::LJpsi, integration< D, F >::LJsigma, integration< D, F >::numParticles, integration< D, F >::pi, particle::pos, r, and SHOW.

Referenced by integration< D, F >::bruteforce().

{
  double rx;
  double ry;

  double r;
  double z;
  double f;
  double fx;
  double fy;

  for (uint i=0; i<numParticles; ++i)
    pi[i].acczero();

//cout << "Before" << endl;
//  for (uint i=0; i<numParticles; ++i)
//    cout << SHOW(pi[i]) << endl;
//cout << endl;

  for (uint i=0; i+1<numParticles; ++i)
  {
    for (uint k=i+1; k<numParticles; ++k)
    {
      rx = pi[i].pos[0] - pi[k].pos[0];
      ry = pi[i].pos[1] - pi[k].pos[1];
    
      r = sqrt(rx*rx+ry*ry);
//cout << SHOW(r) << " ";
      z = LJsigma/r;
//cout << SHOW(z) << " ";
      z *= z;
      z *= z*z;

      // Force divided by r.
      f = LJpsi*24.0/(r*r)*z*(2.0*z-1.0);

if (abs(f)>10)
cout << SHOW(f) << endl;

//cout << SHOW(f) << " ";

      fx = f*rx/r;
      fy = f*ry/r;

//cout << SHOW(fx) << " ";
//cout << SHOW(fy) << " ";
//cout << endl;

      pi[i].acc[0] -= fx;
      pi[i].acc[1] -= fy;

      pi[k].acc[0] += fx;
      pi[k].acc[1] += fy;
    }

  }

//  for (uint i=0; i<numParticles; ++i)
//    cout << SHOW(pi[i]) << endl;

}

template<typename D , typename F >

void integration< D, F >::reset	(	int	argc,
		char **	argv
	)			[inline]

Definition at line 553 of file integration.h.

References integration< D, F >::displayEk, commandline::enablehelp(), integration< D, F >::h, integration< D, F >::histhigh, integration< D, F >::histlow, integration< D, F >::histn, integration< D, F >::LJ, integration< D, F >::LJpsi, integration< D, F >::LJsigma, commandline::mapvar(), integration< D, F >::numParticles, integration< D, F >::pi, integration< D, F >::printstate, integration< D, F >::pu, cell::reset(), integration< D, F >::sampler, integration< D, F >::samplerstate, particlesampler::set(), and integration< D, F >::uniform.

Referenced by test01(), and test02().

{
  commandline c(argc,argv);
  c.mapvar(numParticles,"numParticles");
  if(pi!=0)
    delete[] pi;
  fn.pi = pi = new particle[numParticles];

  C.reset(numParticles);


// Test Code
pu = new uint [numParticles];
for (uint i=0; i<numParticles; ++i)
  pu[i]=i;

  c.mapvar(uniform,"uniform");
  c.mapvar(distribution.vmax,"vmax");
  c.mapvar(distribution.radius,"radius");
  c.mapvar(h,"h");
  c.mapvar(printstate,"printstate");

// <TODO> put this option back in.
//  c.mapvar(zpr,"zpr");
//  if (zpr)
//    zprInit();

  c.mapvar(distribution.box.x0,"x0");
  c.mapvar(distribution.box.x1,"x1");
  c.mapvar(distribution.box.y0,"y0");
  c.mapvar(distribution.box.y1,"y1");

  c.mapvar(displayEk,"displayEk");

  c.mapvar(samplerstate,"samplerstate");

  c.mapvar(LJpsi,"LJpsi");
  c.mapvar(LJsigma,"LJsigma");
  c.mapvar(LJ,"LJ");
  if (LJ)
    uniform=false;

  c.mapvar(histlow,"histlow");
  c.mapvar(histhigh,"histhigh");
  c.mapvar(histn,"histn");
  
  sampler.set(histogram<double>(histlow,histhigh,histn));

  c.enablehelp(cout);

  fn.reset();

  // Initial conditions: no intersecting particles.
  for (uint i=0; i<numParticles; ++i)
  {
    distribution.eval(pi[i]);
    bool flagnoint(true);
    for (uint k=0; (k<i)&&flagnoint; ++k)
    {
      if( pi[i].isIntersecting( pi[k] ) )
      {
        flagnoint=false;
        --i;
      }
    }
  } 

}

template<typename D, typename F>

void integration< D, F >::step

(

)

[inline]

Definition at line 76 of file integration.h.

References integration< D, F >::bruteforce(), integration< D, F >::sampler, integration< D, F >::samplerstate, integration< D, F >::stepcount, integration< D, F >::uniform, and integration< D, F >::uniformgridtest().

Referenced by animate(), keyboard(), and integration< D, F >::steplarge().

  {
    if (!uniform)
      bruteforce();
    else
      uniformgridtest();

    ++stepcount;

    if (samplerstate)
      ++sampler;
  }

template<typename D , typename F >

void integration< D, F >::steplarge

(

)

[inline]

Definition at line 227 of file integration.h.

References integration< D, F >::step().

Referenced by keyboard().

{
  for (uint i=0; i<100; ++i)
    step();
}

template<typename D , typename F >

void integration< D, F >::togglestartstop

(

)

[inline]

Definition at line 235 of file integration.h.

References integration< D, F >::run.

Referenced by keyboard().

{
  if (run)
    run=false;
  else
    run=true;
}

template<typename D , typename F >

void integration< D, F >::uniformgridtest

(

)

[inline]

Definition at line 124 of file integration.h.

References integration< D, F >::bruteforce(), cell::eval(), cell::getsurroundingparticles(), integration< D, F >::h, integration< D, F >::numParticles, integration< D, F >::pi, integration< D, F >::printstate, integration< D, F >::pu, particle::step(), and boxcollision::walls().

Referenced by integration< D, F >::step(), and test02().

{
  boxcollision const & box(distribution.box);
  for (uint i=0; i<numParticles; ++i)
  {
    pi[i].step(h);
    box.walls(pi[i],h);
  }

  C.eval(fn);

if (true&&printstate)
{
uint W(fn.W);
uint count;
int idx=0;
cout << endl;
for (uint i=0; i<W; ++i)
{
  for (uint j=0; j<W; ++j)
  {
   
    count=0;
    for (uint k=0; k<numParticles; ++k)
    {
      if (idx==fn.index(k))
      {
        cout << k << ',';
        ++count;
      }
    }
    if (count==0)
      cout << "*";
    cout << " ";
    ++idx;
  }
  cout << endl;
}
}

  uint sz;
  for (uint i=0; i<numParticles; ++i)
  {
    C.getsurroundingparticles(pu,sz,fn,i);

//cout << SHOW(i) << "   " << SHOW(sz) << "  ";
//print(pu,pu+sz);

    bruteforce(pu,sz);
  }


if (true&&printstate)
{
cout << endl;
for (uint i=0; i<numParticles; ++i)
{
  cout << fn.index(i) << " pi[" << i << "] ";
  C.getsurroundingparticles(pu,sz,fn,i);
  cout << "  ";
  for (uint k=0; k<sz; ++k)
    cout << pu[k] << " ";
  cout << "        ";
  cout << pi[i];
cout << endl;
}
}


if (true&&printstate)
{
uint W(fn.W);
uint count;
int idx=0;
cout << endl;
for (uint i=0; i<W; ++i)
{
  for (uint j=0; j<W; ++j)
  {
   
    count=0;
    for (uint k=0; k<numParticles; ++k)
    {
      if (idx==fn.index(k))
      {
        cout << k << ',';
        ++count;
      }
    }
    if (count==0)
      cout << "*";
    cout << " ";
    ++idx;
  }
  cout << endl;
}
}


}

---

Member Data Documentation

template<typename D, typename F>

bool integration< D, F >::displayEk

Definition at line 37 of file integration.h.

Referenced by integration< D, F >::draw(), integration< D, F >::integration(), and integration< D, F >::reset().

template<typename D, typename F>

framerate integration< D, F >::fr

Definition at line 57 of file integration.h.

Referenced by animate(), and integration< D, F >::draw().

template<typename D, typename F>

float integration< D, F >::h

Definition at line 35 of file integration.h.

Referenced by integration< D, F >::bruteforce(), integration< D, F >::integration(), integration< D, F >::reset(), and integration< D, F >::uniformgridtest().

template<typename D, typename F>

double integration< D, F >::histhigh

Definition at line 47 of file integration.h.

Referenced by integration< D, F >::reset().

template<typename D, typename F>

double integration< D, F >::histlow

Definition at line 46 of file integration.h.

Referenced by integration< D, F >::reset().

template<typename D, typename F>

uint integration< D, F >::histn

Definition at line 48 of file integration.h.

Referenced by integration< D, F >::reset().

template<typename D, typename F>

bool integration< D, F >::LJ

Lennard-Jones potential is enabled with brute force only.

Definition at line 40 of file integration.h.

Referenced by integration< D, F >::bruteforce(), integration< D, F >::integration(), and integration< D, F >::reset().

template<typename D, typename F>

double integration< D, F >::LJpsi

Definition at line 41 of file integration.h.

Referenced by integration< D, F >::integration(), integration< D, F >::LJcalc(), and integration< D, F >::reset().

template<typename D, typename F>

double integration< D, F >::LJsigma

Definition at line 42 of file integration.h.

Referenced by integration< D, F >::integration(), integration< D, F >::LJcalc(), and integration< D, F >::reset().

template<typename D, typename F>

uint integration< D, F >::numParticles

Definition at line 32 of file integration.h.

Referenced by integration< D, F >::bruteforce(), integration< D, F >::draw(), integration< D, F >::integration(), integration< D, F >::LJcalc(), integration< D, F >::reset(), test02(), and integration< D, F >::uniformgridtest().

template<typename D, typename F>

particle* integration< D, F >::pi

Definition at line 53 of file integration.h.

Referenced by integration< D, F >::bruteforce(), integration< D, F >::draw(), integration< D, F >::integration(), integration< D, F >::LJcalc(), integration< D, F >::reset(), and integration< D, F >::uniformgridtest().

template<typename D, typename F>

bool integration< D, F >::printstate

Definition at line 34 of file integration.h.

Referenced by integration< D, F >::integration(), integration< D, F >::reset(), and integration< D, F >::uniformgridtest().

template<typename D, typename F>

uint* integration< D, F >::pu

Definition at line 55 of file integration.h.

Referenced by integration< D, F >::reset(), and integration< D, F >::uniformgridtest().

template<typename D, typename F>

bool integration< D, F >::run

Definition at line 73 of file integration.h.

Referenced by animate(), integration< D, F >::integration(), and integration< D, F >::togglestartstop().

template<typename D, typename F>

particlesampler integration< D, F >::sampler

Definition at line 50 of file integration.h.

Referenced by integration< D, F >::reset(), and integration< D, F >::step().

template<typename D, typename F>

uint integration< D, F >::samplerstate

Definition at line 51 of file integration.h.

Referenced by integration< D, F >::reset(), and integration< D, F >::step().

template<typename D, typename F>

long integration< D, F >::stepcount

Definition at line 29 of file integration.h.

Referenced by animate(), and integration< D, F >::step().

template<typename D, typename F>

gltextmsg integration< D, F >::stepcountmsg

Definition at line 30 of file integration.h.

Referenced by animate(), and integration< D, F >::draw().

template<typename D, typename F>

bool integration< D, F >::uniform

Definition at line 33 of file integration.h.

Referenced by integration< D, F >::integration(), integration< D, F >::reset(), integration< D, F >::step(), and test02().

---

The documentation for this class was generated from the following file:

• nbody/integration.h