SoftGaussParticleConstraint.h

Go to the documentation of this file.

#ifndef __SOFTGAUSSPARTICLECONSTRAINT_H
#define __SOFTGAUSSPARTICLECONSTRAINT_H

#include<vector>
#include<cassert>
#include "BaseSoftConstraint.h"
#include "BaseFitObject.h"

class ParticleFitObject;

//  Class SoftGaussParticleConstraint:

class SoftGaussParticleConstraint: public BaseSoftConstraint {
  public:
    SoftGaussParticleConstraint(double sigma_     
                               );
    virtual ~SoftGaussParticleConstraint() {};
    
    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 void resetFOList(){
      fitobjects.resize (0);
      flags.resize (0);
    }; 
    
    virtual double getValue() const = 0;
    
    virtual double getChi2() const;
    
    virtual double getError() const;
    
    virtual double getSigma() const;
    
    virtual double setSigma(double sigma_     
                           );
    
    virtual void getDerivatives(int idim,      
                                double der[]   
                               ) const = 0;
    virtual void add2ndDerivativesToMatrix(double *M,     
                                           int idim       
                                          ) const;

    virtual void addToGlobalChi2DerVector (double *y,   
                                           int idim     
                                           ) const;
    
    
    virtual void invalidateCache() const 
    {}
    
    void test1stDerivatives ();
    void test2ndDerivatives ();
    
    double num1stDerivative (int ifo,     
                             int ilocal,  
                             double eps   
                            );
    double num2ndDerivative (int ifo1,    
                             int ilocal1, 
                             double eps1, 
                             int ifo2,    
                             int ilocal2, 
                             double eps2  
                            );
                              
    int getVarBasis() const;
  
  protected:
  
    virtual bool secondDerivatives (int i,                        
                                    int j,                        
                                    double *derivatives           
                                   ) const = 0;
    virtual bool firstDerivatives (int i,                        
                                   double *derivatives           
                                  ) const = 0;
  
  
    typedef std::vector <ParticleFitObject *> FitObjectContainer;    
    typedef FitObjectContainer::iterator FitObjectIterator;
    typedef FitObjectContainer::const_iterator ConstFitObjectIterator;
    FitObjectContainer fitobjects;
    std::vector <double> derivatives;
    std::vector <int> flags;
    
    double sigma;

    enum { VAR_BASIS=BaseDefs::VARBASIS_EPXYZ }; // this means that the constraint knows about E,px,py,pz

};

#endif // __SOFTGAUSSPARTICLECONSTRAINT_H