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
  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
  35   TLorentzVector* GetJet(Int_t i);
  36   Int_t GetMultiplicity(Int_t i);
  37   Double_t GetPx(Int_t i);
  38   Double_t GetPy(Int_t i);
  39   Double_t GetPz(Int_t i);
  40   Double_t GetP(Int_t i);
  41   Double_t GetE(Int_t i);
  42   Double_t GetPt(Int_t i);
  43   Double_t GetEta(Int_t i);
  44   Double_t GetPhi(Int_t i);
  45   Double_t GetTheta(Int_t i);
  46   Double_t GetMass(Int_t i);
  47
  48   // Setters
  49   void SetNinput(Int_t i) {fNInput = i;}
  50   void AddJet(Double_t px, Double_t py, Double_t pz, Double_t e);
  51   void SetInJet(Int_t* j);
  52   void SetMultiplicities(Int_t* m);
  53
  54   // others
  55   Bool_t OutOfRange(Int_t i, const char *s) const;
  56   void ClearJets(Option_t *option="");
  57   void PrintJets();
  58
  59  protected:
  60
  61   Int_t fNInput;               // number of input objects
  62   Int_t fNJets;                // number of jets found
  63   TArrayI fInJet;              // i-input object belongs to k-jet
  64   TArrayI fMultiplicities;     // Multiplicity of each jet
  65   TClonesArray* fJets;         // 4-momenta of jets
  66
  67   ClassDef(AliJet,1)
  68 };
  69
  70 #endif