DecayTreeFitter::RecoComposite Class Reference

#include <RecoComposite.h>

Inheritance diagram for DecayTreeFitter::RecoComposite:

Public Types
enum	ParticleType {   kInteractionPoint, kRecoComposite, kRecoResonance, kInternalParticle,   kRecoTrack, kResonance, kRecoPhoton, kMissingParticle }
typedef std::vector < ParticleBase * >	ParticleContainer
typedef std::vector < ParticleBase * >	daucontainer
typedef daucontainer::const_iterator	const_iterator
typedef std::vector< std::pair < const ParticleBase *, int > >	indexmap
typedef std::vector < DecayTreeFitter::Constraint >	constraintlist

Public Member Functions
	RecoComposite (RhoCandidate *bc, const ParticleBase *mother)
virtual	~RecoComposite ()
virtual int	dim () const
int	dimM () const
ErrCode	projectRecoComposite (const FitParams *, Projection &) const
virtual ErrCode	projectConstraint (Constraint::Type, const FitParams *, Projection &) const
virtual ErrCode	initPar1 (FitParams *)
virtual ErrCode	initPar2 (FitParams *)
virtual int	type () const
virtual int	posIndex () const
virtual int	lenIndex () const
virtual int	momIndex () const
virtual bool	hasEnergy () const
virtual bool	hasPosition () const
virtual void	updCache ()
virtual double	chiSquare (const FitParams *fitparams) const
virtual void	addToConstraintList (constraintlist &alist, int depth) const
virtual void	updateIndex (int &offset)
virtual ErrCode	initCov (FitParams *) const
virtual std::string	parname (int index) const
virtual void	print (const FitParams *) const
const ParticleBase *	locate (RhoCandidate *bc) const
RhoCandidate *	particle () const
virtual int	index () const
const ParticleBase *	mother () const
const std::string &	name () const
virtual ErrCode	projectGeoConstraint (const FitParams *, Projection &) const
virtual ErrCode	projectMassConstraint (const FitParams *, Projection &) const
int	eneIndex () const
virtual double	chiSquareD (const FitParams *) const
double	pdtMass () const
double	pdtWidth () const
double	pdtCLifeTime () const
double	pdtTau () const
int	charge () const
const daucontainer &	daughters () const
const_iterator	begin () const
const_iterator	end () const
ParticleBase *	addDaughter (RhoCandidate *, const Configuration &config)
void	removeDaughter (const ParticleBase *pb)
virtual void	retrieveIndexMap (indexmap &anindexmap) const
void	setMother (const ParticleBase *m)
virtual int	nFinalChargedCandidates () const
void	setParticle (RhoCandidate *bc)
void	collectVertexDaughters (daucontainer &particles, int posindex)
bool	setMassConstraint (bool add)
void	setMassConstraint (double mass)

Static Public Member Functions
static ParticleBase *	createParticle (RhoCandidate *bc, const ParticleBase *mother, const Configuration &config)

Protected Member Functions
daucontainer &	daughters ()
ErrCode	initTau (FitParams *par) const
void	makeName (RhoCandidate *bc)
bool	hasMassConstraint () const
void	setIndex (int i)
void	setName (const std::string &n)

Static Protected Member Functions
static double	pdtCLifeTime (RhoCandidate *bc)
static bool	isAResonance (const TParticlePDG *bc)
static double	bFieldOverC ()

Protected Attributes
TVectorD	m_m
TMatrixDSym	m_matrixV
bool	m_hasEnergy

Detailed Description

Definition at line 19 of file RecoComposite.h.

Member Typedef Documentation

typedef daucontainer::const_iterator DecayTreeFitter::ParticleBase::const_iterator

inherited

Definition at line 97 of file ParticleBase.h.

typedef std::vector<DecayTreeFitter::Constraint> DecayTreeFitter::ParticleBase::constraintlist

inherited

Definition at line 110 of file ParticleBase.h.

typedef std::vector<ParticleBase*> DecayTreeFitter::ParticleBase::daucontainer

inherited

Definition at line 96 of file ParticleBase.h.

typedef std::vector< std::pair<const ParticleBase*,int> > DecayTreeFitter::ParticleBase::indexmap

inherited

Definition at line 105 of file ParticleBase.h.

typedef std::vector<ParticleBase*> DecayTreeFitter::ParticleBase::ParticleContainer

inherited

Definition at line 33 of file ParticleBase.h.

Member Enumeration Documentation

enum DecayTreeFitter::ParticleBase::ParticleType

inherited

Enumerator
kInteractionPoint
kRecoComposite
kRecoResonance
kInternalParticle
kRecoTrack
kResonance
kRecoPhoton
kMissingParticle

Definition at line 28 of file ParticleBase.h.

                      {kInteractionPoint,
      kRecoComposite,kRecoResonance,
      kInternalParticle,kRecoTrack,
      kResonance,kRecoPhoton,
      kMissingParticle} ;

Constructor & Destructor Documentation

DecayTreeFitter::RecoComposite::RecoComposite	(	RhoCandidate *	bc,
		const ParticleBase *	mother
	)

Definition at line 20 of file RecoComposite.cxx.

References updCache().

: ParticleBase(bc,amother),m_m(),m_matrixV(),m_hasEnergy(true)
{
  //bool massconstraint = bc && bc->constraint(RecoConstraint::Mass) ;
  //m_hasEnergy = !massconstraint ;
  updCache() ;
}

DecayTreeFitter::RecoComposite::~RecoComposite ( )

virtual

Definition at line 63 of file RecoComposite.cxx.

{}

Member Function Documentation

ParticleBase * DecayTreeFitter::ParticleBase::addDaughter ( RhoCandidate *  cand, const Configuration &  config  )

inherited

Definition at line 94 of file ParticleBase.cxx.

References DecayTreeFitter::ParticleBase::createParticle().

