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