1

/* $Id$ */

     2

     3

/** @file squirrel.cpp allows loading squirrel scripts to control an AI */

     4

     5

#include "squirrel.hpp"

     6

#include <sqstdio.h>

     7

#include <sqstdaux.h>

     8

#include <stdarg.h>

     9

#include "../../debug.h"

    10

    11

/**

    12

 * Here we handle errors that come from squirrel.

    13

 */

    14

static void Squirrel_CompileError(HSQUIRRELVM vm, const SQChar *desc, const SQChar *source, SQInteger line, SQInteger column)

    15

{

    16

#ifdef _SQ64

    17

	printf("Error %s:%ld/%ld: %s\n", source, line, column, desc);

    18

#else

    19

	printf("Error %s:%d/%d: %s\n", source, line, column, desc);

    20

#endif

    21

}

    22

    23

/**

    24

 * Here we handle run-time errors that come from squirrel.

    25

 */

    26

static SQInteger Squirrel_RunError(HSQUIRRELVM vm)

    27

{

    28

	const SQChar *sErr = 0;

    29

    30

	if (sq_gettop(vm) >= 1) {

    31

		if (SQ_SUCCEEDED(sq_getstring(vm, -1, &sErr))) {

    32

			printf("%s\n", sErr);

    33

		} else {

    34

			printf("Unknown error\n");

    35

		}

    36

	}

    37

    38

	return 0;

    39

}

    40

    41

/**

    42

 * Here we handle 'print' command within squirrel script.

    43

 */

    44

static void Squirrel_PrintFunc(HSQUIRRELVM vm, const SQChar *s, ...)

    45

{

    46

	va_list arglist;

    47

	va_start(arglist, s);

    48

	vprintf(s, arglist);

    49

	va_end(arglist);

    50

	printf("\n");

    51

}

    52

    53

Squirrel::Squirrel(const char *script_dir)

    54

{

    55

	char filename[1024];

    56

    57

	this->vm = sq_open(1024);

    58

    59

	/* Handle compile-errors ourself, so we can display it nicely */

    60

	sq_setcompilererrorhandler(this->vm, Squirrel_CompileError);

    61

	sq_notifyallexceptions(this->vm, SQTrue);

    62

	/* Set a good print-function */

    63

	sq_setprintfunc(this->vm, Squirrel_PrintFunc);

    65

	sq_newclosure(this->vm, &Squirrel_RunError, 0);

    66

	sq_seterrorhandler(this->vm);

    67

    68

	/* TODO -- This is ugly! */

    69

	strcpy(filename, "ai/");

    70

	strcat(filename, script_dir);

    71

	strcat(filename, PATHSEP);

    72

	strcat(filename, "main.sq");

    73

    74

	DEBUG(ai, 1, "[squirrel] Starting AI with script in '%s'", filename);

    75

    76

	/* Load and run the script */

    77

	sq_pushroottable(this->vm);

    78

	if (!SQ_SUCCEEDED(sqstd_dofile(this->vm, filename, SQFalse, SQTrue))) {

    79

		DEBUG(ai, 0, "[squirrel] Failed to compile '%s'", filename);

    80

		return;

    81

	}

    82

}

    83

    84

Squirrel::~Squirrel()

    85

{

    86

	/* Clean up the stuff */

    87

	sq_pop(this->vm, 1);

    88

	sq_close(this->vm);

    89

}

    90

    91

/* virtual */ void Squirrel::GameLoop()

    92

{

    93

	/* Store the current top */

    94

	int top = sq_gettop(this->vm);

    95

	/* Go back to root-table */

    96

	sq_pushroottable(this->vm);

    97

	/* Find the function-name inside the script */

    98

	sq_pushstring(this->vm,_SC("GameLoop"), -1);

    99

	if (!SQ_SUCCEEDED(sq_get(this->vm, -2))) {

   100

		DEBUG(ai, 0, "[squirrel] Failed to run GameLoop");

   101

		sq_settop(this->vm, top);

   102

		return;

   103

	}

   104

	/* Call the function */

   105

	sq_pushroottable(this->vm);

   106

	sq_call(this->vm, 1, 0, 0);

   107

	/* Reset the top */

   108

	sq_settop(this->vm, top);

   109

}