Referenced by DecayTreeFitter::InternalParticle::InternalParticle().

{
  m_daughters.push_back( DecayTreeFitter::ParticleBase::createParticle(cand,this,config) ) ;
  return m_daughters.back() ;
}

virtual void DecayTreeFitter::RecoComposite::addToConstraintList ( constraintlist &  alist, int  depth  ) const

inlinevirtual

Implements DecayTreeFitter::ParticleBase.

Reimplemented in DecayTreeFitter::RecoResonance.

Definition at line 47 of file RecoComposite.h.

References DecayTreeFitter::Constraint::composite, dimM(), and DecayTreeFitter::Constraint::geometric.

                                                                             {
      alist.push_back( Constraint(this,Constraint::composite,depth,dimM()) ) ;
      alist.push_back( Constraint(this,Constraint::geometric,depth,3) ) ;
    }

const_iterator DecayTreeFitter::ParticleBase::begin ( ) const

inlineinherited

Definition at line 100 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_daughters.

{ return m_daughters.begin() ; }

double DecayTreeFitter::ParticleBase::bFieldOverC ( )

staticprotectedinherited

Definition at line 582 of file ParticleBase.cxx.

References C(), RhoCalculationTools::GetBz(), and pos.

{ 
  TVector3 pos(0.,0.,0.); //TODO get a sensible position, but let's assume zero first...
  return RhoCalculationTools::GetBz ( pos ) / TMath::C() ; 
}

int DecayTreeFitter::ParticleBase::charge ( ) const

inlineinherited

Definition at line 90 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_charge.

{ return m_charge ; }

double DecayTreeFitter::RecoComposite::chiSquare ( const FitParams *  fitparams ) const

virtual

Definition at line 139 of file RecoComposite.cxx.

References DecayTreeFitter::FitParams::dim(), and p.

{
  Projection p(fitparams->dim(),dimM()) ;
  projectRecoComposite(fitparams,p) ;
  return p.chiSquare() ;
}

double DecayTreeFitter::ParticleBase::chiSquareD ( const FitParams *  fitparams ) const

virtualinherited

Definition at line 551 of file ParticleBase.cxx.

{
  double rc = 0;
  for(daucontainer::const_iterator it = m_daughters.begin() ;
      it != m_daughters.end(); ++it)
    rc += (*it)->chiSquareD(fitparams) ;
  return rc ;
}

void DecayTreeFitter::ParticleBase::collectVertexDaughters ( daucontainer &  particles, int  posindex  )

inherited

Definition at line 227 of file ParticleBase.cxx.

References name, and vtxverbose.

{
  // collect all particles emitted from vertex with position posindex
  if(vtxverbose>=3) {
    std::cout << "DecayTreeFitter::ParticleBase::collectVertexDaughters " << posindex << std::endl ;
  }
  //skip: head of tree, particles from different vertex, resonances
  if( mother() && mother()->posIndex() == posindex && type()!=kRecoResonance && type()!=kResonance )
  {
    particles.push_back( this ) ;
    if(vtxverbose>=3) {
      std::cout << "DecayTreeFitter::ParticleBase::collectVertexDaughters - added a particle "<<name()<<" to vertex " << posindex << std::endl ;
    }
  }
  for( daucontainer::const_iterator idau = daughters().begin() ;
      idau != daughters().end() ; ++idau )
    (*idau)->collectVertexDaughters(particles,posindex ) ; //FIXME: RK Caution here!!
    //collectVertexDaughters(particles,posindex ) ;
}

ParticleBase * DecayTreeFitter::ParticleBase::createParticle ( RhoCandidate *  bc, const ParticleBase *  mother, const Configuration &  config  )

staticinherited

Definition at line 113 of file ParticleBase.cxx.

References RhoCandidate::Charge(), fabs(), RhoCandidate::GetRecoCandidate(), DecayTreeFitter::ParticleBase::index(), RhoCandidate::IsLocked(), DecayTreeFitter::ParticleBase::name(), RhoCandidate::NDaughters(), RhoCandidate::PdgCode(), RhoCandidate::PdtEntry(), DecayTreeFitter::ParticleBase::type(), RhoCandidate::Uid(), and vtxverbose.

Referenced by DecayTreeFitter::ParticleBase::addDaughter().

