PrintMcInfo.cxx

Go to the documentation of this file.

#include "PrintMcInfo/PrintMcInfo.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/IHistogramSvc.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/PropertyMgr.h"
#include "GaudiKernel/SmartDataLocator.h"
#include "GaudiKernel/SmartDataPtr.h"
 
#include "EventModel/Event.h"
#include "EventModel/EventModel.h"
#include "McTruth/McParticle.h"
#include "TMath.h"
#include <fstream>
 
#include "EventModel/EventHeader.h"
#include "EvtRecEvent/EvtRecEvent.h"
#include "EvtRecEvent/EvtRecTrack.h"
 
#include <string>
using namespace std;
 
DECLARE_COMPONENT( PrintMcInfo )
//------------------
PrintMcInfo::PrintMcInfo( const std::string& name, ISvcLocator* pSvcLocator )
    : Algorithm( name, pSvcLocator ) {
 
  declareProperty( "PrintLevel", m_OutputLevel = 0 );
  declareProperty( "FileName", m_FileName );
  declareProperty( "fstreamOption", m_fstreamOption = 1 );
}
PrintMcInfo::~PrintMcInfo() {}
 
StatusCode PrintMcInfo::initialize() {
  // make a map for tanslating the PDG code to its name
  PrintMcInfo::mkmap();
 
  if ( m_fstreamOption == 0 ) { os.open( m_FileName.c_str(), ios::app ); }
  else os.open( m_FileName.c_str() );
 
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "PrintMcInfo::initialize()" << endmsg;
 
  StatusCode status;
  return StatusCode::SUCCESS;
}
StatusCode PrintMcInfo::beginRun() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "PrintMcInfo::beginRun()" << endmsg;
  // add your code here
  return StatusCode::SUCCESS;
}
 
//--------------------execute----------------
 
StatusCode PrintMcInfo::execute() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "PrintMcInfo::execute()" << endmsg;
 
  /*    if(m_fstreamOption==0)
          {
                  os.open(m_FileName.c_str(),ios::app);
          }
          else    os.open(m_FileName.c_str());
  */
  SmartDataPtr<Event::EventHeader> eventHeader( eventSvc(), "/Event/EventHeader" );
  int                              runNo = eventHeader->runNumber();
  int                              event = eventHeader->eventNumber();
  os << "run number:" << runNo << endl << "event number:" << event << endl << endl;
 
  SmartDataPtr<Event::McParticleCol> mcPartCol( eventSvc(), "/Event/MC/McParticleCol" );
  if ( runNo < 0 )
  {
    if ( !mcPartCol )
    {
      log << MSG::ERROR << "Could not retrieve McParticelCol" << endmsg;
      return StatusCode::FAILURE;
    }
 
    Event::McParticleCol::iterator iter_mc = mcPartCol->begin();
    for ( ; iter_mc != mcPartCol->end(); iter_mc++ )
    {
      bool primary = ( *iter_mc )->primaryParticle();
      bool gamma   = ( ( *iter_mc )->particleProperty() ) == 22 ? true : false;
      if ( primary && ( !gamma ) )
      {
        Event::McParticle* pMcPart = ( *iter_mc );
        pMcPart->trackIndex();
        PrintMcInfo::printTree( os, pMcPart, m_OutputLevel, 0 );
        if ( m_OutputLevel > 0 )
        {
          os << endl
             << "\n- - - - - - - - - - - - - -  Information in this decay tree :   particle "
                "level  - - - - - - - - - - - - - -- - -- - -\n"
             << endl;
          PrintMcInfo::printTitle( os, 1 );
          PrintMcInfo::printPartInf( os, pMcPart, 1, 0 );
        }
        if ( m_OutputLevel > 1 )
        {
          os << endl
             << "\n- - - - - - - - - - - - - -  Information in this decay tree :      hit "
                "level  - - - - - - - - - - - - - - - - - - - -\n"
             << endl;
          PrintMcInfo::printPartInf( os, pMcPart, 2, 0 );
        }
      }
    }
    if ( m_OutputLevel > 2 )
    {
      os << endl
         << "\n- - - - - -  - - - - - - - -  Digi Information in this EVENT - - - - - - - - - "
            "- - - - - - - - -  -\n"
         << endl;
      printDigi( os );
    }
 
    os << endl
       << "\n********************************   end of this event   "
          "*********************************\n"
       << endl;
 
  } // end of if(runNo < 0)
  return StatusCode::SUCCESS;
}
 
StatusCode PrintMcInfo::finalize() {
  MsgStream log( msgSvc(), name() );
  log << MSG::INFO << "PrintMcInfo::finalize()" << endmsg;
  return StatusCode::SUCCESS;
}