PWG0/AliCorrectionMatrix.h

   1 #ifndef ALICORRECTIONMATRIX_H
   2 #define ALICORRECTIONMATRIX_H
   3
   4 /* $Id$ */
   5
   6 // ------------------------------------------------------
   7 //
   8 // Class to handle corrections.
   9 //
  10 // ------------------------------------------------------
  11 //
  12 // TODO:
  13 //
  14 // - add options in draw method
  15 //
  16
  17 #include <TNamed.h>
  18
  19 class TH1;
  20
  21 class AliCorrectionMatrix : public TNamed
  22 {
  23 protected: // do not create this baseclass
  24   AliCorrectionMatrix();
  25   AliCorrectionMatrix(const Char_t* name, const Char_t* title);
  26   AliCorrectionMatrix(const AliCorrectionMatrix& c);
  27   virtual ~AliCorrectionMatrix();
  28
  29 public:
  30   AliCorrectionMatrix& operator=(const AliCorrectionMatrix& corrMatrix);
  31   virtual void Copy(TObject& c) const;
  32   virtual Long64_t Merge(TCollection* list);
  33
  34   TH1* GetGeneratedHistogram() { return fhGene; }
  35   TH1* GetMeasuredHistogram()  { return fhMeas; }
  36
  37   void SetGeneratedHistogram(TH1* agene) { fhGene = agene; }
  38   void SetMeasuredHistogram(TH1* ameas)  { fhMeas = ameas; }
  39
  40   void Divide();
  41
  42   void SetAxisTitles(const Char_t* titleX="", const Char_t* titleY="", const Char_t* titleZ="");
  43
  44   virtual Bool_t LoadHistograms(Char_t* fileName, Char_t* dir = ".");
  45   virtual void SaveHistograms();
  46
  47   virtual void DrawHistograms();
  48
  49 protected:
  50   TH1*    fhMeas;  // histogram of measured particles (or tracks)
  51   TH1*    fhGene;  // histogram of generated particles
  52
  53   TH1*    fhCorr;  // correction histogram (ratio generated/measured)
  54
  55   ClassDef(AliCorrectionMatrix,1)
  56 };
  57
  58 #endif
  59