RhoDecayTreeFitter Class Reference

#include <RhoDecayTreeFitter.h>

Inheritance diagram for RhoDecayTreeFitter:

Public Types
enum	FitStatus {   UnFitted = -1, Success = 0, Failed, BadInput,   NonConverged }

Public Member Functions
	RhoDecayTreeFitter (RhoCandidate *bc, int verbosity=0)
	constructor from the particle (decay head) More...
	RhoDecayTreeFitter (RhoCandidate *bc, const RhoVector3Err &pv, int verbosity=0)
	constructor from the particle (decay head) and primary vertex More...
	RhoDecayTreeFitter (RhoCandidate *bc, const RhoLorentzVectorErr &lv, int verbosity=0)
	constructor from the particle (decay head) and pbeam More...
	RhoDecayTreeFitter (RhoCandidate *bc, const RhoLorentzVectorErr &lv, const RhoVector3Err &pv, int verbosity=0)
	constructor from the particle (decay head), pbeam and primary vertex More...
	~RhoDecayTreeFitter ()
	destructor More...
void	setMassConstraint (RhoCandidate *cand, bool add=true)
	Add or remove a mass constraint. More...
void	setMassConstraint (RhoCandidate *cand, double mass)
	Add a constraint to a mass different from the property table mass. More...
Bool_t	Fit (int maxNumberOfIterations=10, double deltaChisquareConverged=0.01)
	Add or remove a mass constraintfor a certain ParticleID. More...
Bool_t	Fit ()
Bool_t	FitAll (int maxit=10, double dchiq=0.01)
Bool_t	FitAll ()
std::string	print () const
	Print the result of the fit. More...
std::ostream &	fillStream (std::ostream &s) const
	Print the result of the fit. More...
RhoCandidate *	particle () const
	The top level particle that is fitted. More...
RhoCandidate *	fitParams (RhoCandidate *p=0) const
double	chiSquare () const
	Total chisquare. More...
ChiSquare	chiSquare (RhoCandidate *p) const
	get the chisquare of everything 'downstream' of a particle More...
int	nDof () const
	Total number of DOFs. More...
FitStatus	status () const
	Status of fit. More...
int	nIter () const
	Number of iterations used by vertex fit. More...
RhoDoubleErr	decayLengthSum (RhoCandidate *, RhoCandidate *) const
bool	updateCand (RhoCandidate *cand) const
	update a particlular candidate in the tree More...
bool	updateTree (RhoCandidate *cand) const
	update a particlular candidate in the tree More...
int	errCode ()
	error code More...
void	setVerbose (int i)
	set the verbosity level (for debugging only) More...
void	SetToleranceZ (double z)
DecayChain *	GetDecayChainDirty ()
RecoTrackStateProvider *	extrapolator ()
	get the extrapolator More...
void	setStateProvider (RecoTrackStateProvider *extrapolator)
	set the track extrapolator More...
double	Chi2Contribution (const RhoCandidate *)
double	GetChi2 () const
int	GetNdf () const
double	GetProb () const
void	SetVerbose (Bool_t v=kTRUE)

Protected Member Functions
RhoDoubleErr	decayLengthSum (const ParticleBase &, const ParticleBase &) const
void	updateCand (const ParticleBase &pb, RhoCandidate *cand) const
std::string	name (RhoCandidate *cand) const
	Name of a particle in the decay tree. More...
DecayChain *	decaychain ()
const DecayChain *	decaychain () const
FitParams *	fitparams ()
const FitParams *	fitparams () const
double	globalChiSquare () const
double	add (RhoCandidate *cand)
double	remove (RhoCandidate *cand)
void	updateIndex ()
void	fit (int maxNumberOfIterations=10, double deltaChisquareConverged=0.01)
	Fit the decay tree. More...
void	fitOneStep ()
	Fit just one step. More...
RhoCandidate *	fittedCand (RhoCandidate *cand, RhoCandidate *headoftree) const
RhoCandidate &	HeadOfTree () const
virtual Bool_t	FitNode (RhoCandidate *b)
RhoCandidate *	CopyCand (RhoCandidate *)
	uppermost particle composite in tree More...
RhoCandidate *	CopyTree (RhoCandidate *)
void	InsertChi2 (const RhoCandidate *bc, const double chi2)
void	SetDaugthersFromComposite (RhoCandidate *cand)
void	FindAndAddFinalStateDaughters (RhoCandidate *cand)
void	SetFourMomentumByDaughters (RhoCandidate *composite)
void	SetDecayVertex (RhoCandidate *composite, const TVector3 &vtx, const TMatrixD &CovVV)

Protected Attributes
RecoTrackStateProvider *	m_extrapolator
	track extrapolator (if needed) More...
Bool_t	fVerbose
RhoCandidate *	fHeadOfTree
std::vector< RhoCandidate * >	fDaughters
double	fChiSquare
int	fNDegreesOfFreedom

Detailed Description

Definition at line 24 of file RhoDecayTreeFitter.h.

Member Enumeration Documentation

enum DecayTreeFitter::Fitter::FitStatus

inherited

Enumerator
UnFitted
Success
Failed
BadInput
NonConverged

Definition at line 59 of file Fitter.h.

                   {
      UnFitted     = -1 ,
      Success      =  0 ,
      Failed            ,
      BadInput          ,
      NonConverged
    } ;

Constructor & Destructor Documentation

RhoDecayTreeFitter::RhoDecayTreeFitter	(	RhoCandidate *	bc,
		int	verbosity = 0
	)

constructor from the particle (decay head)

Definition at line 10 of file RhoDecayTreeFitter.cxx.

: Fitter(bc,new DecayTreeFitter::RecoTrackStateProvider(0.05), verbosity)
{
  // we create the Stateprovider so we have to delete it ourselves later!
}

RhoDecayTreeFitter::RhoDecayTreeFitter	(	RhoCandidate *	bc,
		const RhoVector3Err &	pv,
		int	verbosity = 0
	)

constructor from the particle (decay head) and primary vertex

Definition at line 17 of file RhoDecayTreeFitter.cxx.

: Fitter(bc, pv,new DecayTreeFitter::RecoTrackStateProvider(0.05), verbosity)
{
  // we create the Stateprovider so we have to delete it ourselves later!
}

RhoDecayTreeFitter::RhoDecayTreeFitter	(	RhoCandidate *	bc,
		const RhoLorentzVectorErr &	lv,
		int	verbosity = 0
	)

