commit e787b32debe4e13b363112cbad6325531165e934
author Chris Lattner <clattner@google.com> Wed Aug 08 11:14:57 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 12:57:38 2019 -0700
tree a3af976cd838f208a7d73400beceeb0cdbee2fe3
parent 8bdbebf142ecd50ea412f0ae0314a3e9d4f2aec7

Rework the cloning infrastructure for statements to be able to take and update
an operand mapping, which simplifies it a bit.  Implement cloning for IfStmt,
rename getThenClause() to getThen() which is unambiguous and less repetitive in
use cases.

PiperOrigin-RevId: 207915990

lib/IR/AsmPrinter.cpp

diff --git a/lib/IR/AsmPrinter.cpp b/lib/IR/AsmPrinter.cpp
index c2cd257..3a3b196 100644
--- a/lib/IR/AsmPrinter.cpp
+++ b/lib/IR/AsmPrinter.cpp
@@ -174,10 +174,10 @@
 
 void ModuleState::visitIfStmt(const IfStmt *ifStmt) {
   recordIntegerSetReference(ifStmt->getCondition());
-  for (auto &childStmt : *ifStmt->getThenClause())
+  for (auto &childStmt : *ifStmt->getThen())
     visitStatement(&childStmt);
-  if (ifStmt->hasElseClause())
-    for (auto &childStmt : *ifStmt->getElseClause())
+  if (ifStmt->hasElse())
+    for (auto &childStmt : *ifStmt->getElse())
       visitStatement(&childStmt);
 }
 
@@ -1270,11 +1270,11 @@
   os.indent(numSpaces) << "if (";
   printIntegerSetReference(stmt->getCondition());
   os << ") {\n";
-  print(stmt->getThenClause());
+  print(stmt->getThen());
   os.indent(numSpaces) << "}";
-  if (stmt->hasElseClause()) {
+  if (stmt->hasElse()) {
     os << " else {\n";
-    print(stmt->getElseClause());
+    print(stmt->getElse());
     os.indent(numSpaces) << "}";
   }
 }
@@ -1393,14 +1393,14 @@
 
 void Statement::dump() const { print(llvm::errs()); }
 
-void StmtBlock::print(raw_ostream &os) const {
+void StmtBlock::printBlock(raw_ostream &os) const {
   MLFunction *function = findFunction();
   ModuleState state(function->getContext());
   ModulePrinter modulePrinter(os, state);
   MLFunctionPrinter(function, modulePrinter).print(this);
 }
 
-void StmtBlock::dump() const { print(llvm::errs()); }
+void StmtBlock::dumpBlock() const { printBlock(llvm::errs()); }
 
 void Function::print(raw_ostream &os) const {
   ModuleState state(getContext());