BaseFitter.h

Go to the documentation of this file.

#ifndef __BASEFITTER_H
#define __BASEFITTER_H

#include<vector>
#include<string>
#include<map>

class BaseFitObject;
class BaseConstraint;
class BaseHardConstraint;
class BaseSoftConstraint;
class BaseTracer;

//  Class BaseConstraint:

class BaseFitter {
  public:
    BaseFitter();
    virtual ~BaseFitter();
    virtual void addFitObject (BaseFitObject* fitobject_);
    virtual void addFitObject (BaseFitObject& fitobject_);
    virtual void addConstraint (BaseConstraint* constraint_);
    virtual void addConstraint (BaseConstraint& constraint_);
    virtual void addHardConstraint (BaseHardConstraint* constraint_);
    virtual void addHardConstraint (BaseHardConstraint& constraint_);
    virtual void addSoftConstraint (BaseSoftConstraint* constraint_);
    virtual void addSoftConstraint (BaseSoftConstraint& constraint_);
    virtual std::vector<BaseFitObject*>* getFitObjects();
    virtual std::vector<BaseHardConstraint*>* getConstraints();
    virtual std::vector<BaseSoftConstraint*>* getSoftConstraints();
    virtual double fit() = 0;
    virtual int getError() const = 0;
    virtual double getProbability() const = 0;
    virtual double getChi2() const = 0;
    virtual int    getDoF() const = 0;
    virtual int   getIterations() const = 0;
  
    virtual void reset();
    virtual bool initialize() = 0;
    
    virtual BaseTracer *getTracer();
    virtual const BaseTracer *getTracer() const;
    virtual void setTracer(BaseTracer *newTracer
                          );
    virtual void setTracer(BaseTracer& newTracer
                          );
    
    virtual const double *getGlobalCovarianceMatrix (int& idim 
                                                          ) const;                 
    virtual double *getGlobalCovarianceMatrix (int& idim 
                                              );                 
  
  protected:
    BaseFitter (const BaseFitter& rhs);
    BaseFitter& operator= (const BaseFitter& rhs);
    
    
    typedef std::vector <BaseFitObject *> FitObjectContainer;
    typedef std::vector <BaseHardConstraint *> ConstraintContainer;
    typedef std::vector <BaseSoftConstraint *> SoftConstraintContainer;
    
    typedef FitObjectContainer::iterator FitObjectIterator;
    typedef ConstraintContainer::iterator ConstraintIterator;
    typedef SoftConstraintContainer::iterator SoftConstraintIterator;
    
    FitObjectContainer      fitobjects;
    ConstraintContainer     constraints;
    SoftConstraintContainer softconstraints;    
    
    int     covDim;   
    double *cov;      
    bool    covValid; 

#ifndef FIT_TRACEOFF    
    BaseTracer *tracer;
#endif 
  public:  
#ifndef FIT_TRACEOFF    
    std::map<std::string, double> traceValues;
#endif 

};

#endif // __BASEFITTER_H