RhoKinFitter Class Reference

#include <RhoKinFitter.h>

Inheritance diagram for RhoKinFitter:

Public Member Functions
	RhoKinFitter (RhoCandidate *b)
virtual	~RhoKinFitter ()
void	Add4MomConstraint (TLorentzVector lv)
void	AddMomConstraint (TVector3 v)
void	AddTotEConstraint (double energy)
void	AddTotMomConstraint (double momentum)
void	AddMassConstraint (double mass)
Bool_t	Fit ()
Bool_t	FitAll ()
double	Chi2Diff ()
double	GetPull ()
double	Chi2Contribution (const RhoCandidate &) const
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
void	SetMatrices ()
void	ZeroMatrices ()
void	ReadMatrix ()
void	ReadKinMatrix ()
void	ReadMassKinMatrix ()
void	Read4MomKinMatrix ()
void	ReadMomKinMatrix ()
void	ReadTotEKinMatrix ()
void	ReadTotMomKinMatrix ()
Bool_t	Solve ()
void	SetOutput ()

Private Attributes
TMatrixD	fAl0
TMatrixD	fAl1
TMatrixD	fV_al0
TMatrixD	fV_al1
TMatrixD	fmD
TMatrixD	fmE
TMatrixD	fmd
TMatrixD	fmPull
Int_t	fNvar
Int_t	fNpar
Int_t	fNcon
Int_t	fNc
Int_t	fNiter
Int_t	fNunKnown
Int_t	fNumCon
double	fMass
TLorentzVector	flmm
TVector3	fmm
double	fEc
double	fMom
int	f4MomConstraint
int	fMomConstraint
int	fTotEConstraint
int	fMassConstraint
int	fTotMomConstraint
double	fChi2Diff
double	fPull

Detailed Description

Definition at line 21 of file RhoKinFitter.h.

Constructor & Destructor Documentation

RhoKinFitter::RhoKinFitter

(

RhoCandidate *

b

)

Definition at line 19 of file RhoKinFitter.cxx.

References f4MomConstraint, fMassConstraint, fMomConstraint, fTotEConstraint, and fTotMomConstraint.

                                           :
  RhoFitterBase( b )
{
  f4MomConstraint=-1;
  fMomConstraint=-1;
  fMassConstraint =-1;
  fTotMomConstraint=-1;
  fTotEConstraint=-1;
}

RhoKinFitter::~RhoKinFitter ( )

virtual

Definition at line 29 of file RhoKinFitter.cxx.

{
}

Member Function Documentation

void RhoKinFitter::Add4MomConstraint

(

TLorentzVector

lv

)

Definition at line 34 of file RhoKinFitter.cxx.

References f4MomConstraint, and flmm.

Referenced by ana_complete(), ana_day1(), ana_jpsi(), ana_multi(), anaideal_complete(), PndTripleAnaTask::DsDs2317Analysis(), PndAnaWithTrigger::Exec(), PndScrutAnaTask::Exec(), PndSimpleCombinerTask::Exec(), PndTripleAnaTask::JpsiAnalysis(), PndTripleAnaTask::ThreePiAnalysis(), tut_ana(), tut_ana_fast(), tut_ana_fit(), tut_ana_ntp(), tut_ana_ntp_noqa(), and tut_ana_ntp_qa().

{
  f4MomConstraint = 1;
  flmm=lv;
}

void RhoKinFitter::AddMassConstraint

(

double

mass

)

Definition at line 56 of file RhoKinFitter.cxx.

References fMass, and fMassConstraint.

Referenced by ana_complete(), ana_day1(), ana_multi(), anaideal_complete(), anatut_psi2s(), PndTripleAnaTask::DsDs2317Analysis(), PndMyAnalysisTask::Exec(), tut_ana(), tut_ana_fast(), tut_ana_fit(), tut_ana_ntp(), tut_ana_ntp_noqa(), and tut_ana_ntp_qa().

{
  fMassConstraint = 1;
  fMass=mass;
}

void RhoKinFitter::AddMomConstraint

(

TVector3

v

)

Definition at line 40 of file RhoKinFitter.cxx.

References fmm, fMomConstraint, and v.

{
  fMomConstraint = 1;
  fmm=v;
}

void RhoKinFitter::AddTotEConstraint

(

double

energy

)

