---

/Go4Fit/TGo4Fitter.h

Go to the documentation of this file.

//---------------------------------------------------------------
//        Go4 Release Package v2.10-5 (build 21005) 
//                      03-Nov-2005
//---------------------------------------------------------------
//       The GSI Online Offline Object Oriented (Go4) Project
//       Experiment Data Processing at DVEE department, GSI
//---------------------------------------------------------------
//
//Copyright (C) 2000- Gesellschaft f. Schwerionenforschung, GSI
//                    Planckstr. 1, 64291 Darmstadt, Germany
//Contact:            http://go4.gsi.de
//----------------------------------------------------------------
//This software can be used under the license agreements as stated
//in Go4License.txt file which is part of the distribution.
//----------------------------------------------------------------
#ifndef TGO4FITTER_H
#define TGO4FITTER_H

#include <iostream.h>

#include "TH1.h"
#include "TGraph.h"
#include "TArrayD.h"

#include "TObjArray.h"
#include "TGo4FitParameter.h"
#include "TGo4FitData.h"
#include "TGo4FitDataHistogram.h"
#include "TGo4FitDataGraph.h"
#include "TGo4FitModel.h"
#include "TGo4FitModelGauss1.h"
#include "TGo4FitSlot.h"
#include "TGo4FitterAbstract.h"
#include "TGo4FitterConfig.h"

typedef Double_t (*TUserFitFunction)(Double_t, Double_t, Double_t);

class TGo4Fitter : public TGo4FitterAbstract {
   public:
      enum  { ff_least_squares = 0,
              ff_chi_square    = 1,
              ff_chi_Pearson   = 2,
              ff_chi_Neyman    = 3,
              ff_chi_gamma     = 4,
              ff_ML_Poisson    = 5,
              ff_user          = 6 };

      TGo4Fitter();

      TGo4Fitter(const char* iName, const char* iTitle);

      TGo4Fitter(const char * iName, Int_t iFitFunctionType, Bool_t IsAddStandardActions);

      virtual ~TGo4Fitter();

      void SetFitFunctionType(Int_t iFitFunctionType) { fiFitFunctionType = iFitFunctionType; }

      Int_t GetFitFunctionType() { return fiFitFunctionType; }

      void SetUserFitFunction(TUserFitFunction iFunc) { fxUserFitFunction = iFunc; fiFitFunctionType = ff_user; }

      void SetMemoryUsage(Int_t iMemoryUsage);

      Int_t GetMemoryUsage() { return fiMemoryUsage; }

      virtual void Clear(Option_t* option = 0);

      Int_t GetNumData() const { return fxDatas.GetLast()+1; }

      TGo4FitData* GetData(Int_t n) { return (n>=0) && (n<GetNumData()) ? dynamic_cast<TGo4FitData*> (fxDatas[n]) : 0; }

      const char* GetDataName(Int_t n) { return GetData(n) ? GetData(n)->GetName() : 0; }

      TGo4FitData* FindData(const char* DataName) { return (DataName==0) ? 0 : dynamic_cast<TGo4FitData*> (fxDatas.FindObject(DataName)); }

      TGo4FitData* AddData(TGo4FitData* d);

      TGo4FitDataHistogram* AddH1(const char* DataName, TH1* histo, Bool_t Owned = kFALSE, Double_t lrange = 0., Double_t rrange = 0.);

      TGo4FitDataHistogram* SetH1(const char* DataName, TH1* histo, Bool_t Owned = kFALSE);

      TGo4FitDataGraph* AddGraph(const char* DataName, TGraph* gr, Bool_t Owned = kFALSE, Double_t lrange = 0., Double_t rrange = 0.);

      TGo4FitDataGraph* SetGraph(const char* DataName, TGraph* gr, Bool_t Owned = kFALSE);

      TGo4FitData* RemoveData(const char* DataName, Bool_t IsDel = kFALSE);

      void DeleteAllData();

      Int_t GetNumModel() const { return fxModels.GetLast()+1; }

      TGo4FitModel* GetModel(Int_t n) { return (n>=0) && (n<GetNumModel()) ? dynamic_cast<TGo4FitModel*> (fxModels[n]) : 0; }

      TGo4FitModel* FindModel(const char* ModelName) { return (ModelName==0) ? 0 : dynamic_cast<TGo4FitModel*> (fxModels.FindObject(ModelName)); }

      TGo4FitModel* AddModel(TGo4FitModel* m);

      TGo4FitModel* AddModel(const char* DataName, TGo4FitModel* m);

