McEventCollectionCnv_p7.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// McEventCollectionCnv_p7.h
// Header file for class McEventCollectionCnv_p7
// Author: J.Chapman, P.Clark and A.Verbytskyi
#ifndef GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_p7_H
#define GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_p7_H
 
// STL includes
#include <unordered_map>
 
#ifdef HEPMC3
#include "AtlasHepMC/GenEvent.h"
#include "AtlasHepMC/GenVertex.h"
#include "AtlasHepMC/GenParticle.h"
#include "HepMC3/Data/GenRunInfoData.h"
#else
#ifdef __clang__
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wkeyword-macro"
#endif
#define private public
#define protected public
#include "AtlasHepMC/GenEvent.h"
#include "AtlasHepMC/GenVertex.h"
#include "AtlasHepMC/GenParticle.h"
#undef private
#undef protected
#ifdef __clang__
#pragma clang diagnostic pop
#endif
#endif
#include "GeneratorObjects/McEventCollection.h"
 
// AthenaPoolCnvSvc includes
#include "AthenaPoolCnvSvc/T_AthenaPoolTPConverter.h"
 
// GeneratorObjectsTPCnv includes
#include "GeneratorObjectsTPCnv/McEventCollection_p7.h"
 
#include "GaudiKernel/ServiceHandle.h"
 
class IHepMCWeightSvc;
 
// Forward declaration
class MsgStream;
namespace HepMC { struct DataPool; }
 
class McEventCollectionCnv_p7 : public T_AthenaPoolTPCnvBase<
                                          McEventCollection,
                                          McEventCollection_p7
                                       >
{
 
  typedef T_AthenaPoolTPCnvBase<McEventCollection,
                                McEventCollection_p7> Base_t;

  // Public methods:
 public:

  McEventCollectionCnv_p7();

  McEventCollectionCnv_p7( const McEventCollectionCnv_p7& rhs );

  McEventCollectionCnv_p7& operator=( const McEventCollectionCnv_p7& rhs );

  virtual ~McEventCollectionCnv_p7();
 
  void setPileup();

  virtual void persToTrans( const McEventCollection_p7* persObj,
                            McEventCollection* transObj,
                            MsgStream &log ) ;

  virtual void transToPers( const McEventCollection* transObj,
                            McEventCollection_p7* persObj,
                            MsgStream &log ) ;

  // Protected method:
 protected:
 
  typedef std::unordered_map<HepMC::GenParticlePtr,int> ParticlesMap_t;

  HepMC::GenVertexPtr
  createGenVertex( const McEventCollection_p7& persEvts,
                   const GenVertex_p7& vtx,
                   ParticlesMap_t& bcToPart,
                   HepMC::DataPool& datapools, HepMC::GenEvent* parent=nullptr ) const;

  HepMC::GenParticlePtr
  createGenParticle( const GenParticle_p7& p,
                     ParticlesMap_t& partToEndVtx,
                     HepMC::DataPool& datapools, const HepMC::GenVertexPtr& parent=nullptr, bool add_to_output = true ) const;

#ifdef HEPMC3
  static void writeGenVertex( const HepMC::ConstGenVertexPtr& vtx, McEventCollection_p7& persEvt );
#else
  void writeGenVertex( const HepMC::GenVertex& vtx, McEventCollection_p7& persEvt ) const;
#endif
#ifdef HEPMC3
  static int writeGenParticle( const HepMC::ConstGenParticlePtr& p, McEventCollection_p7& persEvt );
#else
  int writeGenParticle( const HepMC::GenParticle& p, McEventCollection_p7& persEvt ) const;
#endif
 
  bool m_isPileup;
  ServiceHandle<IHepMCWeightSvc> m_hepMCWeightSvc;
};
#endif //> GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_p7_H