1

     2

#include <squirrel.h>

     3

#include "stdafx.h"

     4

#include "debug.h"

     5

#include "squirrel.hpp"

     6

#include "squirrel_std.hpp"

     7

#include "core/alloc_func.hpp"

     8

     9

/* abs() is normally defined to myabs(), which we don't want in this file */

    10

#undef abs

    11

    12

SQInteger SquirrelStd::abs(HSQUIRRELVM vm)

    13

{

    14

	SQInteger tmp;

    15

    16

	sq_getinteger(vm, 2, &tmp);

    17

	sq_pushinteger(vm, ::abs(tmp));

    18

	return 1;

    19

}

    20

    21

SQInteger SquirrelStd::require(HSQUIRRELVM vm)

    22

{

    23

	SQInteger top = sq_gettop(vm);

    24

	const SQChar *filename;

    25

	SQChar *real_filename;

    26

    27

	sq_getstring(vm, 2, &filename);

    28

    29

	/* Get the script-name of the current file, so we can work relative from it */

    30

	SQStackInfos si;

    31

	sq_stackinfos(vm, 1, &si);

    32

	if (si.source == NULL) {

    33

		DEBUG(misc, 0, "[squirrel] Couldn't detect the script-name of the 'require'-caller; this should never happen!");

    34

		return SQ_ERROR;

    35

	}

    36

	real_filename = scstrdup(si.source);

    37

	/* Keep the dir, remove the rest */

    38

	SQChar *s = scstrrchr(real_filename, PATHSEPCHAR);

    39

	if (s != NULL) {

    40

		/* Keep the PATHSEPCHAR there, remove the rest */

    41

		*s++;

    42

		*s = '\0';

    43

	}

    44

	/* And now we concat, so we are relative from the current script

    45

	 * First, we have to make sure we have enough space for the full path */

    46

	real_filename = ReallocT(real_filename, scstrlen(real_filename) + scstrlen(filename) + 1);

    47

	scstrcat(real_filename, filename);

    48

    49

	bool ret = Squirrel::LoadScript(vm, FS2OTTD(real_filename));

    50

    51

	/* Reset the top, so the stack stays correct */

    52

	sq_settop(vm, top);

    53

	free(real_filename);

    54

    55

	return ret ? 0: SQ_ERROR;

    56

}

    57

    58

void squirrel_register_global_std(Squirrel *engine)

    59

{

    60

	/* We don't use squirrel_helper here, as we want to register to the global

    61

	 *  scope and not to a class. */

    62

	engine->AddMethod("require", &SquirrelStd::require, 2, "?s");

    63

}

    64

    65

void squirrel_register_std(Squirrel *engine)

    66

{

    67

	/* We don't use squirrel_helper here, as we want to register to the global

    68

	 *  scope and not to a class. */

    69

	engine->AddMethod("abs", &SquirrelStd::abs, 2, "xi");

    70

}