[u/mrichter/AliRoot.git] / PWG4 / TwoPartCorr / AutoCorr.h

// $Id:$

#ifndef AutoCorr_h
#define AutoCorr_h

#include <Riostream.h>
#include <TTree.h>
#include <TChain.h>
#include <TCanvas.h>
#include <TClonesArray.h>
#include <TH1F.h>
#include <TH2F.h>
#include "Rtypes.h"
#include "TMath.h"
#include "TFile.h"
#include "TSystem.h"
#include "TROOT.h"
#include <vector>
#include <deque>
#include "TRandom.h"
#include "TreeClasses.h"
#include "EventPool.h"

class AutoCorr : public TObject
{
public:
  AutoCorr(){;}
  ~AutoCorr(){;}
  
  // RangeMin,Max specify periodic boundaries
  Double_t DeltaPhi(MyPart* t1, MyPart* t2, 
		    double rangeMin = -TMath::Pi()/2, 
		    double rangeMax = 3*TMath::Pi()/2);
  Double_t DeltaEta(MyPart* t1, MyPart* t2);  
  Bool_t IsTrackOk(MyPart* t);
  Bool_t IsPairOk(MyPart* t1, MyPart* t2);
  Bool_t IsMixedPairOk(MyPart* t1, MyPart* t2);
//  Bool_t IsEventOk(MyHeader *h);

  Int_t InitEventPools(Int_t depth, 
		       Int_t nmultbins, Double_t multbins[], 
		       Int_t nzvtxbins, Double_t zvtxbins[]);

  Int_t UpdatePools(int iEvent, MyHeader* ev, TClonesArray* trk);
  EventPool* GetEventPool(int iMult, int iZvtx);// { return fEvPool.at(iMult).at(iZvtx); }
protected:
  Int_t fNMultBins;
  Int_t fNZvtxBins;

  std::vector<std::vector<EventPool*> > fEvPool; // [fMultBin][fZvtxBin]

  ClassDef(AutoCorr,1)
};
#endif
Commit	Line	Data
88a5a45e	1	// $Id:$
	2
	3	#ifndef AutoCorr_h
	4	#define AutoCorr_h
	5
	6	#include <Riostream.h>
	7	#include <TTree.h>
	8	#include <TChain.h>
	9	#include <TCanvas.h>
	10	#include <TClonesArray.h>
	11	#include <TH1F.h>
	12	#include <TH2F.h>
	13	#include "Rtypes.h"
	14	#include "TMath.h"
	15	#include "TFile.h"
	16	#include "TSystem.h"
	17	#include "TROOT.h"
	18	#include <vector>
	19	#include <deque>
	20	#include "TRandom.h"
	21	#include "TreeClasses.h"
	22	#include "EventPool.h"
	23
	24	class AutoCorr : public TObject
	25	{
	26	public:
	27	AutoCorr(){;}
	28	~AutoCorr(){;}
	29
	30	// RangeMin,Max specify periodic boundaries
	31	Double_t DeltaPhi(MyPart* t1, MyPart* t2,
	32	double rangeMin = -TMath::Pi()/2,
	33	double rangeMax = 3*TMath::Pi()/2);
	34	Double_t DeltaEta(MyPart* t1, MyPart* t2);
	35	Bool_t IsTrackOk(MyPart* t);
	36	Bool_t IsPairOk(MyPart* t1, MyPart* t2);
	37	Bool_t IsMixedPairOk(MyPart* t1, MyPart* t2);
	38	// Bool_t IsEventOk(MyHeader *h);
	39
	40	Int_t InitEventPools(Int_t depth,
	41	Int_t nmultbins, Double_t multbins[],
	42	Int_t nzvtxbins, Double_t zvtxbins[]);
	43
	44	Int_t UpdatePools(int iEvent, MyHeader* ev, TClonesArray* trk);
	45	EventPool* GetEventPool(int iMult, int iZvtx);// { return fEvPool.at(iMult).at(iZvtx); }
	46	protected:
	47	Int_t fNMultBins;
	48	Int_t fNZvtxBins;
	49
	50	std::vector<std::vector<EventPool*> > fEvPool; // [fMultBin][fZvtxBin]
	51
	52	ClassDef(AutoCorr,1)
	53	};
	54	#endif