CbmShieldGenerator.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                CbmShieldGenerator header file                 -----
// -----                Created 15/09/06 by V. Friese                  -----
// -------------------------------------------------------------------------
 
#ifndef FAIRSHIELDGENERATOR_H
#define FAIRSHIELDGENERATOR_H 1
 
 
#include "FairGenerator.h"
 
#include <fstream>
#include <map>
 
class TDatabasePDG;
class FairPrimaryGenerator;
class FairIon;
 
class CbmShieldGenerator : public FairGenerator {
 
public:
  CbmShieldGenerator();
 
 
  CbmShieldGenerator(const char* fileName);
 
 
  virtual ~CbmShieldGenerator();
 
 
  // virtual Bool_t ReadEvent(FairPrimaryGenerator* primGen);
  virtual Bool_t ReadEvent(FairPrimaryGenerator* primGen);
 
  void SetPartType(int partType) { fpartType = partType; }
 
private:
  std::ifstream* fInputFile;  
  const Char_t* fFileName;    
  TDatabasePDG* fPDG;         
 
  int fpartType;  // SELIM
 
  void CloseInput();
 
 
  Int_t RegisterIons();
 
 
  std::map<TString, FairIon*> fIonMap;  
 
 
  CbmShieldGenerator(const CbmShieldGenerator&);
  CbmShieldGenerator& operator=(const CbmShieldGenerator&);
 
  ClassDef(CbmShieldGenerator, 2);
};
 
#endif