ECLExpertModule.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/ECLCluster.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 ECLExpertModule : public Module {
 
  public:
 
    ECLExpertModule();
 
    virtual ~ECLExpertModule();
 
    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);
 
 
    Double32_t  m_ECLminTrkDistance;  //[0.0, 250., 10]
 
    Double32_t  m_ECLZ40;  //[0.0, 1.7, 10]
 
    Double32_t  m_ECLZ51;  //[0.0, 1.2, 10]
 
    Double32_t  m_ECLE1oE9;  //[0.0, 1., 10]
 
    Double32_t  m_ECLE9oE21;  //[0.0, 1., 10]
 
    Double32_t  m_ECLsecondMoment;  //[0.0, 40.0, 10]
 
    Double32_t  m_ECLLAT;  //[0.0, 1., 10]
 
    Double32_t  m_ECLnumberOfCrystals;  //[0.0, 200.0, 10]
 
    Double32_t  m_ECLtime;  //[-1000.0, 1000.0, 12]
 
    Double32_t  m_ECLdeltaTime99;  //[0.0, 1000.0, 12]
 
    Double32_t  m_ECLtheta;  //[0.0, pi, 16]
 
    Double32_t  m_ECLphi;  //[-pi, pi, 16]
 
    Double32_t  m_ECLr;  //[75.0, 300.0, 16]
 
    Double32_t  m_ECLDeltaL;  //[75.0, 300.0, 16]
 
    Double32_t  m_ECLPulseShapeDiscriminationMVA;  //[0.0, 1.0, 18]
 
    Double32_t m_ECLNumberOfHadronDigits;  //[0, 255, 18]
 
    Double32_t m_ECLZMVA;
 
    Double32_t  m_ECLEnergy;  //[-5, 3., 18]
 
    Double32_t  m_ECLlogEnergy;  //[-5, 3., 18]
 
    Double32_t  m_ECLlogEnergyHighestCrystal;  //[-5, 3., 18]
 
    StoreArray<ECLCluster> m_eclClusters;
    StoreArray<KlId> m_klids;
 
    std::vector<float> m_feature_variables;
 
    std::string m_identifier =
      "ECL_fBDT_1xBkgPhiGamma.xml";
    //      "ECL_fBDT_1xBkgpGunMulti.xml";
    //      "ECL_fBDT_test_mixedBkg50kv1.xml";
    //      "ECL_fBDT_mixedBkg10k"; // 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