Updates provided by Magali Estienne

[u/mrichter/AliRoot.git] / JETAN / AliJet.h

#ifndef ALIJET_H
#define ALIJET_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
 
//---------------------------------------------------------------------
// Jet class 
// Stores the output of a jet algorithm
// Author: jgcn@mda.cinvestav.mx
//---------------------------------------------------------------------

#include <Riostream.h>
 
#include <TObject.h>
#include <TArrayI.h>
#include <TArrayF.h>
#include <TRefArray.h>

class TClonesArray;
class TLorentzVector;
 
class AliJet : public TObject
{
 public:
 
  AliJet();
  ~AliJet();

  // Getters
  Int_t         GetNinput() const { return fNInput; }
  Int_t         GetNJets() const {return fNJets;}
  TClonesArray* GetJets() const {return fJets;}
  TArrayI       GetInJet() const {return fInJet;}
  TArrayI       GetMultiplicities() const {return fMultiplicities;}
  TArrayI       GetNCells() const {return fNCells;}
  TArrayF       GetPtFromSignal() const {return fPtFromSignal;}
  TArrayF       GetEtaIn() const { return fEtaIn; }
  TArrayF       GetPhiIn() const { return fPhiIn; }
  TArrayF       GetPtIn() const { return fPtIn; }
  TArrayF       GetPtChargedPtCutIn() const { return fPtChPtCutIn; }
  TArrayF       GetEnTotChargedPtCutIn() const {return fEnTotChPtCutIn; }
  TRefArray*    GetTrackRef() const { return fTrackRef;}
  TArrayI       GetDetectorFlagIn() const { return fDetIn; } 
  Double_t      GetEtAvg() const { return fEtAvg; }

  TLorentzVector* GetJet(Int_t i);
  Int_t    GetMultiplicity(Int_t i) const;
  Int_t    GetNCell(Int_t i) const;
  Double_t GetPx(Int_t i);
  Double_t GetPy(Int_t i);
  Double_t GetPz(Int_t i);
  Double_t GetP(Int_t i);
  Double_t GetE(Int_t i);
  Double_t GetPt(Int_t i);
  Double_t GetEta(Int_t i);
  Double_t GetPhi(Int_t i);
  Double_t GetTheta(Int_t i);
  Double_t GetMass(Int_t i);
   
  // Setters
  void SetNinput(Int_t i) {fNInput = i;}
  void AddJet(Double_t px, Double_t py, Double_t pz, Double_t e);
  void SetMultiplicities(Int_t* m);
  void SetNCells(Int_t* n);
  void SetPtFromSignal(Float_t* p);
  void SetEtaIn(Float_t* eta);
  void SetPhiIn(Float_t* phi);
  void SetPtIn(Float_t* pt);
  void SetInJet(Int_t* idx);
  void SetPtChargedPtCutIn(Float_t* pt2T);
  void SetEnTotChargedPtCutIn(Float_t* en2T);
  void SetTrackReferences(TRefArray* ref) {fTrackRef = ref;}
  void SetDetectorFlagIn(Int_t* detT);
  void SetEtAvg(Double_t et) { fEtAvg = et; }
  // others
  Bool_t OutOfRange(Int_t i, const char *s) const;
  void ClearJets(Option_t *option="");
  void PrintJets();

 protected:

  AliJet(const AliJet& rJet);
  AliJet& operator = (const AliJet& rhs);

  Int_t    fNInput;               // number of input objects
  Int_t    fNJets;                // number of jets found
  Double_t fEtAvg;                // average background et per cell

  TArrayI  fInJet;                // i-input object belongs to k-jet 
  TArrayI  fMultiplicities;       // Multiplicity of each jet
  TArrayI  fNCells;               // Number of cells in jet
  TArrayF  fPtFromSignal;         // percentage of pt from signal
  TClonesArray* fJets;            // 4-momenta of jets

  TArrayF  fEtaIn;                // Arrays of input particles kine:Eta
  TArrayF  fPhiIn;                // Arrays of input particles kine:Phi
  TArrayF  fPtIn;                 // Arrays of input particles kine:Pt
  TArrayF  fPtChPtCutIn;          // Arrays of input particles kin:Pt Charged with pt cut
  TArrayF  fEnTotChPtCutIn;       // Arrays of total energy with pt cut on charged + cut min on cell
  TArrayI  fDetIn;                // Arrays of detector type of each UnitArray
  TRefArray* fTrackRef;           //|| Reference to tracks which could belongs to the jet

  ClassDef(AliJet,2)
};
 
#endif