{
  // This routine interpretes a beta candidate as one of the
  // 'Particles' used by the fitter.

  const TParticlePDG* prop = particle->PdtEntry();

  if(vtxverbose>=5)
    std::cout << "DecayTreeFitter::ParticleBase::createParticle from     " <<particle->PdgCode() << " | " << particle->Uid() << std::endl ;
  ParticleBase* rc=0 ;
  //bool bsconstraint = false ;

  // We refit invalid fits, kinematic fits and composites with beamspot
  // constraint if not at head of tree.
  bool validfit = particle->IsLocked();//TODO needed? && particle.endVertex() != 0 ;
  bool iscomposite = (particle->NDaughters()>0) ;
  bool isresonance = iscomposite && prop && isAResonance(prop) ;

  if(!mother) { // 'head of tree' particles
                //if ( bsconstraint )
                //rc = new InteractionPoint(particle) ;
                //else

    if( iscomposite )
    { if(vtxverbose>=2) std::cout<<" H "; if(vtxverbose>=5) std::cout<<std::endl;
      rc = new InternalParticle(particle,0,config) ; // still need proper head-of-tree class
    }
    else {
      std::cout << "DecayTreeFitter::ParticleBase::createParticle: You are fitting a decay tree that exists of "
        << "a single, non-composite particle and which does not have a beamconstraint."
        << "I do not understand what you want me to do with this." << std::endl ;
      std::cout<<" X ";
      rc = new InternalParticle(particle,0,config) ; // still need proper head-of-tree class
    }
  } else if( !iscomposite ) { // external particles (i.e. tracks/neutrals)
    bool hasrecocand = ( NULL !=particle->GetRecoCandidate() );
    bool ischarged = ( fabs(particle->Charge())>0 );
    //bool hastrack = proto && (proto->GetTrackIndex() >= 0) ;
    //bool hascalo  = proto && (proto->GetEmcNumberOfCrystals() > 0) ;
    //std::cout<<" particle="<<particle<<" proto="<<proto<<" ntrk="<<proto->GetTrackIndex()<<" ncry"<<proto->GetEmcNumberOfCrystals()<<std::endl;
    if( hasrecocand )
    {
      if( ischarged )
      { if(vtxverbose>=2) std::cout<<" T "; if(vtxverbose>=5) std::cout<<std::endl;
        rc = new RecoTrack(particle,mother,config) ;  // reconstructed track
      } else
      { if(vtxverbose>=2) std::cout<<" P "; if(vtxverbose>=5) std::cout<<std::endl;
        rc = new RecoPhoton(particle,mother) ; // reconstructed photon
      }
    } else if( validfit ) {  // fitted composites w/o daughters?
      if( isresonance )
      { if(vtxverbose>=2) std::cout<<" RF "; if(vtxverbose>=5) std::cout<<std::endl;
        rc = new RecoResonance(particle,mother) ;
      } else
      { if(vtxverbose>=2) std::cout<<" CF "; if(vtxverbose>=5) std::cout<<std::endl;
        rc = new RecoComposite(particle,mother) ;
      }
    } else { // missing particle!
      if(vtxverbose>=2) std::cout<<" M "; if(vtxverbose>=5) std::cout<<std::endl;
      rc = new MissingParticle(particle,mother) ;
    }
  } else { // 'internal' particles
    if( validfit /*|| isconversion*/ ) {  // fitted composites
      if( isresonance ) { 
        if(vtxverbose>=2) std::cout<<" Rf "; if(vtxverbose>=5) std::cout<<std::endl;
        rc = new RecoResonance(particle,mother) ;
      } else { 
        if(vtxverbose>=2) std::cout<<" Cf "; if(vtxverbose>=5) std::cout<<std::endl;
        rc = new RecoComposite(particle,mother) ;
      }
    } else {         // unfited composites
      if( isresonance ) { 
        if(vtxverbose>=2) std::cout<<" R "; if(vtxverbose>=5) std::cout<<std::endl;
        rc = new Resonance(particle,mother,config) ;
      } else { 
        if(vtxverbose>=2) std::cout<<" I "; if(vtxverbose>=5) std::cout<<std::endl;
        rc = new InternalParticle(particle,mother,config) ;
      }
    }
  }
  if(vtxverbose>=5)
    std::cout << "DecayTreeFitter::ParticleBase::createParticle finished " <<particle->PdgCode() << " | " << particle->Uid() << std::endl ;

  if(vtxverbose>=2)
    std::cout << "DecayTreeFitter::ParticleBase::createParticle returns type=" << rc->type()
    << " index=" << rc->index() << " with name \""<< rc->name() << "\""<<std::endl ;
  return rc ;
}

const daucontainer& DecayTreeFitter::ParticleBase::daughters ( ) const

inlineinherited

Definition at line 99 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_daughters.

Referenced by DecayTreeFitter::InteractionPoint::InteractionPoint(), and DecayTreeFitter::InternalParticle::InternalParticle().

{ return m_daughters ; }

daucontainer& DecayTreeFitter::ParticleBase::daughters ( )

inlineprotectedinherited

Definition at line 138 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_daughters.

{ return m_daughters ; }

virtual int DecayTreeFitter::RecoComposite::dim ( ) const

inlinevirtual

Implements DecayTreeFitter::ParticleBase.

Reimplemented in DecayTreeFitter::RecoResonance.

Definition at line 26 of file RecoComposite.h.

References m_hasEnergy.

{ return m_hasEnergy ? 8 : 7 ; }// (x,y,z,t,px,py,pz,(E))

int DecayTreeFitter::RecoComposite::dimM ( ) const

inline

Definition at line 29 of file RecoComposite.h.

References m_hasEnergy.

Referenced by DecayTreeFitter::RecoResonance::addToConstraintList(), and addToConstraintList().

{ return m_hasEnergy ? 7 : 6 ; }

const_iterator DecayTreeFitter::ParticleBase::end ( ) const

inlineinherited

Definition at line 101 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_daughters.

Referenced by DecayTreeFitter::ParticleBase::locate().

{ return m_daughters.end() ; }

int DecayTreeFitter::ParticleBase::eneIndex ( ) const

inlineinherited

Definition at line 80 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::hasEnergy(), and DecayTreeFitter::ParticleBase::momIndex().

{ return hasEnergy() ? momIndex()+3 : -1 ; }

virtual bool DecayTreeFitter::RecoComposite::hasEnergy ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::ParticleBase.

Definition at line 41 of file RecoComposite.h.

References m_hasEnergy.

Referenced by DecayTreeFitter::RecoResonance::dim().

{ return m_hasEnergy ; }

bool DecayTreeFitter::ParticleBase::hasMassConstraint ( ) const

inlineprotectedinherited

Definition at line 139 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_hasMassConstraint.

Referenced by DecayTreeFitter::MissingParticle::dim(), and DecayTreeFitter::MissingParticle::hasEnergy().

{ return m_hasMassConstraint ; }

virtual bool DecayTreeFitter::RecoComposite::hasPosition ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::ParticleBase.

Definition at line 42 of file RecoComposite.h.

{ return true ; }

virtual int DecayTreeFitter::ParticleBase::index ( ) const

inlinevirtualinherited

Definition at line 59 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_index.

