SelectorMVA.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 <analysis/modules/CurlTagger/Selector.h>
 
#include <analysis/dataobjects/Particle.h>
#include <vector>
#include <string>
 
#include <mva/dataobjects/DatabaseRepresentationOfWeightfile.h>
#include <mva/methods/FastBDT.h>
#include <mva/interface/Interface.h>
#include <framework/database/DBObjPtr.h>
 
//Root Includes
#include "TFile.h"
#include "TTree.h"
 
namespace Belle2 {
  namespace CurlTagger {
 
    class SelectorMVA : public Selector {
 
    public:
      SelectorMVA(bool belleFlag, bool trainFlag, std::string tFileName);
 
      ~SelectorMVA();
 
      virtual float getResponse(Particle* iPart, Particle* jPart) override;
 
      virtual float getOptimalResponseCut() override;
 
      virtual std::vector<float> getVariables(Particle* iPart, Particle* jPart) override;
 
      virtual void initialize() override;
 
      virtual void collectTrainingInfo(Particle* iPart, Particle* jPart) override;
 
      virtual void finalize() override;
 
      void initializeMVA();
 
    private:
 
      void updateVariables(Particle* iPart, Particle* jPart);
 
      bool m_TrainFlag;
 
      std::string m_TFileName;
 
      TFile* m_TFile;
 
      TTree* m_TTree;
 
      std::unique_ptr<DBObjPtr<DatabaseRepresentationOfWeightfile>> m_weightfile_representation;
 
      MVA::Weightfile m_weightfile;
 
      MVA::GeneralOptions m_generalOptions;
 
      MVA::FastBDTExpert m_expert;
 
      // General Options data - just use same names
      std::string m_identifier;
 
      std::vector<std::string> m_datafiles;
 
      std::vector<std::string> m_variables;
 
      std::string m_target_variable;
 
 
      // MVA Variables
      Float_t m_PPhi;
 
      Float_t m_ChargeProduct;
 
      Float_t m_PtDiffEW;
 
      Float_t m_PzDiffEW;
 
      Float_t m_TrackD0DiffEW;
 
      Float_t m_TrackZ0DiffEW;
 
      Float_t m_TrackTanLambdaDiffEW;
 
      Float_t m_TrackPhi0DiffEW;
 
      Float_t m_TrackOmegaDiffEW;
 
      Bool_t m_IsCurl;
 
    }; //selectorMVA class
 
  } // curlTagger Module namespace
}// Belle 2 namespace