species.h

Go to the documentation of this file.

#ifndef SPECIES_H
#define SPECIES_H
#include "task.h"
#include "numeric.h"
#include <cmath>
#define ID_SPECIES  "Species"
//---------------------------------------------------------------------------------------------------------------------
class Species;
typedef PtrWrapper<Species> WSpecies;
class SpeciesBase;
typedef PtrWrapper<SpeciesBase> WSpeciesBase;

class SpeciesBase : public Task {
 public:
  virtual void MakeRho()=0;
  virtual grid<double>& Rho ()=0;  
  virtual std::string GetClassName () const {return ID_SPECIES;}  
  virtual double GetSize () const = 0;
  virtual double GetSize0 () const = 0;
  virtual double GetCharge () const = 0;
};
class Wall;
class Density;
class Geometry;

class Species : public SpeciesBase {
 private:
  std::string strGeometry;
  std::string strDensity;
  std::string strWall;
  bool Moved;
  
  double initialDensity;
  GridPosition gRhoSize;  
 protected:
  Geometry* pGeom;
  Wall* pWall;
  grid<double> gRho;
  grid<double> gRho1;
  grid<double> gRho2;
  grid<double> gRho3;
  grid<double> gRho4;
  grid<double> gRho5;
  grid<double> gRho6;
  grid<double> gRho7;
  grid<double> gRho8;
  int MaxCount;
  double Size;
  double Size0;
  double Charge;
  double Mass;
  double dt;
 public:
  std::valarray<Position> X;
  std::valarray<Velocity> V;
  int Count;
 public:
  Species ();
  virtual ~Species ();
  virtual void Init ();
  virtual bool Execute ();
  void MakeRho();
  double GetSize () const {return Size;}
  double GetSize0 () const {return Size0;}
  double GetCharge () const {return Charge;}
  double GetMass () const {return Mass;}
  double Getdt () const {return dt;}
  grid<double>& Rho () {return gRho;}
  double Vabs (int p) const {return sqrt(V[p]*V[p]);}
  void SetV (int p, const Velocity & _V) {V[p] = _V;}
  Position GetRx (int nParticle) const {return ((nParticle < Count) ? X[nParticle] : 0.0);}
  void SetMovedFlag () {Moved = true;};
  void ClearMovedFlag () {Moved = false;};
  bool HasMoved () const {return Moved;};
  int Generate (Position aX, Velocity aV);
  void Remove (int nP);
  void Swap (int i1, int i2);
  void ModifySize(double densityContribution);
 protected:
  virtual PARAMETERMAP* MakeParamMap (PARAMETERMAP* pm = NULL);
};
#endif