SVDReconstructionBase.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 <set>
 
namespace Belle2::SVD {
 
  class SVDReconstructionBase {
 
  public:
 
    SVDReconstructionBase(const SVDReconstructionBase&) = default;
 
    explicit SVDReconstructionBase() { buildAlgorithmSets(); };
 
    virtual ~SVDReconstructionBase() {};
 
    bool isTimeAlgorithmAvailable(TString timeAlg)
    {
      return m_timeAlgorithms.find(timeAlg) != m_timeAlgorithms.end();
    }
 
    bool isChargeAlgorithmAvailable(TString chargeAlg)
    {
      return m_chargeAlgorithms.find(chargeAlg) != m_chargeAlgorithms.end();
    }
 
    bool isPositionAlgorithmAvailable(TString positionAlg)
    {
      return m_positionAlgorithms.find(positionAlg) != m_positionAlgorithms.end();
    }
 
 
  protected:
 
 
    std::set<TString> m_timeAlgorithms;
 
    std::set<TString> m_chargeAlgorithms;
 
    std::set<TString> m_positionAlgorithms;
 
    void buildAlgorithmSets()
    {
      m_timeAlgorithms.insert("CoG6");
      m_timeAlgorithms.insert("CoG3");
      m_timeAlgorithms.insert("ELS3");
 
      m_chargeAlgorithms.insert("MaxSample");
      m_chargeAlgorithms.insert("SumSamples");
      m_chargeAlgorithms.insert("ELS3");
 
      m_positionAlgorithms.insert("CoGOnly");
      m_positionAlgorithms.insert("OldDefault");
    }
 
  };
 
}