//
// Header file for implementation of data analysis aft 900 GeV
//
// Author: A. Pulvirenti
//

#ifndef ALIRSNANALYSISPHI900GEV_H
#define ALIRSNANALYSISPHI900GEV_H

#include "AliAnalysisTaskSE.h"
#include "AliRsnTOFT0maker.h"

class TH1I;
class TTree;

class AliStack;
class AliESDEvent;
class AliESDVertex;

class AliRsnAnalysisPhi900GeV : public AliAnalysisTaskSE
{
  public:

    AliRsnAnalysisPhi900GeV(const char *name = "Phi900GeV");
    AliRsnAnalysisPhi900GeV(const AliRsnAnalysisPhi900GeV& copy);
    AliRsnAnalysisPhi900GeV& operator=(const AliRsnAnalysisPhi900GeV& copy);
    virtual ~AliRsnAnalysisPhi900GeV();

    void            SetUseMC(Bool_t yn = kTRUE) {fUseMC = yn;}
    void            SetMaxDCAr(Double_t v) {fDCAr = v;}
    void            SetMaxDCAz(Double_t v) {fDCAz = v;}
    void            SetMaxChi2(Double_t v) {fChi2 = v;}
    void            SetMinNTPC(Int_t    n) {fNTPC = n;}
    void            SetTPCparams(Bool_t isMC);
    void            SetTPCrange(Double_t min, Double_t max) {fMinTPC = min; fMaxTPC = max;}
    void            SetTPCpar(Double_t p0, Double_t p1, Double_t p2, Double_t p3, Double_t p4)
                    {fTPCpar[0]=p0;fTPCpar[1]=p1;fTPCpar[2]=p2;fTPCpar[3]=p3;fTPCpar[4]=p4;}

    void           SetTOFESD(Bool_t yn = kTRUE) {fTOFESD = yn;}
    void           SetTOFSigma(Double_t v) {fTOFSigma = v;}
    void           SetTOFSettings(AliRsnTOFT0maker::ESettings set) {fTOFSettings = set;}

    virtual void    UserCreateOutputObjects();
    virtual void    UserExec(Option_t *option = "");
    virtual void    Terminate(Option_t *option = "");

  private:

    void     ProcessESD(AliESDEvent *esd, const AliESDVertex *v, Double_t time0, AliStack *stack);
    void     ProcessMC(AliStack *stack);
    Double_t AlephBB(Double_t p, Double_t mass = 0.493677) const;
    Double_t RemakeTOFtimeMC(AliESDEvent *& esd);

    Bool_t   fUseMC;  // use MC or data?
    
    Short_t  fPDG;    // PDG code
    Float_t  fIM;     // inv mass
    Float_t  fPt;     // transv momentum
    Float_t  fY;      // rapidity
    Float_t  fEta;    // pseud-o-rapidity
    
    Double_t fDCAr;   // transverse DCA
    Double_t fDCAz;   // longitudinal DCA
    Double_t fChi2;   // track chi2
    Int_t    fNTPC;   // number of TPC clusters

    Double_t fTPCpar[5];  // parameters for TPC bethe-Bloch
    Double_t fMinTPC;     // range for TPC de/dx band - min
    Double_t fMaxTPC;     // range for TPC de/dx band - max

    TTree   *fOutTree[2]; // output tree
    TList   *fOutList;    // list for output event counts
    TH1I    *fHEvents;    // histogram of event types

    Bool_t                       fTOFESD;              //  TOF flag to check if ESD data should be used
    Double_t                     fTOFSigma;            //  TOF default resolution
    AliRsnTOFT0maker            *fTOFmaker;            //! TOF time0 computator
    AliRsnTOFT0maker::ESettings  fTOFSettings;         //  TOF settings

    // ROOT dictionary
    ClassDef(AliRsnAnalysisPhi900GeV,1)
};

#endif