[AddressLowering] NFC: Added print.

Allow in-IR tests to print to stdout.

Author: nate-chandler

lib/SILOptimizer/Mandatory/AddressLowering.cpp

@@ -433,51 +433,58 @@ void ValueStorageMap::replaceValue(SILValue oldValue, SILValue newValue) {
 }
 
 #ifndef NDEBUG
-void ValueStorage::dump() const {
-  llvm::dbgs() << "projectedStorageID: " << projectedStorageID << "\n";
-  llvm::dbgs() << "projectedOperandNum: " << projectedOperandNum << "\n";
-  llvm::dbgs() << "isDefProjection: " << isDefProjection << "\n";
-  llvm::dbgs() << "isUseProjection: " << isUseProjection << "\n";
-  llvm::dbgs() << "isRewritten: " << isRewritten << "\n";
-  llvm::dbgs() << "initializes: " << initializes << "\n";
+void ValueStorage::print(llvm::raw_ostream &OS) const {
+  OS << "projectedStorageID: " << projectedStorageID << "\n";
+  OS << "projectedOperandNum: " << projectedOperandNum << "\n";
+  OS << "isDefProjection: " << isDefProjection << "\n";
+  OS << "isUseProjection: " << isUseProjection << "\n";
+  OS << "isRewritten: " << isRewritten << "\n";
+  OS << "initializes: " << initializes << "\n";
 }
-void ValueStorageMap::ValueStoragePair::dump() const {
-  llvm::dbgs() << "value: ";
-  value->dump();
-  llvm::dbgs() << "address:  ";
+void ValueStorage::dump() const { print(llvm::dbgs()); }
+void ValueStorageMap::ValueStoragePair::print(llvm::raw_ostream &OS) const {
+  OS << "value: ";
+  value->print(OS);
+  OS << "address:  ";
   if (storage.storageAddress)
-    storage.storageAddress->dump();
+    storage.storageAddress->print(OS);
   else
-    llvm::dbgs() << "UNKNOWN!\n";
-  storage.dump();
+    OS << "UNKNOWN!\n";
+  storage.print(OS);
 }
-void ValueStorageMap::dumpProjections(SILValue value) const {
+void ValueStorageMap::ValueStoragePair::dump() const { print(llvm::dbgs()); }
+void ValueStorageMap::printProjections(SILValue value,
+                                       llvm::raw_ostream &OS) const {
   for (auto *pair : getProjections(value)) {
-    pair->dump();
+    pair->print(OS);
   }
 }
-void ValueStorageMap::dump() const {
-  llvm::dbgs() << "ValueStorageMap:\n";
+void ValueStorageMap::dumpProjections(SILValue value) const {
+  printProjections(value, llvm::dbgs());
+}
+void ValueStorageMap::print(llvm::raw_ostream &OS) const {
+  OS << "ValueStorageMap:\n";
   for (unsigned ordinal : indices(valueVector)) {
     auto &valStoragePair = valueVector[ordinal];
-    llvm::dbgs() << "value: ";
-    valStoragePair.value->dump();
+    OS << "value: ";
+    valStoragePair.value->print(OS);
     auto &storage = valStoragePair.storage;
     if (storage.isUseProjection) {
-      llvm::dbgs() << "  use projection: ";
+      OS << "  use projection: ";
       if (!storage.isRewritten)
-        valueVector[storage.projectedStorageID].value->dump();
+        valueVector[storage.projectedStorageID].value->print(OS);
     } else if (storage.isDefProjection) {
-      llvm::dbgs() << "  def projection: ";
+      OS << "  def projection: ";
       if (!storage.isRewritten)
-        valueVector[storage.projectedStorageID].value->dump();
+        valueVector[storage.projectedStorageID].value->print(OS);
     }
     if (storage.storageAddress) {
-      llvm::dbgs() << "  storage: ";
-      storage.storageAddress->dump();
+      OS << "  storage: ";
+      storage.storageAddress->print(OS);
     }
   }
 }
+void ValueStorageMap::dump() const { print(llvm::dbgs()); }
 #endif
 
 //===----------------------------------------------------------------------===//

lib/SILOptimizer/Mandatory/AddressLowering.h

@@ -17,6 +17,10 @@
 #include "llvm/ADT/DenseMap.h"
 #include "llvm/ADT/STLExtras.h"
 
+namespace llvm {
+class raw_ostream;
+}
+
 namespace swift {
 
 /// Track an opaque value's storage. An opaque value is a SILValue with
@@ -199,6 +203,7 @@ struct ValueStorage {
   }
 
 #ifndef NDEBUG
+  void print(llvm::raw_ostream &OS) const;
   void dump() const;
 #endif
 };
@@ -215,6 +220,7 @@ class ValueStorageMap {
     ValueStorage storage;
     ValueStoragePair(SILValue v, ValueStorage s) : value(v), storage(s) {}
 #ifndef NDEBUG
+    void print(llvm::raw_ostream &OS) const;
     void dump() const;
 #endif
   };
@@ -396,7 +402,9 @@ class ValueStorageMap {
   bool isComposingUseProjection(Operand *oper) const;
 
 #ifndef NDEBUG
+  void printProjections(SILValue value, llvm::raw_ostream &OS) const;
   void dumpProjections(SILValue value) const;
+  void print(llvm::raw_ostream &OS) const;
   void dump() const;
 #endif
 };