---

AtlfastProtoJetMaker.h

Go to the documentation of this file.

// AtlfastProtoJet Maker.h
//
// HTP tidied up and put into namespace

#ifndef _Atlfast_AtlfastProtoJetMaker_H
#define _Atlfast_AtlfastProtoJetMaker_H

// Normal STL and physical vectors

#include "CLHEP/Vector/LorentzVector.h"
#include "CLHEP/Vector/ThreeVector.h"
#include <vector>
#include <string>

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

#include "HepMC/GenEvent.h"
#include "HepMC/GenParticle.h"
#include "JetRec/Jet.h"
//Atlfast
#include "AtlfastCode/AtlfastProtoJet.h"
//#include "AtlfastCode/AtlfastTowerProtoJet.h"
#include "AtlfastCode/HepMC_helper.h"
#include "AtlfastCode/TesIO.h"
#include "AtlfastCode/Cell.h"
#include "AtlfastCode/ISmearer.h"
#include "AtlfastCode/JetSmearer.h"
//Transient event store paths
//*******************************
// SET DEFAULTS
//*******************************
#define   DEFAULT_inputCellLocation      "/Event/AtlfastCells"
#define   DEFAULT_inputClusterLocation      "/Event/AtlfastClusters"
#define   DEFAULT_outputTowerPJKey     "/Event/AtlfastTowerProtoJets" 
#define   DEFAULT_outputClusterPJKey     "/Event/AtlfastClusterProtoJets" 
#define   DEFAULT_outputFinalStatePJKey     "/Event/AtlfastFinalStateProtoJets"
#define   DEFAULT_outputTowerJetKey     "/Event/AtlfastTowerJets" 
#define   DEFAULT_outputClusterJetKey     "/Event/AtlfastClusterJets" 
#define   DEFAULT_outputFinalStateJetKey     "/Event/AtlfastFinalStateJets"
#define   DEFAULT_towerPJConstruct true
#define   DEFAULT_clusterPJConstruct true
#define   DEFAULT_finalStatePJConstruct true
#define   DEFAULT_finalStatePJConstruct true
#define   DEFAULT_towerPJSmearOn false
#define   DEFAULT_clusterPJSmearOn false
 
namespace Atlfast {

  class AtlfastProtoJetMaker : public Algorithm {

  public:
    //Gaudi style constructor and execution methods
    AtlfastProtoJetMaker(const std::string& name, ISvcLocator* pSvcLocator);
    ~AtlfastProtoJetMaker();
    
    StatusCode          initialize();
    StatusCode          execute();
    //    StatusCode          finalize() {return StatusCode::SUCCESS;}
    StatusCode          finalize();

  private:
    //Paths in the transient event store to get/put things
    //Currently set in member variables to be overwritten
    //by job options service
    std::string m_inputCellLocation; 
    std::string m_inputClusterLocation;
    std::string m_outputTowerPJKey;
    std::string m_outputClusterPJKey;
    std::string m_outputFinalStatePJKey;
    std::string m_outputTowerJetKey;
    std::string m_outputClusterJetKey;
    std::string m_outputFinalStateJetKey;
    TesIO* m_tesIO;
    bool m_notInit;  
    bool m_towerPJConstruct;
    bool m_clusterPJConstruct;
    bool m_finalStatePJConstruct;
    bool m_towerPJSmearOn;
    bool m_clusterPJSmearOn;
  //--------------------------------------------
  // Smearer helper class
  // which knows all about how to do smearing
  // for my PJs
  //---------------------------------------------
    ISmearer* m_smearer;

  //Methods for filling PJs-----------------------
    template <class T1, class T2> StatusCode FillProtoJets(const std::vector<T1*>&,
                                                           T2*,
                                                           const std::string&,
                                                           const std::string&);
    template <class T1, class T2> StatusCode FillProtoJets(const std::vector<T1*>&,
                                                           T2*,
                                                           const std::string&,
                                                           const std::string&,
                                                           bool);
    template <class T1, class T2> StatusCode FillProtoJets(const std::vector<T1*>&,
                                                           T2*);
  };
//***********************************************************************
  //Fill ProtoJet and put in TES by key NO SMEARING available
  template <class T1, class T2> inline
    StatusCode AtlfastProtoJetMaker::FillProtoJets(const std::vector<T1*>& vec,
                                                   T2* proJ, 
                                                   const std::string& PJoutputLocation, 
                                                   const std::string& JoutputLocation)
    {
      bool smear = false;
      return this->FillProtoJets(vec,proJ,PJoutputLocation, JoutputLocation, smear);
    }//----------------------------------------------------------------------
  //Fill ProtoJet and put in TES by key SMEARING available
  template <class T1, class T2> inline
    StatusCode AtlfastProtoJetMaker::FillProtoJets(const std::vector<T1*>& vec,
                                                   T2* proJ, 
                                                   const std::string& PJoutputLocation, 
                                                   const std::string& JoutputLocation,
                                                   bool smear)
    {
      MsgStream log( messageService(), name() ) ;
      ObjectVector<ProtoJet>* pjs = new ObjectVector<ProtoJet>;
      ObjectVector< ::Jet >* jets = new ObjectVector< ::Jet >;
      typename std::vector<T1*>::const_iterator itr = vec.begin();
      HepLorentzVector hepvec;
      ::Jet* jet;
      for (; itr != vec.end(); ++itr) {
        hepvec = (*itr)->momentum();
        if(smear){
          hepvec = m_smearer->smear(hepvec);
        } 
        proJ = new T2(hepvec);
        pjs->push_back(proJ);
        jet = new ::Jet();
        jet->add(*proJ,1.);
        jets->push_back(jet);
      }
      log << MSG::DEBUG << "Size of vector"<<pjs->size() << endreq;
      //store protojets
      TesIoStat stat = m_tesIO -> store(pjs, PJoutputLocation);
      std::string message = stat ?  "AtlfastProtoJet stored by key":"Failed to store AtlfastProtoJet by key";
      log<<MSG::DEBUG<<message<<endreq;
      if (!stat) {return StatusCode::FAILURE;}
      //store JetRec Jets
      stat = m_tesIO -> store(pjs, JoutputLocation);
      message = stat ?  "JetRec Jet stored by key":"Failed to store JetRec Jet by key";
      log<<MSG::DEBUG<<message<<endreq;
      return (stat)? StatusCode::SUCCESS:StatusCode::FAILURE;
    }//-------------------------------------------------------------------------
  //Fill ProtoJet and put in TES by type doesn't work yet!! dunno why!
  template <class T1, class T2> inline
    StatusCode AtlfastProtoJetMaker::FillProtoJets(const std::vector<T1*>& vec,
                                                   T2* proJ) 
    {
      MsgStream log( messageService(), name() ) ;
      ObjectVector<ProtoJet>* pjs = new ObjectVector<ProtoJet>;
      typename std::vector<T1*>::const_iterator itr = vec.begin();
      for (; itr != vec.end(); ++itr) {
        proJ = new T2((*itr)->momentum());
        pjs->push_back(proJ);
      }
      log << MSG::DEBUG << "Size of vector"<<pjs->size() << endreq;
      //store protojets
      TesIoStat stat = m_tesIO -> store(pjs);
      std::string message = stat ?  "AtlfastProtoJet stored by type":"Failed to store AtlfastProtoJet type";
      log<<MSG::DEBUG<<message<<endreq;
      return (stat)? StatusCode::SUCCESS:StatusCode::FAILURE;
    }

} // end of namespace bracket
#endif

---