TOPModuleT0LLCollectorModule.cc

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#include <top/modules/collectors/TOPModuleT0LLCollectorModule.h>
#include <top/geometry/TOPGeometryPar.h>
#include <top/reconstruction/TOPreco.h>
#include <top/reconstruction/TOPtrack.h>
#include <top/reconstruction/TOPconfigure.h>
 
// framework aux
#include <framework/gearbox/Const.h>
#include <framework/logging/Logger.h>
 
// root
#include <TRandom.h>
#include <TH1F.h>
#include <TH1D.h>
#include <TH2F.h>
 
using namespace std;
 
namespace Belle2 {
  using namespace TOP;
 
  //-----------------------------------------------------------------
  //                 Register module
  //-----------------------------------------------------------------
 
  REG_MODULE(TOPModuleT0LLCollector)
 
  //-----------------------------------------------------------------
  //                 Implementation
  //-----------------------------------------------------------------
 
  TOPModuleT0LLCollectorModule::TOPModuleT0LLCollectorModule()
  {
    // set module description and processing properties
    setDescription("A collector for module T0 calibration with neg. log likelihood "
                   "minimization (method LL). Aimed for the final (precise) calibration"
                   "after initial one with DeltaT method.");
    setPropertyFlags(c_ParallelProcessingCertified);
 
    // module parameters
    addParam("numBins", m_numBins, "number of bins of the search region", 200);
    addParam("timeRange", m_timeRange,
             "time range in which to search for the minimum [ns]", 10.0);
    addParam("minBkgPerBar", m_minBkgPerBar,
             "minimal number of background photons per module", 0.0);
    addParam("scaleN0", m_scaleN0,
             "Scale factor for figure-of-merit N0", 1.0);
    addParam("sigmaSmear", m_sigmaSmear,
             "sigma in [ns] for additional smearing of PDF", 0.0);
    addParam("sample", m_sample,
             "sample type: one of dimuon or bhabha", std::string("dimuon"));
    addParam("deltaEcms", m_deltaEcms,
             "c.m.s energy window (half size) if sample is dimuon or bhabha", 0.1);
    addParam("dr", m_dr, "cut on POCA in r", 2.0);
    addParam("dz", m_dz, "cut on POCA in abs(z)", 4.0);
    addParam("minZ", m_minZ,
             "minimal local z of extrapolated hit", -130.0);
    addParam("maxZ", m_maxZ,
             "maximal local z of extrapolated hit", 130.0);
    addParam("pdfOption", m_pdfOption,
             "PDF option, one of 'rough', 'fine', 'optimal'", std::string("rough"));
 
  }
 
 
  void TOPModuleT0LLCollectorModule::prepare()
  {
    // input collections
 
    m_digits.isRequired();
    m_tracks.isRequired();
    m_extHits.isRequired();
    m_recBunch.isRequired();
 
    // Configure TOP detector for reconstruction
 
    TOPconfigure config;
 
    // Parse PDF option
 
    if (m_pdfOption == "rough") {
      m_PDFOption = TOPreco::c_Rough;
    } else if (m_pdfOption == "fine") {
      m_PDFOption = TOPreco::c_Fine;
    } else if (m_pdfOption == "optimal") {
      m_PDFOption = TOPreco::c_Optimal;
    } else {
      B2ERROR("Unknown PDF option '" << m_pdfOption << "'");
    }
 
    // set track selector
 
    if (m_sample == "dimuon" or m_sample == "bhabha") {
      m_selector = TrackSelector(m_sample);
      m_selector.setDeltaEcms(m_deltaEcms);
      m_selector.setCutOnPOCA(m_dr, m_dz);
      m_selector.setCutOnLocalZ(m_minZ, m_maxZ);
    } else {
      B2ERROR("Invalid sample type '" << m_sample << "'");
    }
 
    // create and register histograms
 
    double tmin = -m_timeRange / 2;
    double tmax =  m_timeRange / 2;
    for (unsigned i = 0; i < c_numSets; i++) {
      for (int slot = 1; slot <= c_numModules; slot++) {
        string name = "chi2_set" + to_string(i) + "_slot" + to_string(slot);
        string title = "chi2 scan, slot" + to_string(slot) + "; t0 [ns]; chi2";
        auto h = new TH1D(name.c_str(), title.c_str(),  m_numBins, tmin, tmax);
        registerObject<TH1D>(name, h);
        m_names[i].push_back(name);
      }
    }
 
    auto h1 = new TH2F("tracks_per_slot", "tracks per slot and sample",
                       c_numModules, 0.5, c_numModules + 0.5, c_numSets, 0, c_numSets);
    h1->SetXTitle("slot number");
    h1->SetYTitle("sample number");
    registerObject<TH2F>("tracks_per_slot", h1);
 
    auto h2 = new TH1F("numHits", "Number of photons per slot",
                       c_numModules, 0.5, c_numModules + 0.5);
    h2->SetXTitle("slot number");
    h2->SetYTitle("hits per slot");
    registerObject<TH1F>("numHits", h2);
 
    auto h3 = new TH2F("timeHits", "Photon times vs. boardstacks",
                       c_numModules * 4, 0.5, c_numModules + 0.5, 200, 0.0, 20.0);
    h3->SetXTitle("slot number");
    h3->SetYTitle("time [ns]");
    registerObject<TH2F>("timeHits", h3);
 
    const auto* geo = TOPGeometryPar::Instance()->getGeometry();
    double bunchTimeSep = geo->getNominalTDC().getSyncTimeBase() / 24;
    auto h4 = new TH1F("offset", "current offset from input files; offset [ns]",
                       200, -bunchTimeSep / 2, bunchTimeSep / 2);
    registerObject<TH1F>("offset", h4);
 
  }
 
 
  void TOPModuleT0LLCollectorModule::collect()
  {
    // bunch must be reconstructed
 
    if (not m_recBunch.isValid()) return;
    if (not m_recBunch->isReconstructed()) return;
 
    // create reconstruction object and set various options
 
    int Nhyp = 1;
    double mass = m_selector.getChargedStable().getMass();
    int pdg = m_selector.getChargedStable().getPDGCode();
    TOPreco reco(Nhyp, &mass, &pdg, m_minBkgPerBar, m_scaleN0);
    reco.setPDFoption(m_PDFOption);
    const auto& tdc = TOPGeometryPar::Instance()->getGeometry()->getNominalTDC();
    double timeMin = tdc.getTimeMin();
    double timeMax = tdc.getTimeMax();
 
    // add photon hits to reconstruction object with time corrected-back for module T0
 
    for (const auto& digit : m_digits) {
      if (digit.getHitQuality() == TOPDigit::c_Good) {
        double t = digit.getTime();
        auto slot = digit.getModuleID();
        if (digit.isModuleT0Calibrated()) t += m_moduleT0->getT0(slot);
        if (digit.hasStatus(TOPDigit::c_BunchOffsetSubtracted)) {
          t += m_recBunch->getAverageOffset();
        }
        reco.addData(digit.getModuleID(), digit.getPixelID(), t, digit.getTimeError());
      }
    }
 
    // loop over reconstructed tracks, make a selection and accumulate log likelihoods
 
    int ntra = 0;
    for (const auto& track : m_tracks) {
 
      // track selection
      TOPtrack trk(&track);
      if (not trk.isValid()) continue;
 
      if (not m_selector.isSelected(trk)) continue;
 
      // run reconstruction
      reco.reconstruct(trk);
      if (reco.getFlag() != 1) continue; // track is not in the acceptance of TOP
 
      // minimization procedure: accumulate
      int sub = gRandom->Integer(c_numSets); // generate sub-sample number
      unsigned module = trk.getModuleID() - 1;
      if (module >= m_names[sub].size()) continue;
      auto h = getObjectPtr<TH1D>(m_names[sub][module]);
      for (int ibin = 0; ibin < h->GetNbinsX(); ibin++) {
        double t0 = h->GetBinCenter(ibin + 1);
        double chi = h->GetBinContent(ibin + 1);
        chi += -2 * reco.getLogL(t0, timeMin, timeMax, m_sigmaSmear);
        h->SetBinContent(ibin + 1, chi);
      }
      auto h1 = getObjectPtr<TH2F>("tracks_per_slot");
      h1->Fill(trk.getModuleID(), sub);
 
      // fill histograms of hits
      auto h2 = getObjectPtr<TH1F>("numHits");
      auto h3 = getObjectPtr<TH2F>("timeHits");
      for (const auto& digit : m_digits) {
        if (digit.getHitQuality() != TOPDigit::c_Good) continue;
        if (digit.getModuleID() != trk.getModuleID()) continue;
        if (digit.getTime() < timeMin) continue;
        if (digit.getTime() > timeMax) continue;
        h2->Fill(digit.getModuleID());
        int bs = digit.getBoardstackNumber();
        h3->Fill((digit.getModuleID() * 4 + bs - 1.5) / 4.0 , digit.getTime());
      }
      ntra++;
    }
 
    // fill just another control histogram
 
    if (ntra > 0) {
      auto h4 = getObjectPtr<TH1F>("offset");
      h4->Fill(m_recBunch->getCurrentOffset());
    }
 
  }
 
}