Mercurial > hg > orthanc-stone
diff Resources/CodeGeneration/testCppHandler/main.cpp @ 495:6405435480ae bgo-commands-codegen
Fixed template to add dump capabilities + started work on an integrated TS/WASM test
author | bgo-osimis |
---|---|
date | Sat, 23 Feb 2019 14:14:32 +0100 |
parents | fc17251477d6 |
children | 4eccf698e52f |
line wrap: on
line diff
--- a/Resources/CodeGeneration/testCppHandler/main.cpp Sat Feb 23 10:18:13 2019 +0100 +++ b/Resources/CodeGeneration/testCppHandler/main.cpp Sat Feb 23 14:14:32 2019 +0100 @@ -1,18 +1,19 @@ -#include <string> -#include <filesystem> -#include <regex> -using namespace std; -namespace fs = std::filesystem; +#include <string> +#include <fstream> +#include <filesystem> +#include <regex> +using namespace std; +namespace fs = std::filesystem; #include <boost/program_options.hpp> -using namespace boost::program_options; - -#include "VsolMessages_generated.hpp" - -/** -Transforms `str` by replacing occurrences of `oldStr` with `newStr`, using -plain text (*not* regular expressions.) -*/ +using namespace boost::program_options; + +#include "VsolMessages_generated.hpp" + +/** +Transforms `str` by replacing occurrences of `oldStr` with `newStr`, using +plain text (*not* regular expressions.) +*/ static inline void ReplaceInString( string& str, const std::string& oldStr, @@ -23,23 +24,76 @@ str.replace(pos, oldStr.length(), newStr); pos += newStr.length(); } -} - -int main(int argc, char** argv) -{ - try - { - string pattern; - +} + +string SlurpFile(const string& fileName) +{ + ifstream ifs(fileName.c_str(), ios::in | ios::binary | ios::ate); + + ifstream::pos_type fileSize = ifs.tellg(); + ifs.seekg(0, ios::beg); + + vector<char> bytes(fileSize); + ifs.read(bytes.data(), fileSize); + + return string(bytes.data(), fileSize); +} + +class MyHandler : public VsolMessages::IHandler +{ +public: + virtual bool Handle(const VsolMessages::A& value) override + { + VsolMessages::StoneDumpValue(cout, value); + return true; + } + virtual bool Handle(const VsolMessages::B& value) override + { + VsolMessages::StoneDumpValue(cout, value); + return true; + } + virtual bool Handle(const VsolMessages::C& value) override + { + VsolMessages::StoneDumpValue(cout, value); + return true; + } + virtual bool Handle(const VsolMessages::Message1& value) override + { + VsolMessages::StoneDumpValue(cout, value); + return true; + } + virtual bool Handle(const VsolMessages::Message2& value) override + { + VsolMessages::StoneDumpValue(cout, value); + return true; + } +}; + +template<typename T> +void ProcessPath(T filePath) +{ + cout << "+--------------------------------------------+\n"; + cout << "| Processing: " << filePath.path().string() << "\n"; + cout << "+--------------------------------------------+\n"; + MyHandler handler; + auto contents = SlurpFile(filePath.path().string()); + VsolMessages::StoneDispatchToHandler(contents, &handler); +} + +int main(int argc, char** argv) +{ + try + { + options_description desc("Allowed options"); desc.add_options() // First parameter describes option name/short name // The second is parameter to option // The third is description ("help,h", "print usage message") - ("pattern,p", value(&pattern), "pattern for input") - ; - + ("pattern,p", value<string>(), "pattern for input") + ; + variables_map vm; store(parse_command_line(argc, argv, desc), vm); @@ -47,26 +101,39 @@ { cout << desc << "\n"; return 0; - } - - // tranform globbing pattern into regex - // we should deal with -, ., *... - string regexPatternStr = pattern; - regex regexPattern(regexPatternStr); - - for (auto& p : fs::directory_iterator(".")) - { - if (regex_match(p.path().string(), regexPattern)) - std::cout << "\"" << p << "\" is a match\n"; - else - std::cout << "\"" << p << "\" is *not* a match\n"; - } - return 0; - - - } + } + + notify(vm); + + string pattern = vm["pattern"].as<string>(); + + // tranform globbing pattern into regex + // we should deal with -, ., *... + string regexPatternStr = pattern; + cout << "Pattern is: " << regexPatternStr << endl; + ReplaceInString(regexPatternStr, "\\", "\\\\"); + ReplaceInString(regexPatternStr, "-", "\\-"); + ReplaceInString(regexPatternStr, ".", "\\."); + ReplaceInString(regexPatternStr, "*", ".*"); + ReplaceInString(regexPatternStr, "?", "."); + cout << "Corresponding regex is: " << regexPatternStr << endl; + + regex regexPattern(regexPatternStr); + + for (auto& p : fs::directory_iterator(".")) + { + auto fileName = p.path().filename().string(); + if (regex_match(fileName, regexPattern)) + { + ProcessPath(p); + } + } + return 0; + + + } catch (exception& e) { cerr << e.what() << "\n"; - } + } } \ No newline at end of file