Definition at line 46 of file RhoKinFitter.cxx.

References energy, fEc, and fTotEConstraint.

{
  fTotEConstraint = 1;
  fEc=energy;
}

void RhoKinFitter::AddTotMomConstraint

(

double

momentum

)

Definition at line 51 of file RhoKinFitter.cxx.

References fMom, and fTotMomConstraint.

{
  fTotMomConstraint = 1;
  fMom=momentum;
}

double RhoKinFitter::Chi2Contribution ( const RhoCandidate &  ) const

inline

Definition at line 37 of file RhoKinFitter.h.

{Error("RhoKinFitter::Chi2Contribution","No chi2 contribution available.");return -9999.;};

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 RhoKinFitter::Chi2Diff ( )

inline

Definition at line 35 of file RhoKinFitter.h.

References fChi2Diff.

{return fChi2Diff;}

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

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(), 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 RhoKinFitter::Fit

(

)

Definition at line 62 of file RhoKinFitter.cxx.

References Bool_t, f4MomConstraint, RhoFitterBase::fDaughters, RhoFitterBase::fHeadOfTree, RhoFitterBase::FindAndAddFinalStateDaughters(), fMassConstraint, fMomConstraint, fNumCon, fTotEConstraint, fTotMomConstraint, ReadMatrix(), SetMatrices(), SetOutput(), Solve(), and ZeroMatrices().

Referenced by ana_complete(), ana_day1(), ana_jpsi(), ana_multi(), anaideal_complete(), anatut_psi2s(), PndTripleAnaTask::DsDs2317Analysis(), PndMyAnalysisTask::Exec(), PndScrutAnaTask::Exec(), PndAnaWithTrigger::Exec(), PndSimpleCombinerTask::Exec(), PndTripleAnaTask::JpsiAnalysis(), PndTripleAnaTask::ThreePiAnalysis(), tut_ana(), tut_ana_fast(), tut_ana_fit(), tut_ana_ntp(), tut_ana_ntp_noqa(), and tut_ana_ntp_qa().

{
  fDaughters.clear();
  FindAndAddFinalStateDaughters(fHeadOfTree);
  //int nd=fDaughters.size(); //unused?

  fNumCon=0;
  if(f4MomConstraint >0) {
    int n4Mom = 4;
    fNumCon = fNumCon + n4Mom;
  }
  if(fMomConstraint >0) {
    int nMom = 3;
    fNumCon = fNumCon + nMom;
  }
  if(fTotEConstraint >0) {
    int nE = 1;
    fNumCon = fNumCon + nE;
  }
  if(fMassConstraint >0) {
    int nMass = 1;
    fNumCon = fNumCon + nMass;
  }
  if(fTotMomConstraint >0) {
    int nTotMom = 1;
    fNumCon = fNumCon + nTotMom;
  }
  SetMatrices();
  ZeroMatrices();
  ReadMatrix();
  Bool_t check = Solve();
  SetOutput();
  return check;
}

Bool_t RhoKinFitter::FitAll ( )

inline

Definition at line 33 of file RhoKinFitter.h.

{Fatal("RhoKinFitter::FitAll()","Not applicable. Please use Fit()."); return kFALSE;};

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

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

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 RhoKinFitter::GetPull ( )

inline

Definition at line 36 of file RhoKinFitter.h.

References fPull.

{return fPull;}

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

void RhoKinFitter::Read4MomKinMatrix ( )

private

Definition at line 381 of file RhoKinFitter.cxx.

References fAl1, RhoFitterBase::fDaughters, flmm, fmD, fmd, fNc, i, m, p1, and sqrt().

Referenced by Solve().

{
  int nd=fDaughters.size();
  TMatrixD alp(fAl1);
  TMatrixD m(nd,1);
  int k,i;
  for (k=0; k<nd; k++) {
    TLorentzVector p1=fDaughters[k]->P4();
    m[k][0]=p1.M();
    double E =sqrt(alp[k*7+0][0]*alp[k*7+0][0]+
                   alp[k*7+1][0]*alp[k*7+1][0]+
                   alp[k*7+2][0]*alp[k*7+2][0]+
                   m[k][0]*m[k][0]);
    for (i=0; i<3; i++) {
      fmD[fNc+i][k*7+i]   = 1;
      fmD[fNc+3][k*7+i]   = alp[k*7+i][0]/E;
    }
//   cout << "value of D" <<  mD[3][k*7+i] << endl;

    for (i=0; i<3; i++) {
      fmd[fNc+i][0] += alp[k*7+i][0];
    }
    fmd[fNc+3][0] += E;
  }
  fmd[fNc+0][0] -= flmm.X();
  fmd[fNc+1][0] -= flmm.Y();
  fmd[fNc+2][0] -= flmm.Z();
  fmd[fNc+3][0] -= flmm.T();
  fNc += 4;
}

