squirrel_std.cpp

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <https://www.gnu.org/licenses/old-licenses/gpl-2.0>.
 */
 
#include "../stdafx.h"
#include <squirrel.h>
#include <sqstdmath.h>
#include "../debug.h"
#include "squirrel_std.hpp"
#include "../string_func.h"
 
#include "../safeguards.h"
 
 
SQInteger SquirrelStd::min(HSQUIRRELVM vm)
{
  SQInteger tmp1, tmp2;
 
  sq_getinteger(vm, 2, &tmp1);
  sq_getinteger(vm, 3, &tmp2);
  sq_pushinteger(vm, std::min(tmp1, tmp2));
  return 1;
}
 
SQInteger SquirrelStd::max(HSQUIRRELVM vm)
{
  SQInteger tmp1, tmp2;
 
  sq_getinteger(vm, 2, &tmp1);
  sq_getinteger(vm, 3, &tmp2);
  sq_pushinteger(vm, std::max(tmp1, tmp2));
  return 1;
}
 
SQInteger SquirrelStd::require(HSQUIRRELVM vm)
{
  SQInteger top = sq_gettop(vm);
  std::string_view filename;
 
  sq_getstring(vm, 2, filename);
 
  /* Get the script-name of the current file, so we can work relative from it */
  SQStackInfos si;
  sq_stackinfos(vm, 1, &si);
 
  /* Keep the dir, remove the rest */
  std::string path{si.source};
  auto p = path.find_last_of(PATHSEPCHAR);
  /* Keep the PATHSEPCHAR there, remove the rest */
  if (p != std::string::npos) path.erase(p + 1);
  path += filename;
#if (PATHSEPCHAR != '/')
  std::transform(path.begin(), path.end(), path.begin(), [](char &c) { return c == '/' ? PATHSEPCHAR : c; });
#endif
 
  Squirrel *engine = (Squirrel *)sq_getforeignptr(vm);
  bool ret = engine->LoadScript(vm, path);
 
  /* Reset the top, so the stack stays correct */
  sq_settop(vm, top);
 
  return ret ? 0 : SQ_ERROR;
}
 
SQInteger SquirrelStd::notifyallexceptions(HSQUIRRELVM vm)
{
  SQBool b;
 
  if (sq_gettop(vm) >= 1) {
    if (SQ_SUCCEEDED(sq_getbool(vm, -1, &b))) {
      sq_notifyallexceptions(vm, b);
      return 0;
    }
  }
 
  return SQ_ERROR;
}
 
void squirrel_register_global_std(Squirrel &engine)
{
  /* We don't use squirrel_helper here, as we want to register to the global
   *  scope and not to a class. */
  engine.AddMethod("require",             &SquirrelStd::require,             ".s");
  engine.AddMethod("notifyallexceptions", &SquirrelStd::notifyallexceptions, ".b");
}
 
void squirrel_register_std(Squirrel &engine)
{
  /* We don't use squirrel_helper here, as we want to register to the global
   *  scope and not to a class. */
  engine.AddMethod("min", &SquirrelStd::min, ".ii");
  engine.AddMethod("max", &SquirrelStd::max, ".ii");
 
  sqstd_register_mathlib(engine.GetVM());
}