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();
 
 
  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 */