RhoKinHyperonFitter Class Reference

#include <RhoKinHyperonFitter.h>

Inheritance diagram for RhoKinHyperonFitter:

Public Member Functions
	RhoKinHyperonFitter (RhoCandidate *b)
virtual	~RhoKinHyperonFitter ()
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 ()
void	SetInputMatrix (TMatrixD M1, TMatrixD V1)
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 RhoKinHyperonFitter.h.

Constructor & Destructor Documentation

RhoKinHyperonFitter::RhoKinHyperonFitter

(

RhoCandidate *

b

)

Definition at line 19 of file RhoKinHyperonFitter.cxx.

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

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

RhoKinHyperonFitter::~RhoKinHyperonFitter ( )

virtual

Definition at line 29 of file RhoKinHyperonFitter.cxx.

{
}

Member Function Documentation

void RhoKinHyperonFitter::Add4MomConstraint

(

TLorentzVector

lv

)

Definition at line 34 of file RhoKinHyperonFitter.cxx.

References f4MomConstraint, and flmm.

{
  f4MomConstraint = 1;
  flmm=lv;
}

void RhoKinHyperonFitter::AddMassConstraint

(

double

mass

)

Definition at line 56 of file RhoKinHyperonFitter.cxx.

References fMass, and fMassConstraint.

{
  fMassConstraint = 1;
  fMass=mass;
}

void RhoKinHyperonFitter::AddMomConstraint

(

TVector3

v

)

Definition at line 40 of file RhoKinHyperonFitter.cxx.

References fmm, fMomConstraint, and v.

{
  fMomConstraint = 1;
  fmm=v;
}

void RhoKinHyperonFitter::AddTotEConstraint

(

double

energy

)

Definition at line 46 of file RhoKinHyperonFitter.cxx.

References energy, fEc, and fTotEConstraint.

{
  fTotEConstraint = 1;
  fEc=energy;
}

void RhoKinHyperonFitter::AddTotMomConstraint

(

double

momentum

)

Definition at line 51 of file RhoKinHyperonFitter.cxx.

References fMom, and fTotMomConstraint.

{
  fTotMomConstraint = 1;
  fMom=momentum;
}

double RhoKinHyperonFitter::Chi2Contribution ( const RhoCandidate &  ) const

inline

Definition at line 38 of file RhoKinHyperonFitter.h.

