McEventCollectionCnv_p4.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// McEventCollectionCnv_p4.h
// Header file for class McEventCollectionCnv_p4
// Author: S.Binet<binet@cern.ch>
#ifndef GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_P4_H
#define GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_P4_H
 
// STL includes
#include <unordered_map>
 
#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
#include "GeneratorObjects/McEventCollection.h"
 
// AthenaPoolCnvSvc includes
#include "AthenaPoolCnvSvc/T_AthenaPoolTPConverter.h"
 
// GeneratorObjectsTPCnv includes
#include "GeneratorObjectsTPCnv/McEventCollection_p4.h"
 
#include "GaudiKernel/ServiceHandle.h"
 
// Forward declaration
class IHepMCWeightSvc; 
class MsgStream;
namespace HepMC { struct DataPool; }
 
class McEventCollectionCnv_p4 : public T_AthenaPoolTPCnvBase<
                                          McEventCollection,
                                          McEventCollection_p4
                                       >
{
 
  typedef T_AthenaPoolTPCnvBase<McEventCollection,
                                McEventCollection_p4> Base_t;

  // Public methods:
 public:

  McEventCollectionCnv_p4();

  McEventCollectionCnv_p4( const McEventCollectionCnv_p4& rhs );

  McEventCollectionCnv_p4& operator=( const McEventCollectionCnv_p4& rhs );

  virtual ~McEventCollectionCnv_p4();
 
  void setPileup();

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

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

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

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

  HepMC::GenParticlePtr
  createGenParticle( const GenParticle_p4& 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_p4& persEvt ) ;
#else
  void writeGenVertex( const HepMC::GenVertex& vtx,
                       McEventCollection_p4& persEvt ) const;
#endif

#ifdef HEPMC3   
  static int writeGenParticle( const HepMC::ConstGenParticlePtr& p,
                        McEventCollection_p4& persEvt ) ;
#else
  int writeGenParticle( const HepMC::GenParticle& p,
                        McEventCollection_p4& persEvt ) const;
 
#endif
  // Protected data:
 protected:
 
  bool m_isPileup;
  ServiceHandle<IHepMCWeightSvc> m_hepMCWeightSvc; 
};
#endif //> GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_P4_H