[u/mrichter/AliRoot.git] / EMCAL / AliEMCALJetFinder.h

#ifndef ALIEMCALJETFINDER_H
#define ALIEMCALJETFINDER_H

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

/* $Id$ */
//*-- Author:
//*-- Andreas Morsch (CERN)

#include <TTask.h>
#include "AliEMCALJet.h"

class TClonesArray;
class TH2F;
class TH1F;
class TCanvas;
class TList;
class AliEMCALHadronCorrection;

class AliEMCALJetFinder : public TTask {
 public:
    AliEMCALJetFinder();
    AliEMCALJetFinder(const char* name, const char *title);
    virtual ~AliEMCALJetFinder();
    virtual void  Init();
    virtual void  Find(Int_t ncell, Int_t ncell_tot, Float_t etc[30000], 
		      Float_t etac[30000], Float_t phic[30000],
		      Float_t min_move, Float_t max_move, Int_t mode,
		      Float_t prec_bg, Int_t ierror);
    virtual void  Find();
    virtual void  FindChargedJet();
    virtual void  FindTracksInJetCone();
    virtual void  Test();
    virtual void  BuildTrackFlagTable();
    virtual Int_t SetTrackFlag(Float_t radius, Int_t pdgCode, Double_t charge);
    // Geometry
    virtual void SetCellSize(Float_t eta, Float_t phi);
    // Parameters
    virtual void SetDebug(Int_t flag = 0) {fDebug = flag;}
    virtual void SetConeRadius(Float_t par);
    virtual void SetEtSeed(Float_t par);
    virtual void SetMinJetEt(Float_t par);
    virtual void SetMinCellEt(Float_t par);
    virtual void SetPtCut(Float_t par = 1.);
    virtual void SetMomentumSmearing(Bool_t flag = kFALSE) {fSmear = flag;}
    virtual void SetEfficiencySim(Bool_t flag = kFALSE)    {fEffic = flag;}
    virtual void SetSamplingFraction(Float_t par) {fSamplingF = par;}
    virtual void SetEnergyWeightingFlag(Bool_t flag) {fWeightingMethod = flag;}
    virtual void SetEMCALWeight(Float_t par) {fEMCALWeight = par;}
    virtual void SetTrackWeight(Float_t par) {fTrackWeight = par;}
    virtual void SetIncludeK0andN(Bool_t flag = kFALSE) {fK0N = flag;}
    // Correction of hadronic energy
    virtual void SetHadronCorrector(AliEMCALHadronCorrection* corr)
	{fHadronCorrector = corr;}
    virtual void SetHadronCorrection(Int_t flag = 1) {fHCorrection = flag;}
    // PAI
    void SetWriteKey(Bool_t flag = kFALSE) {fWrite = flag;}
    void SetMode(Int_t mode = 0) {fMode = mode;}
    void SetMinMove(Float_t minMove = 0.05) {fMinMove = minMove;}
    void SetMaxMove(Float_t maxMove = 0.15) {fMaxMove = maxMove;}
    void SetPrecBg (Float_t precBg = 0.035) {fPrecBg = precBg;}
    void SetParametersForBgSubtraction
    (Int_t mode=0, Float_t minMove=0.05, Float_t maxMove=0.15, Float_t precBg=0.035);
    //    virtual void Print(Option_t* option="") const;    // *MENU*

    Bool_t GetWriteKey() {return fWrite;}
  //AliEMCALJet* GetJetT() {return fJetT[0];}
    AliEMCALJet* GetJetT(Int_t n = 0) {return fJetT[n];}
    virtual void DrawHistsForTuning(Int_t mode=0);           // *MENU*
    virtual void PrintParameters(Int_t mode=0);              // *MENU*
    virtual const Char_t* GetFileNameForParameters(Char_t* dir="RES/");

