[metrics] Add option to export to .csv

Author: kostorr

src/CommandLineUtilities/ProgramMetrics.cxx

@@ -13,6 +13,7 @@
 ///
 /// \author Kostas Alexopoulos ([email protected])
 
+#include <fstream>
 #include <iostream>
 #include "Cru/Constants.h"
 #include "ReadoutCard/ChannelFactory.h"
@@ -41,6 +42,10 @@ class ProgramMetrics: public Program
   virtual void addOptions(boost::program_options::options_description& options)
   {
     Options::addOptionCardId(options);
+    options.add_options()
+      ("csv-out",
+       po::value<std::string>(&mOptions.csvOut),
+       "Path to CSV file to write output");
   }
 
   virtual void run(const boost::program_options::variables_map& map) 
@@ -63,9 +68,14 @@ class ProgramMetrics: public Program
     } else {
       std::cout << "Something went wrong parsing the card id" << std::endl;
     }
+
+    if (mOptions.csvOut != "") {
+      mOutputToCsv = true;
+    }
+
+    std::ofstream csvOut;
 
     std::ostringstream table;
-
     auto formatHeader = "  %-3s %-6s %-10s %-10s %-19s %-20s %-19s %-8s %-17s %-17s\n";
     auto formatRow = "  %-3s %-6s %-10s %-10s %-19s %-20s %-19s %-8s %-17s %-17s\n";
     auto header = (boost::format(formatHeader)
@@ -74,7 +84,13 @@ class ProgramMetrics: public Program
     auto lineFat = std::string(header.length(), '=') + '\n';
     auto lineThin = std::string(header.length(), '-') + '\n';
 
-    table << lineFat << header << lineThin;
+    if (mOutputToCsv) {
+      csvOut.open(mOptions.csvOut);
+      auto csvHeader = "#,Type,PCI Addr,Temp (C),#Dropped Packets,CTP Clock (MHz),Local Clock (MHz),#links,#Wrapper 0 links, #Wrapper 1 links\n"; 
+      csvOut << csvHeader;
+    } else {
+      table << lineFat << header << lineThin;
+    }
 
     int i = 0;
     for (const auto& card : cardsFound) {
@@ -91,18 +107,37 @@ class ProgramMetrics: public Program
       uint32_t links0 = bar2->getLinksPerWrapper(0);
       uint32_t links1 = bar2->getLinksPerWrapper(1);
 
-      auto format = boost::format(formatRow) % i % CardType::toString(card.cardType) % card.pciAddress.toString()
-        % temperature % dropped % ctp_clock % local_clock % links % links0 % links1;
+      if (mOutputToCsv) {
+        auto csvLine =  std::to_string(i) + "," +  CardType::toString(card.cardType) + "," +  card.pciAddress.toString() + "," +  std::to_string(temperature)
+          + "," +  std::to_string(dropped) + "," +  std::to_string(ctp_clock) + "," +  std::to_string(local_clock) + "," +  std::to_string(links) + ","
+          +  std::to_string(links0) + "," +  std::to_string(links1) + "\n";
+        csvOut << csvLine;
+      } else {
+        auto format = boost::format(formatRow) % i % CardType::toString(card.cardType) % card.pciAddress.toString()
+          % temperature % dropped % ctp_clock % local_clock % links % links0 % links1;
 
-      table << format;
+        table << format;
+      }
       i++;
     }
 
-    table << lineFat;
-    std::cout << table.str();
+    if (mOutputToCsv) {
+      csvOut.close();
+    } else {
+      auto lineFat = std::string(header.length(), '=') + '\n';
+      table << lineFat;
+      std::cout << table.str();
+    }
   }
 
-  private:
+private:
+
+  struct OptionsStruct {
+    std::string csvOut = "";
+  } mOptions;
+
+  bool mOutputToCsv = false;
+
 };
 
 int main(int argc, char** argv)