New macro for drawing slat motifs with real contours

[u/mrichter/AliRoot.git] / TRD / AliTRDprobdist.h

diff --git a/TRD/AliTRDprobdist.h b/TRD/AliTRDprobdist.h
index 4333367b393936e73213ea3d2fdcfb18df4b3fb9..f6bf95d39966c2aa3ae47e3562d757c1e2444155 100644 (file)
--- a/TRD/AliTRDprobdist.h
+++ b/TRD/AliTRDprobdist.h
@@ -10,60 +10,58 @@
    Prashant Shukla <shukla@pi0.physi.uni-heidelberg.de>
    -----------------------------------------------------------------*/
 
-#include "TObject.h"
+#include <TNamed.h>
+#include <TFile.h>
+#include <TH1F.h>
 
-const Int_t kNo_Mom=7; 
-const Int_t kNo_EnBins=200;
-const Int_t kNo_TBins=20;
+const Int_t kNMom=11; 
 
-class AliTRDprobdist : public TObject {
+class AliTRDprobdist : public TNamed {
   public:
-    AliTRDprobdist(Int_t multiplicity=1);   // multiplicity can take 
-                         // values 1, 2000, 4000, 6000, 8000
-    //    ~AliTRDprobdist(); 
-    void FillData();      // Fills Data 
-    //    void FillData2000();  // Fills Data of multiplicity 2000
-    //    void FillData4000();  // Fills Data of multiplicity 4000
-    //    void FillData6000();  // Fills Data of multiplicity 6000
-    //    void FillData8000();  // Fills Data of multiplicity 8000
-    Double_t GetBM(Int_t ip) const { return fTrackMomentum[ip];}
-    Double_t GetMeanPI(Int_t ip) const;  // Gets mean of de/dx dist. of pi
-    Double_t GetMeanEL(Int_t ip) const;  // Gets mean of de/dx dist. of e
-    Double_t GetNormalizationPI(Int_t ip) const;  // Gets Norm. of de/dx dist. of pi
-    Double_t GetNormalizationEL(Int_t ip) const;  // Gets Norm. of de/dx dist. of e
-    Double_t GetProbability(Int_t k, Double_t mom, Double_t dedx) const;
-                          // Gets the Probability of having dedx
-    Double_t GetProbabilityT(Int_t k, Double_t mom, Int_t timbin) const;
-                        // Gets the Probability of having timbin
-    void GetData(Int_t ip, Double_t *ebin, Double_t *ppi, Double_t *pel) const { 
-      for(Int_t ie=0; ie<fNEbins; ie++){
-        ebin[ie]=fEnergyLoss[ip][ie];
-        ppi[ie]=fProbPI[ip][ie];
-        pel[ie]=fProbEL[ip][ie];
-      }
-    }
-    void GetDataT(Int_t ip, Int_t *tbin, Double_t *ppi, Double_t *pel) const { 
-      for(Int_t it=0; it<fNTbins; it++){
-        tbin[it]=fTimBin[ip][it];
-        ppi[it]=fProbPIT[ip][it];
-        pel[it]=fProbELT[ip][it];
-      }
-    }
+    AliTRDprobdist(Char_t *responseFile="$ALICE_ROOT/TRD/TRDdEdxHistogramsV1.root");
+                       // Main Constructor
+    AliTRDprobdist(const AliTRDprobdist& pd);  // Copy Constructor
+    virtual ~AliTRDprobdist();               // Destructor
+
+    Double_t GetADCNorm() const {return fADCNorm;}  // Get ADC Normalization
+
+    Double_t GetMomentum(Int_t ip) const {return fTrackMomentum[ip];}
+                      // Gets the momentum for given histogram number ip
+    Double_t GetMean(Int_t iType, Int_t ip) const;        
+                      // Particle type is iType and histogram number is ip         
+    Double_t GetNormalization(Int_t iType, Int_t ip) const;
+
+    TH1F* GetHistogram(Int_t iType, Int_t ip) const;
+
+    Double_t GetProbability(Int_t iType, Double_t mom, Double_t dedx) const;
+                      // Gets the Probability of having dedx
+    Double_t GetProbabilityT(Int_t iType, Double_t mom, Int_t timbin) const;
+                      // Gets the Probability of having timbin
+
+    Int_t GetNbins() const {return fNbins;}         // Number of Energy bins
+    Double_t GetBinSize() const {return fBinSize;}  // Size of Energy bin
+
+    Bool_t UpdateData(Char_t *responseFile) {return ReadData(responseFile);}
+                      // Update the histograms from responseFile
+    void SetADCNorm(Double_t norm){fADCNorm=norm;}  // Set ADC Normalization
+
   protected:
-    Double_t fADCNorm;               // Ratio of de/dx from Det. to prob. dist.
+    Bool_t ReadData(Char_t *responseFile);      // Reads Data
+    Char_t *fpartName[5];             // Names of particle species
+    Double_t fADCNorm;          // Ratio of mean charge from real Det. to prob. dist.
     Int_t fNMom;                     // Number of momenta  
-    Double_t fTrackMomentum[kNo_Mom];           // Track Momentum 
-    Int_t fNEbins;                              // Number of Energy bins
-    Double_t fEnBinSize;                        // Size of Energy bin
-    Double_t fEnergyLoss[kNo_Mom][kNo_EnBins];  // dE/dx 
-    Double_t fProbPI[kNo_Mom][kNo_EnBins];      // Prob. of dEdx  for pi
-    Double_t fProbEL[kNo_Mom][kNo_EnBins];      // Prob. of dEdx  for e
+    Double_t fTrackMomentum[kNMom];          // Track Momentum 
+
+    Int_t fNbins;                   // Number of Energy bins
+    Double_t fBinSize;              // Size of Energy bin
 
-    Int_t fNTbins;                             // Number of Tim bins=20      
-    Double_t fTBinSize;                        // Size of Time Bin =1 
-    Int_t fTimBin[kNo_Mom][kNo_TBins];      // Time Bin  
-    Double_t fProbPIT[kNo_Mom][kNo_TBins];      // Prob. of dEdx for pi
-    Double_t fProbELT[kNo_Mom][kNo_TBins];    // Prob. of dEdx for e
+    TH1F *fh1dEdxEL[kNMom];      // Prob. of dEdx  for e
+    TH1F *fh1dEdxPI[kNMom];      // Prob. of dEdx  for pi
+    TH1F *fh1dEdxMU[kNMom];      // Prob. of dEdx  for muon
+    TH1F *fh1dEdxKA[kNMom];      // Prob. of dEdx  for Kaon
+    TH1F *fh1dEdxPR[kNMom];      // Prob. of dEdx  for proton
+    TH1F *fh1MaxTimBinEL[kNMom];    // Prob. of max Time Bin for e
+    TH1F *fh1MaxTimBinPI[kNMom];    // Prob. of max Time Bin for pi
     ClassDef(AliTRDprobdist,1)
 };