DecayTreeFitter::ParticleBase Class Reference

#include <ParticleBase.h>

Inheritance diagram for DecayTreeFitter::ParticleBase:

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
	ParticleBase (RhoCandidate *bc, const ParticleBase *mother)
	ParticleBase (const std::string &name)
virtual	~ParticleBase ()
virtual int	dim () const =0
virtual void	updateIndex (int &offset)
virtual ErrCode	initPar1 (FitParams *)=0
virtual ErrCode	initPar2 (FitParams *)=0
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
virtual ErrCode	projectConstraint (Constraint::Type, const FitParams *, Projection &) const
virtual int	type () const =0
virtual int	posIndex () const
virtual int	lenIndex () const
virtual int	momIndex () const
virtual bool	hasEnergy () const
virtual bool	hasPosition () 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 void	addToConstraintList (constraintlist &alist, int depth) const =0
virtual int	nFinalChargedCandidates () const
void	setParticle (RhoCandidate *bc)
void	collectVertexDaughters (daucontainer &particles, int posindex)
bool	setMassConstraint (bool add)
void	setMassConstraint (double mass)
ChiSquare	chiSquare (const FitParams *fitparams) const

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

Protected Member Functions
ErrCode	initTau (FitParams *par) const
void	makeName (RhoCandidate *bc)
daucontainer &	daughters ()
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 ()

Private Attributes
RhoCandidate *	m_particle
const ParticleBase *	m_mother
ParticleContainer	m_daughters
const TParticlePDG *	m_prop
int	m_index
double	m_pdtMass
double	m_pdtWidth
double	m_pdtCLifeTime
int	m_charge
std::string	m_name
bool	m_hasMassConstraint

Detailed Description

Definition at line 25 of file ParticleBase.h.

Member Typedef Documentation

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

Definition at line 97 of file ParticleBase.h.

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

Definition at line 110 of file ParticleBase.h.

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

Definition at line 96 of file ParticleBase.h.

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

Definition at line 105 of file ParticleBase.h.

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

Definition at line 33 of file ParticleBase.h.

Member Enumeration Documentation

enum DecayTreeFitter::ParticleBase::ParticleType

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::ParticleBase::ParticleBase	(	RhoCandidate *	bc,
		const ParticleBase *	mother
	)

Definition at line 47 of file ParticleBase.cxx.

References C(), RhoCandidate::Charge(), m_charge, m_name, m_pdtCLifeTime, m_pdtMass, m_pdtWidth, and m_prop.

: m_particle(aParticle),m_mother(aMother),
m_prop(aParticle->PdtEntry()),
m_index(0),m_pdtMass(0),m_pdtWidth(0),m_pdtCLifeTime(0),m_charge(0),
m_name("Unknown"), m_hasMassConstraint(false)
{
  if( m_prop ) {
    m_pdtMass      = m_prop->Mass() ;
    m_pdtWidth     = m_prop->Width() ;
    m_pdtCLifeTime = TMath::C() * m_prop->Lifetime() * 100; // Units: ctau[cm] = C[m/s] * tau[s]
    double fltcharge = m_prop->Charge()/3. ;// TParticlePDG holds charg in |e|/3
    m_charge = fltcharge < 0 ? int(fltcharge-0.5) : int(fltcharge+0.5) ;
    //m_charge = aParticle->Charge()>0 ? 1 : (aParticle->Charge()<0 ? -1 : 0) ;
    m_name = m_prop->GetName() ;//FIXME Does root object name do the job?
  } else {
    m_charge = aParticle->Charge()>0 ? 1 : (aParticle->Charge()<0 ? -1 : 0) ;
  }
}

DecayTreeFitter::ParticleBase::ParticleBase

(

const std::string &

name

)

Definition at line 67 of file ParticleBase.cxx.

: m_particle(0),m_mother(0),
m_prop(0),
m_index(0),m_pdtMass(0),m_pdtWidth(0),m_pdtCLifeTime(0),m_charge(0),
m_name(aname), m_hasMassConstraint(false)
{
}

DecayTreeFitter::ParticleBase::~ParticleBase ( )

virtual

Definition at line 75 of file ParticleBase.cxx.

