TrackSelector.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 <top/reconstruction_cpp/TOPTrack.h>
#include <framework/gearbox/Const.h>
#include <string>
#include <Math/Vector3D.h>
#include <Math/Point3D.h>
 
namespace Belle2 {
  namespace TOP {
 
    class TrackSelector {
 
    public:
 
      enum EType {
        c_undefined = 0,
        c_dimuon = 1,
        c_bhabha = 2,
        c_cosmics = 3
      };
 
      TrackSelector()
      {}
 
      explicit TrackSelector(const std::string& sampleName);
 
      void setMinMomentum(double minMomentum) {m_minMomentum = minMomentum;}
 
      void setDeltaEcms(double deltaEcms) {m_deltaEcms = deltaEcms;}
 
      void setCutOnPOCA(double dr, double dz)
      {
        m_dr = dr;
        m_dz = dz;
      }
 
      void setCutOnLocalZ(double minZ, double maxZ)
      {
        m_minZ = minZ;
        m_maxZ = maxZ;
      }
 
      const std::string& getSampleName() const {return m_sampleName;}
 
      double getMinMomentum() const {return m_minMomentum;}
 
      double getDeltaEcms() const {return m_deltaEcms;}
 
      double getDr() const {return m_dr;}
 
      double getDz() const {return m_dz;}
 
      double getMinZ() const {return m_minZ;}
 
      double getMaxZ() const {return m_maxZ;}
 
      const Const::ChargedStable& getChargedStable() const {return m_chargedStable;}
 
      const ROOT::Math::XYZVector& getPOCAPosition() const {return m_pocaPosition;}
 
      const ROOT::Math::XYZVector& getPOCAMomentum() const {return m_pocaMomentum;}
 
      double getCMSEnergy() const {return m_cmsEnergy;}
 
      const ROOT::Math::XYZPoint& getLocalPosition() const {return m_localPosition;}
 
      const ROOT::Math::XYZVector& getLocalMomentum() const {return m_localMomentum;}
 
      bool isSelected(const TOPTrack& track) const;
 
    private:
 
      EType m_sampleType = c_undefined; 
      std::string m_sampleName; 
      double m_minMomentum = 0; 
      double m_deltaEcms = 0; 
      double m_dr = 0; 
      double m_dz = 0; 
      double m_minZ = 0; 
      double m_maxZ = 0;  
      Const::ChargedStable m_chargedStable = Const::muon; 
      /* cache for track parameters */
      mutable ROOT::Math::XYZVector m_pocaPosition; 
      mutable ROOT::Math::XYZVector m_pocaMomentum; 
      mutable double m_cmsEnergy = 0;  
      mutable ROOT::Math::XYZPoint m_localPosition; 
      mutable ROOT::Math::XYZVector m_localMomentum; 
    };
 
  } // namespace TOP
} // namespace Belle2