src/scripting/wrapper_util.cpp

   1 #include <config.h>
   2
   3 #include <stdexcept>
   4 #include <sstream>
   5 #include "wrapper_util.h"
   6
   7 static void register_function(HSQUIRRELVM v, SQFUNCTION func, const char* name)
   8 {
   9     sq_pushstring(v, name, -1);
  10     sq_newclosure(v, func, 0); //create a new function
  11     sq_createslot(v, -3);
  12 }
  13
  14 static void register_class(HSQUIRRELVM v, WrappedClass* wclass)
  15 {
  16     sq_pushstring(v, wclass->name, -1);
  17     sq_newclass(v, false);
  18     for(WrappedFunction* func = wclass->functions; func->name != 0; ++func) {
  19         register_function(v, func->f, func->name);
  20     }
  21     sq_createslot(v, -3);
  22 }
  23
  24 void register_functions(HSQUIRRELVM v, WrappedFunction* functions)
  25 {
  26     sq_pushroottable(v);
  27     for(WrappedFunction* func = functions; func->name != 0; ++func) {
  28         register_function(v, func->f, func->name);
  29     }
  30     sq_pop(v, 1);
  31 }
  32
  33 void register_classes(HSQUIRRELVM v, WrappedClass* classes)
  34 {
  35     sq_pushroottable(v);
  36     for(WrappedClass* wclass = classes; wclass->name != 0; ++wclass) {
  37         register_class(v, wclass);
  38     }
  39     sq_pop(v, 1);
  40 }
  41
  42 static void print_stack(HSQUIRRELVM v)
  43 {
  44     printf("--------------------------------------------------------------\n");
  45     int count = sq_gettop(v);
  46     for(int i = 1; i <= count; ++i) {
  47         printf("%d: ",i);
  48         switch(sq_gettype(v, i))
  49         {
  50             case OT_NULL:
  51                 printf("null");
  52                 break;
  53             case OT_INTEGER: {
  54                 int val;
  55                 sq_getinteger(v, i, &val);
  56                 printf("integer (%d)", val);
  57                 break;
  58             }
  59             case OT_FLOAT: {
  60                 float val;
  61                 sq_getfloat(v, i, &val);
  62                 printf("float (%f)", val);
  63                 break;
  64             }
  65             case OT_STRING: {
  66                 const char* val;
  67                 sq_getstring(v, i, &val);
  68                 printf("string (%s)", val);
  69                 break;
  70             }
  71             case OT_TABLE:
  72                 printf("table");
  73                 break;
  74             case OT_ARRAY:
  75                 printf("array");
  76                 break;
  77             case OT_USERDATA:
  78                 printf("userdata");
  79                 break;
  80             case OT_CLOSURE:
  81                 printf("closure(function)");
  82                 break;
  83             case OT_NATIVECLOSURE:
  84                 printf("native closure(C function)");
  85                 break;
  86             case OT_GENERATOR:
  87                 printf("generator");
  88                 break;
  89             case OT_USERPOINTER:
  90                 printf("userpointer");
  91                 break;
  92             case OT_CLASS:
  93                 printf("class");
  94                 break;
  95             case OT_INSTANCE:
  96                 printf("instance");
  97                 break;
  98             default:
  99                 printf("unknown?!?");
 100                 break;
 101         }
 102         printf("\n");
 103     }
 104     printf("--------------------------------------------------------------\n");
 105 }
 106
 107 #define check(x)                                    \
 108     if((x) < 0) {                                   \
 109         std::stringstream msg;                      \
 110         sq_getlasterror(v);                         \
 111         const char* error;                          \
 112         sq_getstring(v, -1, &error);                \
 113         msg << "Error: " << error;                  \
 114         throw std::runtime_error(msg.str());        \
 115     }
 116
 117 void expose_object(HSQUIRRELVM v, void* object, const char* type,
 118         const char* name)
 119 {
 120     // part1 of registration of the instance in the root table
 121     sq_pushroottable(v);
 122     sq_pushstring(v, name, -1);
 123
 124     // resolve class name
 125     sq_pushroottable(v);
 126     sq_pushstring(v, type, -1);
 127     print_stack(v);
 128     if(sq_get(v, -2) < 0) {
 129         std::ostringstream msg;
 130         msg << "Couldn't resolve squirrel type '" << type << "'.";
 131         throw std::runtime_error(msg.str());
 132     }
 133     sq_remove(v, -2); // remove roottable
 134     print_stack(v);
 135
 136     // create an instance and set pointer to c++ object
 137     print_stack(v);
 138     check(sq_createinstance(v, -1));
 139     printf("after creatinstance\n");
 140     print_stack(v);
 141     check(sq_setinstanceup(v, -1, object));
 142     printf("after setinstanceup\n");
 143     print_stack(v);
 144
 145     sq_remove(v, -2); // remove class
 146
 147     // part2 of registration of the instance in the root table
 148     print_stack(v);
 149     check(sq_createslot(v, -3));
 150     sq_pop(v, 1);
 151 }
 152