void RhoKinFitter::ReadKinMatrix ( )

private

void RhoKinFitter::ReadMassKinMatrix ( )

private

Definition at line 299 of file RhoKinFitter.cxx.

References a, Double_t, fAl1, RhoFitterBase::fDaughters, fMass, fmD, fmd, fNc, RhoCalculationTools::GetBz(), m, p1, pz, and CAMath::Sqrt().

Referenced by Solve().

{
  int nd=fDaughters.size();

  double Etot = 0.;
  double Px = 0.;
  double Py = 0.;
  double Pz = 0.;

  TMatrixD al1p(fAl1);
  double a;
  TMatrixD m(nd,1);
  for(int k=0; k<nd; ++k) {
    int kN=k*7;
    TLorentzVector p1=fDaughters[k]->P4();
    m[k][0]=p1.M();

    double px = al1p[kN+0][0];
    double py = al1p[kN+1][0];
    double pz = al1p[kN+2][0];
    double E = TMath::Sqrt(px*px+py*py+pz*pz+m[k][0]*m[k][0]);
    Etot += E;
    Px   +=px;
    Py +=py;
//   Px += (px - a*delY);
    //  Py += (py + a*delX);
    Pz += pz;
  }

  fmd[fNc+0][0] = Etot*Etot - Px*Px - Py*Py - Pz*Pz - fMass*fMass ;

  for(int k=0; k<nd; ++k) {
    int kN=k*7;
    TLorentzVector p1=fDaughters[k]->P4();
    m[k][0]=p1.M();
    //double px = al1p[kN+0][0]; //[R.K. 01/2017] unused variable?
    //double py = al1p[kN+1][0]; //[R.K. 01/2017] unused variable?
    //double pz = al1p[kN+2][0]; //[R.K. 01/2017] unused variable?
    //double E = TMath::Sqrt(px*px+py*py+pz*pz+m[k][0]*m[k][0]); //[R.K. 01/2017] unused variable?
    Double_t bField = 0.1*RhoCalculationTools::GetBz(fDaughters[k]->Pos()); // T, assume field in z only
    a = -0.00299792458*bField*fDaughters[k]->GetCharge();
    //Double_t invE = 1./E;

//....................................................
// V.J. - force head mass to constraint mass
fmD[fNc+0][kN+0] = -2.*Px;
fmD[fNc+0][kN+1] = -2.*Py;;
fmD[fNc+0][kN+2] = -2.*Pz;
fmD[fNc+0][kN+3] = 2.*Etot;
fmD[fNc+0][kN+4] = 2.*a*Py;
fmD[fNc+0][kN+5] = -2.*a*Px;
fmD[fNc+0][kN+6] = 0.0;
//....................................................
//    fmD[fNc+0][kN+0] = 2.*(Etot*px*invE-Px);
//    fmD[fNc+0][kN+1] = 2.*(Etot*py*invE-Py);
//    fmD[fNc+0][kN+2] = 2.*(Etot*pz*invE-Pz);
//    fmD[fNc+0][kN+3] = 0.0;
//    fmD[fNc+0][kN+3] = 2* m[k][0]*Etot*invE;
//    //  mD[0][kN+4] = 2.*(Etot*py*invE-Py)*a;
//    // mD[0][kN+5] = 2.*(Etot*px*invE-Px)*a;
//    fmD[fNc+0][kN+4] = 2.*Py*a;
//    fmD[fNc+0][kN+5] = -2.*Px*a;
//    fmD[fNc+0][kN+6] = 0.0;
//................Simple....................
    /*
                  mD[fNc+0][kN+0] = -2.*Px;
                  mD[fNc+0][kN+1] = -2.*Py;;
                  mD[fNc+0][kN+2] = -2.*Pz;
                  mD[fNc+0][kN+3] = 2.*Etot;
          //      mD[fNc+0][kN+3] = 2* m[k][0]*Etot*invE;
          //      mD[fNc+0][kN+4] = 2.*(Etot*py*invE-Py)*a;
          //      mD[fNc+0][kN+5] = 2.*(Etot*px*invE-Px)*a;
                  mD[fNc+0][kN+4] = 0.0;
                  mD[fNc+0][kN+5] = 0.0;
                  mD[fNc+0][kN+6] = 0.0;
    */
//.....................................

  }
  fNc += 1;
}

