OnnxPlugin.cpp

Go to the documentation of this file.

#include "SEImplementation/Plugin/Onnx/OnnxPlugin.h"
#include "SEFramework/Plugin/StaticPlugin.h"
#include "SEImplementation/Plugin/Onnx/OnnxProperty.h"
#include "SEImplementation/Plugin/Onnx/OnnxTaskFactory.h"

namespace SourceXtractor {

static StaticPlugin<OnnxPlugin> onnx_plugin;
Elements::Logging onnx_logger = Elements::Logging::getLogger("MLMeasurement");

std::string OnnxPlugin::getIdString() const {
  return "MLMeasurementPlugin";
}

void OnnxPlugin::registerPlugin(PluginAPI& plugin_api) {
  plugin_api.getTaskFactoryRegistry().registerTaskFactory<OnnxTaskFactory, OnnxProperty>();
  // Note that we do not now the output yet, so we do not register any columns

  plugin_api.getOutputRegistry().registerColumnConverter<OnnxProperty, int>(
            "ml_measurment_flags",
            [](const OnnxProperty&) {
              return 0; // Dummy property for now to avoid crash
            },
            "",
            "flags"
    );
  plugin_api.getOutputRegistry().enableOutput<OnnxProperty>("MLMeasurement", true);
}

} // end of namespace SourceXtractor