{
  for(daucontainer::iterator it = m_daughters.begin() ;
      it != m_daughters.end() ; ++it)
    if(*it) delete *it ;
  m_daughters.clear() ;
}

Member Function Documentation

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

Definition at line 94 of file ParticleBase.cxx.

References createParticle().

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

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

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

pure virtual

Implemented in DecayTreeFitter::RecoTrack, DecayTreeFitter::InteractionPoint, DecayTreeFitter::RecoComposite, DecayTreeFitter::InternalParticle, DecayTreeFitter::RecoResonance, DecayTreeFitter::RecoPhoton, and DecayTreeFitter::MissingParticle.

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

const_iterator DecayTreeFitter::ParticleBase::begin ( ) const

inline

Definition at line 100 of file ParticleBase.h.

References m_daughters.

{ return m_daughters.begin() ; }

double DecayTreeFitter::ParticleBase::bFieldOverC ( )

staticprotected

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

inline

Definition at line 90 of file ParticleBase.h.

References m_charge.

{ return m_charge ; }

ChiSquare DecayTreeFitter::ParticleBase::chiSquare

(

const FitParams *

fitparams

)

const

Definition at line 568 of file ParticleBase.cxx.

References DecayTreeFitter::FitParams::chiSquare().

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

{
  ChiSquare chi2 ;
  // add contribution from daughters
  for(daucontainer::const_iterator it = m_daughters.begin() ;
      it != m_daughters.end() ; ++it) {
    chi2 += (*it)->chiSquare(fitparams) ;
  }
  // add own chisquare, adjust for number of parameters
  chi2 += fitparams->chiSquare( *this ) ;
  chi2 += ChiSquare( 0, -dim() ) ;
  return chi2 ;
}

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

virtual

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
	)

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  )

static

Definition at line 113 of file ParticleBase.cxx.

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

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

inline

Definition at line 99 of file ParticleBase.h.

References m_daughters.

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

{ return m_daughters ; }

daucontainer& DecayTreeFitter::ParticleBase::daughters ( )

inlineprotected

Definition at line 138 of file ParticleBase.h.

References m_daughters.

{ return m_daughters ; }

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

pure virtual

Implemented in DecayTreeFitter::InteractionPoint, DecayTreeFitter::MissingParticle, DecayTreeFitter::RecoParticle, DecayTreeFitter::RecoComposite, DecayTreeFitter::Resonance, DecayTreeFitter::InternalParticle, and DecayTreeFitter::RecoResonance.

const_iterator DecayTreeFitter::ParticleBase::end ( ) const

inline

Definition at line 101 of file ParticleBase.h.

References m_daughters.

Referenced by locate().

{ return m_daughters.end() ; }

int DecayTreeFitter::ParticleBase::eneIndex ( ) const

inline

Definition at line 80 of file ParticleBase.h.

References hasEnergy(), and momIndex().

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

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

inlinevirtual

Reimplemented in DecayTreeFitter::RecoComposite, DecayTreeFitter::InternalParticle, DecayTreeFitter::RecoParticle, and DecayTreeFitter::MissingParticle.

Definition at line 74 of file ParticleBase.h.

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

{ return false ; }

bool DecayTreeFitter::ParticleBase::hasMassConstraint ( ) const

inlineprotected

Definition at line 139 of file ParticleBase.h.

References m_hasMassConstraint.

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

{ return m_hasMassConstraint ; }

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

inlinevirtual

Reimplemented in DecayTreeFitter::RecoComposite, DecayTreeFitter::InternalParticle, and DecayTreeFitter::Resonance.

Definition at line 78 of file ParticleBase.h.

{ return false ; }

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

inlinevirtual

Definition at line 59 of file ParticleBase.h.

References m_index.

Referenced by createParticle(), DecayTreeFitter::DecayChain::index(), DecayTreeFitter::InternalParticle::lenIndex(), DecayTreeFitter::RecoComposite::lenIndex(), DecayTreeFitter::MissingParticle::momIndex(), DecayTreeFitter::RecoParticle::momIndex(), DecayTreeFitter::RecoResonance::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

virtual

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

virtual ErrCode DecayTreeFitter::ParticleBase::initPar1 ( FitParams *  )

pure virtual

