DecayTreeFitter::Constraint Class Reference

#include <Constraint.h>

Inheritance diagram for DecayTreeFitter::Constraint:

Public Types
enum	Type {   unknown =0, beamspot, lifetime, resonance,   composite, track, photon, conversion,   kinematic, massEnergy, geometric, mass,   beamenergy, merged, ntypes }

Public Member Functions
bool	operator< (const Constraint &rhs) const
bool	operator== (const Constraint &rhs) const
Type	type () const
unsigned int	dim () const
bool	isLineair () const
unsigned int	nIter () const
	Constraint ()
	Constraint (const ParticleBase *node, Type atype, int depth, unsigned int adim, int maxniter=1)
virtual	~Constraint ()
virtual ErrCode	project (const FitParams *fitpar, Projection &p) const
virtual ErrCode	filter (FitParams *fitpar) const
virtual ErrCode	filter (FitParams *fitpar, const FitParams *reference) const
virtual void	print (std::ostream &os=std::cout) const
std::string	name () const
void	setWeight (int w)

Protected Member Functions
	Constraint (Constraint::Type atype)
void	setDim (unsigned int d)
void	setNIter (unsigned int d)

Private Attributes
const ParticleBase *	m_node
int	m_depth
Type	m_type
unsigned int	m_dim
int	m_weight
int	m_maxNIter

Detailed Description

Definition at line 22 of file Constraint.h.

Member Enumeration Documentation

enum DecayTreeFitter::Constraint::Type

Enumerator
unknown
beamspot
lifetime
resonance
composite
track
photon
conversion
kinematic
massEnergy
geometric
mass
beamenergy
merged
ntypes

Definition at line 28 of file Constraint.h.

              { unknown=0,
      beamspot,
      lifetime,
      resonance,
      composite,
      track,
      photon,
      conversion,
      kinematic,
      massEnergy,
      geometric,
      mass,
      beamenergy,
      merged,
      ntypes} ;

Constructor & Destructor Documentation

DecayTreeFitter::Constraint::Constraint ( )

inline

Definition at line 55 of file Constraint.h.

: m_node(0),m_depth(0),m_type(unknown),m_dim(0) {}

DecayTreeFitter::Constraint::Constraint ( const ParticleBase *  node, Type  atype, int  depth, unsigned int  adim, int  maxniter = 1  )

inline

Definition at line 57 of file Constraint.h.

    : m_node(node), m_depth(depth), m_type(atype), m_dim(adim),
    m_weight(1), m_maxNIter(maxniter) {}

virtual DecayTreeFitter::Constraint::~Constraint ( )

inlinevirtual

Definition at line 63 of file Constraint.h.

{}

DecayTreeFitter::Constraint::Constraint ( Constraint::Type  atype )

inlineprotected

Definition at line 77 of file Constraint.h.

                                     :
    m_node(0),m_depth(0),m_type(atype),m_dim(0),
    m_weight(0),m_maxNIter(0) {}

Member Function Documentation

unsigned int DecayTreeFitter::Constraint::dim ( ) const

inline

Definition at line 51 of file Constraint.h.

References m_dim.

Referenced by DecayTreeFitter::MergedConstraint::push_back().

{ return m_dim ; }

ErrCode DecayTreeFitter::Constraint::filter ( FitParams *  fitpar ) const

virtual

Definition at line 59 of file Constraint.cxx.

References DecayTreeFitter::FitParams::addChiSquare(), DecayTreeFitter::ErrCode::badsetup, DecayTreeFitter::FitParams::dim(), fabs(), DecayTreeFitter::ErrCode::failure(), kalman, ntypes, p, DecayTreeFitter::FitParams::par(), print(), DecayTreeFitter::ErrCode::Print(), status, and vtxverbose.