      void AddPolynomX(const char* DataName, const char* NamePrefix, Int_t MaxOrder = 1, Int_t GroupIndex = 0, Double_t lrange = 0., Double_t rrange = 0. );

       void AddPolynomX(const char* DataName, const char* NamePrefix, TArrayD& Coef, Int_t GroupIndex = 0);

      void AddPolynoms(const char* DataName, const char* NamePrefix, Int_t MaxOrder = 1, Int_t NumAxis = 1, Int_t GroupIndex = 0);

      TGo4FitModelGauss1* AddGauss1(const char* DataName, const char* ModelName, Double_t iPosition, Double_t iWidth, Double_t iAmpl = 1., Int_t Axis = 0);

      Int_t NumModelsAssosiatedTo(const char* DataName);

      TGo4FitModel* RemoveModel(const char * ModelName, Bool_t IsDel);

      void DeleteModelsAssosiatedTo(const char* DataName);

      void DeleteAllModels();

      void AssignModelTo(const char* ModelName, const char* DataName, Double_t RatioValue = 1., Bool_t FixRatio = kFALSE);

      void ClearModelAssignmentTo(const char* ModelName, const char* DataName = 0);

      void ChangeDataNameInAssignments(const char* oldname, const char* newname);

      Int_t GetNumComp() const { return GetNumData()+GetNumModel(); }

      TGo4FitComponent* GetComp(Int_t n) {  if(n<GetNumData()) return GetData(n); else return GetModel(n-GetNumData()); }

      void EstimateAmplitudes(Int_t NumIters = 1);

      void AddAmplEstimation(Int_t NumIters = 1);

      void AddStandardActions();

      Double_t CalculateFitFunction(Double_t* pars = 0, Int_t FitFunctionType = -1, const char* DataName = 0);

      Int_t CalculateNDF(const char* DataName = 0);

      virtual void FillSlotList(TSeqCollection* list);

      virtual void Print(Option_t* option) const;

      Bool_t CalculatesMomentums(const char* DataName, Bool_t UseRanges, Bool_t SubstractModels, Double_t& first, Double_t& second);
      
      Double_t CalculatesIntegral(const char* DataName, const char* ModelName = 0, Bool_t OnlyCounts = kFALSE);
      
      Double_t CalculatesModelIntegral(const char* ModelName, Bool_t OnlyCounts = kFALSE);

      TObject* CreateDrawObject(const char* ObjName, const char* DataName, Bool_t IsModel = kFALSE, const char* ModelName = 0);

      virtual void Draw(Option_t* option);

      void ProvideLastDrawObjects(TObjArray& lst);

      void PrintAmpls() const;

      void PrintLines() const;

      TString FindNextName(const char* Head, Int_t start, Bool_t isModel = kTRUE);

   protected:

      friend class TGo4FitAmplEstimation;

      // a set of function to access data buffers

      Bool_t ModelBuffersAllocated(TGo4FitModel* model);
      Bool_t DataBuffersAllocated(TGo4FitData* data);

      Int_t GetDataBinsSize(TGo4FitData* data);
      Double_t* GetDataBinsValues(TGo4FitData* data);
      Double_t* GetDataBinsDevs(TGo4FitData* data);
      Double_t* GetDataBinsResult(TGo4FitData* data);
      Double_t* GetModelBinsValues(TGo4FitModel* model, const char* DataName);

      void RebuildAll(Bool_t ForceBuild = kFALSE);


      virtual void CollectAllPars();

      virtual Double_t DoCalculation();
      virtual Int_t DoNDFCalculation();
      Double_t CalculateFCN(Int_t FitFunctionType, TGo4FitData* selectdata = 0);
      Double_t PointFitFunction(Int_t FitFunctionType, Double_t value, Double_t modelvalue, Double_t standdev);

      virtual Bool_t InitFitterData();
      virtual void FinalizeFitterData();

      TObjArray fxDatas;

      TObjArray fxModels;

      Int_t fiFitFunctionType;

      Int_t fiMemoryUsage;

   private:

      void MoveDrawObjectsToROOT();
      void CheckSlotsBeforeDelete(TGo4FitComponent* comp);

      TUserFitFunction fxUserFitFunction;    

      TObjArray* fxDrawObjs;                 


      /*#  TGo4FitModel Models; */

      /*#  TGo4FitData Datas; */
   ClassDef(TGo4Fitter,1)
};


#endif // TGO4FITTER_H



//----------------------------END OF GO4 SOURCE FILE ---------------------

---