includes/gstnvvideotestsrc.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2021-2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */
 
#ifndef _GST_NV_VIDEO_TEST_SRC_H_
#define _GST_NV_VIDEO_TEST_SRC_H_
 
#undef __noinline__
 
#include <stdio.h>
#include <gst/gst.h>
#include <gst/base/base.h>
#include <gst/video/video.h>
 
#if defined(__CUDACC__) || defined(__CUDA_ARCH__) || defined(__CUDA_LIBDEVICE__)
    #define __noinline__ __attribute__((noinline))
#endif
 
#include <nvbufsurface.h>
 
G_BEGIN_DECLS
 
#define GST_TYPE_NV_VIDEO_TEST_SRC          (gst_nv_video_test_src_get_type())
#define GST_NV_VIDEO_TEST_SRC(obj)          (G_TYPE_CHECK_INSTANCE_CAST((obj), GST_TYPE_NV_VIDEO_TEST_SRC, GstNvVideoTestSrc))
#define GST_NV_VIDEO_TEST_SRC_CLASS(klass)  (G_TYPE_CHECK_CLASS_CAST((klass), GST_TYPE_NV_VIDEO_TEST_SRC, GstNvVideoTestSrcClass))
#define GST_IS_NV_VIDEO_TEST_SRC(obj)       (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_NV_VIDEO_TEST_SRC))
#define GST_IS_NV_VIDEO_TEST_SRC_CLASS(obj) (G_TYPE_CHECK_CLASS_TYPE((klass), GST_TYPE_NV_VIDEO_TEST_SRC))
 
typedef enum {
    GST_NV_VIDEO_TEST_SRC_SMPTE,
    GST_NV_VIDEO_TEST_SRC_MANDELBROT,
    GST_NV_VIDEO_TEST_SRC_GRADIENT
} GstNvVideoTestSrcPattern;
 
typedef enum {
    GST_NV_VIDEO_TEST_SRC_FRAMES,
    GST_NV_VIDEO_TEST_SRC_WALL_TIME,
    GST_NV_VIDEO_TEST_SRC_RUNNING_TIME
} GstNvVideoTestSrcAnimationMode;
 
typedef struct _GstNvVideoTestSrc GstNvVideoTestSrc;
typedef struct _GstNvVideoTestSrcClass GstNvVideoTestSrcClass;
 
struct _GstNvVideoTestSrc {
    GstPushSrc parent;
 
    // Plugin parameters.
    GstNvVideoTestSrcPattern pattern;
    GstNvVideoTestSrcAnimationMode animation_mode;
    guint gpu_id;
    NvBufSurfaceMemType memtype;
    gboolean enable_rdma;
 
    // Stream details set during caps negotiation.
    GstCaps *caps;
    GstVideoInfo info;
 
    // Runtime state.
    GstClockTime running_time;
    guint filled_frames;
 
    NvBufSurfaceParams *cuda_surf;
    unsigned int cuda_block_size;
    unsigned int cuda_num_blocks;
    void (*cuda_fill_image)(GstNvVideoTestSrc *src);
 
    // File read related
    gchar *filename;                          /* filename */
    FILE *file_handle;
    guint64 read_position;                    /* position of fd */
    NvBufSurface *file_read_surface = NULL;   /* surface for file read */
    gboolean file_loop;
 
    // Jitter related
    guint max_jitter = 0;        // max jitter in ms
    GstClockTime last_buffer_start_timestamp = 0;
    void *p_fixed_jitter_list = NULL;
};
 
struct _GstNvVideoTestSrcClass {
    GstPushSrcClass parent_class;
};
 
GType gst_nv_video_test_src_get_type(void);
 
G_END_DECLS
 
#endif