TOPDigitVariables.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributor: Jan Strube (jan.strube@desy.de)                           *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <analysis/VariableManager/Manager.h>
#include <vector>
 
class TVector3;
namespace Belle2 {
  class Particle;
  class TOPLikelihood;
  class ExtHit;
  namespace Variable {
    // contains a couple of helper functions that are related to TOP variables
    namespace TOPVariable {
      const TOPLikelihood* getTOPLikelihood(const Particle* particle);
 
      const ExtHit* getExtHit(const Particle* particle);
 
      double getSlotID(const Particle* particle);
 
      TVector3 getLocalPosition(const Particle* particle);
 
      TVector3 getLocalPositionMCMatch(const Particle* particle);
 
      TVector3 getLocalMomentum(const Particle* particle);
 
      double getTOF(const Particle* particle);
 
      double getTOFExpert(const Particle* particle, const std::vector<double>& vars);
 
      double getAverageTimeOfFirst5(const Particle* particle);
 
      int countHits(const Particle* particle, double tmin, double tmax, bool clean = true);
 
      int countRawHits(const Particle* particle, double tmin, double tmax);
 
      double getExpectedPhotonCount(const Particle* particle, int pdg = 0);
 
      double topDigitCount(const Particle* particle);
 
      double extrapTrackToTOPz(const Particle* particle);
 
      double extrapTrackToTOPtheta(const Particle* particle);
 
      double extrapTrackToTOPphi(const Particle* particle);
 
      double topBackgroundDigitCount(const Particle* particle);
 
      double topBackgroundDigitCountRaw(const Particle* particle);
 
      double topRawDigitCount(const Particle* particle);
 
      double topDigitGapSize(const Particle* particle);
 
      double topCountPhotonsAfterLargesGapWithin(const Particle* particle,
                                                 double min, double max);
 
      double topReflectedDigitCount(const Particle* particle);
 
      double getTOPLocalX(const Particle* particle);
 
      double getTOPLocalY(const Particle* particle);
 
      double getTOPLocalZ(const Particle* particle);
 
      double getTOPLocalXMCMatch(const Particle* particle);
 
      double getTOPLocalYMCMatch(const Particle* particle);
 
      double getTOPLocalZMCMatch(const Particle* particle);
 
      double getTOPLocalPhi(const Particle* particle);
 
      double getTOPLocalTheta(const Particle* particle);
 
      double getTOPPhotonCount(const Particle* particle);
 
      double getExpectedTOPPhotonCount(const Particle* particle,
                                       const std::vector<double>& vars);
 
      double countTOPHitsInInterval(const Particle* particle,
                                    const std::vector<double>& vars);
 
      double countTOPHitsInFirst20ns(const Particle* particle);
 
      double countRawTOPHitsInInterval(const Particle* particle,
                                       const std::vector<double>& vars);
 
      double getFlag(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);
 
      //---------------- TOPRecBunch related --------------------
 
      double isTOPRecBunchReconstructed([[maybe_unused]] const Particle* particle);
 
      double TOPRecBunchNumber([[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([[maybe_unused]] const Particle* particle);
    } // TOPVariable
  } // Variable
} // Belle2