Referenced by DecayTreeFitter::ParticleBase::createParticle(), DecayTreeFitter::DecayChain::index(), DecayTreeFitter::InternalParticle::lenIndex(), lenIndex(), DecayTreeFitter::MissingParticle::momIndex(), DecayTreeFitter::RecoParticle::momIndex(), DecayTreeFitter::RecoResonance::momIndex(), DecayTreeFitter::InternalParticle::momIndex(), DecayTreeFitter::Resonance::momIndex(), momIndex(), DecayTreeFitter::InternalParticle::posIndex(), and posIndex().

{ return m_index ; }

ErrCode DecayTreeFitter::ParticleBase::initCov ( FitParams *  fitparams ) const

virtualinherited

Reimplemented in DecayTreeFitter::InteractionPoint, and DecayTreeFitter::RecoPhoton.

Definition at line 248 of file ParticleBase.cxx.

References DecayTreeFitter::FitParams::cov(), name, row, status, and vtxverbose.

Referenced by DecayTreeFitter::Fitter::add(), and DecayTreeFitter::InteractionPoint::initCov().

{
  ErrCode status ;

  if(vtxverbose>=2) {
    std::cout << "DecayTreeFitter::ParticleBase::initCov for " << name() << std::endl ;
  }

  // position
  int posindex = posIndex() ;
  if( posindex>=0 ) {
    const double sigx   = 10; // * Gaudi::Units::cm ;
    const double sigy   = 10; // * Gaudi::Units::cm ;
    const double sigz   = 50; // * Gaudi::Units::cm ;
    fitparams->cov()(posindex+0,posindex+0) = sigx*sigx ;
    fitparams->cov()(posindex+1,posindex+1) = sigy*sigy ;
    fitparams->cov()(posindex+2,posindex+2) = sigz*sigz ;
  }

  // momentum
  int momindex = momIndex() ;
  if(momindex>=0) {
    // TODO: calo at high energy?
    const double sigmom = 10; // * Gaudi::Units::GeV ; // GeV
    int maxrow = hasEnergy() ? 4 : 3 ;
    for(int row=momindex; row<momindex+maxrow; row++)
      fitparams->cov()(row,row) = sigmom*sigmom ;
  }

  // lifetime
  int lenindex = lenIndex() ;
  if(lenindex>=0) {
    const double sigz   = 50; // * Gaudi::Units::cm ;
    fitparams->cov()(lenindex,lenindex) = sigz*sigz ;
  }

  for(daucontainer::const_iterator it = m_daughters.begin() ;
      it != m_daughters.end() ; ++it)
    status |= (*it)->initCov(fitparams) ;
  return status ;
}

ErrCode DecayTreeFitter::RecoComposite::initPar1 ( FitParams *  fitparams )

virtual

Implements DecayTreeFitter::ParticleBase.

Reimplemented in DecayTreeFitter::RecoResonance.

Definition at line 66 of file RecoComposite.cxx.

References i, DecayTreeFitter::FitParams::par(), row, DecayTreeFitter::ErrCode::success, and vtxverbose.

{
  if(vtxverbose>5){std::cout<<"RecoComposite::initPar1: - "<<std::endl;}
  int posindex = posIndex() ;
  int momindex = momIndex() ;

  //quick map for parameters
  int indmap[7]  ;
  for(int i=0; i<3; ++i) indmap[i]   = posindex+i ;
  for(int i=0; i<4; ++i) indmap[i+3] = momindex+i ;
  // copy the 'measurement'
  for(int row=0; row<dimM(); ++row)
    fitparams->par()(indmap[row]) = m_m(row) ;

  return ErrCode::success ;
}

ErrCode DecayTreeFitter::RecoComposite::initPar2 ( FitParams *  fitparams )

virtual

Implements DecayTreeFitter::ParticleBase.

Reimplemented in DecayTreeFitter::RecoResonance.

Definition at line 84 of file RecoComposite.cxx.

References vtxverbose.

{
  if(vtxverbose>5){std::cout<<"RecoComposite::initPar2: - "<<std::endl;}
  // call default lifetime initialization
  return initTau(fitparams) ;
}

ErrCode DecayTreeFitter::ParticleBase::initTau ( FitParams *  par ) const

protectedinherited

Definition at line 522 of file ParticleBase.cxx.

References mom, DecayTreeFitter::FitParams::par(), DecayTreeFitter::ParticleBase::posIndex(), and DecayTreeFitter::ErrCode::success.

{
  int lenindex = lenIndex() ;
  if(lenindex>=0 && hasPosition() ) {
    const ParticleBase* amother = mother() ;
    int momposindex = amother ? amother->posIndex() : -1 ;
    int posindex = posIndex() ;
    int momindex = momIndex() ;
    assert(momposindex>=0) ; // check code logic: no mother -> no tau
                            //assert(fitparams->par(momposindex+0)!=0 ||fitparams->par(momposindex+1)!=0
                            //      ||fitparams->par(momposindex+2)!=0) ; // mother must be initialized

    TVector3 dX,mom ;
    double mom2(0) ;
    for(int irow=0; irow<3; irow++) {
      dX(irow)  = fitparams->par(posindex+irow) - fitparams->par(momposindex+irow) ;
      double px = fitparams->par(momindex+irow) ;
      mom(irow) = px ;
      mom2 += px*px ;
    }
    double tau = dX.Dot(mom)/mom2 ;
    // we don't like 0 and we don't like very negative values
    if( tau==0 ) tau=pdtTau() ;
    //tau = tau==0 ? pdtTau() : std::max(tau,-pdtTau()) ;
    fitparams->par(lenindex) = tau ;
  }
  return ErrCode::success ;
}

bool DecayTreeFitter::ParticleBase::isAResonance ( const TParticlePDG *  bc )

staticprotectedinherited

Definition at line 205 of file ParticleBase.cxx.

