TOPReconstructorModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <mdst/dataobjects/Track.h>
#include <tracking/dataobjects/ExtHit.h>
#include <top/dataobjects/TOPDigit.h>
#include <top/dataobjects/TOPRecBunch.h>
#include <top/dataobjects/TOPBarHit.h>
#include <top/dataobjects/TOPLikelihood.h>
#include <top/dataobjects/TOPPull.h>
#include <top/reconstruction_cpp/PDFConstructor.h>
#include <top/reconstruction_cpp/TOPTrack.h>
#include <top/dbobjects/TOPCalTOFCorrection.h>
#include <framework/gearbox/Const.h>
#include <vector>
#include <string>
#include <limits>
 
namespace Belle2 {

  class TOPReconstructorModule : public Module {
 
  public:

    TOPReconstructorModule();

    virtual void initialize() override;

    virtual void event() override;
 
  private:

    struct PDFCollection {
 
      const TOP::TOPTrack* topTrack = 0; 
      std::vector<TOP::PDFConstructor*> PDFs; 
      const TOP::PDFConstructor* mostProbable = 0; 
      bool isValid = false; 

      PDFCollection(const TOP::TOPTrack& trk, bool deltaRayModeling)
      {
        topTrack = &trk;
        for (const auto& chargedStable : Const::chargedStableSet) {
          auto* pdf = new TOP::PDFConstructor(trk, chargedStable);
          if (not pdf->isValid()) {
            delete pdf;
            for (auto PDF : PDFs) delete PDF;
            PDFs.clear();
            return;
          }
          pdf->switchDeltaRayPDF(deltaRayModeling);
          PDFs.push_back(pdf);
        }
        isValid = true;
      }

      void deletePDFs()
      {
        for (auto pdf : PDFs) delete pdf;
        PDFs.clear();
        isValid = false;
      }

      void setMostProbable()
      {
        if (not isValid) return;
        double maxLogL = -std::numeric_limits<double>::infinity();
        for (const auto pdf : PDFs) {
          double logL = pdf->getLogL().logL;
          if (logL > maxLogL) {
            mostProbable = pdf;
            maxLogL = logL;
          }
        }
      }

      void appendPDFOther(const TOP::PDFConstructor* pdfOther)
      {
        if (not isValid) return;
        for (const auto pdf : PDFs) pdf->appendPDFOther(pdfOther);
      }

      void clearPDFOther()
      {
        if (not isValid) return;
        for (const auto pdf : PDFs) pdf->clearPDFOther();
      }
 
    };
 
    // Module steering parameters
 
    double m_minTime = 0;      
    double m_maxTime = 0;      
    int m_PDGCode = 0;   
    bool m_deltaRayModeling;  
    double m_pTCut;  
    std::string m_topDigitCollectionName; 
    std::string m_topLikelihoodCollectionName; 
    std::string m_topPullCollectionName; 
 
    // datastore objects
 
    StoreArray<TOPDigit> m_digits; 
    StoreArray<Track> m_tracks; 
    StoreArray<ExtHit> m_extHits; 
    StoreArray<TOPBarHit> m_barHits; 
    StoreObjPtr<TOPRecBunch> m_recBunch; 
    StoreArray<TOPLikelihood> m_likelihoods; 
    StoreArray<TOPPull> m_topPulls; 
 
    // database
 
    DBObjPtr<TOPCalTOFCorrection> m_tofCorrections; 
 
  };

} // Belle2 namespace