CovarianceMatrix Class Reference

#include <CovarianceMatrix.h>

Public Types
enum	ObjectType {   kUdscJet, kBJet, kMuon, kElectron,   kMet }

Public Member Functions
	CovarianceMatrix ()
	default constructor More...
	CovarianceMatrix (const std::vector< edm::ParameterSet > &udscResolutions, const std::vector< edm::ParameterSet > &bResolutions, const std::vector< double > &jetEnergyResolutionScaleFactors, const std::vector< double > &jetEnergyResolutionEtaBinning)
	constructor for the fully-hadronic channel More...
	CovarianceMatrix (const std::vector< edm::ParameterSet > &udscResolutions, const std::vector< edm::ParameterSet > &bResolutions, const std::vector< edm::ParameterSet > &lepResolutions, const std::vector< edm::ParameterSet > &metResolutions, const std::vector< double > &jetEnergyResolutionScaleFactors, const std::vector< double > &jetEnergyResolutionEtaBinning)
	constructor for the lepton+jets channel More...
double	getResolution (const TLorentzVector &object, const ObjectType objType, const std::string &whichResolution="")
	get resolution for a given component of an object More...
template<class T >
double	getResolution (const pat::PATObject< T > &object, const std::string &whichResolution, const bool isBJet=false)
	get resolution for a given PAT object More...
template<class T >
TMatrixD	setupMatrix (const pat::PATObject< T > &object, const TopKinFitter::Param param, const std::string &resolutionProvider="")
	return covariance matrix for a PAT object More...
TMatrixD	setupMatrix (const TLorentzVector &object, const ObjectType objType, const TopKinFitter::Param param)
	return covariance matrix for a plain 4-vector More...
	~CovarianceMatrix ()

Private Member Functions
template<class T >
double	getEtaDependentScaleFactor (const pat::PATObject< T > &object)
	get eta dependent smear factor for a PAT object More...
double	getEtaDependentScaleFactor (const TLorentzVector &object)
	get eta-dependent scale factor for a plain 4-vector More...
template<class T >
ObjectType	getObjectType (const pat::PATObject< T > &object, const bool isBJet=false)
	determine type for a given PAT object More...

Private Attributes
std::vector< std::string >	binsB_
std::vector< std::string >	binsLep_
std::vector< std::string >	binsMet_
std::vector< std::string >	binsUdsc_
	vector of strings for the binning of the resolutions More...
std::vector< std::string >	funcEtaB_
std::vector< std::string >	funcEtaLep_
std::vector< std::string >	funcEtaMet_
std::vector< std::string >	funcEtaUdsc_
std::vector< std::string >	funcEtB_
std::vector< std::string >	funcEtLep_
std::vector< std::string >	funcEtMet_
std::vector< std::string >	funcEtUdsc_
	vectors for the resolution functions More...
std::vector< std::string >	funcPhiB_
std::vector< std::string >	funcPhiLep_
std::vector< std::string >	funcPhiMet_
std::vector< std::string >	funcPhiUdsc_
const std::vector< double >	jetEnergyResolutionEtaBinning_
const std::vector< double >	jetEnergyResolutionScaleFactors_
	scale factors for the jet energy resolution More...

Detailed Description

Definition at line 28 of file CovarianceMatrix.h.

Member Enumeration Documentation

ObjectType

enum CovarianceMatrix::ObjectType

Enumerator
kUdscJet
kBJet
kMuon
kElectron
kMet

Definition at line 30 of file CovarianceMatrix.h.

{ kUdscJet, kBJet, kMuon, kElectron, kMet };

Constructor & Destructor Documentation

CovarianceMatrix() [1/3]

CovarianceMatrix::CovarianceMatrix ( )

inline

default constructor

Definition at line 33 of file CovarianceMatrix.h.

{}

CovarianceMatrix() [2/3]

CovarianceMatrix::CovarianceMatrix	(	const std::vector< edm::ParameterSet > &	udscResolutions,
		const std::vector< edm::ParameterSet > &	bResolutions,
		const std::vector< double > &	jetEnergyResolutionScaleFactors,
		const std::vector< double > &	jetEnergyResolutionEtaBinning
	)

constructor for the fully-hadronic channel

Definition at line 7 of file CovarianceMatrix.cc.

