TrackDumper.cxx

Go to the documentation of this file.

// ================================================
// TrackDumper class Implementation
// ================================================
//
// THIS TEXT TO BE REPLACED BY ATLAS STANDARD FORMAT
//
// Namespace Atlfast::
//
#include "AtlfastAlgs/TrackDumper.h"
#include "AtlfastAlgs/GlobalEventData.h"
#include "AtlfastEvent/Track.h"
#include "AtlfastEvent/CollectionDefs.h"


// Gaudi includes
#include "GaudiKernel/DataSvc.h"
#include "GaudiKernel/MsgStream.h"
#include <iostream>
#include <iomanip>
//--------------------------------
// Constructors and destructors
//--------------------------------

namespace Atlfast {
  
  class TrackPrinter{
  public:
    TrackPrinter(MsgStream& log):m_count(0), m_log(log){}
    void operator()(const Track* t){
      //      cout<<setw(10)<<++m_count<<" "<<*t<<endl;
      m_log<< MSG::INFO <<++m_count<<" "<<*t<<endreq;
    }
  private:
    int m_count;
    MsgStream& m_log;
  };

  TrackDumper::TrackDumper (const std::string& name, 
                            ISvcLocator* pSvcLocator ) 
    : Algorithm( name, pSvcLocator ),
      m_tesIO(0) {
    
    // Default paths for entities in the TES
    m_inputLocation    = "/Event/AtlfastTracks";
    declareProperty( "InputLocation",    m_inputLocation ) ;
  }
  
  //----------------------------
  // Destructor
  //----------------------------
  TrackDumper::~TrackDumper() {
    if (m_tesIO) delete m_tesIO;
  } 
  
  
  //---------------------------------
  // initialise() 
  //---------------------------------
  
  StatusCode TrackDumper::initialize(){
    MsgStream log( messageService(), name() ) ;
 
    GlobalEventData* ged = GlobalEventData::Instance();
    //int randSeed = ged->randSeed() ;
    // load the location of the MC in StoreGate
    m_mcLocation       = ged -> mcLocation();
   
    m_tesIO = new TesIO(m_mcLocation, ged->justHardScatter());

    HeaderPrinter hp("Atlfast TrackDumper:", log);
    hp.add("Input Location         ", m_inputLocation);    
    hp.print();
    log << MSG::INFO << "Initialised successfully " << endreq ;    
    return StatusCode::SUCCESS ;
  }
  
  //---------------------------------
  // finalise() 
  //---------------------------------
  
  StatusCode TrackDumper::finalize(){
    
    MsgStream log( messageService(), name() ) ;
    
    log << MSG::INFO << "Finalised successfully " << endreq ;
    
    return StatusCode::SUCCESS ;
  }
  
  
  //----------------------------------------------
  // execute() method called once per event
  //----------------------------------------------
  //
  //  This algorithm creates smeared Tracks passing cuts. 
  //  It scans the list of HepMC::GenParticles from the Monte Carlo truth. 
  //  It creates a Track object by smearing the Trajectory track parameters
  //  corresponding to the HepMC::GenParticle. 
  // 
  //  It then applies kinematic criteria on the properties of the Track
  //  Those which pass the cuts are kept.
  //  Finally all successful ReconstructedParticles are added to 
  //the event store.
  //
  
  StatusCode TrackDumper::execute( ){

    //................................
    // make a message logging stream
    
    MsgStream log( messageService(), name() ) ;
    log << MSG::DEBUG << "Execute() " << endreq;
    

    //.........................................................
    // We now need to access the HepMC event record(s) in the event store
    // and iterate over them to extract HepMC::GenParticles.
    // This method will only select the particles which are required
    // and add them to a local store (mcParticles)
    
    std::vector<Track*> tracks;
    
    m_tesIO->copy<TrackCollection>( tracks, m_inputLocation );
    TrackPrinter printer(log);
    HeaderPrinter hp("TrackDump:", log);
    std::for_each(tracks.begin(), tracks.end(), printer);
    return StatusCode::SUCCESS;
  }
} // end namespace bracket

---