McEventCollectionCnv_p5.h

Go to the documentation of this file.

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

  // Public methods:
 public:

  McEventCollectionCnv_p5();

  McEventCollectionCnv_p5( const McEventCollectionCnv_p5& rhs );

  McEventCollectionCnv_p5& operator=( const McEventCollectionCnv_p5& rhs );

  virtual ~McEventCollectionCnv_p5();
 
  void setPileup();

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

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

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

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

  HepMC::GenParticlePtr
  createGenParticle( const GenParticle_p5& 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_p5& persEvt ) ;
#else
  void writeGenVertex( const HepMC::GenVertex& vtx,
                       McEventCollection_p5& persEvt ) const;
 
#endif
#ifdef HEPMC3
  static int writeGenParticle( const HepMC::ConstGenParticlePtr& p,
                        McEventCollection_p5& persEvt ) ;
#else
  int writeGenParticle( const HepMC::GenParticle& p,
                        McEventCollection_p5& persEvt ) const;
#endif

  // Protected data:
 protected:
 
  bool m_isPileup;
    ServiceHandle<IHepMCWeightSvc> m_hepMCWeightSvc;
};
#endif //> GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_P5_H