References binsB_, binsUdsc_, TtFullHadKinFitProducer_cfi::bResolutions, Exception, funcEtaB_, funcEtaUdsc_, funcEtB_, funcEtUdsc_, funcPhiB_, funcPhiUdsc_, AlCaHLTBitMon_QueryRunRegistry::string, and TtFullHadKinFitProducer_cfi::udscResolutions.

    : jetEnergyResolutionScaleFactors_(jetEnergyResolutionScaleFactors),
      jetEnergyResolutionEtaBinning_(jetEnergyResolutionEtaBinning) {
  for (std::vector<edm::ParameterSet>::const_iterator iSet = udscResolutions.begin(); iSet != udscResolutions.end();
       ++iSet) {
    if (iSet->exists("bin"))
      binsUdsc_.push_back(iSet->getParameter<std::string>("bin"));
    else if (udscResolutions.size() == 1)
      binsUdsc_.push_back("");
    else
      throw cms::Exception("Configuration") << "Parameter 'bin' is needed if more than one PSet is specified!\n";

    funcEtUdsc_.push_back(iSet->getParameter<std::string>("et"));
    funcEtaUdsc_.push_back(iSet->getParameter<std::string>("eta"));
    funcPhiUdsc_.push_back(iSet->getParameter<std::string>("phi"));
  }
  for (std::vector<edm::ParameterSet>::const_iterator iSet = bResolutions.begin(); iSet != bResolutions.end(); ++iSet) {
    if (iSet->exists("bin"))
      binsB_.push_back(iSet->getParameter<std::string>("bin"));
    else if (bResolutions.size() == 1)
      binsB_.push_back("");
    else
      throw cms::Exception("Configuration") << "Parameter 'bin' is needed if more than one PSet is specified!\n";

    funcEtB_.push_back(iSet->getParameter<std::string>("et"));
    funcEtaB_.push_back(iSet->getParameter<std::string>("eta"));
    funcPhiB_.push_back(iSet->getParameter<std::string>("phi"));
  }
}

CovarianceMatrix() [3/3]

CovarianceMatrix::CovarianceMatrix	(	const std::vector< edm::ParameterSet > &	udscResolutions,
		const std::vector< edm::ParameterSet > &	bResolutions,
		const std::vector< edm::ParameterSet > &	lepResolutions,
		const std::vector< edm::ParameterSet > &	metResolutions,
		const std::vector< double > &	jetEnergyResolutionScaleFactors,
		const std::vector< double > &	jetEnergyResolutionEtaBinning
	)

constructor for the lepton+jets channel

Definition at line 40 of file CovarianceMatrix.cc.