{
  ErrCode status ;
  if(m_type<=Constraint::unknown || m_type>=Constraint::ntypes) {
    std::cout << "Constraint::filter(FitParams*): unknown constraint: " << m_type
    << std::endl ;
    status |= ErrCode::badsetup ;
  } else if (m_type!=merged && !m_node) {
    std::cout << "Constraint::filter(FitParams*): constraint without a node"
    << std::endl ;
    status |= ErrCode::badsetup ;
  } else {
    if(vtxverbose>=3) { std::cout << "Constraint::filter(FitParams*): filtering "  ; print() ;}
    // save the unfiltered ('predicted') parameters. we need to
    // store them if we want to iterate constraints.
    const TVectorD* pred(0) ;
    if(m_maxNIter>1) pred = new TVectorD(fitpar->par()) ;

    Projection p(fitpar->dim(),m_dim) ;
    KalmanCalculator kalman ;
    double chisq(0) ;
    int iter(0) ;
    bool finished(false) ;
    if(vtxverbose>=5) { std::cout << "Constraint::filter(FitParams*): starting loop "  ; print() ;}
    while (!finished && !status.failure()) {
      p.reset() ;
      status |= project(fitpar,p) ;
      if(!status.failure()) {
        status |= kalman.init( p.r(), p.H(), fitpar, p.V() ) ;
        //status |= kalman.init( p.r(), p.H(), fitpar, (&p.V()) ? ( p.V().NonZeros()>0 ? &p.V() : 0 ) : 0  ) ;
        if( !status.failure()) {
          if(iter==0 || !pred) {
            kalman.updatePar( fitpar ) ;
          } else {
            kalman.updatePar( *pred, fitpar ) ;
          }
          const double dchisqconverged = 0.001 ;
          double newchisq = kalman.chisq() ;
          double dchisq = newchisq - chisq ;
          bool diverging = iter > 0 && dchisq>0  ;
          bool converged = fabs(dchisq) < dchisqconverged ;
          finished  = ++iter >= m_maxNIter || diverging || converged ;

          if(vtxverbose>=3) {
            std::cout << "chi2,niter: "
            << iter << " "<< std::setprecision(7)
            << std::setw(12) << chisq << " "
            << std::setw(12)<< newchisq << " "
            << std::setw(12)<< dchisq << " "
            << diverging << " "
            << converged << " ";
            status.Print(std::cout);
            std::cout << std::endl ;
          }
          chisq = newchisq ;
        }
      }
    }
    if(!status.failure()) {
      kalman.updateCov( fitpar ) ;
      fitpar->addChiSquare( kalman.chisq(), m_dim, p.particle() ) ;
    }
    if(pred) delete pred ;
    if(vtxverbose>=4 &&m_node&&
       m_node->mother() ) { m_node->mother()->print(fitpar) ; }
  }
  if(vtxverbose>=5) { std::cout << "Constraint::filter(FitParams*): done"  ; print() ;}
  return status ;
}

ErrCode DecayTreeFitter::Constraint::filter ( FitParams *  fitpar, const FitParams *  reference  ) const

virtual

Definition at line 131 of file Constraint.cxx.

References DecayTreeFitter::FitParams::addChiSquare(), DecayTreeFitter::ErrCode::badsetup, DecayTreeFitter::KalmanCalculator::chisq(), DecayTreeFitter::FitParams::dim(), DecayTreeFitter::ErrCode::failure(), DecayTreeFitter::KalmanCalculator::init(), kalman, name, ntypes, p, DecayTreeFitter::FitParams::par(), print(), DecayTreeFitter::ErrCode::Print(), status, DecayTreeFitter::KalmanCalculator::updateCov(), DecayTreeFitter::KalmanCalculator::updatePar(), and vtxverbose.

