gstdatarepo.c

Go to the documentation of this file.

/* SPDX-License-Identifier: LGPL-2.1-only */
#include "gstdatarepo.h"
#include "gstdatareposrc.h"
#include "gstdatareposink.h"
 
GstDataRepoDataType
gst_data_repo_get_data_type_from_caps (const GstCaps * caps)
{
  const gchar *name;
 
  g_return_val_if_fail (GST_IS_CAPS (caps), GST_DATA_REPO_DATA_UNKNOWN);
 
  name = gst_structure_get_name (gst_caps_get_structure (caps, 0));
  g_return_val_if_fail (name != NULL, GST_DATA_REPO_DATA_UNKNOWN);
 
  if (g_ascii_strcasecmp (name, "other/tensors") == 0) {
    return GST_DATA_REPO_DATA_TENSOR;
  } else if (g_ascii_strcasecmp (name, "video/x-raw") == 0) {
    return GST_DATA_REPO_DATA_VIDEO;
  } else if (g_ascii_strcasecmp (name, "audio/x-raw") == 0) {
    return GST_DATA_REPO_DATA_AUDIO;
  } else if (g_ascii_strcasecmp (name, "text/x-raw") == 0) {
    return GST_DATA_REPO_DATA_TEXT;
  } else if (g_ascii_strcasecmp (name, "application/octet-stream") == 0) {
    return GST_DATA_REPO_DATA_OCTET;
  } else if (g_ascii_strcasecmp (name, "image/png") == 0
      || g_ascii_strcasecmp (name, "image/jpeg") == 0
      || g_ascii_strcasecmp (name, "image/tiff") == 0
      || g_ascii_strcasecmp (name, "image/gif") == 0) {
    return GST_DATA_REPO_DATA_IMAGE;
  }
 
  GST_ERROR ("Could not get a data type from caps.");
  return GST_DATA_REPO_DATA_UNKNOWN;
}
 
static gboolean
plugin_init (GstPlugin * plugin)
{
  if (!gst_element_register (plugin, "datareposrc", GST_RANK_NONE,
          GST_TYPE_DATA_REPO_SRC))
    return FALSE;
 
  if (!gst_element_register (plugin, "datareposink", GST_RANK_NONE,
          GST_TYPE_DATA_REPO_SINK))
    return FALSE;
 
  return TRUE;
}
 
#ifndef PACKAGE
#define PACKAGE "NNStreamer MLOps Data Repository Plugins"
#endif
 
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
    GST_VERSION_MINOR,
    datarepo,
    "NNStreamer MLOps Data Repository plugin library",
    plugin_init, VERSION, "LGPL", PACKAGE,
    "https://github.com/nnstreamer/nnstreamer")