5

#ifndef SQUIRREL_CLASS_HPP

     6

#define SQUIRREL_CLASS_HPP

     7

     8

/**

     9

 * The template to define classes in Squirrel. It takes care of the creation

    10

 *  and calling of such classes, to make the AI Layer cleaner while having a

    11

 *  powerful script as possible AI language.

    12

 */

    13

template <class CL>

    14

class DefSQClass {

    15

private:

    16

	const char *classname;

    17

    18

public:

    19

	DefSQClass(const char *_classname) :

    20

		classname(_classname)

    21

	{}

    22

    23

	/**

    24

	 * The destructor, to delete the real instance.

    25

	 */

    26

	static SQInteger Destructor(SQUserPointer p, SQInteger size)

    27

	{

    28

		/* Remove the real instance too */

    29

		if (p != NULL) delete (CL *)p;

    30

		return 0;

    31

	}

    32

    33

	/**

    34

	 * The constructor, to create the real instance.

    35

	 */

    36

	static SQInteger Constructor(HSQUIRRELVM vm)

    37

	{

    38

		/* Create the real instance */

    39

		CL *instance = new CL();

    40

		sq_setinstanceup(vm, -1, instance);

    41

		sq_setreleasehook(vm, -1, &DefSQClass::Destructor);

    42

		return 0;

    43

	}

    44

    45

	/**

    46

	 * This defines a method inside a class for Squirrel.

    47

	 */

    48

	template <typename Func>

    49

	void DefSQFunction(HSQUIRRELVM vm, Func function_proc, const char *function_name)

    50

	{

    51

//		int nparam = 1;

    52

//		const char *params = "x";

    53

		void *ptr;

    54

    55

		sq_pushstring(vm, function_name, -1);

    56

		/* Store the real function-pointer in the user-data */

    57

		ptr = sq_newuserdata(vm, sizeof(function_proc));

    58

		memcpy(ptr, &function_proc, sizeof(function_proc));

    59

		sq_newclosure(vm, SQConvert::DefSQCallback<CL, Func>, 1);

    60

//		sq_setparamscheck(vm, nparam, params);

    61

		sq_createslot(vm, -3);

    62

	}

    63

    64

	void PreRegister(HSQUIRRELVM vm)

    65

	{

    66

		/* Prepare the new class */

    67

		sq_pushroottable(vm);

    68

		sq_pushstring(vm, this->classname, -1);

    69

		sq_newclass(vm, SQFalse);

    70

    71

		/* Register a constructor */

    72

		sq_pushstring(vm, "constructor", -1);

    73

		sq_newclosure(vm, &DefSQClass::Constructor, 0);

    74

//		sq_setparamscheck(vm, nparam, params);

    75

		sq_createslot(vm, -3);

    76

	}

    77

    78

	void PostRegister(HSQUIRRELVM vm)

    79

	{

    80

		/* Create the class and all his methods */

    81

		sq_createslot(vm, -3);

    82

	}

    83

};

    84

    85

#endif /* SQUIRREL_CLASS_HPP */