References binsB_, binsLep_, binsMet_, binsUdsc_, TtFullHadKinFitProducer_cfi::bResolutions, Exception, funcEtaB_, funcEtaLep_, funcEtaMet_, funcEtaUdsc_, funcEtB_, funcEtLep_, funcEtMet_, funcEtUdsc_, funcPhiB_, funcPhiLep_, funcPhiMet_, funcPhiUdsc_, mps_fire::i, jetEnergyResolutionEtaBinning_, jetEnergyResolutionScaleFactors_, AlCaHLTBitMon_QueryRunRegistry::string, and TtFullHadKinFitProducer_cfi::udscResolutions.

    : jetEnergyResolutionScaleFactors_(jetEnergyResolutionScaleFactors),
      jetEnergyResolutionEtaBinning_(jetEnergyResolutionEtaBinning) {
  if (jetEnergyResolutionScaleFactors_.size() + 1 != jetEnergyResolutionEtaBinning_.size())
    throw cms::Exception("Configuration") << "The number of scale factors does not fit to the number of eta bins!\n";
  for (unsigned int i = 0; i < jetEnergyResolutionEtaBinning_.size(); i++)
    if (jetEnergyResolutionEtaBinning_[i] < 0. && i < jetEnergyResolutionEtaBinning_.size() - 1)
      throw cms::Exception("Configuration") << "eta binning in absolut values required!\n";

  for (std::vector<edm::ParameterSet>::const_iterator iSet = udscResolutions.begin(); iSet != udscResolutions.end();
       ++iSet) {
    if (iSet->exists("bin"))
      binsUdsc_.push_back(iSet->getParameter<std::string>("bin"));
    else if (udscResolutions.size() == 1)
      binsUdsc_.push_back("");
    else
      throw cms::Exception("Configuration") << "Parameter 'bin' is needed if more than one PSet is specified!\n";

    funcEtUdsc_.push_back(iSet->getParameter<std::string>("et"));
    funcEtaUdsc_.push_back(iSet->getParameter<std::string>("eta"));
    funcPhiUdsc_.push_back(iSet->getParameter<std::string>("phi"));
  }
  for (std::vector<edm::ParameterSet>::const_iterator iSet = bResolutions.begin(); iSet != bResolutions.end(); ++iSet) {
    if (iSet->exists("bin"))
      binsB_.push_back(iSet->getParameter<std::string>("bin"));
    else if (bResolutions.size() == 1)
      binsB_.push_back("");
    else
      throw cms::Exception("Configuration") << "Parameter 'bin' is needed if more than one PSet is specified!\n";

    funcEtB_.push_back(iSet->getParameter<std::string>("et"));
    funcEtaB_.push_back(iSet->getParameter<std::string>("eta"));
    funcPhiB_.push_back(iSet->getParameter<std::string>("phi"));
  }
  for (std::vector<edm::ParameterSet>::const_iterator iSet = lepResolutions.begin(); iSet != lepResolutions.end();
       ++iSet) {
    if (iSet->exists("bin"))
      binsLep_.push_back(iSet->getParameter<std::string>("bin"));
    else if (lepResolutions.size() == 1)
      binsLep_.push_back("");
    else
      throw cms::Exception("Configuration") << "Parameter 'bin' is needed if more than one PSet is specified!\n";

    funcEtLep_.push_back(iSet->getParameter<std::string>("et"));
    funcEtaLep_.push_back(iSet->getParameter<std::string>("eta"));
    funcPhiLep_.push_back(iSet->getParameter<std::string>("phi"));
  }
  for (std::vector<edm::ParameterSet>::const_iterator iSet = metResolutions.begin(); iSet != metResolutions.end();
       ++iSet) {
    if (iSet->exists("bin"))
      binsMet_.push_back(iSet->getParameter<std::string>("bin"));
    else if (metResolutions.size() == 1)
      binsMet_.push_back("");
    else
      throw cms::Exception("Configuration") << "Parameter 'bin' is needed if more than one PSet is specified!\n";

    funcEtMet_.push_back(iSet->getParameter<std::string>("et"));
    funcEtaMet_.push_back(iSet->getParameter<std::string>("eta"));
    funcPhiMet_.push_back(iSet->getParameter<std::string>("phi"));
  }
}

~CovarianceMatrix()

CovarianceMatrix::~CovarianceMatrix ( )

inline

Definition at line 47 of file CovarianceMatrix.h.

{}

Member Function Documentation

getEtaDependentScaleFactor() [1/2]

template<class T >

double CovarianceMatrix::getEtaDependentScaleFactor ( const pat::PATObject< T > &  object )

private

get eta dependent smear factor for a PAT object

Definition at line 162 of file CovarianceMatrix.h.

References funct::abs(), PVValHelper::eta, mps_fire::i, jetEnergyResolutionEtaBinning_, and jetEnergyResolutionScaleFactors_.

Referenced by setupMatrix().

                                                                                 {
  double etaDependentScaleFactor = 1.;
  for (unsigned int i = 0; i < jetEnergyResolutionEtaBinning_.size(); i++) {
    if (std::abs(object.eta()) >= jetEnergyResolutionEtaBinning_[i] && jetEnergyResolutionEtaBinning_[i] >= 0.) {
      if (i == jetEnergyResolutionEtaBinning_.size() - 1) {
        edm::LogWarning("CovarianceMatrix") << "object eta (" << std::abs(object.eta()) << ") beyond last eta bin ("
                                            << jetEnergyResolutionEtaBinning_[i] << ") using scale factor 1.0!";
        etaDependentScaleFactor = 1.;
        break;
      }
      etaDependentScaleFactor = jetEnergyResolutionScaleFactors_[i];
    } else
      break;
  }
  return etaDependentScaleFactor;
}

getEtaDependentScaleFactor() [2/2]

double CovarianceMatrix::getEtaDependentScaleFactor ( const TLorentzVector &  object )

private

get eta-dependent scale factor for a plain 4-vector

Definition at line 338 of file CovarianceMatrix.cc.

