1 files changed, 54 insertions, 0 deletions
diff --git a/include/llvm/Profile/ProfileDataWriter.h b/include/llvm/Profile/ProfileDataWriter.h
new file mode 100644
index 0000000000..7e7802a83c
--- /dev/null
+++ b/include/llvm/Profile/ProfileDataWriter.h
@@ -0,0 +1,54 @@
+//=-- ProfileDataWriter.h - Instrumented profiling writer ---------*- C++ -*-=//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file contains support for writing profiling data for instrumentation
+// based PGO and coverage.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_PROFILE_PROFILEDATA_WRITER_H__
+#define LLVM_PROFILE_PROFILEDATA_WRITER_H__
+
+#include "llvm/ADT/StringMap.h"
+#include "llvm/Support/DataTypes.h"
+#include "llvm/Support/raw_ostream.h"
+
+#include <vector>
+
+namespace llvm {
+
+struct __attribute__((packed)) ProfileDataHeader {
+  char     Magic[4];
+  uint32_t Version;
+  uint32_t DataStart;
+  uint32_t Padding;
+  uint64_t MaxFunctionCount;
+};
+
+/// Writer for instrumentation based profile data
+class ProfileDataWriter {
+  StringMap<size_t> FunctionOffsets;
+  std::vector<uint64_t> FunctionData;
+  uint32_t DataStart;
+  uint64_t MaxFunctionCount;
+
+  void write32(raw_ostream &OS, uint32_t Value);
+  void write64(raw_ostream &OS, uint64_t Value);
+public:
+  ProfileDataWriter()
+      : DataStart(sizeof(ProfileDataHeader)), MaxFunctionCount(0) {}
+
+  void addFunctionCounts(StringRef FuncName, uint64_t FunctionHash,
+                         uint64_t NumCounters, const uint64_t *Counters);
+  void write(raw_ostream &OS);
+};
+
+} // end namespace llvm
+
+#endif // LLVM_PROFILE_PROFILEDATA_WRITER_H__