[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskKMeans.h

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

/* $Id$ */

//-------------------------------------------------------------------------
//     Analysis Task that uses the Soft K-Means Algorithm to find clusters in
//     the eta-phi space of Minimum Bias. No pt information is used for the clustering.
//     
//
//     Author: Andreas Morsch (CERN)
//     andreas.morsch@cern.ch
//-------------------------------------------------------------------------

class TH1F;
class TH2F;
class TList;
class TProfile;

class AliESDEvent;
class AliESDtrack;
class AliESDtrackCuts;


#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskKMeans : public AliAnalysisTaskSE {
 public:
    AliAnalysisTaskKMeans();
  AliAnalysisTaskKMeans(const char *name);
  virtual ~AliAnalysisTaskKMeans() {}
  virtual void     UserCreateOutputObjects();
  virtual void     UserExec(Option_t *option);
  virtual void     Terminate(Option_t *);
  virtual void     SetCuts(AliESDtrackCuts* cuts) {fCuts = cuts;}
  virtual Double_t DeltaPhi(Double_t phi1, Double_t phi2);
  virtual Double_t DeltaR(Double_t phi1, Double_t eta1, Double_t phi2, Double_t eta2);
  
 private:
  // Others
  TList*           fHists;         // Histograms
  TH1F*            fH1CEta;        // Eta distribution of clusters
  TH1F*            fH1CPhi;        // Phi distribution of clusters  
  TH1F*            fH1CEtaR;       // Eta distribution of clusters for rndm evnt
  TH2F*            fH2N1N2;        // Cluster sizes 
  TH1F*            fH1Pt;          // pt outside clusters
  TH1F*            fH1PtC;         // pt outside clusters
  TH1F*            fH1PtC1;        // pt dr > 0.4
  TH1F*            fH1PtC2;        // pt dr > 0.2 
  TH1F*            fH1SPt;         // sum pt
  TH1F*            fH1SPtC;        // sum pt
  TH1F*            fH1DPhi;        // Dphi wr to cluster
  TH1F*            fH1DR;          // DR   wr to cluster
  TH1F*            fH1DRR;         // DR   wr to cluster from rndm events   
  TH2F*            fH2DPhiEta;     // eta-phi wr to cluster
  TH2F*            fH2DPhiEtaR;    // eta-phi wr to cluster for rndm events 
  TH2F*            fH2DPhiEtaL;    // eta-phi of leading particle
  TH2F*            fH2DPhiEtaC;    // eta-phi of Clusters
  TH2F*            fH2DPhiEtaCR;   // eta-phi of Clusters
  
  AliESDtrackCuts* fCuts;             // List of cuts
  ClassDef(AliAnalysisTaskKMeans, 1); // A k-means clustering analysis
};

#endif
Commit	Line	Data
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
	17	class TH1F;
	18	class TH2F;
	19	class TList;
	20	class TProfile;
	21
	22	class AliESDEvent;
	23	class AliESDtrack;
	24	class AliESDtrackCuts;
	25
	26
	27	#include "AliAnalysisTaskSE.h"
	28
	29	class 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);
	40
	41	private:
	42	// Others
	43	TList* fHists; // Histograms
	44	TH1F* fH1CEta; // Eta distribution of clusters
	45	TH1F* fH1CPhi; // Phi distribution of clusters
	46	TH1F* fH1CEtaR; // Eta distribution of clusters for rndm evnt
	47	TH2F* fH2N1N2; // Cluster sizes
	48	TH1F* fH1Pt; // pt outside clusters
	49	TH1F* fH1PtC; // pt outside clusters
	50	TH1F* fH1PtC1; // pt dr > 0.4
	51	TH1F* fH1PtC2; // pt dr > 0.2
a51f90df	52	TH1F* fH1SPt; // sum pt
a51f90df	53	TH1F* fH1SPtC; // sum pt
70f2ce9d	54	TH1F* fH1DPhi; // Dphi wr to cluster
	55	TH1F* fH1DR; // DR wr to cluster
	56	TH1F* fH1DRR; // DR wr to cluster from rndm events
	57	TH2F* fH2DPhiEta; // eta-phi wr to cluster
	58	TH2F* fH2DPhiEtaR; // eta-phi wr to cluster for rndm events
	59	TH2F* fH2DPhiEtaL; // eta-phi of leading particle
a51f90df	60	TH2F* fH2DPhiEtaC; // eta-phi of Clusters
a51f90df	61	TH2F* fH2DPhiEtaCR; // eta-phi of Clusters
70f2ce9d	62
	63	AliESDtrackCuts* fCuts; // List of cuts
	64	ClassDef(AliAnalysisTaskKMeans, 1); // A k-means clustering analysis
	65	};
	66
	67	#endif