from Per Thomas - add new raw ana classes in LinkDef also
[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskKMeans.h
CommitLineData
70f2ce9d 1#ifndef AliAnalysisTaskKMeans_cxx
2#define AliAnalysisTaskKMeans_cxx
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6/* $Id$ */
7
8//-------------------------------------------------------------------------
9// Analysis Task that uses the Soft K-Means Algorithm to find clusters in
10// the eta-phi space of Minimum Bias. No pt information is used for the clustering.
11//
12//
13// Author: Andreas Morsch (CERN)
14// andreas.morsch@cern.ch
15//-------------------------------------------------------------------------
16
17class TH1F;
18class TH2F;
19class TList;
20class TProfile;
21
22class AliESDEvent;
23class AliESDtrack;
24class AliESDtrackCuts;
25
26
27#include "AliAnalysisTaskSE.h"
28
29class AliAnalysisTaskKMeans : public AliAnalysisTaskSE {
30 public:
31 AliAnalysisTaskKMeans();
32 AliAnalysisTaskKMeans(const char *name);
33 virtual ~AliAnalysisTaskKMeans() {}
34 virtual void UserCreateOutputObjects();
35 virtual void UserExec(Option_t *option);
36 virtual void Terminate(Option_t *);
37 virtual void SetCuts(AliESDtrackCuts* cuts) {fCuts = cuts;}
38 virtual Double_t DeltaPhi(Double_t phi1, Double_t phi2);
39 virtual Double_t DeltaR(Double_t phi1, Double_t eta1, Double_t phi2, Double_t eta2);
320bb308 40 virtual void SetK(Int_t k) {fK = k;}
70f2ce9d 41 private:
42 // Others
320bb308 43 Int_t fK; // K
70f2ce9d 44 TList* fHists; // Histograms
45 TH1F* fH1CEta; // Eta distribution of clusters
46 TH1F* fH1CPhi; // Phi distribution of clusters
47 TH1F* fH1CEtaR; // Eta distribution of clusters for rndm evnt
48 TH2F* fH2N1N2; // Cluster sizes
49 TH1F* fH1Pt; // pt outside clusters
50 TH1F* fH1PtC; // pt outside clusters
51 TH1F* fH1PtC1; // pt dr > 0.4
52 TH1F* fH1PtC2; // pt dr > 0.2
a51f90df 53 TH1F* fH1SPt; // sum pt
54 TH1F* fH1SPtC; // sum pt
70f2ce9d 55 TH1F* fH1DPhi; // Dphi wr to cluster
56 TH1F* fH1DR; // DR wr to cluster
57 TH1F* fH1DRR; // DR wr to cluster from rndm events
58 TH2F* fH2DPhiEta; // eta-phi wr to cluster
59 TH2F* fH2DPhiEtaR; // eta-phi wr to cluster for rndm events
60 TH2F* fH2DPhiEtaL; // eta-phi of leading particle
5a9b5891 61 TH2F* fH2DPhiEtaLR; // eta-phi of leading particle
a51f90df 62 TH2F* fH2DPhiEtaC; // eta-phi of Clusters
63 TH2F* fH2DPhiEtaCR; // eta-phi of Clusters
320bb308 64 TH1F* fH1Resp; // responsibility
65 TH1F* fH1RespR; // responsibility
70f2ce9d 66 AliESDtrackCuts* fCuts; // List of cuts
67 ClassDef(AliAnalysisTaskKMeans, 1); // A k-means clustering analysis
68};
69
70#endif