{Error("RhoKinHyperonFitter::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 RhoKinHyperonFitter::Chi2Diff ( )

inline

Definition at line 36 of file RhoKinHyperonFitter.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(), 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 RhoKinHyperonFitter::Fit

(

)

Definition at line 62 of file RhoKinHyperonFitter.cxx.

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

{
  fDaughters.clear();
  //FindAndAddFinalStateDaughters(fHeadOfTree);
  SetDaugthersFromComposite(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;
  }

if(fMassConstraint >0|| fMomConstraint>0)
{
  SetMatrices();
  ZeroMatrices();
  ReadMatrix();
}
  Bool_t check = Solve();
  SetOutput();
  return check;
}

Bool_t RhoKinHyperonFitter::FitAll ( )

inline

Definition at line 33 of file RhoKinHyperonFitter.h.

{Fatal("RhoKinHyperonFitter::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 RhoKinHyperonFitter::GetPull ( )

inline

Definition at line 37 of file RhoKinHyperonFitter.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 RhoKinHyperonFitter::Read4MomKinMatrix ( )

private

Definition at line 442 of file RhoKinHyperonFitter.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);
  fmD.Zero();
  fmd.Zero();
  int k,i;
  for (k=0; k<nd; k++) {

   TLorentzVector p1=fDaughters[k]->P4();
   m[k][0]=p1.M();

  // double E =alp[k*7+3][0];
   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;
     //if(i==3){
     //fmD[fNc][k*7+i]=E/alp[k*7][0];
     //fmD[fNc+1][k*7+i]=E/alp[k*7+1][0];
     //fmD[fNc+2][k*7+i]=E/alp[k*7+2][0];
     //fmD[fNc+3][k*7+i]=1;
     //}

    }

//  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 RhoKinHyperonFitter::ReadKinMatrix ( )

private

void RhoKinHyperonFitter::ReadMassKinMatrix ( )

private

Definition at line 410 of file RhoKinHyperonFitter.cxx.

References fAl1, fMass, fmD, fmd, and pz.

Referenced by Solve().

{

  fmD.Zero();
  //double Etot = 0.;
  //double Px = 0.;
  //double Py = 0.;
  //double Pz = 0.;
  TMatrixD al1p(fAl1);
  //double a;


    double px = al1p[0][0];
    double py = al1p[1][0];
    double pz = al1p[2][0];
    double E =  al1p[3][0];

  fmd[0][0] = E*E - px*px - py*py - pz*pz - fMass*fMass ;

// V.J. - force head mass to constraint mass
//....................................................
                  fmD[0][0] = -2.*px;
                  fmD[0][1] = -2.*py;;
                  fmD[0][2] = -2.*pz;
                  fmD[0][3] = 2.*E;
                  fmD[0][4] = 0.0;
                  fmD[0][5] = 0.0;
                  fmD[0][6] = 0.0;
//.....................................

}

void RhoKinHyperonFitter::ReadMatrix ( )

private

Definition at line 309 of file RhoKinHyperonFitter.cxx.

References RhoCandidate::Cov7(), fAl0, RhoFitterBase::fHeadOfTree, fMassConstraint, fMomConstraint, fV_al0, i, jj, p1, p2, RhoCandidate::P4(), and RhoCandidate::Pos().

Referenced by Fit().

{

//Here the m.c.f is treated differently with other fit, e.g. 4c

if(fMassConstraint>0||fMomConstraint>0){

//px,py,pz,E,x,y,z
    TLorentzVector p1=fHeadOfTree->P4();
    TVector3 p2=fHeadOfTree->Pos();
    fAl0[0][0]=p1.X();
    fAl0[1][0]=p1.Y();
    fAl0[2][0]=p1.Z();
    fAl0[3][0]=p1.E();
    fAl0[4][0]=p2.X();
    fAl0[5][0]=p2.Y();
    fAl0[6][0]=p2.Z();


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

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

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

    p3Cov=p1Cov;
//cout<<"in readMatrix before trans"<<endl;
//   p3Cov.Print();
//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<<"after trans"<<endl;
 //    p4Cov.Print();
    for(int i=0; i<7; i++) {
      for (int j=0; j<7; j++) {
        fV_al0[i][j]  = p4Cov[i][j];
      }
    }

 }

 }

void RhoKinHyperonFitter::ReadMomKinMatrix ( )

private

Definition at line 490 of file RhoKinHyperonFitter.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 RhoKinHyperonFitter::ReadTotEKinMatrix ( )

private

Definition at line 512 of file RhoKinHyperonFitter.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 RhoKinHyperonFitter::ReadTotMomKinMatrix ( )

private

Definition at line 536 of file RhoKinHyperonFitter.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 Fit(), RhoKinVtxFitter::FitNode(), RhoKinHyperonVtxFitter::FitNode(), and 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 RhoKinHyperonFitter::SetInputMatrix	(	TMatrixD	M1,
		TMatrixD	V1
	)

Definition at line 371 of file RhoKinHyperonFitter.cxx.

References f4MomConstraint, fAl0, fMassConstraint, fMomConstraint, fNumCon, fTotEConstraint, fTotMomConstraint, fV_al0, SetMatrices(), and ZeroMatrices().

{

// Here these Matrix must be read in from the vertex fit
  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();
fAl0=M1;
fV_al0=V1;

}

void RhoKinHyperonFitter::SetMatrices ( )

private

Definition at line 102 of file RhoKinHyperonFitter.cxx.

References fAl0, fAl1, RhoFitterBase::fDaughters, RhoFitterBase::fHeadOfTree, fMassConstraint, fmD, fmd, fmE, fMomConstraint, fmPull, fNc, fNcon, fNiter, fNpar, fNumCon, fNvar, fV_al0, fV_al1, and RhoFitterBase::SetDaugthersFromComposite().

Referenced by Fit(), and SetInputMatrix().

{
  int nd;
 
  if(fMassConstraint>0 || fMomConstraint>0) {nd=1;}
  else{
  fDaughters.clear();
  //FindAndAddFinalStateDaughters(fHeadOfTree);
  SetDaugthersFromComposite(fHeadOfTree);
  nd=fDaughters.size(); 
  //cout<<"SetMatrix nd="<<nd<<endl;
  }
 
  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 RhoKinHyperonFitter::SetOutput ( )

private

Definition at line 198 of file RhoKinHyperonFitter.cxx.

References f4MomConstraint, fAl0, RhoFitterBase::fDaughters, RhoFitterBase::fHeadOfTree, fMassConstraint, fMomConstraint, RhoFitterBase::fNDegreesOfFreedom, fTotEConstraint, fTotMomConstraint, fV_al0, i, m, p1, pos, RhoCandidate::SetCov7(), RhoCandidate::SetP4(), RhoCandidate::SetP7(), and sqrt().

Referenced by Fit().

{


  fNDegreesOfFreedom = 0;
  if(f4MomConstraint >0) 
    //fNDegreesOfFreedom += 4+fDaughters.size();
    fNDegreesOfFreedom += 4;
  else {
    if(fMomConstraint >0) 
      fNDegreesOfFreedom += 3;
    if(fTotEConstraint >0 || fMassConstraint >0 || fTotMomConstraint >0) 
      fNDegreesOfFreedom += 1;
  }



if(fMassConstraint>0 || fMomConstraint>0){
  int nd=1;
  TMatrixD m(nd,1);
    TVector3 pos(fAl0[4][0],fAl0[5][0],fAl0[6][0]);
    TLorentzVector mom4(fAl0[0][0],fAl0[1][0],fAl0[2][0],fAl0[3][0]);
    fHeadOfTree->SetP7(pos,mom4);
      

    TMatrixD p1Cov(7,7);
    TMatrixD HeadCov(7,7);
    for(int i=0; i<7; i++) {
      for (int j=0; j<7; j++) {
        
        p1Cov[i][j]= fV_al0[i][j];
        //fHeadOfTree->SetCov7(p1Cov); //New covariance matrix without correlations
      }
    }
  
//Change from px,py,pz,E,x,y,z
    //         to x,y,z,px,py,pz,E
    for(int i=0; i<7; i++) {
      for(int j=0; j<7; j++) {
        if(i>=4) {
          if(j>=4) {
            HeadCov[i-4][j-4] = p1Cov[i][j];
          } else { HeadCov[i-4][j+3] = p1Cov[i][j]; }
        } else {
          if(j>=4) {
            HeadCov[i+3][j-4] = p1Cov[i][j];
          } else { HeadCov[i+3][j+3] = p1Cov[i][j]; }
        }
      }
    }


fHeadOfTree->SetCov7(HeadCov);
  return;

}else{//in the 4C case, we need to output the p7 and cov7 of the daughters
    int nd=fDaughters.size();
    TMatrixD m(nd,1);
    TLorentzVector tot(0,0.0,0.0,0.0);
    for (int k=0; k<nd; k++) {
    TLorentzVector p1=fDaughters[k]->P4();
    m[k][0]=p1.M();
    TVector3 pos(fAl0[k*7+4][0],fAl0[k*7+5][0],fAl0[k*7+6][0]);
    double E =sqrt(fAl0[k*7+0][0]*fAl0[k*7+0][0]+
                   fAl0[k*7+1][0]*fAl0[k*7+1][0]+
                   fAl0[k*7+2][0]*fAl0[k*7+2][0]+
                   m[k][0]*m[k][0]);

    TLorentzVector mom4(fAl0[k*7+0][0],fAl0[k*7+1][0],fAl0[k*7+2][0],E);
    fDaughters[k]->SetP7(pos,mom4);


    TMatrixD p1Cov(7,7);
    TMatrixD p2Cov(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
      }
    }

//Change from px,py,pz,E,x,y,z
//         to x,y,z,px,py,pz,E

    for(int i=0; i<7; i++) {
      for(int j=0; j<7; j++) {
        if(i>=4) {
          if(j>=4) {
            p2Cov[i-4][j-4] = p1Cov[i][j];
          } else { p2Cov[i-4][j+3] = p1Cov[i][j]; }
        } else {
          if(j>=4) {
            p2Cov[i+3][j-4] = p1Cov[i][j];
          } else { p2Cov[i+3][j+3] = p1Cov[i][j]; }
        }
      }
    }
tot+=mom4;
fDaughters[k]->SetCov7(p2Cov);
  }
fHeadOfTree->SetP4(tot);
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 RhoKinHyperonFitter::Solve ( )

private

Definition at line 146 of file RhoKinHyperonFitter.cxx.

References f4MomConstraint, fabs(), fAl0, fAl1, RhoFitterBase::fChiSquare, fMassConstraint, fmD, fmd, fMomConstraint, fTotEConstraint, fTotMomConstraint, fV_al0, fV_al1, Read4MomKinMatrix(), ReadMassKinMatrix(), ReadMomKinMatrix(), ReadTotEKinMatrix(), and ReadTotMomKinMatrix().

Referenced by Fit().

{
  double ierr; // used to check inversions
  fAl1=fAl0;
  TMatrixD chi2_0(1,1);
  chi2_0[0][0]=10000000.;// Xinying Song added a very large initial chi2
  int j1Max=40;

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

  TMatrixD Vd_inv = fmD*fV_al0*mD_t;
  TMatrixD Vd = Vd_inv.Invert(&ierr);
  TMatrixD lam = Vd* ( fmD*(fAl0 - fAl1) + fmd);
  TMatrixD lam_t=lam;
  lam_t.T();
  TMatrixD al_new=fAl0-fV_al0*mD_t*lam;
  TMatrixD V_al_new=fV_al0-fV_al0*mD_t*Vd*fmD*fV_al0;
//  TMatrixD chi2_new = lam_t*(fmD*(fAl0-fAl1)+fmd );
  TMatrixD chi2_new = lam_t*(fmD*(fAl0-al_new)+fmd );

  if(TMath::IsNaN(chi2_new[0][0])) continue;
  double delchi2 = chi2_0[0][0]- chi2_new[0][0];


 if( chi2_new[0][0] < chi2_0[0][0]|| fabs(delchi2)<1.){
   fAl1= al_new;
 //  TMatrixD V_al_new=fV_al0-fV_al0*mD_t*Vd*fmD*fV_al0;
 // fV_al1= V_al_new;
   chi2_0[0][0]= chi2_new[0][0];
  fChiSquare=chi2_new[0][0];
  if(fabs(delchi2)<0.01||j1==j1Max-1){ fV_al1= V_al_new; break;}
   }
}

fAl0=fAl1;
fV_al0=fV_al1;
return kTRUE;


}

void RhoKinHyperonFitter::ZeroMatrices ( )

private

Definition at line 133 of file RhoKinHyperonFitter.cxx.

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

Referenced by Fit(), and SetInputMatrix().

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

Member Data Documentation

int RhoKinHyperonFitter::f4MomConstraint

private

Definition at line 77 of file RhoKinHyperonFitter.h.

Referenced by Add4MomConstraint(), Fit(), RhoKinHyperonFitter(), SetInputMatrix(), SetOutput(), and Solve().

TMatrixD RhoKinHyperonFitter::fAl0

private

Definition at line 53 of file RhoKinHyperonFitter.h.

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

TMatrixD RhoKinHyperonFitter::fAl1

private

Definition at line 54 of file RhoKinHyperonFitter.h.

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

double RhoKinHyperonFitter::fChi2Diff

private

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

double RhoKinHyperonFitter::fEc

private

Definition at line 74 of file RhoKinHyperonFitter.h.

Referenced by AddTotEConstraint(), and ReadTotEKinMatrix().

RhoCandidate* RhoFitterBase::fHeadOfTree

protectedinherited

Definition at line 62 of file RhoFitterBase.h.

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

TLorentzVector RhoKinHyperonFitter::flmm

private

Definition at line 72 of file RhoKinHyperonFitter.h.

Referenced by Add4MomConstraint(), and Read4MomKinMatrix().

double RhoKinHyperonFitter::fMass

private

Definition at line 71 of file RhoKinHyperonFitter.h.

Referenced by AddMassConstraint(), and ReadMassKinMatrix().

int RhoKinHyperonFitter::fMassConstraint

private

Definition at line 80 of file RhoKinHyperonFitter.h.

Referenced by AddMassConstraint(), Fit(), ReadMatrix(), RhoKinHyperonFitter(), SetInputMatrix(), SetMatrices(), SetOutput(), and Solve().

TMatrixD RhoKinHyperonFitter::fmD

private

Definition at line 58 of file RhoKinHyperonFitter.h.

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

TMatrixD RhoKinHyperonFitter::fmd

private

Definition at line 60 of file RhoKinHyperonFitter.h.

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

TMatrixD RhoKinHyperonFitter::fmE

private

Definition at line 59 of file RhoKinHyperonFitter.h.

Referenced by SetMatrices(), and ZeroMatrices().

TVector3 RhoKinHyperonFitter::fmm

private

Definition at line 73 of file RhoKinHyperonFitter.h.

Referenced by AddMomConstraint(), and ReadMomKinMatrix().

double RhoKinHyperonFitter::fMom

private

Definition at line 75 of file RhoKinHyperonFitter.h.

Referenced by AddTotMomConstraint(), and ReadTotMomKinMatrix().

int RhoKinHyperonFitter::fMomConstraint

private

Definition at line 78 of file RhoKinHyperonFitter.h.

Referenced by AddMomConstraint(), Fit(), ReadMatrix(), RhoKinHyperonFitter(), SetInputMatrix(), SetMatrices(), SetOutput(), and Solve().

TMatrixD RhoKinHyperonFitter::fmPull

private

Definition at line 61 of file RhoKinHyperonFitter.h.

Referenced by SetMatrices(), and ZeroMatrices().

Int_t RhoKinHyperonFitter::fNc

private

Definition at line 66 of file RhoKinHyperonFitter.h.

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

Int_t RhoKinHyperonFitter::fNcon

private

Definition at line 65 of file RhoKinHyperonFitter.h.

Referenced by SetMatrices().

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

Int_t RhoKinHyperonFitter::fNiter

private

Definition at line 67 of file RhoKinHyperonFitter.h.

Referenced by SetMatrices().

Int_t RhoKinHyperonFitter::fNpar

private

Definition at line 64 of file RhoKinHyperonFitter.h.

Referenced by SetMatrices().

Int_t RhoKinHyperonFitter::fNumCon

private

Definition at line 69 of file RhoKinHyperonFitter.h.

Referenced by Fit(), SetInputMatrix(), and SetMatrices().

Int_t RhoKinHyperonFitter::fNunKnown

private

Definition at line 68 of file RhoKinHyperonFitter.h.

Int_t RhoKinHyperonFitter::fNvar

private

Definition at line 63 of file RhoKinHyperonFitter.h.

Referenced by SetMatrices().

double RhoKinHyperonFitter::fPull

private

Definition at line 84 of file RhoKinHyperonFitter.h.

Referenced by GetPull().

int RhoKinHyperonFitter::fTotEConstraint

private

Definition at line 79 of file RhoKinHyperonFitter.h.

Referenced by AddTotEConstraint(), Fit(), RhoKinHyperonFitter(), SetInputMatrix(), SetOutput(), and Solve().

int RhoKinHyperonFitter::fTotMomConstraint

private

Definition at line 81 of file RhoKinHyperonFitter.h.

Referenced by AddTotMomConstraint(), Fit(), RhoKinHyperonFitter(), SetInputMatrix(), SetOutput(), and Solve().

TMatrixD RhoKinHyperonFitter::fV_al0

private

Definition at line 55 of file RhoKinHyperonFitter.h.

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

TMatrixD RhoKinHyperonFitter::fV_al1

private

Definition at line 56 of file RhoKinHyperonFitter.h.

Referenced by SetMatrices(), Solve(), 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:

• RhoKinHyperonFitter.h
• RhoKinHyperonFitter.cxx