From 81202236d40adfaf9c1bda4cced3f389b230d35e Mon Sep 17 00:00:00 2001
From: Felix Weiglhofer <weiglhofer@fias.uni-frankfurt.de>
Date: Wed, 14 Jun 2023 14:57:25 +0000
Subject: [PATCH] Options.cxx: Add flag for output-file.
---
algo/base/Options.cxx | 4 ++++
algo/base/Options.h | 4 ++++
reco/app/cbmreco/main.cxx | 12 ++++++++++--
3 files changed, 18 insertions(+), 2 deletions(-)
diff --git a/algo/base/Options.cxx b/algo/base/Options.cxx
index 0bdeba4501..4c1747f50a 100644
--- a/algo/base/Options.cxx
+++ b/algo/base/Options.cxx
@@ -59,6 +59,10 @@ Options::Options(int argc, char** argv)
po::options_description generic("Other options");
// clang-format off
generic.add_options()
+ ("output,o", po::value(&fOutputFile)->default_value("")->value_name("<file>"),
+ "write results to file")
+ ("write-per-ts,W", po::bool_switch(&fWritePerTimeslice)->default_value(false),
+ "Write results to file per timeslice (resulting files are named <file>_<tsnr>). Requires -o.")
("device,d", po::value(&fDevice)->default_value("cpu")->value_name("<device>"),
"select device (cpu, cuda0, cuda1, hip0, ...)")
("log-level,l", po::value(&fLogLevel)->default_value(info)->value_name("<level>"),
diff --git a/algo/base/Options.h b/algo/base/Options.h
index 58c5574ec3..1da91be39a 100644
--- a/algo/base/Options.h
+++ b/algo/base/Options.h
@@ -23,6 +23,8 @@ namespace cbm::algo
fs::path ParamsDir() const { return fParamsDir; }
const std::string& InputLocator() const { return fInputLocator; }
+ fs::path OutputFile() const { return fOutputFile; }
+ bool WritePerTimeslice() const { return fWritePerTimeslice; }
severity_level LogLevel() const { return fLogLevel; }
fs::path LogFile() const { return fLogFile; }
const std::string& Device() const { return fDevice; }
@@ -49,6 +51,8 @@ namespace cbm::algo
private:
std::string fParamsDir; // TODO: can we make this a std::path?
std::string fInputLocator;
+ std::string fOutputFile;
+ bool fWritePerTimeslice = false;
severity_level fLogLevel;
std::string fLogFile;
std::string fDevice;
diff --git a/reco/app/cbmreco/main.cxx b/reco/app/cbmreco/main.cxx
index 63aea0df5e..dd694294c7 100644
--- a/reco/app/cbmreco/main.cxx
+++ b/reco/app/cbmreco/main.cxx
@@ -46,7 +46,7 @@ int main(int argc, char** argv)
reco.Init(opts);
- Archive archive(opts.Detectors());
+ Archive archive({fles::SubsystemIdentifier::STS}); // TODO: use opts.Detector() once detector flag is merged
fles::TimesliceAutoSource source {opts.InputLocator()};
int tsIdx = 0;
int num_ts = opts.NumTimeslices();
@@ -57,12 +57,20 @@ int main(int argc, char** argv)
continue;
}
auto result = reco.Run(*ts);
- archive.TimesliceResults().push_back(result);
+
+ if (opts.WritePerTimeslice() && !opts.OutputFile().empty()) {
+ Archive tsResults({fles::SubsystemIdentifier::STS}); // TODO: use opts.Detector() once detector flag is merged
+ tsResults.TimesliceResults().push_back(result);
+ // TODO: handle file endings
+ tsResults.Store(fmt::format("{}_{}", opts.OutputFile().string(), ts->index()));
+ }
+ archive.TimesliceResults().push_back(result); // TODO: result should be moved not copied
tsIdx++;
if (num_ts > 0 && tsIdx >= num_ts) { break; }
}
+ if (!opts.WritePerTimeslice() && !opts.OutputFile().empty()) archive.Store(opts.OutputFile());
reco.Finalize();
return 0;
--
GitLab