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 "ProtoJetEvent/ProtoJet.h"  
//#include "ProtoJetEvent/ProtoJetCollection.h"
//Atlfast
#include "AtlfastUtils/TesIO.h" //needed for TesIOstat definition

class ISvcLocator;

namespace Atlfast {
  class ISmearer;
  class AtlfastProtoJet;

  using std::string;

  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();

  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;
    TesIO* m_tesIO;
    bool m_towerPJConstruct;
    bool m_clusterPJConstruct;
    bool m_finalStatePJConstruct;
    bool m_towerPJSmearOn;
    bool m_clusterPJSmearOn;

    //holds the Stotegate MC location
    std::string m_mcLocation;


  //--------------------------------------------
  // 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&);
    template <class T1, class T2> StatusCode FillProtoJets(const std::vector<T1*>&,
                                                           T2*,
                                                           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){
    bool smear = false;
    return this->FillProtoJets(vec,
                               proJ,
                               PJoutputLocation, 
                               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, 
                                        bool smear){
    MsgStream log( messageService(), name() ) ;
    ProtoJetCollection<ProtoJet>* pjs = new ProtoJetCollection<ProtoJet>;
    typename std::vector<T1*>::const_iterator itr = vec.begin();
    HepLorentzVector hepvec;
      for (; itr != vec.end(); ++itr) {
      hepvec = (*itr)->momentum();
      if(smear){
        hepvec = m_smearer->smear(hepvec);
      } 
      proJ = new T2(hepvec);
      pjs->push_back(proJ);
      }
    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;}
    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

---