constructor from the particle (decay head) and pbeam

Definition at line 24 of file RhoDecayTreeFitter.cxx.

: Fitter(bc, lv,new DecayTreeFitter::RecoTrackStateProvider(0.05), verbosity)
{
  // we create the Stateprovider so we have to delete it ourselves later!
}

RhoDecayTreeFitter::RhoDecayTreeFitter	(	RhoCandidate *	bc,
		const RhoLorentzVectorErr &	lv,
		const RhoVector3Err &	pv,
		int	verbosity = 0
	)

constructor from the particle (decay head), pbeam and primary vertex

Definition at line 32 of file RhoDecayTreeFitter.cxx.

: Fitter(bc, lv, pv,new DecayTreeFitter::RecoTrackStateProvider(0.05), verbosity)
{
  // we create the Stateprovider so we have to delete it ourselves later!
}

RhoDecayTreeFitter::~RhoDecayTreeFitter

(

)

destructor

Definition at line 40 of file RhoDecayTreeFitter.cxx.

{
//  if(m_extrapolator) delete m_extrapolator;
}

Member Function Documentation

double DecayTreeFitter::Fitter::add ( RhoCandidate *  cand )

protectedinherited

Definition at line 324 of file Fitter.cxx.

References DecayTreeFitter::ParticleBase::addToConstraintList(), bp, DecayTreeFitter::ParticleBase::initCov(), DecayTreeFitter::ParticleBase::initPar1(), DecayTreeFitter::ParticleBase::initPar2(), offset(), and DecayTreeFitter::ParticleBase::updateIndex().

{
  // first obtain a map
  //ParticleBase::indexmap indexmap ;
  //m_decaychain->mother()->addToMap( indexmap ) ;
  // add this track
std::cout<<"Fitter::add(RhoCandidate*) - start"<<std::endl;
  ParticleBase* bp = m_decaychain->mother()->addDaughter(cand, Configuration()) ;
  int offset = m_fitparams->dim() ;
  double deltachisq(999) ;
  if( bp ) {
    bp->updateIndex(offset) ;
    // reassign indices
    //int offset(0) ;
    //m_decaychain->mother()->updIndex(offset) ;
    // resize the fitparams
std::cout<<"Fitter::add(RhoCandidate*) - resize"<<std::endl;
    m_fitparams->resize(offset) ;
    // initialize the added track, filter and return the delta chisq
    bp->initPar1(m_fitparams) ;
    bp->initPar2(m_fitparams) ;
    bp->initCov(m_fitparams) ;

std::cout<<"Fitter::add(RhoCandidate*) - constrains"<<std::endl;
    ParticleBase::constraintlist constraints ;
    bp->addToConstraintList(constraints,0) ;
    double chisq = m_fitparams->chiSquare() ;
    ErrCode estatus ;
    for( ParticleBase::constraintlist::const_iterator it = constraints.begin() ;
        it != constraints.end(); ++it)
      estatus |= it->filter(m_fitparams) ;

    deltachisq = m_fitparams->chiSquare() - chisq ;
    fChiSquare = m_fitparams->chiSquare() ;

std::cout<<"Fitter::add(RhoCandidate*) - init dchain"<<std::endl;
    // we want this somewhere else, but too much work now
    decaychain()->initConstraintList() ;

    //    print() ;
  } else {
    std::cout << "cannot add track to this vertex ..."
    << m_decaychain->mother()->type() << std::endl ;
  }
  return deltachisq ;
}

Double_t RhoFitterBase::Chi2Contribution ( const RhoCandidate *  b )

inherited

access to the fitted candidates

Definition at line 86 of file RhoFitterBase.cxx.

References Double_t, RhoFitterBase::fChi2Map, uid(), and RhoCandidate::Uid().

{
  if(!b) return -999.;
  Int_t uid = b->Uid();
  Double_t chi2=fChi2Map[uid];
  return chi2 >=0.0 ? chi2 : -1.;
}

double DecayTreeFitter::Fitter::chiSquare ( ) const

inlineinherited

Total chisquare.

Definition at line 124 of file Fitter.h.

References RhoFitterBase::fChiSquare.

{ return fChiSquare ; }

ChiSquare DecayTreeFitter::Fitter::chiSquare ( RhoCandidate *  p ) const

inherited

get the chisquare of everything 'downstream' of a particle

Definition at line 912 of file Fitter.cxx.

{
  return m_decaychain->chiSquare(aParticle, m_fitparams) ;
}

RhoCandidate * RhoFitterBase::CopyCand ( RhoCandidate *  b )

protectedinherited

uppermost particle composite in tree

Definition at line 51 of file RhoFitterBase.cxx.

References RhoFactory::Instance(), RhoFactory::NewCandidate(), RhoCandidate::RemoveAssociations(), and RhoCandidate::SetFit().

Referenced by RhoFitterBase::CopyTree().

{
  RhoCandidate* newCand = RhoFactory::Instance()->NewCandidate ( b );
  newCand->RemoveAssociations();
  b->SetFit(newCand);//ready to be modified
  return newCand;
}

RhoCandidate * RhoFitterBase::CopyTree ( RhoCandidate *  head )

protectedinherited

Definition at line 61 of file RhoFitterBase.cxx.

References RhoFitterBase::CopyCand(), RhoCandidate::Daughter(), i, RhoCandidate::IsComposite(), RhoCandidate::NDaughters(), and RhoCandidate::SetMotherLink().

Referenced by RhoFitterBase::RhoFitterBase().

{
  //std::cout<<"\n\tcopy tree "<<head->Uid()<<" "<<&head<<" "<<head->PdgCode()<<" "<<head->NDaughters()<<"...";
  RhoCandidate* headcopy=CopyCand(head);
  RhoCandidate* daucopy=0;
  RhoCandidate* dau=0;
  for(Int_t i=0;i<head->NDaughters();i++)
  {
    dau=head->Daughter(i);
    //std::cout<<"  daugter "<<dau->Uid()<<" "<<i<<" "<<dau->PdgCode()<<" at "<<dau<<"  ";
    if(dau == head) {
      std::cout<<endl<<"*** Candidate is its own mother???  *** \n"<<std::endl;
      std::cout<<"  print:   "<<*head<<std::endl;;
    }
    assert(dau != head);
    if(dau->IsComposite()) daucopy=CopyTree(dau);
        else daucopy=CopyCand(dau);
        //std::cout<<"CopyTree: copied candidate "<<dau->Uid()<<std::endl;
    daucopy->SetMotherLink(headcopy); //daughter link is set automatically, too
  }
  return headcopy;
}

