src/Scripting/nasal-props.cxx

   1
   2 #ifdef HAVE_CONFIG_H
   3 #  include "config.h"
   4 #endif
   5
   6 #include <simgear/nasal/nasal.h>
   7 #include <simgear/props/props.hxx>
   8
   9 #include <Main/globals.hxx>
  10
  11 #include "NasalSys.hxx"
  12
  13 // Implementation of a Nasal wrapper for the SGPropertyNode class,
  14 // using the Nasal "ghost" (er... Garbage collection Handle for
  15 // OutSide Thingy) facility.
  16 //
  17 // Note that these functions appear in Nasal with prepended
  18 // underscores.  They work on the low-level "ghost" objects and aren't
  19 // intended for use from user code, but from Nasal code you will find
  20 // in props.nas.  That is where the Nasal props.Node class is defined,
  21 // which provides a saner interface along the lines of SGPropertyNode.
  22
  23 static void propNodeGhostDestroy(void* ghost)
  24 {
  25     SGPropertyNode_ptr* prop = (SGPropertyNode_ptr*)ghost;
  26     delete prop;
  27 }
  28
  29 naGhostType PropNodeGhostType = { propNodeGhostDestroy };
  30
  31 static naRef propNodeGhostCreate(naContext c, SGPropertyNode* n)
  32 {
  33     if(!n) return naNil();
  34     SGPropertyNode_ptr* ghost = new SGPropertyNode_ptr(n);
  35     return naNewGhost(c, &PropNodeGhostType, ghost);
  36 }
  37
  38 naRef FGNasalSys::propNodeGhost(SGPropertyNode* handle)
  39 {
  40     return propNodeGhostCreate(_context, handle);
  41 }
  42
  43 #define NASTR(s) s ? naStr_fromdata(naNewString(c),(char*)(s),strlen(s)) : naNil()
  44
  45 //
  46 // Standard header for the extension functions.  It turns the "ghost"
  47 // found in arg[0] into a SGPropertyNode_ptr*, and then "unwraps" the
  48 // vector found in the second argument into a normal-looking args
  49 // array.  This allows the Nasal handlers to do things like:
  50 //   Node.getChild = func { _getChild(me.ghost, arg) }
  51 //
  52 #define NODEARG()                                                       \
  53     if(argc < 2 || !naIsGhost(args[0]) ||                               \
  54        naGhost_type(args[0]) != &PropNodeGhostType)                       \
  55         naRuntimeError(c, "bad argument to props function");            \
  56     SGPropertyNode_ptr* node = (SGPropertyNode_ptr*)naGhost_ptr(args[0]); \
  57     naRef argv = args[1]
  58
  59 static naRef f_getType(naContext c, naRef me, int argc, naRef* args)
  60 {
  61     NODEARG();
  62     char* t = "unknown";
  63     switch((*node)->getType()) {
  64     case SGPropertyNode::NONE:   t = "NONE";   break;
  65     case SGPropertyNode::ALIAS:  t = "ALIAS";  break;
  66     case SGPropertyNode::BOOL:   t = "BOOL";   break;
  67     case SGPropertyNode::INT:    t = "INT";    break;
  68     case SGPropertyNode::LONG:   t = "LONG";   break;
  69     case SGPropertyNode::FLOAT:  t = "FLOAT";  break;
  70     case SGPropertyNode::DOUBLE: t = "DOUBLE"; break;
  71     case SGPropertyNode::STRING: t = "STRING"; break;
  72     case SGPropertyNode::UNSPECIFIED: t = "UNSPECIFIED"; break;
  73     }
  74     return NASTR(t);
  75 }
  76
  77 static naRef f_getName(naContext c, naRef me, int argc, naRef* args)
  78 {
  79     NODEARG();
  80     return NASTR((*node)->getName());
  81 }
  82
  83 static naRef f_getIndex(naContext c, naRef me, int argc, naRef* args)
  84 {
  85     NODEARG();
  86     return naNum((*node)->getIndex());
  87 }
  88
  89 static naRef f_getValue(naContext c, naRef me, int argc, naRef* args)
  90 {
  91     NODEARG();
  92     switch((*node)->getType()) {
  93     case SGPropertyNode::BOOL:   case SGPropertyNode::INT:
  94     case SGPropertyNode::LONG:   case SGPropertyNode::FLOAT:
  95     case SGPropertyNode::DOUBLE:
  96         return naNum((*node)->getDoubleValue());
  97     case SGPropertyNode::STRING:
  98     case SGPropertyNode::UNSPECIFIED:
  99         return NASTR((*node)->getStringValue());
 100     default:
 101         return naNil();
 102     }
 103 }
 104
 105 static naRef f_setValue(naContext c, naRef me, int argc, naRef* args)
 106 {
 107     NODEARG();
 108     naRef val = naVec_get(argv, 0);
 109     if(naIsString(val)) (*node)->setStringValue(naStr_data(val));
 110     else                (*node)->setDoubleValue(naNumValue(val).num);
 111     return naNil();
 112 }
 113
 114 static naRef f_setIntValue(naContext c, naRef me, int argc, naRef* args)
 115 {
 116     NODEARG();
 117     // Original code:
 118     //   int iv = (int)naNumValue(naVec_get(argv, 0)).num;
 119
 120     // Junk to pacify the gcc-2.95.3 optimizer:
 121     naRef tmp0 = naVec_get(argv, 0);
 122     naRef tmp1 = naNumValue(tmp0);
 123     double tmp2 = tmp1.num;
 124     int iv = (int)tmp2;
 125
 126     (*node)->setIntValue(iv);
 127     return naNil();
 128 }
 129
 130 static naRef f_setBoolValue(naContext c, naRef me, int argc, naRef* args)
 131 {
 132     NODEARG();
 133     naRef val = naVec_get(argv, 0);
 134     (*node)->setBoolValue(naTrue(val) ? true : false);
 135     return naNil();
 136 }
 137
 138 static naRef f_setDoubleValue(naContext c, naRef me, int argc, naRef* args)
 139 {
 140     NODEARG();
 141     (*node)->setDoubleValue(naNumValue(naVec_get(argv, 0)).num);
 142     return naNil();
 143 }
 144
 145 static naRef f_getParent(naContext c, naRef me, int argc, naRef* args)
 146 {
 147     NODEARG();
 148     SGPropertyNode* n = (*node)->getParent();
 149     if(!n) return naNil();
 150     return propNodeGhostCreate(c, n);
 151 }
 152
 153 static naRef f_getChild(naContext c, naRef me, int argc, naRef* args)
 154 {
 155     NODEARG();
 156     naRef child = naVec_get(argv, 0);
 157     if(!naIsString(child)) return naNil();
 158     naRef idx = naNumValue(naVec_get(argv, 1));
 159     bool create = naTrue(naVec_get(argv, 2));
 160     SGPropertyNode* n;
 161     try {
 162         if(naIsNil(idx) || !naIsNum(idx)) {
 163             n = (*node)->getChild(naStr_data(child), create);
 164         } else {
 165             n = (*node)->getChild(naStr_data(child), (int)idx.num, create);
 166         }
 167     } catch (const string& err) {
 168         naRuntimeError(c, (char *)err.c_str());
 169         return naNil();
 170     }
 171     if(!n) return naNil();
 172     return propNodeGhostCreate(c, n);
 173 }
 174
 175 static naRef f_getChildren(naContext c, naRef me, int argc, naRef* args)
 176 {
 177     NODEARG();
 178     naRef result = naNewVector(c);
 179     if(naIsNil(argv) || naVec_size(argv) == 0) {
 180         // Get all children
 181         for(int i=0; i<(*node)->nChildren(); i++)
 182             naVec_append(result, propNodeGhostCreate(c, (*node)->getChild(i)));
 183     } else {
 184         // Get all children of a specified name
 185         naRef name = naVec_get(argv, 0);
 186         if(!naIsString(name)) return naNil();
 187         try {
 188             vector<SGPropertyNode_ptr> children
 189                 = (*node)->getChildren(naStr_data(name));
 190             for(unsigned int i=0; i<children.size(); i++)
 191                 naVec_append(result, propNodeGhostCreate(c, children[i]));
 192         } catch (const string& err) {
 193             naRuntimeError(c, (char *)err.c_str());
 194             return naNil();
 195         }
 196     }
 197     return result;
 198 }
 199
 200 static naRef f_removeChild(naContext c, naRef me, int argc, naRef* args)
 201 {
 202     NODEARG();
 203     naRef child = naVec_get(argv, 0);
 204     naRef index = naVec_get(argv, 1);
 205     if(!naIsString(child) || !naIsNum(index)) return naNil();
 206     try {
 207         (*node)->removeChild(naStr_data(child), (int)index.num, false);
 208     } catch (const string& err) {
 209         naRuntimeError(c, (char *)err.c_str());
 210     }
 211     return naNil();
 212 }
 213
 214 static naRef f_removeChildren(naContext c, naRef me, int argc, naRef* args)
 215 {
 216     NODEARG();
 217     naRef result = naNewVector(c);
 218     if(naIsNil(argv) || naVec_size(argv) == 0) {
 219         // Remove all children
 220         for(int i = (*node)->nChildren() - 1; i >=0; i--)
 221             naVec_append(result, propNodeGhostCreate(c, (*node)->removeChild(i)));
 222     } else {
 223         // Remove all children of a specified name
 224         naRef name = naVec_get(argv, 0);
 225         if(!naIsString(name)) return naNil();
 226         try {
 227             vector<SGPropertyNode_ptr> children
 228                 = (*node)->removeChildren(naStr_data(name), false);
 229             for(unsigned int i=0; i<children.size(); i++)
 230                 naVec_append(result, propNodeGhostCreate(c, children[i]));
 231         } catch (const string& err) {
 232             naRuntimeError(c, (char *)err.c_str());
 233             return naNil();
 234         }
 235     }
 236     return result;
 237 }
 238
 239 static naRef f_getNode(naContext c, naRef me, int argc, naRef* args)
 240 {
 241     NODEARG();
 242     naRef path = naVec_get(argv, 0);
 243     bool create = naTrue(naVec_get(argv, 1));
 244     if(!naIsString(path)) return naNil();
 245     SGPropertyNode* n;
 246     try {
 247         n = (*node)->getNode(naStr_data(path), create);
 248     } catch (const string& err) {
 249         naRuntimeError(c, (char *)err.c_str());
 250         return naNil();
 251     }
 252     return propNodeGhostCreate(c, n);
 253 }
 254
 255 static naRef f_new(naContext c, naRef me, int argc, naRef* args)
 256 {
 257     return propNodeGhostCreate(c, new SGPropertyNode());
 258 }
 259
 260 static naRef f_globals(naContext c, naRef me, int argc, naRef* args)
 261 {
 262     return propNodeGhostCreate(c, globals->get_props());
 263 }
 264
 265 static struct {
 266     naCFunction func;
 267     char* name;
 268 } propfuncs[] = {
 269     { f_getType, "_getType" },
 270     { f_getName, "_getName" },
 271     { f_getIndex, "_getIndex" },
 272     { f_getValue, "_getValue" },
 273     { f_setValue, "_setValue" },
 274     { f_setIntValue, "_setIntValue" },
 275     { f_setBoolValue, "_setBoolValue" },
 276     { f_setDoubleValue, "_setDoubleValue" },
 277     { f_getParent, "_getParent" },
 278     { f_getChild, "_getChild" },
 279     { f_getChildren, "_getChildren" },
 280     { f_removeChild, "_removeChild" },
 281     { f_removeChildren, "_removeChildren" },
 282     { f_getNode, "_getNode" },
 283     { f_new, "_new" },
 284     { f_globals, "_globals" },
 285     { 0, 0 }
 286 };
 287
 288 naRef FGNasalSys::genPropsModule()
 289 {
 290     naRef namespc = naNewHash(_context);
 291     for(int i=0; propfuncs[i].name; i++)
 292         hashset(namespc, propfuncs[i].name,
 293                 naNewFunc(_context, naNewCCode(_context, propfuncs[i].func)));
 294     return namespc;
 295 }