minor bug fix (A. Colla)

[u/mrichter/AliRoot.git] / HBTAN / AliHBTCorrFitFctn.h

diff --git a/HBTAN/AliHBTCorrFitFctn.h b/HBTAN/AliHBTCorrFitFctn.h
index f15eb73442b57e35a0a0d9669fc223d2928150ba..6ab6a7b9390a53a097886b92032ed0c492e02047 100644 (file)
--- a/HBTAN/AliHBTCorrFitFctn.h
+++ b/HBTAN/AliHBTCorrFitFctn.h
@@ -1,5 +1,6 @@
 #ifndef ALIHBTCORRFITFCTN_H
 #define ALIHBTCORRFITFCTN_H
+
 //____________________________________________________________
 ///////////////////////////////////////////////////////////////
 //                                                           //
@@ -11,24 +12,36 @@
 #include "AliHBTFunction.h"
 #include <TNtuple.h>
 
-class AliHBTCorrFitFctn: public AliHBTTwoPairFctn1D
+class AliHBTCorrFitFctn: public AliHBTOnePairFctn1D
 {
 //Q Invaraint Correlation Function
 //It writes Ntuple that is input for CorrFit
  public:
-   AliHBTCorrFitFctn(Int_t nbins = 300, Double_t maxXval = 0.3, Double_t minXval = 0.0);
+   AliHBTCorrFitFctn();
+   AliHBTCorrFitFctn(Int_t  fit, Int_t  norm);
    virtual ~AliHBTCorrFitFctn(){delete fNtuple;}
-   void ProcessSameEventParticles(AliHBTPair* trackpair, AliHBTPair* partpair);
-   void ProcessDiffEventParticles(AliHBTPair* trackpair, AliHBTPair* partpair);
+   void ProcessSameEventParticles(AliHBTPair* trackpair);
+   void ProcessDiffEventParticles(AliHBTPair* trackpair);
    
    TH1* GetResult();
-   void WriteFunction();
+   virtual Int_t WriteFunction();
+
+   void SetMaxNumberOfPairs(Int_t  fit, Int_t  norm){fNMaxPairsFitArea = fit; fNMaxPairsNormArea = norm;}
+   void SetFitRange(Float_t max) {fFitRangeMax = max;}
+   void SetNormalizationRange(Float_t min, Float_t max) { fNormRangeMin = min; fNormRangeMax= max;}
  protected:
-   Double_t GetValue(AliHBTPair* /*trackpair*/, AliHBTPair* /*partpair*/) const {return 0.0;}//not usable
+   Double_t GetValue(AliHBTPair* /*trackpair*/) const {return 0.0;}//not usable
    
    TNtuple* fNtuple;//ntuple for storig pairs
-   Int_t    fNPairsFitArea;//number of pairs in fitting area
+
+   Int_t    fNPairsFitArea;//current number of pairs in fitting area
+   Int_t    fNMaxPairsFitArea;//current number of pairs in fitting area
+   Float_t  fFitRangeMax;
+   
    Int_t    fNPairsNormArea;//number of pairs in normalization area
+   Int_t    fNMaxPairsNormArea;//number of pairs in normalization area
+   Float_t  fNormRangeMin;
+   Float_t  fNormRangeMax;
   public:
    ClassDef(AliHBTCorrFitFctn,1)
 };