DecayTreeFitter::FitParams Class Reference

#include <FitParams.h>

Public Member Functions
	FitParams (int dim)
	FitParams (const FitParams &par)
virtual	~FitParams ()
TMatrixDSym &	cov ()
TVectorD &	par ()
double &	par (int row)
TMatrixDSym	cov (const std::vector< int > &indexVec) const
TVectorD	par (const std::vector< int > &indexVec) const
const TMatrixDSym &	cov () const
const TVectorD &	par () const
const double &	par (int row) const
int &	nConstraintsVec (int row)
int	dim () const
double	chiSquare () const
int	nConstraints () const
int	nDof () const
double	err (int row) const
void	resize (int newdim)
void	reset (int newdim)
void	resetPar ()
void	resetCov (double scale=100)
void	print () const
bool	testCov () const
void	addChiSquare (double chi2, int nconstraints, const ParticleBase *p)
ChiSquare	chiSquare (const ParticleBase &p) const

Protected Member Functions
	FitParams ()

Private Attributes
int	m_dim
TVectorD	m_par
TMatrixDSym	m_cov
double	m_chiSquare
int	m_nConstraints
std::vector< int >	m_nConstraintsVec
std::map< const ParticleBase *, ChiSquare >	m_chiSquareMap

Detailed Description

Definition at line 24 of file FitParams.h.

Constructor & Destructor Documentation

DecayTreeFitter::FitParams::FitParams

(

int

dim

)

Definition at line 15 of file FitParams.cxx.

: m_dim(adim),m_par(adim),m_cov(adim),//m_scale(adim),
m_chiSquare(0),m_nConstraints(0),m_nConstraintsVec(adim,0)
{
  //for(int i=0;i<adim;i++) m_scale[i][i]=1.;
}

DecayTreeFitter::FitParams::FitParams

(

const FitParams &

par

)

Definition at line 22 of file FitParams.cxx.

References cov(), m_cov, m_par, and par().

: m_dim(fpar.dim()),m_par(fpar.dim()),m_cov(fpar.dim()),
m_chiSquare(0),m_nConstraints(0),m_nConstraintsVec(fpar.dim(),0)
{
    m_par = fpar.par();
    m_cov = fpar.cov();
    //FIXME don't copy constraintsvector etc? Those are cleared anyways. I guess
}

DecayTreeFitter::FitParams::~FitParams ( )

virtual

Definition at line 31 of file FitParams.cxx.

                                     {    
  //std::cout<<"FitParams::~FitParams A ("<<&m_nConstraintsVec<<"); n="<<m_nConstraintsVec.size()<<"   c="<<m_nConstraintsVec.capacity() <<std::endl;
  //m_nConstraintsVec.clear();
  //std::cout<<"FitParams::~FitParams B; n="<<m_nConstraintsVec.size()<<"   c="<<m_nConstraintsVec.capacity() <<std::endl;
  //m_chiSquareMap.clear() ;
}

DecayTreeFitter::FitParams::FitParams ( )

inlineprotected

Definition at line 67 of file FitParams.h.

{};

Member Function Documentation

void DecayTreeFitter::FitParams::addChiSquare	(	double	chi2,
		int	nconstraints,
		const ParticleBase *	p
	)

Definition at line 141 of file FitParams.cxx.

References p.

Referenced by DecayTreeFitter::Constraint::filter().

{
  m_chiSquare += chisq;
  m_nConstraints += nconstraints ;
  if( p ) m_chiSquareMap[ p ] += ChiSquare(chisq,nconstraints) ;
}

double DecayTreeFitter::FitParams::chiSquare ( ) const

inline

Definition at line 51 of file FitParams.h.

References m_chiSquare.

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

{ return m_chiSquare ; }

ChiSquare DecayTreeFitter::FitParams::chiSquare

(

const ParticleBase &

p

)

const

Definition at line 148 of file FitParams.cxx.

{
  std::map<const ParticleBase*, ChiSquare>::const_iterator it = m_chiSquareMap.find( &p ) ;
  return it != m_chiSquareMap.end() ? it->second : ChiSquare() ;
}

TMatrixDSym& DecayTreeFitter::FitParams::cov ( )

inline

Definition at line 34 of file FitParams.h.

References m_cov.

Referenced by FitParams(), DecayTreeFitter::KalmanCalculator::init(), DecayTreeFitter::RecoPhoton::initCov(), DecayTreeFitter::InteractionPoint::initCov(), DecayTreeFitter::RecoTrack::initCov(), DecayTreeFitter::ParticleBase::initCov(), DecayTreeFitter::ParticleBase::print(), and DecayTreeFitter::KalmanCalculator::updateCov().

{ return m_cov ; }

TMatrixDSym DecayTreeFitter::FitParams::cov

(

const std::vector< int > &

indexVec

)

