[u/mrichter/AliRoot.git] / PWGGA / PHOSTasks / PHOS_PbPb / AliAnalysisTaskPi0FlowMC.h

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

// Inclusion of AliPHOSHijingEfficiency,
// by Dmitri Peressounko, 05.02.2013
// Authors: Henrik Qvigstad, Dmitri Peressounko
// Date   : 20.06.2013
/* $Id$ */


#ifndef ALIANALYSISTASKPI0FLOWMC_H
#define ALIANALYSISTASKPI0FLOWMC_H

class TParticle;

#include <AliAnalysisTaskPi0Flow.h>


class AliAnalysisTaskPi0FlowMC : public AliAnalysisTaskPi0Flow
{
public:
  AliAnalysisTaskPi0FlowMC(const char* name = "AliAnalysisTaskPi0Flow", Period period = kUndefinedPeriod);
  virtual ~AliAnalysisTaskPi0FlowMC();

protected: // Override:
  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t *option);
  // Pi0FlowTask
  virtual void SelectPhotonClusters();
  virtual void FillSelectedClusterHistograms();
  virtual void ConsiderPi0s();
  virtual void ConsiderPi0sMix();
  virtual void ProcessMC();

protected: // member functions:
  AliAnalysisTaskPi0FlowMC(const AliAnalysisTaskPi0FlowMC&); // not implemented
  AliAnalysisTaskPi0FlowMC& operator=(const AliAnalysisTaskPi0FlowMC&); // not implemented
  
  AliStack* GetMCStack();

protected: // member variables:
  AliStack* fStack;
  
  void FillMCHist();
  
  virtual Double_t PrimaryWeight(Int_t primary);
  virtual Double_t PrimaryParticleWeight(TParticle * particle);
  void FillSecondaries() ;
  Int_t FindPrimary(AliVCluster* clu,  Bool_t& sure);
  Int_t FindCommonParent(Int_t iPart, Int_t jPart) ;
  Bool_t HaveParent(Int_t iPart, Int_t pdgParent);
  Bool_t InPi0mass(Double_t m, Double_t pt);

  void FillAllHistograms(const char* particleName, AliCaloPhoton* ph1);

  static const Double_t kRCut = 1.;
  enum ParticleID {kEta=221};


  ClassDef(AliAnalysisTaskPi0FlowMC, 1); // PHOS analysis task
};

#endif // ALIANALYSISTASKPI0FLOWMC_H
Commit	Line	Data
444647ad	1	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	2	* See cxx source for full Copyright notice */
	3
	4	// Inclusion of AliPHOSHijingEfficiency,
	5	// by Dmitri Peressounko, 05.02.2013
	6	// Authors: Henrik Qvigstad, Dmitri Peressounko
7ea85581	7	// Date : 20.06.2013
444647ad	8	/* $Id$ */
	9
	10
	11	#ifndef ALIANALYSISTASKPI0FLOWMC_H
	12	#define ALIANALYSISTASKPI0FLOWMC_H
	13
	14	class TParticle;
	15
	16	#include <AliAnalysisTaskPi0Flow.h>
	17
	18
	19	class AliAnalysisTaskPi0FlowMC : public AliAnalysisTaskPi0Flow
	20	{
	21	public:
	22	AliAnalysisTaskPi0FlowMC(const char* name = "AliAnalysisTaskPi0Flow", Period period = kUndefinedPeriod);
	23	virtual ~AliAnalysisTaskPi0FlowMC();
	24
abbd081b	25	protected: // Override:
	26	virtual void UserCreateOutputObjects();
	27	virtual void UserExec(Option_t *option);
	28	// Pi0FlowTask
	29	virtual void SelectPhotonClusters();
	30	virtual void FillSelectedClusterHistograms();
	31	virtual void ConsiderPi0s();
	32	virtual void ConsiderPi0sMix();
	33	virtual void ProcessMC();
	34
444647ad	35	protected: // member functions:
	36	AliAnalysisTaskPi0FlowMC(const AliAnalysisTaskPi0FlowMC&); // not implemented
	37	AliAnalysisTaskPi0FlowMC& operator=(const AliAnalysisTaskPi0FlowMC&); // not implemented
444647ad	38
444647ad	39	AliStack* GetMCStack();
abbd081b	40
444647ad	41	protected: // member variables:
	42	AliStack* fStack;
	43
	44	void FillMCHist();
	45
7ea85581	46	virtual Double_t PrimaryWeight(Int_t primary);
7ea85581	47	virtual Double_t PrimaryParticleWeight(TParticle * particle);
444647ad	48	void FillSecondaries() ;
abbd081b	49	Int_t FindPrimary(AliVCluster* clu, Bool_t& sure);
444647ad	50	Int_t FindCommonParent(Int_t iPart, Int_t jPart) ;
	51	Bool_t HaveParent(Int_t iPart, Int_t pdgParent);
	52	Bool_t InPi0mass(Double_t m, Double_t pt);
	53
	54	void FillAllHistograms(const char* particleName, AliCaloPhoton* ph1);
	55
	56	static const Double_t kRCut = 1.;
	57	enum ParticleID {kEta=221};
abbd081b	58
abbd081b	59
444647ad	60	ClassDef(AliAnalysisTaskPi0FlowMC, 1); // PHOS analysis task
	61	};
	62
	63	#endif // ALIANALYSISTASKPI0FLOWMC_H