Lepjets_Event.h

Go to the documentation of this file.

//
// $Id: Lepjets_Event.h,v 1.1 2011/05/26 09:46:53 mseidel Exp $
//
// 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