FinalStateParticleHistogramMaker.h

Go to the documentation of this file.

// ================================================
// FinalStateParticleHistogramMaker class description
// ================================================
//
// THIS TEXT TO BE REPLACED BY ATLAS STANDARD FORMAT
//
//
// This version....
//
//
// Namespace Atlfast::
//
// class: FinalStateParticleHistogramMaker
//
// Description: 
//
//  formatted output of collection of Reconstructed Particles from the TES
//  for debugging purposes
// ................................................................
//

#ifndef ATLFAST_FINALSTATERPARTICLEHISTOGRAMMAKER_H
#define ATLFAST_FINALSTATERPARTICLEHISTOGRAMMAKER_H

// STL
#include <vector>
#include <string>
#include <utility>
#include <map>

// Gaudi 
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/IAlgorithm.h"
#include "GaudiKernel/Algorithm.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"

#include "AIDA/IHistogram1D.h"

// Other
#include "CLHEP/Vector/LorentzVector.h"
#include "HepMC/GenVertex.h"

// Atlfast
#include "AtlfastEvent/Cell.h"
#include "AtlfastEvent/Cell.h"
#include "AtlfastEvent/CollectionDefs.h"
#include "AtlfastEvent/SimpleKinematic.h"
#include "AtlfastEvent/ParticleCodes.h"

#include "AtlfastUtils/TesIO.h"
#include "AtlfastUtils/FunctionObjects.h"
#include "AtlfastUtils/HepMC_helper/HepMC_helper.h"

#include "AtlfastAlgs/GlobalEventData.h"


namespace HepMC_helper{
  class IMCselector;
  class MCselectorWrapper;
}

namespace Atlfast {

  using std::string;
  using std::pair;
  using std::vector;
  using std::map;

  class FillHistograms{
    
  public:
    FillHistograms(TesIO* tesIOPtr, 
                   MsgStream &log, 
                   map< string, vector<IHistogram1D*> > &histContainer):
      m_tesIO(tesIOPtr), 
      m_log(log), 
      m_histogramContainer(histContainer){}
    void operator()(pair< string, HepMC_helper::MCselectorWrapper*> &pairPtr);
    
  private:

    void Fill(MCparticleCollection&, string&);

    TesIO* m_tesIO;
    MsgStream &m_log;
    map< string, vector<IHistogram1D*> > &m_histogramContainer;
    
  };
    
  class FinalStateParticleHistogramMaker : public Algorithm {
    
  public:
    
    //-------------------------
    // Constructors/Destructors
    //
    // Gaudi requires that the constructor takes certain arguments
    // (and passes them directly to the constructor of the base class)
    //-------------------------
    FinalStateParticleHistogramMaker( const std::string& name, ISvcLocator* pSvcLocator ) ; 
    virtual ~FinalStateParticleHistogramMaker();
    
    
    //------------------------------------------------------
    // Methods used by Gaudi to run the algorithm
    //------------------------------------------------------
    virtual StatusCode initialize() ;
    virtual StatusCode execute() ;
    virtual StatusCode finalize() ;
    
    
    
  private:

    void PutIntoContainer(string&, HepMC_helper::MCselectorWrapper*); 
    void BookHistograms(string&);
    void SetHistogramLimits();

    void AllSelectorSetups(string&,HepMC_helper::MCselectorWrapper*);
    void Z0SelectorSetups(string&,HepMC_helper::MCselectorWrapper*);
    void VisibleToCalSelectorSetups(string&,HepMC_helper::MCselectorWrapper*,GlobalEventData*);
    void InvisibleToAtlasSelectorSetups(string &,HepMC_helper::MCselectorWrapper*,GlobalEventData*);
    void IsFinalStateSelectorSetups(string&,HepMC_helper::MCselectorWrapper*);
    void BSelectorSetups(string&,HepMC_helper::MCselectorWrapper*);
    void ElectronSelectorSetups(string&,HepMC_helper::MCselectorWrapper*);
    void PhotonSelectorSetups(string&,HepMC_helper::MCselectorWrapper*);
    void MuonSelectorSetups(string&,HepMC_helper::MCselectorWrapper*);
    void DefaultReconstructedParticleSelectorSetups(HepMC_helper::MCselectorWrapper*&, 
                                                    int particleType, double ptMin, double etaMax);
    
    std::string m_inputLocation ;
    TesIO* m_tesIO;
    //holds the Stotegate MC location
    std::string m_mcLocation;

    // Histogram limits
    int m_multBins; double m_multLow, m_multHigh;     // Multiplicity
    int m_eBins; double m_eLow, m_eHigh;              // Energy
    int m_pxBins; double m_pxLow, m_pxHigh;           // px
    int m_pyBins; double m_pyLow, m_pyHigh;           // py
    int m_pzBins; double m_pzLow, m_pzHigh;           // pz
    int m_eSumBins; double m_eSumLow, m_eSumHigh;     // Energy sum
    int m_pxSumBins; double m_pxSumLow, m_pxSumHigh;  // px sum
    int m_pySumBins; double m_pySumLow, m_pySumHigh;  // py sum
    int m_pzSumBins; double m_pzSumLow, m_pzSumHigh;  // pz sum
        
    vector< pair< string, HepMC_helper::MCselectorWrapper* > > m_selectorContainer;
    map< string, vector<IHistogram1D*> > m_histogramContainer;
    //-------------------------------
    // Default Values
    //-------------------------------
    
  };
  
}  //end of namespace bracket

#endif

---