TOPDigitVariables.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 <analysis/VariableManager/Manager.h>
#include <vector>
#include <Math/Vector3D.h>
#include <Math/Point3D.h>
 
namespace Belle2 {
  class Particle;
  class TOPLikelihood;
  class ExtHit;
  class TOPBarHit;
  class TOPLikelihoodScanResult;
  namespace Variable {
    namespace TOPVariable {
 
      //---------------- helpers --------------------
 
      const TOPLikelihood* getTOPLikelihood(const Particle* particle);
 
      const ExtHit* getExtHit(const Particle* particle);
 
      const TOPBarHit* getBarHit(const Particle* particle);
 
      const TOPLikelihoodScanResult* getTOPLikelihoodScanResult(const Particle* particle);
 
 
      int getSlotID(const Particle* particle);
 
 
      //---------------- ExtHit (or TOPBarHit) based --------------------
 
      double topSlotID(const Particle* particle);
 
      double topSlotIDMCMatch(const Particle* particle);
 
      bool getLocalPosition(const Particle* particle, ROOT::Math::XYZPoint& result);
 
      double getTOPLocalX(const Particle* particle);
 
      double getTOPLocalY(const Particle* particle);
 
      double getTOPLocalZ(const Particle* particle);
 
      bool getLocalPositionMCMatch(const Particle* particle, ROOT::Math::XYZPoint& result);
 
      double getTOPLocalXMCMatch(const Particle* particle);
 
      double getTOPLocalYMCMatch(const Particle* particle);
 
      double getTOPLocalZMCMatch(const Particle* particle);
 
      bool getLocalMomentum(const Particle* particle, ROOT::Math::XYZVector& result);
 
      double getTOPLocalPhi(const Particle* particle);
 
      double getTOPLocalTheta(const Particle* particle);
 
      bool getLocalMomentumMCMatch(const Particle* particle, ROOT::Math::XYZVector& result);
 
      double getTOPLocalPhiMCMatch(const Particle* particle);
 
      double getTOPLocalThetaMCMatch(const Particle* particle);
 
      double computeTOF(const Particle* particle, int pdg);
 
      double getTOF(const Particle* particle);
 
      double getTOFMCMatch(const Particle* particle);
 
      double getTOFExpert(const Particle* particle, const std::vector<double>& vars);
 
 
      //---------------- Helix extrapolation --------------------
 
      double extrapTrackToTOPz(const Particle* particle);
 
      double extrapTrackToTOPtheta(const Particle* particle);
 
      double extrapTrackToTOPphi(const Particle* particle);
 
 
      //---------------- TOPDigit based --------------------
 
      double countHits(const Particle* particle, double tmin, double tmax, bool clean);
 
      double topDigitCount(const Particle* particle);
 
      double topDigitCountMCMatch(const Particle* particle);
 
      double topSignalDigitCount(const Particle* particle);
 
      double topBackgroundDigitCount(const Particle* particle);
 
      double topRawDigitCount(const Particle* particle);
 
      double countTOPHitsInInterval(const Particle* particle, const std::vector<double>& vars);
 
      double countTOPHitsInIntervalMCMatch(const Particle* particle, const std::vector<double>& vars);
 
      double countRawTOPHitsInInterval(const Particle* particle, const std::vector<double>& vars);
 
 
      //---------------- TOPLikelihood based --------------------
 
      double getFlag(const Particle* particle);
 
      double getModuleID(const Particle* particle);
 
      double getEmissionX(const Particle* particle);
 
      double getEmissionZ(const Particle* particle);
 
      double getTOPPhotonCount(const Particle* particle);
 
      double expectedPhotonCount(const Particle* particle, int pdg);
 
      double getExpectedPhotonCount(const Particle* particle);
 
      double getExpectedPhotonCountExpert(const Particle* particle,
                                          const std::vector<double>& vars);
 
      double effectiveSignalYield(const Particle* particle, int pdg);
 
      double getEffectiveSignalYield(const Particle* particle);
 
      double getEffectiveSignalYieldExpert(const Particle* particle,
                                           const std::vector<double>& vars);
 
      double getEstimatedBkgCount(const Particle* particle);
 
      double getElectronLogL(const Particle* particle);
 
      double getMuonLogL(const Particle* particle);
 
      double getPionLogL(const Particle* particle);
 
      double getKaonLogL(const Particle* particle);
 
      double getProtonLogL(const Particle* particle);
 
      double getDeuteronLogL(const Particle* particle);
 
 
      //---------------- TOPLikelihoodScanResult based --------------------
 
      double getLogLScanMass(const Particle* particle);
 
      double getLogLScanMassLowerInterval(const Particle* particle);
 
      double getLogLScanMassUpperInterval(const Particle* particle);
 
      double getLogLScanThreshold(const Particle* particle);
 
      double getLogLScanExpectedSignalPhotons(const Particle* particle);
 
 
      //---------------- TOPRecBunch based --------------------
 
      double isTOPRecBunchReconstructed([[maybe_unused]] const Particle* particle);
 
      double isTOPRecBunchFilled([[maybe_unused]] const Particle* particle);
 
      double TOPRecBunchNumber([[maybe_unused]] const Particle* particle);
 
      double TOPRecBucketNumber([[maybe_unused]] const Particle* particle);
 
      double isTOPRecBunchNumberEQsim([[maybe_unused]] const Particle* particle);
 
      double TOPRecBunchCurrentOffset([[maybe_unused]] const Particle* particle);
 
      double TOPRecBunchTrackCount([[maybe_unused]] const Particle* particle);
 
      double TOPRecBunchUsedTrackCount([[maybe_unused]] const Particle* particle);
 
 
      //-------------- Event based -----------------------------------
 
      double TOPRawPhotonsInSlot([[maybe_unused]] const Particle* particle, const std::vector<double>& vars);
 
      double TOPGoodPhotonsInSlot([[maybe_unused]] const Particle* particle, const std::vector<double>& vars);
 
      double TOPTracksInSlot(const Particle* particle);
 
    } // TOPVariable
  } // Variable
} // Belle2