---

EventHeaderMaker.h

Go to the documentation of this file.

// ================================================
// EventHeaderMaker class description
// ================================================
//
// THIS TEXT TO BE REPLACED BY ATLAS STANDARD FORMAT
//
//
// This version....
//
//
// Namespace Atlfast::
//
// class: EventHeaderMaker
//
// Description: 
//
//  Algorithm which makes the Atlfast Event Header
//  This must be run as the final algorithm in the 
//  Atlfast suite as it requires the information from
//  all the earlier makers.
//
//
// ................................................................
//

#ifndef __ATLFAST_EventHeaderMaker__
#define __ATLFAST_EventHeaderMaker__

// STL
#include <string>
#include <iterator>
#include <vector>
//#include <stl>

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

// Other
#include "CLHEP/Vector/LorentzVector.h"
#include "HepMC/GenEvent.h"
#include "HepMC/GenParticle.h"

// Atlfast
#include "AtlfastCode/ReconstructedParticle.h"
#include "AtlfastCode/CommonData.h"
#include "AtlfastCode/TesIO.h"
#include "AtlfastCode/HepMC_helper.h"
#include "AtlfastCode/Jet.h"

//*****************************************
// Default paths in the Transient event store to get/put entities
//
//[Note:  This will change as the current system is not considered acceptable.]
//******************************************

#define DEFAULT_beamEnergy                7000.0;

#define DEFAULT_lumi          1   // Luminosity option (1=low, 2=high)
#define DEFAULT_randSeed    12345 // Random number seed





//places to look in the TES 
#define DEFAULT_electronLocation          "/Event/AtlfastIsolatedElectrons"
#define DEFAULT_photonLocation            "/Event/AtlfastIsolatedPhotons"
#define DEFAULT_isolatedMuonLocation      "/Event/AtlfastIsolatedMuons"
#define DEFAULT_nonIsolatedMuonLocation   "/Event/AtlfastNonIsolatedMuons"
#define DEFAULT_jetLocation               "/Event/AtlfastJets"
#define DEFAULT_cellLocation              "/Event/AtlfastCells"
#define DEFAULT_clusterLocation           "/Event/AtlfastClusters"
#define DEFAULT_mcTruthLocation           "/Event/McEventCollection"
#define DEFAULT_outputLocation            "/Event/AtlfastEventHeader"
#define DEFAULT_commonDataLocation        "/Event/AtlfastCommonData"
#define DEFAULT_missingMomentumLocation   "/Event/AtlfastMissingMomentum"
#define DEFAULT_runNumber                 1000;
#define DEFAULT_evtNumber                 0;
#define DEFAULT_testMode                  false;

namespace Atlfast {

  class EventHeaderMaker : public Algorithm{
    
  public:
    
    //-------------------------
    // Constructors/Destructors
    //
    // Gaudi requires that the constructor takes certain arguments
    // (and passes them directly to the constructor of the base class)
    //-------------------------
    EventHeaderMaker( const std::string& name, ISvcLocator* pSvcLocator ) ; 
    virtual ~EventHeaderMaker();
    
    
    //------------------------------------------------------
    // Methods used by Gaudi to run the algorithm
    //------------------------------------------------------
    
    StatusCode initialize() ;
    StatusCode execute() ;
    StatusCode finalize() ;
    
    
    
  private:
    
    
    //-------------------------------
    // Private methods
    //-------------------------------
    
    template <class Type> int numberInList(std::string tesAddress, Type);
    int numberJetFlavor(std::string tesAddress, int);
    HepLorentzVector missingMomentum(MsgStream&);
    HepLorentzVector escapedMomentum(MsgStream&);
    template <class Type> 
      HepLorentzVector momentumSum(std::string tesAddress, Type dummy);
    template <class Type> 
      HepLorentzVector smearAndSum(std::string tesAddress, Type dummy);
    template <class Type, class AssType> 
      HepLorentzVector assocMomentumSum(std::string tesAddress, 
                                        Type dum1, AssType dum2);
    HepLorentzVector unusedClusterSum(); 
    HepLorentzVector unusedMuonSum();    
    HepLorentzVector unusedCellSum();    
    
    
    //-------------------------------
    // Data members
    //-------------------------------
    