void RhoKinFitter::ReadMatrix ( )

private

Definition at line 233 of file RhoKinFitter.cxx.

References fAl0, RhoFitterBase::fDaughters, fV_al0, i, jj, p1, p2, and sqrt().

Referenced by Fit().

{
  int nd =fDaughters.size();
  for (int k=0; k<nd; k++) {
    int kN=k*7;
//px,py,pz,E,x,y,z
    TLorentzVector p1=fDaughters[k]->P4();
    TVector3 p2=fDaughters[k]->Pos();
    fAl0[kN+0][0]=p1.X();
    fAl0[kN+1][0]=p1.Y();
    fAl0[kN+2][0]=p1.Z();
//    al0[kN+3][0]=p1.E();
    fAl0[kN+4][0]=p2.X();
    fAl0[kN+5][0]=p2.Y();
    fAl0[kN+6][0]=p2.Z();

    double fm=fDaughters[k]->Mass();
    fAl0[kN+3][0]=sqrt(fAl0[kN+0][0]*fAl0[kN+0][0]+ fAl0[kN+1][0]*fAl0[kN+1][0]+fAl0[kN+2][0]*fAl0[kN+2][0]+fm*fm);

// Read Covariance Matrix .... Can read 6x6 matrices..................
    TMatrixD p1Cov(7,7);
    TMatrixD p2Cov(6,6);
    TMatrixD p3Cov(7,7);
    TMatrixD p4Cov(7,7);
    p1Cov=fDaughters[k]->Cov7(); //Cov Matrix x,y,z,px,py,pz,E

    for (int ii=0; ii<6; ii++) {for(int jj=0; jj<6; jj++) {p2Cov[ii][jj]=p1Cov[ii][jj];}} //test

    //Extend matrix for energy for each candidates .....6x6 to 7x7

    TMatrixD J(7,6) ;
    J.Zero();
    TMatrixD J_t(6,7);
    for (int ii=0; ii<6; ii++) {for(int jj=0; jj<6; jj++) {J[ii][jj] = 1;}}
    for(int i=3; i<6; ++i) {J[6][i] = fAl0[kN+i-3][0]/fAl0[kN+3][0];}
//  for (int i=3; i <6; i++) { J[6][i] = -al0[kN+i-3][0]/fm;}
//        J[6][6] =al0[kN+6-3][0] /fm;
//   p3Cov= J*p2Cov*(J_t.Transpose(J));
    p3Cov=p1Cov;

//Change to px,py,pz,E,x,y,z
    for(int i=0; i<7; i++) {
      for(int j=0; j<7; j++) {
        if(i>=3) {
          if(j>=3) {
            p4Cov[i-3][j-3] = p3Cov[i][j];
          } else { p4Cov[i-3][j+4] = p3Cov[i][j]; }
        } else {
          if(j>=3) {
            p4Cov[i+4][j-3] = p3Cov[i][j];
          } else { p4Cov[i+4][j+4] = p3Cov[i][j]; }
        }
      }
    }

// cout<<"p2Cov"<<endl;
    // p2Cov.Print();
    for(int i=0; i<7; i++) {
      for (int j=0; j<7; j++) {
        fV_al0[k*7+i][k*7+j]  = p4Cov[i][j];
      }
    }
  }
}

void RhoKinFitter::ReadMomKinMatrix ( )

private

Definition at line 413 of file RhoKinFitter.cxx.

References fAl1, RhoFitterBase::fDaughters, fmD, fmd, fmm, fNc, i, and m.

Referenced by Solve().

