GenerativePlugin.h

// (c) Copyright 2022 CERN for the benefit of the LHCb Collaboration.
//
// This software is distributed under the terms of the GNU General Public
// Licence version 3 (GPL Version 3), copied verbatim in the file "LICENCE".
//
// In applying this licence, CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#pragma once
//STL
#include <vector>
#include <string>
 
// SQLamarr
#include "SQLamarr/BasePlugin.h"
 
namespace SQLamarr
{
  class GenerativePlugin: public BasePlugin
  {
    public:
      GenerativePlugin (
          SQLite3DB& db,
          const std::string& library,
          const std::string& function_name,
          const std::string& select_query,
          const std::string& output_table,
          const std::vector<std::string> outputs,
          unsigned int n_random,
          const std::vector<std::string> reference_keys = {"ref_id"}
          ) 
          : BasePlugin(db, library, function_name, select_query, 
              output_table, outputs, reference_keys)
          , m_func(load_func<genfunc>(function_name))
          , m_n_random (n_random) 
          {}
 
 
    private:
      virtual 
      void eval_parametrization (float* output, const float* input) override;
 
      typedef float *(*genfunc)(float *, const float*, const float*);
      genfunc m_func;
 
      unsigned int m_n_random;
  };
}