    // Access to Results
    virtual Int_t   Njets();
    virtual Float_t JetEnergy(Int_t);
    virtual Float_t JetPhiL(Int_t);
    virtual Float_t JetPhiW(Int_t);
    virtual Float_t JetEtaL(Int_t);  
    virtual Float_t JetEtaW(Int_t);
    TH2F*   GetLego()  {return fLego;}
    TH2F*   GetLegoB() {return fLegoB;}
    TH2F*   GetLegoEMCAL() {return fhLegoEMCAL;}
    TH2F*   GetLegoTracks() {return fhLegoTracks;}
    TH2F*   GethEff() {return fhEff;}
    TH1F*   GetCellEt() {return fhCellEt;}
    TH1F*   GetCellEMCALEt() {return fhCellEMCALEt;}
    TH1F*   GetTrackPt() {return fhTrackPt;}
    TH1F*   GetTrackPtBcut() {return fhTrackPtBcut;}
    TList*  GetHistsList() {return fHistsList;}
    Int_t   GetNChTpc() {return fNChTpc;}
    Bool_t GetEnergyWeightingFlag() {return fWeightingMethod ;}
    Float_t GetEMCALWeight() {return fEMCALWeight;}
    Float_t GetTrackWeight() {return fTrackWeight;}
    void    DrawLego(Char_t *opt="lego");         // *MENU*
    void    DrawLegoEMCAL(Char_t *opt="lego");    // *MENU*
    void    DrawLegos();                          // *MENU*
    Bool_t  IsThisPartonsOrDiQuark(Int_t pdg);
    // I/O
    virtual void SetOutputFileName(char* name) {fOutFileName = name;}
    virtual void FillFromHits(Int_t flag = 0);
    virtual void FillFromHitFlaggedTracks(Int_t flag = 0);
    virtual void FillFromDigits(Int_t flag = 0);
    virtual void FillFromTracks(Int_t flag = 0, Int_t ich = 0);
    virtual void FillFromParticles();
    virtual void FillFromPartons();

    virtual void SaveBackgroundEvent();
    virtual void InitFromBackground();
    virtual void AddJet(const AliEMCALJet& jet);
    virtual void WriteJets();
    virtual void ResetJets();
    virtual TClonesArray* Jets() {return fJets;}
 private:
    virtual void BookLego();
    Float_t WeightedJetEnergy(Float_t eta, Float_t phi);
    Float_t EMCALConeEnergy(Float_t eta, Float_t phi);
    Float_t TrackConeEnergy(Float_t eta, Float_t phi);
    Float_t HCConeEnergy(Float_t eta, Float_t phi);
    virtual void DumpLego();
    virtual void ResetMap();
    virtual Float_t PropagatePhi(Float_t pt, Float_t charge, Bool_t& curls);
    virtual void RearrangeParticlesMemory(Int_t npart);
 protected:
    Bool_t                         fWrite;           // Key for writing
    Bool_t                         fWeightingMethod; // Key for writing
    TClonesArray*                  fJets;            //! List of Jets
    TH2F*                          fLego;            //! Lego Histo
    TH2F*                          fLegoB;           //! Lego Histo Backg
    TH2F*                          fhLegoTracks;     //! Lego for Tracks
    TH2F*                          fhLegoEMCAL;      //! Lego for EMCAL itself
    TH2F*                          fhLegoHadrCorr;   //! Lego for hadron correction
    TH2F*                          fhEff;            //! Hist. for controling eff.
    TH1F*                          fhCellEt;         //! Et distr. for cells from fLego
    TH1F*                          fhCellEMCALEt;    //! Et distr. for cells from fLegoEMCAL
    TH1F*                          fhTrackPt;        //! Pt distr. for charge particles
    TH1F*                          fhTrackPtBcut;    //! Pt distr. for charge particles + cut due to magnetic field
    TH1F*                          fhChPartMultInTpc;//! Ch. part. multiplicity in TPC acceptance
    TCanvas*                       fC1;              //! first canvas for drawing
    TList*                         fHistsList;       //! List of hists - 4-mar-2002
    AliEMCALJet*                   fJetT[10];        //! Jet temporary storage
    AliEMCALHadronCorrection*      fHadronCorrector; //! Pointer to hadronic correction
    Int_t                          fHCorrection;     //  Hadron correction flag
    Int_t                          fDebug;           //! Debug flag
    Int_t                          fBackground;      //! Background flag
    Float_t                        fConeRadius;      //  Cone radius
    Float_t                        fPtCut;           //  Pt cut on charged tracks
    Float_t                        fEtSeed;          //  Min. Et for seed
    Float_t                        fMinJetEt;        //  Min Et of jet
    Float_t                        fMinCellEt;       //  Min Et in one cell
    Float_t                        fSamplingF;       //  Sampling Fraction
    Bool_t                         fSmear;           //  Flag for momentum smearing
    Bool_t                         fEffic;           //  Flag for efficiency simulation
    Bool_t                         fK0N;             //  Flag for efficiency simulation
    Int_t                          fNjets;           //! Number of Jetsp
    Float_t                        fDeta;            //! eta cell size 
    Float_t                        fDphi;            //! phi cell size
    Int_t                          fNcell;           //! number of cells
    Int_t                          fNtot;            //! total number of cells
    Int_t                          fNbinEta;         //! number of cells in eta
    Int_t                          fNbinPhi;         //! number of cells in phi
    Float_t                        fEtaMin;          //! minimum eta  
    Float_t                        fEtaMax;          //! maximum eta
    Float_t                        fPhiMin;          //! minimun phi
    Float_t                        fPhiMax;          //! maximum phi
    Float_t                        fEtCell[30000];   //! Cell Energy
    Float_t                        fEtaCell[30000];  //! Cell eta
    Float_t                        fPhiCell[30000];  //! Cell phi
    Int_t                          fNt;              //! number of tracks
    Int_t                          fNChTpc;          //! number of ch.part in TPC

