Minimiser.h

Go to the documentation of this file.

#ifndef MINIMISER_HH
#define MINIMISER_HH
// author: Jonas Rademacker (Jonas.Rademacker@bristol.ac.uk)
// status:  Mon 9 Feb 2009 19:17:55 GMT

#include "TMinuit.h"
#include "TMatrixTSym.h"
#include "Mint/NamedParameter.h"
#include "Mint/IMinimisable.h"

#include <iostream>
#include <vector>

class TGraph;
class TH2D;

namespace MINT{

class IMinuitParameter;
class FitParameter;

class Minimiser : public TMinuit{
  mutable Double_t arglist[10];
  mutable Int_t ierflg;
 protected:
  bool  _useAnalyticGradient;  
  static Minimiser* _defaultMinimiser;
  static int _defaultMaxCalls;

  MinuitParameterSet* _parSet;
  //  MinuitParameterSet _floating;

  IMinimisable* _theFunction;
  int _maxCalls;

  int _printLevel;
  double _errdef ;

  bool init();  
  bool MakeSpace(int needSpace);
  void TMinInit();

  bool updateFitParameters(Double_t* p);
  bool setParametersToResult();
  bool endOfFit();
  //  bool detachParameters();

  Int_t Eval(Int_t npar
         , Double_t *grad
         , Double_t &fval
         , Double_t *par
         , Int_t flag
         );
 public:
  static Minimiser* getDefaultMinimiser();

  Minimiser(IMinimisable* fitFunction=0, const double errdef = 1.);
  virtual ~Minimiser();
  bool attachFunction(IMinimisable* fcn);
  IMinimisable* theFunction(){ return _theFunction;}
  const IMinimisable* theFunction() const{ return _theFunction;}

  const MinuitParameterSet* parSet()const{ return _parSet;}
  MinuitParameterSet* parSet(){ return _parSet;}

  unsigned int nPars() const;
  IMinuitParameter* getParPtr(unsigned int i);
  const IMinuitParameter* getParPtr(unsigned int i) const;

  bool OK() const;
  bool parsOK() const;
  bool fcnOK() const;

  double getFCNVal();
  void FCNGradient(std::vector<double>& grad);

  bool initialiseVariables();

  bool setPrintLevel(int level=-1);
  bool temporarilyQuiet();
  bool resetPrintLevel();

  bool SetSomeMinuitOptions();
  bool CallMigrad();
  bool CallMinos();
  bool CallSeek(int maxCalls = 100, int devs = 5);
  bool CallSimplex(int maxCalls = 300, double tolerance = 1. );
  bool CallImprove(int maxCalls= 1500, int searches= 5);

  bool prepFit();
  bool doFit();
  bool doMinosFit();
  bool doSeekFit(int maxCalls = 100, int devs = 5);
  bool doSimplexFit(int maxCalls = 300, double tolerance = 1. );

  bool scanMarked();
  bool scanAll();
  TGraph* scan(int i, double from=0, double to=0);
  TGraph* scan(IMinuitParameter& fp, double from=0, double to=0);


  void setMaxCalls(int maxCalls);
  int getMaxCalls() const;

  void printResultVsInput(std::ostream& os = std::cout) const;

  // get number of free parameters from TMinuit call:
  // GetNumFreePars()

  TMatrixTSym<double> covMatrix();
  TMatrixTSym<double> covMatrixFull();

  int status() ;

  enum FitStatus {
    NOTCALCULATED = 0, 
    NOTACCURATE, 
    NOTPOSDEF, 
    CONVERGED 
  } ;

  bool isStatus(Minimiser::FitStatus) ;

  bool isStatusAtLeast(Minimiser::FitStatus) ;

  int printStatus() ;

  bool isConverged() ;

  TGraph* contour(unsigned, unsigned, float nsigma = 1., unsigned npoints = 40) ;

  
};
}//namespace MINT
#endif
//