---

AtlfastProtoJetMaker.h

Go to the documentation of this file.

// AtlfastProtoJet Maker.h
//

#ifndef ATLFAST_ATLFASTPROTOJETMAKER_H
#define ATLFAST_ATLFASTPROTOJETMAKER_H

// Normal STL and physical vectors

#include "CLHEP/Vector/LorentzVector.h"  //used by template function
#include <vector>
#include <string>

// Gaudi
#include "GaudiKernel/Algorithm.h"   //needed for inheritance
#include "GaudiKernel/MsgStream.h"   //def used by inline functions

#include "JetRec/Jet.h"   //definition used by inline funtions
//Atlfast
#include "AtlfastCode/TesIO.h" //needed for TesIOstat definition
//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
 
class ISvcLocator;

namespace Atlfast {
  class ISmearer;
  class AtlfastProtoJet;

  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_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;
    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(jets, 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);
    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

---