PndCorrDistGenerator.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----               PndCorrDistGenerator header file                -----
// -----               Created 23/05/07  by Aida Galoyan               -----
// -------------------------------------------------------------------------

#ifndef PNDCORRDISTGENERATOR_H
#define PNDCORRDISTGENERATOR_H 1

#include "FairGenerator.h"
#include "THParticle.h"

// using namespace std;

class TClonesArray;
class TFile;
class TH2F;
class TTree;
class TVirtualMCStack;
class FairPrimaryGenerator;

class PndCorrDistGenerator : public FairGenerator {

 public:
  PndCorrDistGenerator();

  PndCorrDistGenerator(const Char_t *fileName);

  virtual ~PndCorrDistGenerator();

  virtual Bool_t ReadEvent(FairPrimaryGenerator *primGen);
  void SetParam() { fParam = kTRUE; };
  void SetThetaRange(Double_t thetLow = 0., Double_t thetHigh = 0.)
  {
    fTheLow = thetLow;
    fTheHigh = thetHigh;
  };

 private:
  Int_t iEvent;             
  const Char_t *fFileName;  
  TFile *fInputFile;        
  TH2F *fInputHist;         
  TClonesArray *fParticles; 
  Int_t fPdgType;
  Bool_t fParam;
  Double_t fTheLow, fTheHigh;

  void CloseInput();

  Double_t MaxBoltDistP(Double_t MeanP);

  Double_t MeanMomentum(Double_t thet);

  ClassDef(PndCorrDistGenerator, 3);
};

#endif