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

//
// This class is a part of a package of high level QA monitoring for TRD.
//
// S. Radomski
// radomski@physi.uni-heidelberg.de
// March 2008
//

#include "AliAnalysisTask.h"  

class TTree; 
class TH1D; 
class TH2D;
class TLorentzVector;

class AliExternalTrackParam;
class AliKFParticle;
class AliESDEvent; 
class AliESDtrack;

class AliTRDqaJPsi : public AliAnalysisTask {

 public:

  AliTRDqaJPsi();
  AliTRDqaJPsi(const char *name);
  AliTRDqaJPsi(const AliTRDqaJPsi &trd);
  AliTRDqaJPsi &operator=(const AliTRDqaJPsi & /*g*/) { return *this; };
  virtual ~AliTRDqaJPsi() {}
   
  virtual void Exec(Option_t * opt = "");
  virtual void ConnectInputData(Option_t *);
  virtual void CreateOutputObjects();
  virtual void Terminate(Option_t * opt = "");

 private:
 
  TTree        * fChain;             //!pointer to the analyzed TTree or TChain
  AliESDEvent  * fESD;               //! Declaration of leave types

  TObjArray * fOutputContainer; //! output data container
  
  // histograms

  static const Int_t fgknSteps = 5;   // number of analysis steps (arbitrary)

  TH1D *fStatus[fgknSteps];      // track status
  TH1D *fnTracks[2*fgknSteps];   // number of tracks
  TH1D *fPt[2*fgknSteps];        // transverse momentum
  TH1D *fPID[2*fgknSteps];       // PID LQ
  TH1D *fAngleSM[fgknSteps];     // difference in SM ID
  
  //TH2D *fnGoodTracks;          // correlation of the final number of Pos and Neg tracks
  TH1D *fInvMass[fgknSteps];     // invariant mass using different cuts
  TH1D *fInvMassVec[fgknSteps];  // invariant mass
  TH1D *fInvMassDiff[fgknSteps]; // invariant mass difference

  TH2D *fPtAngle[fgknSteps];     // pt angle 

  // tracks
  AliKFParticle *fTracks[1000];  // tracks
  TLorentzVector *fVec[1000];    // Lorentz vector for the tracks
  Int_t fInSample[1000][fgknSteps]; // in sample?
  Int_t fSM[1000];                  // TRD sector
  Int_t fnKFtracks;                 //[2];          
  
  // helper functions
  void FillHist(AliESDtrack *track, Int_t step);
  TLorentzVector *CreateVector(AliESDtrack *track);


  ClassDef(AliTRDqaJPsi, 0); // a TRD analysis task 
};
#endif // ALITRDQAJPSI_H