PndGiBuuGenerator.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 PndGiBuuGenerator header file                  -----
// -----               Created 02/08/12  by T.Stockmanns                -----
// -------------------------------------------------------------------------

#ifndef PND_GiBuuGENERATOR_H
#define PND_GiBuuGENERATOR_H

#include "FairGenerator.h"
#include "PndGiBuuDecayerAbs.h"

#include "TString.h"
#include <fstream>
#include <map>

class FairPrimaryGenerator;

class PndGiBuuGenerator : public FairGenerator {

 public:
  PndGiBuuGenerator(const Char_t *fileName, const Char_t *pidDataFileName);

  virtual ~PndGiBuuGenerator();

  virtual Bool_t ReadEvent(FairPrimaryGenerator *primGen);

  void RegisterDecayer(Int_t pdgCode, PndGiBuuDecayerAbs *decayer) { fDecayerMap[pdgCode] = decayer; }

 protected:
  Bool_t Init();

 private:
  PndGiBuuGenerator(){};

  void FillPidMap();
  Int_t GetPdgParticleId(Int_t GiBuuId, Int_t charge);
  void WriteoutDecayParticle(Int_t oldEventNr, Int_t oldRunId, FairPrimaryGenerator *primGen);

  std::map<std::pair<Int_t, Int_t>, Int_t> fMapGiBuuToPdg; 

  Int_t fEvent;                   
  Int_t fRunId;                   
  TString fFileName;              
  TString fPidDataFileName;       
  std::ifstream *fInputAsciiFile; 

  std::map<Int_t, PndGiBuuDecayerAbs *> fDecayerMap; 

  Double_t fPx, fPy, fPz; 
  Int_t fPdg;             

  void CloseInput();

  ClassDef(PndGiBuuGenerator, 1);
};

#endif