Top_Fit.h

Go to the documentation of this file.

//
//
// 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