CbmEventGenerator.h

Go to the documentation of this file.

/* Copyright (C) 2019-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese [committer] */

 
 
#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();
 

  virtual void AddTrack(Int_t pdgid, Double_t px, Double_t py, Double_t pz, Double_t vx, Double_t vy, Double_t vz,
                        Int_t parent = -1, Bool_t wanttracking = true, Double_t e = -9e9, Double_t tof = 0.,
                        Double_t weight = 0., TMCProcess proc = kPPrimary);
 

  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 */