Change for to use statements instead of expressions for init and step

grammar/grammar.y

@@ -80,9 +80,9 @@ statement(A) ::= T_IF T_LPAREN expr(E) T_RPAREN statement(S).
 													{ A = new IfStatement(E, S); }
 statement(A) ::= T_RIF T_LPAREN expr(P) T_RPAREN statement(S).
 													{ A = new RandomIfStatement(P, S); }
-statement(A) ::= T_FOR T_LPAREN expr(INIT) T_SEMICOLON expr(COND) T_SEMICOLON expr(STEP) T_RPAREN statement(S).
+statement(A) ::= T_FOR T_LPAREN statement(INIT) T_SEMICOLON expr(COND) T_SEMICOLON statement(STEP) T_RPAREN statement(S).
 													{ A = new ForStatement(INIT, COND, STEP, S); }
-statement(A) ::= T_RFOR T_LPAREN expr(INIT) T_SEMICOLON expr(P) T_SEMICOLON expr(STEP) T_RPAREN statement(S).
+statement(A) ::= T_RFOR T_LPAREN statement(INIT) T_SEMICOLON expr(P) T_SEMICOLON statement(STEP) T_RPAREN statement(S).
 													{ A = new RandomForStatement(INIT, P, STEP, S); }
 statement(A) ::= T_RETURN expr(E) T_SEMICOLON.		{ A = new ReturnStatement(E); }
 statement(A) ::= T_BEGIN statements(S) T_END.		{ A = new Scope(S); }

inc/AST/ForStatement.h

@@ -6,19 +6,19 @@
 
 class ForStatement : public Statement {
 public:
-	ForStatement(Expression* init, Expression* cond, Expression* step, Statement* stmt);
+	ForStatement(Statement* init, Expression* cond, Statement* step, Statement* stmt);
 	virtual ~ForStatement();
 
 	virtual void accept(ASTVisitor* visitor);
 
-	Expression* getInit();
+	Statement *getInit();
 	Expression* getCond();
-	Expression* getStep();
+	Statement* getStep();
 	Statement* getStmt();
 private:
-	Expression* init_;
+	Statement* init_;
 	Expression* cond_;
-	Expression* step_;
+	Statement* step_;
 	Statement* stmt_;
 };
 

inc/AST/RandomForStatement.h

@@ -7,19 +7,19 @@
 class RandomForStatement : public Statement
 {
 public:
-	RandomForStatement(Expression* init, Expression* prob, Expression* step, Statement* stmt);
+	RandomForStatement(Statement *init, Expression* prob, Statement *step, Statement* stmt);
 	virtual ~RandomForStatement();
 
 	virtual void accept(ASTVisitor *visitor);
 
-	Expression* getInit();
+	Statement* getInit();
 	Expression* getProb();
-	Expression* getStep();
+	Statement *getStep();
 	Statement* getStmt();
 private:
-	Expression* init_;
+	Statement* init_;
 	Expression* prob_;
-	Expression* step_;
+	Statement* step_;
 	Statement* stmt_;
 };
 

src/AST/ForStatement.cpp

@@ -1,7 +1,7 @@
 #include "AST/ForStatement.h"
 #include "AST/ASTVisitor.h"
 
-ForStatement::ForStatement(Expression *init, Expression *cond, Expression *step, Statement *stmt) : init_(init), cond_(cond), step_(step), stmt_(stmt) {
+ForStatement::ForStatement(Statement *init, Expression *cond, Statement *step, Statement *stmt) : init_(init), cond_(cond), step_(step), stmt_(stmt) {
 	//
 }
 
@@ -13,7 +13,7 @@ void ForStatement::accept(ASTVisitor* visitor) {
 	visitor->visit(this);
 }
 
-Expression* ForStatement::getInit() {
+Statement* ForStatement::getInit() {
 	return init_;
 }
 
@@ -21,7 +21,7 @@ Expression* ForStatement::getCond() {
 	return cond_;
 }
 
-Expression* ForStatement::getStep() {
+Statement* ForStatement::getStep() {
 	return step_;
 }
 

src/AST/RandomForStatement.cpp

@@ -1,7 +1,7 @@
 #include "AST/RandomForStatement.h"
 #include "AST/ASTVisitor.h"
 
-RandomForStatement::RandomForStatement(Expression *init, Expression *prob, Expression *step, Statement *stmt) : init_(init),prob_(prob),step_(step),stmt_(stmt)
+RandomForStatement::RandomForStatement(Statement *init, Expression *prob, Statement *step, Statement *stmt) : init_(init),prob_(prob),step_(step),stmt_(stmt)
 {
 	//
 }
@@ -14,7 +14,7 @@ void RandomForStatement::accept(ASTVisitor *visitor) {
 	visitor->visit(this);
 }
 
-Expression* RandomForStatement::getInit() {
+Statement *RandomForStatement::getInit() {
 	return init_;
 }
 
@@ -22,7 +22,7 @@ Expression* RandomForStatement::getProb() {
 	return prob_;
 }
 
-Expression* RandomForStatement::getStep() {
+Statement* RandomForStatement::getStep() {
 	return step_;
 }
 

src/test.cpp

@@ -127,7 +127,7 @@ int main() {
 			lexertl::lookup (state_machine, results);
 			std::string s(results.start, results.end);
 			if (results.id != T_WHITESPACE) {
-				std::cout << "Id: " << results.id << ", Token: " << s << std::endl;
+				//std::cout << "Id: " << results.id << ", Token: " << s << std::endl;
 				scullyParser(parser, results.id, new Token(s), cv);
 			}
 		} while (results.id != 0);