botc: comparison parser.cxx

-:9e4f785501db
+:47e686c96d8f
 #include "variables.h"
 #define MUST_TOPLEVEL if (g_CurMode != MODE_TOPLEVEL) \
 	ParserError ("%s-statements may only be defined at top level!", token.chars());
-#define MUST_NOT_TOPLEVEL if (g_CurMode != MODE_TOPLEVEL) \
+#define MUST_NOT_TOPLEVEL if (g_CurMode == MODE_TOPLEVEL) \
 	ParserError ("%s-statements may not be defined at top level!", token.chars());
 int g_NumStates = 0;
 int g_NumEvents = 0;
 int g_CurMode = MODE_TOPLEVEL;
 			continue;
 		}
 		// Label
 		if (!PeekNext().compare (":")) {
+			MUST_NOT_TOPLEVEL
 			if (IsKeyword (token))
 				ParserError ("label name `%s` conflicts with keyword\n", token.chars());
 			if (FindCommand (token))
 				ParserError ("label name `%s` conflicts with command name\n", token.chars());
 			if (FindGlobalVariable (token))
 			continue;
 		}
 		// Goto
 		if (!token.icompare ("goto")) {
+			MUST_NOT_TOPLEVEL
 			// Get the name of the label
 			MustNext ();
 			// Find the mark this goto statement points to
 			unsigned int m = w->FindMark (MARKTYPE_LABEL, token);
 			continue;
 		}
 		// If
 		if (!token.icompare ("if")) {
+			MUST_NOT_TOPLEVEL
+			PushBlockStack ();
 			// Condition
 			MustNext ("(");
 			// Read the expression and write it. Store it to memory too for else statements.
 			MustNext ();
 			w->WriteBuffer (c);
 			MustNext (")");
 			MustNext ("{");
+			// Add a mark - to here temporarily - and add a reference to it.
+			// Upon a closing brace, the mark will be adjusted.
+			unsigned int marknum = w->AddMark (MARKTYPE_IF, "");
 			// Use DH_IFNOTGOTO - if the expression is not true, we goto the mark
 			// we just defined - and this mark will be at the end of the block.
-			AddBlockMark (w, DH_IFNOTGOTO);
+			w->Write<word> (DH_IFNOTGOTO);
+			w->AddReference (marknum);
+			// Store it in the block stack
+			blockstack[g_BlockStackCursor].mark1 = marknum;
+			blockstack[g_BlockStackCursor].type = BLOCKTYPE_IF;
+			continue;
+		}
+		// While
+		if (!token.compare ("while")) {
+			printf ("begin while loop\n");
+			MUST_NOT_TOPLEVEL
+			PushBlockStack ();
+			// While loops need two marks - one at the start of the loop and one at the
+			// end. The condition is checked at the very start of the loop, if it fails,
+			// we use goto to skip to the end of the loop. At the end, we loop back to
+			// the beginning with a go-to statement.
+			unsigned int mark1 = w->AddMark (MARKTYPE_INTERNAL, ""); // start
+			unsigned int mark2 = w->AddMark (MARKTYPE_INTERNAL, ""); // end
+			// Condition
+			MustNext ("(");
+			MustNext ();
+			DataBuffer* expr = ParseExpression (TYPE_INT);
+			MustNext (")");
+			MustNext ("{");
+			// Write condition
+			w->WriteBuffer (expr);
+			// Instruction to go to the end if it fails
+			w->Write<word> (DH_IFNOTGOTO);
+			w->AddReference (mark2);
+			// Store the needed stuff
+			blockstack[g_BlockStackCursor].mark1 = mark1;
+			blockstack[g_BlockStackCursor].mark2 = mark2;
+			blockstack[g_BlockStackCursor].type = BLOCKTYPE_WHILE;
 			continue;
 		}
 		if (!token.compare ("}")) {
 			// Closing brace
 			// If we're in the block stack, we're descending down from it now
 			if (g_BlockStackCursor > 0) {
-				// Adjust its closing mark so that it's here.
+				BlockInformation* info = &blockstack[g_BlockStackCursor];
-				unsigned int marknum = blockstack[g_BlockStackCursor];
+				switch (info->type) {
-				if (marknum != MAX_MARKS)
+				case BLOCKTYPE_IF:
-					w->MoveMark (marknum);
+					printf ("end if\n");
-				w->AddMark (MARKTYPE_INTERNAL, "");
+					// Adjust the closing mark.
+					w->MoveMark (info->mark1);
+					break;
+				case BLOCKTYPE_WHILE:
+					printf ("end while loop\n");
+					// Write down the instruction to go back to the start of the loop
+					w->Write (DH_GOTO);
+					w->AddReference (info->mark1);
+					// Move the closing mark here since we're at the end of the while loop
+					w->MoveMark (info->mark2);
+					// Offset it by 4 since the parser doesn't like having marks directly afte refs
+					w->OffsetMark (info->mark2, sizeof (word)*2);
+				}
+				// Descend down the stack
 				g_BlockStackCursor--;
 				continue;
 			}
 			int dataheader =	(g_CurMode == MODE_EVENT) ? DH_ENDEVENT :
 	retbuf->Write<word> (var->index);
 	return retbuf;
 }
-void ScriptReader::AddBlockMark (ObjWriter* w, word dataheader) {
+void ScriptReader::PushBlockStack () {
-	// Add a mark - to here temporarily - and add a reference to it.
+	BlockInformation* info = &blockstack[g_BlockStackCursor];
-	// Upon a closing brace, the mark will be adjusted.
+	info->type = 0;
-	unsigned int marknum = w->AddMark (MARKTYPE_IF, "");
+	info->mark1 = 0;
+	info->mark2 = 0;
-	// Write the header and store reference to the mark
-	w->Write<word> (dataheader);
-	w->AddReference (marknum);
-	// Store it in the block stack
 	g_BlockStackCursor++;
-	blockstack[g_BlockStackCursor] = marknum;
+}
-}

Mercurial > botc / file comparison

comparison: parser.cxx

parser.cxx