    Int_t                          fNtS;             //! number of tracks selected
    Int_t*                         fTrackList;       //! List of selected tracks
    Float_t*                       fPtT;             //! Pt   of tracks 
    Float_t*                       fEtaT;            //! Eta  of tracks
    Float_t*                       fPhiT;            //! Phi  of tracks
    Int_t*                         fPdgT;            //! PDG code of tracks
 
    Int_t                          fNtB;             //! number of tracks in Bg
    Int_t*                         fTrackListB;      //! List of selected tracks in Bg
    Float_t*                       fPtB;             //! Pt   of tracks in Bg
    Float_t*                       fEtaB;            //! Eta  of tracks in Bg
    Float_t*                       fPhiB;            //! Phi  of tracks in Bg
    Int_t*                         fPdgB;            //! PDG  of tracks in Bg

    // parameter for jet_finder_ua1
    Int_t                          fMode;            // key for BG subtraction
    Float_t                        fMinMove;         // min cone move 
    Float_t                        fMaxMove;         // max cone move
    Float_t                        fPrecBg;          // max value of change for BG (in %)
    Int_t                          fError;           // error variables 

    char*                          fOutFileName;     //! Output file name
    TFile*                         fOutFile;         //! Output file
    TFile*                         fInFile;          //! Output file
    Int_t                          fEvent;           //! Processed event
    Float_t			   fEMCALWeight;
    Float_t			   fTrackWeight;
    ClassDef(AliEMCALJetFinder,3)        // JetFinder for EMCAL
}
;
#endif // ALIEMCALJetFinder_H
Commit	Line	Data
471f69dc	1	#ifndef ALIEMCALJETFINDER_H
	2	#define ALIEMCALJETFINDER_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id$ */
	8	//*-- Author:
	9	//*-- Andreas Morsch (CERN)
	10
	11	#include <TTask.h>
	12	#include "AliEMCALJet.h"
	13
	14	class TClonesArray;
	15	class TH2F;
