AtlfastProtoJetMaker.cxx

Go to the documentation of this file.

// ================================================
// AtlfastProtoJetMaker 
// ================================================
//
// THIS TEXT TO BE REPLACED BY ATLAS STANDARD FORMAT
//
//
// Namespace Atlfast
//

#include "AtlfastAlgs/AtlfastProtoJetMaker.h"
#include "AtlfastAlgs/ISmearer.h"
#include "AtlfastAlgs/JetSmearer.h"
#include "AtlfastAlgs/GlobalEventData.h"

#include "AtlfastUtils/HeaderPrinter.h"
#include "AtlfastUtils/HepMC_helper/IsFinalState.h"

#include "AtlfastEvent/AtlfastProtoJet.h"
#include "AtlfastEvent/ICell.h"
#include "AtlfastEvent/Jet.h"
#include "AtlfastEvent/ICluster.h"
#include "AtlfastEvent/CollectionDefs.h"

#include <cmath>
//#include <iomanip.h>
// CLHEP,HepMC
#include "GeneratorObjects/McEventCollection.h"
// Gaudi includes
#include "GaudiKernel/DataSvc.h"
#include "GaudiKernel/Chrono.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/ISvcLocator.h"
//Generator includes
#include "GeneratorObjects/McEventCollection.h"

namespace Atlfast {

  AtlfastProtoJetMaker:: AtlfastProtoJetMaker(const std::string& name, ISvcLocator* pSvcLocator):
    Algorithm(name,pSvcLocator){
    
    //default paths for entities in the TES
    m_inputCellLocation     = "/Event/AtlfastCells";
    m_inputClusterLocation  = "/Event/AtlfastClusters";

    m_outputTowerPJKey      = "/Event/AtlfastTowerProtoJets";
    m_outputClusterPJKey    = "/Event/AtlfastClusterProtoJets";
    m_outputFinalStatePJKey = "/Event/AtlfastFinalStateProtoJets";


    m_towerPJConstruct      = true;
    m_clusterPJConstruct    = true;
    m_finalStatePJConstruct = true;

    m_towerPJSmearOn = false;
    m_clusterPJSmearOn = false;

    declareProperty("InputCellLocation",      m_inputCellLocation);
    declareProperty("InputClusterLocation",   m_inputClusterLocation);

    declareProperty("OutputTowerPJKey",       m_outputTowerPJKey);
    declareProperty("OutputClusterPJKey",     m_outputClusterPJKey);
    declareProperty("OutputFinalStatePJKey",  m_outputFinalStatePJKey);

    declareProperty("TowerPJConstruct",       m_towerPJConstruct);
    declareProperty("ClusterPJConstruct",     m_clusterPJConstruct);
    declareProperty("FinalStatePJConstruct",  m_finalStatePJConstruct);

    declareProperty("TowerPJSmearOn",         m_towerPJSmearOn);
    declareProperty("ClusterPJSmearOn",       m_clusterPJSmearOn);
  }
  
  //__________________________________________________________________________
  AtlfastProtoJetMaker::~AtlfastProtoJetMaker()
  {
    MsgStream log( messageService(), name() ) ;
    log << MSG::INFO << "AtlfastProtoJetMaker destructor called" << endreq;
  }
  //__________________________________________________________________________
  StatusCode AtlfastProtoJetMaker::initialize()
  {
    
    MsgStream log( messageService(), name() );
    log << MSG::DEBUG <<"AtlfastProtoJetllMaker initialize()" << endreq;
    //get the Global Event Data using singleton pattern
    GlobalEventData* ged = GlobalEventData::Instance();

    int lumi                  = ged->lumi();
    int randSeed              = ged->randSeed() ;
    double barrelForwardEta   = ged->barrelForwardEta();
    m_mcLocation              = ged->mcLocation();  
    m_tesIO = new TesIO(m_mcLocation, ged->justHardScatter());

    m_smearer = new JetSmearer(randSeed, 
                               lumi,
                               0., 
                               0., 
                               barrelForwardEta); 
      
    HeaderPrinter hp("Atlfast-ProtoJet Maker:", log);
    hp.add("Input Cell Key                  ", m_inputCellLocation);    
    hp.add("Input Cluster Key               ", m_inputClusterLocation);    
    hp.add("Output  Tower PJ Key            ", m_outputTowerPJKey);    
    hp.add("Output Cluster PJ Key           ", m_outputClusterPJKey);    
    hp.add("Output FinalState PJ Key        ", m_outputFinalStatePJKey);
    hp.add("Construct Tower ProtoJets       ", m_towerPJConstruct);
    hp.add("Construct Cluster ProtoJets     ", m_clusterPJConstruct);
    hp.add("Construct Final State ProtoJets ", m_finalStatePJConstruct);
    hp.add("Smear Tower  ProtoJets          ", m_towerPJSmearOn);
    hp.add("Smear Cluster ProtoJets         ", m_clusterPJSmearOn);
    hp.print();
    

    return StatusCode::SUCCESS; 
  }
  
  StatusCode AtlfastProtoJetMaker::finalize()
  {
    MsgStream log( messageService(), name() );
    log << MSG::DEBUG <<"AtlfastProtoJetllMaker finalize()" << endreq;
    return StatusCode::SUCCESS; 
  }
  
  //_________________________________________________________________________
  StatusCode AtlfastProtoJetMaker::execute()
  {
    //.............................................
    
    MsgStream log( messageService(), name() );
    std::string message;
    log << MSG::DEBUG << "AtlfastProtoJetMaker execute()" << endreq;

    AtlfastProtoJet* proJ;
     //do protojet filling from now on---------------------
    if(m_towerPJConstruct){
    //Get the cells from storeGate
      std::vector<const ICell*> cells;
      TesIoStat stat = 
        m_tesIO->copy<ITwoCptCellCollection>(cells,m_inputCellLocation);
      message = stat?  "Found Cells in TES":"No Cells in TES";
      log<<MSG::DEBUG << message <<endreq;
      if(stat){
        if(!this->FillProtoJets(cells,proJ,m_outputTowerPJKey,m_towerPJSmearOn)){
          return StatusCode::FAILURE;
        }
      }
    }

    if(m_clusterPJConstruct){
    //Get the clusters from storeGate
      std::vector<ICluster*> clusters;
      TesIoStat stat = 
        m_tesIO->copy<IClusterCollection>(clusters,m_inputClusterLocation);
      message = stat?  "Found Clusters in TES":"No Clusters in TES";
      log<<MSG::DEBUG << message <<endreq;
      if(stat){
        if(!this->FillProtoJets(clusters,proJ,m_outputClusterPJKey,m_clusterPJSmearOn)){
          return StatusCode::FAILURE;
        }
      }
    }
    if(m_finalStatePJConstruct){
    //Get the FinalStateParticles from storeGate
      HepMC_helper::IsFinalState ifs;
      MCparticleCollection particles;
      TesIoStat stat = m_tesIO->getMC(particles, &ifs);
      message = stat?  "Found FS Particles in TES":"No FS Particles in TES";
      log<<MSG::DEBUG << message <<endreq;
      if(stat){
        if(!this->FillProtoJets(particles,proJ,m_outputFinalStatePJKey)){
          return StatusCode::FAILURE;
        }
      }
    }
  return StatusCode::SUCCESS;
  }

 
} // end of namespace bracket

---