From e7175e606f61a823727951864dc5f3dcb3052920 Mon Sep 17 00:00:00 2001 From: Jonathan Lifflander Date: Thu, 24 Mar 2022 15:23:21 -0700 Subject: [PATCH] #1717: lb: add .br extension for compressed data files --- src/vt/configs/arguments/args.cc | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/src/vt/configs/arguments/args.cc b/src/vt/configs/arguments/args.cc index 1f3b92a0a7..098982630c 100644 --- a/src/vt/configs/arguments/args.cc +++ b/src/vt/configs/arguments/args.cc @@ -776,7 +776,9 @@ std::tuple ArgConfig::parseToConfig( } namespace { -static std::string buildFile(std::string const& file, std::string const& dir) { +static std::string buildFile( + std::string const& file, std::string const& dir, bool add_br_extension +) { std::string name = file; std::size_t rank = name.find("%p"); auto str_rank = std::to_string(theContext()->getNode()); @@ -785,17 +787,22 @@ static std::string buildFile(std::string const& file, std::string const& dir) { } else { name.replace(rank, 2, str_rank); } + if (add_br_extension) { + if (name.substr(name.length()-3, 3) != ".br") { + name = name + ".br"; + } + } return dir + "/" + name; } } /* end anon namespace */ std::string AppConfig::getLBStatsFileOut() const { - return buildFile(vt_lb_stats_file, vt_lb_stats_dir); + return buildFile(vt_lb_stats_file, vt_lb_stats_dir, vt_lb_stats_compress); } std::string AppConfig::getLBStatsFileIn() const { - return buildFile(vt_lb_stats_file_in, vt_lb_stats_dir_in); + return buildFile(vt_lb_stats_file_in, vt_lb_stats_dir_in, false); } }} /* end namespace vt::arguments */