c44ecd4c	16	class TH1F;
c44ecd4c	17	class TCanvas;
b561574c	18	class TList;
b561574c	19	class AliEMCALHadronCorrection;
471f69dc	20
	21	class AliEMCALJetFinder : public TTask {
	22	public:
3bc109a9	23	AliEMCALJetFinder();
	24	AliEMCALJetFinder(const char* name, const char *title);
	25	virtual ~AliEMCALJetFinder();
08a3fb06	26	virtual void Init();
a942c029	27	virtual void Find(Int_t ncell, Int_t ncell_tot, Float_t etc[30000],
3bc109a9	28	Float_t etac[30000], Float_t phic[30000],
	29	Float_t min_move, Float_t max_move, Int_t mode,
	30	Float_t prec_bg, Int_t ierror);
a942c029	31	virtual void Find();
759ae494	32	virtual void FindChargedJet();
a942c029	33	virtual void FindTracksInJetCone();
a942c029	34	virtual void Test();
d594b22e	35	virtual void BuildTrackFlagTable();
d594b22e	36	virtual Int_t SetTrackFlag(Float_t radius, Int_t pdgCode, Double_t charge);
3bc109a9	37	// Geometry
	38	virtual void SetCellSize(Float_t eta, Float_t phi);
	39	// Parameters
d594b22e	40	virtual void SetDebug(Int_t flag = 0) {fDebug = flag;}
3bc109a9	41	virtual void SetConeRadius(Float_t par);
	42	virtual void SetEtSeed(Float_t par);
	43	virtual void SetMinJetEt(Float_t par);
	44	virtual void SetMinCellEt(Float_t par);
d594b22e	45	virtual void SetPtCut(Float_t par = 1.);
	46	virtual void SetMomentumSmearing(Bool_t flag = kFALSE) {fSmear = flag;}
	47	virtual void SetEfficiencySim(Bool_t flag = kFALSE) {fEffic = flag;}
5649a778	48	virtual void SetSamplingFraction(Float_t par) {fSamplingF = par;}
7deee4ae	49	virtual void SetEnergyWeightingFlag(Bool_t flag) {fWeightingMethod = flag;}
	50	virtual void SetEMCALWeight(Float_t par) {fEMCALWeight = par;}
	51	virtual void SetTrackWeight(Float_t par) {fTrackWeight = par;}
d53b7b0b	52	virtual void SetIncludeK0andN(Bool_t flag = kFALSE) {fK0N = flag;}
d594b22e	53	// Correction of hadronic energy
	54	virtual void SetHadronCorrector(AliEMCALHadronCorrection* corr)
	55	{fHadronCorrector = corr;}
	56	virtual void SetHadronCorrection(Int_t flag = 1) {fHCorrection = flag;}
b7a73953	57	// PAI
	58	void SetWriteKey(Bool_t flag = kFALSE) {fWrite = flag;}
	59	void SetMode(Int_t mode = 0) {fMode = mode;}
	60	void SetMinMove(Float_t minMove = 0.05) {fMinMove = minMove;}
	61	void SetMaxMove(Float_t maxMove = 0.15) {fMaxMove = maxMove;}
	62	void SetPrecBg (Float_t precBg = 0.035) {fPrecBg = precBg;}
	63	void SetParametersForBgSubtraction
	64	(Int_t mode=0, Float_t minMove=0.05, Float_t maxMove=0.15, Float_t precBg=0.035);
	65	// virtual void Print(Option_t* option="") const; // MENU
	66
	67	Bool_t GetWriteKey() {return fWrite;}
668c80d5	68	//AliEMCALJet* GetJetT() {return fJetT[0];}
668c80d5	69	AliEMCALJet* GetJetT(Int_t n = 0) {return fJetT[n];}
c44ecd4c	70	virtual void DrawHistsForTuning(Int_t mode=0); // MENU
	71	virtual void PrintParameters(Int_t mode=0); // MENU
	72	virtual const Char_t* GetFileNameForParameters(Char_t* dir="RES/");
b7a73953	73
a942c029	74	// Access to Results
3bc109a9	75	virtual Int_t Njets();
	76	virtual Float_t JetEnergy(Int_t);
	77	virtual Float_t JetPhiL(Int_t);
	78	virtual Float_t JetPhiW(Int_t);
	79	virtual Float_t JetEtaL(Int_t);
	80	virtual Float_t JetEtaW(Int_t);
b561574c	81	TH2F* GetLego() {return fLego;}
b561574c	82	TH2F* GetLegoB() {return fLegoB;}
c44ecd4c	83	TH2F* GetLegoEMCAL() {return fhLegoEMCAL;}
7deee4ae	84	TH2F* GetLegoTracks() {return fhLegoTracks;}
c44ecd4c	85	TH2F* GethEff() {return fhEff;}
	86	TH1F* GetCellEt() {return fhCellEt;}
	87	TH1F* GetCellEMCALEt() {return fhCellEMCALEt;}
	88	TH1F* GetTrackPt() {return fhTrackPt;}
	89	TH1F* GetTrackPtBcut() {return fhTrackPtBcut;}
b561574c	90	TList* GetHistsList() {return fHistsList;}
b561574c	91	Int_t GetNChTpc() {return fNChTpc;}
7deee4ae	92	Bool_t GetEnergyWeightingFlag() {return fWeightingMethod ;}
	93	Float_t GetEMCALWeight() {return fEMCALWeight;}
	94	Float_t GetTrackWeight() {return fTrackWeight;}
c44ecd4c	95	void DrawLego(Char_t opt="lego"); // MENU*
	96	void DrawLegoEMCAL(Char_t opt="lego"); // MENU*
	97	void DrawLegos(); // MENU
b561574c	98	Bool_t IsThisPartonsOrDiQuark(Int_t pdg);
3bc109a9	99	// I/O
08a3fb06	100	virtual void SetOutputFileName(char* name) {fOutFileName = name;}
3bc109a9	101	virtual void FillFromHits(Int_t flag = 0);
d594b22e	102	virtual void FillFromHitFlaggedTracks(Int_t flag = 0);
d594b22e	103	virtual void FillFromDigits(Int_t flag = 0);
3bc109a9	104	virtual void FillFromTracks(Int_t flag = 0, Int_t ich = 0);
c44ecd4c	105	virtual void FillFromParticles();
b7a73953	106	virtual void FillFromPartons();
b7a73953	107
a942c029	108	virtual void SaveBackgroundEvent();
a942c029	109	virtual void InitFromBackground();
3bc109a9	110	virtual void AddJet(const AliEMCALJet& jet);
	111	virtual void WriteJets();
	112	virtual void ResetJets();
	113	virtual TClonesArray* Jets() {return fJets;}
471f69dc	114	private:
3bc109a9	115	virtual void BookLego();
7deee4ae	116	Float_t WeightedJetEnergy(Float_t eta, Float_t phi);
	117	Float_t EMCALConeEnergy(Float_t eta, Float_t phi);
	118	Float_t TrackConeEnergy(Float_t eta, Float_t phi);
	119	Float_t HCConeEnergy(Float_t eta, Float_t phi);
3bc109a9	120	virtual void DumpLego();
3bc109a9	121	virtual void ResetMap();
26dadf3a	122	virtual Float_t PropagatePhi(Float_t pt, Float_t charge, Bool_t& curls);
c44ecd4c	123	virtual void RearrangeParticlesMemory(Int_t npart);
471f69dc	124	protected:
b7a73953	125	Bool_t fWrite; // Key for writing
7deee4ae	126	Bool_t fWeightingMethod; // Key for writing
d594b22e	127	TClonesArray* fJets; //! List of Jets
d594b22e	128	TH2F* fLego; //! Lego Histo
c44ecd4c	129	TH2F* fLegoB; //! Lego Histo Backg
	130	TH2F* fhLegoTracks; //! Lego for Tracks
	131	TH2F* fhLegoEMCAL; //! Lego for EMCAL itself
	132	TH2F* fhLegoHadrCorr; //! Lego for hadron correction
	133	TH2F* fhEff; //! Hist. for controling eff.
	134	TH1F* fhCellEt; //! Et distr. for cells from fLego
	135	TH1F* fhCellEMCALEt; //! Et distr. for cells from fLegoEMCAL
	136	TH1F* fhTrackPt; //! Pt distr. for charge particles
	137	TH1F* fhTrackPtBcut; //! Pt distr. for charge particles + cut due to magnetic field
b561574c	138	TH1F* fhChPartMultInTpc;//! Ch. part. multiplicity in TPC acceptance
c44ecd4c	139	TCanvas* fC1; //! first canvas for drawing
b561574c	140	TList* fHistsList; //! List of hists - 4-mar-2002
d594b22e	141	AliEMCALJet* fJetT[10]; //! Jet temporary storage
	142	AliEMCALHadronCorrection* fHadronCorrector; //! Pointer to hadronic correction
	143	Int_t fHCorrection; // Hadron correction flag
	144	Int_t fDebug; //! Debug flag
a942c029	145	Int_t fBackground; //! Background flag
d594b22e	146	Float_t fConeRadius; // Cone radius
	147	Float_t fPtCut; // Pt cut on charged tracks
	148	Float_t fEtSeed; // Min. Et for seed
	149	Float_t fMinJetEt; // Min Et of jet
	150	Float_t fMinCellEt; // Min Et in one cell
a942c029	151	Float_t fSamplingF; // Sampling Fraction
d594b22e	152	Bool_t fSmear; // Flag for momentum smearing
d594b22e	153	Bool_t fEffic; // Flag for efficiency simulation
d53b7b0b	154	Bool_t fK0N; // Flag for efficiency simulation
b561574c	155	Int_t fNjets; //! Number of Jetsp
d594b22e	156	Float_t fDeta; //! eta cell size
	157	Float_t fDphi; //! phi cell size
	158	Int_t fNcell; //! number of cells
	159	Int_t fNtot; //! total number of cells
	160	Int_t fNbinEta; //! number of cells in eta
	161	Int_t fNbinPhi; //! number of cells in phi
be9787fe	162	Float_t fEtaMin; //! minimum eta
	163	Float_t fEtaMax; //! maximum eta
	164	Float_t fPhiMin; //! minimun phi
	165	Float_t fPhiMax; //! maximum phi
d594b22e	166	Float_t fEtCell[30000]; //! Cell Energy
	167	Float_t fEtaCell[30000]; //! Cell eta
	168	Float_t fPhiCell[30000]; //! Cell phi
d594b22e	169	Int_t fNt; //! number of tracks
b561574c	170	Int_t fNChTpc; //! number of ch.part in TPC
b561574c	171
a942c029	172	Int_t fNtS; //! number of tracks selected
b561574c	173	Int_t* fTrackList; //! List of selected tracks
a942c029	174	Float_t* fPtT; //! Pt of tracks
	175	Float_t* fEtaT; //! Eta of tracks
	176	Float_t* fPhiT; //! Phi of tracks
975127ed	177	Int_t* fPdgT; //! PDG code of tracks
975127ed	178
a942c029	179	Int_t fNtB; //! number of tracks in Bg
b561574c	180	Int_t* fTrackListB; //! List of selected tracks in Bg
a942c029	181	Float_t* fPtB; //! Pt of tracks in Bg
	182	Float_t* fEtaB; //! Eta of tracks in Bg
	183	Float_t* fPhiB; //! Phi of tracks in Bg
975127ed	184	Int_t* fPdgB; //! PDG of tracks in Bg
b7a73953	185
b7a73953	186	// parameter for jet_finder_ua1
c44ecd4c	187	Int_t fMode; // key for BG subtraction
b7a73953	188	Float_t fMinMove; // min cone move
b7a73953	189	Float_t fMaxMove; // max cone move
b7a73953	190	Float_t fPrecBg; // max value of change for BG (in %)
	191	Int_t fError; // error variables
	192
08a3fb06	193	char* fOutFileName; //! Output file name
	194	TFile* fOutFile; //! Output file
	195	TFile* fInFile; //! Output file
	196	Int_t fEvent; //! Processed event
7deee4ae	197	Float_t fEMCALWeight;
	198	Float_t fTrackWeight;
	199	ClassDef(AliEMCALJetFinder,3) // JetFinder for EMCAL
3bc109a9	200	}
3bc109a9	201	;
471f69dc	202	#endif // ALIEMCALJetFinder_H