Implemented in DecayTreeFitter::InteractionPoint, DecayTreeFitter::RecoComposite, DecayTreeFitter::Resonance, DecayTreeFitter::InternalParticle, DecayTreeFitter::RecoPhoton, DecayTreeFitter::RecoResonance, DecayTreeFitter::RecoParticle, and DecayTreeFitter::MissingParticle.

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

virtual ErrCode DecayTreeFitter::ParticleBase::initPar2 ( FitParams *  )

pure virtual

Implemented in DecayTreeFitter::RecoTrack, DecayTreeFitter::RecoComposite, DecayTreeFitter::Resonance, DecayTreeFitter::InternalParticle, DecayTreeFitter::RecoPhoton, DecayTreeFitter::RecoResonance, and DecayTreeFitter::MissingParticle.

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

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

protected

Definition at line 522 of file ParticleBase.cxx.

References mom, DecayTreeFitter::FitParams::par(), 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 )

staticprotected

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

inlinevirtual

Reimplemented in DecayTreeFitter::RecoComposite, DecayTreeFitter::Resonance, DecayTreeFitter::InternalParticle, and DecayTreeFitter::RecoResonance.

Definition at line 70 of file ParticleBase.h.

Referenced by DecayTreeFitter::Fitter::decayLengthSum(), DecayTreeFitter::DecayChain::lenIndex(), and DecayTreeFitter::Fitter::updateCand().

{ return -1 ; }

const ParticleBase * DecayTreeFitter::ParticleBase::locate

(

RhoCandidate *

bc

)

const

Definition at line 353 of file ParticleBase.cxx.

References end(), and locate().

Referenced by DecayTreeFitter::Fitter::decayLengthSum(), DecayTreeFitter::DecayChain::locate(), 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 )

protected

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

inlinevirtual

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

Definition at line 71 of file ParticleBase.h.

Referenced by eneIndex(), DecayTreeFitter::DecayChain::momIndex(), DecayTreeFitter::InternalParticle::projectConversionConstraint(), DecayTreeFitter::InternalParticle::projectMassConstraintTwoBody(), and DecayTreeFitter::Fitter::updateCand().

{ return -1 ; }

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

inline

Definition at line 60 of file ParticleBase.h.

References 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

inline

Definition at line 61 of file ParticleBase.h.

References m_name.

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

{ return m_name ; }

int DecayTreeFitter::ParticleBase::nFinalChargedCandidates ( ) const

virtual

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

virtual

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

inline

Definition at line 57 of file ParticleBase.h.

References m_particle.

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

{ return m_particle ; }

double DecayTreeFitter::ParticleBase::pdtCLifeTime ( ) const

inline

Definition at line 88 of file ParticleBase.h.

References m_pdtCLifeTime.

{ return m_pdtCLifeTime ; }

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

staticprotected

double DecayTreeFitter::ParticleBase::pdtMass ( ) const

inline

Definition at line 86 of file ParticleBase.h.

References m_pdtMass.

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

{ return m_pdtMass ; }

double DecayTreeFitter::ParticleBase::pdtTau ( ) const

inline

Definition at line 89 of file ParticleBase.h.

References m_pdtCLifeTime, and m_pdtMass.

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

double DecayTreeFitter::ParticleBase::pdtWidth ( ) const

inline

Definition at line 87 of file ParticleBase.h.

References m_pdtWidth.

{ return m_pdtWidth ; }

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

inlinevirtual

Reimplemented in DecayTreeFitter::RecoComposite, DecayTreeFitter::Resonance, DecayTreeFitter::InternalParticle, and DecayTreeFitter::RecoResonance.

Definition at line 69 of file ParticleBase.h.

Referenced by DecayTreeFitter::DecayChain::chiSquare(), initTau(), DecayTreeFitter::RecoResonance::posIndex(), DecayTreeFitter::Resonance::posIndex(), DecayTreeFitter::DecayChain::posIndex(), and DecayTreeFitter::Fitter::updateCand().

{ return -1 ; }

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

virtual

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::ParticleBase::projectConstraint ( Constraint::Type  atype, const FitParams *  , Projection &    ) const

virtual