const

Definition at line 82 of file FitParams.cxx.

References col, and row.

                                                                              {
  int nrow = indexVec.size() ;
  TMatrixDSym thecov(nrow) ;
  for(int row=0; row<nrow; row++)
    //for(int col=0; col<=row ; col++)//fix for TmatrixDSym (which is not really symmetric)
    for(int col=0; col<nrow ; col++)
      thecov(row,col) = m_cov(indexVec[row],indexVec[col]) ;
  return thecov ;
}

const TMatrixDSym& DecayTreeFitter::FitParams::cov ( ) const

inline

Definition at line 41 of file FitParams.h.

References m_cov.

{ return m_cov ; }

int DecayTreeFitter::FitParams::dim ( ) const

inline

Definition at line 50 of file FitParams.h.

References m_dim.

Referenced by DecayTreeFitter::RecoParticle::chiSquare(), DecayTreeFitter::InteractionPoint::chiSquare(), DecayTreeFitter::RecoComposite::chiSquare(), DecayTreeFitter::Constraint::filter(), DecayTreeFitter::KalmanCalculator::init(), DecayTreeFitter::DecayChain::init(), and nDof().

{ return m_dim ; }

double DecayTreeFitter::FitParams::err ( int  row ) const

inline

Definition at line 55 of file FitParams.h.

References m_cov, and sqrt().

{ return sqrt(m_cov(row,row)) ; }

int DecayTreeFitter::FitParams::nConstraints ( ) const

inline

Definition at line 53 of file FitParams.h.

References m_nConstraints.

Referenced by nDof().

{ return m_nConstraints ; }

int& DecayTreeFitter::FitParams::nConstraintsVec ( int  row )

inline

Definition at line 47 of file FitParams.h.

References m_nConstraintsVec.

Referenced by DecayTreeFitter::KalmanCalculator::updateCov().

{ return m_nConstraintsVec[row/*-1*/] ; }

int DecayTreeFitter::FitParams::nDof ( ) const

inline

Definition at line 54 of file FitParams.h.

References dim(), and nConstraints().

{ return nConstraints() - dim() ; }

TVectorD& DecayTreeFitter::FitParams::par ( )

inline

Definition at line 35 of file FitParams.h.

References m_par.

Referenced by DecayTreeFitter::Constraint::filter(), FitParams(), DecayTreeFitter::KalmanCalculator::init(), DecayTreeFitter::InternalParticle::initMom(), DecayTreeFitter::MissingParticle::initPar1(), DecayTreeFitter::InternalParticle::initPar1(), DecayTreeFitter::RecoResonance::initPar1(), DecayTreeFitter::RecoComposite::initPar1(), DecayTreeFitter::InteractionPoint::initPar1(), DecayTreeFitter::RecoPhoton::initPar2(), DecayTreeFitter::InternalParticle::initPar2(), DecayTreeFitter::RecoTrack::initPar2(), DecayTreeFitter::RecoPhoton::initParPhoton(), DecayTreeFitter::ParticleBase::initTau(), DecayTreeFitter::ParticleBase::print(), DecayTreeFitter::InteractionPoint::projectBeamConstraint(), DecayTreeFitter::InternalParticle::projectConversionConstraint(), DecayTreeFitter::ParticleBase::projectGeoConstraint(), DecayTreeFitter::InteractionPoint::projectIPConstraint(), DecayTreeFitter::InternalParticle::projectKineConstraint(), DecayTreeFitter::ParticleBase::projectMassConstraint(), DecayTreeFitter::InternalParticle::projectMassConstraintTwoBody(), DecayTreeFitter::RecoComposite::projectRecoComposite(), DecayTreeFitter::RecoPhoton::projectRecoConstraint(), DecayTreeFitter::RecoTrack::projectRecoConstraint(), DecayTreeFitter::KalmanCalculator::updatePar(), and DecayTreeFitter::RecoTrack::updCache().

{ return m_par ; }

double& DecayTreeFitter::FitParams::par ( int  row )

inline

Definition at line 36 of file FitParams.h.

References m_par.

{ return m_par(row) ; }

TVectorD DecayTreeFitter::FitParams::par

(

const std::vector< int > &

indexVec

)

const

Definition at line 92 of file FitParams.cxx.

References row.

                                                                           {
  int nrow = indexVec.size() ;
  TVectorD thepar(nrow) ;
  for(int row=0; row<nrow; row++)
    thepar(row) = m_par(indexVec[row]) ;
  return thepar ;
}

const TVectorD& DecayTreeFitter::FitParams::par ( ) const

inline

Definition at line 42 of file FitParams.h.

References m_par.

{ return m_par ; }

const double& DecayTreeFitter::FitParams::par ( int  row ) const

inline

Definition at line 43 of file FitParams.h.

References m_par.