    // the following data members tell the
    // algorithm where to retrieve its input particles from
    // in the TES. These are written by the job Options service.
    std::string m_electronLocation ;
    std::string m_photonLocation ;
    std::string m_isolatedMuonLocation;
    std::string m_nonIsolatedMuonLocation;
    std::string m_jetLocation;
    std::string m_cellLocation;
    std::string m_clusterLocation;
    std::string m_mcTruthLocation;
    std::string m_outputLocation;
    std::string m_commonDataLocation;
    std::string m_missingMomentumLocation;
    
    
    bool m_testMode;
    int m_runNumber;
    int m_evtNumber;
    double m_beamEnergy; // beam energy in GeV
    std::vector <int> m_escapingParticles; // PDG ids for escaping particles
    
    
    int           m_lumi;
    
    //htp random number seed will need to be removed eventually
    //    currently used to set up Smearer's random number seed
    //    SHOULD BE REPLACED BY FRAMEWORK SERVICE!! 
    int           m_randSeed;
    double        m_barrelForwardEta;
    
    double        m_notInit;
    TesIO*        m_tesIO;
    HepMC_helper::IMCselector*  m_visibleToAtlas;
  };

  //-------------------------------------------------------
  // numberInList
  //-------------------------------------------------------
  template <class Type> inline 
    int EventHeaderMaker::numberInList(std::string tesAddress, 
                                       Type /* dummy */) {
    std::vector<Type*> aList;
    if( !m_tesIO->copy(aList, tesAddress) ) throw "Error in numberInList";
    //if( !m_tesIO->get(aList) ) throw "Error in numberInList";
    return aList.size();
  }
  
  //-------------------------------------------------------
  // numberJetFlavor
  //-------------------------------------------------------
  inline
    int EventHeaderMaker::numberJetFlavor(
                                          std::string tesAddress, 
                                          int flavor) {
    std::vector<Jet*> theJets;
    int temp=0;
    if(m_tesIO->copy(theJets, tesAddress)){
      //if(m_tesIO->get(theJets)){
      std::vector<Jet*>::const_iterator jet  = theJets.begin();
      for (; jet != theJets.end(); ++jet) {
        if( (*jet)->pdg_id()==flavor ) ++temp;
      }
    }
    return temp;
  }
  
  //-------------------------------------------------------
  // momentumSum
  //-------------------------------------------------------
  template <class Type> inline HepLorentzVector
    EventHeaderMaker::momentumSum(std::string tesAddress, Type /*type info*/) {

    std::vector<Type*> aList;

    if(!m_tesIO->copy(aList, tesAddress) ) throw "Error in momentumSum";

    HepLorentzVector temp(0.0,0.0,0.0,0.0);
    typename std::vector<Type*>::const_iterator iter = aList.begin();

    for (; iter != aList.end(); ++iter) temp += (*iter)->momentum();

    return temp;
  }
  
  //-------------------------------------------------------
  // assocMomentumSum
  //-------------------------------------------------------
  template <class Type, class AssType> inline HepLorentzVector 
    EventHeaderMaker::assocMomentumSum(std::string tesAddress, 
                                       Type /* dum1 */,
                                       AssType /* dum2 */) {
    
    HepLorentzVector temp(0.0,0.0,0.0,0.0);
    std::vector<Type*> aList;
    if(!m_tesIO->copy(aList, tesAddress)) throw "Error in assoc Momentum Sum";
    //if(!m_tesIO->get(aList)) throw "Error in assoc Momentum Sum";
    
    typename std::vector<Type*>::const_iterator iter  = aList.begin();
    for (; iter != aList.end(); ++iter){
      SmartRefVector<AssType> aList  = (*iter)->associations(AssType());
      typename SmartRefVector<AssType>::iterator aIter = aList.begin();
      for(; aIter<aList.end(); ++aIter) temp+=(*aIter)->momentum();
    }
    return temp;  
  }
} // end of namespace bracket

#endif

---