TrackConstraint.h

Go to the documentation of this file.

#ifndef __TrackConstraint_H
#define __TrackConstraint_H

#include<vector>
#include<cassert>
#include "BaseConstraint.h"

class TrackFitObject;

//  Class TrackConstraint:

class TrackConstraint: public BaseConstraint {
  public:
    inline TrackConstraint();
    virtual ~TrackConstraint() {};
    
    virtual void setFOList(std::vector <TrackFitObject*> *fitobjects_ 
                          ){
      for (int i = 0; i < (int) fitobjects_->size(); i++) {
        fitobjects.push_back ((*fitobjects_)[i]);
        flags.push_back (1);
        sign.push_back (i == 0 ? 1 : -1);
      }  
    }; 
    virtual void addToFOList(TrackFitObject& fitobject, int flag = 1
                             ){
      fitobjects.push_back (&fitobject);
      flags.push_back (flag);
      sign.push_back (sign.size() == 0 ? 1 : -1);
    }; 
    virtual double getValue() const = 0;
    
    virtual void getDerivatives(int idim, double der[]) const = 0;
    virtual void add1stDerivativesToMatrix(double *M,      
                                           int idim        
                                           ) const 
    {assert (false);}
    virtual void add2ndDerivativesToMatrix (double *M,      
                                            int idim,       
                                            double lambda   
                                            ) const
    {assert (false);}
    
    virtual void addToGlobalChi2DerVector (double *y,   
                                           int idim,    
                                           double lambda //< The lambda value
                                           ) const
    {assert (false);};
    
    virtual int  getGlobalNum() const {return globalNum;}
    virtual void setGlobalNum(int iglobal) {globalNum = iglobal;}
    
    virtual void invalidateCache() const {};
  
  protected:
    typedef std::vector <TrackFitObject *> FitObjectContainer;    
    typedef FitObjectContainer::iterator FitObjectIterator;
    typedef FitObjectContainer::const_iterator ConstFitObjectIterator;
    FitObjectContainer fitobjects;
    std::vector <double> derivatives;
    std::vector <int> flags;
    std::vector <double> sign;
    
    int globalNum;

};

TrackConstraint::TrackConstraint() {
  invalidateCache();
}

#endif // __TrackConstraint_H