Loop parsing

src/Parser/Statement/Statement.h

@@ -13,6 +13,7 @@ enum class StatementKind {
     RETURN,
     FUNCTION,
     VARIABLE,
+    LOOP,
     META_EXTERN_FUNCTION,
     INVALID
 };

src/Parser/Statement/StatementLoop.cpp

@@ -0,0 +1,39 @@
+#include "StatementLoop.h"
+
+#include "Parser/Expression/Expression.h"
+#include "Parser/Statement/StatementBlock.h"
+
+StatementLoop::StatementLoop(shared_ptr<Statement> initStatement, shared_ptr<Expression> preConditionExpression, shared_ptr<Expression> postConditionExpression, shared_ptr<StatementBlock> bodyBlockStatement):
+Statement(StatementKind::LOOP), initStatement(initStatement), preConditionExpression(preConditionExpression), postConditionExpression(postConditionExpression), bodyBlockStatement(bodyBlockStatement) { }
+
+shared_ptr<Statement> StatementLoop::getInitStatement() {
+    return initStatement;
+}
+
+shared_ptr<Expression> StatementLoop::getPreConditionExpression() {
+    return preConditionExpression;
+}
+
+shared_ptr<Expression> StatementLoop::getPostConditionExpression() {
+    return postConditionExpression;
+}
+
+shared_ptr<StatementBlock> StatementLoop::getBodyBlockStatement() {
+    return bodyBlockStatement;
+}
+
+string StatementLoop::toString(int indent) {
+    string value;
+    for (int ind=0; ind<indent; ind++)
+        value += "  ";
+    value += "REP(";
+    if (initStatement != nullptr)
+        value += initStatement->toString(0), ", ";
+    if (preConditionExpression != nullptr)
+        value += preConditionExpression->toString(0) + ", ";
+    if (postConditionExpression != nullptr)
+        value += postConditionExpression->toString(0);
+    value += "):\n";
+    value += bodyBlockStatement->toString(indent+1);
+    return value;
+}

src/Parser/Statement/StatementLoop.h

@@ -0,0 +1,20 @@
+#include "Statement.h"
+
+class Expression;
+class StatementBlock;
+
+class StatementLoop: public Statement {
+private:
+    shared_ptr<Statement> initStatement;
+    shared_ptr<Expression> preConditionExpression;
+    shared_ptr<Expression> postConditionExpression;
+    shared_ptr<StatementBlock> bodyBlockStatement;
+
+public:
+    StatementLoop(shared_ptr<Statement> initStatement, shared_ptr<Expression> preConditionExpression, shared_ptr<Expression> postConditionExpression, shared_ptr<StatementBlock> bodyBlockStatement);
+    shared_ptr<Statement> getInitStatement();
+    shared_ptr<Expression> getPreConditionExpression();
+    shared_ptr<Expression> getPostConditionExpression();
+    shared_ptr<StatementBlock> getBodyBlockStatement();
+    string toString(int indent) override;
+};