References C(), and vtxverbose.

                                                                  {
  bool rc = false ;
  switch(prop->PdgCode()) {
    case 11: // bremstrahlung is treated as a resonance
    case -11:
      rc = true ;
      break ;
    //case 22: // conversions are not treated as a resonance
    //case 310: // K shorts count as "stable" in PDT Table
      //rc = false;
      //break ;
    default: // this should take care of the pi0
      rc = (prop->Lifetime()>0) && (100.*TMath::C()*prop->Lifetime() < 0.0001); //[cm]
      if(rc) break;
      if(vtxverbose>4)std::cout<<"ParticleBase::isAResonance l."<<__LINE__<<": ctau="<<100.*TMath::C()*prop->Lifetime()<<" cm"<<std::endl;
      if(prop->Stable()) return false;
      //rc = prop.ctau() < 0.001*Gaudi::Units::mm ;
      break;
  }
  return rc ;
}

virtual int DecayTreeFitter::RecoComposite::lenIndex ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::ParticleBase.

Reimplemented in DecayTreeFitter::RecoResonance.

Definition at line 38 of file RecoComposite.h.

References DecayTreeFitter::ParticleBase::index().

{ return index()+3 ; }

const ParticleBase * DecayTreeFitter::ParticleBase::locate ( RhoCandidate *  bc ) const

inherited

Definition at line 353 of file ParticleBase.cxx.

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

Referenced by DecayTreeFitter::Fitter::decayLengthSum(), DecayTreeFitter::DecayChain::locate(), DecayTreeFitter::ParticleBase::locate(), DecayTreeFitter::Fitter::name(), and DecayTreeFitter::Fitter::updateCand().

{
  // does there exist an 'iscloneof' in lhcb?
  const ParticleBase* rc = m_particle == abc ? this : 0 ;
  for(daucontainer::const_iterator it = m_daughters.begin() ;
      !rc && it != m_daughters.end(); ++it)
    rc = (*it)->locate(abc) ;
  return rc ;
}

void DecayTreeFitter::ParticleBase::makeName ( RhoCandidate *  bc )

protectedinherited

virtual int DecayTreeFitter::RecoComposite::momIndex ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::ParticleBase.

Reimplemented in DecayTreeFitter::RecoResonance.

Definition at line 39 of file RecoComposite.h.

References DecayTreeFitter::ParticleBase::index().

{ return index()+4 ; }

const ParticleBase* DecayTreeFitter::ParticleBase::mother ( ) const

inlineinherited

Definition at line 60 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_mother.

Referenced by DecayTreeFitter::DecayChain::chiSquare(), DecayTreeFitter::Fitter::decayLengthSum(), DecayTreeFitter::InternalParticle::dim(), DecayTreeFitter::InternalParticle::lenIndex(), DecayTreeFitter::InternalParticle::momIndex(), DecayTreeFitter::RecoResonance::posIndex(), DecayTreeFitter::Resonance::posIndex(), DecayTreeFitter::Fitter::remove(), and DecayTreeFitter::Fitter::updateCand().

{ return m_mother ; }

const std::string& DecayTreeFitter::ParticleBase::name ( ) const

inlineinherited

Definition at line 61 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_name.

Referenced by DecayTreeFitter::ParticleBase::createParticle(), and DecayTreeFitter::Fitter::name().

{ return m_name ; }

int DecayTreeFitter::ParticleBase::nFinalChargedCandidates ( ) const

virtualinherited

Reimplemented in DecayTreeFitter::RecoTrack.

Definition at line 560 of file ParticleBase.cxx.

Referenced by DecayTreeFitter::sortByType().

                                                               {
  int rc=0;
  for(daucontainer::const_iterator it = m_daughters.begin() ;
      it != m_daughters.end() ; ++it)
    rc +=(*it)->nFinalChargedCandidates() ;
  return rc ;
}

std::string DecayTreeFitter::ParticleBase::parname ( int  index ) const

virtualinherited

Reimplemented in DecayTreeFitter::InternalParticle, DecayTreeFitter::RecoResonance, DecayTreeFitter::Resonance, DecayTreeFitter::MissingParticle, and DecayTreeFitter::RecoParticle.

Definition at line 291 of file ParticleBase.cxx.

References name.

Referenced by DecayTreeFitter::MissingParticle::parname(), DecayTreeFitter::RecoParticle::parname(), DecayTreeFitter::Resonance::parname(), DecayTreeFitter::RecoResonance::parname(), and DecayTreeFitter::InternalParticle::parname().

{
  std::string rc = name() ;
  switch(thisindex) {
    case 0: rc += " x  " ; break ;
    case 1: rc += " y  " ; break ;
    case 2: rc += " z  " ; break ;
    case 3: rc += " len" ; break ;
    case 4: rc += " px " ; break ;
    case 5: rc += " py " ; break ;
    case 6: rc += " pz " ; break ;
    case 7: rc += " E  " ; break ;
    default: ;
  }
  return rc ;
}

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

inlineinherited

Definition at line 57 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_particle.

Referenced by DecayTreeFitter::compTrkTransverseMomentum(), DecayTreeFitter::InteractionPoint::InteractionPoint(), DecayTreeFitter::DecayChain::locate(), and DecayTreeFitter::sortByType().

{ return m_particle ; }

double DecayTreeFitter::ParticleBase::pdtCLifeTime ( ) const

inlineinherited

Definition at line 88 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_pdtCLifeTime.

{ return m_pdtCLifeTime ; }

static double DecayTreeFitter::ParticleBase::pdtCLifeTime ( RhoCandidate *  bc )

staticprotectedinherited

double DecayTreeFitter::ParticleBase::pdtMass ( ) const

inlineinherited

Definition at line 86 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_pdtMass.

Referenced by DecayTreeFitter::InternalParticle::projectMassConstraintTwoBody(), and DecayTreeFitter::Fitter::updateCand().

{ return m_pdtMass ; }

double DecayTreeFitter::ParticleBase::pdtTau ( ) const

inlineinherited

Definition at line 89 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_pdtCLifeTime, and DecayTreeFitter::ParticleBase::m_pdtMass.

