Upload Audio_Effects_SDK/samples/utils/ConfigReader.cpp with huggingface_hub

Audio_Effects_SDK/samples/utils/ConfigReader.cpp

@@ -0,0 +1,130 @@
+/*
+Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
+
+NVIDIA CORPORATION and its licensors retain all intellectual property
+and proprietary rights in and to this software, related documentation
+and any modifications thereto. Any use, reproduction, disclosure or
+distribution of this software and related documentation without an express
+license agreement from NVIDIA CORPORATION is strictly prohibited.
+*/
+
+#include "ConfigReader.hpp"
+
+#include <fstream>
+#include <iostream>
+#include <sstream>
+
+bool ConfigReader::Load(const std::string & config_filename)
+{
+  std::ifstream configFile;
+  configFile.open(config_filename, std::ios_base::in);
+  int line_number = 0;
+  if (configFile.is_open()) {
+    // read stream line by line
+    for (std::string line; std::getline(configFile, line);) {
+      line_number++;
+
+      // Strip whitespace
+      std::size_t start = line.find_first_not_of(" \n\r\t");
+      if (start == std::string::npos) {
+        continue;
+      }
+      line = line.substr(start);
+      // Ignore comments
+      if (line[0] == '#') {
+        continue;
+      }
+
+      // Get the name
+      std::size_t end = line.find_first_of(" \t");
+      if (end == std::string::npos) {
+        std::cerr << "Invalid config file at line " << line_number << std::endl;
+        return false;
+      }
+      std::string name = line.substr(start, end - start);
+      line = line.substr(end);
+
+      // Strip whitespace between name value pair
+      start = line.find_first_not_of(" \t");
+      if (start == std::string::npos) {
+        std::cerr << "Invalid config file at line " << line_number << std::endl;
+        return false;
+      }
+      std::string value = line.substr(start);
+#ifdef __linux__
+      // Remove trailing \r s
+      if (value.size() && value[value.size() -1] == '\r')
+      {
+        value.pop_back();
+      }
+#endif
+      config_dict_.insert({ name, value });
+    }
+    configFile.close();
+  }
+  else {
+    std::cerr << "Unable to open file: " << config_filename << std::endl;
+    return false;
+  }
+
+  loaded_ = true;
+  return true;
+}
+
+bool ConfigReader::IsConfigValueAvailable(const std::string& name) const {
+  if (!loaded_) {
+    std::cout << "Not loaded" << std::endl;
+    return false;
+  }
+
+  return config_dict_.find(name) != config_dict_.end();
+}
+
+bool ConfigReader::GetConfigValue(const std::string& name, std::string* value) const {
+  if (!loaded_) {
+    std::cout << "Not loaded" << std::endl;
+    return false;
+  }
+
+  auto iter = config_dict_.find(name);
+  if (iter == config_dict_.end()) {
+    std::cerr << "Unable to find \"" << name << " effect\" configuration" << std::endl;
+    return false;
+  }
+
+  *value = iter->second;
+  return true;
+}
+
+std::string ConfigReader::GetConfigValue(const std::string & name) const
+{
+  std::string value;
+  if (GetConfigValue(name, &value) == false) {
+    std::cerr << "Config value not found" << std::endl;
+    return value;
+  }
+
+  return value;
+}
+
+std::vector<std::string> ConfigReader::GetConfigValueList(const std::string & name) const
+{
+  std::string value_list;
+  if (GetConfigValue(name, &value_list) == false) {
+    std::cerr << "Config value list not found" << std::endl;
+    return std::vector<std::string>();
+  }
+
+  std::vector<std::string> list;
+  // Split string around spaces.
+  std::istringstream ss(value_list);
+  do {
+    std::string value;
+    ss >> value;
+    if (!value.empty())
+      list.push_back(value);
+  } while (ss);
+
+  return list;
+}
+