Project CMSSW displayed by LXR CMSSW_15_1_X_2025-03-21-2300/​TopQuarkAnalysis/​TopHitFit/​interface/​RunHitFit.h	Source navigation Diff markup Identifier search General search
	Version: CMSSW_15_1_X_2025-03-21-2300 CMSSW_15_1_X_2025-03-20-2300 CMSSW_15_1_X_2025-03-19-2300 CMSSW_15_1_X_2025-03-18-2300 CMSSW_15_1_X_2025-03-17-2300 CMSSW_15_1_X_2025-03-16-2300 Revert   Change

File indexing completed on 2024-04-06 12:31:17

 //
 //
 
 /**
     @file RunHitFit.h
 
     @brief Template class of experiment-independent interface to HitFit.
 
     @author Haryo Sumowidagdo <Suharyo.Sumowidagdo@cern.ch>
 
     @par Creation date:
     May 2009.
 
     @par Modification History:
     Nov 2009: Haryo Sumowidagdo <Suharyo.Sumowidagdo@cern.ch>:
     Add doxygen tags for automatic generation of documentation.
  */
 
 #ifndef HITFIT_RUNHITFIT_H
 #define HITFIT_RUNHITFIT_H
 
 #include <algorithm>
 
 #include "TopQuarkAnalysis/TopHitFit/interface/Defaults_Text.h"
 #include "TopQuarkAnalysis/TopHitFit/interface/LeptonTranslatorBase.h"
 #include "TopQuarkAnalysis/TopHitFit/interface/JetTranslatorBase.h"
 #include "TopQuarkAnalysis/TopHitFit/interface/METTranslatorBase.h"
 #include "TopQuarkAnalysis/TopHitFit/interface/Fit_Result.h"
 #include "TopQuarkAnalysis/TopHitFit/interface/Lepjets_Event.h"
 #include "TopQuarkAnalysis/TopHitFit/interface/Top_Fit.h"
 
 // Explanation about the MIN/MAX definitions:
 //
 // For a given number of jets, there is a corresponding number of
 // permutations how to assign each jet in the event to the corresponding
 // parton-level jet.
 // The number of permutations up to 10 jets are given below for Tt and
 // TtH events.
 //
 // NJet         Npermutation (Tt)       Npermutation (TtH)
 // 4            24                      --
 // 5            120                     --
 // 6            360                     360
 // 7            840                     2520
 // 8            1680                    10080
 // 9            3024                    30240
 // 10           5040                    75600
 //
 // The formulas for the number of permutations for Tt and TtH events
 // given n jets in the event are
 //
 //         n!
 // Tt:  -------- ; n >= 4
 //      (n - 4)!
 //
 //          n!
 // TtH: ---------- ; n >= 6
 //      (n - 6)!2!
 //
 // The current MAX settings are chosen for a maximum number of 8 jets
 // Increasing this limit should be done with caution, as it will
 // increase the number of permutations rapidly.
 //
 
 namespace hitfit {
 
   /**
     @brief Template class of experiment-independent interface to HitFit.
     This class is intended to be used inside the programming environment of
     a specific experiment, where each type of physics objects has its
     own class/type.  For using HitFit with generic four-vector classes,
     user can't use this class and have to use the Top_Fit class directly.
     The reason is: this class is designed assuming electron and muon are
     represented by different object type, a situation which is guaranteed to
     happen in any experiments.
     The class contains some static integer constants to limit the maximum
     amount of jets in an event before fitting. See the description of those
     constants for details.  The numbers of permutations for \f$t\bar{t}\f$ and
     \f$t\bar{t}H\f$ as a function of the number of jets
     \f$N_{\mathrm{jet}}\f$ in the event for a few values of are
 
     <center>
     <table border="2">
 
     <tr>
     <th>\f$N_{\mathrm{jet}}\f$</th>
     <th>\f$N_{t\bar{t}}\f$</th>
     <th>\f$N_{t\bar{t}H}\f$</th>
     </tr>
 
     <tr>
     <td align="right">4</td>
     <td align="right">24</td>
     <td align="right">N/A</td>
     </tr>
 
     <tr>
     <td align="right">5</td>
     <td align="right">120</td>
     <td align="right">N/A</td>
     </tr>
 
     <tr>
     <td align="right">6</td>
     <td align="right">360</td>
     <td align="right">360</td>
     </tr>
 
     <tr>
     <td align="right">7</td>
     <td align="right">840</td>
     <td align="right">2520</td>
     </tr>
 
     <tr>
     <td align="right">8</td>
     <td align="right">1680</td>
     <td align="right">20160</td>
     </tr>
 
     </table>
     </center>
 
     If adjusting the limits defined by the static constants is desired, then
     please the following formulas.
 
     The number for possible
     permutations, \f$N_{t\bar{t}}\f$, as a function of number of jets,
     \f$n\f$, for \f$t\bar{t}\f$ event is given by:
     \f[
     N_{t\bar{t}}(n) = \frac{n!}{(n-4)!};~ n \ge 4
     \f]
     The number for possible permutations, \f$N_{t\bar{t}H}\f$, as a
     function of number of jets, \f$n\f$, for \f$t\bar{t}H\f$ is given by:
     \f[
     N_{t\bar{t}}(n) = \frac{n!}{(n-6)!2!};~ n \ge 6
     \f]
 
     @param AElectron The typename of the electron physics object class to
     be translated into HitFit's Lepjets_Event_Lep.
 
     @param AMuon The typename of the muon physics object class to
     be translated into HitFit's Lepjets_Event_Lep.
 
     @param AJet The typename of the jet physics object class to
     be translated into HitFit's Lepjets_Event_Jet.
 
     @param AMet The typename of the missing transverse energy physics
     object class be translated into HitFit's Fourvec.
 
  */
   template <class AElectron, class AMuon, class AJet, class AMet>
   class RunHitFit {
   private:
     /**
        The translator from AElectron to Lepjets_Event_Lep.
      */
     LeptonTranslatorBase<AElectron> _ElectronTranslator;
 
     /**
        The translator from AMuon to Lepjets_Event_Lep.
      */
     LeptonTranslatorBase<AMuon> _MuonTranslator;
 
     /**
        The translator from AJet to Lepjets_Event_Jet.
      */
     JetTranslatorBase<AJet> _JetTranslator;
 
     /**
        The translator from AMet to Fourvec.
      */
     METTranslatorBase<AMet> _METTranslator;
 
     /**
        The internal event.
        The internal event only contains lepton and missing
        transverse energy.
      */
     Lepjets_Event _event;
 
     /**
        The internal array of jets.
        Jets are kept in this array and not added into the internal
        event. The reason is: the jet energy correction applied to a jet
        is dependent on the assumed jet type (b or light) in the
        permutation.  Therefore the decision is to store
        jets in their original format/data type.
 
        Before a fit to a particular permutation is done,
        this class convert the jets in this array into
        Lepjets_Event_Jet format, taking into consideration
        the assumed jet type and applying the appropriate jet energy
        correction.
      */
     std::vector<AJet> _jets;
 
     /**
        Boolean flag which sets whether to use jet resolution
        read from file or jet resolution embedded in the physics objects.
 
        This flag is only set when the FIRST jet is added into the event.
 
        By default this flag is set to FALSE if user does not specify anything
        about which resolution to be used.
      */
     bool _jetObjRes;
 
     /**
        The interface between the event and the fitting algorithm.
      */
     Top_Fit _Top_Fit;
 
     /**
        The array of events with permutation information before fitting.
      */
     std::vector<Lepjets_Event> _Unfitted_Events;
 
     /**
        The results of the kinematic fit.
      */
     std::vector<Fit_Result> _Fit_Results;
 
   public:
     /**
        @brief Constructor.
 
        @param el The function object to translate from AElectron to
        Lepjets_Event_Lep.
 
        @param mu The function object to translate from AMuon to
        Lepjets_Event_Lep.
 
        @param jet The function object to translate from AJet to
        Lepjets_Event_Jet.
 
        @param met The function object to translate from AMet to
        Fourvec.
 
        @param default_file The path of ASCII text files which contains the
        parameter settings for this instance of RunHitFit.
 
        @param lepw_mass The mass to which the leptonic \f$ W- \f$ boson should be
        constrained to.  A value of zero means this constraint will be removed.
 
        @param hadw_mass The mass to which the hadronic \f$ W- \f$ boson should be
        constrained to.  A value of zero means this constraint will be removed.
 
        @param top_mass The mass to which the top quark should be constrained
        to.  A value of zero means this constraint will be removed.
      */
     RunHitFit(const LeptonTranslatorBase<AElectron>& el,
               const LeptonTranslatorBase<AMuon>& mu,
               const JetTranslatorBase<AJet>& jet,
               const METTranslatorBase<AMet>& met,
               const std::string default_file,
               double lepw_mass,
               double hadw_mass,
               double top_mass)
         : _ElectronTranslator(el),
           _MuonTranslator(mu),
           _JetTranslator(jet),
           _METTranslator(met),
           _event(0, 0),
           _jetObjRes(false),
           _Top_Fit(Top_Fit_Args(Defaults_Text(default_file)), lepw_mass, hadw_mass, top_mass) {}
 
     /**
        @brief Destructor.
      */
     ~RunHitFit() {}
 
     /**
        @brief Clear the internal event, fit results, and jets.
      */
     void clear() {
       _event = Lepjets_Event(0, 0);
       _jets.clear();
       _jetObjRes = false;
       _Unfitted_Events.clear();
       _Fit_Results.clear();
     }
 
     /**
        @brief Add one electron into the internal event.
 
        @param electron The electron to be added into the internal event.
 
        @param useObjRes Boolean parameter to indicate if the
        user would like to use the resolution embedded in the object,
        and not the resolution read when instantiating the class.
      */
     void AddLepton(const AElectron& electron, bool useObjRes = false) {
       _event.add_lep(_ElectronTranslator(electron, electron_label, useObjRes));
       return;
     }
 
     /**
        @brief Add one muon into the internal event.
 
        @param muon The muon to be added into the internal event.
 
        @param useObjRes Boolean parameter to indicate if the
        user would like to use the resolution embedded in the object,
        and not the resolution read when instantiating the class.
     */
     void AddLepton(const AMuon& muon, bool useObjRes = false) {
       _event.add_lep(_MuonTranslator(muon, muon_label, useObjRes));
       return;
     }
 
     /**
        @brief Add one jet into the internal event.  This function will
        do nothing if the internal event has already contained the maximally
        allowed number of jets.
 
        Explanation about this function: This function does not directly add
        the jet into the internal event.  Rather, this function store the
        jet in an internal array.
        The reason is: jet energy correction
        and resolution depends on the jet type in the permutation.
        Therefore RunHitFit will only add jet into the event after a specific
        jet permutation has been determined.
        This is done in the FitAllPermutation function().
 
        @param jet The jet to be added into the internal event.
 
        @param useObjRes Boolean parameter to indicate if the
        user would like to use the resolution embedded in the object,
        and not the resolution read when instantiating the class.
     */
     void AddJet(const AJet& jet, bool useObjRes = false) {
       // Only set flag when adding the first jet
       // the additional jets then WILL be treated in the
       // same way like the first jet.
       if (_jets.empty()) {
         _jetObjRes = useObjRes;
       }
 
       if (_jets.size() < MAX_HITFIT_JET) {
         _jets.push_back(jet);
       }
       return;
     }
 
     /**
        @brief Set the missing transverse energy of the internal event.
      */
     void SetMet(const AMet& met, bool useObjRes = false) {
       _event.met() = _METTranslator(met, useObjRes);
       _event.kt_res() = _METTranslator.KtResolution(met, useObjRes);
       return;
     }
 
     /**
        @brief Set the \f$k_{T}\f$ resolution of the internal event.
 
        @param res The resolution.
      */
     void SetKtResolution(const Resolution& res) {
       _event.kt_res() = res;
       return;
     }
 
     /**
        @brief Set the \f$E_{T}\!\!\!\!/\f$ resolution of the internal event.
 
        @param res The \f$E_{T}\!\!\!\!/\f$ resolution, same as \f$k_{T}\f$
        resolution.
      */
     void SetMETResolution(const Resolution& res) {
       SetKtResolution(res);
       return;
     }
 
     /**
        @brief Return a constant reference to the underlying Top_Fit object.
      */
     const Top_Fit& GetTopFit() const { return _Top_Fit; }
 
     /**
        @brief Fit all permutations of the internal event.  Returns the
        number of permutations.
      */
     std::vector<Fit_Result>::size_type FitAllPermutation() {
       if (_jets.size() < MIN_HITFIT_JET) {
         // For ttbar lepton+jets, a minimum of MIN_HITFIT_JETS jets
         // is required
         return 0;
       }
 
       if (_jets.size() > MAX_HITFIT_JET) {
         // Restrict the maximum number of jets in the fit
         // to prevent loop overflow
         return 0;
       }
 
       _Unfitted_Events.clear();
       _Fit_Results.clear();
 
       // Prepare the array of jet types for permutation
       std::vector<int> jet_types(_jets.size(), unknown_label);
       jet_types[0] = lepb_label;
       jet_types[1] = hadb_label;
       jet_types[2] = hadw1_label;
       jet_types[3] = hadw1_label;
 
       if (_Top_Fit.args().do_higgs_flag() && _jets.size() >= MIN_HITFIT_TTH) {
         jet_types[4] = higgs_label;
         jet_types[5] = higgs_label;
       }
 
       std::stable_sort(jet_types.begin(), jet_types.end());
 
       do {
         // begin loop over all jet permutation
         for (int nusol = 0; nusol != 2; nusol++) {
           // loop over two neutrino solution
           bool nuz = bool(nusol);
 
           // Copy the event
           Lepjets_Event fev = _event;
 
           // Add jets into the event, with the assumed type
           // in accord with the permutation.
           // The translator _JetTranslator will correctly
           // return object of Lepjets_Event_Jet with
           // jet energy correction applied in accord with
           // the assumed jet type (b or light).
           for (size_t j = 0; j != _jets.size(); j++) {
             fev.add_jet(_JetTranslator(_jets[j], jet_types[j], _jetObjRes));
           }
 
           // Clone fev (intended to be fitted event)
           // to ufev (intended to be unfitted event)
           Lepjets_Event ufev = fev;
 
           // Set jet types.
           fev.set_jet_types(jet_types);
           ufev.set_jet_types(jet_types);
 
           // Store the unfitted event
           _Unfitted_Events.push_back(ufev);
 
           // Prepare the placeholder for various kinematic quantities
           double umwhad;
           double utmass;
           double mt;
           double sigmt;
           Column_Vector pullx;
           Column_Vector pully;
 
           // Do the fit
           double chisq = _Top_Fit.fit_one_perm(fev, nuz, umwhad, utmass, mt, sigmt, pullx, pully);
           // Store output of the fit
           _Fit_Results.push_back(Fit_Result(chisq, fev, pullx, pully, umwhad, utmass, mt, sigmt));
 
         }  // end loop over two neutrino solution
 
       } while (std::next_permutation(jet_types.begin(), jet_types.end()));
       // end loop over all jet permutations
 
       return _Fit_Results.size();
     }
 
     /**
         @brief Return the unfitted events for all permutations.
      */
     std::vector<Lepjets_Event> GetUnfittedEvent() { return _Unfitted_Events; }
 
     /**
         @brief Return the results of fitting all permutations of the
         internal event.
      */
     std::vector<Fit_Result> GetFitAllPermutation() { return _Fit_Results; }
 
     /**
        Minimum number of jet as input to HitFit in Tt event
      */
     static const unsigned int MIN_HITFIT_JET = 4;
 
     /**
        Minimum number of jet as input to HitFit in TtH event
      */
     static const unsigned int MIN_HITFIT_TTH = 6;
 
     /**
        Maximum number of jet as input to HitFit in each event
      */
     static const unsigned int MAX_HITFIT_JET = 8;
 
     /**
        Maximum number of HitFit permutation in each event.
      */
     static const unsigned int MAX_HITFIT = 1680;
 
     /**
        Maximum number of fitted variables in HitFit in each event
      */
     static const unsigned int MAX_HITFIT_VAR = 32;
   };
 
 }  // namespace hitfit
 
 #endif  // #ifndef RUNHITFIT_H

This page was automatically generated by the 2.2.1 LXR engine. The LXR team