{
  int nd=fDaughters.size();
  TMatrixD alp(fAl1);
  TMatrixD m(nd,1);
  int k,i;
  for (k=0; k<nd; k++) {
    for (i=0; i<3; i++) {
      fmD[fNc+i][k*7+i]   = 1;
    }
    for (i=0; i<3; i++) {
      fmd[fNc+i][0] += alp[k*7+i][0];
    }
  }
  fmd[fNc+0][0]-= fmm.X();
  fmd[fNc+1][0]-= fmm.Y();
  fmd[fNc+2][0]-= fmm.Z();

  fNc += 3;
}

void RhoKinFitter::ReadTotEKinMatrix ( )

private

Definition at line 435 of file RhoKinFitter.cxx.

References fAl1, RhoFitterBase::fDaughters, fEc, fmD, fmd, fNc, i, m, p1, and sqrt().

Referenced by Solve().

{
  int nd=fDaughters.size();
  TMatrixD alp(fAl1);
  TMatrixD m(nd,1);
  int k,i;
  for (k=0; k<nd; k++) {
    TLorentzVector p1=fDaughters[k]->P4();
    m[k][0]=p1.M();
    double E =sqrt(alp[k*7+0][0]*alp[k*7+0][0]+
                   alp[k*7+1][0]*alp[k*7+1][0]+
                   alp[k*7+2][0]*alp[k*7+2][0]+
                   m[k][0]*m[k][0]);
    for (i=0; i<3; i++) {
      fmD[fNc+0][k*7+i]   = 0.;
      // mD[0][k*7+i]  = alp[k*7+i][0]/E;
    }
    fmD[fNc+0][k*7+3]   = 1.;
    fmd[fNc+0][0] += E;
  }
  fmd[fNc+0][0] -= fEc;
  fNc +=1;
}

void RhoKinFitter::ReadTotMomKinMatrix ( )

private

Definition at line 459 of file RhoKinFitter.cxx.

References fAl1, RhoFitterBase::fDaughters, fmD, fmd, fMom, fNc, i, m, and sqrt().

Referenced by Solve().

{
  int nd=fDaughters.size();
  TMatrixD alp(fAl1);
  TMatrixD m(nd,1);
  int k,i;
  for (k=0; k<nd; k++) {
    double Ptot =sqrt(alp[k*7+0][0]*alp[k*7+0][0]+
                      alp[k*7+1][0]*alp[k*7+1][0]+
                      alp[k*7+2][0]*alp[k*7+2][0]);
    for (i=0; i<3; i++) {
      // mD[0][k*7+i]   = 1.;
      fmD[fNc+0][k*7+i]  = alp[k*7+i][0]/Ptot;
    }
    fmd[fNc+0][0] += Ptot;
  }
  fmd[fNc+0][0] -= fMom;
  fNc +=1;
}

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 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 RhoKinFitter::SetMatrices ( )

private

Definition at line 97 of file RhoKinFitter.cxx.

References fAl0, fAl1, RhoFitterBase::fDaughters, fmD, fmd, fmE, fmPull, fNc, fNcon, fNiter, fNpar, fNumCon, fNvar, fV_al0, and fV_al1.

Referenced by Fit().

{
  int nd=fDaughters.size();

  fNvar=7;
  fNpar = nd*fNvar;
  fNcon= fNumCon;
  fNc=0;
  fNiter=0;


  fAl0.ResizeTo(7*nd,1);
  fV_al0.ResizeTo(fNpar,fNpar);
  fAl1.ResizeTo(7*nd,1);
  fV_al1.ResizeTo(fNpar,fNpar);
  fmD.ResizeTo(fNcon,fNpar);
  fmE.ResizeTo(fNcon,3);
  fmd.ResizeTo(fNcon,1);
  fmPull.ResizeTo(7*nd,1);
}

void RhoKinFitter::SetOutput ( )

private

Definition at line 173 of file RhoKinFitter.cxx.

References a, Double_t, f4MomConstraint, fAl0, RhoFitterBase::fDaughters, RhoFitterBase::fHeadOfTree, fMassConstraint, fMomConstraint, RhoFitterBase::fNDegreesOfFreedom, fTotEConstraint, fTotMomConstraint, fV_al0, RhoCalculationTools::GetBz(), i, m, p1, pos, and RhoFitterBase::SetFourMomentumByDaughters().

Referenced by Fit().

