From Philippe & Laurent: new variant of MUON visualization.

[u/mrichter/AliRoot.git] / STAT / AliTMinuitToolkit.h

diff --git a/STAT/AliTMinuitToolkit.h b/STAT/AliTMinuitToolkit.h
index 39863312753c06d74a63bf7333a361e0358aba5c..b9b5f269c67626a6b11f9ccb81004515ad8a062f 100644 (file)
--- a/STAT/AliTMinuitToolkit.h
+++ b/STAT/AliTMinuitToolkit.h
@@ -34,17 +34,26 @@ public:
   void SetMaxCalls(Int_t calls) {fMaxCalls = calls;};
   void SetTolerance(Double_t tol) {fPrecision = tol;};
   void SetPoints(TMatrixD * points) {fPoints = points;};
-  void Test();
+  void SetWeights(TVectorD * weights) {fWeights = weights;};
+  static void Test();
   static void FitterFCN(int &npar, double  *dummy, double &fchisq, double *gin, int iflag);
   void Fit();
+  void EnableRobust(Bool_t b) {fUseRobust = b;};
+  void EnableRobust(Bool_t b, Double_t sigma) {fUseRobust = b; fExpectedSigma = sigma;}; 
   //
   TMatrixD * GetPoints() {return fPoints;};
+  TVectorD * GetWeights() {return fWeights;};
   TFormula * GetFormula() {return fFormula;};
   TVectorD * GetParameters() {return fParam;};
   TFormula * GetWeightFunction() {return fWeightFunction;};
   TMatrixD * GetCovarianceMatrix() {return fCovar;};
+  Bool_t     GetStatus() {return fUseRobust;};
+  Double_t   GetExpectedSigma() {return fExpectedSigma;};
  
 private:
+  //
+  AliTMinuitToolkit(const AliTMinuitToolkit&); // fake copy constr. -- suppress warnings
+  AliTMinuitToolkit& operator=(const AliTMinuitToolkit&); // fake -- suppress warnings
   //
   TFormula        * fFormula;            // formula of the fitted function
   TFormula        * fWeightFunction;     // weight function, must be defined between 0 and 1
@@ -52,12 +61,15 @@ private:
   //
   //
   TMatrixD        * fPoints;             // fitted points
+  TVectorD        * fWeights;            // weights of the points
   TVectorD        * fParam;              // parameter values
   TMatrixD        * fParamLimits;        // limits of the parameters (up, down)
   TMatrixD        * fCovar;              // covariance matrix
   Double_t        * fChi2;               // chi-square
   Int_t             fMaxCalls;           // maximum number of calls, default value depends on fit algorithm
   Double_t          fPrecision;          // normalized tolerance, default value depends on fit algorithm
+  Bool_t            fUseRobust;          // switch on/off robust option, default: kFalse
+  Double_t          fExpectedSigma;      // expected sigma to normalize robust fitting
   //  
   ClassDef(AliTMinuitToolkit,1);
 };