---

IKinematicDumper.cxx

Go to the documentation of this file.

// ================================================
//IKinematicDumper class Implementation
// ================================================
//
// THIS TEXT TO BE REPLACED BY ATLAS STANDARD FORMAT
//
// Namespace Atlfast::
//
#include "AtlfastAlgs/IKinematicDumper.h"
#include "AtlfastEvent/Cluster.h"
#include "AtlfastEvent/Jet.h"
#include "AtlfastEvent/CollectionDefs.h"
#include "AtlfastEvent/ReconstructedParticle.h"
#include "AtlfastUtils/FunctionObjects.h"

#include <cmath> 
#include <algorithm>
#include <iomanip.h>
// Gaudi includes
#include "GaudiKernel/DataSvc.h"

//--------------------------------
// Constructors and destructors
//--------------------------------

namespace Atlfast {
  using std::cout;
  using std::endl;
  using std::ios;

  IKinematicDumper::IKinematicDumper
  ( const std::string& name, ISvcLocator* pSvcLocator ) 
    : Algorithm( name, pSvcLocator ){
    
    
    // Default paths for entities in the TES
    m_clusterLocation    = "/Event/AtlfastClusters";
    m_jetLocation        = "/Event/AtlfastJets";
    m_imuonLocation      = "/Event/AtlfastIsolatedMuons";
    m_nimuonLocation      = "/Event/AtlfastNonIsolatedMuons";

    declareProperty( "clusterLocation", m_clusterLocation ) ;
    declareProperty( "jetLocation",     m_jetLocation ) ;
    declareProperty( "imuonLocation",   m_imuonLocation ) ;
    declareProperty( "nimuonLocation",  m_nimuonLocation ) ;
    
  }
  
  
  IKinematicDumper::~IKinematicDumper() {} 
  
  
  //---------------------------------
  // initialise() 
  //---------------------------------
  
  StatusCode IKinematicDumper::initialize()
  {
    MsgStream log( messageService(), name() ) ;
    log<<MSG::DEBUG<<"Initialising "<<endreq;
    //set up TesIO
    //    m_tesIO = new TesIO(eventDataService());
    m_tesIO = new TesIO();
    
    return StatusCode::SUCCESS ;
  }
  
  //---------------------------------
  // finalise() 
  //---------------------------------
  
  StatusCode IKinematicDumper::finalize()
  {
    // .. put any finalisation in here...
    
    return StatusCode::SUCCESS ;
  }
  
  
  //----------------------------------------------
  // execute() method called once per event
  //----------------------------------------------
  //
  
  StatusCode IKinematicDumper::execute( )
  {
    //................................
    // make a message logging stream
    
    MsgStream log( messageService(), name() ) ;
    log<<MSG::DEBUG<<"Executing "<<endreq;
    


    //IsolatedMuons: read from TES and dump
    std::vector<ReconstructedParticle*> imuons;
    m_tesIO->copy<ReconstructedParticleCollection>(imuons, m_imuonLocation);
    std::sort(imuons.begin(), imuons.end(), SortAttribute::AscendingEta());
    log<<MSG::INFO<<"Dumping  Isolated Muons "<<imuons.size()<<endreq;
    log<<MSG::INFO<<"----------------------- "<<endreq;    
    std::vector<ReconstructedParticle*>::const_iterator 
      rp_iter = imuons.begin();
    this->header();
    for(;rp_iter!=imuons.end();++rp_iter) this->dump(*rp_iter);

    //NonIsolatedMuons: read from TES and dump
    std::vector<ReconstructedParticle*> nimuons;
    m_tesIO->copy<ReconstructedParticleCollection> (nimuons, m_nimuonLocation);
    std::sort(nimuons.begin(), nimuons.end(), SortAttribute::AscendingEta());
    log<<MSG::INFO<<"Dumping  NonIsolated Muons "<<nimuons.size()<<endreq;
    log<<MSG::INFO<<"-------------------------- "<<endreq;    
    rp_iter = nimuons.begin();
    this->header();
    for(;rp_iter!=nimuons.end();++rp_iter) this->dump(*rp_iter);
    //Clusters: read from TES and dump
    std::vector<Cluster*> clusters;
    m_tesIO->copy<ClusterCollection> (clusters, m_clusterLocation);
    std::sort(clusters.begin(), clusters.end(), SortAttribute::AscendingEta());
    log<<MSG::INFO<<"Dumping  Clusters "<<clusters.size()<<endreq;
    log<<MSG::INFO<<"------------------- "<<endreq;    
    std::vector<Cluster*>::const_iterator cl_iter = clusters.begin();
    this->header();
    for(;cl_iter!=clusters.end();++cl_iter) this->dump(*cl_iter);
    //Jets: read from TES and dump
    std::vector<Jet*> jets;
    m_tesIO->copy<JetCollection>(jets, m_jetLocation);
    std::sort(jets.begin(), jets.end(), SortAttribute::AscendingEta());
    log<<MSG::INFO<<"Dumping  Jets "<<jets.size()<<endreq;
    log<<MSG::INFO<<"------------------- "<<endreq;    
    std::vector<Jet*>::const_iterator jet_iter = jets.begin();
    this->header();
    for(;jet_iter!=jets.end();++jet_iter) this->dump(*jet_iter);

    log<<MSG::INFO<<"End of execute "<<endreq;    
    return StatusCode::SUCCESS;
  }
  
  void IKinematicDumper::header()const{
    cout<<"     eta      phi       pt        e"<<endl;
    cout<<"-----------------------------------"<<endl;
  }
  void IKinematicDumper::dump(const IKinematic* p)const{
    cout<<setiosflags(ios::fixed);
    cout<<setprecision(3);
    cout
        <<setw(8)<<setprecision(3)<<p->momentum().pseudoRapidity()<<" "
        <<setw(8)<<setprecision(3)<<p->momentum().phi()<<" "
        <<setw(8)<<setprecision(3)<<p->momentum().perp()<<" "
        <<setw(8)<<setprecision(3)<<p->momentum().e()<<" "
        <<endl;
  }
  //  template <class T>
  //  std::vector<T*> assocs(Jet j, T type){
  //    std::vector<T*> v = j->associations(T);
  //    return v;
};  // end of namespace bracket

---