{
  // filter but linearize around reference
  ErrCode status ;
  if(m_type<=Constraint::unknown || m_type>=Constraint::ntypes) {
    std::cout << "Constraint::filter(FitParams*,FitParams*): unknown constraint: " << m_type
    << std::endl ;
    status |= ErrCode::badsetup ;
  } else if (m_type!=merged && !m_node) {
    std::cout << "Constraint::filter(FitParams*,FitParams*): filter constraint without a node"
    << std::endl ;
    status |= ErrCode::badsetup ;
  } else {
    if(vtxverbose>=3) { std::cout << "Constraint::filter(FitParams*,FitParams*): filtering (par&refpar) "  ; print() ;}

    // project using the reference
    Projection p(fitpar->dim(),m_dim) ;
    if(vtxverbose>=5) { std::cout << "Constraint::filter(FitParams*,FitParams*): project"<<std::endl;}
    status = project(reference,p) ;
    if(vtxverbose>=5) { std::cout << "Constraint::filter(FitParams*,FitParams*): residual"<<std::endl;}
    // now update the residual
    p.r() += p.H() * (fitpar->par() - reference->par()) ;
    if(vtxverbose>=5) { std::cout << "Constraint::filter(FitParams*,FitParams*): kalman"<<std::endl;}
    // now call the Kalman update as usual
    KalmanCalculator kalman ;
    status |= kalman.init( p.r(), p.H(), fitpar, p.V() ) ;
    //status |= kalman.init( p.r(), p.H(), fitpar, (&p.V()) ? ( p.V().NonZeros()>0 ? &p.V() : 0 ) : 0  ) ;
    kalman.updatePar( fitpar ) ;
    kalman.updateCov( fitpar ) ;
    fitpar->addChiSquare( kalman.chisq(), m_dim, p.particle() ) ;
    if(vtxverbose>=3) { std::cout << "Constraint::filter(FitParams*,FitParams*): \""<<name()<<"\" Chisquare contribution: "  <<kalman.chisq()<<" with "<<m_dim<<" dimensions from particle "<<p.particle()->name()<<std::endl;}
    if(vtxverbose>=5) { std::cout << "Constraint::filter(FitParams*,FitParams*): done"<<std::endl;}
  }
  if( status.failure()){
    std::cout << "Constraint::filter(FitParams*,FitParams*): error filtering constraint: "
    << name() << " "; status.Print(std::cout); std::cout << std::endl ;}

  return status ;
}

bool DecayTreeFitter::Constraint::isLineair ( ) const

inline

Definition at line 52 of file Constraint.h.

References m_maxNIter.

{ return m_maxNIter <=1 ; }

std::string DecayTreeFitter::Constraint::name

(

)

const

Definition at line 179 of file Constraint.cxx.

References track.

{
  std::string rc = "unknown constraint!" ;
  switch(m_type)
  {
    case beamspot:     rc = "beamspot" ; break ;
    case beamenergy:   rc = "beamenergy" ; break ;
    case composite:    rc = "composite" ; break ;
    case resonance:    rc = "resonance" ; break ;
    case track:        rc = "track" ; break ;
    case photon:       rc = "photon" ; break ;
    case kinematic:    rc = "kinematic" ; break ;
    case geometric:    rc = "geometric" ; break ;
    case mass:         rc = "mass" ; break ;
    case massEnergy:   rc = "massEnergy" ; break ;
    case lifetime:     rc = "lifetime" ; break ;
    case merged:       rc = "merged" ; break ;
    case conversion:   rc = "conversion" ; break ;
    case ntypes:
    case unknown:
      break ;
  }
  return rc ;
}

unsigned int DecayTreeFitter::Constraint::nIter ( ) const

inline

Definition at line 53 of file Constraint.h.

References m_maxNIter.

Referenced by DecayTreeFitter::MergedConstraint::push_back().

{ return m_maxNIter ; }

bool DecayTreeFitter::Constraint::operator<

(

const Constraint &

rhs

)

const

Definition at line 20 of file Constraint.cxx.

References m_depth, and m_type.

