1 files changed, 20 insertions, 1 deletions
diff --git a/llvm/tools/llvm-cov/SourceCoverageView.h b/llvm/tools/llvm-cov/SourceCoverageView.h
index 03c0742..57b5d94 100644
--- a/llvm/tools/llvm-cov/SourceCoverageView.h
+++ b/llvm/tools/llvm-cov/SourceCoverageView.h
@@ -101,7 +101,7 @@ struct LineCoverageStats {
 ///
 /// A source coverage view and its nested sub-views form a file-oriented
 /// representation of code coverage data. This view can be printed out by a
-/// renderer which implements the Rendering Interface.
+/// renderer which implements both the File Creation and Rendering interfaces.
 class SourceCoverageView {
   /// A function or file name.
   StringRef SourceName;
@@ -122,6 +122,19 @@ class SourceCoverageView {
   /// on display.
   std::vector<InstantiationView> InstantiationSubViews;
 
+public:
+  /// @name File Creation Interface
+  /// @{
+
+  /// \brief Create a file to print a coverage view into.
+  virtual Expected<std::unique_ptr<raw_ostream>>
+  createOutputFile(StringRef Path, bool InToplevel) = 0;
+
+  /// \brief Close a file which has been used to print a coverage view.
+  virtual void closeOutputFile(std::unique_ptr<raw_ostream> OS) = 0;
+
+  /// @}
+
 protected:
   struct LineRef {
     StringRef Line;
@@ -183,6 +196,12 @@ protected:
   /// digits.
   static std::string formatCount(uint64_t N);
 
+  /// \brief If directory output is enabled, create a file with \p Path as the
+  /// suffix. Otherwise, return stdout.
+  static Expected<std::unique_ptr<raw_ostream>>
+  createOutputStream(const CoverageViewOptions &Opts, StringRef Path,
+                     StringRef Extension, bool InToplevel);
+
   SourceCoverageView(StringRef SourceName, const MemoryBuffer &File,
                      const CoverageViewOptions &Options,
                      coverage::CoverageData &&CoverageInfo)