{
  int nd=fDaughters.size();
  TMatrixD m(nd,1);
  fNDegreesOfFreedom = 0;
  if(f4MomConstraint >0) 
    fNDegreesOfFreedom += 4;
  else {
    if(fMomConstraint >0) 
      fNDegreesOfFreedom += 3;
    if(fTotEConstraint >0 || fMassConstraint >0 || fTotMomConstraint >0) 
      fNDegreesOfFreedom += 1;
  }
  double  sumA=0;
  double a;

  for (int k=0; k<nd; k++) {
    Double_t bField = 0.1*RhoCalculationTools::GetBz(fDaughters[k]->Pos()); // T, assume field in z only
    a = -0.00299792458*bField*fDaughters[k]->GetCharge();
    sumA += a;
    m[k][0]=fDaughters[k]->P4().M();
    TVector3 pos(fAl0[k*7+4][0],fAl0[k*7+5][0],fAl0[k*7+6][0]);
    TLorentzVector mom4(fAl0[k*7+0][0],fAl0[k*7+1][0],fAl0[k*7+2][0],fAl0[k*7+3][0]);
    TLorentzVector p1;
    p1.SetXYZM(fAl0[k*7+0][0],fAl0[k*7+1][0],fAl0[k*7+2][0],m[k][0]);
    // fDaughters[k]->SetP4(p1);
    //fDaughters[k]->SetP7(pos,mom4);
    fDaughters[k]->SetP7(pos,p1);
      

    TMatrixD p1Cov(7,7);
    for(int i=0; i<7; i++) {
      for (int j=0; j<7; j++) {
        
        p1Cov[i][j]= fV_al0[k*7+i][k*7+j];
        fDaughters[k]->SetCov7(p1Cov); //New covariance matrix without correlations
      }
    }
  }

// For the composite particle ..............................
//  double fpx=0,fpy=0,fpz=0,fe=0;
//  for (int k=0; k<nd; k++) {
//    fpx+= fAl0[k*7+0][0];
//    fpy+= fAl0[k*7+1][0];
//    fpz+= fAl0[k*7+2][0];
//    fe += fAl0[k*7+3][0];
//    //   fe +=fDaughters[k]->P4().E(); // Energy from the initial daughter particles
//  }
//  TLorentzVector sum(fpx,fpy,fpz,fe);
//  fHeadOfTree->SetP4(sum);

  //iteratively set the foumomenta in the tree
  SetFourMomentumByDaughters(fHeadOfTree);
  return;
}

void RhoFitterBase::SetVerbose ( Bool_t  v = kTRUE )

inlineinherited

Definition at line 52 of file RhoFitterBase.h.

References RhoFitterBase::fVerbose, and v.

{fVerbose=v;}

Bool_t RhoKinFitter::Solve ( )

private

Definition at line 132 of file RhoKinFitter.cxx.

References f4MomConstraint, fAl0, fAl1, RhoFitterBase::fChiSquare, fMassConstraint, fmD, fmd, fMomConstraint, fmPull, fNcon, fPull, fTotEConstraint, fTotMomConstraint, fV_al0, i, Read4MomKinMatrix(), ReadMassKinMatrix(), ReadMomKinMatrix(), ReadTotEKinMatrix(), ReadTotMomKinMatrix(), and sqrt().

Referenced by Fit().

{
  //int nd=fDaughters.size(); //unused?
  double ierr; // used to check inversions
  fAl1=fAl0;
//   int j1Max=50;
//  for(Int_t j1=0;j1<j1Max;++j1)
//  {
  if(f4MomConstraint >0) { Read4MomKinMatrix();}
  if(fMomConstraint >0) { ReadMomKinMatrix();}
  if(fTotEConstraint >0) { ReadTotEKinMatrix();}
  if(fMassConstraint >0) { ReadMassKinMatrix();}
  if(fTotMomConstraint >0) { ReadTotMomKinMatrix();}

  TMatrixD mD_t=fmD;
  mD_t.T();
//  mD_t=mD_t.Transpose(mD);

// mD_t.Print();
  TMatrixD Vd_inv = fmD*fV_al0*mD_t;

  TMatrixD Vd = Vd_inv.Invert(&ierr);
  //Vd.Print();
// TMatrixD lam=Vd*md;
  TMatrixD lam = Vd* ( fmD*(fAl1 - fAl0) + fmd);
  TMatrixD al_new=fAl0-fV_al0*mD_t*lam;
//  al_new.Print();
  TMatrixD V_al_new=fV_al0-fV_al0*mD_t*Vd*fmD*fV_al0;
  double chi2=0.;
  for (int i=0; i<fNcon; i++) { chi2+=lam[i][0]*fmd[i][0]; }
  fChiSquare=chi2;
  double covdif=(fV_al0[0][0]-V_al_new[0][0]);
  if (covdif > 0 ) {        fmPull[0][0] =(fAl0[0][0]-al_new[0][0])/sqrt(covdif);}
  fAl0=al_new;
  fV_al0=V_al_new;
  fPull=fmPull[0][0];
  return kTRUE;
}

