StJets.h

// -*- mode: c++;-*-
// $Id: StJets.h,v 1.4 2010/06/30 17:51:58 pibero Exp $
#ifndef StJets_h
#define StJets_h

#include <vector>
using std::vector;

#include "TObject.h"
#include "TClonesArray.h"

#include "TrackToJetIndex.h"
#include "TowerToJetIndex.h"

class StJet;

class StJets : public TObject
{
public:

  StJets();
  virtual ~StJets();
    
  void Clear(bool clearTracks = false);
  void Clear(Option_t* opt);
        
  void setBemcCorrupt(bool v) { mCorrupt = v; }
  bool bemcCorrupt() const { return mCorrupt; }

  int nJets() const { return mJets->GetEntriesFast(); }

  TClonesArray* jets() { return mJets; }

  void addJet(StJet& jet);

  TClonesArray* tracks() { return mTrackToJetIndices; }
  TClonesArray* towers() { return mTowerToJetIndices; }

  void addTrackToIndex(TrackToJetIndex& t2j);
  void addTowerToIndex(TowerToJetIndex& t2j);
    
  TObjArray tracks(int jetIndex) const;
  TObjArray towers(int jetIndex) const;
  vector<TLorentzVector*> particles(int jetIndex) const; // for backward compatibility

  int eventId    () const { return mEventId    ; }
  int eventNumber() const { return mEventNumber; }
  int runId      () const { return mRunId      ; }
  int runNumber  () const { return mRunNumber  ; }

  void seteventId    (int v) { mEventId     = v; }
  void seteventNumber(int v) { mEventNumber = v; }
  void setrunId      (int v) { mRunId       = v; }
  void setrunNumber  (int v) { mRunNumber   = v; }

  //    bool isSameEvent(const StMuDst*);

  int nDylanPoints() const { return mDylanPoints; }
  void setDylanPoints(int v) { mDylanPoints = v; }

  double sumEmcE() const { return mSumEmcE; }
  void setSumEmcE(double v) { mSumEmcE = v; }

  double e     (int jetIndex) const;
  double et    (int jetIndex) const;
  double p     (int jetIndex) const;
  double pt    (int jetIndex) const;
  double phi   (int jetIndex) const;
  double eta   (int jetIndex) const;
  int    nCell (int jetIndex) const;
  int    charge(int jetIndex) const;

private:
  int    mDylanPoints;
  double mSumEmcE;
  int    mEventId;
  int    mEventNumber;
  int    mRunId;
  int    mRunNumber;
  bool   mCorrupt;

  bool inBounds(int);
  StJet* jet(int);
    
  TClonesArray* mJets;
  TClonesArray* mTrackToJetIndices;
  TClonesArray* mTowerToJetIndices;
    
  ClassDef(StJets,2);
};

//inlines
inline void StJets::Clear(Option_t* opt)
{
  TObject::Clear(opt);
  mEventId = mEventNumber = mRunId = mRunNumber = 0;
  mCorrupt = false;
}

#endif // StJets_h