max/poet
1
0
Fork 0
mirror of https://git.gfz-potsdam.de/naaice/poet.git synced 2025-12-16 04:48:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Refactor command line argument handling and improve script sourcing in initializer.cpp

Browse Source
This commit is contained in:
Max Luebke 2024-04-08 12:49:58 +00:00
parent 89cc239583
commit d5e4b03339
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/initializer.cpp
View File

@ -12,13 +12,15 @@
int main(int argc, char **argv) { int main(int argc, char **argv) {
argh::parser cmdl(argc, argv); argh::parser cmdl({"-o", "--output"});
cmdl.parse(argc, argv);
if (cmdl[{"-h", "--help"}] || cmdl.pos_args().size() != 2) { if (cmdl[{"-h", "--help"}] || cmdl.pos_args().size() != 2) {
std::cout << "Usage: " << argv[0] std::cout << "Usage: " << argv[0]
<< " [-o, --output output_file] " << " [-o, --output output_file]"
"[--setwd] " " [-s, --setwd] "
"<script.R>" " <script.R>"
<< std::endl; << std::endl;
return EXIT_SUCCESS; return EXIT_SUCCESS;
} }
@ -50,11 +52,13 @@ int main(int argc, char **argv) {
} }
std::string output_file; std::string output_file;
cmdl({"-o", "--output"}, cmdl({"-o", "--output"},
in_file_name.substr(0, in_file_name.find_last_of('.')) + ".rds") >> curr_path + "/" +
in_file_name.substr(0, in_file_name.find_last_of('.')) + ".rds") >>
output_file; output_file;
if (cmdl[{"--setwd"}]) { if (cmdl[{"-s", "--setwd"}]) {
const std::string dir_path = Rcpp::as<std::string>( const std::string dir_path = Rcpp::as<std::string>(
Rcpp::Function("dirname")(normalized_path_script)); Rcpp::Function("dirname")(normalized_path_script));
@ -69,7 +73,8 @@ int main(int argc, char **argv) {
// replace file extension by .rds // replace file extension by .rds
Rcpp::Function save("saveRDS"); Rcpp::Function save("saveRDS");
save(init.exportList(), Rcpp::wrap(curr_path + "/" + output_file));
save(init.exportList(), Rcpp::wrap(output_file));
std::cout << "Saved result to " << output_file << std::endl; std::cout << "Saved result to " << output_file << std::endl;