artdaq  v3_07_01
TransferTest.hh
1 #ifndef ARTDAQ_TEST_DAQRATE_TRANSFERTEST_HH
2 #define ARTDAQ_TEST_DAQRATE_TRANSFERTEST_HH
3 
4 #include <chrono>
5 #include <cmath>
6 #include <string>
7 #include <vector>
8 
9 #include "artdaq-utilities/Plugins/MetricManager.hh"
10 #include "fhiclcpp/ParameterSet.h"
11 
12 namespace artdaq {
17 {
18 public:
36  explicit TransferTest(fhicl::ParameterSet psi);
37 
42  int runTest();
43 
44 private:
45  std::pair<size_t, double> do_sending(int thread_index);
46 
47  std::pair<size_t, double> do_receiving();
48 
49  //Helper functions
50  const std::vector<std::string> suffixes{" B", " KB", " MB", " GB", " TB"};
51 
52  std::string formatBytes(double bytes, size_t suffixIndex = 0);
53 
54  int senders_;
55  int receivers_;
56  int sending_threads_;
57  int sends_each_sender_;
58  int receives_each_receiver_; // Should be sends_each_sender * sending_threads * sending_ranks / receiving_ranks
59  int buffer_count_;
60  int error_count_max_;
61  size_t fragment_size_;
62  std::chrono::steady_clock::time_point start_time_;
63  fhicl::ParameterSet ps_;
64  bool validate_mode_;
65  int partition_number_;
66 };
67 
68 inline std::string TransferTest::formatBytes(double bytes, size_t suffixIndex)
69 {
70  auto b = fabs(bytes);
71 
72  if (b > 1024.0 && suffixIndex < suffixes.size())
73  {
74  return formatBytes(bytes / 1024.0, suffixIndex + 1);
75  }
76 
77  return std::to_string(bytes) + suffixes[suffixIndex];
78 }
79 } // namespace artdaq
80 #endif //ARTDAQ_TEST_DAQRATE_TRANSFERTEST_HH
int runTest()
Run the test as configured.
TransferTest(fhicl::ParameterSet psi)
TransferTest Constructor.
Definition: TransferTest.cc:14
Test a set of TransferInterface plugins.
Definition: TransferTest.hh:16