/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_7/src/TopQuarkAnalysis/TopHitFit/src/Top_Fit.cc

Go to the documentation of this file.

//
// $Id: Top_Fit.cc,v 1.1 2011/05/26 09:47:00 mseidel Exp $
//
// File: src/Top_Fit.cc
// Purpose: Handle jet permutations.
// Created: Jul, 2000, sss, based on run 1 mass analysis code.
//
// XXX handle merging jets.
// XXX btagging for ttH.
//
// CMSSW File      : src/Top_Fit.cc
// Original Author : Scott Stuart Snyder <snyder@bnl.gov> for D0
// Imported to CMSSW by Haryo Sumowidagdo <Suharyo.Sumowidagdo@cern.ch>
//


#include "TopQuarkAnalysis/TopHitFit/interface/Top_Fit.h"
#include "TopQuarkAnalysis/TopHitFit/interface/Lepjets_Event.h"
#include "TopQuarkAnalysis/TopHitFit/interface/Top_Decaykin.h"
#include "TopQuarkAnalysis/TopHitFit/interface/Defaults.h"
#include "TopQuarkAnalysis/TopHitFit/interface/Fit_Results.h"
#include "TopQuarkAnalysis/TopHitFit/interface/fourvec.h"
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cassert>

using std::cout;
using std::endl;
using std::abs;
using std::next_permutation;
using std::stable_sort;
using std::vector;
using std::ostream;


