DecayTreeFitter::RecoResonance Class Reference

#include <RecoResonance.h>

Inheritance diagram for DecayTreeFitter::RecoResonance:

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
	RecoResonance (RhoCandidate *bc, const ParticleBase *mother)
virtual	~RecoResonance ()
virtual int	dim () 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	momIndex () const
virtual int	lenIndex () const
virtual std::string	parname (int index) const
virtual void	addToConstraintList (constraintlist &alist, int depth) const
int	dimM () const
ErrCode	projectRecoComposite (const FitParams *, Projection &) const
virtual bool	hasEnergy () const
virtual bool	hasPosition () const
virtual void	updCache ()
virtual double	chiSquare (const FitParams *fitparams) const
virtual void	updateIndex (int &offset)
virtual ErrCode	initCov (FitParams *) 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 17 of file RecoResonance.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::RecoResonance::RecoResonance	(	RhoCandidate *	bc,
		const ParticleBase *	mother
	)

Definition at line 18 of file RecoResonance.cxx.

: RecoComposite(bc,aMother) {}

DecayTreeFitter::RecoResonance::~RecoResonance ( )

virtual

Definition at line 22 of file RecoResonance.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::RecoResonance::addToConstraintList ( constraintlist &  alist, int  depth  ) const

inlinevirtual

Reimplemented from DecayTreeFitter::RecoComposite.

Definition at line 36 of file RecoResonance.h.

References DecayTreeFitter::RecoComposite::dimM(), and DecayTreeFitter::Constraint::resonance.

                                                                             {
      alist.push_back( Constraint(this,Constraint::resonance,depth,dimM()) ) ; }

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

virtualinherited

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::RecoResonance::dim ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::RecoComposite.

Definition at line 23 of file RecoResonance.h.

References DecayTreeFitter::RecoComposite::hasEnergy().

{ return hasEnergy() ? 4 : 3 ; } // (px,py,pz,(E))

int DecayTreeFitter::RecoComposite::dimM ( ) const

inlineinherited

Definition at line 29 of file RecoComposite.h.

References DecayTreeFitter::RecoComposite::m_hasEnergy.

Referenced by addToConstraintList(), and DecayTreeFitter::RecoComposite::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

inlinevirtualinherited

Reimplemented from DecayTreeFitter::ParticleBase.

Definition at line 41 of file RecoComposite.h.

References DecayTreeFitter::RecoComposite::m_hasEnergy.

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

inlinevirtualinherited

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(), DecayTreeFitter::RecoComposite::lenIndex(), DecayTreeFitter::MissingParticle::momIndex(), DecayTreeFitter::RecoParticle::momIndex(), momIndex(), DecayTreeFitter::Resonance::momIndex(), DecayTreeFitter::InternalParticle::momIndex(), DecayTreeFitter::RecoComposite::momIndex(), DecayTreeFitter::InternalParticle::posIndex(), and DecayTreeFitter::RecoComposite::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::RecoResonance::initPar1 ( FitParams *  fitparams )

virtual

Reimplemented from DecayTreeFitter::RecoComposite.

Definition at line 25 of file RecoResonance.cxx.

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

{
  if(vtxverbose>5){std::cout<<"RecoResonance::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' -> this overwrites mother position !
  for(int row=0; row<dimM(); row++)
    fitparams->par()(indmap[row]) = m_m(row) ;
  return ErrCode::success ;
}

ErrCode DecayTreeFitter::RecoResonance::initPar2 ( FitParams *  )

virtual

Reimplemented from DecayTreeFitter::RecoComposite.

Definition at line 43 of file RecoResonance.cxx.

References DecayTreeFitter::ErrCode::success, and vtxverbose.

{
  if(vtxverbose>5){std::cout<<"RecoResonance::initPar2: - "<<std::endl;}
  // nothing to do!
  return ErrCode::success ;
}

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::RecoResonance::lenIndex ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::RecoComposite.

Definition at line 32 of file RecoResonance.h.

{ return -1 ; }

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::RecoResonance::momIndex ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::RecoComposite.

Definition at line 31 of file RecoResonance.h.

References DecayTreeFitter::ParticleBase::index().

{ return index() ; }

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(), 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::RecoResonance::parname ( int  index ) const

virtual

Reimplemented from DecayTreeFitter::ParticleBase.

Definition at line 66 of file RecoResonance.cxx.

References DecayTreeFitter::ParticleBase::parname().

{
  return ParticleBase::parname(ind+4) ;
}

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::RecoResonance::posIndex ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::RecoComposite.

Definition at line 30 of file RecoResonance.h.

References DecayTreeFitter::ParticleBase::mother(), and DecayTreeFitter::ParticleBase::posIndex().

{ return mother()->posIndex()  ; }

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::RecoResonance::projectConstraint ( Constraint::Type  aType, const FitParams *  fitparams, Projection &  p  ) const

virtual

Reimplemented from DecayTreeFitter::RecoComposite.

Definition at line 51 of file RecoResonance.cxx.

References DecayTreeFitter::ParticleBase::projectConstraint(), and DecayTreeFitter::Constraint::resonance.

{
  ErrCode aStatus ;
  switch(aType) {
    case Constraint::resonance:
      aStatus |= projectRecoComposite(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

inherited

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::RecoResonance::type ( ) const

inlinevirtual

Reimplemented from DecayTreeFitter::RecoComposite.

Definition at line 28 of file RecoResonance.h.

References DecayTreeFitter::ParticleBase::kRecoResonance.

{ return kRecoResonance ; }

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

virtualinherited

Definition at line 41 of file RecoComposite.cxx.

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

Referenced by DecayTreeFitter::RecoComposite::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

protectedinherited

Definition at line 56 of file RecoComposite.h.

Referenced by DecayTreeFitter::RecoComposite::dim(), DecayTreeFitter::RecoComposite::dimM(), and DecayTreeFitter::RecoComposite::hasEnergy().

TVectorD DecayTreeFitter::RecoComposite::m_m

protectedinherited

Definition at line 54 of file RecoComposite.h.

TMatrixDSym DecayTreeFitter::RecoComposite::m_matrixV

protectedinherited

Definition at line 55 of file RecoComposite.h.

---

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

• RecoResonance.h
• RecoResonance.cxx