ParticleConstraint.h

Go to the documentation of this file.

#ifndef __PARTICLECONSTRAINT_H
#define __PARTICLECONSTRAINT_H

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

class ParticleFitObject;

//  Class ParticleConstraint:

class ParticleConstraint: public BaseConstraint {
  public:
    inline ParticleConstraint();
    virtual ~ParticleConstraint() {};
    
    virtual void setFOList(std::vector <ParticleFitObject*> *fitobjects_ 
                          ){
      for (int i = 0; i < (int) fitobjects_->size(); i++) {
        fitobjects.push_back ((*fitobjects_)[i]);
        flags.push_back (1);
      }  
    }; 
    virtual void addToFOList(ParticleFitObject& fitobject, int flag = 1
                             ){
      fitobjects.push_back (&fitobject);
      flags.push_back (flag);
    }; 
    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 <ParticleFitObject *> FitObjectContainer;    
    typedef FitObjectContainer::iterator FitObjectIterator;
    typedef FitObjectContainer::const_iterator ConstFitObjectIterator;
    FitObjectContainer fitobjects;
    std::vector <double> derivatives;
    std::vector <int> flags;
    
    int globalNum;

};

ParticleConstraint::ParticleConstraint() {
  invalidateCache();
}

#endif // __PARTICLECONSTRAINT_H

---