namespace hitfit {


//*************************************************************************
// Argument handling.
//


Top_Fit_Args::Top_Fit_Args (const Defaults& defs)
//
// Purpose: Constructor.
//
// Inputs:
//   defs -        The Defaults instance from which to initialize.
//
  : _print_event_flag (defs.get_bool ("print_event_flag")),
    _do_higgs_flag (defs.get_bool ("do_higgs_flag")),
    _jet_mass_cut (defs.get_float ("jet_mass_cut")),
    _mwhad_min_cut (defs.get_float ("mwhad_min_cut")),
    _mwhad_max_cut (defs.get_float ("mwhad_max_cut")),
    _mtdiff_max_cut (defs.get_float ("mtdiff_max_cut")),
    _nkeep (defs.get_int ("nkeep")),
    _solve_nu_tmass (defs.get_bool ("solve_nu_tmass")),
    _args (defs)
   {
}


bool Top_Fit_Args::print_event_flag () const
//
// Purpose: Return the print_event_flag parameter.
//          See the header for documentation.
//
{
  return _print_event_flag;
}


bool Top_Fit_Args::do_higgs_flag () const
//
// Purpose: Return the do_higgs_flag parameter.
//          See the header for documentation.
//
{
  return _do_higgs_flag;
}


double Top_Fit_Args::jet_mass_cut () const
//
// Purpose: Return the jet_mass_cut parameter.
//          See the header for documentation.
//
{
  return _jet_mass_cut;
}


double Top_Fit_Args::mwhad_min_cut () const
//
// Purpose: Return the mwhad_min_cut parameter.
//          See the header for documentation.
//
{
  return _mwhad_min_cut;
}


double Top_Fit_Args::mwhad_max_cut () const
//
// Purpose: Return the mwhad_max_cut parameter.
//          See the header for documentation.
//
{
  return _mwhad_max_cut;
}


double Top_Fit_Args::mtdiff_max_cut () const
//
// Purpose: Return the mtdiff_max_cut parameter.
//          See the header for documentation.
//
{
  return _mtdiff_max_cut;
}


int Top_Fit_Args::nkeep () const
//
// Purpose: Return the nkeep parameter.
//          See the header for documentation.
//
{
  return _nkeep;
}


bool Top_Fit_Args::solve_nu_tmass () const
//
// Purpose: Return the solve_nu_tmass parameter
//          See the header for documentation.
//
{
  return _solve_nu_tmass;
}


const Constrained_Top_Args& Top_Fit_Args::constrainer_args () const
//
// Purpose: Return the contained subobject parameters.
//
{
  return _args;
}



//*************************************************************************
// Helper functions.
//


namespace {


bool test_for_bad_masses (const Lepjets_Event& ev,
                          const Top_Fit_Args& args,
                          double mwhad,
                          double umthad,
                          double umtlep)
//
// Purpose: Apply mass cuts to see if this event should be rejected
//          without fitting.
//
// Inputs:
//   ev -          The event to test.
//   args -        Parameter setting.
//   mwhad -       The hadronic W mass.
//   umthad -      The hadronic top mass.
//   umtlep -      The leptonic top mass.
//
// Returns:
//   True if the event should be rejected.
//
{

  // Reject the event if any jet's mass is too large.
  if (ev.sum (lepb_label).m()  > args.jet_mass_cut() ||
      ev.sum (hadb_label).m()  > args.jet_mass_cut() ||
      ev.sum (hadw1_label).m() > args.jet_mass_cut() ||
      ev.sum (hadw2_label).m() > args.jet_mass_cut()) {
      return true;
  }

  // Reject if if the hadronic W mass is outside the window.
  if (mwhad < args.mwhad_min_cut()) {
      return true;
  }

  // Reject if if the hadronic W mass is outside the window.
  if (mwhad > args.mwhad_max_cut()) {
      return true;
  }

  // And if the two top masses are too far apart.
  if (abs (umthad - umtlep) > args.mtdiff_max_cut()) {
      return true;
  }

  // It's ok.
  return false;
}


vector<int> classify_jetperm (const vector<int>& jet_types,
                              const Lepjets_Event& ev)
//
// Purpose: Classify a jet permutation, to decide on what result
//          lists it should be put.
//
// Inputs:
//   jet_types -   Vector of jet types.
//   ev -          The original event being fit.
//
// Returns:
//   A list_flags vector, appropriate to pass to Fit_Results::push.
//
{
  // Start by assuming it's on all the lists.
  // We'll clear the flags if we see that it actually doesn't
  // belong.
  vector<int> out (n_lists);
  out[all_list] = 1;
  out[noperm_list] = 1;
  out[semicorrect_list] = 1;
  out[limited_isr_list] = 1;
  out[topfour_list] = 1;
  out[btag_list] = 1;
  out[htag_list] = 1;

  // Loop over jets.
  assert (jet_types.size() == ev.njets());
  for (vector<int>::size_type i=0; i < jet_types.size(); i++)
  {
    {
      int t1 =   jet_types[i];    // Current type of this jet.
      int t2 = ev.jet(i).type();  // `Correct' type of this jet.

      // Consider hadw1_label and hadw2_label the same.
      if (t1 == hadw2_label) t1 = hadw1_label;
      if (t2 == hadw2_label) t2 = hadw1_label;

      // If they're not the same, the permutation isn't correct.
      if (t1 != t2) out[noperm_list] = 0;

      // Test for a semicorrect permutation.
      // Here, all hadronic-side jets are considered equivalent.
      if (t1 == hadw1_label) t1 = hadb_label;
      if (t2 == hadw1_label) t2 = hadb_label;
      if (t1 != t2) out[semicorrect_list] = 0;
    }

    if (jet_types[i] == isr_label && i <= 2)
      out[limited_isr_list] = 0;

    if ((jet_types[i] == isr_label && i <= 3) ||
        (jet_types[i] != isr_label && i >= 4))
      out[topfour_list] = 0;

    if ((ev.jet(i).svx_tag() || ev.jet(i).slt_tag()) &&
        ! (jet_types[i] == hadb_label || jet_types[i] == lepb_label))
      out[btag_list] = 0;

    if ((ev.jet(i).svx_tag() || ev.jet(i).slt_tag()) &&
        ! (jet_types[i] == hadb_label  || jet_types[i] == lepb_label ||
           jet_types[i] == higgs_label))
      out[htag_list] = 0;
  }
  return out;
}


void set_jet_types (const vector<int>& jet_types,
                    Lepjets_Event& ev)
//
// Purpose: Update EV with a new set of jet types.
//
// Inputs:
//   jet_types -   Vector of new jet types.
//   ev -          The event to update.
//
// Outputs:
//   ev -          The updated event.
//
{
  assert (ev.njets() == jet_types.size());
  bool saw_hadw1 = false;
  for (vector<int>::size_type i=0; i < ev.njets(); i++) {
    int t = jet_types[i];
    if (t == hadw1_label) {
      if (saw_hadw1)
        t = hadw2_label;
      saw_hadw1 = true;
    }
    ev.jet (i).type() = t;
  }
}


} // unnamed namespace


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


Top_Fit::Top_Fit (const Top_Fit_Args& args,
                  double lepw_mass,
                  double hadw_mass,
                  double top_mass)
//
// Purpose: Constructor.
//
// Inputs:
//   args -        The parameter settings for this instance.
//   lepw_mass -   The mass to which the leptonic W should be constrained,
//                 or 0 to skip this constraint.
//   hadw_mass -   The mass to which the hadronic W should be constrained,
//                 or 0 to skip this constraint.
//   top_mass -    The mass to which the top quarks should be constrained,
//                 or 0 to skip this constraint.
//
  : _args (args),
    _constrainer (args.constrainer_args(),
                  lepw_mass, hadw_mass, top_mass),
    _lepw_mass(lepw_mass),
    _hadw_mass (hadw_mass)
{
}


double Top_Fit::fit_one_perm (Lepjets_Event& ev,
                              bool& nuz,
                              double& umwhad,
                              double& utmass,
                              double& mt,
                              double& sigmt,
                              Column_Vector& pullx,
                              Column_Vector& pully)
//
// Purpose: Fit a single jet permutation.
//
// Inputs:
//   ev -          The event to fit.
//                 The object labels must have already been assigned.
//   nuz -         Boolean flag to indicate which neutrino solution to be
//                 used.
//                 false = use smaller neutrino z solution
//                 true  = use larger neutrino z solution
//
// Outputs:
//   ev-           The event after the fit.
//   umwhad -      Hadronic W mass before fitting.
//   utmass -      Top mass before fitting, averaged from both sides.
//   mt -          Top mass after fitting.
//   sigmt -       Top mass uncertainty after fitting.
//   pullx -       Vector of pull quantities for well-measured variables.
//   pully -       Vector of pull quantities for poorly-measured variables.
//
// Returns:
//   The fit chisq, or < 0 if the fit didn't converge.
//
// Adaptation note by Haryo Sumowidagdo:
//   This function is rewritten in order to make its purpose reflects
//   the function's name.  The function nows only fit one jet permutation
//   with one neutrino solution only.
//
//
{
  mt = 0;
  sigmt = 0;

  // Find the neutrino solutions by requiring either:
  // 1) that the leptonic top have the same mass as the hadronic top.
  // 2) that the mass of the lepton and neutrino is equal to the W mass

  umwhad = Top_Decaykin::hadw (ev) . m();
  double umthad = Top_Decaykin::hadt (ev) . m();
  double nuz1, nuz2;

  if (_args.solve_nu_tmass()) {
      Top_Decaykin::solve_nu_tmass (ev, umthad, nuz1, nuz2);
  }
  else {
      Top_Decaykin::solve_nu (ev, _lepw_mass, nuz1, nuz2);
  }

  // Set up to use the selected neutrino solution
  if (!nuz) {
      ev.met().setZ(nuz1);
  }
  else {
      ev.met().setZ(nuz2);
  }

  // Note: We have set the neutrino Pz, but we haven't set the neutrino energy.
  // Remember that originally the neutrino energy was equal to
  // sqrt(nu_px*nu_px + nu_py*nu_py).  Calculating the invariant mass squared
  // for the neutrino will give negative mass squared.
  // Therefore we need to adjust (increase) the neutrino energy in order to
  // make its mass remain zero.

  adjust_e_for_mass(ev.met(),0);

  // Find the unfit top mass as the average of the two sides.
  double umtlep = Top_Decaykin::lept (ev) . m();
  utmass = (umthad + umtlep) / 2;

  // Trace, if requested.
  if (_args.print_event_flag()) {
    cout << "Top_Fit::fit_one_perm() : Before fit:\n";
    Top_Decaykin::dump_ev (cout, ev);
  }

  // Maybe reject this event.
  if (_hadw_mass > 0 && test_for_bad_masses (ev, _args, umwhad,
                                             umthad, umtlep))
  {
    cout << "Top_Fit: bad mass comb.\n";
    return -999;
  }

  // Do the fit.
  double chisq = _constrainer.constrain (ev, mt, sigmt, pullx, pully);

  // Trace, if requested.
  if (_args.print_event_flag()) {
    cout << "Top_Fit::fit_one_perm() : After fit:\n";
    cout << "chisq: " << chisq << " mt: " << mt << " ";
    Top_Decaykin::dump_ev (cout, ev);
  }

  // Done!
  return chisq;
}


Fit_Results Top_Fit::fit (const Lepjets_Event& ev)
//
// Purpose: Fit all jet permutations for EV.
//
// Inputs:
//   ev -          The event to fit.
//
// Returns:
//   The results of the fit.
//
{
  // Make a new Fit_Results object.
  Fit_Results res (_args.nkeep(), n_lists);

  // Set up the vector of jet types.
  vector<int> jet_types (ev.njets(), isr_label);
  assert (ev.njets() >= 4);
  jet_types[0] = lepb_label;
  jet_types[1] = hadb_label;
  jet_types[2] = hadw1_label;
  jet_types[3] = hadw1_label;

  if (_args.do_higgs_flag() && ev.njets() >= 6) {
    jet_types[4] = higgs_label;
    jet_types[5] = higgs_label;
  }

  // Must be in sorted order.
  stable_sort (jet_types.begin(), jet_types.end());

  do {

    // Loop over the two possible neutrino solution
    for (int nusol = 0 ; nusol != 2 ; nusol++) {

    // Set up the neutrino solution to be used
    bool nuz = bool(nusol);

    // Copy the event.
    Lepjets_Event fev = ev;

    // Install the new jet types.
    set_jet_types (jet_types, fev);

    // Figure out on what lists this permutation should go.
    vector<int> list_flags = classify_jetperm (jet_types, ev);

    // Set up the output variables for fit results.
    double umwhad, utmass, mt, sigmt;
    Column_Vector pullx;
    Column_Vector pully;
    double chisq;

    // Tracing.
    cout << "Top_Fit::fit(): Before fit: (";
    for (vector<int>::size_type i=0; i < jet_types.size(); i++) {
        if (i) cout << " ";
        cout << jet_types[i];
    }
    cout << " nuz = " << nuz ;
    cout << ") " << std::endl;

    // Do the fit.
    chisq = fit_one_perm (fev, nuz, umwhad, utmass, mt, sigmt, pullx, pully);

    // Print the result, if requested.
    if (_args.print_event_flag()) {
        cout << "Top_Fit::fit(): After fit:\n";
        char buf[256];
        sprintf (buf, "chisq: %8.3f  mt: %6.2f pm %5.2f %c\n",
             chisq, mt, sigmt, (list_flags[noperm_list] ? '*' : ' '));
        cout << buf;
    }

    // Add it to the results.
    res.push (chisq, fev, pullx, pully, umwhad, utmass, mt, sigmt, list_flags);

    } // end of for loop over the two neutrino solution

    // Step to the next permutation.
  } while (next_permutation (jet_types.begin(), jet_types.end()));

  return res;
}


std::ostream& operator<< (std::ostream& s, const Top_Fit& fitter)
//
// Purpose: Print the object to S.
//
// Inputs:
//   s -           The stream to which to write.
//   fitter -      The object to write.
//
// Returns:
//   The stream S.
//
{
  return s << fitter._constrainer;
}


const Top_Fit_Args& Top_Fit::args() const
{
    return _args;
}

} // namespace hitfit