DecayChain* DecayTreeFitter::Fitter::decaychain ( )

inlineprotectedinherited

Definition at line 187 of file Fitter.h.

References DecayTreeFitter::Fitter::m_decaychain.

{ return m_decaychain  ; }

const DecayChain* DecayTreeFitter::Fitter::decaychain ( ) const

inlineprotectedinherited

Definition at line 189 of file Fitter.h.

References DecayTreeFitter::Fitter::m_decaychain.

{ return m_decaychain  ; }

RhoDoubleErr DecayTreeFitter::Fitter::decayLengthSum ( RhoCandidate *  candA, RhoCandidate *  candB  ) const

inherited

Compute the decay length sum of two particles in the decay tree (useful for e.g. B->DD)

Definition at line 514 of file Fitter.cxx.

References DecayTreeFitter::ParticleBase::lenIndex(), DecayTreeFitter::ParticleBase::locate(), and DecayTreeFitter::ParticleBase::mother().

{
  // returns the decaylengthsum of two particles (use ful for e.g. B->DD)
  RhoDoubleErr rc(0,0) ;
  const ParticleBase* pbA = m_decaychain->locate(candA) ;
  const ParticleBase* pbB = m_decaychain->locate(candB) ;
  if(pbA && pbB && pbA->mother() && pbB->mother() ) {
    int lenindexA = pbA->lenIndex() ;
    int lenindexB = pbB->lenIndex() ;
    if( lenindexA>=0 && lenindexB>=0) {
      double lenA    = m_fitparams->par()(lenindexA) ;
      double lenB    = m_fitparams->par()(lenindexB) ;
      double cov =
      m_fitparams->cov()(lenindexA, lenindexA) +
      m_fitparams->cov()(lenindexB, lenindexB) +
      2*m_fitparams->cov()(lenindexA, lenindexB) ;

      // rc = VtxDoubleErr(lenA+lenB,std::sqrt(cov)) ;
      rc = RhoDoubleErr( lenA+lenB , cov) ;
    }
  }
  return rc ;
}

RhoDoubleErr DecayTreeFitter::Fitter::decayLengthSum ( const ParticleBase &  , const ParticleBase &    ) const

protectedinherited

int DecayTreeFitter::Fitter::errCode ( )

inlineinherited

error code

Definition at line 158 of file Fitter.h.

References DecayTreeFitter::Fitter::m_errCode.

{ return m_errCode ; }

RecoTrackStateProvider* DecayTreeFitter::Fitter::extrapolator ( )

inlineinherited

get the extrapolator

Definition at line 172 of file Fitter.h.

References DecayTreeFitter::Fitter::m_extrapolator.

{ return m_extrapolator ; }

std::ostream & DecayTreeFitter::Fitter::fillStream ( std::ostream &  s ) const

inherited

Print the result of the fit.

Definition at line 908 of file Fitter.cxx.

References print().

{ return s << print() ; }

void RhoFitterBase::FindAndAddFinalStateDaughters ( RhoCandidate *  cand )

protectedinherited

Definition at line 149 of file RhoFitterBase.cxx.

References RhoCandidate::Daughter(), RhoFitterBase::fDaughters, RhoCandidate::IsComposite(), RhoCandidate::IsLocked(), and RhoCandidate::NDaughters().

Referenced by Rho4CFitter::Fit(), RhoKinFitter::Fit(), and Rho4CFitter::FitConserveMasses().

{
  RhoCandidate* tc;
  for (int k=0;k<cand->NDaughters();k++) {
    tc=cand->Daughter(k);
    if (!tc->IsComposite() || tc->IsLocked()) { fDaughters.push_back(tc); }
    else { FindAndAddFinalStateDaughters(tc); }
  }
  return;
}

Bool_t RhoFitterBase::Fit ( )

inherited

This function has to be overridden by the real algoritms implementations

Definition at line 95 of file RhoFitterBase.cxx.

References RhoFitterBase::fChi2Map, RhoFitterBase::fHeadOfTree, and RhoFitterBase::FitNode().

Referenced by ana_check(), ana_check_psi(), ana_complete(), ana_day1(), ana_invariantmass_2pi_tpc(), ana_multi(), anaDMesonsCharged(), anaideal_complete(), anatut_psi2s(), PndTripleAnaTask::DsDs2317Analysis(), PndPmtTask::Exec(), PndProdAnaTask::Exec(), PndLLbarAnaTask::Exec(), PndMyAnalysisTask::Exec(), PndSimpleCombinerTask::Exec(), DecayTreeFitter::Fitter::FitAll(), PndTripleAnaTask::JpsiAnalysis(), newana_check_eta(), poormantracks(), run_ana_eta_c_stt(), run_ana_eta_c_stt_v2(), run_ana_eta_c_tpc(), run_ana_invariantmass_2pi_stt(), run_ana_invariantmass_2pi_tpc(), run_ana_invariantmass_4pi_stt(), run_ana_invariantmass_4pi_tpc(), run_ana_mertens_evt7(), tut_ana(), tut_ana_fast(), tut_ana_fit(), tut_ana_ntp(), tut_ana_ntp_noqa(), and tut_ana_ntp_qa().

{
  fChi2Map.clear();
  return FitNode(fHeadOfTree);
}

Bool_t DecayTreeFitter::Fitter::Fit ( int  maxNumberOfIterations = 10, double  deltaChisquareConverged = 0.01  )

inlineinherited

Add or remove a mass constraintfor a certain ParticleID.

Add a constraint to a mass different from the property table mass Fit the decay tree

Definition at line 103 of file Fitter.h.

References DecayTreeFitter::Fitter::fit(), DecayTreeFitter::Fitter::m_particle, DecayTreeFitter::Fitter::m_status, and DecayTreeFitter::Fitter::updateTree().

    {
      fit(maxNumberOfIterations,deltaChisquareConverged); 
      bool check = updateTree(m_particle);
      check = check && (m_status==0);
      return check;
    };

void DecayTreeFitter::Fitter::fit ( int  maxNumberOfIterations = 10, double  deltaChisquareConverged = 0.01  )

protectedinherited

