SVDTimeGroupingConfiguration.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/logging/Logger.h>
 
#include "TObject.h"
#include "TString.h"
 
#include <vector>
 
namespace Belle2 {
  struct SVDTimeGroupingParameters: public TObject {
    Float_t tRange[2];
    Int_t   rebinningFactor;
    Float_t fillSigmaN;
    Float_t limitSigma[2];
    Float_t fitRangeHalfWidth;
    Float_t removeSigmaN;
    Float_t fracThreshold;
    Int_t   maxGroups;
    Float_t expectedSignalTime[3];
    Float_t signalLifetime;
    Int_t   numberOfSignalGroups;
    Bool_t  formSingleSignalGroup;
    Float_t acceptSigmaN;
    Bool_t  writeGroupInfo;
    Bool_t  includeOutOfRangeClusters;
    std::vector<Float_t> clsSigma[3][2];
 
    TString _COMMENT; 
    ClassDef(SVDTimeGroupingParameters, 1); 
  };
 
  class SVDTimeGroupingConfiguration: public TObject {
  public:
 
    SVDTimeGroupingConfiguration(const TString& uniqueID = "",
                                 const std::vector<TString>& description = {})
      : m_uniqueID(uniqueID), m_description(description)
    {
 
      setTimeGroupingParameters("CoG3", 6).tRange[0] = -160;
      setTimeGroupingParameters("CoG3", 6).tRange[1] = 160;
      setTimeGroupingParameters("CoG3", 6).rebinningFactor = 2;
      setTimeGroupingParameters("CoG3", 6).fillSigmaN = 3;
      setTimeGroupingParameters("CoG3", 6).limitSigma[0] = 1;
      setTimeGroupingParameters("CoG3", 6).limitSigma[1] = 15;
      setTimeGroupingParameters("CoG3", 6).fitRangeHalfWidth = 5;
      setTimeGroupingParameters("CoG3", 6).removeSigmaN = 5;
      setTimeGroupingParameters("CoG3", 6).fracThreshold = 0.05;
      setTimeGroupingParameters("CoG3", 6).maxGroups = 20;
      setTimeGroupingParameters("CoG3", 6).expectedSignalTime[0] = -50;
      setTimeGroupingParameters("CoG3", 6).expectedSignalTime[1] = 0;
      setTimeGroupingParameters("CoG3", 6).expectedSignalTime[2] = 50;
      setTimeGroupingParameters("CoG3", 6).signalLifetime = 30;
      setTimeGroupingParameters("CoG3", 6).numberOfSignalGroups = 1;
      setTimeGroupingParameters("CoG3", 6).formSingleSignalGroup = false;
      setTimeGroupingParameters("CoG3", 6).acceptSigmaN = 5;
      setTimeGroupingParameters("CoG3", 6).writeGroupInfo = true;
      setTimeGroupingParameters("CoG3", 6).includeOutOfRangeClusters = true;
      setTimeGroupingParameters("CoG3", 6).clsSigma[0][0] = {3.49898, 2.94008, 3.46766, 5.3746, 6.68848, 7.35446, 7.35983, 7.71601, 10.6172, 13.4805};
      setTimeGroupingParameters("CoG3", 6).clsSigma[0][1] = {6.53642, 3.76216, 3.30086, 3.95969, 5.49408, 7.07294, 8.35687, 8.94839, 9.23135, 10.485};
      setTimeGroupingParameters("CoG3", 6).clsSigma[1][0] = {3.49898, 2.94008, 3.46766, 5.3746, 6.68848, 7.35446, 7.35983, 7.71601, 10.6172, 13.4805};
      setTimeGroupingParameters("CoG3", 6).clsSigma[1][1] = {6.53642, 3.76216, 3.30086, 3.95969, 5.49408, 7.07294, 8.35687, 8.94839, 9.23135, 10.485};
      setTimeGroupingParameters("CoG3", 6).clsSigma[2][0] = {3.49898, 2.94008, 3.46766, 5.3746, 6.68848, 7.35446, 7.35983, 7.71601, 10.6172, 13.4805};
      setTimeGroupingParameters("CoG3", 6).clsSigma[2][1] = {6.53642, 3.76216, 3.30086, 3.95969, 5.49408, 7.07294, 8.35687, 8.94839, 9.23135, 10.485};
 
 
      // initialize all other algorithms same as CoG3-6sample.
      // values must be checked in the globaltag before use
 
      setTimeGroupingParameters("ELS3", 6) = getTimeGroupingParameters("CoG3", 6);
      setTimeGroupingParameters("CoG6", 6) = getTimeGroupingParameters("CoG3", 6);
 
      setTimeGroupingParameters("CoG3", 3) = getTimeGroupingParameters("CoG3", 6);
      setTimeGroupingParameters("ELS3", 3) = getTimeGroupingParameters("CoG3", 6);
      setTimeGroupingParameters("CoG6", 3) = getTimeGroupingParameters("CoG3", 6);
    };
 
    TString get_uniqueID() const {return m_uniqueID;}
 
    SVDTimeGroupingParameters getTimeGroupingParameters(const TString& alg,
                                                        const Int_t& numberOfAcquiredSamples) const
    {
      if (numberOfAcquiredSamples == 6) {
        if (alg == "CoG3")
          return m_parsForCoG3In6Samples;
        else if (alg == "ELS3")
          return m_parsForELS3In6Samples;
        else if (alg == "CoG6")
          return m_parsForCoG6In6Samples;
      } else if (numberOfAcquiredSamples == 3) {
        if (alg == "CoG3")
          return m_parsForCoG3In3Samples;
        else if (alg == "ELS3")
          return m_parsForELS3In3Samples;
        else if (alg == "CoG6")
          return m_parsForCoG6In3Samples;
      }
      B2FATAL("This state is forbidden");
    };
 
 
 
    SVDTimeGroupingParameters& setTimeGroupingParameters(const TString& alg,
                                                         const Int_t& numberOfAcquiredSamples)
    {
      if (numberOfAcquiredSamples == 6) {
        if (alg == "CoG3")
          return m_parsForCoG3In6Samples;
        else if (alg == "ELS3")
          return m_parsForELS3In6Samples;
        else if (alg == "CoG6")
          return m_parsForCoG6In6Samples;
      } else if (numberOfAcquiredSamples == 3) {
        if (alg == "CoG3")
          return m_parsForCoG3In3Samples;
        else if (alg == "ELS3")
          return m_parsForELS3In3Samples;
        else if (alg == "CoG6")
          return m_parsForCoG6In3Samples;
      }
      B2FATAL("This state is forbidden");
    };
 
    static std::string name;
 
  private:
 
    TString m_uniqueID;
    std::vector<TString> m_description;
 
    SVDTimeGroupingParameters m_parsForCoG6In6Samples;
    SVDTimeGroupingParameters m_parsForCoG3In6Samples;
    SVDTimeGroupingParameters m_parsForELS3In6Samples;
 
    SVDTimeGroupingParameters m_parsForCoG6In3Samples;
    SVDTimeGroupingParameters m_parsForCoG3In3Samples;
    SVDTimeGroupingParameters m_parsForELS3In3Samples;
 
    ClassDef(SVDTimeGroupingConfiguration, 1); 
  };
 
}