OPALFitterGSL.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * See https://github.com/tferber/OrcaKinfit, forked from                 *
 * https://github.com/iLCSoft/MarlinKinfit                                *
 *                                                                        *
 * Further information about the fit engine and the user interface        *
 * provided in MarlinKinfit can be found at                               *
 * https://www.desy.de/~blist/kinfit/doc/html/                            *
 * and in the LCNotes LC-TOOL-2009-001 and LC-TOOL-2009-004 available     *
 * from http://www-flc.desy.de/lcnotes/                                   *
 *                                                                        *
 * Adopted by: Torben Ferber (torben.ferber@desy.de) (TF)                 *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef __OPALFITTERGSL_H
#define __OPALFITTERGSL_H
 
#include<vector>
#include "analysis/OrcaKinFit/BaseFitter.h"
 
#include <gsl/gsl_vector.h>
#include <gsl/gsl_matrix.h>
#include <gsl/gsl_permutation.h>
 
namespace Belle2 {
  namespace OrcaKinFit {
 
    class BaseFitObject;
    class BaseConstraint;
    class OPALFitterGSL : public BaseFitter {
    public:
      OPALFitterGSL();
      virtual ~OPALFitterGSL();
      virtual double fit() override;
 
 
      virtual int getError() const override;
 
      virtual double getProbability() const override;
      virtual double getChi2() const override;
      virtual int    getDoF() const override;
      virtual int  getIterations() const override;
 
      virtual int    getNcon() const;
 
      virtual int    getNsoft() const;
 
      virtual int    getNpar() const;
 
      virtual int    getNunm() const;
 
      virtual bool initialize() override;
 
      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;
 
    };
 
  }// end OrcaKinFit namespace
} // end Belle2 namespace
 
#endif // __OPALFITTERGSL_H