Fit the decay tree.

Definition at line 171 of file Fitter.cxx.

References fabs(), DecayTreeFitter::ErrCode::fastdivergingfit, max(), print(), DecayTreeFitter::ErrCode::slowdivergingfit, and vtxverbose.

Referenced by DecayTreeFitter::Fitter::Fit().

{
if(vtxverbose>5){std::cout<<"Fitter::fit(): - start"<<std::endl;}
//FIXME it's not used?
  //    m_map.clear() ;

  const int maxndiverging=3 ;
  //const double dChisqQuit = nDof() ; // if chi2 increases by more than this --> fit failed

  // initialize
  fChiSquare = -1 ;
  // m_errCode.reset() ;
  m_errCode = 0 ;

  if( m_status == UnFitted )
  { 
    if(vtxverbose>5){std::cout<<"Fitter::fit(): - unfitted: init decaychain"<<std::endl;}
    m_errCode = m_decaychain->init(m_fitparams).flag() ; 
  }
  if(vtxverbose>3){std::cout<<"Fitter:fit(): decaychain init ended with errcode "<<m_errCode<<std::endl;}
  // if(m_errCode.failure()) {
  if ( 0 != m_errCode )
  {
    // the input tracks are too far apart
    m_status = BadInput ;

  } else {
    // reset the status flag
    m_status = UnFitted ;

    int ndiverging=0 ;
    bool finished = false ;

    for(m_niter=0; m_niter<nitermax && !finished; ++m_niter) 
    {
      TVectorD prevpar = m_fitparams->par() ;
      bool firstpass = m_niter==0 ;
      if(vtxverbose>5){std::cout<<"Fitter::fit(): - filtering"<<std::endl;}
      m_errCode = m_decaychain->filter(m_fitparams,firstpass).flag() ;
      if(vtxverbose>5){std::cout<<"Fitter::fit(): - filtering done"<<std::endl;}
      double chisq = m_fitparams->chiSquare() ;
      double deltachisq = chisq - fChiSquare ;
      // if chi2 increases by more than this --> fit failed
      const double dChisqQuit = std::max(double(2*nDof()),2*fChiSquare) ;

      // if(m_errCode.failure()) {
      if( 0 != m_errCode ) {
        finished = true ;
        m_status = Failed ;
        if(vtxverbose>5){std::cout<<"Fitter::fit(): - ow, we got an error: "<<m_errCode<<std::endl;}
      } else {
        if( m_niter>0 ) {
          if( fabs( deltachisq ) < dChisqConv ) {
            fChiSquare = chisq ;
            m_status = Success ;
            finished = true ;
            if(vtxverbose>5){std::cout<<"Fitter::fit(): - yay! we filtered successfully"<<std::endl;}
          } else if( m_niter>1 && deltachisq > dChisqQuit ) {
            m_fitparams->par() = prevpar ;
            m_status  = Failed ;
            m_errCode = ErrCode::fastdivergingfit ;
            finished = true ;
            if(vtxverbose>5){std::cout<<"Fitter::fit(): - fast diverging"<<std::endl;}
          } else if( deltachisq > 0 && ++ndiverging>=maxndiverging) {
            m_fitparams->par() = prevpar ;
            m_status = NonConverged ;
            m_errCode = ErrCode::slowdivergingfit ;
            finished = true ;
            if(vtxverbose>5){std::cout<<"Fitter::fit(): - slow diverging"<<std::endl;}
          } else if( deltachisq > 0 ) {
            if(vtxverbose>5){std::cout<<"Fitter::fit(): - well, smaller steps now"<<std::endl;}
            // make a half step and reduce stepsize
            m_fitparams->par()   += prevpar ;
            m_fitparams->par()   *= 0.5 ;
            //if( m_niter>10) m_fitparams->scale() *= 0.5 ;
          }
        }
        if ( deltachisq < 0 ) ndiverging=0 ; // start over with counting
        if(!finished) fChiSquare = chisq ;
      }
      fNDegreesOfFreedom = m_fitparams->nDof();

      if(vtxverbose>=1) {
        std::cout << "Fitter::fit(): step, stat, errcode, ndf (ncon, dim) , chiSquare, dchisq: "
        << std::setw(3) << m_niter
        << std::setw(3) << m_status<<"\t"
        << std::setw(3) << m_errCode<<"\t"
        << std::setw(3) << nDof() << " ("
        << std::setw(3) << m_fitparams->nConstraints() << ", "
        << std::setw(3) << m_fitparams->dim() << ") "
        << std::setprecision(6)
        << std::setw(12) << chisq
        << std::setw(12) << deltachisq << std::endl ;
      }

      if(vtxverbose>=4) {
        std::cout << print()            << std::endl;
        std::cout << "press a key ...." << std::endl ;
        getchar() ;
      }
    }

    if( m_niter == nitermax && m_status != Success )
    { m_status = NonConverged ; }

    //m_decaychain->mother()->forceP4Sum(m_fitparams) ;

    if(m_status == Success ) {
      if(vtxverbose>5){std::cout<<"Fitter::fit(): - test cov"<<std::endl;}
      if( !(m_fitparams->testCov() ) ) {
        //static int counter(10) ;
        //if( --counter>=0)
        std::cout << "DecayTreeFitterter::Fitter: Error matrix not positive definite. "
          << "Changing status to failed." << std::endl ;
        m_status = Failed ;
        //print() ;
      }
    }
  }
  if(vtxverbose>5){std::cout<<"Fitter::fit(): - done"<<std::endl;}
}

Bool_t RhoFitterBase::FitAll ( )

inherited

Definition at line 101 of file RhoFitterBase.cxx.

References RhoFitterBase::fHeadOfTree, RhoCandidate::IsLocked(), and RhoFitterBase::IterateAndFit().

{
  if(fHeadOfTree->IsLocked())
  {
    Warning("RhoFitterBase::FitAll","You tried to fit a locked candidate. Retuning kFALSE now.");
    return kFALSE;
  }
  return IterateAndFit(fHeadOfTree);
}

Bool_t DecayTreeFitter::Fitter::FitAll ( int  maxit = 10, double  dchiq = 0.01  )

inlineinherited

Definition at line 110 of file Fitter.h.

References RhoFitterBase::Fit().

{ return Fit(maxit,dchiq); };

Bool_t RhoFitterBase::FitNode ( RhoCandidate *  b )

