TriggerObject.h

Go to the documentation of this file.

#ifndef HLTReco_TriggerObject_h
#define HLTReco_TriggerObject_h

#include "DataFormats/Candidate/interface/Particle.h"
#include <cmath>
#include <vector>

namespace trigger
{

  class TriggerObject {

  private:
    int id_;
    float pt_, eta_, phi_, mass_;

  public:
    TriggerObject(): id_(), pt_(), eta_(), phi_(), mass_() { }
    TriggerObject(int id, float pt, float eta, float phi, float mass):
      id_(id), pt_(pt), eta_(eta), phi_(phi), mass_(mass) { }

    template <typename T>
    TriggerObject(int id, const T& o):
    id_(id), pt_(o.pt()), eta_(o.eta()), phi_(o.phi()), mass_(o.mass()) { }
    template <typename T>
    TriggerObject(const T& o):
    id_(o.pdgId()), pt_(o.pt()), eta_(o.eta()), phi_(o.phi()), mass_(o.mass()) { }

    void setId  (int     id) {id_  =id;}
    void setPt  (float   pt) {pt_  =pt;}
    void setEta (float  eta) {eta_ =eta;}
    void setPhi (float  phi) {phi_ =phi;}
    void setMass(float mass) {mass_=mass;}

    int   id() const {return id_;}
    float pt() const {return pt_;}
    float eta() const {return eta_;}
    float phi() const {return phi_;}
    float mass() const {return mass_;}

    float px() const {return pt_*std::cos(phi_);}
    float py() const {return pt_*std::sin(phi_);}
    float pz() const {return pt_*std::sinh(eta_);}
    float p () const {return pt_*std::cosh(eta_);}
    float energy() const {return std::sqrt(std::pow(mass_,2)+std::pow(p(),2));}
    // et = energy/cosh(eta)
    float et() const {return std::sqrt(std::pow(mass_/std::cosh(eta_),2)+std::pow(pt_,2));}

    reco::Particle particle(reco::Particle::Charge q=0, 
      const reco::Particle::Point & vertex = reco::Particle::Point(0,0,0),
      int status=0, bool integerCharge=true) const {
      return reco::Particle(q,
        reco::Particle::LorentzVector(px(),py(),pz(),energy()),
        vertex,id(),status,integerCharge);
    }

  };


  typedef std::vector<TriggerObject> TriggerObjectCollection;

}

#endif