CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
List of all members | Public Member Functions | Private Attributes
CandMassKinFitter Struct Reference

#include <CandMassKinFitter.h>

Inheritance diagram for CandMassKinFitter:
CustomKinFitter

Public Member Functions

 CandMassKinFitter (double m)
 
reco::FitQuality set (reco::Candidate &) const
 

Private Attributes

double mass_
 

Detailed Description

Definition at line 6 of file CandMassKinFitter.h.

Constructor & Destructor Documentation

◆ CandMassKinFitter()

CandMassKinFitter::CandMassKinFitter ( double  m)
inlineexplicit

Definition at line 7 of file CandMassKinFitter.h.

7 : mass_(m) {}

Member Function Documentation

◆ set()

FitQuality CandMassKinFitter::set ( reco::Candidate c) const

Definition at line 11 of file CandMassKinFitter.cc.

11  {
12  TKinFitter fitter("CandMassFit", "CandMassFit");
13  TString name("dau0");
14  size_t daus = c.numberOfDaughters();
15  vector<TMatrixD> errors(daus, TMatrix(3, 3));
16  vector<TVector3> momenta(daus);
17  vector<TFitParticleMCCart*> particles(daus, nullptr);
18  TFitConstraintM constraint("MassConstraint", "MassConstraint", nullptr, nullptr, mass_);
19  for (size_t i = 0; i < daus; ++i) {
20  const Candidate& dau = *c.daughter(i);
21  const Particle::LorentzVector& p4 = dau.p4();
22  TMatrixD& err = errors[i];
23  TVector3& mom = momenta[i];
24  mom = TVector3(p4.px(), p4.py(), p4.pz());
25  TrackRef trk = dau.get<TrackRef>();
26  // dummy errors for now...
27  // should play with track parametrization...
28  err.Zero();
29  err(0, 0) = 0.1;
30  err(1, 1) = 0.1;
31  err(2, 2) = 0.1;
32  fitter.addMeasParticle(particles[i] = new TFitParticleMCCart(name, name, &mom, dau.mass(), &err));
33  name[3]++;
34  constraint.addParticle1(particles[i]);
35  }
36  fitter.addConstraint(&constraint);
37  fitter.setMaxNbIter(30);
38  fitter.setMaxDeltaS(1e-2);
39  fitter.setMaxF(1e-1);
40  fitter.setVerbosity(0);
41  fitter.fit();
42  // if ( fitter->getStatus() != 0 ) throw ...
43  TLorentzVector sum(0, 0, 0, 0);
44  for (size_t i = 0; i < daus; ++i) {
45  Candidate& dau = *c.daughter(i);
47  const TLorentzVector* p4 = part->getCurr4Vec();
48  dau.setP4(Particle::LorentzVector(p4->X(), p4->Y(), p4->Z(), p4->T()));
49  sum += *p4;
50  delete particles[i];
51  }
52  c.setP4(Particle::LorentzVector(sum.X(), sum.Y(), sum.Z(), sum.T()));
53  return FitQuality(fitter.getS(), fitter.getNDF());
54 }

References TKinFitter::addConstraint(), TKinFitter::addMeasParticle(), HltBtagPostValidation_cff::c, HLT_2018_cff::constraint, MillePedeFileConverter_cfg::e, runTheMatrix::err, MessageLogger_cfi::errors, TKinFitter::fit(), reco::Candidate::get(), TKinFitter::getNDF(), TKinFitter::getS(), mps_fire::i, reco::Candidate::mass(), Skims_PA_cff::name, reco::Candidate::p4(), p4, ecalTrigSettings_cff::particles, TKinFitter::setMaxDeltaS(), TKinFitter::setMaxF(), TKinFitter::setMaxNbIter(), reco::Candidate::setP4(), and TKinFitter::setVerbosity().

Member Data Documentation

◆ mass_

double CandMassKinFitter::mass_
private

Definition at line 11 of file CandMassKinFitter.h.

mps_fire.i
i
Definition: mps_fire.py:355
reco::Candidate::setP4
virtual void setP4(const LorentzVector &p4)=0
set 4-momentum
reco::Candidate::mass
virtual double mass() const =0
mass
TFitConstraintM
Definition: TFitConstraintM.h:11
MessageLogger_cfi.errors
errors
Definition: MessageLogger_cfi.py:18
reco::Candidate::get
T get() const
get a component
Definition: Candidate.h:221
TKinFitter
Definition: TKinFitter.h:13
ecalTrigSettings_cff.particles
particles
Definition: ecalTrigSettings_cff.py:11
edm::Ref< TrackCollection >
CandMassKinFitter::mass_
double mass_
Definition: CandMassKinFitter.h:11
part
part
Definition: HCALResponse.h:20
errors
Definition: errors.py:1
visualization-live-secondInstance_cfg.m
m
Definition: visualization-live-secondInstance_cfg.py:72
runTheMatrix.err
err
Definition: runTheMatrix.py:288
FitQuality
FitQuality
Definition: Fit.h:30
p4
double p4[4]
Definition: TauolaWrapper.h:92
TFitParticleMCCart
Definition: TFitParticleMCCart.h:9
HltBtagPostValidation_cff.c
c
Definition: HltBtagPostValidation_cff.py:31
reco::Candidate
Definition: Candidate.h:27
reco::JetExtendedAssociation::LorentzVector
math::PtEtaPhiELorentzVectorF LorentzVector
Definition: JetExtendedAssociation.h:25
HLT_2018_cff.constraint
constraint
Definition: HLT_2018_cff.py:50466
Skims_PA_cff.name
name
Definition: Skims_PA_cff.py:17
reco::Candidate::p4
virtual const LorentzVector & p4() const =0
four-momentum Lorentz vector
MillePedeFileConverter_cfg.e
e
Definition: MillePedeFileConverter_cfg.py:37