McEventCollectionCnv_p6.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2022 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>
 
#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_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;

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