protectedvirtualinherited

Reimplemented in RhoKinHyperonVtxFitter, RhoKinVtxFitter, and RhoKalmanVtxFitter.

Definition at line 130 of file RhoFitterBase.cxx.

Referenced by RhoFitterBase::Fit(), and RhoFitterBase::IterateAndFit().

{
  Warning("RhoFitterBase::FitNode","Method not implemented in %s",this->GetName());
  return kFALSE;
}

void DecayTreeFitter::Fitter::fitOneStep ( )

protectedinherited

Fit just one step.

Definition at line 294 of file Fitter.cxx.

References vtxverbose.

{
  bool firstpass = m_status==UnFitted ;
  if( firstpass ) m_decaychain->init(m_fitparams) ;
  m_decaychain->filter(m_fitparams,firstpass) ;
  fChiSquare = m_fitparams->chiSquare() ;
  fNDegreesOfFreedom = m_fitparams->nDof();
  if(vtxverbose>=1)
    std::cout << "In Fitter::fitOneStep(): " << m_status << " " << firstpass << " " << fChiSquare << " " << fNDegreesOfFreedom << std::endl ;
  m_status = Success ;
}

RhoCandidate* DecayTreeFitter::Fitter::fitParams ( RhoCandidate *  p = 0 ) const

inlineinherited

Currently the only accessor to the actual fitted data

Parameters

p	(INPUT) the particle the fitted parameters ( 0 for invaild parameters/fits)

Definition at line 121 of file Fitter.h.

References p.

{return p->GetFit();};

FitParams* DecayTreeFitter::Fitter::fitparams ( )

inlineprotectedinherited

Definition at line 188 of file Fitter.h.

References DecayTreeFitter::Fitter::m_fitparams.

{ return m_fitparams   ; }

const FitParams* DecayTreeFitter::Fitter::fitparams ( ) const

inlineprotectedinherited

Definition at line 190 of file Fitter.h.

References DecayTreeFitter::Fitter::m_fitparams.

{ return m_fitparams   ; }

RhoCandidate* DecayTreeFitter::Fitter::fittedCand ( RhoCandidate *  cand, RhoCandidate *  headoftree  ) const

protectedinherited

double RhoFitterBase::GetChi2 ( ) const

inlineinherited

Definition at line 48 of file RhoFitterBase.h.

References RhoFitterBase::fChiSquare.

Referenced by ana_check(), ana_check_psi(), ana_complete(), ana_day1(), ana_multi(), anaideal_complete(), anatut_psi2s(), PndTripleAnaTask::DsDs2317Analysis(), PndPmtTask::Exec(), PndProdAnaTask::Exec(), PndLLbarAnaTask::Exec(), PndAnaWithTrigger::Exec(), PndMyAnalysisTask::Exec(), PndScrutAnaTask::Exec(), PndSimpleCombinerTask::Exec(), PndTripleAnaTask::JpsiAnalysis(), newana_check_eta(), poormantracks(), PndRhoTupleQA::qaFitter(), run_ana_eta_c_stt(), run_ana_eta_c_stt_v2(), run_ana_eta_c_tpc(), PndTripleAnaTask::ThreePiAnalysis(), tut_ana(), tut_ana_fast(), tut_ana_fit(), tut_ana_ntp(), tut_ana_ntp_noqa(), and tut_ana_ntp_qa().

{return fChiSquare;};

DecayChain* DecayTreeFitter::Fitter::GetDecayChainDirty ( )

inlineinherited

Definition at line 165 of file Fitter.h.

References DecayTreeFitter::Fitter::m_decaychain.

{ return m_decaychain  ; }

int RhoFitterBase::GetNdf ( ) const

inlineinherited

Definition at line 49 of file RhoFitterBase.h.

References RhoFitterBase::fNDegreesOfFreedom.

Referenced by poormantracks(), and PndRhoTupleQA::qaFitter().

{return fNDegreesOfFreedom;};

double RhoFitterBase::GetProb ( ) const

inlineinherited

Definition at line 50 of file RhoFitterBase.h.

References RhoFitterBase::fChiSquare, and RhoFitterBase::fNDegreesOfFreedom.

Referenced by ana_complete(), ana_day1(), ana_multi(), anaideal_complete(), PndPmtTask::Exec(), PndLLbarAnaTask::Exec(), poormantracks(), PndRhoTupleQA::qaFitter(), tut_ana(), tut_ana_fast(), tut_ana_fit(), tut_ana_ntp(), tut_ana_ntp_noqa(), and tut_ana_ntp_qa().

{return TMath::Prob(fChiSquare,fNDegreesOfFreedom);};

double DecayTreeFitter::Fitter::globalChiSquare ( ) const

protectedinherited

Definition at line 427 of file Fitter.cxx.

{
  return m_decaychain->chiSquare(m_fitparams) ;
}

RhoCandidate& RhoFitterBase::HeadOfTree ( ) const

inlineprotectedinherited

Definition at line 57 of file RhoFitterBase.h.

References RhoFitterBase::fHeadOfTree.

{ return *fHeadOfTree; }

void RhoFitterBase::InsertChi2 ( const RhoCandidate *  bc, const double  chi2  )

inlineprotectedinherited

Definition at line 66 of file RhoFitterBase.h.

References RhoFitterBase::fChi2Map, and RhoCandidate::Uid().

Referenced by RhoKalmanVtxFitter::Calculate().

{fChi2Map[ bc->Uid()] = chi2;}

std::string DecayTreeFitter::Fitter::name ( RhoCandidate *  cand ) const

protectedinherited

Name of a particle in the decay tree.

Definition at line 539 of file Fitter.cxx.

References DecayTreeFitter::ParticleBase::locate(), and DecayTreeFitter::ParticleBase::name().

{
  const ParticleBase* pb = m_decaychain->locate(cand) ;
  return pb ? pb->name() : "Not found" ;
}

int DecayTreeFitter::Fitter::nDof ( ) const

inherited

Total number of DOFs.

Definition at line 319 of file Fitter.cxx.

                                  {
  return fNDegreesOfFreedom;
}

int DecayTreeFitter::Fitter::nIter ( ) const

inlineinherited

Number of iterations used by vertex fit.

Definition at line 130 of file Fitter.h.

References DecayTreeFitter::Fitter::m_niter.

{ return m_niter  ; }

RhoCandidate* DecayTreeFitter::Fitter::particle ( ) const