References funct::abs(), mps_fire::i, jetEnergyResolutionEtaBinning_, and jetEnergyResolutionScaleFactors_.

                                                                                {
  double etaDependentScaleFactor = 1.;
  for (unsigned int i = 0; i < jetEnergyResolutionEtaBinning_.size(); i++) {
    if (std::abs(object.Eta()) >= jetEnergyResolutionEtaBinning_[i] && jetEnergyResolutionEtaBinning_[i] >= 0.) {
      if (i == jetEnergyResolutionEtaBinning_.size() - 1) {
        edm::LogWarning("CovarianceMatrix") << "object eta (" << std::abs(object.Eta()) << ") beyond last eta bin ("
                                            << jetEnergyResolutionEtaBinning_[i] << ") using scale factor 1.0!";
        etaDependentScaleFactor = 1.;
        break;
      }
      etaDependentScaleFactor = jetEnergyResolutionScaleFactors_[i];
    } else
      break;
  }
  return etaDependentScaleFactor;
}

getObjectType()

template<class T >

CovarianceMatrix::ObjectType CovarianceMatrix::getObjectType ( const pat::PATObject< T > &  object, const bool  isBJet = false  )

private

determine type for a given PAT object

Definition at line 137 of file CovarianceMatrix.h.

References ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), Exception, kBJet, kElectron, kMet, kMuon, kUdscJet, and electrons_cff::objType.

Referenced by getResolution(), and setupMatrix().

                                                                                                           {
  ObjectType objType;
  // jets
  if constexpr (std::is_same_v<T, reco::Jet>) {
    if (isBJet)
      objType = kBJet;
    else
      objType = kUdscJet;
  }
  // muons
  else if constexpr (std::is_same_v<T, reco::Muon>)
    objType = kMuon;
  // electrons
  else if constexpr (std::is_same_v<T, reco::GsfElectron>)
    objType = kElectron;
  // MET
  else if constexpr (std::is_same_v<T, reco::MET>)
    objType = kMet;
  // catch anything else
  else
    throw cms::Exception("UnsupportedObject") << "The object given is not supported!\n";
  return objType;
}

getResolution() [1/2]

double CovarianceMatrix::getResolution	(	const TLorentzVector &	object,
		const ObjectType	objType,
		const std::string &	whichResolution = ""
	)

get resolution for a given component of an object

Definition at line 107 of file CovarianceMatrix.cc.

References binsB_, binsLep_, binsMet_, binsUdsc_, EcalCondDBWriter_cfi::Energy, Exception, funcEtaB_, funcEtaLep_, funcEtaMet_, funcEtaUdsc_, funcEtB_, funcEtLep_, funcEtMet_, funcEtUdsc_, funcPhiB_, funcPhiLep_, funcPhiMet_, funcPhiUdsc_, mps_fire::i, kBJet, kElectron, kMet, kMuon, kUdscJet, and electrons_cff::objType.

Referenced by getResolution(), and setupMatrix().

                                                                         {
  std::vector<std::string>*bins_, *funcEt_, *funcEta_, *funcPhi_;

  switch (objType) {
    case kUdscJet:
      bins_ = &binsUdsc_;
      funcEt_ = &funcEtUdsc_;
      funcEta_ = &funcEtaUdsc_;
      funcPhi_ = &funcPhiUdsc_;
      break;
    case kBJet:
      bins_ = &binsB_;
      funcEt_ = &funcEtB_;
      funcEta_ = &funcEtaB_;
      funcPhi_ = &funcPhiB_;
      break;
    case kMuon:
      bins_ = &binsLep_;
      funcEt_ = &funcEtLep_;
      funcEta_ = &funcEtaLep_;
      funcPhi_ = &funcPhiLep_;
      break;
    case kElectron:
      bins_ = &binsLep_;
      funcEt_ = &funcEtLep_;
      funcEta_ = &funcEtaLep_;
      funcPhi_ = &funcPhiLep_;
      break;
    case kMet:
      bins_ = &binsMet_;
      funcEt_ = &funcEtMet_;
      funcEta_ = &funcEtaMet_;
      funcPhi_ = &funcPhiMet_;
      break;
    default:
      throw cms::Exception("UnsupportedObject") << "The object given is not supported!\n";
  }

  int selectedBin = -1;
  reco::LeafCandidate candidate;
  for (unsigned int i = 0; i < bins_->size(); ++i) {
    StringCutObjectSelector<reco::LeafCandidate> select_(bins_->at(i));
    candidate = reco::LeafCandidate(
        0, reco::LeafCandidate::LorentzVector(object.Px(), object.Py(), object.Pz(), object.Energy()));
    if (select_(candidate)) {
      selectedBin = i;
      break;
    }
  }
  double res = 0;
  if (selectedBin >= 0) {
    if (whichResolution == "et")
      res = StringObjectFunction<reco::LeafCandidate>(funcEt_->at(selectedBin)).operator()(candidate);
    else if (whichResolution == "eta")
      res = StringObjectFunction<reco::LeafCandidate>(funcEta_->at(selectedBin)).operator()(candidate);
    else if (whichResolution == "phi")
      res = StringObjectFunction<reco::LeafCandidate>(funcPhi_->at(selectedBin)).operator()(candidate);
    else
      throw cms::Exception("ProgrammingError") << "Only 'et', 'eta' and 'phi' resolutions supported!\n";
  }
  return res;
}

