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(int idim, double *M) const {assert (false);};
    virtual void add2ndDerivativesToMatrix(int idim, double *M, double lambda) 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

---