OPALFitterGSL.h

Go to the documentation of this file.

// Class OPALFitter
//
// Author: Jenny Boehme
// Last update: $Date: 2010/10/29 14:55:27 $
//          by: $Author: mbeckman $
// 
// Description: kinematic fit a la WWFGO
//               

#ifndef __OPALFITTERGSL_H
#define __OPALFITTERGSL_H

#include<vector>
#include "BaseFitter.h"

#include <gsl/gsl_vector.h>
#include <gsl/gsl_matrix.h>
#include <gsl/gsl_permutation.h>

class BaseFitObject;
class BaseConstraint;
class OPALFitterGSL : public BaseFitter {
  public:
    OPALFitterGSL();
    virtual ~OPALFitterGSL();
    virtual double fit();
    

    virtual int getError() const;
    
    virtual double getProbability() const;
    virtual double getChi2() const;
    virtual int    getDoF() const;
    virtual int  getIterations() const;

    virtual int    getNcon() const;
    
    virtual int    getNsoft() const;
    
    virtual int    getNpar() const;
    
    virtual int    getNunm() const;
    
    virtual bool initialize();
  
    virtual void setDebug (int debuglevel);
    
  protected:
    
    
    virtual bool updateFitObjects (double etaxi[]);
    enum {NPARMAX=50, NCONMAX=20, NUNMMAX=20};
    
    int npar;      // total number of parameters
    int nmea;      // total number of measured parameters
    int nunm;      // total number of unmeasured parameters
    int ncon;      // total number of constraints
    int ierr;      // Error status
    int nit;       // Number of iterations

    double fitprob;   // fit probability
    double chi2;      // final chi2
    
    static void ini_gsl_permutation (gsl_permutation *&p, unsigned int size);
    static void ini_gsl_vector (gsl_vector *&v, int unsigned size);
    static void ini_gsl_matrix (gsl_matrix *&m, int unsigned size1, unsigned int size2);
    
    static void debug_print (gsl_matrix *m, const char *name);
    static void debug_print (gsl_vector *v, const char *name);
    
  private:
    gsl_vector *f; 
    gsl_vector *r; 
    gsl_matrix *Fetaxi; 
    gsl_matrix *S; 
    gsl_matrix *Sinv; 
    gsl_matrix *SinvFxi; 
    gsl_matrix *SinvFeta; 
    gsl_matrix *W1; 
    gsl_matrix *G; 
    gsl_matrix *H; 
    gsl_matrix *HU; 
    gsl_matrix *IGV; 
    gsl_matrix *V; 
    gsl_matrix *VLU; 
    gsl_matrix *Vinv; 
    gsl_matrix *Vnew; 
    gsl_matrix *Minv; 
    gsl_matrix *dxdt; 
    gsl_matrix *Vdxdt; 
    gsl_vector *dxi; 
    gsl_vector *Fxidxi;
    gsl_vector *lambda;
    gsl_vector *FetaTlambda;
    gsl_vector *etaxi;
    gsl_vector *etasv;
    gsl_vector *y;           
    gsl_vector *y_eta;           
    gsl_vector *Vinvy_eta;           
    //    gsl_matrix *Feta; //<-- DJeans: not used/needed
    gsl_matrix *FetaV; 
    
    gsl_permutation *permS; 
    gsl_permutation *permU; 
    gsl_permutation *permV; 
    
    int debug;
    
};

#endif // __OPALFITTERGSL_H