[u/mrichter/AliRoot.git] / EVGEN / AliDimuCombinator.h

#ifndef _AliDimuCombinator_H
#define _AliDimuCombinator_H
#include "GParticle.h"
#include <TBrowser.h>
#include <TList.h>
#include <TTree.h>
#include <TROOT.h>


class AliDimuCombinator:
    public TObject 
{
public:
    AliDimuCombinator(TClonesArray* Partarray){
	fPartArray=Partarray;
	fNParticle=fPartArray->GetEntriesFast();
	
	fimuon1 =0;
	fimuon2 =0;
	fmuon1  =0;
	fmuon2  =0;
	fimin1  = 0;
	fimin2  = 0;
	fimax1  = fNParticle;
	fimax2  = fNParticle;
	fPtMin  =0;
	fEtaMin =-10;
	fEtaMax =-10;
	fRate1=1.;
	fRate2=1.;
    }
//    
//  Iterators
//  Single muons
    GParticle* FirstMuon();
    GParticle* NextMuon();
//  Single muons selected
    GParticle* FirstMuonSelected();
    GParticle* NextMuonSelected();
//  Dimuons    
    void FirstMuonPair(GParticle* & muon1, GParticle* & muon2);
    void NextMuonPair(GParticle* & muon1, GParticle* & muon2);
//  Dimuons selected    
    void FirstMuonPairSelected(GParticle* & muon1, GParticle* & muon2);
    void NextMuonPairSelected(GParticle* & muon1, GParticle* & muon2);
//  Loop over all prticles    
    void ResetRange();
//  Set two ranges for dimuon loop    
    void SetFirstRange (Int_t from, Int_t to);
    void SetSecondRange(Int_t from, Int_t to);    
//  Cuts
    void SetPtMin(Float_t ptmin) {fPtMin=ptmin;}
    void SetEtaCut(Float_t etamin, Float_t etamax){fEtaMin=etamin; fEtaMax=etamax;}      Bool_t Selected(GParticle* part);
    Bool_t Selected(GParticle* part1, GParticle* part2);
// Kinematics
    Float_t Mass(GParticle* part1, GParticle* part);
    Float_t PT(GParticle* part1, GParticle* part);
    Float_t Pz(GParticle* part1, GParticle* part);
    Float_t Y(GParticle* part1, GParticle* part);
// Response
    void SmearGauss(Float_t width, Float_t & value);
// Weight
    Bool_t  Correlated(GParticle* part1, GParticle* part2);
    void    SetRate(Float_t rate){fRate1=rate;}
    void    SetRate(Float_t rate1, Float_t rate2 ){fRate1=rate1; fRate2=rate2;}
    Float_t Weight(GParticle* part);
    Float_t Weight(GParticle* part1, GParticle* part);
    Float_t Decay_Prob(GParticle* part);
    
 private:
    void FirstPartner();
    void NextPartner();
    void FirstPartnerSelected();
    void NextPartnerSelected();
    Int_t Origin(GParticle* part);
    GParticle* Parent(GParticle* part);
    GParticle* Partner();
    Int_t Type(GParticle *part){return part->GetKF();}
private:
    TClonesArray *fPartArray;
    Int_t fNParticle;
    Int_t fimuon1;
    Int_t fimuon2;
    Int_t fimin1;
    Int_t fimin2;
    Int_t fimax1;
    Int_t fimax2;
    Float_t fRate1;
    Float_t fRate2;
    GParticle *fmuon1;
    GParticle *fmuon2;
    Float_t fPtMin;
    Float_t fEtaMin;
    Float_t fEtaMax;
  ClassDef(AliDimuCombinator,1) // Dimuon Combinator
};
#endif
Commit	Line	Data
dafbc1c5	1	#ifndef _AliDimuCombinator_H
dafbc1c5	2	#define _AliDimuCombinator_H
fe4da5cc	3	#include "GParticle.h"
	4	#include <TBrowser.h>
	5	#include <TList.h>
	6	#include <TTree.h>
	7	#include <TROOT.h>
	8
	9
