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