---

TesIO.cxx

Go to the documentation of this file.

#include "AtlfastUtils/TesIO.h"
#include <cmath> 
#include <algorithm>
#include "AtlfastUtils/FunctionObjects.h"
#include "GeneratorObjects/McEventCollection.h"
#include "AtlfastUtils/HepMC_helper/IMCselector.h"
#include "AtlfastUtils/HepMC_helper/All.h"
//#include "AtlfastCode/HepMC_helper/MCselectorWrapper.h"
#include "AtlfastEvent/ReconstructedParticleCollection.h"
#include "HepMC/GenEvent.h"
#include "HepMC/GenParticle.h"


namespace Atlfast {

  TesIoStat TesIO::getMC(MCparticleCollection& mcParticles) const {
    const HepMC_helper::IMCselector* selector = new HepMC_helper::All();
    TesIoStat stat = this->getMC(mcParticles, selector);
    return stat;
  }


  TesIoStat TesIO::getMC(MCparticleCollection& mcParticles, 
                        const HepMC_helper::IMCselector* selector) const {
    
    const DataHandle<McEventCollection> mcCollptr;
    TesIoStat stat = this->getDH(mcCollptr);
    if(!stat){
      return TesIoStat(StatusCode::FAILURE,"Could not find MC ptcles in TES");
    }
      
    //    HepMC_helper::MCselectorWrapper wrappedSelector(selector);
    
    // Iterate over all McEvent records
    McEventCollection::const_iterator itr;
    for (itr = mcCollptr->begin(); itr!=mcCollptr->end(); ++itr) {
      // Access the HepMC record which is wrapped within McEvent
      HepMC::GenEvent* genEvt = (*itr)->pGenEvt()  ;
      if(genEvt == 0) return TesIoStat(StatusCode::FAILURE,
                                       "Could not find HEP Gen event");
      
      // Copy all final state particles of the correct type into the 
      // collection
      
      //      HepMC::copy_if( genEvt->particles_begin(), 
      //              genEvt->particles_end(),
      //              back_inserter( mcParticles),
      //              wrappedSelector
      //              );
      //      HepMC::copy_if( genEvt->particles_begin(), 
      //              genEvt->particles_end(),
      //              back_inserter( mcParticles),
      //              *(selector->create())
      //              );
            GenEvent::particle_const_iterator it= genEvt->particles_begin();
            GenEvent::particle_const_iterator en= genEvt->particles_end();
            for(; it!=en; ++it){
              if((*selector)(*it)){mcParticles.push_back(*it);}
            }
    }
    
    return TesIoStat(StatusCode::SUCCESS);
    
  }
}//namespace

---