McEventCollection.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef GENERATOROBJECTSMCEVENTCOLLECTION_H
#define GENERATOROBJECTSMCEVENTCOLLECTION_H 1
 
 
#include "AthContainers/DataVector.h"
#include "AthenaKernel/CLASS_DEF.h"
#include "AtlasHepMC/GenEvent.h"
#include "AtlasHepMC/GenParticle.h"
#include "AtlasHepMC/GenVertex.h"
#include "AtlasHepMC/GenRanges.h"
 
#include <iostream>
class McEventCollection : public DataVector<HepMC::GenEvent>
{
  typedef DataVector<HepMC::GenEvent>::iterator EventIterator;
  typedef DataVector<HepMC::GenEvent>::const_iterator EventConstIterator;
 
 public:
  McEventCollection ();
 
  ~McEventCollection ();
 
  McEventCollection (const McEventCollection& in);
 
 
  McEventCollection& operator=(const McEventCollection&);
  
  const HepMC::GenEvent* find(int evtNumber) const;
};
 
inline
McEventCollection::McEventCollection (void)
  :     DataVector<HepMC::GenEvent>       ()
{}
 
inline McEventCollection::McEventCollection(const McEventCollection& in) : DataVector<HepMC::GenEvent>()
{
  McEventCollection::operator=(in);
}
 
 
inline McEventCollection& McEventCollection::operator=(const McEventCollection& in)
{
  // Force a deep copy on the DataVector
  //
  for (const HepMC::GenEvent* ev : in)
  {
    HepMC::GenEvent* nev = new HepMC::GenEvent(*ev);
#ifdef HEPMC3
    auto ri = ev->run_info();
    if (ri) {
      std::shared_ptr<HepMC3::GenRunInfo> nri =  std::make_shared<HepMC3::GenRunInfo>(*(ri.get()));
      nev->set_run_info(nri);
    }
    // Fill barcodes attribute in copied GenEvent
    HepMC::fillBarcodesAttribute(nev);
#endif
    DataVector<HepMC::GenEvent>::push_back(nev);
    
  }
 
  return *this;
}
 
inline
McEventCollection::~McEventCollection (void)
{}
 
 
CLASS_DEF(McEventCollection, 133273, 1)
 
#endif