mongoose.cpp

extern "C" {
#include <TH.h>
#include <luaT.h>
}

#include "THpp.hpp"
using namespace std;
using namespace TH;

#include "mongoose.h"

static int InitMongoose(lua_State* L) {
  setLuaState(L);
  string tty = FromLuaStack<string>(1);
  Mongoose* mongoose = new Mongoose(tty);
  if (mongoose->file == NULL)
    lua_pushnil(L);
  else
    lua_pushlightuserdata(L, (void*)mongoose);
  return 1;
}

static int ReleaseMongoose(lua_State* L) {
  setLuaState(L);
  Mongoose* mongoose = (Mongoose*)lua_touserdata(L, 1);
  delete mongoose;
  return 0;
}

//============================================================
// Register functions in LUA
//


#define torch_(NAME)        TH_CONCAT_3(torch_, Real, NAME)
#define torch_Tensor        TH_CONCAT_STRING_3(torch.,Real,Tensor)
#define libmongoose_(NAME)  TH_CONCAT_3(libmongoose_, Real, NAME)

static const luaL_reg libmongoose_init [] = {
  {"initMongoose", InitMongoose},
  {"releaseMongoose", ReleaseMongoose},
  {NULL, NULL}
};

#include "generic/mongoose.cpp"
#include "THGenerateFloatTypes.h"

LUA_EXTERNC DLL_EXPORT int luaopen_libmongoose(lua_State *L) {
  luaL_register(L, "libmongoose", libmongoose_init);
  
  libmongoose_FloatMain_init(L);
  libmongoose_DoubleMain_init(L);
  
  return 1;
}