CLHEPHitFitTranslator.cc

Go to the documentation of this file.

//
//  $Id: CLHEPHitFitTranslator.cc,v 1.1 2011/05/26 09:46:59 mseidel Exp $
//

#include <cstdlib>
#include "CLHEP/Vector/LorentzVector.h"

#include "TopQuarkAnalysis/TopHitFit/interface/LeptonTranslatorBase.h"
#include "TopQuarkAnalysis/TopHitFit/interface/JetTranslatorBase.h"
#include "TopQuarkAnalysis/TopHitFit/interface/METTranslatorBase.h"

namespace hitfit {

template<>
LeptonTranslatorBase<CLHEP::HepLorentzVector>::LeptonTranslatorBase()
{

    std::string CMSSW_BASE(getenv("CMSSW_BASE"));
    std::string resolution_filename = CMSSW_BASE +
        std::string("/src/TopQuarkAnalysis/TopHitFit/data/exampleElectronResolution.txt");
    resolution_ = EtaDepResolution(resolution_filename);

} // LeptonTranslatorBase<CLHEP::HepLorentzVector>::LeptonTranslatorBase()


template<>
LeptonTranslatorBase<CLHEP::HepLorentzVector>::LeptonTranslatorBase(const std::string& ifile)
{

    std::string CMSSW_BASE(getenv("CMSSW_BASE"));
    std::string resolution_filename;

    if (ifile.empty()) {
        resolution_filename = CMSSW_BASE +
        std::string("/src/TopQuarkAnalysis/TopHitFit/data/exampleElectronResolution.txt");
    } else {
        resolution_filename = ifile ;
    }

    resolution_ = EtaDepResolution(resolution_filename);

} // LeptonTranslatorBase<CLHEP::HepLorentzVector>::LeptonTranslatorBase(const std::string& ifile)


template<>
LeptonTranslatorBase<CLHEP::HepLorentzVector>::~LeptonTranslatorBase()
{
}


template<>
Lepjets_Event_Lep
LeptonTranslatorBase<CLHEP::HepLorentzVector>::operator()(const CLHEP::HepLorentzVector& lepton,
                                                          int type /* = hitfit::lepton_label */,
                                                          bool useObjEmbRes /* = false */)
{

    Fourvec p(lepton.px(),lepton.py(),lepton.pz(),lepton.e());

    double lepton_eta(lepton.eta());
    Vector_Resolution lepton_resolution = resolution_.GetResolution(lepton_eta);
    Lepjets_Event_Lep retlep(p,
                             type,
                             lepton_resolution);
    return retlep;

} // Lepjets_Event_Lep LeptonTranslatorBase<CLHEP::HepLorentzVector>::operator()(const CLHEP::HepLorentzVector& lepton)


template<>
const EtaDepResolution&
LeptonTranslatorBase<CLHEP::HepLorentzVector>::resolution() const
{
    return resolution_;
}


template<>
bool
LeptonTranslatorBase<CLHEP::HepLorentzVector>::CheckEta(const CLHEP::HepLorentzVector& lepton) const
{
    return resolution_.CheckEta(lepton.eta());
}


template<>
JetTranslatorBase<CLHEP::HepLorentzVector>::JetTranslatorBase()
{

    std::string CMSSW_BASE(getenv("CMSSW_BASE"));
    std::string udsc_resolution_filename = CMSSW_BASE +
        std::string("/src/TopQuarkAnalysis/TopHitFit/data/exampleJetResolution.txt");
    std::string b_resolution_filename = udsc_resolution_filename;

    udscResolution_ = EtaDepResolution(udsc_resolution_filename);
    bResolution_    = EtaDepResolution(b_resolution_filename);

} // JetTranslatorBase<CLHEP::HepLorentzVector>::JetTranslatorBase()


template<>
JetTranslatorBase<CLHEP::HepLorentzVector>::JetTranslatorBase(const std::string& udscFile,
                                                              const std::string& bFile)
{

    std::string CMSSW_BASE(getenv("CMSSW_BASE"));
    std::string udsc_resolution_filename;
    std::string b_resolution_filename;

    if (udscFile.empty()) {
        udsc_resolution_filename = CMSSW_BASE +
            std::string("/src/TopQuarkAnalysis/TopHitFit/data/exampleJetResolution.txt");
    } else {
        udsc_resolution_filename = udscFile;
    }

    if (bFile.empty()) {
        b_resolution_filename = CMSSW_BASE +
            std::string("/src/TopQuarkAnalysis/TopHitFit/data/exampleJetResolution.txt");
    } else {
        b_resolution_filename = bFile;
    }

    udscResolution_ = EtaDepResolution(udsc_resolution_filename);
    bResolution_    = EtaDepResolution(b_resolution_filename);

} // JetTranslatorBase<CLHEP::HepLorentzVector>::JetTranslatorBase(const std::string& udscFile,const std::string& bFile)


template<>
JetTranslatorBase<CLHEP::HepLorentzVector>::~JetTranslatorBase()
{
} // JetTranslatorBase<CLHEP::HepLorentzVector>::~JetTranslatorBase()


template<>
Lepjets_Event_Jet
JetTranslatorBase<CLHEP::HepLorentzVector>::operator()(const CLHEP::HepLorentzVector& jet,
                                                       int type /* = hitfit::unknown_label */,
                                                       bool useObjEmbRes /* = false */)
{

    Fourvec p(jet.px(),jet.py(),jet.pz(),jet.e());

    double            jet_eta        = jet.eta();
    Vector_Resolution jet_resolution;

    if (type == hitfit::lepb_label||type == hitfit::hadb_label||type== hitfit::higgs_label) {
        jet_resolution = bResolution_.GetResolution(jet_eta);
    } else {
        jet_resolution = udscResolution_.GetResolution(jet_eta);
    }

    Lepjets_Event_Jet retjet(p,
                             type,
                             jet_resolution);
    return retjet;

} // Lepjets_Event_Jet JetTranslatorBase<CLHEP::HepLorentzVector>::operator()(const CLHEP::HepLorentzVector& j,int type)


template<>
const EtaDepResolution&
JetTranslatorBase<CLHEP::HepLorentzVector>::udscResolution() const
{
    return udscResolution_;
}


template<>
const EtaDepResolution&
JetTranslatorBase<CLHEP::HepLorentzVector>::bResolution() const
{
    return bResolution_;
}


template<>
bool
JetTranslatorBase<CLHEP::HepLorentzVector>::CheckEta(const CLHEP::HepLorentzVector& jet) const
{
    return udscResolution_.CheckEta(jet.eta()) && bResolution_.CheckEta(jet.eta());
}


template<>
METTranslatorBase<CLHEP::HepLorentzVector>::METTranslatorBase()
{
    resolution_ = Resolution(std::string("0,0,12"));
} // METTranslatorBase<CLHEP::HepLorentzVector>::METTranslatorBase()


template<>
METTranslatorBase<CLHEP::HepLorentzVector>::METTranslatorBase(const std::string& ifile)
{
    const Defaults_Text defs(ifile);
    std::string resolution_string(defs.get_string("met_resolution"));
    resolution_ = Resolution(resolution_string);

} // METTranslatorBase<CLHEP::HepLorentzVector>::METTranslatorBase(const std::string& ifile)


template<>
METTranslatorBase<CLHEP::HepLorentzVector>::~METTranslatorBase()
{
} // METTranslatorBase<CLHEP::HepLorentzVector>::~METTranslatorBase()


template<>
Fourvec
METTranslatorBase<CLHEP::HepLorentzVector>::operator()(const CLHEP::HepLorentzVector& m,
                                                       bool useObjEmbRes /* = false */)
{

    return Fourvec (m.px(),m.py(),0.0,m.e());

} // Fourvec METTranslatorBase<CLHEP::HepLorentzVector>::operator()(const CLHEP::HepLorentzVector& m)



template<>
Resolution
METTranslatorBase<CLHEP::HepLorentzVector>::KtResolution(const CLHEP::HepLorentzVector& m,
                                                         bool useObjEmbRes /* = false */) const
{
    return resolution_;
} // Resolution METTranslatorBase<CLHEP::HepLorentzVector>::KtResolution(const CLHEP::HepLorentzVector& m)



template<>
Resolution
METTranslatorBase<CLHEP::HepLorentzVector>::METResolution(const CLHEP::HepLorentzVector& m,
                                                          bool useObjEmbRes /* = false */) const
{
    return KtResolution(m,useObjEmbRes);
} // Resolution METTranslatorBase<CLHEP::HepLorentzVector>::METResolution(const CLHEP::HepLorentzVector& m)


} // namespace hitfit