source: PSPA/Interface_Web/trunk/pspaWT/include/particleBeam.h @ 143

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

#include<cmath>

#include <vector>

#include "bareParticle.h"
#include "mathematicalTools.h"
#include "mathematicalConstants.h"
#include "PhysicalConstants.h"
#include "nomdElements.h"

using namespace std;

struct particle
{
  float xx, xxp, begamx,yy,yyp,begamy,z, begamz,phi,wz;
  float phi0, ksi1,ksi2,ksi3;
  int ne,np,ngood,npart;

  int readFromParmelaFile(FILE* fp)
  {
    return fscanf(fp, " %e %e %e %e %e %e %e %e %e %e %d %d %d %d %e %e %e %e \n", &xx, &xxp, &begamx,&yy,&yyp,&begamy,&z, &begamz,&phi,&wz,&ne,&np,&ngood,&npart,&phi0, &ksi1,&ksi2,&ksi3);
  }
  void imprim()
  {
    printf( " %e %e %e %e %e %e %e %e %e %e %d %d %d, %d %e %e %e %e \n", xx, xxp, begamx,yy,yyp,begamy,z, begamz,phi,wz,ne,np,ngood,npart,phi0, ksi1,ksi2,ksi3);
  }
};



class particleBeam 
{

  bool particleRepresentationOk_;
  bool momentRepresentationOk_;

  bareParticle referenceParticle_;

  vector<bareParticle> goodPartic_;

      vector< vector<double> > rij_transportMoments_;
      vector<double> centroid_;
      double P0Transport_;


      void readTransportMoments(ifstream& inp);

      void impressionDesMoments() const;


 public:

 particleBeam()  
   {
     rij_transportMoments_.resize(6);
     unsigned dim=0;
     unsigned k;
     for ( k=0; k < 6; k++)
       {
         rij_transportMoments_.at(k).resize(++dim); 
       }
     P0Transport_ = 0.0;
     particleRepresentationOk_ = false;
     momentRepresentationOk_ = false;

   }

  ~particleBeam() {;}

  bool setFromParmela(unsigned numeroElement,double referencefrequency);
      void buildMomentRepresentation();

      //       bool setFromTransport(ifstream& inp, unsigned nblignes);
          bool  setFromTransport(string elementLabel, typedElement elemType);

  inline void clear() 
  {
    goodPartic_.clear();
     P0Transport_ = 0.0;
     particleRepresentationOk_ = false;
     momentRepresentationOk_ = false;
  }

inline  int getNbParticles() const
  {
    return goodPartic_.size();
  }

 inline const vector< vector<double> >&  getTransportMoments() const  { return rij_transportMoments_;}
 inline double getP0Transport() const { return P0Transport_;}
  inline bool particleRepresentationOk() const {return particleRepresentationOk_;}
 inline bool momentRepresentationOk() const {return momentRepresentationOk_;}


inline void  addParticle( bareParticle p)
    {
      goodPartic_.push_back(p);
    }



inline const vector<bareParticle>& getParticleVector() const
 {
   return goodPartic_;
 }

inline vector<bareParticle>& getParticleVector() 
 {
   return goodPartic_;
 }

 void getVariance(double& varx, double& vary, double& varz) const;
 

 void printAllXYZ() const;


 void Zrange(double& zmin, double& zmax) const;

 void donneesDessinEllipseXxp(vector<double>& xcor, vector<double>& ycor);


 virtual string FileOutputFlow() const;


 virtual bool FileInput(ifstream& ifs);


};


#endif