From 638177c8d11d2aaf6136e606ce5478c310052ba9 Mon Sep 17 00:00:00 2001
From: Barzan Hayati <barzanhayati@gmail.com>
Date: Mon, 30 Jun 2025 19:53:25 +0000
Subject: [PATCH] Create nvvideoconvert

---
 CMakeLists.txt                   |  5 +++--
 src/nv_video_convert_manager.cpp | 18 ++++++++++++++++++
 src/nv_video_convert_manager.hpp | 10 ++++++++++
 src/pipeline_manager.cpp         |  1 +
 src/pipeline_manager.hpp         |  3 +++
 5 files changed, 35 insertions(+), 2 deletions(-)
 create mode 100644 src/nv_video_convert_manager.cpp
 create mode 100644 src/nv_video_convert_manager.hpp

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 4b26576..f0300a2 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -52,10 +52,11 @@ include_directories(${PROJECT_SOURCE_DIR}/streammux_manager.hpp)
 include_directories(${PROJECT_SOURCE_DIR}/source_bin.hpp)
 include_directories(${PROJECT_SOURCE_DIR}/gstds_example_manager.hpp)
 include_directories(${PROJECT_SOURCE_DIR}/tiler_manager.hpp)
+include_directories(${PROJECT_SOURCE_DIR}/nv_video_convert_manager.hpp)
 
 
-
-set(SOURCES src/main.cpp  src/camera_manager.cpp src/pipeline_manager.cpp src/streammux_manager.cpp src/source_bin.cpp src/gstds_example_manager.cpp src/tiler_manager.cpp)
+set(SOURCES src/main.cpp  src/camera_manager.cpp src/pipeline_manager.cpp src/streammux_manager.cpp src/source_bin.cpp src/gstds_example_manager.cpp src/tiler_manager.cpp
+            src/nv_video_convert_manager.cpp)
 # Create the executable
 add_executable(${PROJECT_NAME}  ${SOURCES})
 
diff --git a/src/nv_video_convert_manager.cpp b/src/nv_video_convert_manager.cpp
new file mode 100644
index 0000000..cf51961
--- /dev/null
+++ b/src/nv_video_convert_manager.cpp
@@ -0,0 +1,18 @@
+#include "nv_video_convert_manager.hpp"
+
+#define SET_GPU_ID(object, gpu_id) \
+    g_object_set(G_OBJECT(object), "gpu-id", gpu_id, NULL);
+#define GPU_ID 0
+
+NvVideoConvertManager::NvVideoConvertManager() {}
+
+bool NvVideoConvertManager::create_nv_video_convert() {
+    /* Use convertor to convert from NV12 to RGBA as required by nvosd */
+    nvvidconv = gst_element_factory_make("nvvideoconvert", "nvvideo-converter");
+    SET_GPU_ID(nvvidconv, GPU_ID);
+    if (!nvvidconv) {
+        g_printerr("Could not create nvvideoconvert.Exiting. \n");
+        return false;
+    }
+    return true;
+}
\ No newline at end of file
diff --git a/src/nv_video_convert_manager.hpp b/src/nv_video_convert_manager.hpp
new file mode 100644
index 0000000..d857a12
--- /dev/null
+++ b/src/nv_video_convert_manager.hpp
@@ -0,0 +1,10 @@
+#include <gst/gst.h>
+
+class NvVideoConvertManager {
+   private:
+   public:
+    GstElement *nvvidconv = NULL;
+    NvVideoConvertManager();
+    bool create_nv_video_convert();
+    ~NvVideoConvertManager();
+};
\ No newline at end of file
diff --git a/src/pipeline_manager.cpp b/src/pipeline_manager.cpp
index fe87f5e..ef0d5a4 100644
--- a/src/pipeline_manager.cpp
+++ b/src/pipeline_manager.cpp
@@ -61,6 +61,7 @@ bool PipelineManager::create_pipeline_elements(int num_sources,
     tiler_manager->create_tiler(num_sources,
                                 streammux_manager->MUXER_OUTPUT_WIDTH,
                                 streammux_manager->MUXER_OUTPUT_HEIGHT);
+    nv_video_convert_manager->create_nv_video_convert();
 
     return true;
 }
\ No newline at end of file
diff --git a/src/pipeline_manager.hpp b/src/pipeline_manager.hpp
index df086d0..451a283 100644
--- a/src/pipeline_manager.hpp
+++ b/src/pipeline_manager.hpp
@@ -3,6 +3,7 @@
 
 #include "cuda_runtime_api.h"
 #include "gstds_example_manager.hpp"
+#include "nv_video_convert_manager.hpp"
 #include "source_bin.hpp"
 #include "streammux_manager.hpp"
 #include "tiler_manager.hpp"
@@ -16,6 +17,8 @@ class PipelineManager {
     StreammuxManager *streammux_manager = new StreammuxManager();
     GstdsExampleManager *gstds_example_manager = new GstdsExampleManager();
     TilerManager *tiler_manager = new TilerManager();
+    NvVideoConvertManager *nv_video_convert_manager =
+        new NvVideoConvertManager();
 
    public:
     int current_device = -1;