[tools/onnx-subgraph] add initial main.cpp

tools/onnx_subgraph/CMakeLists.txt

@@ -14,6 +14,9 @@ find_package(Python3 COMPONENTS Interpreter Development REQUIRED)
 include_directories(${CMAKE_CURRENT_BINARY_DIR})
 include_directories(${Python3_INCLUDE_DIRS})
 
+add_executable(onnx-subgraph src/main.cpp)
+target_link_libraries(onnx-subgraph ${Python3_LIBRARIES})
+
 set(ONNX_SUBGRAPH_FILES                                                 
     test_model_download.sh
     extract_onnx.py

tools/onnx_subgraph/src/main.cpp

@@ -0,0 +1,76 @@
+/*
+ * Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved
+ *
+ * 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.
+ */
+#include <iostream>
+#include <string>
+#include <filesystem>
+namespace fs = std::filesystem;
+
+int main(int argc, char *argv[])
+{
+  std::string onnxFile;
+  std::string confFile;
+  if (argc == 3)
+  {
+    std::string arg1 = argv[1];
+    if (arg1.substr(0, 7) == "--onnx=")
+    {
+      onnxFile = arg1.substr(7);
+      if (onnxFile.empty())
+      {
+        std::cout << "No ONNX file provided." << std::endl;
+        return -1;
+      }
+
+      if (!fs::exists(onnxFile))
+      {
+        std::cout << onnxFile << " not exists." << std::endl;
+        return -1;
+      }
+      else
+      {
+        std::cout << onnxFile << " exists." << std::endl;
+      }
+    }
+
+    std::string arg2 = argv[2];
+    if (arg2.substr(0, 7) == "--conf=")
+    {
+      confFile = arg2.substr(7);
+      if (confFile.empty())
+      {
+        std::cout << "No conf file provided." << std::endl;
+        return -1;
+      }
+
+      if (!fs::exists(confFile))
+      {
+        std::cout << confFile << " not exists." << std::endl;
+        return -1;
+      }
+      else
+      {
+        std::cout << confFile << " exists." << std::endl;
+      }
+    }
+  }
+  else
+  {
+    printf("Please set valide args: ./onnx-subgraph --onnx=xxx.onnx --conf=xxx.json\n");
+    return -1;
+  }
+
+  return 0;
+}