3 files changed, 24 insertions, 3 deletions
diff --git a/AST/StmtSerialization.cpp b/AST/StmtSerialization.cpp
index 7d82d0547e..c6af9a50c7 100644
--- a/AST/StmtSerialization.cpp
+++ b/AST/StmtSerialization.cpp
@@ -52,6 +52,9 @@ Stmt* Stmt::Materialize(llvm::Deserializer& D) {
     case DefaultStmtClass:
       return DefaultStmt::directMaterialize(D);
       
+    case IfStmtClass:
+      return IfStmt::directMaterialize(D);
+      
     case IntegerLiteralClass:
       return IntegerLiteral::directMaterialize(D);
       
@@ -182,6 +185,21 @@ DefaultStmt* DefaultStmt::directMaterialize(llvm::Deserializer& D) {
   return stmt;
 }
 
+void IfStmt::directEmit(llvm::Serializer& S) const {
+  S.Emit(IfLoc);
+  S.EmitOwnedPtr(getCond());
+  S.EmitOwnedPtr(getThen());
+  S.EmitOwnedPtr(getElse());
+}
+
+IfStmt* IfStmt::directMaterialize(llvm::Deserializer& D) {
+  SourceLocation L = SourceLocation::ReadVal(D);
+  Expr* Cond = D.ReadOwnedPtr<Expr>();
+  Stmt* Then = D.ReadOwnedPtr<Stmt>();
+  Stmt* Else = D.ReadOwnedPtr<Stmt>();
+  return new IfStmt(L,Cond,Then,Else);
+}
+
 void IntegerLiteral::directEmit(llvm::Serializer& S) const {
   S.Emit(Loc);
   S.Emit(getType());
diff --git a/include/clang/AST/Expr.h b/include/clang/AST/Expr.h
index 9b6a8e345b..6a06e5702a 100644
--- a/include/clang/AST/Expr.h
+++ b/include/clang/AST/Expr.h
@@ -340,7 +340,7 @@ public:
   virtual child_iterator child_begin();
   virtual child_iterator child_end();
   
-  void directEmit(llvm::Serializer& S) const;
+  virtual void directEmit(llvm::Serializer& S) const;
   static ParenExpr* directMaterialize(llvm::Deserializer& D);
 };
 
diff --git a/include/clang/AST/Stmt.h b/include/clang/AST/Stmt.h
index 46059c56bf..523eda69a0 100644
--- a/include/clang/AST/Stmt.h
+++ b/include/clang/AST/Stmt.h
@@ -402,6 +402,9 @@ public:
   // Iterators
   virtual child_iterator child_begin();
   virtual child_iterator child_end();
+  
+  virtual void directEmit(llvm::Serializer& S) const;
+  static IfStmt* directMaterialize(llvm::Deserializer& D);
 };
 
 /// SwitchStmt - This represents a 'switch' stmt.
@@ -448,7 +451,7 @@ public:
   virtual child_iterator child_begin();
   virtual child_iterator child_end();
   
-  void directEmit(llvm::Serializer& S) const;
+  virtual void directEmit(llvm::Serializer& S) const;
   static SwitchStmt* directMaterialize(llvm::Deserializer& D);
 };
 
@@ -644,7 +647,7 @@ public:
   virtual child_iterator child_begin();
   virtual child_iterator child_end();
   
-  void directEmit(llvm::Serializer& S) const;
+  virtual void directEmit(llvm::Serializer& S) const;
   static BreakStmt* directMaterialize(llvm::Deserializer& D);
 };