{
  // the simple way
  return m_type < rhs.m_type ||
  (m_type == rhs.m_type && m_depth < rhs.m_depth) ;

  // this is probably the second most complicated routine: how do we
  // order the constraints. there is one very special case:
  // Ks->pipi0 requires the pi0 mass constraints at the very
  // end. otherwise, it just doesn't work. in all other cases, we
  // prefer to fit 'down' the tree'. the 'external' constraints must
  // be filtered first, but soft pions must be fitted after the
  // geometric constraints of the D. You see, this is horrible.

  // if either of the two is external, or either of the two is a
  // mass constraint, we order by m_typem_

  //  if( (m_type <= Constraint::composite ||
  //      rhs.m_type <= Constraint::composite ) ||
  //    (m_type >= Constraint::mass ||
  //     rhs.m_type >= Constraint::mass ) ) {
  //      return m_type < rhs.m_type ||
  //      (m_type == rhs.m_type && m_depth < rhs.m_depth) ;
  //   }
  //return m_depth < rhs.m_depth  ||
  //(m_depth == rhs.m_depth && m_type < rhs.m_type ) ;

}

bool DecayTreeFitter::Constraint::operator== ( const Constraint &  rhs ) const

inline

Definition at line 46 of file Constraint.h.

References m_type.

                                                 {
      return m_type == rhs.m_type ; }

void DecayTreeFitter::Constraint::print ( std::ostream &  os = std::cout ) const

virtual

Reimplemented in DecayTreeFitter::MergedConstraint.

Definition at line 171 of file Constraint.cxx.

{
  os << "node index = " << m_node->index()
  << " name = " << m_node->name().c_str()
  << " constraint type = " << m_type
  << " depth = " << m_depth << std::endl ;
}

ErrCode DecayTreeFitter::Constraint::project ( const FitParams *  fitpar, Projection &  p  ) const

virtual

Reimplemented in DecayTreeFitter::MergedConstraint.

Definition at line 51 of file Constraint.cxx.

References DecayTreeFitter::Projection::setParticle().

{
  // this one will be overruled by the MergedConstraint
  p.setParticle( *m_node ) ;
  return m_node->projectConstraint(m_type,fitpar,p) ;
}

void DecayTreeFitter::Constraint::setDim ( unsigned int  d )

inlineprotected

Definition at line 80 of file Constraint.h.

References d, and m_dim.

Referenced by DecayTreeFitter::MergedConstraint::MergedConstraint(), and DecayTreeFitter::MergedConstraint::push_back().

{ m_dim = d ; }

void DecayTreeFitter::Constraint::setNIter ( unsigned int  d )

inlineprotected

Definition at line 81 of file Constraint.h.

References d, and m_maxNIter.

Referenced by DecayTreeFitter::MergedConstraint::push_back().

{ m_maxNIter = d ; }

void DecayTreeFitter::Constraint::setWeight ( int  w )

inline

Definition at line 73 of file Constraint.h.

References m_weight.

{ m_weight = w<0 ? -1 : 1 ; }

Type DecayTreeFitter::Constraint::type ( ) const

inline

Definition at line 50 of file Constraint.h.

References m_type.

{ return m_type ; }

Member Data Documentation

int DecayTreeFitter::Constraint::m_depth

private

Definition at line 84 of file Constraint.h.

Referenced by operator<().

unsigned int DecayTreeFitter::Constraint::m_dim

private

Definition at line 86 of file Constraint.h.

Referenced by dim(), and setDim().

int DecayTreeFitter::Constraint::m_maxNIter

private

Definition at line 89 of file Constraint.h.

Referenced by isLineair(), nIter(), and setNIter().

const ParticleBase* DecayTreeFitter::Constraint::m_node

private

Definition at line 83 of file Constraint.h.

Type DecayTreeFitter::Constraint::m_type

private

Definition at line 85 of file Constraint.h.

Referenced by operator<(), operator==(), and type().

int DecayTreeFitter::Constraint::m_weight

private

Definition at line 88 of file Constraint.h.

Referenced by setWeight().

---

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

• Constraint.h
• Constraint.cxx