[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / charmFlow / AliHFAfterBurner.h

#ifndef ALIANALYSISTASKAFTERBURNER_H
#define ALIANALYSISTASKAFTERBURNER_H

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

//*************************************************************************
// AliAnalysisTaskSEHFv2 gives the needed tools for the D 
// mesons v2 analysis 
// Authors: Giacomo Ortona, ortona@to.infn.it,
//          
//*************************************************************************

/* $Id$ */

class TH1F;
class TH2D;
class AliRDHFCuts;
class TVector2;

class AliHFAfterBurner : public TObject
{
 public:

  AliHFAfterBurner();
  AliHFAfterBurner(Int_t decChannel);

  virtual ~AliHFAfterBurner();

  Double_t GetNewAngle(AliAODRecoDecayHF *d,TClonesArray *arrayMC);
  Double_t NewtonMethodv2(Double_t phi,Double_t v2,Double_t phi0);
  Double_t NewtonMethodv2(Double_t phi,Double_t v2){
    return NewtonMethodv2(phi,v2,phi);}
  void SetMCv2(Float_t v2sig,Float_t v2bkg);
  Int_t CheckOrigin(const AliAODMCParticle* mcPart,TClonesArray *arrayMC) const;
  Double_t GetPhi(Double_t phi,Float_t v2);
  Float_t GetPhi02Pi(Float_t phi);
  void SetPrecision(Bool_t prec){fPrecisionNewton=prec;}
  void SetUseNewton(Bool_t newt){fUseNewton=newt;}
  void SetDecChannel(Int_t decch);
  void SetEventPlane(Double_t ep){fEventPlane=ep;}
  void SetEventPlane();
  void SetEventPlaneMethod(Int_t method);

  Bool_t GetIsSignal() const {return fSignal;}
  Double_t GetEventPlane() const {return fEventPlane;}
 private:
  AliHFAfterBurner(const AliHFAfterBurner &source);
  AliHFAfterBurner& operator=(const AliHFAfterBurner& source); 

  Float_t fSigv2;       //v2 for signal
  Float_t fBkgv2;       //v2 for background
  Bool_t fUseNewton;    //Switch between Newton and fast methods
  Double_t fPrecisionNewton;//precision to stop Newton method
  Int_t fDecChannel;   //D2H decay channel
  Bool_t fSignal;      //kTRUE if candidate is signal
  Double_t fEventPlane; //event plane
  Int_t fMethodEP;     //switch between different EP method calculation

  ClassDef(AliHFAfterBurner,1); // AliFlowAB class
};
#endif
Commit	Line	Data
ed352b10	1	#ifndef ALIANALYSISTASKAFTERBURNER_H
	2	#define ALIANALYSISTASKAFTERBURNER_H
	3
	4	/* Copyright(c) 1998-2010, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	//*************************************************************************
	8	// AliAnalysisTaskSEHFv2 gives the needed tools for the D
	9	// mesons v2 analysis
	10	// Authors: Giacomo Ortona, ortona@to.infn.it,
	11	//
	12	//*************************************************************************
	13
	14	/* $Id$ */
	15
	16	class TH1F;
	17	class TH2D;
	18	class AliRDHFCuts;
	19	class TVector2;
	20
	21	class AliHFAfterBurner : public TObject
	22	{
	23	public:
	24
	25	AliHFAfterBurner();
	26	AliHFAfterBurner(Int_t decChannel);
	27
	28	virtual ~AliHFAfterBurner();
	29
	30	Double_t GetNewAngle(AliAODRecoDecayHF d,TClonesArray arrayMC);
	31	Double_t NewtonMethodv2(Double_t phi,Double_t v2,Double_t phi0);
	32	Double_t NewtonMethodv2(Double_t phi,Double_t v2){
	33	return NewtonMethodv2(phi,v2,phi);}
	34	void SetMCv2(Float_t v2sig,Float_t v2bkg);
	35	Int_t CheckOrigin(const AliAODMCParticle* mcPart,TClonesArray *arrayMC) const;
	36	Double_t GetPhi(Double_t phi,Float_t v2);
	37	Float_t GetPhi02Pi(Float_t phi);
	38	void SetPrecision(Bool_t prec){fPrecisionNewton=prec;}
	39	void SetUseNewton(Bool_t newt){fUseNewton=newt;}
	40	void SetDecChannel(Int_t decch);
	41	void SetEventPlane(Double_t ep){fEventPlane=ep;}
	42	void SetEventPlane();
	43	void SetEventPlaneMethod(Int_t method);
	44
262b44fa	45	Bool_t GetIsSignal() const {return fSignal;}
262b44fa	46	Double_t GetEventPlane() const {return fEventPlane;}
ed352b10	47	private:
	48	AliHFAfterBurner(const AliHFAfterBurner &source);
	49	AliHFAfterBurner& operator=(const AliHFAfterBurner& source);
	50
	51	Float_t fSigv2; //v2 for signal
	52	Float_t fBkgv2; //v2 for background
	53	Bool_t fUseNewton; //Switch between Newton and fast methods
	54	Double_t fPrecisionNewton;//precision to stop Newton method
	55	Int_t fDecChannel; //D2H decay channel
	56	Bool_t fSignal; //kTRUE if candidate is signal
	57	Double_t fEventPlane; //event plane
	58	Int_t fMethodEP; //switch between different EP method calculation
	59
	60	ClassDef(AliHFAfterBurner,1); // AliFlowAB class
	61	};
	62	#endif