getResolution() [2/2]

template<class T >

double CovarianceMatrix::getResolution ( const pat::PATObject< T > &  object, const std::string &  whichResolution, const bool  isBJet = false  )

inline

get resolution for a given PAT object

Definition at line 60 of file CovarianceMatrix.h.

References HBHEDarkening_cff::energy, getObjectType(), getResolution(), multPhiCorr_741_25nsDY_cfi::px, and multPhiCorr_741_25nsDY_cfi::py.

                                                                                                                   {
    return getResolution(TLorentzVector(object.px(), object.py(), object.pz(), object.energy()),
                         getObjectType(object, isBJet),
                         whichResolution);
  }

setupMatrix() [1/2]

template<class T >

TMatrixD CovarianceMatrix::setupMatrix	(	const pat::PATObject< T > &	object,
		const TopKinFitter::Param	param,
		const std::string &	resolutionProvider = ""
	)

return covariance matrix for a PAT object

Definition at line 88 of file CovarianceMatrix.h.

References ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), HBHEDarkening_cff::energy, getEtaDependentScaleFactor(), getObjectType(), TopKinFitter::kEMom, TopKinFitter::kEtEtaPhi, TopKinFitter::kEtThetaPhi, electrons_cff::objType, funct::pow(), multPhiCorr_741_25nsDY_cfi::px, and multPhiCorr_741_25nsDY_cfi::py.

                                                                            {
  // This part is for pat objects with resolutions embedded
  if (object.hasKinResolution()) {
    TMatrixD CovM3(3, 3);
    CovM3.Zero();
    TMatrixD CovM4(4, 4);
    CovM4.Zero();
    TMatrixD* CovM = &CovM3;
    switch (param) {
      case TopKinFitter::kEtEtaPhi:
        CovM3(0, 0) = pow(object.resolEt(resolutionProvider), 2);
        if constexpr (std::is_same_v<T, reco::Jet>)
          CovM3(0, 0) *= getEtaDependentScaleFactor(object);
        else if constexpr (std::is_same_v<T, reco::MET>)
          CovM3(1, 1) = pow(9999., 2);
        else
          CovM3(1, 1) = pow(object.resolEta(resolutionProvider), 2);
        CovM3(2, 2) = pow(object.resolPhi(resolutionProvider), 2);
        CovM = &CovM3;
        break;
      case TopKinFitter::kEtThetaPhi:
        CovM3(0, 0) = pow(object.resolEt(resolutionProvider), 2);
        if constexpr (std::is_same_v<T, reco::Jet>)
          CovM3(0, 0) *= getEtaDependentScaleFactor(object);
        CovM3(1, 1) = pow(object.resolTheta(resolutionProvider), 2);
        CovM3(2, 2) = pow(object.resolPhi(resolutionProvider), 2);
        CovM = &CovM3;
        break;
      case TopKinFitter::kEMom:
        CovM4(0, 0) = pow(1, 2);
        CovM4(1, 1) = pow(1, 2);
        CovM4(2, 2) = pow(1, 2);
        CovM4(3, 3) = pow(1, 2);
        CovM = &CovM4;
        break;
    }
    return *CovM;
  }
  // This part is for objects without resolutions embedded
  else {
    const ObjectType objType = getObjectType(object, (resolutionProvider == "bjets"));
    const TLorentzVector p4(object.px(), object.py(), object.pz(), object.energy());
    return setupMatrix(p4, objType, param);
  }
}

