fix PID reference figures style (AlexW)

[u/mrichter/AliRoot.git] / TRD / qaRec / AliTRDpidChecker.h

diff --git a/TRD/qaRec/AliTRDpidChecker.h b/TRD/qaRec/AliTRDpidChecker.h
index d57290d24e2e7943033950e8e363170c085ccbf4..54981641ec5c5dde868dc08349c225b0372383bd 100644 (file)
--- a/TRD/qaRec/AliTRDpidChecker.h
+++ b/TRD/qaRec/AliTRDpidChecker.h
@@ -6,67 +6,98 @@
 // Task to check PID performance of the TRD
 //
 // Author : Alex Wilk <wilka@uni-muenster.de>
+//          Alex Bercuci <A.Bercuci@gsi.de>
+//          Markus Fasel <M.Fasel@gsi.de>
 //
 ///////////////////////////////////////////////////////
 
-#include "AliPID.h"
-#include "../Cal/AliTRDCalPID.h"
+#ifndef ROOT_TAxis
+#include "TAxis.h"
+#endif
 
 #ifndef ALITRDRECOTASK_H
 #include "AliTRDrecoTask.h"
 #endif
 
-class TObjArray;
-class TList;
-class TClonesArray;
-class TTreeSRedirector;
 class AliTRDReconstructor;
+class AliTRDpidUtil;
 class AliTRDpidChecker : public AliTRDrecoTask 
 {
-
-  enum{
-    kLQlikelihood    = 0                                           // place for 2-dim LQ electron likelihood distributions
-    ,kNNlikelihood = 1 * AliTRDCalPID::kNMom * AliPID::kSPECIES  // place for NN electron likelihood distributions
-    ,kdEdx         = 2 * AliTRDCalPID::kNMom * AliPID::kSPECIES  // place for the dE/dx spectra
-    ,kPH           = 3 * AliTRDCalPID::kNMom * AliPID::kSPECIES  // place for pulse height spectra
-    ,kMomentum     = 4 * AliTRDCalPID::kNMom * AliPID::kSPECIES  // place for the momentum distribution
-    ,kMomentumBin  = kMomentum +1                                // place for the momentum distribution
-    ,kGraphLQ      = kMomentumBin +1                             // place for the 2-dim LQ pion efficiencies
-    ,kGraphLQerr   = kGraphLQ +1                                 // place for the 2-dim LQ pion efficiency errors
-    ,kGraphNN      = kGraphLQerr +1                              // place for the NN pion efficiencies
-    ,kGraphNNerr   = kGraphNN +1                                 // place for the NN pion efficiency errors
-  };
-
+  // Plots registered for this task
   enum{
-    kGraphStart = kGraphLQ
+     kEfficiency     =  0     // pi Efficiency plot
+    ,kdEdx           =  1     // dE/dx spectra
+    ,kdEdxSlice      =  2     // dE/dx spectra
+    ,kPH             =  3     // pulse height spectra
+    ,kNClus          =  4     //  number of clusters per track
+    ,kMomentum       =  5     // momentum distribution
+    ,kMomentumBin    =  6     // momentum distribution
+    ,kThresh         =  7     // threshold in efficiency
   };
-
 public:
   AliTRDpidChecker();
   virtual ~AliTRDpidChecker();
   
-  void    CreateOutputObjects();
-  void    Exec(Option_t *option);
-  void    GetRefFigure(Int_t ifig, Int_t &first, Int_t &last, Option_t *opt);  
-  Bool_t  PostProcess();
-  void    Terminate(Option_t *);
+  virtual void    CreateOutputObjects();
+  virtual Bool_t  GetRefFigure(Int_t ifig);
+  virtual Bool_t  PostProcess();
+  virtual void    Terminate(Option_t *);
 
+  TH1 *PlotLQ(const AliTRDtrackV1 *track = 0x0);
+  TH1 *PlotNN(const AliTRDtrackV1 *track = 0x0);
+  TH1 *PlotESD(const AliTRDtrackV1 *track = 0x0);
+  TH1 *PlotdEdx(const AliTRDtrackV1 *track = 0x0);
+  TH1 *PlotdEdxSlice(const AliTRDtrackV1 *track = 0x0);
+  TH1 *PlotPH(const AliTRDtrackV1 *track = 0x0);
+  TH1 *PlotNClus(const AliTRDtrackV1 *track = 0x0);
+  TH1 *PlotMom(const AliTRDtrackV1 *track = 0x0);
+  TH1 *PlotMomBin(const AliTRDtrackV1 *track = 0x0);
+
+  TObjArray *GetGraphs() { return fGraph; };
+  virtual TObjArray *Histos();
+  void EvaluatePionEfficiency(TObjArray *histoContainer, TObjArray *results, Float_t electron_efficiency);
+  inline void SetMomentumBinning(Int_t nBins, Double_t *bins);
+  inline Int_t FindBin(Int_t species, Double_t momentum);
+  inline Bool_t IsInRange(Double_t momentum);
 
 private:
   AliTRDpidChecker(const AliTRDpidChecker&);               // not implemented
   AliTRDpidChecker& operator=(const AliTRDpidChecker&);    // not implemented
 
-  Double_t GetPionEfficiency(Int_t Index1, Int_t Index2);  // calculates the pion efficiency
-  Double_t GetError(Int_t Index1, Int_t Index2);           // calculates the error
+  Int_t  CalcPDG(AliTRDtrackV1* track = 0x0);
+  Bool_t CheckTrackQuality(const AliTRDtrackV1* track = 0x0);
   
-
   AliTRDReconstructor *fReconstructor;     //! reconstructor needed for recalculation the PID
-
-  enum{
-    kBins = 12001                // binning of the likelihood histograms
-  };
-
+  AliTRDpidUtil       *fUtil;              //! utility class for PID calculations
+  TObjArray           *fGraph;             //! array of graphs filled in PostProcess
+  TObjArray           *fEfficiency;        //! array of histograms with efficiency
+  TAxis               *fMomentumAxis;      //! helper mementum binning
   ClassDef(AliTRDpidChecker, 1); // TRD PID checker
 };
 
+//________________________________________________________________________
+inline void AliTRDpidChecker::SetMomentumBinning(Int_t nBins, Double_t *bins){
+  //
+  // Set the Momentum Bins
+  //
+  if(fMomentumAxis) delete fMomentumAxis;
+  fMomentumAxis = new TAxis(nBins, bins);
+}
+
+//________________________________________________________________________
+inline Int_t AliTRDpidChecker::FindBin(Int_t species, Double_t momentum){
+  //
+  // Find the Bin in the 2D Histogram
+  //
+  return species * fMomentumAxis->GetNbins() + (fMomentumAxis->FindBin(momentum) -1);
+}
+
+//________________________________________________________________________
+inline Bool_t AliTRDpidChecker::IsInRange(Double_t momentum){
+  //
+  // Check Whether momentum is in the defined Range
+  //
+  return (momentum >= fMomentumAxis->GetXmin() && momentum <= fMomentumAxis->GetXmax());
+}
+
 #endif