gRPC-based Endoscopy Tool Tracking application with Data Flow Tracking

Signed-off-by: Victor Chang <[email protected]>
nvidia-holoscan · Nov 14, 2024 · 788adb7 · 788adb7
1 parent 78fc53b
commit 788adb7
Show file tree

Hide file tree

Showing 25 changed files with 1,612 additions and 4 deletions.
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -915,6 +915,76 @@
         },
         //#endregion ucx_h264_endoscopy_tool_tracking
 
+
+        //#region grpc_endoscopy_tool_tracking
+        {
+            "name": "(gdb) grpc_endoscopy_tool_tracking/cpp (edge)",
+            "type": "cppdbg",
+            "request": "launch",
+            "preLaunchTask": "Build grpc_endoscopy_tool_tracking (delay 3s)",
+            "program": "${workspaceFolder}/build/grpc_endoscopy_tool_tracking/applications/distributed/grpc/grpc_endoscopy_tool_tracking/cpp/grpc_endoscopy_tool_tracking_edge",
+            "environment": [
+                {
+                    "name": "HOLOSCAN_INPUT_PATH",
+                    "value": "${env:HOLOHUB_DATA_DIR}/endoscopy"
+                },
+                {
+                    "name": "HOLOSCAN_LOG_LEVEL",
+                    "value": "INFO"
+                }
+            ],
+            "stopAtEntry": false,
+            "cwd": "${workspaceFolder}/build/grpc_endoscopy_tool_tracking/applications/distributed/grpc/grpc_endoscopy_tool_tracking/cpp",
+            "externalConsole": false,
+            "MIMode": "gdb",
+            "setupCommands": [
+                {
+                    "description": "Enable pretty-printing for gdb",
+                    "text": "-enable-pretty-printing",
+                    "ignoreFailures": true
+                }
+            ],
+            "presentation": {
+                "hidden": false,
+                "group": "grpc_endoscopy_tool_tracking",
+                "order": 1
+            }
+        },
+        {
+            "name": "(gdb) grpc_endoscopy_tool_tracking/cpp (cloud)",
+            "type": "cppdbg",
+            "request": "launch",
+            "preLaunchTask": "Build grpc_endoscopy_tool_tracking",
+            "program": "${workspaceFolder}/build/grpc_endoscopy_tool_tracking/applications/distributed/grpc/grpc_endoscopy_tool_tracking/cpp/grpc_endoscopy_tool_tracking_cloud",
+            "environment": [
+                {
+                    "name": "HOLOSCAN_INPUT_PATH",
+                    "value": "${env:HOLOHUB_DATA_DIR}/endoscopy"
+                },
+                {
+                    "name": "HOLOSCAN_LOG_LEVEL",
+                    "value": "INFO"
+                }
+            ],
+            "stopAtEntry": false,
+            "cwd": "${workspaceFolder}/build/grpc_endoscopy_tool_tracking/applications/distributed/grpc/grpc_endoscopy_tool_tracking/cpp",
+            "externalConsole": false,
+            "MIMode": "gdb",
+            "setupCommands": [
+                {
+                    "description": "Enable pretty-printing for gdb",
+                    "text": "-enable-pretty-printing",
+                    "ignoreFailures": true
+                }
+            ],
+            "presentation": {
+                "hidden": false,
+                "group": "grpc_endoscopy_tool_tracking",
+                "order": 1
+            }
+        },
+        //#endregion grpc_endoscopy_tool_tracking
+
         //#region grpc_h264_endoscopy_tool_tracking
         {
             "name": "(gdb) grpc_h264_endoscopy_tool_tracking/cpp (edge)",
@@ -1227,6 +1297,20 @@
                 "order": 11
             }
         },
+        {
+            "name": "(compound) grpc_endoscopy_tool_tracking/cpp (cloud & edge)",
+            "configurations": [
+                "(gdb) grpc_endoscopy_tool_tracking/cpp (cloud)",
+                "(gdb) grpc_endoscopy_tool_tracking/cpp (edge)"
+            ],
+            "preLaunchTask": "Build grpc_endoscopy_tool_tracking",
+            "stopAll": true,
+            "presentation": {
+                "hidden": false,
+                "group": "grpc_endoscopy_tool_tracking",
+                "order": 11
+            }
+        },
         {
             "name": "(compound) grpc_h264_endoscopy_tool_tracking/cpp (cloud & edge)",
             "configurations": [

diff --git a/.vscode/tasks.json b/.vscode/tasks.json
@@ -272,6 +272,53 @@
                 "group": "ucx_endoscopy_tool_tracking"
             }
         },
