IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE Samples] restored support for multiple -i args (#9190)

Browse Source 
* restored support for  multiple -i args

* replaced return with break, moved return out of cycle
This commit is contained in:
Ivan Vikhrev 2021-12-14 11:37:50 +03:00 committed by GitHub
parent e06d9fdbb9
commit b6176fa768
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
samples/cpp/common/utils/src/args_helper.cpp
View File

@ -67,22 +67,30 @@ void readInputFilesArguments(std::vector<std::string>& files, const std::string&
*/ */
void parseInputFilesArguments(std::vector<std::string>& files) { void parseInputFilesArguments(std::vector<std::string>& files) {
std::vector<std::string> args = gflags::GetArgvs(); std::vector<std::string> args = gflags::GetArgvs();
auto args_it = begin(args);
const auto is_image_arg = [](const std::string& s) { const auto is_image_arg = [](const std::string& s) {
return s == "-i" || s == "--images"; return s == "-i" || s == "--images";
}; };
const auto is_arg = [](const std::string& s) { const auto is_arg = [](const std::string& s) {
return s.front() == '-'; return s.front() == '-';
}; };
const auto img_start = std::find_if(begin(args), end(args), is_image_arg);
while (args_it != args.end()) {
const auto img_start = std::find_if(args_it, end(args), is_image_arg);
if (img_start == end(args)) { if (img_start == end(args)) {
return; break;
} }
const auto img_begin = std::next(img_start); const auto img_begin = std::next(img_start);
const auto img_end = std::find_if(img_begin, end(args), is_arg); const auto img_end = std::find_if(img_begin, end(args), is_arg);
for (auto img = img_begin; img != img_end; ++img) { for (auto img = img_begin; img != img_end; ++img) {
readInputFilesArguments(files, *img); readInputFilesArguments(files, *img);
} }
args_it = img_end;
}
if (files.empty()) {
return;
}
size_t max_files = 20; size_t max_files = 20;
if (files.size() < max_files) { if (files.size() < max_files) {
slog::info << "Files were added: " << files.size() << slog::endl; slog::info << "Files were added: " << files.size() << slog::endl;