RhoFitterBase Class Reference

#include <RhoFitterBase.h>

Inheritance diagram for RhoFitterBase:

Public Member Functions
	RhoFitterBase ()
	RhoFitterBase (RhoCandidate *decayTree)
	RhoFitterBase (const RhoFitterBase &)
	~RhoFitterBase ()
Bool_t	Fit ()
Bool_t	FitAll ()
double	Chi2Contribution (const RhoCandidate *)
double	GetChi2 () const
int	GetNdf () const
double	GetProb () const
void	SetVerbose (Bool_t v=kTRUE)

Protected Member Functions
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
Bool_t	fVerbose
RhoCandidate *	fHeadOfTree
std::vector< RhoCandidate * >	fDaughters
double	fChiSquare
int	fNDegreesOfFreedom

Private Member Functions
Bool_t	IterateAndFit (RhoCandidate *b)

Private Attributes
std::map< Int_t, Double_t >	fChi2Map
	! each particle's contribution to the chi^2 More...

Detailed Description

Definition at line 21 of file RhoFitterBase.h.

Constructor & Destructor Documentation

RhoFitterBase::RhoFitterBase ( )

inline

constructor

Definition at line 28 of file RhoFitterBase.h.

{};

RhoFitterBase::RhoFitterBase

(

RhoCandidate *

decayTree

)

Definition at line 30 of file RhoFitterBase.cxx.

References CopyTree(), fChi2Map, and fHeadOfTree.

                                                  : fVerbose ( kFALSE ), fChiSquare(-9999), fNDegreesOfFreedom(-9999)
{
  //fHeadOfTree=CopyCand( head );
  //TODO do we want to copy the trere here and now?
  fHeadOfTree=CopyTree(head);
  fChi2Map.clear();
}

RhoFitterBase::RhoFitterBase

(

const RhoFitterBase &

other

)

Definition at line 42 of file RhoFitterBase.cxx.

References CopyTree(), and fHeadOfTree.

                                                          : TObject(other), fVerbose ( other.fVerbose )
{
  //fHeadOfTree=CopyCand(*other.fHeadOfTree);
  //TODO do we want to copy the tree here and now?
  fHeadOfTree=CopyTree(other.fHeadOfTree);
}

RhoFitterBase::~RhoFitterBase

(

)

Definition at line 38 of file RhoFitterBase.cxx.

{
}

Member Function Documentation

Double_t RhoFitterBase::Chi2Contribution

(

const RhoCandidate *

b

)

access to the fitted candidates

Definition at line 86 of file RhoFitterBase.cxx.

References Double_t, 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.;
}

RhoCandidate * RhoFitterBase::CopyCand ( RhoCandidate *  b )

protected

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 CopyTree().

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

RhoCandidate * RhoFitterBase::CopyTree ( RhoCandidate *  head )

protected

Definition at line 61 of file RhoFitterBase.cxx.

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

Referenced by 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;
}

void RhoFitterBase::FindAndAddFinalStateDaughters ( RhoCandidate *  cand )

protected

Definition at line 149 of file RhoFitterBase.cxx.

References RhoCandidate::Daughter(), 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

(

)

This function has to be overridden by the real algoritms implementations

Definition at line 95 of file RhoFitterBase.cxx.

References fChi2Map, fHeadOfTree, and 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 RhoFitterBase::FitAll

(

)

Definition at line 101 of file RhoFitterBase.cxx.

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

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

Bool_t RhoFitterBase::FitNode ( RhoCandidate *  b )

protectedvirtual

Reimplemented in RhoKinHyperonVtxFitter, RhoKinVtxFitter, and RhoKalmanVtxFitter.

Definition at line 130 of file RhoFitterBase.cxx.

Referenced by Fit(), and IterateAndFit().

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

double RhoFitterBase::GetChi2 ( ) const

inline

Definition at line 48 of file RhoFitterBase.h.

References 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;};

int RhoFitterBase::GetNdf ( ) const

inline

Definition at line 49 of file RhoFitterBase.h.

References fNDegreesOfFreedom.

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

{return fNDegreesOfFreedom;};

double RhoFitterBase::GetProb ( ) const

inline

Definition at line 50 of file RhoFitterBase.h.

References fChiSquare, and 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);};

RhoCandidate& RhoFitterBase::HeadOfTree ( ) const

inlineprotected

Definition at line 57 of file RhoFitterBase.h.

References fHeadOfTree.

{ return *fHeadOfTree; }

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

inlineprotected

Definition at line 66 of file RhoFitterBase.h.

References fChi2Map, and RhoCandidate::Uid().

Referenced by RhoKalmanVtxFitter::Calculate().

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

Bool_t RhoFitterBase::IterateAndFit ( RhoCandidate *  b )

private

Definition at line 112 of file RhoFitterBase.cxx.

References Bool_t, RhoCandidate::Daughter(), FitNode(), i, RhoCandidate::IsComposite(), RhoCandidate::IsLocked(), and RhoCandidate::NDaughters().

Referenced by FitAll().

{
  RhoCandidate* dau=0;
  Bool_t check = kFALSE;
  for(Int_t i=0;i<b->NDaughters();i++)
  {
    dau=b->Daughter(i);
    if (dau->IsComposite() && !dau->IsLocked())
    {
      check = IterateAndFit(dau);
      if(kFALSE==check) return kFALSE;
    }
  }
  check = FitNode(b);
  if(kFALSE==check) return kFALSE;
  return kTRUE;
}

void RhoFitterBase::SetDaugthersFromComposite ( RhoCandidate *  cand )

protected

Definition at line 136 of file RhoFitterBase.cxx.

References RhoCandidate::Daughter(), 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  )

protected

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 )

protected

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 RhoFitterBase::SetVerbose ( Bool_t  v = kTRUE )

inline

Definition at line 52 of file RhoFitterBase.h.

References fVerbose, and v.

{fVerbose=v;}

Member Data Documentation

std::map<Int_t,Double_t> RhoFitterBase::fChi2Map

private

! each particle's contribution to the chi^2

Definition at line 80 of file RhoFitterBase.h.

Referenced by Chi2Contribution(), Fit(), InsertChi2(), and RhoFitterBase().

double RhoFitterBase::fChiSquare

protected

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(), GetChi2(), GetProb(), Rho4CFitter::Rho4CFitter(), RhoKinFitter::Solve(), and RhoKinHyperonFitter::Solve().

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

protected

Definition at line 69 of file RhoFitterBase.h.

Referenced by RhoKinVtxFitter::Compute(), RhoKinHyperonVtxFitter::Compute(), Rho4CFitter::Do4CFit(), Rho4CFitter::Do4CFitWithMassConservation(), FindAndAddFinalStateDaughters(), Rho4CFitter::Fit(), RhoKinHyperonFitter::Fit(), RhoKinFitter::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(), 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

protected

Definition at line 62 of file RhoFitterBase.h.

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

int RhoFitterBase::fNDegreesOfFreedom

protected

Definition at line 75 of file RhoFitterBase.h.

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

Bool_t RhoFitterBase::fVerbose

protected

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 SetVerbose().

---

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

• RhoFitterBase.h
• RhoFitterBase.cxx