tensor_common.h

Go to the documentation of this file.

 
#ifndef __GST_TENSOR_COMMON_H__
#define __GST_TENSOR_COMMON_H__
 
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
 
#include <glib.h>
#include <stdint.h>
#include <gst/gst.h>
#include <gst/base/gstadapter.h>
#include <gst/base/gstcollectpads.h>
 
#include "tensor_typedef.h"
#include "nnstreamer_log.h"
#include "nnstreamer_plugin_api.h"
 
#ifdef HAVE_ORC
#include <orc/orcfunctions.h>
 
#define nns_memcpy(d,s,n) do { \
    if ((n) > 100) orc_memcpy ((d), (s), (n)); \
    else memcpy ((d), (s), (n)); \
  } while (0)
 
#define nns_memset orc_memset
#else
#define nns_memcpy memcpy
#define nns_memset memset
#endif
 
G_BEGIN_DECLS
 
typedef enum
{
  SYNC_NOSYNC = 0,
  SYNC_SLOWEST = 1,
  SYNC_BASEPAD = 2,
  SYNC_REFRESH = 3,
  SYNC_END,
} tensor_time_sync_mode;
 
typedef struct _tensor_sync_basepad_data{
  guint sink_id;
  GstClockTime duration;
} tensor_sync_basepad_data;
 
typedef struct _tensor_time_sync_data {
  tensor_time_sync_mode mode;
  gchar *option;
  union {
    tensor_sync_basepad_data data_basepad;
  };
} tensor_time_sync_data;
 
typedef struct
{
  GstCollectData collect;
  GstBuffer *buffer;
} GstTensorCollectPadData;
 
typedef struct
{
  GstPad *pad;
  GstClockTime last_ts;
  GstFlowReturn last_ret;
  guint nth;
} GstTensorPad;
 
extern void
gst_tensor_parse_config_file (const gchar *config_path, const GObject *object);
 
extern tensor_time_sync_mode
gst_tensor_time_sync_get_mode (const gchar * str);
 
extern const gchar *
gst_tensor_time_sync_get_mode_string (tensor_time_sync_mode mode);
 
extern gboolean
gst_tensor_time_sync_set_option_data (tensor_time_sync_data * sync);
 
extern gboolean
gst_tensor_time_sync_get_current_time (GstCollectPads * collect, tensor_time_sync_data * sync, GstClockTime * current_time, GstBuffer * tensors_buf);
 
extern void
gst_tensor_time_sync_flush (GstCollectPads * collect);
 
extern gboolean
gst_tensor_time_sync_buffer_from_collectpad (GstCollectPads * collect, tensor_time_sync_data * sync, GstClockTime current_time, GstBuffer * tensors_buf, GstTensorsConfig * configs, gboolean * is_eos);
 
extern GstBuffer *
gst_tensor_buffer_from_config (GstBuffer * in, GstTensorsConfig * config);
 
extern GstCaps *
gst_tensor_pad_caps_from_config (GstPad * pad, const GstTensorsConfig * config);
 
extern GstCaps *
gst_tensor_pad_possible_caps_from_config (GstPad * pad, const GstTensorsConfig * config);
 
extern tensor_format
gst_tensor_pad_get_format (GstPad *pad);
 
#define gst_tensor_pad_caps_is_static(p) (gst_tensor_pad_get_format (p) == _NNS_TENSOR_FORMAT_STATIC)
 
#define gst_tensor_pad_caps_is_flexible(p) (gst_tensor_pad_get_format (p) == _NNS_TENSOR_FORMAT_FLEXIBLE)
 
#define gst_tensor_pad_caps_is_sparse(p) (gst_tensor_pad_get_format (p) == _NNS_TENSOR_FORMAT_SPARSE)
 
extern GHashTable *
gst_tensor_aggregation_init (void);
 
extern void
gst_tensor_aggregation_clear (GHashTable * table, const guint32 key);
 
extern void
gst_tensor_aggregation_clear_all (GHashTable * table);
 
extern GstAdapter *
gst_tensor_aggregation_get_adapter (GHashTable * table, const guint32 key);
 
/******************************************************
 ************ Commonly used debugging macros **********
 ******************************************************
 */
#define silent_debug(self, ...) do { \
    if (DBG) { \
      GST_DEBUG_OBJECT (self, __VA_ARGS__); \
    } \
  } while (0)
 
#define silent_debug_caps(self, caps, msg) do { \
  if (DBG) { \
    if (caps) { \
      gchar *caps_s_string = gst_caps_to_string (caps); \
      GST_DEBUG_OBJECT (self, msg " = %s\n", caps_s_string); \
      g_free (caps_s_string); \
    } \
  } \
} while (0)
 
G_END_DECLS
#endif /* __GST_TENSOR_COMMON_H__ */