SetPtFromSignal added. (G. Contreras)
[u/mrichter/AliRoot.git] / JETAN / AliJet.h
1 #ifndef ALIJET_H
2 #define ALIJET_H
3  
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6  
7  
8 //---------------------------------------------------------------------
9 // Jet class 
10 // Stores the output of a jet algorithm
11 // Author: jgcn@mda.cinvestav.mx
12 //---------------------------------------------------------------------
13  
14 #include <TObject.h>
15 #include <TArrayI.h>
16 #include <TArrayF.h>
17
18 class TClonesArray;
19 class TLorentzVector;
20  
21 class AliJet : public TObject
22 {
23  public:
24  
25   AliJet();
26   ~AliJet();
27
28   // Getters
29   Int_t GetNinput() const { return fNInput; }
30   Int_t GetNJets() const {return fNJets;}
31   TClonesArray* GetJets() const {return fJets;}
32   TArrayI GetInJet() const {return fInJet;}
33   TArrayI GetMultiplicities() const {return fMultiplicities;}
34   TArrayF GetPtFromSignal() const {return fPtFromSignal;}
35
36   TLorentzVector* GetJet(Int_t i);
37   Int_t GetMultiplicity(Int_t i);
38   Double_t GetPx(Int_t i);
39   Double_t GetPy(Int_t i);
40   Double_t GetPz(Int_t i);
41   Double_t GetP(Int_t i);
42   Double_t GetE(Int_t i);
43   Double_t GetPt(Int_t i);
44   Double_t GetEta(Int_t i);
45   Double_t GetPhi(Int_t i);
46   Double_t GetTheta(Int_t i);
47   Double_t GetMass(Int_t i);
48    
49   // Setters
50   void SetNinput(Int_t i) {fNInput = i;}
51   void AddJet(Double_t px, Double_t py, Double_t pz, Double_t e);
52   void SetInJet(Int_t* j);
53   void SetMultiplicities(Int_t* m);
54   void SetPtFromSignal(Float_t* p);
55
56   // others
57   Bool_t OutOfRange(Int_t i, const char *s) const;
58   void ClearJets(Option_t *option="");
59   void PrintJets();
60
61  protected:
62
63   Int_t fNInput;               // number of input objects
64   Int_t fNJets;                // number of jets found
65   TArrayI fInJet;              // i-input object belongs to k-jet 
66   TArrayI fMultiplicities;     // Multiplicity of each jet
67   TArrayF fPtFromSignal;       // percentage of pt from signal
68   TClonesArray* fJets;         // 4-momenta of jets
69
70   ClassDef(AliJet,1)
71 };
72  
73 #endif