StandardHistogramMaker.cxx

Go to the documentation of this file.

// ================================================
// StandardHistogramMaker class Implementation
// ================================================
//
// THIS TEXT TO BE REPLACED BY ATLAS STANDARD FORMAT
//
// Namespace ATLF::
//

#include "AtlfastAlgs/StandardHistogramMaker.h"
#include "AtlfastAlgs/GlobalEventData.h"
#include "AtlfastEvent/ReconstructedParticle.h"
#include "AtlfastEvent/CollectionDefs.h"
#include "AtlfastEvent/EventHeader.h"
// standard library includes
#include <math.h> 

// Gaudi includes
#include "GaudiKernel/DataSvc.h"
#include "GaudiKernel/IHistogramSvc.h"

namespace Atlfast {

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

StandardHistogramMaker::StandardHistogramMaker 
  ( const std::string& name, ISvcLocator* pSvcLocator ) 
    : Algorithm( name, pSvcLocator )
{




  // Default paths for entities in the TES
  m_electronLocation         = "/Event/AtlfastElectrons";
  m_photonLocation           = "/Event/AtlfastPhotons";
  m_muonLocation             = "/Event/AtlfastNonIsolatedMuons";
  m_isolatedElectronLocation = "/Event/AtlfastIsolatedElectrons";
  m_isolatedPhotonLocation   = "/Event/AtlfastIsolatedPhotons";
  m_isolatedMuonLocation     = "/Event/AtlfastIsolatedMuons";
  m_cellLocation             = "/Event/AtlfastCells";
  m_clusterLocation          = "/Event/AtlfastClusters";
  m_jetLocation              = "/Event/AtlfastJets";
  m_eventHeaderLocation      = "/Event/AtlfastEventHeader";
  m_mcTruthLocation          = "/Event/McEventCollection";


  // Default number at which to start the histograms
  m_histStart                    = 0;

  declareProperty( "ElectronLocation",         m_electronLocation ) ;
  declareProperty( "PhotonLocation",           m_photonLocation );
  declareProperty( "MuonLocation",             m_muonLocation );
  declareProperty( "IsolatedElectronLocation", m_isolatedElectronLocation );
  declareProperty( "IsolatedPhotonLocation",   m_isolatedPhotonLocation );
  declareProperty( "IsolatedMuonLocation",     m_isolatedMuonLocation );
  declareProperty( "CellLocation",             m_cellLocation );
  declareProperty( "ClusterLocation",          m_clusterLocation );
  declareProperty( "JetLocation",              m_jetLocation );
  declareProperty( "EventHeaderLocation",      m_eventHeaderLocation );  
  declareProperty( "McTruthLocation",          m_mcTruthLocation );

  declareProperty( "HistogramStart", m_histStart );


}
StandardHistogramMaker::~StandardHistogramMaker(){}

//---------------------------------
// initialise() 
//---------------------------------

StatusCode StandardHistogramMaker::initialize()
{
  // .. put any initialisation in here...
  MsgStream log(messageService(), name());
  log << MSG::DEBUG << "in initialize()" << endreq;
  
  m_nHist = m_histStart;
  this->bookNonIsolatedElectronHistograms();
  this->bookIsolatedElectronHistograms();
  this->bookNonIsolatedPhotonHistograms();
  this->bookIsolatedPhotonHistograms();
  this->bookNonIsolatedMuonHistograms();
  this->bookIsolatedMuonHistograms();
  this->bookCellHistograms();
  this->bookClusterHistograms();
  this->bookJetHistograms();
  this->bookEventHeaderHistograms();

  log << MSG::DEBUG << "Booked histograms OK" << endreq;

  GlobalEventData* ged = GlobalEventData::Instance();
  // load the location of the MC in StoreGate
  m_mcLocation       = ged -> mcLocation();


  //  m_tesIO = new TesIO(eventDataService());
  m_tesIO = new TesIO(m_mcLocation, ged->justHardScatter());
  return StatusCode::SUCCESS ;
}



//---------------------------------
// finalise() 
//---------------------------------

StatusCode StandardHistogramMaker::finalize()
{
  // .. put any finalisation in here...
  delete m_tesIO;
  return StatusCode::SUCCESS ;
}




//----------------------------------------------
// execute() method called once per event
//----------------------------------------------

StatusCode StandardHistogramMaker::execute( )
{

  //................................
  // make a message logging stream

  MsgStream log( messageService(), name() ) ;
  log << MSG::DEBUG << "Executing " << endreq;



  this->kinHistogramFiller<
    ReconstructedParticleCollection>(
                                     m_electronLocation,
                                     m_h_electronMultiplicity,
                                     m_h_electronPx,
                                     m_h_electronPy,
                                     m_h_electronPz,
                                     m_h_electronE,
                                     m_h_electronPt
                                     );

  this->kinHistogramFiller<
    ReconstructedParticleCollection>(
                                     m_isolatedElectronLocation,
                                     m_h_isolatedElectronMultiplicity,
                                     m_h_isolatedElectronPx,
                                     m_h_isolatedElectronPy,
                                     m_h_isolatedElectronPz,
                                     m_h_isolatedElectronE,
                                     m_h_isolatedElectronPt
                                     );

  this->kinHistogramFiller<
    ReconstructedParticleCollection>(
                                     m_photonLocation,
                                     m_h_photonMultiplicity,
                                     m_h_photonPx,
                                     m_h_photonPy,
                                     m_h_photonPz,
                                     m_h_photonE,
                                     m_h_photonPt
                                     );


  
  this->kinHistogramFiller<
    ReconstructedParticleCollection>(
                                     m_isolatedPhotonLocation,
                                     m_h_isolatedPhotonMultiplicity,
                                     m_h_isolatedPhotonPx,
                                     m_h_isolatedPhotonPy,
                                     m_h_isolatedPhotonPz,
                                     m_h_isolatedPhotonE,
                                     m_h_isolatedPhotonPt
                                     );

  
  this->kinHistogramFiller<
    ReconstructedParticleCollection>(
                                     m_muonLocation,
                                     m_h_muonMultiplicity,
                                     m_h_muonPx,
                                     m_h_muonPy,
                                     m_h_muonPz,
                                     m_h_muonE,
                                     m_h_muonPt
                                     );

    
   this->kinHistogramFiller<
     ReconstructedParticleCollection>(
                                      m_isolatedMuonLocation,
                                      m_h_isolatedMuonMultiplicity,
                                      m_h_isolatedMuonPx,
                                      m_h_isolatedMuonPy,
                                      m_h_isolatedMuonPz,
                                      m_h_isolatedMuonE,
                                      m_h_isolatedMuonPt
                                      );



  this->kinHistogramFiller<ITwoCptCellCollection>(m_cellLocation,
                                                  m_h_cellMultiplicity,
                                                  m_h_cellPx,
                                                  m_h_cellPy,
                                                  m_h_cellPz,
                                                  m_h_cellE,
                                                  m_h_cellPt
                                                  );
  
  this->kinHistogramFiller<IClusterCollection>(m_clusterLocation,
                                               m_h_clusterMultiplicity,
                                               m_h_clusterPx,
                                               m_h_clusterPy,
                                               m_h_clusterPz,
                                               m_h_clusterE,
                                               m_h_clusterPt
                                               );
  
  this->kinHistogramFiller<JetCollection>(m_jetLocation,
                                          m_h_jetMultiplicity,
                                          m_h_jetPx,
                                          m_h_jetPy,
                                          m_h_jetPz,
                                          m_h_jetE,
                                          m_h_jetPt
                                          );
  


  /*    
  const EventHeader* eventHeader(0);
  if(!(m_tesIO->getDH(eventHeader))){
    log << MSG::ERROR<<"Could not find the event header in the TES"<<endreq;
    }
  
  if(eventHeader){
    m_h_njetb->fill(float(eventHeader->nBJets()), 1.0);
    m_h_njetc->fill(float(eventHeader->nCJets()), 1.0);
    m_h_njettau->fill(float(eventHeader->nTauJets()), 1.0);
    m_h_pxmiss->fill(float(eventHeader->pMiss().x()), 1.0);
    m_h_pymiss->fill(float(eventHeader->pMiss().y()), 1.0);
    m_h_pxnue->fill(float(eventHeader->pEscaped().x()), 1.0);
    m_h_pynue->fill(float(eventHeader->pEscaped().y()), 1.0);
    
    
    HepMC::WeightContainer weights = eventHeader->firstFewWeights(3);
    m_h_mcWeight0->fill(float(weights[0]), 1.0);
    m_h_mcWeight1->fill(float(weights[1]), 1.0);
    m_h_mcWeight2->fill(float(weights[2]), 1.0);
  } else {
    log << MSG::ERROR << " Invalid EventHeader data handle" << endreq;
  }
  
  */
  return StatusCode::SUCCESS ;
}

void StandardHistogramMaker::bookNonIsolatedElectronHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking electron histograms" << endreq;

