CbmEventGenerator.h

Go to the documentation of this file.

 
#ifndef CBMEVENTGENERATOR_H
#define CBMEVENTGENERATOR_H 1
 
 
#include "CbmBeamProfile.h"
#include "CbmTarget.h"
#include <FairPrimaryGenerator.h>
#include <memory>
 
class FairGenericStack;
 
 
class CbmEventGenerator : public FairPrimaryGenerator {
 
public:
  CbmEventGenerator();
 
 
  virtual ~CbmEventGenerator();
 
 
  void ForceVertexAtZ(Double_t zVertex);
 
 
  void ForceVertexInTarget(Bool_t choice = kTRUE) {
    fForceVertexInTarget = choice;
  }
 
 
  const CbmBeamProfile& GetBeamProfile() { return fBeamProfile; }
 
 
  virtual Bool_t GenerateEvent(FairGenericStack* stack);
 
 
  virtual void Print(Option_t* opt = "") const;
 
 
  void SetBeamAngle(Double_t meanThetaX,
                    Double_t meanThetaY,
                    Double_t sigmaThetaX = -1.,
                    Double_t sigmaThetaY = -1.);
 
 
  void SetBeamPosition(Double_t meanX,
                       Double_t meanY,
                       Double_t sigmaX = -1.,
                       Double_t sigmaY = -1.,
                       Double_t zF     = 0.);
 
 
  void SetTarget(std::shared_ptr<const CbmTarget> target) { fTarget = target; }
 
 
private:
  CbmBeamProfile fBeamProfile;               
  std::shared_ptr<const CbmTarget> fTarget;  
  Bool_t fForceVertexInTarget;  
  Bool_t fForceVertexAtZ;       
  Double_t fVertexZ;            
 
 
  virtual void MakeBeamAngle() {};
 
 
  virtual void MakeVertex();
 
 
  void MakeVertexAtZ();
 
 
  virtual void MakeVertexInFocalPlane();
 
 
  virtual void MakeVertexInTarget();
 
 
  ClassDef(CbmEventGenerator, 2);
};
 
 
#endif /* CBMEVENTGENERATOR_H */