VertexFitObject.h

Go to the documentation of this file.

#ifndef __VERTEXFITOBJECT_H
#define __VERTEXFITOBJECT_H

#include "jbltools/kinfit/BaseFitObject.h"

#include <iostream>
#include <cassert>
#include <vector>

class ThreeVector;
class FourVector;
class TrackFitObject;
class BaseFitter;
class BaseConstraint;

// Class VertexFitObject

class VertexFitObject: public BaseFitObject {
  public:
    VertexFitObject(const char *name_,
                    double x,
                    double y,
                    double z
                   );
    VertexFitObject (const VertexFitObject& rhs);
    VertexFitObject& operator= (const VertexFitObject& rhs);
    virtual ~VertexFitObject();
    
    virtual bool   setParam (int ilocal,         
                             double par_,        
                             bool measured_,     
                             bool fixed_ = false 
                            );  
    virtual bool   setParam (int ilocal,         
                             double par_         
                            );  
    virtual bool   setMParam (int ilocal,         
                              double mpar_        
                             );  
    virtual bool   setError (int ilocal,    
                             double err_    
                             );
    virtual bool   setCov (int ilocal,    
                           int jlocal,    
                           double cov_    
                          );
    virtual bool setGlobalParNum (int ilocal, int iglobal); 
    
    virtual bool fixParam (int ilocal,    
                           bool fix=true  
                          );
    
    virtual double getParam (int ilocal     
                            ) const;
    virtual double getMParam (int ilocal     
                            ) const;
    virtual double getError (int ilocal     
                            ) const;
    virtual double getCov (int ilocal,    
                           int jlocal     
                          ) const;
    virtual bool isParamMeasured (int ilocal) const;
    virtual bool isParamFixed (int ilocal) const;
    virtual int getGlobalParNum(int ilocal) const;
    virtual int getNPar() const;
    
    virtual const char *getName () const;
    
    virtual void setName (const char *name_);
    
    virtual const char *getParamName (int ilocal     
                                     ) const;
   
    
    virtual void getVertexEx (ThreeVector& p
                             ) const;
    virtual ThreeVector getVertex() const; 
    
    virtual void getVertexDerivativeEx (int ilocal, 
                                        ThreeVector& p
                                       ) const;
    virtual ThreeVector getVertexDerivative (int ilocal 
                                            ) const;

    virtual void addToGlobCov(double *cov, int idim) const; 
        
    
    virtual double getChi2() const;
    virtual double getDChi2DParam(int ilocal) const {assert (0);};
    virtual double getD2Chi2DParam2(int ilocal1, int ilocal2) const {assert (0);};
    
    virtual void addToGlobalChi2DerMatrix (int idim, double *M) const {assert (0);};
    virtual void addToGlobalDerMatrix (int idim, double c, double *M) const {assert (0);};
    
    virtual std::ostream& print (std::ostream& os   
                                ) const;
    
    virtual TrackFitObject *getTrack (int i
                                     ) const {return tracks[i].track;}
        
    virtual void addTrack (TrackFitObject *track,   
                           bool inbound,            
                           bool measured            
                          );
    
    enum  {
      VX = 1, VY = 2, VZ = 4, VXY = 3, VXYZ=7,
      PX = 8, PY = 16, PZ = 32, PXY = 24, PXYZ = 56,
      E = 64, EPXYZ = 120,
      THE_FULL_MONTY = 127
    };
    
    virtual void addConstraints (BaseFitter& fitter, 
                                 int mask = THE_FULL_MONTY
                                );
    
    virtual ThreeVector estimatePosition ();
    
    virtual void initForFit();
            
  protected:
    virtual void addVertexConstraints (BaseFitter& fitter, int axis);
    virtual void addMomentumConstraint (BaseFitter& fitter, int axis);
  
    virtual void initCov();
    
    virtual bool calculateCovInv() const;
    
    virtual void calculateChi2() const;
    virtual void copy (const VertexFitObject& source);
  
  
    enum {NPAR = 3};
    
    double par[NPAR];
    double mpar[NPAR];
    double err[NPAR];
    bool measured[NPAR];
    bool fixed[NPAR];
    int globalParNum [NPAR];
    double cov [NPAR][NPAR];    
    mutable double covinv [NPAR][NPAR];   
    mutable bool covinvvalid; 
    
    // Cache variables
    mutable double chi2;               
    mutable double resid[NPAR];     
    mutable bool   chi2contr[NPAR]; 
    
    char *name;                        
    
    struct TrackDescriptor {
      TrackFitObject *track;
      bool inbound;
      bool measured;
      TrackDescriptor (TrackFitObject *track_, bool inbound_, bool measured_)
        : track (track_), inbound (inbound_), measured (measured_) 
        {}
      TrackDescriptor ()
        : track (0), inbound (0), measured (0) 
        {}
    };
    
    typedef std::vector<TrackDescriptor> TContainer;
    typedef TContainer::iterator TIterator;
    TContainer tracks;
    typedef std::vector<BaseConstraint *> CContainer;
    typedef CContainer::iterator CIterator;
    CContainer constraints;
    
};
    
#endif // __VERTEXFITOBJECT_H

---