void RhoKinFitter::ZeroMatrices ( )

private

Definition at line 119 of file RhoKinFitter.cxx.

References fAl0, fAl1, fmD, fmd, fmE, fmPull, fV_al0, and fV_al1.

Referenced by Fit().

{
  fAl0.Zero();
  fV_al0.Zero();
  fAl1.Zero();
  fV_al1.Zero();
  fmPull.Zero();
  fmD.Zero();
  fmd.Zero();
  fmE.Zero();
}

Member Data Documentation

int RhoKinFitter::f4MomConstraint

private

Definition at line 76 of file RhoKinFitter.h.

Referenced by Add4MomConstraint(), Fit(), RhoKinFitter(), SetOutput(), and Solve().

TMatrixD RhoKinFitter::fAl0

private

Definition at line 52 of file RhoKinFitter.h.

Referenced by ReadMatrix(), SetMatrices(), SetOutput(), Solve(), and ZeroMatrices().

TMatrixD RhoKinFitter::fAl1

private

Definition at line 53 of file RhoKinFitter.h.

Referenced by Read4MomKinMatrix(), ReadMassKinMatrix(), ReadMomKinMatrix(), ReadTotEKinMatrix(), ReadTotMomKinMatrix(), SetMatrices(), Solve(), and ZeroMatrices().

double RhoKinFitter::fChi2Diff

private

Definition at line 82 of file RhoKinFitter.h.

Referenced by Chi2Diff().

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(), 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(), RhoKinHyperonFitter::Fit(), Fit(), Rho4CFitter::FitConserveMasses(), RhoKinVtxFitter::GetCovariance(), RhoKinHyperonVtxFitter::GetCovariance(), Read4MomKinMatrix(), RhoKinHyperonFitter::Read4MomKinMatrix(), RhoKinVtxFitter::ReadKinMatrix(), RhoKinHyperonVtxFitter::ReadKinMatrix(), RhoKinVtxFitter::ReadMassKinMatrix(), ReadMassKinMatrix(), RhoKinHyperonVtxFitter::ReadMassKinMatrix(), RhoKinVtxFitter::ReadMatrix(), ReadMatrix(), RhoKinHyperonVtxFitter::ReadMatrix(), ReadMomKinMatrix(), RhoKinHyperonFitter::ReadMomKinMatrix(), ReadTotEKinMatrix(), RhoKinHyperonFitter::ReadTotEKinMatrix(), ReadTotMomKinMatrix(), RhoKinHyperonFitter::ReadTotMomKinMatrix(), RhoFitterBase::SetDaugthersFromComposite(), RhoKinVtxFitter::SetMatrices(), SetMatrices(), RhoKinHyperonFitter::SetMatrices(), RhoKinHyperonVtxFitter::SetMatrices(), RhoKinVtxFitter::SetOutput(), SetOutput(), RhoKinHyperonFitter::SetOutput(), RhoKinHyperonVtxFitter::SetOutput(), RhoKinHyperonVtxFitter::TransportToPoca(), RhoKinVtxFitter::TransportToVertex(), and RhoKinHyperonVtxFitter::TransportToVertex().

double RhoKinFitter::fEc

private

Definition at line 73 of file RhoKinFitter.h.

Referenced by AddTotEConstraint(), and ReadTotEKinMatrix().

RhoCandidate* RhoFitterBase::fHeadOfTree

protectedinherited

Definition at line 62 of file RhoFitterBase.h.

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

TLorentzVector RhoKinFitter::flmm

private

Definition at line 71 of file RhoKinFitter.h.

Referenced by Add4MomConstraint(), and Read4MomKinMatrix().

