nnstreamer.c

Go to the documentation of this file.

 
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
 
#include <gst/gst.h>
 
#include <elements/gsttensor_aggregator.h>
#include <elements/gsttensor_converter.h>
#include <elements/gsttensor_crop.h>
#include <elements/gsttensor_debug.h>
#include <elements/gsttensor_decoder.h>
#include <elements/gsttensor_demux.h>
#include <elements/gsttensor_if.h>
#include <elements/gsttensor_merge.h>
#include <elements/gsttensor_mux.h>
#include <elements/gsttensor_rate.h>
#include <elements/gsttensor_reposink.h>
#include <elements/gsttensor_reposrc.h>
#include <elements/gsttensor_sink.h>
#include <elements/gsttensor_sparsedec.h>
#include <elements/gsttensor_sparseenc.h>
#include <elements/gsttensor_split.h>
#include <elements/gsttensor_transform.h>
#include <elements/gsttensor_trainer.h>
 
#ifdef _ENABLE_SRC_IIO
#include <elements/gsttensor_srciio.h>
#endif
 
#include <tensor_filter/tensor_filter.h>
#if defined(ENABLE_NNSTREAMER_EDGE)
#include <tensor_query/tensor_query_serversrc.h>
#include <tensor_query/tensor_query_serversink.h>
#include <tensor_query/tensor_query_client.h>
#endif
 
#define NNSTREAMER_INIT(plugin,name,type) \
  do { \
    if (!gst_element_register (plugin, "tensor_" # name, GST_RANK_NONE, GST_TYPE_TENSOR_ 
      GST_ERROR ("Failed to register nnstreamer plugin : tensor_" # name); \
      return FALSE; \
    } \
  } while (0)
 
static gboolean
gst_nnstreamer_init (GstPlugin * plugin)
{
  NNSTREAMER_INIT (plugin, aggregator, AGGREGATOR);
  NNSTREAMER_INIT (plugin, converter, CONVERTER);
  NNSTREAMER_INIT (plugin, crop, CROP);
  NNSTREAMER_INIT (plugin, debug, DEBUG);
  NNSTREAMER_INIT (plugin, decoder, DECODER);
  NNSTREAMER_INIT (plugin, demux, DEMUX);
  NNSTREAMER_INIT (plugin, filter, FILTER);
  NNSTREAMER_INIT (plugin, merge, MERGE);
  NNSTREAMER_INIT (plugin, mux, MUX);
  NNSTREAMER_INIT (plugin, reposink, REPOSINK);
  NNSTREAMER_INIT (plugin, reposrc, REPOSRC);
  NNSTREAMER_INIT (plugin, sink, SINK);
  NNSTREAMER_INIT (plugin, sparse_enc, SPARSE_ENC);
  NNSTREAMER_INIT (plugin, sparse_dec, SPARSE_DEC);
  NNSTREAMER_INIT (plugin, split, SPLIT);
  NNSTREAMER_INIT (plugin, transform, TRANSFORM);
  NNSTREAMER_INIT (plugin, if, IF);
  NNSTREAMER_INIT (plugin, rate, RATE);
  NNSTREAMER_INIT (plugin, trainer, TRAINER);
#if defined(ENABLE_NNSTREAMER_EDGE)
  NNSTREAMER_INIT (plugin, query_serversrc, QUERY_SERVERSRC);
  NNSTREAMER_INIT (plugin, query_serversink, QUERY_SERVERSINK);
  NNSTREAMER_INIT (plugin, query_client, QUERY_CLIENT);
#endif
#ifdef _ENABLE_SRC_IIO
  NNSTREAMER_INIT (plugin, src_iio, SRC_IIO);
#endif
  return TRUE;
}
 
#ifndef PACKAGE
#define PACKAGE "nnstreamer"
#endif
 
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
    GST_VERSION_MINOR,
    nnstreamer,
    "NNStreamer plugin library allows neural networks in GStreamer pipelines. Use nnstreamer-check utility for more information of the current NNStreamer installation.",
    gst_nnstreamer_init, VERSION, "LGPL", "nnstreamer",
    "https://github.com/nnstreamer/nnstreamer");