VignetPlugin.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_PLUGIN_VIGNETPLUGIN_H_
#define _SEIMPLEMENTATION_PLUGIN_VIGNETPLUGIN_H_
 
#include <NdArray/NdArray.h>
#include "SEFramework/Plugin/Plugin.h"
#include "SEImplementation/Plugin/Vignet/VignetTaskFactory.h"
#include "Vignet.h"
#include "VignetArray.h"
 
namespace SourceXtractor {
 
template<typename T>
using NdArray = Euclid::NdArray::NdArray<T>;
 
class VignetPlugin : public Plugin {
public:
  virtual ~VignetPlugin() = default;
 
  virtual void registerPlugin(PluginAPI& plugin_api) {
    plugin_api.getTaskFactoryRegistry().registerTaskFactory<VignetTaskFactory, Vignet, VignetArray>();
    plugin_api.getOutputRegistry().registerColumnConverter<VignetArray, NdArray<DetectionImage::PixelType >>(
      "vignet",
      [](const VignetArray& prop) {
        return prop.getVignets();
      },
      "count",
      "The object vignet data"
    );
    plugin_api.getOutputRegistry().enableOutput<VignetArray>("Vignet");
  }
 
  virtual std::string getIdString() const {
    return "Vignet";
  }
 
private:
}; // end of VignetPlugin class
 
}  // namespace SourceXtractor
#endif /* _SEIMPLEMENTATION_PLUGIN_VIGNETPLUGIN_H_ */