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 <mdst/dataobjects/KLMCluster.h>
#include <mdst/dataobjects/KlId.h>
 
#include <mva/dataobjects/DatabaseRepresentationOfWeightfile.h>
#include <mva/interface/Weightfile.h>
#include <mva/interface/Expert.h>
 
#include <string>
 
 
namespace Belle2 {
  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