[u/mrichter/AliRoot.git] / HMPID / AliAnalysisTaskJetsHMPID.h

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

//=================================================================================
// AliAnalysysTaskJetsHMPID - Class performing PID analysis in jets with the HMPID
// A set of histograms is created.
//=================================================================================

#ifndef AliAnalysisTaskJetsHMPID_H
#define AliAnalysisTaskJetsHMPID_H

#include <TList.h>
#include <TH2F.h>
#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskJetsHMPID : public AliAnalysisTaskSE
{
  public:
    AliAnalysisTaskJetsHMPID();
    AliAnalysisTaskJetsHMPID(const Char_t* name);
    AliAnalysisTaskJetsHMPID& operator= (const AliAnalysisTaskJetsHMPID& c);
    AliAnalysisTaskJetsHMPID(const AliAnalysisTaskJetsHMPID& c);
    virtual ~AliAnalysisTaskJetsHMPID();

    virtual void   ConnectInputData(Option_t *);
    virtual void   UserCreateOutputObjects();
    virtual void   UserExec(Option_t *option);
    virtual void   Terminate(Option_t *);

    void SetJetBranch(const Char_t *br){ fJetBranch=br; }
    void SetBkgBranch(const Char_t *br){ fBkgBranch=br; }
    void SetJetPtCut(Float_t ptcut)    { fJetPtCut=ptcut; }

 protected:

 private:
  TString       fJetBranch;          // jet branch to read
  TString       fBkgBranch;          // background branch to read
  Float_t       fJetPtCut;           // jet pT threshold

  AliAODEvent   *fAOD;               // AOD object
  TList         *fHistList;          // list of histograms

  TH2F          *fThetaChJet;        // theta Cherenkov distribution in the jets
  TH2F          *fThetaChBkg;        // theta Cherenkov distribution out of the jets
  TH2F          *fThetaChRndCone;    // theta Cherenkov distribution in random cone from background
  TH1F          *fEvSelDPhi;         // Delta phi jet-HMPID in different events selected
  TH1F          *fJetsPt;            // pT of jets into the HMPID
  TH1F          *fRndConePt;         // pT of random cones into the HMPID
  TH1F          *fAwayJetPt;         // pT of jets on the away side
  TH2F          *fJetsEtaPhi;        // eta and phi of jets into the HMPID
  TH2F          *fTrksEtaPhiJet;     // eta and phi of jet tracks into the HMPID
  TH2F          *fTrksEtaPhiBkg;     // eta and phi of bkg tracks into the HMPID

  TTree         *fTree;              // tree with useful data for subsequent analysis
  Float_t        fTrackPt;           // track pt
  Float_t        fJetPt;             // jet pt
  Float_t        fPionBkg;           // pions probability out of the jets
  Float_t        fKaonBkg;           // kaons probability out of the jets
  Float_t        fProtBkg;           // prots probability out of the jets
  Float_t        fPionJet;           // pions probability in the jets
  Float_t        fKaonJet;           // kaons probability in the jets
  Float_t        fProtJet;           // prots probability in the jets

  ClassDef(AliAnalysisTaskJetsHMPID, 1);
};

#endif
Commit	Line	Data
bfcb155c	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	//=================================================================================
	17	// AliAnalysysTaskJetsHMPID - Class performing PID analysis in jets with the HMPID
	18	// A set of histograms is created.
	19	//=================================================================================
	20
	21	#ifndef AliAnalysisTaskJetsHMPID_H
	22	#define AliAnalysisTaskJetsHMPID_H
	23
	24	#include <TList.h>
	25	#include <TH2F.h>
	26	#include "AliAnalysisTaskSE.h"
	27
	28	class AliAnalysisTaskJetsHMPID : public AliAnalysisTaskSE
	29	{
	30	public:
	31	AliAnalysisTaskJetsHMPID();
	32	AliAnalysisTaskJetsHMPID(const Char_t* name);
	33	AliAnalysisTaskJetsHMPID& operator= (const AliAnalysisTaskJetsHMPID& c);
	34	AliAnalysisTaskJetsHMPID(const AliAnalysisTaskJetsHMPID& c);
	35	virtual ~AliAnalysisTaskJetsHMPID();
	36
	37	virtual void ConnectInputData(Option_t *);
	38	virtual void UserCreateOutputObjects();
	39	virtual void UserExec(Option_t *option);
	40	virtual void Terminate(Option_t *);
	41
	42	void SetJetBranch(const Char_t *br){ fJetBranch=br; }
	43	void SetBkgBranch(const Char_t *br){ fBkgBranch=br; }
	44	void SetJetPtCut(Float_t ptcut) { fJetPtCut=ptcut; }
	45
	46	protected:
	47
	48	private:
	49	TString fJetBranch; // jet branch to read
	50	TString fBkgBranch; // background branch to read
	51	Float_t fJetPtCut; // jet pT threshold
	52
	53	AliAODEvent *fAOD; // AOD object
	54	TList *fHistList; // list of histograms
	55
	56	TH2F *fThetaChJet; // theta Cherenkov distribution in the jets
	57	TH2F *fThetaChBkg; // theta Cherenkov distribution out of the jets
	58	TH2F *fThetaChRndCone; // theta Cherenkov distribution in random cone from background
	59	TH1F *fEvSelDPhi; // Delta phi jet-HMPID in different events selected
	60	TH1F *fJetsPt; // pT of jets into the HMPID
	61	TH1F *fRndConePt; // pT of random cones into the HMPID
	62	TH1F *fAwayJetPt; // pT of jets on the away side
	63	TH2F *fJetsEtaPhi; // eta and phi of jets into the HMPID
	64	TH2F *fTrksEtaPhiJet; // eta and phi of jet tracks into the HMPID
65	TH2F *fTrksEtaPhiBkg; // eta and phi of bkg tracks into the HMPID
66
67	TTree *fTree; // tree with useful data for subsequent analysis
68	Float_t fTrackPt; // track pt
69	Float_t fJetPt; // jet pt
70	Float_t fPionBkg; // pions probability out of the jets
71	Float_t fKaonBkg; // kaons probability out of the jets
72	Float_t fProtBkg; // prots probability out of the jets
73	Float_t fPionJet; // pions probability in the jets
74	Float_t fKaonJet; // kaons probability in the jets
75	Float_t fProtJet; // prots probability in the jets
76
77	ClassDef(AliAnalysisTaskJetsHMPID, 1);
78	};
79
80	#endif