ProtoJet.h

Go to the documentation of this file.

#ifndef JetAlgorithms_ProtoJet_h
#define JetAlgorithms_ProtoJet_h

#include "RecoParticleFlow/PFRootEvent/interface/JetRecoTypes.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Math/interface/LorentzVector.h"

#include <vector>
#include <string>

class ProtoJet {
public:
  typedef JetReco::InputItem Constituent;
  typedef JetReco::InputCollection Constituents;

  typedef math::XYZTLorentzVector LorentzVector;
  ProtoJet();

  //**Constructor. Runs off an array of CaloTower* */
  ProtoJet(const Constituents& theConstituents);

 //**Full Constructor */
  ProtoJet(const LorentzVector& fP4, const Constituents& fConstituents); 

  ~ProtoJet() {}

  void setJetArea (float fArea);
  float jetArea () const;

  void setPileup (float fEnergy);
  float pileup () const;

  void setNPasses (int fPasses);
  int nPasses () const;


  // The Jet four-vector as a true Lorentz vector
  double px() const {return mP4.Px();}
  double py() const {return mP4.Py();}
  double pz() const {return mP4.Pz();}
  double e() const {return mP4.E();}
  double energy() const {return e();}
  
  // Standard quantities derived from the Jet Lorentz vector
  double p() const {return mP4.P();}
  double pt() const {return mP4.Pt();}
  double et() const {return mP4.Et();}
  double m() const {return mP4.M();}
  double phi() const {return mP4.Phi();}
  double eta() const {return mP4.Eta();}
  double y() const {return p() > 0 ? mP4.Rapidity() : 0;}
  int numberOfConstituents() const {return mConstituents.size();};


  const LorentzVector& p4() const {return mP4;}
  
  
  const Constituents& getTowerList();
  Constituents getTowerList() const;

  const Constituents& getPresortedTowerList() const;

  void putTowers(const Constituents& towers);

  void reorderTowers ();

  void calculateLorentzVector() {calculateLorentzVectorERecombination();}
  void calculateLorentzVectorERecombination();
  void calculateLorentzVectorEtRecombination();
  
  


private:
  LorentzVector mP4;
  Constituents mConstituents;
  bool mOrdered;
  float mJetArea;
  float mPileupEnergy;
  int mPassNumber;
};

#endif