+        {
+            "type": "shell",
+            "label": "Build grpc_endoscopy_tool_tracking",
+            "command": "./run",
+            "args": [
+                "build",
+                "grpc_endoscopy_tool_tracking",
+                "--type",
+                "debug"
+            ],
+            "options": {
+                "cwd": "${env:WORKSPACE_DIR}"
+            },
+            "group": "build",
+            "problemMatcher": [],
+            "detail": "CMake template build task",
+            "presentation": {
+                "echo": true,
+                "reveal": "silent",
+                "focus": true,
+                "panel": "dedicated",
+                "showReuseMessage": false,
+                "clear": true,
+                "group": "grpc_endoscopy_tool_tracking"
+            }
+        },
+        {
+            "type": "shell",
+            "label": "Build grpc_endoscopy_tool_tracking (delay 3s)",
+            "command": "sleep 3",
+            "dependsOn": "Build grpc_endoscopy_tool_tracking",
+            "options": {
+                "cwd": "${env:WORKSPACE_DIR}"
+            },
+            "group": "build",
+            "problemMatcher": [],
+            "detail": "CMake template build task",
+            "presentation": {
+                "echo": true,
+                "reveal": "silent",
+                "focus": true,
+                "panel": "dedicated",
+                "showReuseMessage": false,
+                "clear": true,
+                "group": "grpc_endoscopy_tool_tracking"
+            }
+        },
         {
             "type": "shell",
             "label": "Build h264_endoscopy_tool_tracking",
@@ -332,6 +379,53 @@
             "problemMatcher": [],
             "detail": "CMake template build task"
         },
+        {
+            "type": "shell",
+            "label": "Build grpc_endoscopy_tool_tracking",
+            "command": "./run",
+            "args": [
+                "build",
+                "grpc_endoscopy_tool_tracking",
+                "--type",
+                "debug"
+            ],
+            "options": {
+                "cwd": "${env:WORKSPACE_DIR}"
+            },
+            "group": "build",
+            "problemMatcher": [],
+            "detail": "CMake template build task",
+            "presentation": {
+                "echo": true,
+                "reveal": "silent",
+                "focus": true,
+                "panel": "dedicated",
+                "showReuseMessage": false,
+                "clear": true,
+                "group": "grpc_endoscopy_tool_tracking"
+            }
+        },
+        {
+            "type": "shell",
+            "label": "Build grpc_endoscopy_tool_tracking (delay 3s)",
+            "command": "sleep 3",
+            "dependsOn": "Build grpc_endoscopy_tool_tracking",
+            "options": {
+                "cwd": "${env:WORKSPACE_DIR}"
+            },
+            "group": "build",
+            "problemMatcher": [],
+            "detail": "CMake template build task",
+            "presentation": {
+                "echo": true,
+                "reveal": "silent",
+                "focus": true,
+                "panel": "dedicated",
+                "showReuseMessage": false,
+                "clear": true,
+                "group": "grpc_endoscopy_tool_tracking"
+            }
+        },
         {
             "type": "shell",
             "label": "Build grpc_h264_endoscopy_tool_tracking",

diff --git a/applications/distributed/grpc/CMakeLists.txt b/applications/distributed/grpc/CMakeLists.txt
@@ -14,6 +14,12 @@
 # limitations under the License.
 
 
+
+add_holohub_application(grpc_endoscopy_tool_tracking DEPENDS
+                        OPERATORS lstm_tensor_rt_inference
+                                  tool_tracking_postprocessor
+                                  grpc_operators)
+
 add_holohub_application(grpc_h264_endoscopy_tool_tracking DEPENDS
                         OPERATORS video_encoder
                                   tensor_to_video_buffer

diff --git a/applications/distributed/grpc/grpc_endoscopy_tool_tracking/CMakeLists.txt b/applications/distributed/grpc/grpc_endoscopy_tool_tracking/CMakeLists.txt
@@ -0,0 +1,35 @@
+# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+cmake_minimum_required(VERSION 3.20)
+project(grpc_endoscopy_tool_tracking LANGUAGES NONE)
+
+# Download the endoscopy sample data
+if(HOLOHUB_DOWNLOAD_DATASETS)
+  include(holoscan_download_data)
+  holoscan_download_data(endoscopy
+    URL https://api.ngc.nvidia.com/v2/resources/nvidia/clara-holoscan/holoscan_endoscopy_sample_data/versions/20230222/zip
+    DOWNLOAD_NAME holoscan_endoscopy_sample_data_20230222.zip
+    URL_MD5 d54f84a562d29ed560a87d2607eba973
+    DOWNLOAD_DIR ${HOLOHUB_DATA_DIR}
+    GENERATE_GXF_ENTITIES
+    GXF_ENTITIES_HEIGHT 480
+    GXF_ENTITIES_WIDTH 854
+    GXF_ENTITIES_CHANNELS 3
+    GXF_ENTITIES_FRAMERATE 30
+  )
+endif()
+
+add_subdirectory(cpp)