Implemented serialization for ForStmt.

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@43817 91177308-0d34-0410-b5e6-96231b3b80d8

2 files changed, 23 insertions, 0 deletions

diff --git a/AST/StmtSerialization.cpp b/AST/StmtSerialization.cpp
index d7b497155a..61e70ecde8 100644
--- a/AST/StmtSerialization.cpp
+++ b/AST/StmtSerialization.cpp
@@ -54,6 +54,9 @@ Stmt* Stmt::Materialize(llvm::Deserializer& D) {
     
     case DoStmtClass:
       return DoStmt::directMaterialize(D);
+
+    case ForStmtClass:
+      return ForStmt::directMaterialize(D);
       
     case IfStmtClass:
       return IfStmt::directMaterialize(D);
@@ -204,6 +207,23 @@ DoStmt* DoStmt::directMaterialize(llvm::Deserializer& D) {
   return new DoStmt(Body,Cond,DoLoc);
 }
 
+void ForStmt::directEmit(llvm::Serializer& S) const {
+  S.Emit(ForLoc);
+  S.EmitOwnedPtr(getInit());
+  S.EmitOwnedPtr(getCond());
+  S.EmitOwnedPtr(getInc());
+  S.EmitOwnedPtr(getBody());
+}
+
+ForStmt* ForStmt::directMaterialize(llvm::Deserializer& D) {
+  SourceLocation ForLoc = SourceLocation::ReadVal(D);
+  Stmt* Init = D.ReadOwnedPtr<Stmt>();
+  Expr* Cond = D.ReadOwnedPtr<Expr>();
+  Expr* Inc = D.ReadOwnedPtr<Expr>();
+  Stmt* Body = D.ReadOwnedPtr<Stmt>();
+  return new ForStmt(Init,Cond,Inc,Body,ForLoc);
+}
+
 void IfStmt::directEmit(llvm::Serializer& S) const {
   S.Emit(IfLoc);
   S.EmitOwnedPtr(getCond());
diff --git a/include/clang/AST/Stmt.h b/include/clang/AST/Stmt.h
index 284501269d..ed44687da0 100644
--- a/include/clang/AST/Stmt.h
+++ b/include/clang/AST/Stmt.h
@@ -565,6 +565,9 @@ public:
   // Iterators
   virtual child_iterator child_begin();
   virtual child_iterator child_end();
+  
+  virtual void directEmit(llvm::Serializer& S) const;
+  static ForStmt* directMaterialize(llvm::Deserializer& D);
 };
 
 /// GotoStmt - This represents a direct goto.