[u/mrichter/AliRoot.git] / ANALYSIS / AliBalance.h

#ifndef ALIBALANCE_H
#define ALIBALANCE_H
/*  See cxx source for full Copyright notice */


/* $Id$ */

//-------------------------------------------------------------------------
//                          Class AliBalance
//   This is the class for the Balance Function analysis
//
//    Origin: Panos Christakoglou, UOA-CERN, Panos.Christakoglou@cern.ch
//-------------------------------------------------------------------------

#include <TROOT.h>
#include <TObject.h>
#include <TLorentzVector.h>

#define MAXIMUM_NUMBER_OF_STEPS	1024

class AliBalance : public TObject
{
 private:
  Double_t *fCharge; //matrix with the charge of each track
  Int_t fNtrack; //number of tracks to compute the BF

  TLorentzVector *fV; //4-momentum vector - info for tracks used to compute the BF
  
  Int_t fNumberOfBins; //number of bins of the analyzed interval
  Int_t fAnalysisType; //0:y - 1:eta - 2:Qlong - 3:Qout - 4:Qside - 5:Qinv - 6:phi
  Int_t fAnalyzedEvents; //number of events that have been analyzed
 
  Double_t fP2_Start, fP2_Stop, fP2_Step; //inerval info
 	
  Double_t fNnn[MAXIMUM_NUMBER_OF_STEPS]; //N(--)
  Double_t fNpp[MAXIMUM_NUMBER_OF_STEPS]; //N(++)
  Double_t fNpn[MAXIMUM_NUMBER_OF_STEPS]; //N(+-)
  Double_t fNn, fNp; //number of pos./neg. inside the analyzed interval
  
  Double_t fB[MAXIMUM_NUMBER_OF_STEPS]; //BF matrix
  Double_t ferror[MAXIMUM_NUMBER_OF_STEPS]; //error of the BF
  
 public:
  AliBalance();
  
  AliBalance(Double_t P2_Start, Double_t P2_Stop, Int_t P2_Steps);
  
  ~AliBalance();
  
  void SetParticles(TLorentzVector *P, Double_t *charge, Int_t dim);
  void SetNumberOfBins(Int_t ibins);
  void SetAnalysisType(Int_t iType);
  void SetInterval(Double_t P2_Start, Double_t P2_Stop);

  void CalculateBalance();
  
  Double_t GetNp() { return 1.0*fNp; }
  Double_t GetNn() { return 1.0*fNn; }
  Double_t GetNnn(Int_t p2) { return 1.0*fNnn[p2]; }
  Double_t GetNpp(Int_t p2) { return 1.0*fNpp[p2]; }
  Double_t GetNpn(Int_t p2) { return 1.0*fNpn[p2]; }
 
  Double_t GetBalance(Int_t p2);
  Double_t GetError(Int_t p2);

  void PrintResults();
 
  ClassDef(AliBalance, 1)
} ;

#endif
Commit	Line	Data
cd286c84	1	#ifndef ALIBALANCE_H
	2	#define ALIBALANCE_H
	3	/* See cxx source for full Copyright notice */
	4
	5
	6	/* $Id$ */
	7
	8	//-------------------------------------------------------------------------
	9	// Class AliBalance
	10	// This is the class for the Balance Function analysis
	11	//
	12	// Origin: Panos Christakoglou, UOA-CERN, Panos.Christakoglou@cern.ch
	13	//-------------------------------------------------------------------------
	14
	15	#include <TROOT.h>
	16	#include <TObject.h>
	17	#include <TLorentzVector.h>
	18
	19	#define MAXIMUM_NUMBER_OF_STEPS 1024
	20
	21	class AliBalance : public TObject
	22	{
	23	private:
	24	Double_t *fCharge; //matrix with the charge of each track
	25	Int_t fNtrack; //number of tracks to compute the BF
	26
	27	TLorentzVector *fV; //4-momentum vector - info for tracks used to compute the BF
	28
	29	Int_t fNumberOfBins; //number of bins of the analyzed interval
	30	Int_t fAnalysisType; //0:y - 1:eta - 2:Qlong - 3:Qout - 4:Qside - 5:Qinv - 6:phi
	31	Int_t fAnalyzedEvents; //number of events that have been analyzed
	32
	33	Double_t fP2_Start, fP2_Stop, fP2_Step; //inerval info
	34
	35	Double_t fNnn[MAXIMUM_NUMBER_OF_STEPS]; //N(--)
	36	Double_t fNpp[MAXIMUM_NUMBER_OF_STEPS]; //N(++)
	37	Double_t fNpn[MAXIMUM_NUMBER_OF_STEPS]; //N(+-)
	38	Double_t fNn, fNp; //number of pos./neg. inside the analyzed interval
	39
	40	Double_t fB[MAXIMUM_NUMBER_OF_STEPS]; //BF matrix
	41	Double_t ferror[MAXIMUM_NUMBER_OF_STEPS]; //error of the BF
	42
	43	public:
	44	AliBalance();
	45
	46	AliBalance(Double_t P2_Start, Double_t P2_Stop, Int_t P2_Steps);
	47
	48	~AliBalance();
	49
	50	void SetParticles(TLorentzVector P, Double_t charge, Int_t dim);
	51	void SetNumberOfBins(Int_t ibins);
	52	void SetAnalysisType(Int_t iType);
	53	void SetInterval(Double_t P2_Start, Double_t P2_Stop);
	54
94d49dd3	55	void CalculateBalance();
cd286c84	56
	57	Double_t GetNp() { return 1.0*fNp; }
	58	Double_t GetNn() { return 1.0*fNn; }
	59	Double_t GetNnn(Int_t p2) { return 1.0*fNnn[p2]; }
	60	Double_t GetNpp(Int_t p2) { return 1.0*fNpp[p2]; }
	61	Double_t GetNpn(Int_t p2) { return 1.0*fNpn[p2]; }
	62
	63	Double_t GetBalance(Int_t p2);
	64	Double_t GetError(Int_t p2);
	65
	66	void PrintResults();
	67
	68	ClassDef(AliBalance, 1)
	69	} ;
	70
	71	#endif