dafbc1c5	10	class AliDimuCombinator:
fe4da5cc	11	public TObject
	12	{
	13	public:
dafbc1c5	14	AliDimuCombinator(TClonesArray* Partarray){
fe4da5cc	15	fPartArray=Partarray;
	16	fNParticle=fPartArray->GetEntriesFast();
	17
	18	fimuon1 =0;
	19	fimuon2 =0;
	20	fmuon1 =0;
	21	fmuon2 =0;
	22	fimin1 = 0;
	23	fimin2 = 0;
	24	fimax1 = fNParticle;
	25	fimax2 = fNParticle;
	26	fPtMin =0;
	27	fEtaMin =-10;
	28	fEtaMax =-10;
	29	fRate1=1.;
	30	fRate2=1.;
	31	}
	32	//
7d566a7d	33	// Iterators
7d566a7d	34	// Single muons
fe4da5cc	35	GParticle* FirstMuon();
fe4da5cc	36	GParticle* NextMuon();
7d566a7d	37	// Single muons selected
fe4da5cc	38	GParticle* FirstMuonSelected();
fe4da5cc	39	GParticle* NextMuonSelected();
7d566a7d	40	// Dimuons
fe4da5cc	41	void FirstMuonPair(GParticle* & muon1, GParticle* & muon2);
fe4da5cc	42	void NextMuonPair(GParticle* & muon1, GParticle* & muon2);
7d566a7d	43	// Dimuons selected
fe4da5cc	44	void FirstMuonPairSelected(GParticle* & muon1, GParticle* & muon2);
fe4da5cc	45	void NextMuonPairSelected(GParticle* & muon1, GParticle* & muon2);
7d566a7d	46	// Loop over all prticles
fe4da5cc	47	void ResetRange();
7d566a7d	48	// Set two ranges for dimuon loop
fe4da5cc	49	void SetFirstRange (Int_t from, Int_t to);
	50	void SetSecondRange(Int_t from, Int_t to);
	51	// Cuts
7d566a7d	52	void SetPtMin(Float_t ptmin) {fPtMin=ptmin;}
	53	void SetEtaCut(Float_t etamin, Float_t etamax){fEtaMin=etamin; fEtaMax=etamax;} Bool_t Selected(GParticle* part);
	54	Bool_t Selected(GParticle* part1, GParticle* part2);
fe4da5cc	55	// Kinematics
	56	Float_t Mass(GParticle* part1, GParticle* part);
	57	Float_t PT(GParticle* part1, GParticle* part);
	58	Float_t Pz(GParticle* part1, GParticle* part);
	59	Float_t Y(GParticle* part1, GParticle* part);
	60	// Response
	61	void SmearGauss(Float_t width, Float_t & value);
fe4da5cc	62	// Weight
7d566a7d	63	Bool_t Correlated(GParticle* part1, GParticle* part2);
fe4da5cc	64	void SetRate(Float_t rate){fRate1=rate;}
	65	void SetRate(Float_t rate1, Float_t rate2 ){fRate1=rate1; fRate2=rate2;}
	66	Float_t Weight(GParticle* part);
	67	Float_t Weight(GParticle* part1, GParticle* part);
7d566a7d	68	Float_t Decay_Prob(GParticle* part);
7d566a7d	69
fe4da5cc	70	private:
	71	void FirstPartner();
	72	void NextPartner();
	73	void FirstPartnerSelected();
	74	void NextPartnerSelected();
7d566a7d	75	Int_t Origin(GParticle* part);
7d566a7d	76	GParticle* Parent(GParticle* part);
fe4da5cc	77	GParticle* Partner();
	78	Int_t Type(GParticle *part){return part->GetKF();}
	79	private:
	80	TClonesArray *fPartArray;
	81	Int_t fNParticle;
	82	Int_t fimuon1;
	83	Int_t fimuon2;
	84	Int_t fimin1;
	85	Int_t fimin2;
	86	Int_t fimax1;
	87	Int_t fimax2;
	88	Float_t fRate1;
	89	Float_t fRate2;
	90	GParticle *fmuon1;
	91	GParticle *fmuon2;
	92	Float_t fPtMin;
	93	Float_t fEtaMin;
	94	Float_t fEtaMax;
dafbc1c5	95	ClassDef(AliDimuCombinator,1) // Dimuon Combinator
fe4da5cc	96	};
	97	#endif
	98
	99
	100
	101