  m_h_electronMultiplicity = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Electron Multiplicity",10,0.0,10.0);

  m_h_electronPx = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Electron Px",10,0.0,100000.0);
  m_h_electronPy = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Electron Py",10,0.0,100000.0);
  m_h_electronPz = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Electron Pz",10,0.0,100000.0);
  m_h_electronE = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Electron  E",10,0.0,100000.0);
  m_h_electronPt = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Electron Pt",10,0.0,100000.0);

  //MC truth electrons
  

  return;
}


void StandardHistogramMaker::bookIsolatedElectronHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking isolated electron histograms" << endreq;

  m_h_isolatedElectronMultiplicity = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Electron Multiplicity",10,0.0,10.0);

  m_h_isolatedElectronPx = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Electron Px",10,0.0,100000.0);

  m_h_isolatedElectronPy = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Electron Py",10,0.0,100000.0);

  m_h_isolatedElectronPz = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Electron Pz",10,0.0,100000.0);

  m_h_isolatedElectronE = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Electron  E",10,0.0,100000.0);

  m_h_isolatedElectronPt = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Electron Pt",10,0.0,100000.0);
  return;
}


void StandardHistogramMaker::bookNonIsolatedPhotonHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking photon histograms" << endreq;

  m_h_photonMultiplicity = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Photon Multiplicity",20,0.0,20.0);

  m_h_photonPx = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Photon Px",10,0.0,100000.0);

  m_h_photonPy = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Photon Py",10,0.0,100000.0);

  m_h_photonPz = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Photon Pz",10,0.0,100000.0);

  m_h_photonE = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Photon  E",10,0.0,100000.0);

  m_h_photonPt = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Photon Pt",10,0.0,100000.0);
  return;
}


