MissingMassTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// vim: ts=2 sw=2
// Local include(s)
#include "DiTauMassTools/MissingMassTool.h"
 
// EDM include(s):
#include "xAODTau/TauJet.h"
 
using namespace DiTauMassTools;
using ROOT::Math::PtEtaPhiMVector;
 
// Default constructor
MissingMassTool::MissingMassTool(const std::string& name) : asg::AsgTool(name)
 
{
  declareProperty("Decorate",           m_decorate=false, "Activate EventInfo decoration");
  declareProperty("FloatStoppingCrit",      m_float_stop=true, "Activate floating stopping criterion");
  declareProperty("CalibSet",           m_calib_set="2019", "Calibration: 2019");
  // default negative. Only set parameter if positive
  // so that the default are in MissingMassCalculator code
  declareProperty("NsigmaMET",          m_n_sigma_met=-1);
  declareProperty("UseTailCleanup",     m_tail_cleanup=-1); 
  declareProperty("UseVerbose",         m_use_verbose=-1);
  declareProperty("NiterFit2",          m_niter_fit_2=-1);
  declareProperty("NiterFit3",          m_niter_fit_3=-1);
  // Property not there in latest MMC tags
  declareProperty("UseTauProbability",      m_use_tau_probability=-1);
  declareProperty("UseMnuProbability",      m_use_mnu_probability=false);
  declareProperty("UseDefaults",        m_use_defaults=-1);
  declareProperty("UseEfficiencyRecovery",  m_use_efficiency_recovery=-1);
  declareProperty("UseMETDphiLL",       m_use_met_param_dphiLL = false);
  declareProperty("ParamFilePath",      m_param_file_path = "MMC_params_v1_fixed.root");
  declareProperty("BeamEnergy",     m_beam_energy = 6500.0);
  declareProperty("LFVLeplepRefit", m_lfv_leplep_refit = true);
  declareProperty("SaveLlhHisto", m_save_llh_histo = false, "Save MMC LLh histograms for debugging purpose. If enabled, it can slow down MMC running time");
}
 
// Copy constructor
MissingMassTool::MissingMassTool(const MissingMassTool& other): asg::AsgTool(other.name() + "_copy") {}
 
 
StatusCode MissingMassTool::initialize()
 
{
  ATH_MSG_INFO("Initialize MissingMassTool");
 
  MMCCalibrationSet::e aset;
 
  if (m_calib_set == "2015HIGHMASS") {
    aset = MMCCalibrationSet::MMC2015HIGHMASS;
  } else if (m_calib_set == "UPGRADE") {
    aset = MMCCalibrationSet::UPGRADE;
  } else if (m_calib_set == "LFV") {
    aset = MMCCalibrationSet::LFVMMC2012;
  } else if (m_calib_set == "2016MC15C") {
    aset = MMCCalibrationSet::MMC2016MC15C;
  } else if (m_calib_set == "2019") {
    aset = MMCCalibrationSet::MMC2019;
  } else {
    return StatusCode::FAILURE;
  }
  m_MMC = new MissingMassCalculator(aset, m_param_file_path);
  // set properties if non negative
  if (!m_float_stop) m_MMC->SetUseFloatStopping(false);
  if (m_n_sigma_met>=0) m_MMC->SetNsigmaMETscan(m_n_sigma_met);
  if (m_tail_cleanup>=0) m_MMC->preparedInput.SetUseTailCleanup(m_tail_cleanup);
  if (m_use_verbose>=0) m_MMC->preparedInput.SetUseVerbose(m_use_verbose);
  if (m_niter_fit_2>=0) m_MMC->SetNiterFit2(m_niter_fit_2);
  if (m_niter_fit_3>=0) m_MMC->SetNiterFit3(m_niter_fit_3);
  if (m_use_tau_probability>=0) m_MMC->Prob->SetUseTauProbability(m_use_tau_probability);
  if (m_use_defaults>=0) m_MMC->preparedInput.SetUseDefaults(m_use_defaults);
  if (m_use_efficiency_recovery>=0) m_MMC->SetUseEfficiencyRecovery(m_use_efficiency_recovery);
  if (m_use_met_param_dphiLL) m_MMC->Prob->SetUseDphiLL(m_use_met_param_dphiLL);
  if (m_use_mnu_probability) m_MMC->Prob->SetUseMnuProbability(m_use_mnu_probability);
  if (m_beam_energy) m_MMC->SetBeamEnergy(m_beam_energy);
  if (!m_lfv_leplep_refit) m_MMC->SetLFVLeplepRefit(false);
  if (m_save_llh_histo) m_MMC->SaveLlhHisto(true);
 
  // could be made a property but maybe not with the enum
  // What about a string argument ?
 
  return StatusCode::SUCCESS;
}
 
StatusCode MissingMassTool::finalize()
 
{
  ATH_MSG_INFO("Finalize MissingMassTool");
  delete m_MMC;
  //delete Output;
 
  return StatusCode::SUCCESS;
}
 
 
// generic method
CP::CorrectionCode MissingMassTool::apply(const xAOD::EventInfo& ei,
    const xAOD::IParticle* part1,
    const xAOD::IParticle* part2,
    const xAOD::MissingET* met,
    const int & njets)
{
  // This is actually where the work is done
  m_MMC->SetEventNumber(ei.eventNumber());
  m_MMC->RunMissingMassCalculator(part1, part2, met, njets);
 
  // Very dry decoration - MET and resonance vectors are retrieved
  // in dedicated method (see MissingMassTool.h)
 
  if (m_decorate) {
    int aFitStatus = m_MMC->OutputInfo.GetFitStatus();
    static const SG::Decorator<int> dec_mmc_fit_status ("mmc_fit_status");
    static const SG::Decorator<double> dec_mmc_maxw_mass ("mmc_maxw_mass");
    static const SG::Decorator<double> dec_mmc_mlm_mass ("mmc_mlm_mass");
    static const SG::Decorator<double> dec_mmc_mlnu3p_mass ("mmc_mlnu3p_mass");
    static const SG::Decorator<PtEtaPhiMVector> dec_mmc_mlnu3p_4vect ("mmc_mlnu3p_4vect");
    dec_mmc_fit_status(ei) = aFitStatus;
    dec_mmc_maxw_mass(ei)  = aFitStatus==1 ? m_MMC->OutputInfo.GetFittedMass(MMCFitMethod::MAXW) : -1;
    dec_mmc_mlm_mass(ei)   = aFitStatus==1 ? m_MMC->OutputInfo.GetFittedMass(MMCFitMethod::MLM) : -1;
    dec_mmc_mlnu3p_mass(ei) = aFitStatus==1 ? m_MMC->OutputInfo.GetFittedMass(MMCFitMethod::MLNU3P) : -1;
    PtEtaPhiMVector null4V(0.,0.,0.,0.);
    dec_mmc_mlnu3p_4vect(ei) = aFitStatus==1 ? m_MMC->OutputInfo.GetResonanceVec(MMCFitMethod::MLNU3P) : null4V;
  }
 
  return CP::CorrectionCode::Ok;
}