Utility.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Thomas Keck                                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <mva/interface/Interface.h>
#include <mva/interface/Options.h>
#include <mva/interface/Weightfile.h>
 
namespace Belle2 {
  namespace MVA {
 
    void loadRootDictionary();
 
    void download(const std::string& identifier, const std::string& filename, int experiment = 0, int run = 0, int event = 0);
 
    void upload(const std::string& filename, const std::string& identifier, int exp1 = 0, int run1 = 0, int exp2 = -1, int run2 = -1);
 
    void upload_array(std::vector<std::string>& filenames, const std::string& identifier, int exp1 = 0, int run1 = 0, int exp2 = -1,
                      int run2 = -1);
 
    bool available(const std::string& filename, int experiment = 0, int run = 0, int event = 0);
 
    void extract(const std::string& filename, const std::string& directory);
 
    std::string info(const std::string& filename);
 
    void save_custom_weightfile(const GeneralOptions& general_options, const SpecificOptions& specific_options,
                                const std::string& custom_weightfile, const std::string& output_identifier = "");
 
    void teacher(const GeneralOptions& general_options, const SpecificOptions& specific_options,
                 const MetaOptions& meta_options = MetaOptions());
 
    void expert(const std::vector<std::string>& filenames, const std::vector<std::string>& datafile, const std::string& treename,
                const std::string& outputfile, int experiment = 0, int run = 0, int event = 0, bool copy_target = true);
 
    std::unique_ptr<Belle2::MVA::Expert> teacher_dataset(GeneralOptions general_options, const SpecificOptions& specific_options,
                                                         Dataset& data);
 
    std::unique_ptr<Belle2::MVA::Expert> teacher_splot(const GeneralOptions& general_options, const SpecificOptions& specific_options,
                                                       const MetaOptions& meta_options);
 
    std::unique_ptr<Belle2::MVA::Expert> teacher_sideband_substraction(const GeneralOptions& general_options,
        const SpecificOptions& specific_options,
        const MetaOptions& meta_options);
 
    std::unique_ptr<Belle2::MVA::Expert> teacher_reweighting(const GeneralOptions& general_options,
                                                             const SpecificOptions& specific_options,
                                                             const MetaOptions& meta_options);
 
 
  }
}