--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/src/main.cc Mon Jan 13 23:44:15 2014 +0200
@@ -0,0 +1,281 @@
+/*
+ Copyright (c) 2012-2014, Santeri Piippo
+ All rights reserved.
+
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions are met:
+
+ * Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+
+ * Neither the name of the <organization> nor the
+ names of its contributors may be used to endorse or promote products
+ derived from this software without specific prior written permission.
+
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
+ DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+*/
+
+#include "main.h"
+#include "scriptreader.h"
+#include "object_writer.h"
+#include "events.h"
+#include "commands.h"
+#include "stringtable.h"
+#include "variables.h"
+#include "data_buffer.h"
+#include "object_writer.h"
+#include "parser.h"
+#include "lexer.h"
+
+// List of keywords
+const string_list g_Keywords =
+{
+ "bool",
+ "break",
+ "case",
+ "continue",
+ "const",
+ "default",
+ "do",
+ "else",
+ "event",
+ "for",
+ "goto",
+ "if",
+ "int",
+ "mainloop",
+ "onenter",
+ "onexit",
+ "state",
+ "switch",
+ "str"
+ "void",
+ "while",
+
+ // These ones aren't implemented yet but I plan to do so, thus they are
+ // reserved. Also serves as a to-do list of sorts for me. >:F
+ "enum", // Would enum actually be useful? I think so.
+ "func", // Would function support need external support from zandronum?
+ "return",
+};
+
+// databuffer global variable
+int g_NextMark = 0;
+
+int main (int argc, char** argv)
+{
+ try
+ {
+ // Intepret command-line parameters:
+ // -l: list commands
+ // I guess there should be a better way to do this.
+ if (argc == 2 && !strcmp (argv[1], "-l"))
+ {
+ command_info* comm;
+ init_commands();
+ printf ("Begin list of commands:\n");
+ printf ("------------------------------------------------------\n");
+
+ for (command_info* comm : get_commands())
+ print ("%1\n", get_command_signature (comm));
+
+ printf ("------------------------------------------------------\n");
+ printf ("End of command list\n");
+ exit (0);
+ }
+
+ // Print header
+ string header;
+ string headerline;
+ header = format ("%1 version %2.%3", APPNAME, VERSION_MAJOR, VERSION_MINOR);
+
+ for (int i = 0; i < (header.len() / 2) - 1; ++i)
+ headerline += "-=";
+
+ headerline += '-';
+ print ("%1\n%2\n", header, headerline);
+
+ if (argc < 2)
+ {
+ fprintf (stderr, "usage: %s <infile> [outfile] # compiles botscript\n", argv[0]);
+ fprintf (stderr, " %s -l # lists commands\n", argv[0]);
+ exit (1);
+ }
+
+ string outfile;
+
+ if (argc < 3)
+ outfile = ObjectFileName (argv[1]);
+ else
+ outfile = argv[2];
+
+ // If we'd end up writing into an existing file,
+ // ask the user if we want to overwrite it
+ if (fexists (outfile))
+ {
+ // Additional warning if the paths are the same
+ string warning;
+#ifdef FILE_CASEINSENSITIVE
+
+ if (+outfile == +string (argv[1]))
+#else
+ if (outfile == argv[1])
+#endif
+ {
+ warning = "\nWARNING: Output file is the same as the input file. ";
+ warning += "Answering yes here will destroy the source!\n";
+ warning += "Continue nevertheless?";
+ }
+
+ printf ("output file `%s` already exists! overwrite?%s (y/n) ", outfile.chars(), warning.chars());
+
+ char ans;
+ fgets (&ans, 1, stdin);
+
+ if (ans != 'y')
+ {
+ printf ("abort\n");
+ exit (1);
+ }
+ }
+
+ // Read definitions
+ printf ("Reading definitions...\n");
+ init_events();
+ init_commands();
+
+ // Prepare reader and writer
+ botscript_parser* r = new botscript_parser;
+ object_writer* w = new object_writer;
+
+ // We're set, begin parsing :)
+ printf ("Parsing script...\n");
+ r->parse_botscript (argv[1], w);
+ printf ("Script parsed successfully.\n");
+
+ // Parse done, print statistics and write to file
+ int globalcount = g_GlobalVariables.size();
+ int stringcount = num_strings_in_table();
+ int NumMarks = w->MainBuffer->count_marks();
+ int NumRefs = w->MainBuffer->count_references();
+ print ("%1 / %2 strings written\n", stringcount, g_max_stringlist_size);
+ print ("%1 / %2 global variables\n", globalcount, g_max_global_vars);
+ print ("%1 / %2 bytecode marks\n", NumMarks, MAX_MARKS); // TODO: nuke
+ print ("%1 / %2 bytecode references\n", NumRefs, MAX_MARKS); // TODO: nuke
+ print ("%1 / %2 events\n", g_NumEvents, g_max_events);
+ print ("%1 state%s1\n", g_NumStates);
+
+ w->write_to_file (outfile);
+
+ // Clear out the junk
+ delete r;
+ delete w;
+
+ // Done!
+ exit (0);
+ }
+ catch (script_error& e)
+ {
+ lexer* lx = lexer::get_main_lexer();
+ string fileinfo;
+
+ if (lx != null && lx->has_valid_token())
+ {
+ lexer::token* tk = lx->get_token();
+ fileinfo = format ("%1:%2:%3: ", tk->file, tk->line, tk->column);
+ }
+
+ fprint (stderr, "%1error: %2\n", fileinfo, e.what());
+ }
+}
+
+// ============================================================================
+// Utility functions
+
+// ============================================================================
+// Does the given file exist?
+bool fexists (string path)
+{
+ if (FILE* test = fopen (path, "r"))
+ {
+ fclose (test);
+ return true;
+ }
+
+ return false;
+}
+
+// ============================================================================
+// Mutates given filename to an object filename
+string ObjectFileName (string s)
+{
+ // Locate the extension and chop it out
+ int extdot = s.last (".");
+
+ if (extdot >= s.len() - 4)
+ s -= (s.len() - extdot);
+
+ s += ".o";
+ return s;
+}
+
+// ============================================================================
+// Is the given argument a reserved keyword?
+bool IsKeyword (string s)
+{
+ for (int u = 0; u < NumKeywords(); u++)
+ if (s.to_uppercase() == g_Keywords[u].to_uppercase())
+ return true;
+
+ return false;
+}
+
+int NumKeywords()
+{
+ return sizeof (g_Keywords) / sizeof (const char*);
+}
+
+// ============================================================================
+type_e GetTypeByName (string t)
+{
+ t = t.to_lowercase();
+ return (t == "int") ? TYPE_INT :
+ (t == "str") ? TYPE_STRING :
+ (t == "void") ? TYPE_VOID :
+ (t == "bool") ? TYPE_BOOL :
+ TYPE_UNKNOWN;
+}
+
+
+// ============================================================================
+// Inverse operation - type name by value
+string GetTypeName (type_e type)
+{
+ switch (type)
+ {
+ case TYPE_INT: return "int"; break;
+
+ case TYPE_STRING: return "str"; break;
+
+ case TYPE_VOID: return "void"; break;
+
+ case TYPE_BOOL: return "bool"; break;
+
+ case TYPE_UNKNOWN: return "???"; break;
+ }
+
+ return "";
+}