{ return m_pdtMass >0 ? m_pdtCLifeTime/m_pdtMass : 0 ; }

double DecayTreeFitter::ParticleBase::pdtWidth ( ) const

inlineinherited

Definition at line 87 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_pdtWidth.

{ return m_pdtWidth ; }

virtual int DecayTreeFitter::RecoComposite::posIndex ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::ParticleBase.

Reimplemented in DecayTreeFitter::RecoResonance.

Definition at line 37 of file RecoComposite.h.

References DecayTreeFitter::ParticleBase::index().

{ return index()   ; }

void DecayTreeFitter::ParticleBase::print ( const FitParams *  fitpar ) const

virtualinherited

Definition at line 309 of file ParticleBase.cxx.

References DecayTreeFitter::FitParams::cov(), i, name, DecayTreeFitter::FitParams::par(), pz, and sqrt().

{
  std::cout << std::setw(5) << "[" << type() << "]" << std::setw(15) << std::flush << name().c_str()
  << std::setw(15)<< " val"  << std::setw(15)<< "err" << std::setw(15) << "sigma^2"<< std::endl ;
  std::cout << std::setprecision(5) ;
  for(int i=0; i<dim(); i++) {
    int theindex = index()+i ;
    std::cout << std::setw(2) << theindex << " "
    << std::setw(20) << parname(i).c_str()
    << std::setw(15) << fitpar->par()(theindex)
    << std::setw(15) << sqrt(fitpar->cov()(theindex,theindex))
    << std::setw(15) << fitpar->cov()(theindex,theindex) <<std::endl ;
  }
  if( hasEnergy() ) {
    int momindex = momIndex() ;
    double E  = fitpar->par()(momindex+3) ;
    double px = fitpar->par()(momindex+0) ;
    double py = fitpar->par()(momindex+1) ;
    double pz = fitpar->par()(momindex+2) ;
    double mass2 = E*E-px*px-py*py-pz*pz ;
    double mass = mass2>0 ? sqrt(mass2) : -sqrt(-mass2) ;

    // TODO this does not work? fitpar->cov()).GetSub(momindex+0,momindex+3)
    TMatrixDSym cov = fitpar->cov().GetSub(momindex+0,momindex+3,momindex+0,momindex+3);
    //HepSymMatrix cov = fitpar->cov().sub(momindex+0,momindex+3) ;
    TVectorD G(4) ; //was at G(4,0) .. ??why
    G(0) = -px/mass ;
    G(1) = -py/mass ;
    G(2) = -pz/mass ;
    G(3) =   E/mass ;
    double massvar = cov.Similarity(G) ;
    std::cout << std::setw(2)<< "-"<<" " << std::setw(20) << "mass: "
    << std::setw(15) << mass
    << std::setw(15) << sqrt(massvar)
    << std::setw(15) << massvar << std::endl ;
  }

  for(daucontainer::const_iterator it = m_daughters.begin() ;
      it != m_daughters.end() ; ++it)
    (*it)->print(fitpar) ;

}

ErrCode DecayTreeFitter::RecoComposite::projectConstraint ( Constraint::Type  aType, const FitParams *  fitparams, Projection &  p  ) const

virtual

Reimplemented from DecayTreeFitter::ParticleBase.

Reimplemented in DecayTreeFitter::RecoResonance.

Definition at line 120 of file RecoComposite.cxx.

References DecayTreeFitter::Constraint::composite, DecayTreeFitter::Constraint::geometric, and DecayTreeFitter::ParticleBase::projectConstraint().

{
  ErrCode aStatus ;
  switch(aType) {
    case Constraint::composite:
      aStatus |= projectRecoComposite(fitparams,p) ;
      break ;
    case Constraint::geometric:
      aStatus |= projectGeoConstraint(fitparams,p) ;
      break ;
    default:
      aStatus |= ParticleBase::projectConstraint(aType,fitparams,p) ;
  }
  return aStatus ;
}

ErrCode DecayTreeFitter::ParticleBase::projectGeoConstraint ( const FitParams *  fitparams, Projection &  p  ) const

virtualinherited

Definition at line 387 of file ParticleBase.cxx.

References DecayTreeFitter::Projection::H(), mom, p2, DecayTreeFitter::FitParams::par(), RhoCalculationTools::PrintMatrix(), pz, DecayTreeFitter::Projection::r(), row, sqrt(), DecayTreeFitter::ErrCode::success, DecayTreeFitter::Projection::V(), and vtxverbose.

