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 "AtlfastEvent/CollectionDefs.h"
#include "HepMC/GenEvent.h"
#include "HepMC/GenParticle.h"

namespace Atlfast {


  //Wrapper class for  polymorphic selectors:  STL
  //passes function objects by value...(Meyers, Effective STL item 38)
  
  class SelectorBridge{
  public:
    SelectorBridge(const HepMC_helper::IMCselector* s):m_selector(s){}
    bool operator()(const GenParticle* p) {
      return m_selector->operator()(p);
    }
  private:
    const HepMC_helper::IMCselector* m_selector;
  };
  
  TesIoStat TesIO::getMC(MCparticleCollection& mcParticles) const {
    HepMC_helper::All selector;
    TesIoStat stat = this->getMC(mcParticles, &selector);
    return stat;
  }

  
  TesIoStat TesIO::getMC(MCparticleCollection& mcParticles, 
                         const HepMC_helper::IMCselector* selector) const {
    
    const McEventCollection* mcCollptr(0);
    TesIoStat stat;
    if(m_mcLocation==""){
      // std::cout<<"MCLOCATION1=BLANK"<<std::endl;
      stat = this->getDH(mcCollptr);
    }else{
      // std::cout<<"MCLOCATION1=NONBLANK"<<std::endl;
      stat = this->getDH(mcCollptr,m_mcLocation);
    }
    
    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) {
      
      // For 12.0.4, GenEvents are not aware of their status (hard scatter,
      // pileup, etc.. The following line ensures only the first event in the 
      // McEventCollection is considered until the situation changes and 
      // IsFromHardScatter can do more than just return true!
      
      if ( m_justHardScatter && itr != mcCollptr->begin() ) continue;
      
      //16/10/03 Genevt has been removed
      // Access the HepMC record which is wrapped within McEvent
      //16/10/03      HepMC::GenEvent* genEvt = (*itr)->pGenEvt()  ;
      //16/10/03      if(genEvt == 0) return TesIoStat(StatusCode::FAILURE,
      //16/10/03                                       "Could not find HEP Gen event");
      
      
      
      /* idiot test - range inserter DOES NOT WORL!!    
      HepMC::GenEvent::particle_const_iterator j= (*itr)->particles_begin();
      HepMC::GenEvent::particle_const_iterator jj= (*itr)->particles_end();
      MCparticleCollection junk(j, jj);
      MCparticleCollection junk1;
      for(;j!=jj;++j){junk1.push_back(*j);}
      */
      //Copy ALL particles into mcParticles

        /*
       mcParticles.insert(mcParticles.end(),
                          (*itr)->particles_begin(),
                          (*itr)->particles_end()                        
                          );
        */

      //partition moves the unselected particles to the back of
      //the vector. Use erase to regain space and reset the constianer end.
      
      //mcParticles.erase(
      //                std::partition(mcParticles.begin(),
      //                               mcParticles.end(),
      //                               SelectorBridge(selector)
      //                               ),
      //                mcParticles.end()
      //                );
      
      
    
      
      HepMC::GenEvent::particle_const_iterator it= (*itr)->particles_begin();
      HepMC::GenEvent::particle_const_iterator en= (*itr)->particles_end();
      for(; it!=en; ++it){
        if((*selector)(*it)){mcParticles.push_back(*it);}
      }
      
    }
    return TesIoStat(StatusCode::SUCCESS);
  }
  TesIoStat 
  TesIO::getMCweightContainers(MCweightContainerCollection& 
                               mcWeightContainers)const {
    
    const McEventCollection* mcCollptr(0);
    TesIoStat stat;
    if(m_mcLocation==""){
      // std::cout<<"MCLOCATION2=BLANK"<<std::endl;
    stat = this->getDH(mcCollptr);
    }else{
    stat = this->getDH(mcCollptr, m_mcLocation);
    // std::cout<<"MCLOCATION2=NONBLANK"<<std::endl;
    }
    if(!stat){
      return TesIoStat(StatusCode::FAILURE,"Could not find MCEventCollection");
    }
      
    //    HepMC_helper::MCselectorWrapper wrappedSelector(selector);
    
    // Iterate over all McEvent records
    McEventCollection::const_iterator itr;
    for (itr = mcCollptr->begin(); itr!=mcCollptr->end(); ++itr) {

      //having a real hard time (as usual) with HepMC containers
      //intead, do a little copying...
      
      MCweightContainer container( ((*itr)->weights()).begin(), 
                                   ((*itr)->weights()).end()
                                   ); 
      mcWeightContainers.push_back( container );
    }
      
    return TesIoStat(StatusCode::SUCCESS);
  }
}//namespace

---