#ifndef ALICORRECTIONMATRIX_H
#define ALICORRECTIONMATRIX_H

/* $Id$ */

// ------------------------------------------------------
//
// Class to handle corrections.
//
// ------------------------------------------------------
//
// TODO:
//
// - add options in draw method
//

#include <TNamed.h>

class TH1;

class AliCorrectionMatrix : public TNamed
{
protected: // do not create this baseclass
  AliCorrectionMatrix();
  AliCorrectionMatrix(const Char_t* name, const Char_t* title);
  AliCorrectionMatrix(const AliCorrectionMatrix& c);
  virtual ~AliCorrectionMatrix();
  AliCorrectionMatrix& operator=(const AliCorrectionMatrix& corrMatrix);

public:
  virtual void Copy(TObject& c) const;
  virtual Long64_t Merge(TCollection* list);

  TH1* GetGeneratedHistogram() { return fhGene; }
  TH1* GetMeasuredHistogram()  { return fhMeas; }
  TH1* GetCorrectionHistogram() { return fhCorr; }

  void SetGeneratedHistogram(TH1* agene) { fhGene = agene; }
  void SetMeasuredHistogram(TH1* ameas)  { fhMeas = ameas; }
  void SetCorrectionHistogram(TH1* acorr) { fhCorr = acorr; }

  void Divide();
  void Multiply();
  void SetCorrectionToUnity();
  void ResetErrorsOnCorrections();
  void Scale(Double_t factor);

  void Add(AliCorrectionMatrix* aMatrixToAdd, Float_t c=1);

  void SetAxisTitles(const Char_t* titleX="", const Char_t* titleY="", const Char_t* titleZ="");

  virtual Bool_t LoadHistograms(const Char_t* dir = 0);
  virtual void SaveHistograms();

  virtual void DrawHistograms(const Char_t* canvasName = 0);

  virtual void ReduceInformation();

  virtual void Reset(Option_t* option = "");

protected:
  TH1*    fhMeas;  // histogram of measured particles (or tracks)
  TH1*    fhGene;  // histogram of generated particles

  TH1*    fhCorr;  // correction histogram (ratio generated/measured)

  ClassDef(AliCorrectionMatrix,1)
};

#endif