CandMassKinFitter Struct Reference

#include <CandMassKinFitter.h>

Inheritance diagram for CandMassKinFitter:

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 ( double  m )

inlineexplicit

Definition at line 7 of file CandMassKinFitter.h.

: mass_( m ) { }

Member Function Documentation

FitQuality CandMassKinFitter::set

(

reco::Candidate &

c

)

const

Definition at line 11 of file CandMassKinFitter.cc.

References TKinFitter::addConstraint(), TKinFitter::addMeasParticle(), TFitConstraintM::addParticle1(), pfSecondaryVertexTagInfos_cfi::constraint, reco::Candidate::daughter(), MillePedeFileConverter_cfg::e, benchmark_cfg::errors, TKinFitter::fit(), reco::Candidate::get(), TAbsFitParticle::getCurr4Vec(), TKinFitter::getNDF(), TKinFitter::getS(), mps_fire::i, reco::Candidate::mass(), dataset::name, reco::Candidate::numberOfDaughters(), reco::Candidate::p4(), p4, HadronAndPartonSelector_cfi::particles, TKinFitter::setMaxDeltaS(), TKinFitter::setMaxF(), TKinFitter::setMaxNbIter(), reco::Candidate::setP4(), and TKinFitter::setVerbosity().

                                                       {
  TKinFitter fitter("CandMassFit", "CandMassFit");
  TString name("dau0");
  size_t daus = c.numberOfDaughters();
  vector<TMatrixD> errors(daus, TMatrix(3,3));
  vector<TVector3> momenta(daus);
  vector<TFitParticleMCCart *> particles(daus, nullptr);
  TFitConstraintM constraint( "MassConstraint", 
                  "MassConstraint", nullptr, nullptr , mass_);
  for ( size_t i = 0; i < daus; ++ i ) {
    const Candidate & dau = * c.daughter( i );
    const Particle::LorentzVector& p4 = dau.p4();
    TMatrixD & err = errors[i];
    TVector3 & mom = momenta[i];
    mom = TVector3( p4.px(), p4.py(), p4.pz() );
    TrackRef trk = dau.get<TrackRef>();
    // dummy errors for now...
    // should play with track parametrization...
    err.Zero();
    err(0,0) = 0.1;
    err(1,1) = 0.1;
    err(2,2) = 0.1;
    fitter.addMeasParticle( particles[i] = new TFitParticleMCCart( name, name, & mom, dau.mass(), & err ) );
    name[3] ++;
    constraint.addParticle1( particles[i] );
  } 
  fitter.addConstraint(& constraint);
  fitter.setMaxNbIter( 30 );
  fitter.setMaxDeltaS( 1e-2 );
  fitter.setMaxF( 1e-1 );
  fitter.setVerbosity( 0 );         
  fitter.fit();
  // if (  fitter->getStatus() != 0 ) throw ...
  TLorentzVector sum( 0, 0, 0, 0 );
  for( size_t i = 0; i < daus; ++ i ) {
    Candidate & dau = * c.daughter( i );
    TFitParticleMCCart * part =  particles[i];
    const TLorentzVector * p4 = part->getCurr4Vec();
    dau.setP4( Particle::LorentzVector( p4->X(), p4->Y(), p4->Z(), p4->T() ) );
    sum += * p4;
    delete particles[i];
  }
  c.setP4( Particle::LorentzVector( sum.X(), sum.Y(), sum.Z(), sum.T() ) );
  return FitQuality( fitter.getS(), fitter.getNDF());
}

Member Data Documentation

double CandMassKinFitter::mass_

private

Definition at line 10 of file CandMassKinFitter.h.