PrimaryGenerator.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_EVENTGEN_PRIMARYGENERATOR_H_
#define ALICEO2_EVENTGEN_PRIMARYGENERATOR_H_
 
#include "FairPrimaryGenerator.h"
#include "DataFormatsCalibration/MeanVertexObject.h"
#include "SimConfig/SimConfig.h"
 
class TFile;
class TTree;
class TString;
 
namespace o2
{
namespace dataformats
{
class MCEventHeader;
}
} // namespace o2
 
namespace o2
{
namespace eventgen
{
 
class PrimaryGenerator : public FairPrimaryGenerator
{
 
 public:
  PrimaryGenerator() = default;
  ~PrimaryGenerator() override;
 
  Bool_t GenerateEvent(FairGenericStack* pStack) override;
 
  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 mother1 = -1, Int_t mother2 = -1,
                Int_t daughter1 = -1, Int_t daughter2 = -1,
                Bool_t wanttracking = true,
                Double_t e = -9e9, Double_t tof = 0.,
                Double_t weight = 0., TMCProcess proc = kPPrimary, Int_t generatorStatus = 0);
 
  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) override;
 
  Bool_t Init() override;
 
  Bool_t embedInto(TString fname);
 
  void setEmbedIndex(int idx) { mEmbedIndex = idx; }
 
  void setExternalVertexForNextEvent(double x, double y, double z);
 
  // sets the vertex mode; if mode is kCCDB, a valid MeanVertexObject pointer must be given at the same time
  void setVertexMode(o2::conf::VertexMode const& mode, o2::dataformats::MeanVertexObject const* obj = nullptr);
  // if we apply vertex smearing
  void setApplyVertex(bool onoff) { mApplyVertex = onoff; }
 
  // set identifier and description
  void setGeneratorId(int id) { mGeneratorId = id; }
  void setGeneratorDescription(std::string const& desc) { mGeneratorDescription = desc; }
 
 protected:
  // PrimaryGenerator(const PrimaryGenerator&) = default;
  // PrimaryGenerator& operator=(const PrimaryGenerator&) = default;
 
  void setInteractionDiamond(const Double_t* xyz, const Double_t* sigmaxyz);
 
  void setInteractionVertex(const o2::dataformats::MCEventHeader* event);
 
  void fixInteractionVertex();
 
  float mExternalVertexX = 0, mExternalVertexY = 0, mExternalVertexZ = 0; // holding vertex fixed from outside
  bool mHaveExternalVertex = false;                                       // true of user fixed external vertex from outside for next event
 
  TFile* mEmbedFile = nullptr;
  TTree* mEmbedTree = nullptr;
  Int_t mEmbedEntries = 0;
  Int_t mEmbedIndex = 0;
  o2::dataformats::MCEventHeader* mEmbedEvent = nullptr;
 
  bool mApplyVertex = true;
  o2::conf::VertexMode mVertexMode = o2::conf::VertexMode::kDiamondParam; // !vertex mode
  std::unique_ptr<o2::dataformats::MeanVertexObject> mMeanVertex;
 
 private:
  void setGeneratorInformation();
  // generator identifier and description
  int mGeneratorId = -1;
  std::string mGeneratorDescription;
 
  ClassDefOverride(PrimaryGenerator, 3);
 
}; 
/*****************************************************************/
/*****************************************************************/
 
} /* namespace eventgen */
} /* namespace o2 */
 
#endif /* ALICEO2_EVENTGEN_PRIMARYGENERATOR_H_ */