TrackRecordGenerator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef TRACKRECORDGENERATOR_H
#define TRACKRECORDGENERATOR_H
 
#include "GeneratorModules/GenModule.h"
#include "GeneratorModules/GenData.h"
#include <memory>
 
// Members
#include "CLHEP/Vector/LorentzVector.h"
#include "AtlasHepMC/Polarization.h"
 
#include <vector>
#include <string>
 
class TrackRecordGenerator : public GenModule {
 public:
   TrackRecordGenerator(const std::string& name, ISvcLocator* pSvcLocator);
   virtual ~TrackRecordGenerator();
   virtual StatusCode callGenerator();
   virtual StatusCode fillEvt(HepMC::GenEvent* evt);
 
 private:
 
   // Event record holders
   int m_events;
   std::vector<int> m_pdgCode;
   std::vector<CLHEP::HepLorentzVector> m_fourPos;
   std::vector<CLHEP::HepLorentzVector> m_fourMom;
   CLHEP::Hep3Vector m_center;
   std::vector<HepMC::Polarization> m_polarization;
 
   // Options for track records 
   double m_smearTR;  
   double m_smearTRp; 
   std::string m_recordName;  
   bool m_stopParticles;      
   float m_stopped_tminus, m_stopped_tplus; 
   bool m_add_cL; 
   std::shared_ptr<GenData> m_gendata{nullptr};
};
 
#endif