Lepjets_Event.h

Go to the documentation of this file.

//
//
// File: hitfit/Lepjets_Event.h
// Purpose: Represent a simple `event' consisting of leptons and jets.
// Created: Jul, 2000, sss, based on run 1 mass analysis code.
//
// An instance of this class holds a list of `leptons' (as represented
// by Lepjets_Event_Lep) and `jets' (as represented by Lepjets_Event_Jet).
// We also record:
//
//   - Missing Et
//   - z-vertex
//   - run and event number
//
// CMSSW File      : interface/Lepjets_Event.h
// Original Author : Scott Stuart Snyder <snyder@bnl.gov> for D0
// Imported to CMSSW by Haryo Sumowidagdo <Suharyo.Sumowidagdo@cern.ch>
//
 
#ifndef HITFIT_LEPJETS_EVENT_H
#define HITFIT_LEPJETS_EVENT_H
 
#include "TopQuarkAnalysis/TopHitFit/interface/Lepjets_Event_Jet.h"
#include "TopQuarkAnalysis/TopHitFit/interface/Lepjets_Event_Lep.h"
#include <vector>
#include <iosfwd>
 
namespace hitfit {
 
  class Lepjets_Event
  //
  // Purpose: Represent a simple `event' consisting of leptons and jets.
  //
  {
  public:
    // Constructor.
    Lepjets_Event(int runnum, int evnum);
 
    // Get the run and event number.
 
    const int& runnum() const;
 
    int& runnum();
 
    const int& evnum() const;
 
    int& evnum();
 
    // Get the length of the lepton and jet lists.
    std::vector<Lepjets_Event_Lep>::size_type nleps() const;
 
    std::vector<Lepjets_Event_Jet>::size_type njets() const;
 
    // Access leptons and jets.
 
    Lepjets_Event_Lep& lep(std::vector<Lepjets_Event_Lep>::size_type i);
 
    Lepjets_Event_Jet& jet(std::vector<Lepjets_Event_Jet>::size_type i);
 
    const Lepjets_Event_Lep& lep(std::vector<Lepjets_Event_Lep>::size_type i) const;
 
    const Lepjets_Event_Jet& jet(std::vector<Lepjets_Event_Jet>::size_type i) const;
 
    // Access missing Et.
    Fourvec& met();
 
    const Fourvec& met() const;
 
    // Access kt resolution.
 
    Resolution& kt_res();
 
    const Resolution& kt_res() const;
 
    // Access the z-vertex.
 
    double zvertex() const;
 
    double& zvertex();
 
    // Access the isMC flag.
    bool isMC() const;
 
    void setMC(bool isMC);
 
    // Access the discriminants.
    double& dlb();
 
    double dlb() const;
 
    double& dnn();
 
    double dnn() const;
 
    // Sum all objects (leptons or jets) with type TYPE.
    Fourvec sum(int type) const;
 
    // Calculate kt --- sum of all objects plus missing Et.
    Fourvec kt() const;
 
    // Add new objects to the event.
    void add_lep(const Lepjets_Event_Lep& lep);
 
    void add_jet(const Lepjets_Event_Jet& jet);
 
    // Smear the objects in the event according to their resolutions.
    void smear(CLHEP::HepRandomEngine& engine, bool smear_dir = false);
 
    // Sort according to pt.
    void sort();
 
    // Get jet types
    std::vector<int> jet_types() const;
 
    // Set jet types
    bool set_jet_types(const std::vector<int>&);
 
    // Remove objects failing pt and eta cuts.
    int cut_leps(double pt_cut, double eta_cut);
 
    int cut_jets(double pt_cut, double eta_cut);
 
    // Remove all but the first N jets.
    void trimjets(std::vector<Lepjets_Event_Jet>::size_type n);
 
    // Dump this object.
    std::ostream& dump(std::ostream& s, bool full = false) const;
 
    std::string jet_permutation() const;
 
  private:
    // The lepton and jet lists.
 
    std::vector<Lepjets_Event_Lep> _leps;
 
    std::vector<Lepjets_Event_Jet> _jets;
 
    // Other event state.
    Fourvec _met;
 
    Resolution _kt_res;
 
    double _zvertex;
 
    bool _isMC;
 
    int _runnum;
 
    int _evnum;
 
    double _dlb;
 
    double _dnn;
  };
 
  // Print the object.
  std::ostream& operator<<(std::ostream& s, const Lepjets_Event& ev);
 
}  // namespace hitfit
 
#endif  // not HITFIT_LEPJETS_EVENT_H