{
  // implements the constraint

  //  vec{x}_decay = vec{x}_production + decaylength * vec{p} / p
  int posindexmother = mother()->posIndex() ;
  int posindex = posIndex();
  int lenindex = lenIndex() ;
  int momindex = momIndex() ;
  assert(posindexmother>=0 && posindex>=0 && lenindex>=0 && momindex>=0) ;


  // size of momentum
  double px = fitparams->par()(momindex+0) ;
  double py = fitparams->par()(momindex+1) ;
  double pz = fitparams->par()(momindex+2) ;
  double p2 = px*px+py*py+pz*pz ;
  double mom  = std::sqrt(p2) ;
  
  double dvx = fitparams->par()(posindexmother+0) - fitparams->par()(posindex+0);
  double dvy = fitparams->par()(posindexmother+1) - fitparams->par()(posindex+1);
  double dvz = fitparams->par()(posindexmother+2) - fitparams->par()(posindex+2);
  double len=dvx*dvx+dvy*dvy+dvz*dvz;
  len=std::sqrt(len);
  
  // lineair approximation is fine for now
  for(int row=0; row<3; row++) {
    double posxmother = fitparams->par()(posindexmother+row) ;
    double posx       = fitparams->par()(posindex+row) ;
    double momx       = fitparams->par()(momindex+row) ;
    p.r(row) = posxmother - posx + len*momx/mom ;
    p.H(row,posindexmother+row) = 1 ;
    p.H(row,posindex+row)       = -1 ;
    p.H(row,lenindex)         = momx/mom ;
  }
  // still need these as well
  p.H(0,momindex+0)  = len/mom*( 1 - px*px/p2 ) ;
  p.H(0,momindex+1)  = len/mom*( 0 - px*py/p2 ) ;
  p.H(0,momindex+2)  = len/mom*( 0 - px*pz/p2 ) ;
  p.H(1,momindex+0)  = len/mom*( 0 - py*px/p2 ) ;
  p.H(1,momindex+1)  = len/mom*( 1 - py*py/p2 ) ;
  p.H(1,momindex+2)  = len/mom*( 0 - py*pz/p2 ) ;
  p.H(2,momindex+0)  = len/mom*( 0 - pz*px/p2 ) ;
  p.H(2,momindex+1)  = len/mom*( 0 - pz*py/p2 ) ;
  p.H(2,momindex+2)  = len/mom*( 1 - pz*pz/p2 ) ;
  //if( charge()!=0 ) 
  //{
    //double lambda = bFieldOverC() * charge() ;
    //double tau = fitparams->par(lenIndex());
    //double pt = sqrt(px*px+py*py) ;
    //const double posprecision = 1e-4 ; // 1mu
    //if( fabs(pt*lambda*tau*tau) > posprecision ) 
    //{
      //double sinlt = sin(lambda*tau) ;
      //double coslt = cos(lambda*tau) ;
      //double px1 = px*coslt - py*sinlt ;
      //double py1 = py*coslt + px*sinlt ;
     
      //p.r(0) += -tau*px + (py1-py)/lambda ;
      //p.r(1) += -tau*py - (px1-px)/lambda ;
     
      //p.H(0,lenindex+0) += -px + px1 ;
      //p.H(0,momindex+0) += -tau + sinlt/lambda ;
      //p.H(0,momindex+1) +=        (coslt-1)/lambda ;
      //p.H(1,lenindex+0) += -py + py1 ;
      //p.H(1,momindex+0) +=      - (coslt-1)/lambda ;
      //p.H(1,momindex+1) += -tau + sinlt/lambda ;

      //std::cout << "Using helix for position of particle: " << name().c_str() << " "
           //<< lambda << " " << lambda*tau
           //<< "  delta-x,y: " << -tau*px + (py1-py)/lambda << "  "
           //<< -tau*py - (px1-px)/lambda << std::endl ;
    //}
  //}

  //p.setParticle( *mother() ) ;  // adds geoconstraint chi2 to the balance of the mother particle. Why?
  if(vtxverbose>6){
    std::cout<<"ParticleBase::projectConstraint(): projection is:"<<std::endl;
    std::cout<<"r "; p.r().Print();
    std::cout<<"V "; p.V().Print();
    std::cout<<"H "; RhoCalculationTools::PrintMatrix(p.H());
    }
  return ErrCode::success ;
}

ErrCode DecayTreeFitter::ParticleBase::projectMassConstraint ( const FitParams *  fitparams, Projection &  p  ) const

virtualinherited

Definition at line 477 of file ParticleBase.cxx.

References DecayTreeFitter::Projection::H(), DecayTreeFitter::FitParams::par(), pz, DecayTreeFitter::Projection::r(), DecayTreeFitter::ErrCode::success, DecayTreeFitter::Projection::Vfast(), and vtxverbose.

{
  if(vtxverbose>6){std::cout<<"ParticleBase::projectMassConstraint():"<<std::endl;}
  double mass  = pdtMass() ;
  double mass2 = mass*mass ;
  int momindex = momIndex() ;

  // initialize the value
  double px = fitparams->par()(momindex+0) ;
  double py = fitparams->par()(momindex+1) ;
  double pz = fitparams->par()(momindex+2) ;
  double E  = fitparams->par()(momindex+3) ;
  if(vtxverbose>6){std::cout<<"px="<<px<<", py="<<py<<", pz="<<pz<<", E="<<E<<", pdtmass="<<mass<<std::endl;}
  p.r(0) = E*E-px*px-py*py-pz*pz-mass2 ;

  // calculate the projection matrix
  p.H(0,momindex+0) = -2.0*px ;
  p.H(0,momindex+1) = -2.0*py ;
  p.H(0,momindex+2) = -2.0*pz ;
  p.H(0,momindex+3) =  2.0*E ;

  // set the variance in the residual
  double width = pdtWidth() ;
  p.Vfast(0,0) = 4*mass*mass*width*width ;

  return ErrCode::success ;
}

ErrCode DecayTreeFitter::RecoComposite::projectRecoComposite	(	const FitParams *	fitparams,
		Projection &	p
	)		const

Definition at line 92 of file RecoComposite.cxx.

References col, DecayTreeFitter::Projection::H(), i, DecayTreeFitter::FitParams::par(), RhoCalculationTools::PrintMatrix(), DecayTreeFitter::Projection::r(), row, DecayTreeFitter::ErrCode::success, DecayTreeFitter::Projection::V(), DecayTreeFitter::Projection::Vfast(), and vtxverbose.

{
  int posindex = posIndex() ;
  int momindex = momIndex() ;

  // quick map for parameters
  int indmap[7]  ;
  for(int i=0; i<3; i++) indmap[i]   = posindex+i ;
  for(int i=0; i<4; i++) indmap[i+3] = momindex+i ;
  for(int row=0; row<dimM(); row++) {
    //p.r(row) = fitparams->par()(indmap[row]) - m_m(row) ;
    p.r(row) = m_m(row) - fitparams->par()(indmap[row]) ; //RK turned sign. It's "measurement - step before".
    p.H(row,indmap[row]) = 1 ;
    for(int col=0; col<dimM()/*=row*/; col++)
      p.Vfast(row,col) = m_matrixV(row,col) ;
  }
  if(vtxverbose>6){
    std::cout<<"RecoComposite::projectRecoComposite(): projection is:"<<posindex<<std::endl;
    std::cout<<"r "; p.r().Print();
    std::cout<<"V "; p.V().Print();
    std::cout<<"H "; RhoCalculationTools::PrintMatrix(p.H());
    }

  return ErrCode::success ;
}

