gsttensor_transform.h

Go to the documentation of this file.

 
#ifndef __GST_TENSOR_TRANSFORM_H__
#define __GST_TENSOR_TRANSFORM_H__
 
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
 
#include <gst/gst.h>
#include <gst/base/gstbasetransform.h>
#include <tensor_common.h>
#include <tensor_data.h>
 
G_BEGIN_DECLS
 
#define GST_TYPE_TENSOR_TRANSFORM \
  (gst_tensor_transform_get_type())
#define GST_TENSOR_TRANSFORM(obj) \
  (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_TENSOR_TRANSFORM,GstTensorTransform))
#define GST_TENSOR_TRANSFORM_CLASS(klass) \
  (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_TENSOR_TRANSFORM,GstTensorTransformClass))
#define GST_IS_TENSOR_TRANSFORM(obj) \
  (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_TENSOR_TRANSFORM))
#define GST_IS_TENSOR_TRANSFORM_CLASS(klass) \
  (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_TENSOR_TRANSFORM))
#define GST_TENSOR_TRANSFORM_CAST(obj)  ((GstTensorTransform *)(obj))
 
typedef struct _GstTensorTransform GstTensorTransform;
typedef struct _GstTensorTransformClass GstTensorTransformClass;
 
typedef enum _tensor_transform_mode
{
  GTT_DIMCHG = 0,     /* Dimension Change. "dimchg" */
  GTT_TYPECAST,       /* Type change. "typecast" */
  GTT_ARITHMETIC,     /* Arithmetic. "arithmetic" */
  GTT_TRANSPOSE,      /* Transpose. "transpose" */
  GTT_STAND,          /* Standardization. "stand" */
  GTT_CLAMP,          /* Clamp, "clamp" */
  GTT_PADDING,        /* Padding, "padding" */
 
  GTT_UNKNOWN = -1,   /* Unknown/Not-implemented-yet Mode. "unknown" */
} tensor_transform_mode;
 
typedef enum
{
  GTT_OP_TYPECAST = 0,
  GTT_OP_ADD = 1,
  GTT_OP_MUL = 2,
  GTT_OP_DIV = 3,
 
  GTT_OP_UNKNOWN
} tensor_transform_operator;
 
typedef enum
{
  STAND_DEFAULT = 0,
  STAND_DC_AVERAGE = 1,
  STAND_END,
} tensor_transform_stand_mode;
 
typedef enum
{
  PADDING_LEFT = 0,
  PADDING_RIGHT = 1,
  PADDING_TOP = 2,
  PADDING_BOTTOM = 3,
  PADDING_FRONT = 4,
  PADDING_BACK = 5,
  PADDING_END,
} tensor_transform_padding_axis;
 
typedef struct _tensor_transform_dimchg {
  int from;
  int to;
} tensor_transform_dimchg;
 
typedef struct _tensor_transform_typecast {
  tensor_type to; 
} tensor_transform_typecast;
 
typedef struct
{
  tensor_transform_operator op;
  int applying_ch;
  tensor_data_s value;
} tensor_transform_operator_s;
 
typedef struct _tensor_transform_arithmetic {
  tensor_type out_type;
  gboolean per_channel_arith;
  guint ch_dim;
} tensor_transform_arithmetic;
 
typedef struct _tensor_transform_transpose {
  uint8_t trans_order[NNS_TENSOR_RANK_LIMIT];
} tensor_transform_transpose;
 
typedef struct _tensor_transform_stand {
  tensor_transform_stand_mode mode;
  tensor_type out_type;
  gboolean per_channel;
} tensor_transform_stand;
 
typedef struct _tensor_transform_clamp {
  double min, max;
} tensor_transform_clamp;
 
typedef struct _tensor_transform_padding {
  guint pad[NNS_TENSOR_RANK_LIMIT];
  tensor_layout layout;
} tensor_transform_padding;
 
struct _GstTensorTransform
{
  GstBaseTransform element; 
  gboolean silent;  
  tensor_transform_mode mode; 
  gchar *option; 
  union {
    tensor_transform_dimchg data_dimchg; 
    tensor_transform_typecast data_typecast; 
    tensor_transform_arithmetic data_arithmetic; 
    tensor_transform_transpose data_transpose; 
    tensor_transform_stand data_stand; 
    tensor_transform_clamp data_clamp; 
    tensor_transform_padding data_padding; 
  };
  gboolean loaded; 
  gboolean acceleration; 
  GSList *operators; 
  GstTensorsConfig in_config; 
  GstTensorsConfig out_config; 
  GList *apply; 
};
 
struct _GstTensorTransformClass
{
  GstBaseTransformClass parent_class;   
};
 
GType gst_tensor_transform_get_type (void);
 
G_END_DECLS
 
#endif /* __GST_TENSOR_TRANSFORM_H__ */