source: PSPA/Interface_Web/trunk/pspaWT/sources/controler/include/particleBeam.h @ 354

#ifndef PARTICLEBEAM_SEEN
#define  PARTICLEBEAM_SEEN
#include <iostream>
#include <sstream>
#include <stdio.h>
#include<cmath>
#include <vector>

#include "bareParticle.h"
#include "nomdElements.h"
#include "beam2Moments.h"

using namespace std;

class particleBeam 
{
  bool particleRepresentationOk_;
  bool momentRepresentationOk_;
  beam2Moments rij_;
  double P0Transport_;

  // la representation du faisceau de particules sera a preciser ulterieurement 
  // Pour l'instant une particule courante est definie par : 
  // x
  // y
  // deltaz = z - z0 (z0 est le z de la particule de reference)
  // gamma.betax
  // gamma.betay
  // gamma.betaz
  vector<bareParticle> goodPartic_;

  // par ailleurs, la particules de reference est definie par x,y,z,gamma.betax, gamma.betay, gamma.betaz
  // elle est dupliquee dans le faisceau prorement dit (goodPartic_) avec z = z-z0 et ne doit, par consequent,
  // pas etre prise en compte dans les calculs de trajectoire.
  bareParticle referenceParticle_;
  

  vector<double> centroid_;

  void impressionDesMoments() const;
  void razDesMoments();
  void particlesPhaseSpaceComponent(vector<double>& coord, unsigned index);

 public:

  particleBeam(); 
  virtual ~particleBeam() {;}

  void buildMomentRepresentation();
  void clear();
  int getNbParticles() const;
  const beam2Moments&  getTransportMoments() const;
  double getSigmaTransportij(unsigned i, unsigned j);
  double getUnnormalizedEmittanceX();
  double getP0Transport() const;
  double referenceKineticEnergyMeV() const;
  bool particleRepresentationOk() const;
  bool momentRepresentationOk() const;
  void  addParticle( bareParticle p);
  const vector<bareParticle>& getParticleVector() const;
  vector<bareParticle>& getParticleVector(); 
  double getXmaxRms();
  void getVariance(double& varx, double& vary, double& varz) const;
  void set2Moments(beam2Moments& moments);
  void setWithParticles(vector<double>& centroid, bareParticle& referencePart, vector<bareParticle>& particles);
  void printAllXYZ() const;
  void Zrange(double& zmin, double& zmax) const;
  //  void donneesDessinEllipseXxp(vector<double>& xcor, vector<double>& ycor);
  void donneesDessinEllipse(vector<double>& xcor, vector<double>& ycor, unsigned indexAbs, unsigned indexOrd);
  void particlesPhaseSpaceData(vector<double>& xcor, vector<double>& ycor, unsigned indexAbs, unsigned indexOrd);
  virtual string FileOutputFlow() const;
  virtual bool FileInput(ifstream& ifs);
  void histogramme(vector<double>&xcor,vector<int>& hist,int& cnts,double out[3]);
};
#endif