1 files changed, 84 insertions, 0 deletions
diff --git a/llvm/tools/llvm-exegesis/lib/Analysis.cpp b/llvm/tools/llvm-exegesis/lib/Analysis.cpp
new file mode 100644
index 00000000000..d6b39027445
--- /dev/null
+++ b/llvm/tools/llvm-exegesis/lib/Analysis.cpp
@@ -0,0 +1,84 @@
+//===-- Analysis.cpp --------------------------------------------*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "BenchmarkResult.h"
+#include "Analysis.h"
+#include "llvm/Support/FormatVariadic.h"
+#include <vector>
+
+namespace exegesis {
+
+static const char kCsvSep = ',';
+
+static void writeCsvEscaped(llvm::raw_ostream& OS, const std::string& S) {
+  if (std::find(S.begin(), S.end(), kCsvSep) == S.end()) {
+    OS << S;
+  } else {
+    // Needs escaping.
+    OS << '"';
+    for (const char C : S) {
+      if (C == '"')
+        OS << "\"\"";
+      else
+        OS << C;
+    }
+    OS << '"';
+  }
+}
+
+// Prints a row representing an instruction, along with scheduling info and
+// point coordinates (measurements).
+static void printInstructionRow(const InstructionBenchmark &Point,
+                         const llvm::MCSubtargetInfo &STI,
+                         const size_t ClusterId, llvm::raw_ostream &OS) {
+  OS << ClusterId << kCsvSep;
+  writeCsvEscaped(OS, Point.AsmTmpl.Name);
+  // FIXME: Print the sched class once InstructionBenchmark separates key into
+  // (mnemonic, mode, opaque).
+  for (const auto &Measurement : Point.Measurements) {
+    OS << kCsvSep;
+    writeCsvEscaped(OS, llvm::formatv("{0:F}", Measurement.Value));
+  }
+  OS << "\n";
+}
+
+static void printCluster(const std::vector<InstructionBenchmark> &Points,
+                  const llvm::MCSubtargetInfo &STI, const size_t ClusterId,
+                  const InstructionBenchmarkClustering::Cluster &Cluster,
+                  llvm::raw_ostream &OS) {
+  // Print all points.
+  for (const auto &PointId : Cluster.PointIndices) {
+    printInstructionRow(Points[PointId], STI, ClusterId, OS);
+  }
+}
+
+llvm::Error
+printAnalysisClusters(const InstructionBenchmarkClustering &Clustering,
+                      const llvm::MCSubtargetInfo &STI, llvm::raw_ostream &OS) {
+  if (Clustering.getPoints().empty())
+    return llvm::Error::success();
+
+  // Write the header.
+  OS << "cluster_id;key,sched_class";
+  for (const auto &Measurement : Clustering.getPoints().front().Measurements) {
+    OS << kCsvSep;
+    writeCsvEscaped(OS, Measurement.Key);
+  }
+  OS << "\n";
+
+  // Write the points.
+  for (size_t I = 0, E = Clustering.getValidClusters().size(); I < E; ++I) {
+    printCluster(Clustering.getPoints(), STI, I,
+                 Clustering.getValidClusters()[I], OS);
+    OS << "\n\n";
+  }
+  return llvm::Error::success();
+}
+
+} // namespace exegesis