inlineinherited

The top level particle that is fitted.

Definition at line 116 of file Fitter.h.

References DecayTreeFitter::Fitter::m_particle.

{ return m_particle ; }

std::string DecayTreeFitter::Fitter::print ( ) const

inherited

Print the result of the fit.

Definition at line 307 of file Fitter.cxx.

References s, and status.

{
  std::ostringstream s ;
  m_decaychain->mother()->print(m_fitparams) ;
  s << "chisq,ndof,ncontr,niter,status,errcode: "
  << chiSquare() << " "
  << nDof() << " " << m_fitparams->nConstraints() << " "
  << nIter() << " " << status() << " " << m_errCode << std::endl ;
  return s.str() ;
}

double DecayTreeFitter::Fitter::remove ( RhoCandidate *  cand )

protectedinherited

Definition at line 371 of file Fitter.cxx.

References DecayTreeFitter::ParticleBase::addToConstraintList(), DecayTreeFitter::ParticleBase::mother(), and DecayTreeFitter::ParticleBase::removeDaughter().

{
  ParticleBase* pb = const_cast<ParticleBase*>(m_decaychain->locate(cand)) ;
  ErrCode estatus ;
  double dchisq(0) ;
  if(pb) {
    // filter with negative weight
    ParticleBase::constraintlist constraints ;
    pb->addToConstraintList(constraints,0) ;
    double chisq = m_fitparams->chiSquare() ;
    for( ParticleBase::constraintlist::iterator it = constraints.begin() ;
        it != constraints.end(); ++it) {
      it->setWeight(-1) ;
      estatus |= it->filter(m_fitparams) ;
    }
    dchisq = chisq - m_fitparams->chiSquare() ;
    // remove
    ParticleBase* themother = const_cast<ParticleBase*>(pb->mother()) ;
    if(themother) themother->removeDaughter(pb);
  }
  return dchisq ;
}

void RhoFitterBase::SetDaugthersFromComposite ( RhoCandidate *  cand )

protectedinherited

Definition at line 136 of file RhoFitterBase.cxx.

References RhoCandidate::Daughter(), RhoFitterBase::fDaughters, RhoCandidate::IsComposite(), and RhoCandidate::NDaughters().

Referenced by RhoKinHyperonFitter::Fit(), RhoKinVtxFitter::FitNode(), RhoKinHyperonVtxFitter::FitNode(), and RhoKinHyperonFitter::SetMatrices().

{
  fDaughters.clear();
  if(cand->IsComposite()){
    RhoCandidate* tc;
    for (int k=0;k<cand->NDaughters();k++) {
      tc=cand->Daughter(k);
      fDaughters.push_back(tc);
    }
  }
  return;
}

void RhoFitterBase::SetDecayVertex ( RhoCandidate *  composite, const TVector3 &  vtx, const TMatrixD &  CovVV  )

protectedinherited

Definition at line 178 of file RhoFitterBase.cxx.

References RhoCandidate::SetDecayVtx().

Referenced by RhoKalmanVtxFitter::Calculate(), RhoKinVtxFitter::SetOutput(), and RhoKinHyperonVtxFitter::SetOutput().

{
  RhoError decaypointcov(CovVV);
  RhoVector3Err decayvertex(vtx,decaypointcov);
  composite->SetDecayVtx(decayvertex);
}

void RhoFitterBase::SetFourMomentumByDaughters ( RhoCandidate *  composite )

protectedinherited

Definition at line 160 of file RhoFitterBase.cxx.

References RhoCandidate::Cov7(), RhoCandidate::Daughter(), RhoCandidate::IsComposite(), RhoCandidate::IsLocked(), RhoCandidate::NDaughters(), RhoCandidate::P4(), RhoCandidate::SetCov7(), and RhoCandidate::SetP4().

Referenced by RhoKalmanVtxFitter::Calculate(), Rho4CFitter::Fit(), Rho4CFitter::FitConserveMasses(), RhoKinVtxFitter::SetOutput(), and RhoKinFitter::SetOutput().

{
  RhoCandidate* tc;
  TLorentzVector tmpLV;
  TMatrixD tmpCov(7,7);
  // Sum daughter fourmomenta, dive into nodes if necessary
  for (int k=0;k<composite->NDaughters();k++) {
    tc=composite->Daughter(k);
    if (tc->IsComposite() && !tc->IsLocked()) { SetFourMomentumByDaughters(tc); }
    tmpLV += tc->P4();
    tmpCov = tmpCov + tc->Cov7();
  }
  composite->SetP4(tmpLV);
  composite->SetCov7(tmpCov);
  //std::cout<<" Base fitter cov7 from tc "<<tc->Uid()<<"/"<<tc->Charge()<<"/"<<tc->PdgCode()<<": ";tmpCov.Print();
  return;
}

void DecayTreeFitter::Fitter::setMassConstraint ( RhoCandidate *  cand, bool  add = true  )

inherited

Add or remove a mass constraint.

Definition at line 394 of file Fitter.cxx.

{
  m_decaychain->setMassConstraint(bc,badd) ;
  m_status = UnFitted ;
}

void DecayTreeFitter::Fitter::setMassConstraint ( RhoCandidate *  cand, double  mass  )

inherited

Add a constraint to a mass different from the property table mass.

Definition at line 400 of file Fitter.cxx.

{
  m_decaychain->setMassConstraint(bc,mass) ;
  m_status = UnFitted ;
}

void DecayTreeFitter::Fitter::setStateProvider ( RecoTrackStateProvider *  extrapolator )

inherited

set the track extrapolator

Definition at line 154 of file Fitter.cxx.

{ m_extrapolator = aExtrapolator; }

void DecayTreeFitter::Fitter::SetToleranceZ ( double  z )

inlineinherited

Definition at line 161 of file Fitter.h.

References DecayTreeFitter::Fitter::m_extrapolator, and DecayTreeFitter::RecoTrackStateProvider::SetToleranceZ().

{if(m_extrapolator) m_extrapolator->SetToleranceZ(z);};

void RhoFitterBase::SetVerbose ( Bool_t  v = kTRUE )

inlineinherited

Definition at line 52 of file RhoFitterBase.h.

References RhoFitterBase::fVerbose, and v.

{fVerbose=v;}

void DecayTreeFitter::Fitter::setVerbose ( int  i )

inherited

