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 <string>
 
 
namespace Belle2 {
  class ECLCluster;
  class KlId;
  class DatabaseRepresentationOfWeightfile;
 
  namespace MVA {
    class Weightfile;
    class Expert;
    class SingleDataset;
  }
 

  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