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 "../../debug.h"

     9

    10

#define SQUIRREL_CLASS

    11

#include "convert.hpp"

    12

#include "squirrel_class.hpp"

    13

#include "../core/ai_base.hpp"

    14

#include "../core/ai_company.hpp"

    15

#include "../core/ai_map.hpp"

    16

#include "../core/ai_town.hpp"

    17

    18

SQInteger Squirrel::SQDestructor(SQUserPointer p, SQInteger size)

    19

{

    20

	return 1;

    21

}

    22

    23

SQInteger Squirrel::SQConstructor(HSQUIRRELVM vm)

    24

{

    25

	DEBUG(ai, 0, "[squirrel] Please do NOT define your AI class yourself; use RegisterAI() instead");

    26

	return SQ_ERROR;

    27

}

    28

    29

SQInteger Squirrel::SQGetTick(HSQUIRRELVM vm)

    30

{

    31

	SQUserPointer self;

    32

    33

	if (!SquirrelEngine::GetInstance(vm, &self)) {

    34

		DEBUG(ai, 0, "[squirrel] Failed to find instance pointer");

    35

		return SQ_ERROR;

    36

	}

    37

    38

	/* Return the value */

    39

	sq_pushinteger(vm, ((Squirrel *)self)->GetTick());

    40

	return 1;

    41

}

    42

    43

SQInteger Squirrel::SQRegisterAI(HSQUIRRELVM vm)

    44

{

    45

	const SQChar *classname = NULL;

    46

	sq_getstring(vm, -1, &classname);

    47

    48

	/* Check if the class really exists */

    49

	int oldtop = sq_gettop(vm);

    50

	sq_pushroottable(vm);

    51

	sq_pushstring(vm, classname, -1);

    52

	if (SQ_FAILED(sq_get(vm, -2))) {

    53

		DEBUG(ai, 0, "[squirrel] Failed to find class '%s'", classname);

    54

		return SQ_ERROR;

    55

	}

    56

	sq_settop(vm, oldtop);

    57

    58

	/* Retreive the 'this' pointer from the global space */

    59

	Squirrel *self = (Squirrel *)sq_getforeignptr(vm);

    60

	if (self == NULL) {

    61

		DEBUG(ai, 0, "[squirrel] Currently only one RegisterAI call is allowed!");

    62

		return SQ_ERROR;

    63

	}

    64

	/* Make sure we only register one AI in this script */

    65

	sq_setforeignptr(vm, NULL);

    66

    67

	/* Create the new class */

    68

	if (!self->engine->CreateClassInstance(classname, self, &self->SQ_instance)) {

    69

		return SQ_ERROR;

    70

	}

    71

    72

	return 0;

    73

}

    74

    75

void Squirrel::RegisterClasses()

    76

{

    77

	HSQUIRRELVM vm = this->engine->GetVM();

    78

	SQInteger top = sq_gettop(vm);

    79

    80

	/* Store 'this' so we can find it back later on */

    81

	sq_setforeignptr(vm, this);

    82

    83

	this->engine->AddClassBegin("AIController");

    84

    85

	/* Create a constructor, so we can assign the right 'this' */

    86

	this->engine->AddMethod("constructor", &Squirrel::SQConstructor, 1, "x");

    87

	/* Assign other functions as AIController won't be done via the normal way */

    88

	this->engine->AddMethod("GetTick", &Squirrel::SQGetTick, 1, "x");

    89

    90

	this->engine->AddClassEnd();

    91

    92

	/* Register the 'RegisterAI()' function */

    93

	sq_pushroottable(vm);

    94

	this->engine->AddMethod("RegisterAI", &Squirrel::SQRegisterAI, 2, "ts");

    95

   100

   101

	/* Reset top */

   102

	sq_settop(vm, top);

   103

}

   104

   105

Squirrel::Squirrel(const char *script)

   106

{

   107

	this->engine = new SquirrelEngine();

   108

	this->RegisterClasses();

   109

	this->engine->LoadScript(script);

   110

}

   111

   112

Squirrel::~Squirrel()

   113

{

   114

	delete this->engine;

   115

}

   116

   117

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

   118

{

   119

	this->engine->CallMethod(this->SQ_instance, "GameLoop");

   120

}