McEventCollectionCnv_p6.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
// McEventCollectionCnv_p6.h
// Header file for class McEventCollectionCnv_p6
// Author: J.Chapman
#ifndef GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_p6_H
#define GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_p6_H
 
// STL includes
#include <unordered_map>
 
# include "AtlasHepMC/GenEvent.h"
# include "AtlasHepMC/GenVertex.h"
# include "AtlasHepMC/GenParticle.h"
# include "HepMC3/Data/GenRunInfoData.h"
#include "GeneratorObjects/McEventCollection.h"
 
// AthenaPoolCnvSvc includes
#include "AthenaPoolCnvSvc/T_AthenaPoolTPConverter.h"
 
// GeneratorObjectsTPCnv includes
#include "GeneratorObjectsTPCnv/McEventCollection_p6.h"
 
#include "GaudiKernel/ServiceHandle.h"
 
class IHepMCWeightSvc;
 
// Forward declaration
class MsgStream;
namespace HepMC { struct DataPool; }
 
class McEventCollectionCnv_p6 : public T_AthenaPoolTPCnvBase<
                                          McEventCollection,
                                          McEventCollection_p6
                                       >
{
 
  typedef T_AthenaPoolTPCnvBase<McEventCollection,
                                McEventCollection_p6> Base_t;

  // Public methods:
 public:

  McEventCollectionCnv_p6();

  McEventCollectionCnv_p6( const McEventCollectionCnv_p6& rhs );

  McEventCollectionCnv_p6& operator=( const McEventCollectionCnv_p6& rhs );

  virtual ~McEventCollectionCnv_p6();
 
  void setPileup();

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

  virtual void transToPers( const McEventCollection* transObj,
                            McEventCollection_p6* persObj,
                            MsgStream &log ) ;
 
 protected:
 
  typedef std::unordered_map<HepMC::GenParticlePtr,int> ParticlesMap_t;

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

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

  static void writeGenVertex( const HepMC::ConstGenVertexPtr& vtx, McEventCollection_p6& persEvt );
  static int writeGenParticle( const HepMC::ConstGenParticlePtr& p, McEventCollection_p6& persEvt );
 
  bool m_isPileup;
  ServiceHandle<IHepMCWeightSvc> m_hepMCWeightSvc;
};
#endif //> GENERATOROBJECTSTPCNV_MCEVENTCOLLECTIONCNV_p6_H