KLMExpertModule.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/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <framework/database/DBObjPtr.h>
 
#include <string>
 
 
namespace Belle2 {
  class KLMCluster;
  class KlId;
  class DatabaseRepresentationOfWeightfile;
 
  namespace MVA {
    class Weightfile;
    class Expert;
    class SingleDataset;
  }
 

  class KLMExpertModule : public Module {
 
  public:

    KLMExpertModule();

    virtual ~KLMExpertModule();

    virtual void initialize() override;

    virtual void beginRun() override;

    virtual void event() override;

    virtual void terminate() override
    {
      m_expert.reset();
      m_dataset.reset();
    }
 
  private:

    void init_mva(MVA::Weightfile& weightfile);
 

    // KLM variables
    float m_KLMnCluster{ -1.};
    float m_KLMnLayer{ -1.};
    float m_KLMnInnermostLayer{ -1.};
    float m_KLMglobalZ{0.};
    float m_KLMtime{ -1.};
    float m_KLMavInterClusterDist{ -1.};
    float m_KLMhitDepth{ -1.};
    float m_KLMenergy{ -1.};
    float m_KLMnextCluster{ -1.};
    float m_KLMTrackSepDist{ -1.};
    float m_KLMTrackSepAngle{ -1.};

    float m_KLMInitialTrackSepAngle{ -1.};
    float m_KLMTrackRotationAngle{ -1.};
    float m_KLMTrackClusterSepAngle{ -1.};
 
    // variables of closest ECL cluster with respect to KLM cluster
    float m_KLMECLDist{ -1.};
    float m_KLMECLE{ -1.};
    float m_KLMECLdeltaL{ -1.};   // new
    float m_KLMECLminTrackDist{ -1.}; //new
    float m_KLMECLE9oE25{ -1.};
    float m_KLMECLTiming{ -1.};
    float m_KLMECLTerror{ -1.};
    float m_KLMECLEerror{ -1.};

    float m_KLMECLZMVA{ -1.};
    float m_KLMECLZ40{ -1.};
    float m_KLMECLZ51{ -1.};

    StoreArray<KLMCluster> m_klmClusters;
    StoreArray<KlId> m_klids;

    std::vector<float> m_feature_variables;

    std::string m_identifier =
      "KLM_fBDT_10xbkg100k"; // assuming 1 times beambkg and generic BBbar events

    std::unique_ptr<DBObjPtr<DatabaseRepresentationOfWeightfile>>
                                                               m_weightfile_representation;
    std::unique_ptr<MVA::Expert> m_expert;
    std::unique_ptr<MVA::SingleDataset> m_dataset;
 
 
  }; // end class

} // end namespace Belle2