setupMatrix() [2/2]

TMatrixD CovarianceMatrix::setupMatrix	(	const TLorentzVector &	object,
		const ObjectType	objType,
		const TopKinFitter::Param	param
	)

return covariance matrix for a plain 4-vector

Definition at line 172 of file CovarianceMatrix.cc.

References res::HelperMET::a(), res::HelperMuon::a(), res::HelperElectron::a(), res::HelperJet::a(), res::HelperMET::b(), res::HelperElectron::b(), res::HelperMuon::b(), res::HelperJet::b(), binsLep_, binsMet_, binsUdsc_, res::HelperMET::c(), res::HelperElectron::c(), res::HelperMuon::c(), res::HelperJet::c(), res::HelperJet::d(), res::HelperElectron::et(), res::HelperMET::et(), res::HelperMuon::et(), res::HelperJet::et(), res::HelperElectron::eta(), res::HelperMuon::eta(), res::HelperJet::eta(), PVValHelper::eta, Exception, getEtaDependentScaleFactor(), getResolution(), res::HelperJet::kB, kBJet, kElectron, TopKinFitter::kEMom, TopKinFitter::kEtEtaPhi, TopKinFitter::kEtThetaPhi, kMet, kMuon, res::HelperJet::kUds, kUdscJet, electrons_cff::objType, res::HelperMuon::phi(), res::HelperElectron::phi(), res::HelperJet::phi(), res::HelperMET::phi(), funct::pow(), DiDispStaMuonMonitor_cfi::pt, res::HelperMuon::theta(), res::HelperElectron::theta(), and res::HelperJet::theta().

                                                                      {
  TMatrixD CovM3(3, 3);
  CovM3.Zero();
  TMatrixD CovM4(4, 4);
  CovM4.Zero();
  const double pt = object.Pt();
  const double eta = object.Eta();
  switch (objType) {
    case kUdscJet:
      // if object is a non-b jet
      {
        res::HelperJet jetRes;
        switch (param) {
          case TopKinFitter::kEMom:
            CovM4(0, 0) = pow(jetRes.a(pt, eta, res::HelperJet::kUds), 2);
            CovM4(1, 1) = pow(jetRes.b(pt, eta, res::HelperJet::kUds), 2);
            CovM4(2, 2) = pow(jetRes.c(pt, eta, res::HelperJet::kUds), 2);
            CovM4(3, 3) = pow(jetRes.d(pt, eta, res::HelperJet::kUds), 2);
            return CovM4;
          case TopKinFitter::kEtEtaPhi:
            if (binsUdsc_.empty()) {
              CovM3(0, 0) = pow(jetRes.et(pt, eta, res::HelperJet::kUds), 2);
              CovM3(0, 0) *= pow(getEtaDependentScaleFactor(object), 2);
              CovM3(1, 1) = pow(jetRes.eta(pt, eta, res::HelperJet::kUds), 2);
              CovM3(2, 2) = pow(jetRes.phi(pt, eta, res::HelperJet::kUds), 2);
            } else {
              CovM3(0, 0) = pow(getResolution(object, objType, "et"), 2);
              CovM3(0, 0) *= pow(getEtaDependentScaleFactor(object), 2);
              CovM3(1, 1) = pow(getResolution(object, objType, "eta"), 2);
              CovM3(2, 2) = pow(getResolution(object, objType, "phi"), 2);
            }
            return CovM3;
          case TopKinFitter::kEtThetaPhi:
            CovM3(0, 0) = pow(jetRes.et(pt, eta, res::HelperJet::kUds), 2);
            CovM3(0, 0) *= pow(getEtaDependentScaleFactor(object), 2);
            CovM3(1, 1) = pow(jetRes.theta(pt, eta, res::HelperJet::kUds), 2);
            CovM3(2, 2) = pow(jetRes.phi(pt, eta, res::HelperJet::kUds), 2);
            return CovM3;
        }
      }
      break;
    case kBJet:
      // if object is a b jet
      {
        res::HelperJet jetRes;
        switch (param) {
          case TopKinFitter::kEMom:
            CovM4(0, 0) = pow(jetRes.a(pt, eta, res::HelperJet::kB), 2);
            CovM4(1, 1) = pow(jetRes.b(pt, eta, res::HelperJet::kB), 2);
            CovM4(2, 2) = pow(jetRes.c(pt, eta, res::HelperJet::kB), 2);
            CovM4(3, 3) = pow(jetRes.d(pt, eta, res::HelperJet::kB), 2);
            return CovM4;
          case TopKinFitter::kEtEtaPhi:
            if (binsUdsc_.empty()) {
              CovM3(0, 0) = pow(jetRes.et(pt, eta, res::HelperJet::kB), 2);
              CovM3(0, 0) *= pow(getEtaDependentScaleFactor(object), 2);
              CovM3(1, 1) = pow(jetRes.eta(pt, eta, res::HelperJet::kB), 2);
              CovM3(2, 2) = pow(jetRes.phi(pt, eta, res::HelperJet::kB), 2);
            } else {
              CovM3(0, 0) = pow(getResolution(object, objType, "et"), 2);
              CovM3(0, 0) *= pow(getEtaDependentScaleFactor(object), 2);
              CovM3(1, 1) = pow(getResolution(object, objType, "eta"), 2);
              CovM3(2, 2) = pow(getResolution(object, objType, "phi"), 2);
            }
            return CovM3;
          case TopKinFitter::kEtThetaPhi:
            CovM3(0, 0) = pow(jetRes.et(pt, eta, res::HelperJet::kB), 2);
            CovM3(0, 0) *= pow(getEtaDependentScaleFactor(object), 2);
            CovM3(1, 1) = pow(jetRes.theta(pt, eta, res::HelperJet::kB), 2);
            CovM3(2, 2) = pow(jetRes.phi(pt, eta, res::HelperJet::kB), 2);
            return CovM3;
        }
      }
      break;
    case kMuon:
      // if object is a muon
      {
        res::HelperMuon muonRes;
        switch (param) {
          case TopKinFitter::kEMom:
            CovM3(0, 0) = pow(muonRes.a(pt, eta), 2);
            CovM3(1, 1) = pow(muonRes.b(pt, eta), 2);
            CovM3(2, 2) = pow(muonRes.c(pt, eta), 2);
            return CovM3;
          case TopKinFitter::kEtEtaPhi:
            if (binsLep_.empty()) {
              CovM3(0, 0) = pow(muonRes.et(pt, eta), 2);
              CovM3(1, 1) = pow(muonRes.eta(pt, eta), 2);
              CovM3(2, 2) = pow(muonRes.phi(pt, eta), 2);
            } else {
              CovM3(0, 0) = pow(getResolution(object, objType, "et"), 2);
              CovM3(1, 1) = pow(getResolution(object, objType, "eta"), 2);
              CovM3(2, 2) = pow(getResolution(object, objType, "phi"), 2);
            }
            return CovM3;
          case TopKinFitter::kEtThetaPhi:
            CovM3(0, 0) = pow(muonRes.et(pt, eta), 2);
            CovM3(1, 1) = pow(muonRes.theta(pt, eta), 2);
            CovM3(2, 2) = pow(muonRes.phi(pt, eta), 2);
            return CovM3;
        }
      }
      break;
    case kElectron: {
      // if object is an electron
      res::HelperElectron elecRes;
      switch (param) {
        case TopKinFitter::kEMom:
          CovM3(0, 0) = pow(elecRes.a(pt, eta), 2);
          CovM3(1, 1) = pow(elecRes.b(pt, eta), 2);
          CovM3(2, 2) = pow(elecRes.c(pt, eta), 2);
          return CovM3;
        case TopKinFitter::kEtEtaPhi:
          if (binsLep_.empty()) {
            CovM3(0, 0) = pow(elecRes.et(pt, eta), 2);
            CovM3(1, 1) = pow(elecRes.eta(pt, eta), 2);
            CovM3(2, 2) = pow(elecRes.phi(pt, eta), 2);
          } else {
            CovM3(0, 0) = pow(getResolution(object, objType, "et"), 2);
            CovM3(1, 1) = pow(getResolution(object, objType, "eta"), 2);
            CovM3(2, 2) = pow(getResolution(object, objType, "phi"), 2);
          }
          return CovM3;
        case TopKinFitter::kEtThetaPhi:
          CovM3(0, 0) = pow(elecRes.et(pt, eta), 2);
          CovM3(1, 1) = pow(elecRes.theta(pt, eta), 2);
          CovM3(2, 2) = pow(elecRes.phi(pt, eta), 2);
          return CovM3;
      }
    } break;
    case kMet:
      // if object is met
      {
        res::HelperMET metRes;
        switch (param) {
          case TopKinFitter::kEMom:
            CovM3(0, 0) = pow(metRes.a(pt), 2);
            CovM3(1, 1) = pow(metRes.b(pt), 2);
            CovM3(2, 2) = pow(metRes.c(pt), 2);
            return CovM3;
          case TopKinFitter::kEtEtaPhi:
            if (binsMet_.empty()) {
              CovM3(0, 0) = pow(metRes.et(pt), 2);
              CovM3(1, 1) = pow(9999., 2);
              CovM3(2, 2) = pow(metRes.phi(pt), 2);
            } else {
              CovM3(0, 0) = pow(getResolution(object, objType, "et"), 2);
              CovM3(1, 1) = pow(getResolution(object, objType, "eta"), 2);
              CovM3(2, 2) = pow(getResolution(object, objType, "phi"), 2);
            }
            return CovM3;
          case TopKinFitter::kEtThetaPhi:
            CovM3(0, 0) = pow(metRes.et(pt), 2);
            CovM3(1, 1) = pow(9999., 2);
            CovM3(2, 2) = pow(metRes.phi(pt), 2);
            return CovM3;
        }
      }
      break;
  }
  cms::Exception("Logic") << "Something went wrong while trying to setup a covariance matrix!\n";
  return CovM4;  //should never get here
}

