Top_Fit.h

Go to the documentation of this file.

//
// $Id: Top_Fit.h,v 1.1 2011/05/26 09:46:53 mseidel Exp $
//
// File: hitfit/Top_Fit.h
// Purpose: Handle jet permutations.
// Created: Jul, 2000, sss, based on run 1 mass analysis code.
//
// CMSSW File      : interface/Top_Fit.h
// Original Author : Scott Stuart Snyder <snyder@bnl.gov> for D0
// Imported to CMSSW by Haryo Sumowidagdo <Suharyo.Sumowidagdo@cern.ch>
//


#ifndef HITFIT_TOP_FIT_H
#define HITFIT_TOP_FIT_H


#include "TopQuarkAnalysis/TopHitFit/interface/Constrained_Top.h"
#include "TopQuarkAnalysis/TopHitFit/interface/matutil.h"
#include <iosfwd>


namespace hitfit {


class Lepjets_Event;
class Fit_Results;


//
// Indices for the various results lists we make in Fit_Results.
//
enum Lists
{
  all_list = 0,         // All events.
  noperm_list = 1,      // All jet assignments are correct.
  semicorrect_list = 2, // Jets assigned to the correct top.
  limited_isr_list = 3, // Top three jets are not ISR.
  topfour_list = 4,     // Top four jets are not ISR, any other jets are ISR.
  btag_list = 5,        // All tagged jets were assigned as b-jets.
  htag_list = 6,        // All tagged jets were assigned as b-jets or higgs.
  n_lists = 7
};


//*************************************************************************


class Top_Fit_Args
//
// Purpose: Hold on to parameters for the Top_Fit class.
//
//   bool print_event_flag - If true, print the event after the fit.
//   bool do_higgs_flag    - If true, fit ttH events.
//   double jet_mass_cut   - Reject events with jet masses larger than this.
//   double mwhad_min_cut  - Reject events with hadronic W mass before
//                           fitting smaller than this.
//   double mwhad_max_cut  - Reject events with hadronic W mass before
//                           fitting larger than this.
//   double mtdiff_max_cut - Reject events where the difference between
//                           leptonic and hadronic top masses before fitting
//                           is larger than this.
//   int nkeep             - Number of results to keep in each list.
//   bool solve_nu_tmass   - If true, use hadronic top mass to constrain
//                           the neutrino pz.  Otherwise constrain the
//                           neutrino + lepton mass to W mass.
//
{
public:
  // Constructor.  Initialize from a Defaults object.
  Top_Fit_Args (const Defaults& defs);

  // Retrieve parameter values.
  bool print_event_flag () const;

  bool do_higgs_flag () const;

  double jet_mass_cut () const;

  double mwhad_min_cut () const;

  double mwhad_max_cut () const;

  double mtdiff_max_cut () const;

  int nkeep () const;

  bool solve_nu_tmass() const;

  // Arguments for subobjects.
  const Constrained_Top_Args& constrainer_args () const;


private:
  // Hold on to parameter values.
  bool _print_event_flag;

  bool _do_higgs_flag;

  double _jet_mass_cut;

  double _mwhad_min_cut;

  double _mwhad_max_cut;

  double _mtdiff_max_cut;

  int _nkeep;

  bool _solve_nu_tmass;

  Constrained_Top_Args _args;

};


//*************************************************************************

class Top_Fit
//
// Purpose: Handle jet permutations.
//
{
public:
  // Constructor.
  // LEPW_MASS, HADW_MASS, and TOP_MASS are the masses to which
  // those objects should be constrained.  To remove a constraint,
  // set the mass to 0.
  Top_Fit (const Top_Fit_Args& args,
           double lepw_mass,
           double hadw_mass,
           double top_mass);

  // Fit a single jet permutation.  Return the results for that fit.
  double fit_one_perm (Lepjets_Event& ev,
                       bool& nuz,
                       double& umwhad,
                       double& utmass,
                       double& mt,
                       double& sigmt,
                       Column_Vector& pullx,
                       Column_Vector& pully);

  // Fit all jet permutations in EV.
  Fit_Results fit (const Lepjets_Event& ev);

  // Print.
  friend std::ostream& operator<< (std::ostream& s, const Top_Fit& fitter);

  const Top_Fit_Args& args() const;

private:
  // The object state.
  const Top_Fit_Args _args;
  Constrained_Top _constrainer;
  double _lepw_mass;
  double _hadw_mass;
};


} // namespace hitfit


#endif // not HITFIT_TOP_FIT_H