set the verbosity level (for debugging only)

Definition at line 158 of file Fitter.cxx.

References i, and vtxverbose.

{ vtxverbose = i ; }

FitStatus DecayTreeFitter::Fitter::status ( ) const

inlineinherited

Status of fit.

Definition at line 128 of file Fitter.h.

References DecayTreeFitter::Fitter::m_status.

{ return m_status ; }

bool DecayTreeFitter::Fitter::updateCand ( RhoCandidate *  cand ) const

inherited

update a particlular candidate in the tree

return an updated decay tree. this is not a final solution. will try to move more info to Particle methods to retrieve the result in terms of RhoCandidates (note: mother vertex is not updated, and decay length cannot be stored anywhere. Use fitParams instead methods to retrieve the result in terms of RhoCandidates (note: mother vertex is not updated, and decay length cannot be stored anywhere. Use fitParams instead

Definition at line 875 of file Fitter.cxx.

References DecayTreeFitter::ParticleBase::locate().

{
  // assigns fitted parameters to a candidate
  const ParticleBase* pb = m_decaychain->locate(aParticle) ;
  if(pb) updateCand(*pb,aParticle) ;
  //else {
  // this error message does not make sense, because it is also
  // triggered for daughters that were simply not refitted. we
  // have to do something about that.
  //       VtxPrintTree printer(&myvtxprint) ;
  //       ErrMsg(error) << "cann't find candidate " << std::endl
  //       << printer.print(cand)
  //       << "in tree " << std::endl
  //       << printer.print(*_bc)
  //       << endmsg;
  return pb != 0 ;
}

void DecayTreeFitter::Fitter::updateCand ( const ParticleBase &  pb, RhoCandidate *  cand  ) const

protectedinherited

Definition at line 736 of file Fitter.cxx.

References col, energy, RhoCandidate::GetFit(), DecayTreeFitter::ParticleBase::hasEnergy(), i, RhoFactory::Instance(), DecayTreeFitter::ParticleBase::lenIndex(), DecayTreeFitter::ParticleBase::momIndex(), DecayTreeFitter::ParticleBase::mother(), RhoFactory::NewCandidate(), DecayTreeFitter::ParticleBase::pdtMass(), pos, DecayTreeFitter::ParticleBase::posIndex(), RhoCandidate::RemoveAssociations(), row, RhoCandidate::SetCov7(), RhoCandidate::SetFit(), RhoCandidate::SetP4(), RhoCandidate::SetPos(), and sqrt().

{
  // Update the FittedCand of the RhoCandidate
  // This replaces the previous updateCand() as well as fitParams() functions
  int posindex = ( pb.mother() ? pb.mother()->posIndex() : pb.posIndex() ) ;
  if(posindex<0) posindex = pb.posIndex() ;
  int momindex = pb.momIndex() ;
  int lenindex = pb.lenIndex() ;
  TVector3 pos(m_fitparams->par()(posindex+0),
               m_fitparams->par()(posindex+1),
               m_fitparams->par()(posindex+2)) ;
  TLorentzVector p4 ;
  p4.SetPx( m_fitparams->par()(momindex+0) ) ;
  p4.SetPy( m_fitparams->par()(momindex+1) ) ;
  p4.SetPz( m_fitparams->par()(momindex+2) ) ;
  // double decaylength = lenindex>=0 ? m_fitparams->par()(lenindex) : 0 ; // Unused
  TMatrixDSym cov8(8) ;
  //
  if( pb.hasEnergy() ) {
    // if particle has energy, get full p4 from fitparams
    p4.SetE( m_fitparams->par()(momindex+3) ) ;
    int parmap[8] ;
    for(int i=0; i<3; ++i) parmap[i]   = posindex + i ;
    for(int i=0; i<4; ++i) parmap[i+3] = momindex + i ;
    parmap[7] = lenindex ;
    int maxrow = lenindex >=0 ? 8 : 7 ;
    for(int row=0; row<maxrow; ++row)
      //for(int col=0; col<=row; ++col) //TMatrixDSym is not really symmetric :-(
      for(int col=0; col<maxrow; ++col)
        cov8(row,col) = m_fitparams->cov()(parmap[row],parmap[col]) ;
  } else {
    // if not, use the pdttable mass
    TMatrixDSym cov7(7) ;
    int parmap[8] ;
    for(int i=0; i<3; ++i) parmap[i]   = posindex + i ;
    for(int i=0; i<3; ++i) parmap[i+3] = momindex + i ;
    parmap[6] = lenindex ;
    int maxrow = lenindex >=0 ? 7 : 6 ;
    for(int row=0; row<maxrow; ++row)
      //for(int col=0; col<=row; ++col)
      for(int col=0; col<maxrow; ++col)
        cov7(row,col) = m_fitparams->cov()(parmap[row],parmap[col]) ;

    // now fill the jacobian
    double mass = pb.pdtMass() ;
    double energy2 = mass*mass ;
    for(int row=0; row<3; ++row) {
      double px = m_fitparams->par()(momindex+row) ;
      energy2 += px*px ;
    }
    double energy = std::sqrt(energy2) ;
    TMatrixD jacobian(8,7);
    for(int col=0; col<7; ++col)
      jacobian(col,col) = 1;
    for(int col=3; col<6; ++col)
      jacobian(6,col) = m_fitparams->par()(parmap[col])/energy ;

    p4.SetE(energy) ;
    cov8 = cov7.Similarity(jacobian);
  }
  //    // VtxFitParams vtxfitparams(pb.charge(),pos,p4,decaylength,cov8) ;
  RhoCandidate* fittedcand=aParticle->GetFit();
  if(!fittedcand)
  {
    fittedcand = RhoFactory::Instance()->NewCandidate ( aParticle );
    fittedcand->RemoveAssociations();
    aParticle->SetFit(fittedcand);//ready to be modified
  }
  fittedcand->SetPos(pos);
  fittedcand->SetP4(p4);
  fittedcand->SetCov7(cov8.GetSub(0,6,0,6));
}

void DecayTreeFitter::Fitter::updateIndex ( )

protectedinherited

Definition at line 418 of file Fitter.cxx.

References offset().

{
 std::cout<<"Fitter::updateIndex() start"<<std::endl;
  int offset=0 ;
  m_decaychain->mother()->updateIndex(offset) ;
 std::cout<<"Fitter::updateIndex() resize"<<std::endl;
  m_fitparams->resize(offset) ;
}

bool DecayTreeFitter::Fitter::updateTree ( RhoCandidate *  cand ) const

inherited

update a particlular candidate in the tree

Definition at line 894 of file Fitter.cxx.

References RhoCandidate::Daughter(), and RhoCandidate::NDaughters().

Referenced by DecayTreeFitter::Fitter::Fit().

{
  bool rc ;
  if ( (rc = updateCand(p) ) ) {
    //      BOOST_FOREACH( RhoCandidate* daughter, p.daughters() ) {
    //      updateTree( const_cast<RhoCandidate*>(*daughter) ) ;
    for(int iDau=0;iDau<p->NDaughters();iDau++){
      updateTree( const_cast<RhoCandidate*>(p->Daughter(iDau)) ) ;
    }
  }
  return rc ;
}

Member Data Documentation

double RhoFitterBase::fChiSquare

protectedinherited

Definition at line 74 of file RhoFitterBase.h.

Referenced by RhoKalmanVtxFitter::Calculate(), DecayTreeFitter::Fitter::chiSquare(), RhoKinVtxFitter::Compute(), RhoKinHyperonVtxFitter::Compute(), Rho4CFitter::Do4CFit(), Rho4CFitter::Do4CFitWithMassConservation(), RhoKalmanVtxFitter::FitNode(), RhoKalmanVtxFitter::FitVertexFast(), RhoFitterBase::GetChi2(), RhoFitterBase::GetProb(), Rho4CFitter::Rho4CFitter(), RhoKinFitter::Solve(), and RhoKinHyperonFitter::Solve().

std::vector<RhoCandidate*> RhoFitterBase::fDaughters

protectedinherited

Definition at line 69 of file RhoFitterBase.h.

Referenced by RhoKinVtxFitter::Compute(), RhoKinHyperonVtxFitter::Compute(), Rho4CFitter::Do4CFit(), Rho4CFitter::Do4CFitWithMassConservation(), RhoFitterBase::FindAndAddFinalStateDaughters(), Rho4CFitter::Fit(), RhoKinFitter::Fit(), RhoKinHyperonFitter::Fit(), Rho4CFitter::FitConserveMasses(), RhoKinVtxFitter::GetCovariance(), RhoKinHyperonVtxFitter::GetCovariance(), RhoKinFitter::Read4MomKinMatrix(), RhoKinHyperonFitter::Read4MomKinMatrix(), RhoKinVtxFitter::ReadKinMatrix(), RhoKinHyperonVtxFitter::ReadKinMatrix(), RhoKinVtxFitter::ReadMassKinMatrix(), RhoKinFitter::ReadMassKinMatrix(), RhoKinHyperonVtxFitter::ReadMassKinMatrix(), RhoKinVtxFitter::ReadMatrix(), RhoKinFitter::ReadMatrix(), RhoKinHyperonVtxFitter::ReadMatrix(), RhoKinFitter::ReadMomKinMatrix(), RhoKinHyperonFitter::ReadMomKinMatrix(), RhoKinFitter::ReadTotEKinMatrix(), RhoKinHyperonFitter::ReadTotEKinMatrix(), RhoKinFitter::ReadTotMomKinMatrix(), RhoKinHyperonFitter::ReadTotMomKinMatrix(), RhoFitterBase::SetDaugthersFromComposite(), RhoKinVtxFitter::SetMatrices(), RhoKinFitter::SetMatrices(), RhoKinHyperonFitter::SetMatrices(), RhoKinHyperonVtxFitter::SetMatrices(), RhoKinVtxFitter::SetOutput(), RhoKinFitter::SetOutput(), RhoKinHyperonFitter::SetOutput(), RhoKinHyperonVtxFitter::SetOutput(), RhoKinHyperonVtxFitter::TransportToPoca(), RhoKinVtxFitter::TransportToVertex(), and RhoKinHyperonVtxFitter::TransportToVertex().

RhoCandidate* RhoFitterBase::fHeadOfTree

protectedinherited

Definition at line 62 of file RhoFitterBase.h.

Referenced by Rho4CFitter::Fit(), RhoKinFitter::Fit(), RhoKinHyperonFitter::Fit(), RhoFitterBase::Fit(), RhoFitterBase::FitAll(), Rho4CFitter::FitConserveMasses(), RhoFitterBase::HeadOfTree(), RhoKinHyperonFitter::ReadMatrix(), RhoFitterBase::RhoFitterBase(), RhoKalmanVtxFitter::RhoKalmanVtxFitter(), RhoKinHyperonFitter::SetMatrices(), RhoKinFitter::SetOutput(), and RhoKinHyperonFitter::SetOutput().

int RhoFitterBase::fNDegreesOfFreedom

protectedinherited

Definition at line 75 of file RhoFitterBase.h.

Referenced by RhoKalmanVtxFitter::Calculate(), RhoKinVtxFitter::Compute(), RhoKinHyperonVtxFitter::Compute(), Rho4CFitter::Do4CFit(), Rho4CFitter::Do4CFitWithMassConservation(), RhoKalmanVtxFitter::FitVertexFast(), RhoFitterBase::GetNdf(), RhoFitterBase::GetProb(), RhoKinFitter::SetOutput(), and RhoKinHyperonFitter::SetOutput().

Bool_t RhoFitterBase::fVerbose

protectedinherited

Definition at line 58 of file RhoFitterBase.h.

Referenced by RhoKalmanVtxFitter::Calculate(), RhoKinVtxFitter::Compute(), RhoKinHyperonVtxFitter::Compute(), RhoKalmanVtxFitter::FitVertexFast(), RhoKalmanVtxFitter::RhoKalmanVtxFitter(), RhoKalmanVtxFitter::SetDebug(), RhoKinVtxFitter::SetOutput(), RhoKinHyperonVtxFitter::SetOutput(), RhoKalmanVtxFitter::SetSilent(), and RhoFitterBase::SetVerbose().

RecoTrackStateProvider* DecayTreeFitter::Fitter::m_extrapolator

protectedinherited

track extrapolator (if needed)

Definition at line 235 of file Fitter.h.

Referenced by DecayTreeFitter::Fitter::extrapolator(), and DecayTreeFitter::Fitter::SetToleranceZ().

---

The documentation for this class was generated from the following files:

• RhoDecayTreeFitter.h
• RhoDecayTreeFitter.cxx