{ return m_par(row) ; }

void DecayTreeFitter::FitParams::print

(

)

const

Definition at line 72 of file FitParams.cxx.

References row, and sqrt().

                                           {
  std::cout << std::setw(3) << "index" << std::setw(15) << "val" << std::setw(15) << "err" << std::endl ;
  std::cout << std::setprecision(5) ;
  for(int row=0; row<m_dim; row++)
    std::cout << std::setw(3) << row
    << std::setw(15) << m_par(row)
    << std::setw(15) << sqrt(m_cov(row,row)) << std::endl ;
  std::cout << std::setprecision(10) ;
} ;

void DecayTreeFitter::FitParams::reset

(

int

newdim

)

Definition at line 100 of file FitParams.cxx.

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

{
  resetPar();
  resetCov();
  if( newdim != m_dim )
  {
    m_par.ResizeTo(newdim);
    m_cov.ResizeTo(newdim,newdim);
    m_nConstraintsVec.resize(newdim,0) ;
  }
}

void DecayTreeFitter::FitParams::resetCov

(

double

scale = 100

)

Definition at line 43 of file FitParams.cxx.

References col, and row.

Referenced by DecayTreeFitter::DecayChain::filter(), and DecayTreeFitter::DecayChain::init().

                                                     {

  if(aScale > 0)
  {
    for(int row=0; row<m_dim; row++) {
      for(int col=0; col<row; col++) {
        m_cov(row,col) = 0 ;
        m_cov(col,row) = 0 ;
      }
      m_cov(row,row) *= aScale ;
      if(m_cov(row,row) < 0 ) m_cov(row,row)*=-1 ;
      if(m_cov(row,row) == 0 ) m_cov(row,row)=aScale ;
    }
  }
  m_chiSquare=0 ;
  m_nConstraints=0 ;
  for(int row=0; row<m_dim; row++) {
    nConstraintsVec(row) = 0 ;
  }
  m_chiSquareMap.clear() ;
}

void DecayTreeFitter::FitParams::resetPar

(

)

Definition at line 38 of file FitParams.cxx.

References row.

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

                                        {
  for(int row=0; row<m_dim; row++)
    m_par(row) = 0 ;
}

void DecayTreeFitter::FitParams::resize

(

int

newdim

)

Definition at line 112 of file FitParams.cxx.

{
  if( newdim > m_dim ) {
    m_dim = newdim ;
    // very expensive, but okay ...
    //      TVectorD newpar(newdim) ;
    //      newpar.sub(1,m_par);
    //
    //      TMatrixDSym newcov(newdim) ;
    //      newcov.sub(1,m_cov) ;
    //
    //      //      HepVector newpar(newdim,0) ;
    //      //       HepSymMatrix newcov(newdim,0) ;
    //      //       std::cout << newpar << std::endl ;
    //      //       for(int row=0; row<m_dim ; row++) {
    //      //  newpar(row) = m_par(row) ;
    //      //  for(int col=0; col<=row; col++)
    //      //  //        newcov(row,col) = m_cov(row,col) ;
    //      //       }
    //      //      std::cout << m_par << " " << newpar << std::endl ;
    //
    //      m_par = newpar ;
    //      m_cov = newcov ;
    m_par.ResizeTo(newdim);
    m_cov.ResizeTo(newdim,newdim);
    m_nConstraintsVec.resize(newdim,0) ;
  }
}

bool DecayTreeFitter::FitParams::testCov

(

)

const

Definition at line 65 of file FitParams.cxx.

References row.

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

                                             {
  bool okay=true ;
  for(int row=0; row<m_dim && okay; row++)
    okay = m_cov(row,row)>=0 ;
  return okay ;
}

Member Data Documentation

double DecayTreeFitter::FitParams::m_chiSquare

private

Definition at line 73 of file FitParams.h.

Referenced by chiSquare().

std::map<const ParticleBase*, ChiSquare> DecayTreeFitter::FitParams::m_chiSquareMap

private

Definition at line 76 of file FitParams.h.

TMatrixDSym DecayTreeFitter::FitParams::m_cov

private

Definition at line 71 of file FitParams.h.

Referenced by cov(), err(), and FitParams().

int DecayTreeFitter::FitParams::m_dim

private

Definition at line 67 of file FitParams.h.

Referenced by dim().

int DecayTreeFitter::FitParams::m_nConstraints

private

Definition at line 74 of file FitParams.h.

Referenced by nConstraints().

std::vector<int> DecayTreeFitter::FitParams::m_nConstraintsVec

private

Definition at line 75 of file FitParams.h.

Referenced by nConstraintsVec().

TVectorD DecayTreeFitter::FitParams::m_par

private

Definition at line 70 of file FitParams.h.

Referenced by FitParams(), and par().

---

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

• FitParams.h
• FitParams.cxx