void DecayTreeFitter::ParticleBase::removeDaughter ( const ParticleBase *  pb )

inherited

Definition at line 100 of file ParticleBase.cxx.

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

{
  daucontainer::iterator it = std::find(m_daughters.begin(),m_daughters.end(),pb) ;
  if(it != m_daughters.end() ) {
    ParticleBase* _dp = *it ;
    m_daughters.erase(it) ;
    if(_dp) delete _dp ;
  } else {
    std::cout << "ERROR: cannot remove particle, because not found ..." << std::endl ;
  }
}

void DecayTreeFitter::ParticleBase::retrieveIndexMap ( indexmap &  anindexmap ) const

virtualinherited

Definition at line 377 of file ParticleBase.cxx.

{

  anindexmap.push_back(std::pair<const ParticleBase*,int>(this,index())) ;
  for(daucontainer::const_iterator it = m_daughters.begin() ;
      it != m_daughters.end() ; ++it)
    (*it)->retrieveIndexMap(anindexmap) ;
}

void DecayTreeFitter::ParticleBase::setIndex ( int  i )

inlineprotectedinherited

Definition at line 141 of file ParticleBase.h.

References i, and DecayTreeFitter::ParticleBase::m_index.

{ m_index = i ; }

bool DecayTreeFitter::ParticleBase::setMassConstraint ( bool  add )

inlineinherited

Definition at line 118 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_hasMassConstraint.

Referenced by DecayTreeFitter::DecayChain::setMassConstraint().

                                     {
      std::swap(add,m_hasMassConstraint) ;
      return add != m_hasMassConstraint ;
    }

void DecayTreeFitter::ParticleBase::setMassConstraint ( double  mass )

inlineinherited

Definition at line 123 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_hasMassConstraint, and DecayTreeFitter::ParticleBase::m_pdtMass.

                                          {
      m_hasMassConstraint = true ;
      m_pdtMass = mass ;
    }

void DecayTreeFitter::ParticleBase::setMother ( const ParticleBase *  m )

inlineinherited

Definition at line 108 of file ParticleBase.h.

References m, and DecayTreeFitter::ParticleBase::m_mother.

{ m_mother = m ; }

void DecayTreeFitter::ParticleBase::setName ( const std::string &  n )

inlineprotectedinherited

Definition at line 142 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_name, and n.

{ m_name = n ; }

void DecayTreeFitter::ParticleBase::setParticle ( RhoCandidate *  bc )

inlineinherited

Definition at line 113 of file ParticleBase.h.

References DecayTreeFitter::ParticleBase::m_particle.

{ m_particle = bc ; }

virtual int DecayTreeFitter::RecoComposite::type ( ) const

inlinevirtual

Implements DecayTreeFitter::ParticleBase.

Reimplemented in DecayTreeFitter::RecoResonance.

Definition at line 35 of file RecoComposite.h.

References DecayTreeFitter::ParticleBase::kRecoComposite.

{ return kRecoComposite ; }

void DecayTreeFitter::ParticleBase::updateIndex ( int &  offset )

virtualinherited

Definition at line 83 of file ParticleBase.cxx.

References offset(), and vtxverbose.

Referenced by DecayTreeFitter::Fitter::add(), and DecayTreeFitter::DecayChain::DecayChain().

{
  if(vtxverbose>=5){std::cout<<"ParticleBase::updateIndex() start"<<std::endl;}
  // first the daughters
  for(const_iterator it = begin(); it!= end(); ++it) (*it)->updateIndex(offset) ;
  // now the real work
  m_index = offset ;
  offset += dim() ;
  if(vtxverbose>=5){std::cout<<"ParticleBase::updateIndex() end"<<std::endl;}
}

void DecayTreeFitter::RecoComposite::updCache ( )

virtual

Definition at line 41 of file RecoComposite.cxx.

References convertToHepSymMatrix(), mom, name, particle, pos, and vtxverbose.

Referenced by RecoComposite().

{
  // cache par7 (x,y,z,px,py,pz,E) cov7
  TVector3 pos = particle()->Pos() ;
  TLorentzVector mom = particle()->P4() ;
  m_m = TVectorD(dimM()) ;
  m_m(0) = pos.x() ;
  m_m(1) = pos.y() ;
  m_m(2) = pos.z() ;
  m_m(3) = mom.Px() ;
  m_m(4) = mom.Py() ;
  m_m(5) = mom.Pz() ;
  if(hasEnergy()) m_m(6) = mom.E() ;
  TMatrixDSym cov7 = convertToHepSymMatrix( particle()->Cov7()) ;//FIXME The conversion shall be checked
  cov7.GetSub(0,dimM(),m_matrixV) ; // so either 7 or 6, depending on mass constraint
  if(vtxverbose>=4) {
    std::cout << "cov matrix of external candidate: " << name().c_str()
    << " " << dimM() << std::endl;
    m_matrixV.Print();
  }
}

Member Data Documentation

bool DecayTreeFitter::RecoComposite::m_hasEnergy

protected

Definition at line 56 of file RecoComposite.h.

Referenced by dim(), dimM(), and hasEnergy().

TVectorD DecayTreeFitter::RecoComposite::m_m

protected

Definition at line 54 of file RecoComposite.h.

TMatrixDSym DecayTreeFitter::RecoComposite::m_matrixV

protected

Definition at line 55 of file RecoComposite.h.

---

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

• RecoComposite.h
• RecoComposite.cxx