#ifndef ALITRDRESOLUTION_H
#define ALITRDRESOLUTION_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id: AliTRDresolution.h 27496 2008-07-22 08:35:45Z cblume $ */

////////////////////////////////////////////////////////////////////////////
//                                                                        //
//  TRD Resolution performance                                            //
//                                                                        //
//  Authors:                                                              //
//    Alexandru Bercuci <A.Bercuci@gsi.de>                                //
//    Markus Fasel <M.Fasel@gsi.de>                                       //
//                                                                        //
////////////////////////////////////////////////////////////////////////////

#ifndef ALITRDRECOTASK_H
#include "AliTRDrecoTask.h"
#endif

class TH1;
class TH2;
class TF1;
class TGraphErrors;
class TObjArray;
class TDatabasePDG;
class AliTRDReconstructor;
class AliTRDgeometry;
class AliTRDrecoParam;
class AliTRDseedV1;
class AliTRDtrackInfo;
class AliTRDresolution : public AliTRDrecoTask
{
public:
  enum ETRDresolutionPlot {
     kCharge     =  0 // charge resolution
    ,kCluster    =  1 // cluster - track
    ,kTrack      =  2 // tracklet - track residuals/pulls
    ,kTrackIn    =  3 // tracklet - track residuals/pulls at lower TRD entrance 
    ,kTrackOut   =  4 // tracklet - track residuals/pulls at lower TRD entrance during refit
    ,kMCcluster  =  5 // cluster-mc resolution/pulls
    ,kMCtracklet =  6 // tracklet-mc resolution/pulls
    ,kMCtrackIn  =  7 // TPC track monitor
    ,kMCtrackOut =  8 // TOF/HMPID track monitor
    ,kMCtrack    =  9 // TRD track monitor
    ,kNviews     = 10 // total number of resolution views
    ,kNprojs     = 70 // total number of projections for all views
  };
  enum ETRDresolutionSteer {
    kVerbose  = 0
    ,kVisual  = 1
  };
  enum ETRDresolutionOutSlots {
     kClToTrk    = 2
    ,kTrkltToTrk = 3
    ,kClToMC     = 4
    ,kTrkltToMC  = 5
    ,kNOutSlots  = 4
  };

  AliTRDresolution();
  AliTRDresolution(char* name);
  virtual ~AliTRDresolution();
  
  void    UserCreateOutputObjects();
  static Float_t GetPtThreshold() {return fgPtThreshold;}
  Bool_t  GetRefFigure(Int_t ifig);
  TObjArray*  Histos(); 
  TObjArray*  Results(Int_t i=0) const {return i ? fGraphS : fGraphM;} 
  void    UserExec(Option_t * opt);
  Bool_t  IsVerbose() const {return TESTBIT(fStatus, kVerbose);}
  Bool_t  IsVisual() const {return TESTBIT(fStatus, kVisual);}
  Bool_t  PostProcess();

  TH1*    PlotCharge(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotCluster(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotTracklet(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotTrackIn(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotTrackOut(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotMC(const AliTRDtrackV1 *t=NULL);

  static void SetPtThreshold(Float_t pt) {fgPtThreshold = pt;}
  void    SetRecoParam(AliTRDrecoParam *r);
  void    SetVerbose(Bool_t v = kTRUE) {v ? SETBIT(fStatus ,kVerbose): CLRBIT(fStatus ,kVerbose);}
  void    SetVisual(Bool_t v = kTRUE) {v ? SETBIT(fStatus, kVisual) : CLRBIT(fStatus, kVisual);}

  void    Terminate(Option_t * opt);
  Bool_t  GetGraph(Float_t *bb, ETRDresolutionPlot ip, Int_t idx=-1, Bool_t kLEG=kTRUE, const Char_t *explain=NULL);
  Bool_t  GetGraphArray(Float_t *bb, ETRDresolutionPlot ip, Int_t idx, Bool_t kLEG=kTRUE, Int_t n=0, Int_t *sel=NULL, const Char_t *explain=NULL);
  
private:
  AliTRDresolution(const AliTRDresolution&);
  AliTRDresolution& operator=(const AliTRDresolution&);

  void    AdjustF1(TH1 *h, TF1 *f);
  TObjArray*  BuildMonitorContainerCluster(const char* name);
  TObjArray*  BuildMonitorContainerTracklet(const char* name);
  TObjArray*  BuildMonitorContainerTrack(const char* name);
  void    GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
  Bool_t  Process(TH2* const h2, TF1 *f, Float_t k, TGraphErrors **g);
  Bool_t  Process2D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1., Int_t gidx=-1);
  Bool_t  Process2Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Process3D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Process3DL(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Process3Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt);

  UChar_t             fStatus;          // steer parameter of the task
  UShort_t            fIdxPlot;         //! plot counter (internal)
  UShort_t            fIdxFrame;        //! frame counter (internal)
  static Char_t const *fgPerformanceName[kNviews]; // name of performance plot
  static UChar_t const fgNproj[kNviews]; // number of projections per task
  static UChar_t const fgNcomp[kNprojs]; // number of projections per task
  static Char_t const *fgAxTitle[kNprojs][4]; // Title for all ref histos
  static Float_t      fgPtThreshold;     //! pt threshold for some performance plots
  static const Int_t  fgkNresYsegm;      //! number of segments for saving y resolution
  static const Char_t *fgkResYsegmName;  //! name of segment for saving y resolution
  AliTRDReconstructor *fReconstructor;  //! local reconstructor
  AliTRDgeometry      *fGeo;            //! TRD geometry
  TDatabasePDG        *fDBPDG;          //! PDG database
  TObjArray           *fGraphS;         //! result holder - sigma values
  TObjArray           *fGraphM;         //! result holder - mean values

  // calibration containers
  TObjArray           *fCl;     //! cluster2track calib
  TObjArray           *fTrklt;  //! tracklet2track calib
  TObjArray           *fMCcl;   //! cluster2mc calib
  TObjArray           *fMCtrklt;//! tracklet2mc calib
  
  ClassDef(AliTRDresolution, 5) // TRD tracking resolution task
};
#endif