gstdatareposrc.h

Go to the documentation of this file.

/* SPDX-License-Identifier: LGPL-2.1-only */
#ifndef __GST_DATA_REPO_SRC_H__
#define __GST_DATA_REPO_SRC_H__
 
#include <sys/types.h>
#include <gst/gst.h>
#include <gst/base/gstpushsrc.h>
#include <json-glib/json-glib.h>
#include <tensor_typedef.h>
#include "gstdatarepo.h"
 
G_BEGIN_DECLS
#define GST_TYPE_DATA_REPO_SRC \
  (gst_data_repo_src_get_type())
#define GST_DATA_REPO_SRC(obj) \
  (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_DATA_REPO_SRC,GstDataRepoSrc))
#define GST_DATA_REPO_SRC_CLASS(klass) \
  (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_DATA_REPO_SRC,GstDataRepoSrcClass))
#define GST_IS_DATA_REPO_SRC(obj) \
  (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_DATA_REPO_SRC))
#define GST_IS_DATA_REPO_SRC_CLASS(klass) \
  (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_DATA_REPO_SRC))
 
typedef struct _GstDataRepoSrc GstDataRepoSrc;
typedef struct _GstDataRepoSrcClass GstDataRepoSrcClass;
 
struct _GstDataRepoSrc {
 
  GstPushSrc parent;            
  GstPad *src_pad;
 
  gboolean is_start;            
  gboolean successful_read;     
  gint fd;                      
  gint file_size;               
  guint64 read_position;        
  guint64 fd_offset;            
  guint64 start_offset;         
  guint64 last_offset;          
  gsize tensors_size[NNS_TENSOR_SIZE_LIMIT];   
  gsize tensors_offset[NNS_TENSOR_SIZE_LIMIT]; 
  gint current_sample_index;    
  gboolean first_epoch_is_done;
  guint total_samples;           
  guint num_samples;             
  gsize sample_size;             
  GstDataRepoDataType data_type; 
  /* property */
  gchar *filename;              
  gchar *json_filename;         
  gchar *tensors_seq_str;       
  guint start_sample_index;     
  guint stop_sample_index;      
  guint epochs;                 
  gboolean is_shuffle;          
  GArray *shuffled_index_array; 
  guint array_index;            
  guint tensors_seq[NNS_TENSOR_SIZE_LIMIT];  
  guint tensors_seq_cnt;
  gboolean need_changed_caps;   
  GstCaps *caps;                
  /* flexible tensors */
  GstTensorsConfig config;          
  JsonArray *sample_offset_array;   
  JsonArray *tensor_size_array;     
  JsonArray *tensor_count_array;    
  JsonParser *parser;               
  guint sample_offset_array_len;
  guint tensor_size_array_len;
  guint tensor_count_array_len;
 
  GstClockTime running_time;    
  gint rate_n, rate_d;
  guint64 n_frame;
};
 
struct _GstDataRepoSrcClass {
  GstPushSrcClass parent_class;
};
 
GType gst_data_repo_src_get_type (void);
 
G_END_DECLS
#endif /* __GST_DATA_REPO_SRC_H__ */