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 /* $Id$ */
   8
   9 //---------------------------------------------------------------------
  10 // Jet class
  11 // Stores the output of a jet algorithm
  12 // Author: jgcn@mda.cinvestav.mx
  13 //---------------------------------------------------------------------
  14
  15 #include <TObject.h>
  16 #include <TArrayI.h>
  17 #include <TArrayF.h>
  18
  19 class TClonesArray;
  20 class TLorentzVector;
  21
  22 class AliJet : public TObject
  23 {
  24  public:
  25
  26   AliJet();
  27   ~AliJet();
  28
  29   // Getters
  30   Int_t GetNinput() const {return fNInput;}
  31   Int_t GetNJets() const {return fNJets;}
  32   TClonesArray* GetJets() const {return fJets;}
  33   TArrayI GetInJet() const {return fInJet;}
  34   TArrayI GetMultiplicities() const {return fMultiplicities;}
  35   TArrayI GetNCells() const {return fNCells;}
  36   TArrayF GetPtFromSignal() const {return fPtFromSignal;}
  37   TArrayF GetEtaIn() const {return fEtaIn;}
  38   TArrayF GetPhiIn() const {return fPhiIn;}
  39   TArrayF GetPtIn() const {return fPtIn;}
  40   Double_t GetEtAvg() const {return fEtAvg;}
  41
  42   TLorentzVector* GetJet(Int_t i);
  43   Int_t GetMultiplicity(Int_t i) const;
  44   Int_t GetNCell(Int_t i) const;
  45   Double_t GetPx(Int_t i);
  46   Double_t GetPy(Int_t i);
  47   Double_t GetPz(Int_t i);
  48   Double_t GetP(Int_t i);
  49   Double_t GetE(Int_t i);
  50   Double_t GetPt(Int_t i);
  51   Double_t GetEta(Int_t i);
  52   Double_t GetPhi(Int_t i);
  53   Double_t GetTheta(Int_t i);
  54   Double_t GetMass(Int_t i);
  55
  56   // Setters
  57   void SetNinput(Int_t i) {fNInput = i;}
  58   void AddJet(Double_t px, Double_t py, Double_t pz, Double_t e);
  59   void SetMultiplicities(Int_t* m);
  60   void SetNCells(Int_t* n);
  61   void SetPtFromSignal(Float_t* p);
  62   void SetEtaIn(Float_t* eta);
  63   void SetPhiIn(Float_t* phi);
  64   void SetPtIn(Float_t* pt);
  65   void SetInJet(Int_t* idx);
  66   void SetEtAvg(Double_t et) { fEtAvg = et; }
  67
  68   // others
  69   Bool_t OutOfRange(Int_t i, const char *s) const;
  70   void ClearJets(Option_t *option="");
  71   void PrintJets();
  72
  73  protected:
  74
  75   AliJet(const AliJet& rJet);
  76   AliJet& operator = (const AliJet& rhs);
  77
  78   Int_t fNInput;               // number of input objects
  79   Int_t fNJets;                // number of jets found
  80   Double_t fEtAvg;             // average background et per cell
  81
  82   TArrayI fInJet;              // i-input object belongs to k-jet
  83   TArrayI fMultiplicities;     // Multiplicity of each jet
  84   TArrayI fNCells;             // Number of cells in jet
  85   TArrayF fPtFromSignal;       // percentage of pt from signal
  86   TClonesArray* fJets;         // 4-momenta of jets
  87
  88   TArrayF fEtaIn;              // Arrays of input particles kine:Eta
  89   TArrayF fPhiIn;              // Arrays of input particles kine:Phi
  90   TArrayF fPtIn;               // Arrays of input particles kine:Pt
  91
  92   ClassDef(AliJet,1)
  93 };
  94
  95 #endif