Member Data Documentation

binsB_

std::vector<std::string> CovarianceMatrix::binsB_

private

Definition at line 68 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

binsLep_

std::vector<std::string> CovarianceMatrix::binsLep_

private

Definition at line 68 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), getResolution(), and setupMatrix().

binsMet_

std::vector<std::string> CovarianceMatrix::binsMet_

private

Definition at line 68 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), getResolution(), and setupMatrix().

binsUdsc_

std::vector<std::string> CovarianceMatrix::binsUdsc_

private

vector of strings for the binning of the resolutions

Definition at line 68 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), getResolution(), and setupMatrix().

funcEtaB_

std::vector<std::string> CovarianceMatrix::funcEtaB_

private

Definition at line 71 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcEtaLep_

std::vector<std::string> CovarianceMatrix::funcEtaLep_

private

Definition at line 71 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcEtaMet_

std::vector<std::string> CovarianceMatrix::funcEtaMet_

private

Definition at line 71 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcEtaUdsc_

std::vector<std::string> CovarianceMatrix::funcEtaUdsc_

private

Definition at line 71 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcEtB_

std::vector<std::string> CovarianceMatrix::funcEtB_

private

Definition at line 70 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcEtLep_

std::vector<std::string> CovarianceMatrix::funcEtLep_