double RhoKinFitter::fMass

private

Definition at line 70 of file RhoKinFitter.h.

Referenced by AddMassConstraint(), and ReadMassKinMatrix().

int RhoKinFitter::fMassConstraint

private

Definition at line 79 of file RhoKinFitter.h.

Referenced by AddMassConstraint(), Fit(), RhoKinFitter(), SetOutput(), and Solve().

TMatrixD RhoKinFitter::fmD

private

Definition at line 57 of file RhoKinFitter.h.

Referenced by Read4MomKinMatrix(), ReadMassKinMatrix(), ReadMomKinMatrix(), ReadTotEKinMatrix(), ReadTotMomKinMatrix(), SetMatrices(), Solve(), and ZeroMatrices().

TMatrixD RhoKinFitter::fmd

private

Definition at line 59 of file RhoKinFitter.h.

Referenced by Read4MomKinMatrix(), ReadMassKinMatrix(), ReadMomKinMatrix(), ReadTotEKinMatrix(), ReadTotMomKinMatrix(), SetMatrices(), Solve(), and ZeroMatrices().

TMatrixD RhoKinFitter::fmE

private

Definition at line 58 of file RhoKinFitter.h.

Referenced by SetMatrices(), and ZeroMatrices().

TVector3 RhoKinFitter::fmm

private

Definition at line 72 of file RhoKinFitter.h.

Referenced by AddMomConstraint(), and ReadMomKinMatrix().

double RhoKinFitter::fMom

private

Definition at line 74 of file RhoKinFitter.h.

Referenced by AddTotMomConstraint(), and ReadTotMomKinMatrix().

int RhoKinFitter::fMomConstraint

private

Definition at line 77 of file RhoKinFitter.h.

Referenced by AddMomConstraint(), Fit(), RhoKinFitter(), SetOutput(), and Solve().

TMatrixD RhoKinFitter::fmPull

private

Definition at line 60 of file RhoKinFitter.h.

Referenced by SetMatrices(), Solve(), and ZeroMatrices().

Int_t RhoKinFitter::fNc

private

Definition at line 65 of file RhoKinFitter.h.

Referenced by Read4MomKinMatrix(), ReadMassKinMatrix(), ReadMomKinMatrix(), ReadTotEKinMatrix(), ReadTotMomKinMatrix(), and SetMatrices().

Int_t RhoKinFitter::fNcon

private

Definition at line 64 of file RhoKinFitter.h.

Referenced by SetMatrices(), and Solve().

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(), SetOutput(), and RhoKinHyperonFitter::SetOutput().

Int_t RhoKinFitter::fNiter

private

Definition at line 66 of file RhoKinFitter.h.

Referenced by SetMatrices().

Int_t RhoKinFitter::fNpar

private

Definition at line 63 of file RhoKinFitter.h.

Referenced by SetMatrices().

Int_t RhoKinFitter::fNumCon

private

Definition at line 68 of file RhoKinFitter.h.

Referenced by Fit(), and SetMatrices().

Int_t RhoKinFitter::fNunKnown

private

Definition at line 67 of file RhoKinFitter.h.

Int_t RhoKinFitter::fNvar

private

Definition at line 62 of file RhoKinFitter.h.

Referenced by SetMatrices().

double RhoKinFitter::fPull

private

Definition at line 83 of file RhoKinFitter.h.

Referenced by GetPull(), and Solve().

int RhoKinFitter::fTotEConstraint

private

Definition at line 78 of file RhoKinFitter.h.

Referenced by AddTotEConstraint(), Fit(), RhoKinFitter(), SetOutput(), and Solve().

int RhoKinFitter::fTotMomConstraint

private

Definition at line 80 of file RhoKinFitter.h.

Referenced by AddTotMomConstraint(), Fit(), RhoKinFitter(), SetOutput(), and Solve().

TMatrixD RhoKinFitter::fV_al0

private

Definition at line 54 of file RhoKinFitter.h.

Referenced by ReadMatrix(), SetMatrices(), SetOutput(), Solve(), and ZeroMatrices().

TMatrixD RhoKinFitter::fV_al1

private

Definition at line 55 of file RhoKinFitter.h.

Referenced by SetMatrices(), and ZeroMatrices().

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

---

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

• RhoKinFitter.h
• RhoKinFitter.cxx