void StandardHistogramMaker::bookIsolatedPhotonHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking isolated photon histograms" << endreq;

  m_h_isolatedPhotonMultiplicity = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Photon Multiplicity",20,0.0,20.0);


  m_h_isolatedPhotonPx = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Photon Px",10,0.0,100000.0);

  m_h_isolatedPhotonPy = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Photon Py",10,0.0,100000.0);

  m_h_isolatedPhotonPz = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Photon Pz",10,0.0,100000.0);

  m_h_isolatedPhotonE = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Photon E",10,0.0,100000.0);

  m_h_isolatedPhotonPt = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Photon Pt",10,0.0,100000.0);
  return;
}


void StandardHistogramMaker::bookNonIsolatedMuonHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking muon histograms" << endreq;

  m_h_muonMultiplicity = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Muon Multiplicity",20,0.0,20.0);

  m_h_muonPx = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Muon Px",10,0.0,100000.0);

  m_h_muonPy = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Muon Py",10,0.0,100000.0);

  m_h_muonPz = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Muon Pz",10,0.0,100000.0);

  m_h_muonE = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Muon  E",10,0.0,100000.0);

  m_h_muonPt = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Muon Pt",10,0.0,100000.0);
  return;
}


void StandardHistogramMaker::bookIsolatedMuonHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking isolated isolated muon histograms" << endreq;

  m_h_isolatedMuonMultiplicity = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Muon Multiplicity",20,0.0,20.0);


  m_h_isolatedMuonPx = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Muon Px",10,0.0,100000.0);

  m_h_isolatedMuonPy = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Muon Py",10,0.0,100000.0);

  m_h_isolatedMuonPz = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Muon Pz",10,0.0,100000.0);

  m_h_isolatedMuonE = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Muon E",10,0.0,100000.0);

  m_h_isolatedMuonPt = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Isolated Muon Pt",10,0.0,100000.0);
  return;
}

void StandardHistogramMaker::bookCellHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking Cell histograms" << endreq;

  m_h_cellMultiplicity = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cell Multiplicity",50,0.0,1000.0);

  m_h_cellPx = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cell Px",10,0.0,20000.0);

  m_h_cellPy = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cell Py",10,0.0,20000.0);

  m_h_cellPz = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cell Pz",10,0.0,100000.0);

  m_h_cellE = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cell E",10,0.0,100000.0);

  m_h_cellPt = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cell Pt",10,0.0,50000.0);

  return;
}

void StandardHistogramMaker::bookClusterHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking Cluster histograms" << endreq;

  m_h_clusterMultiplicity = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cluster Multiplicity",20,0.0,20.0);

  m_h_clusterPx = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cluster Px",10,0.0,100000.0);

  m_h_clusterPy = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cluster Py",10,0.0,100000.0);

  m_h_clusterPz = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cluster Pz",10,0.0,100000.0);

  m_h_clusterE = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cluster E",10,0.0,100000.0);

  m_h_clusterPt = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Cluster Pt",10,0.0,100000.0);

  return;
}

void StandardHistogramMaker::bookJetHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking Jet histograms" << endreq;

  m_h_jetMultiplicity = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Jet Multiplicity",20,0.0,20.0);

  m_h_jetPx = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Jet Px",10,0.0,100000.0);

  m_h_jetPy = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Jet Py",10,0.0,100000.0);

  m_h_jetPz = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Jet Pz",10,0.0,100000.0);

  m_h_jetE = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Jet E",50,0.0,500000.0);

  m_h_jetPt = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Jet Pt",20,0.0,200000.0);

  return;
}

void StandardHistogramMaker::bookEventHeaderHistograms () {
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "Booking Event Header histograms" << endreq;

  m_h_njetb = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Event Header njetb",20,0.0,20.0);

  m_h_njetc = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Event Header njetc",20,0.0,20.0);

  m_h_njettau = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Event Header njettau",20,0.0,20.0);

  m_h_pxmiss = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Event Header pxmiss",20,0.0,100000.);

  m_h_pymiss = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Event Header pymiss",20,0.0,100000.);


  m_h_pxnue = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Event Header pxnue",20,0.0,100000.);

  m_h_mcWeight0 = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Event Header mcWeight0",10,0.0,1.);

  m_h_mcWeight1 = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Event Header mcWeight1",10,0.0,1.);

  m_h_mcWeight2 = 
    histoSvc()->book("/stat/simple/",
                     ++m_nHist,"Event Header mcWeight2",10,0.0,1.);

  return;
}






} // end of namespace bracket

---