Add the FormatStmt class.

isanbard · Oct 28, 2011 · 087e96b · 087e96b
1 parent 885135a
commit 087e96b
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 2 deletions.
diff --git a/include/flang/AST/Stmt.h b/include/flang/AST/Stmt.h
@@ -32,10 +32,16 @@ class Stmt {
 public:
   enum StmtTy {
     Program,
+
+    // Specification Part
     Use,
     Import,
-    Implicit,
+
-    Parameter,
+      // Implicit Part
+      Implicit,
+      Parameter,
+      Format,
+
     Asynchronous,
     EndProgram,
     Assignment,
@@ -271,6 +277,22 @@ class ParameterStmt : public Stmt {
   }
 };
 
+/// FormatStmt -
+///
+class FormatStmt : public Stmt {
+  FormatSpec *FS;
+
+  FormatStmt(SMLoc Loc, FormatSpec *fs, ExprResult StmtLabel);
+public:
+  static FormatStmt *Create(ASTContext &C, SMLoc Loc, FormatSpec *fs,
+                            ExprResult StmtLabel);
+
+  static bool classof(const FormatStmt*) { return true; }
+  static bool classof(const Stmt *S) {
+    return S->getStatementID() == Format;
+  }
+};
+
 /// AsynchronousStmt - Specifies the asynchronous attribute for a list of
 /// objects.
 ///

diff --git a/lib/AST/Stmt.cpp b/lib/AST/Stmt.cpp
@@ -138,6 +138,18 @@ ParameterStmt *ParameterStmt::Create(ASTContext &C, SMLoc Loc,
   return new (C) ParameterStmt(Loc, NC, CE, StmtLabel);
 }
 
+//===----------------------------------------------------------------------===//
+// Format Statement
+//===----------------------------------------------------------------------===//
+
+FormatStmt::FormatStmt(SMLoc Loc, FormatSpec *fs, ExprResult StmtLabel)
+  : Stmt(Format, Loc, StmtLabel), FS(fs) {}
+
+FormatStmt *FormatStmt::Create(ASTContext &C, SMLoc Loc, FormatSpec *fs,
+                               ExprResult StmtLabel) {
+  return new (C) FormatStmt(Loc, fs, StmtLabel);
+}
+
 //===----------------------------------------------------------------------===//
 // Asynchronous Statement
 //===----------------------------------------------------------------------===//