private

Definition at line 70 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcEtMet_

std::vector<std::string> CovarianceMatrix::funcEtMet_

private

Definition at line 70 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcEtUdsc_

std::vector<std::string> CovarianceMatrix::funcEtUdsc_

private

vectors for the resolution functions

Definition at line 70 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcPhiB_

std::vector<std::string> CovarianceMatrix::funcPhiB_

private

Definition at line 72 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcPhiLep_

std::vector<std::string> CovarianceMatrix::funcPhiLep_

private

Definition at line 72 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcPhiMet_

std::vector<std::string> CovarianceMatrix::funcPhiMet_

private

Definition at line 72 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

funcPhiUdsc_

std::vector<std::string> CovarianceMatrix::funcPhiUdsc_

private

Definition at line 72 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getResolution().

jetEnergyResolutionEtaBinning_

const std::vector<double> CovarianceMatrix::jetEnergyResolutionEtaBinning_

private

Definition at line 75 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getEtaDependentScaleFactor().

jetEnergyResolutionScaleFactors_

const std::vector<double> CovarianceMatrix::jetEnergyResolutionScaleFactors_

private

scale factors for the jet energy resolution

Definition at line 74 of file CovarianceMatrix.h.

Referenced by CovarianceMatrix(), and getEtaDependentScaleFactor().