Reimplemented in DecayTreeFitter::InteractionPoint, DecayTreeFitter::InternalParticle, DecayTreeFitter::RecoParticle, DecayTreeFitter::RecoComposite, and DecayTreeFitter::RecoResonance.

Definition at line 507 of file ParticleBase.cxx.

References DecayTreeFitter::ErrCode::badsetup, and name.

Referenced by DecayTreeFitter::RecoResonance::projectConstraint(), DecayTreeFitter::RecoComposite::projectConstraint(), DecayTreeFitter::RecoParticle::projectConstraint(), and DecayTreeFitter::InternalParticle::projectConstraint().

{
  std::cout << "ParticleBase::projectConstraint(): no method to project this constaint: "
  << name() << " " << type() << " " << atype << std::endl ;
  return ErrCode::badsetup ;
}

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

virtual

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

virtual

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

void DecayTreeFitter::ParticleBase::removeDaughter

(

const ParticleBase *

pb

)

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

virtual

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 )

inlineprotected

Definition at line 141 of file ParticleBase.h.

References i, and m_index.

{ m_index = i ; }

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

inline

Definition at line 118 of file ParticleBase.h.

References m_hasMassConstraint.

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

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

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

inline

Definition at line 123 of file ParticleBase.h.

References m_hasMassConstraint, and m_pdtMass.

                                          {
      m_hasMassConstraint = true ;
      m_pdtMass = mass ;
    }

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

inline

Definition at line 108 of file ParticleBase.h.

References m, and m_mother.

{ m_mother = m ; }

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

inlineprotected

Definition at line 142 of file ParticleBase.h.

References m_name, and n.

{ m_name = n ; }

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

inline

Definition at line 113 of file ParticleBase.h.

References m_particle.

{ m_particle = bc ; }

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

pure virtual

Implemented in DecayTreeFitter::InteractionPoint, DecayTreeFitter::RecoTrack, DecayTreeFitter::RecoComposite, DecayTreeFitter::MissingParticle, DecayTreeFitter::RecoPhoton, DecayTreeFitter::InternalParticle, DecayTreeFitter::RecoResonance, and DecayTreeFitter::Resonance.

Referenced by createParticle(), and DecayTreeFitter::sortByType().

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

virtual

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

Member Data Documentation

int DecayTreeFitter::ParticleBase::m_charge

private

Definition at line 152 of file ParticleBase.h.

Referenced by charge(), and ParticleBase().

ParticleContainer DecayTreeFitter::ParticleBase::m_daughters

private

Definition at line 146 of file ParticleBase.h.

Referenced by begin(), daughters(), and end().

bool DecayTreeFitter::ParticleBase::m_hasMassConstraint

private

Definition at line 154 of file ParticleBase.h.

Referenced by hasMassConstraint(), and setMassConstraint().

int DecayTreeFitter::ParticleBase::m_index

private

Definition at line 148 of file ParticleBase.h.

Referenced by index(), and setIndex().

const ParticleBase* DecayTreeFitter::ParticleBase::m_mother

private

Definition at line 145 of file ParticleBase.h.

Referenced by mother(), and setMother().

std::string DecayTreeFitter::ParticleBase::m_name

private

Definition at line 153 of file ParticleBase.h.

Referenced by name(), ParticleBase(), and setName().

RhoCandidate* DecayTreeFitter::ParticleBase::m_particle

private

Definition at line 144 of file ParticleBase.h.

Referenced by particle(), and setParticle().

double DecayTreeFitter::ParticleBase::m_pdtCLifeTime

private

Definition at line 151 of file ParticleBase.h.

Referenced by ParticleBase(), pdtCLifeTime(), and pdtTau().

double DecayTreeFitter::ParticleBase::m_pdtMass

private

Definition at line 149 of file ParticleBase.h.

Referenced by ParticleBase(), pdtMass(), pdtTau(), and setMassConstraint().

double DecayTreeFitter::ParticleBase::m_pdtWidth

private

Definition at line 150 of file ParticleBase.h.

Referenced by ParticleBase(), and pdtWidth().

const TParticlePDG* DecayTreeFitter::ParticleBase::m_prop

private

Definition at line 147 of file ParticleBase.h.

Referenced by ParticleBase().

---

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

• ParticleBase.h
• ParticleBase.cxx