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

#ifndef ALIANALYSISTASKKMEANS_H
#define ALIANALYSISTASKKMEANS_H
 /* 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;
class AliKMeansResult;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskKMeans : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskKMeans();
  AliAnalysisTaskKMeans(const char *name);
  AliAnalysisTaskKMeans(const AliAnalysisTaskKMeans &res);
  AliAnalysisTaskKMeans& operator=(const AliAnalysisTaskKMeans& trclass);
  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);
  virtual void     SetK(Int_t k) {fK = k;} 
  virtual void     SetMinimumMultiplicity(Int_t k) {fNMin = k;} 
  virtual void     SetEtaPhi(TH2F* h2) {fH2REtaPhi = h2;}
 private:
  // Others
  Int_t            fK;             // K                        
  Int_t            fNMin;          // Minimum multipicity                         
  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*            fH1PtAS;        //! away-side peak 
  TH1F*            fH1PtR;         //! away-side peak 
  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*            fH2DPhiEtaLR;   //! eta-phi of leading particle
  TH2F*            fH2DPhiEtaC;    //! eta-phi of Clusters
  TH2F*            fH2DPhiEtaCR;   //! eta-phi of Clusters
  TH1F*            fH1Resp;        //! responsibility
  TH1F*            fH1RespR;       //! responsibility
  TH2F*            fH2Sigma;       //! sigma
  TH2F*            fH2SigmaR;      //! sigma random
  TH1F*            fHDensity;      //! Particle density
  TH1F*            fHCSize;        //! Cluster Size
  TH1F*            fHNCluster;     //! Number of clusters
  TH2F*            fHPtDensity;    //! Pt vs density
  TH1F*            fHDPhi;         //! Phi Correlation 
  TH2F*            fH2EtaPhi;      //! eta phi 
  TH2F*            fH2REtaPhi;     //  eta phi 
  AliKMeansResult* fA[10];         //! Array of results
  AliKMeansResult* fB[10];         //! Array of results
  AliESDtrackCuts* fCuts;          // List of cuts
  ClassDef(AliAnalysisTaskKMeans, 1); // A k-means clustering analysis
};

#endif
Commit	Line	Data
e42693ad	1	#ifndef ALIANALYSISTASKKMEANS_H
e42693ad	2	#define ALIANALYSISTASKKMEANS_H
70f2ce9d	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;
70f2ce9d	21	class AliESDEvent;
	22	class AliESDtrack;
	23	class AliESDtrackCuts;
19c5a36a	24	class AliKMeansResult;
70f2ce9d	25
	26	#include "AliAnalysisTaskSE.h"
	27
	28	class AliAnalysisTaskKMeans : public AliAnalysisTaskSE {
	29	public:
e42693ad	30	AliAnalysisTaskKMeans();
70f2ce9d	31	AliAnalysisTaskKMeans(const char *name);
e42693ad	32	AliAnalysisTaskKMeans(const AliAnalysisTaskKMeans &res);
e42693ad	33	AliAnalysisTaskKMeans& operator=(const AliAnalysisTaskKMeans& trclass);
70f2ce9d	34	virtual ~AliAnalysisTaskKMeans() {}
	35	virtual void UserCreateOutputObjects();
	36	virtual void UserExec(Option_t *option);
	37	virtual void Terminate(Option_t *);
	38	virtual void SetCuts(AliESDtrackCuts* cuts) {fCuts = cuts;}
	39	virtual Double_t DeltaPhi(Double_t phi1, Double_t phi2);
	40	virtual Double_t DeltaR(Double_t phi1, Double_t eta1, Double_t phi2, Double_t eta2);
320bb308	41	virtual void SetK(Int_t k) {fK = k;}
42dc9410	42	virtual void SetMinimumMultiplicity(Int_t k) {fNMin = k;}
b31787ca	43	virtual void SetEtaPhi(TH2F* h2) {fH2REtaPhi = h2;}
70f2ce9d	44	private:
70f2ce9d	45	// Others
320bb308	46	Int_t fK; // K
42dc9410	47	Int_t fNMin; // Minimum multipicity
19c5a36a	48	TList* fHists; //! Histograms
	49	TH1F* fH1CEta; //! Eta distribution of clusters
	50	TH1F* fH1CPhi; //! Phi distribution of clusters
	51	TH1F* fH1CEtaR; //! Eta distribution of clusters for rndm evnt
	52	TH2F* fH2N1N2; //! Cluster sizes
	53	TH1F* fH1Pt; //! pt outside clusters
	54	TH1F* fH1PtC; //! pt outside clusters
	55	TH1F* fH1PtC1; //! pt dr > 0.4
	56	TH1F* fH1PtC2; //! pt dr > 0.2
	57	TH1F* fH1PtAS; //! away-side peak
	58	TH1F* fH1PtR; //! away-side peak
	59	TH1F* fH1SPt; //! sum pt
	60	TH1F* fH1SPtC; //! sum pt
	61	TH1F* fH1DPhi; //! Dphi wr to cluster
	62	TH1F* fH1DR; //! DR wr to cluster
	63	TH1F* fH1DRR; //! DR wr to cluster from rndm events
	64	TH2F* fH2DPhiEta; //! eta-phi wr to cluster
	65	TH2F* fH2DPhiEtaR; //! eta-phi wr to cluster for rndm events
	66	TH2F* fH2DPhiEtaL; //! eta-phi of leading particle
	67	TH2F* fH2DPhiEtaLR; //! eta-phi of leading particle
	68	TH2F* fH2DPhiEtaC; //! eta-phi of Clusters
	69	TH2F* fH2DPhiEtaCR; //! eta-phi of Clusters
	70	TH1F* fH1Resp; //! responsibility
	71	TH1F* fH1RespR; //! responsibility
	72	TH2F* fH2Sigma; //! sigma
	73	TH2F* fH2SigmaR; //! sigma random
	74	TH1F* fHDensity; //! Particle density
	75	TH1F* fHCSize; //! Cluster Size
	76	TH1F* fHNCluster; //! Number of clusters
	77	TH2F* fHPtDensity; //! Pt vs density
b31787ca	78	TH1F* fHDPhi; //! Phi Correlation
	79	TH2F* fH2EtaPhi; //! eta phi
	80	TH2F* fH2REtaPhi; // eta phi
f2b8ec84	81	AliKMeansResult* fA[10]; //! Array of results
f2b8ec84	82	AliKMeansResult* fB[10]; //! Array of results
b31787ca	83	AliESDtrackCuts* fCuts; // List of cuts
70f2ce9d	84	ClassDef(AliAnalysisTaskKMeans, 1); // A k-means clustering analysis
	85	};
	86
	87	#endif