7 # include <sys/time.h> // gettimeofday
12 #include <sys/types.h>
17 #include <simgear/nasal/nasal.h>
18 #include <simgear/props/props.hxx>
19 #include <simgear/math/sg_random.h>
20 #include <simgear/misc/sg_path.hxx>
21 #include <simgear/misc/interpolator.hxx>
22 #include <simgear/structure/commands.hxx>
24 #include <Main/globals.hxx>
25 #include <Main/fg_props.hxx>
27 #include "NasalSys.hxx"
29 static FGNasalSys* nasalSys = 0;
32 // Read and return file contents in a single buffer. Note use of
33 // stat() to get the file size. This is a win32 function, believe it
34 // or not. :) Note the REALLY IMPORTANT use of the "b" flag to fopen.
35 // Text mode brain damage will kill us if we're trying to do bytewise
37 static char* readfile(const char* file, int* lenOut)
40 if(stat(file, &data) != 0) return 0;
41 FILE* f = fopen(file, "rb");
43 char* buf = new char[data.st_size];
44 *lenOut = fread(buf, 1, data.st_size, f);
46 if(*lenOut != data.st_size) {
47 // Shouldn't happen, but warn anyway since it represents a
48 // platform bug and not a typical runtime error (missing file,
50 SG_LOG(SG_NASAL, SG_ALERT,
51 "ERROR in Nasal initialization: " <<
52 "short count returned from fread() of " << file <<
53 ". Check your C library!");
60 FGNasalSys::FGNasalSys()
66 _nextGCKey = 0; // Any value will do
68 _purgeListeners = false;
71 // Does a naCall() in a new context. Wrapped here to make lock
72 // tracking easier. Extension functions are called with the lock, but
73 // we have to release it before making a new naCall(). So rather than
74 // drop the lock in every extension function that might call back into
75 // Nasal, we keep a stack depth counter here and only unlock/lock
76 // around the naCall if it isn't the first one.
77 naRef FGNasalSys::call(naRef code, naRef locals)
79 naContext ctx = naNewContext();
80 if(_callCount) naModUnlock();
82 naRef result = naCall(ctx, code, 0, 0, naNil(), locals);
86 if(_callCount) naModLock();
91 FGNasalSys::~FGNasalSys()
94 map<int, FGNasalListener *>::iterator it, end = _listener.end();
95 for(it = _listener.begin(); it != end; ++it)
98 naFreeContext(_context);
102 bool FGNasalSys::parseAndRun(const char* sourceCode)
104 naRef code = parse("FGNasalSys::parseAndRun()", sourceCode,
112 FGNasalScript* FGNasalSys::parseScript(const char* src, const char* name)
114 FGNasalScript* script = new FGNasalScript();
115 script->_gcKey = -1; // important, if we delete it on a parse
116 script->_nas = this; // error, don't clobber a real handle!
120 sprintf(buf, "FGNasalScript@%p", (void *)script);
124 script->_code = parse(name, src, strlen(src));
125 if(naIsNil(script->_code)) {
130 script->_gcKey = gcSave(script->_code);
134 // Utility. Sets a named key in a hash by C string, rather than nasal
136 void FGNasalSys::hashset(naRef hash, const char* key, naRef val)
138 naRef s = naNewString(_context);
139 naStr_fromdata(s, (char*)key, strlen(key));
140 naHash_set(hash, s, val);
143 // The get/setprop functions accept a *list* of strings and walk
144 // through the property tree with them to find the appropriate node.
145 // This allows a Nasal object to hold onto a property path and use it
146 // like a node object, e.g. setprop(ObjRoot, "size-parsecs", 2.02). This
147 // is the utility function that walks the property tree.
148 // Future enhancement: support integer arguments to specify array
150 static SGPropertyNode* findnode(naContext c, naRef* vec, int len)
152 SGPropertyNode* p = globals->get_props();
154 for(int i=0; i<len; i++) {
156 if(!naIsString(a)) return 0;
157 p = p->getNode(naStr_data(a));
160 } catch (const string& err) {
161 naRuntimeError(c, (char *)err.c_str());
167 // getprop() extension function. Concatenates its string arguments as
168 // property names and returns the value of the specified property. Or
169 // nil if it doesn't exist.
170 static naRef f_getprop(naContext c, naRef me, int argc, naRef* args)
172 const SGPropertyNode* p = findnode(c, args, argc);
173 if(!p) return naNil();
175 switch(p->getType()) {
176 case SGPropertyNode::BOOL: case SGPropertyNode::INT:
177 case SGPropertyNode::LONG: case SGPropertyNode::FLOAT:
178 case SGPropertyNode::DOUBLE:
179 return naNum(p->getDoubleValue());
181 case SGPropertyNode::STRING:
182 case SGPropertyNode::UNSPECIFIED:
184 naRef nastr = naNewString(c);
185 const char* val = p->getStringValue();
186 naStr_fromdata(nastr, (char*)val, strlen(val));
189 case SGPropertyNode::ALIAS: // <--- FIXME, recurse?
195 // setprop() extension function. Concatenates its string arguments as
196 // property names and sets the value of the specified property to the
198 static naRef f_setprop(naContext c, naRef me, int argc, naRef* args)
201 char buf[BUFLEN + 1];
205 for(int i=0; i<argc-1; i++) {
206 naRef s = naStringValue(c, args[i]);
207 if(naIsNil(s)) return naNil();
208 strncpy(p, naStr_data(s), buflen);
210 buflen = BUFLEN - (p - buf);
211 if(i < (argc-2) && buflen > 0) {
217 SGPropertyNode* props = globals->get_props();
218 naRef val = args[argc-1];
220 if(naIsString(val)) props->setStringValue(buf, naStr_data(val));
222 naRef n = naNumValue(val);
224 naRuntimeError(c, "setprop() value is not string or number");
225 props->setDoubleValue(buf, n.num);
227 } catch (const string& err) {
228 naRuntimeError(c, (char *)err.c_str());
234 // print() extension function. Concatenates and prints its arguments
235 // to the FlightGear log. Uses the highest log level (SG_ALERT), to
236 // make sure it appears. Is there better way to do this?
237 static naRef f_print(naContext c, naRef me, int argc, naRef* args)
241 for(int i=0; i<n; i++) {
242 naRef s = naStringValue(c, args[i]);
243 if(naIsNil(s)) continue;
244 buf += naStr_data(s);
246 SG_LOG(SG_GENERAL, SG_ALERT, buf);
247 return naNum(buf.length());
250 // fgcommand() extension function. Executes a named command via the
251 // FlightGear command manager. Takes a single property node name as
253 static naRef f_fgcommand(naContext c, naRef me, int argc, naRef* args)
255 if(argc < 2 || !naIsString(args[0]) || !naIsGhost(args[1]))
256 naRuntimeError(c, "bad arguments to fgcommand()");
257 naRef cmd = args[0], props = args[1];
258 SGPropertyNode_ptr* node = (SGPropertyNode_ptr*)naGhost_ptr(props);
259 return naNum(globals->get_commands()->execute(naStr_data(cmd), *node));
262 // settimer(func, dt, simtime) extension function. Falls through to
263 // FGNasalSys::setTimer(). See there for docs.
264 static naRef f_settimer(naContext c, naRef me, int argc, naRef* args)
266 nasalSys->setTimer(c, argc, args);
270 // setlistener(func, property, bool) extension function. Falls through to
271 // FGNasalSys::setListener(). See there for docs.
272 static naRef f_setlistener(naContext c, naRef me, int argc, naRef* args)
274 return nasalSys->setListener(c, argc, args);
277 // removelistener(int) extension function. Falls through to
278 // FGNasalSys::removeListener(). See there for docs.
279 static naRef f_removelistener(naContext c, naRef me, int argc, naRef* args)
281 return nasalSys->removeListener(c, argc, args);
284 // Returns a ghost handle to the argument to the currently executing
286 static naRef f_cmdarg(naContext c, naRef me, int argc, naRef* args)
288 return nasalSys->cmdArgGhost();
291 // Sets up a property interpolation. The first argument is either a
292 // ghost (SGPropertyNode_ptr*) or a string (global property path) to
293 // interpolate. The second argument is a vector of pairs of
294 // value/delta numbers.
295 static naRef f_interpolate(naContext c, naRef me, int argc, naRef* args)
297 SGPropertyNode* node;
298 naRef prop = argc > 0 ? args[0] : naNil();
299 if(naIsString(prop)) node = fgGetNode(naStr_data(prop), true);
300 else if(naIsGhost(prop)) node = *(SGPropertyNode_ptr*)naGhost_ptr(prop);
303 naRef curve = argc > 1 ? args[1] : naNil();
304 if(!naIsVector(curve)) return naNil();
305 int nPoints = naVec_size(curve) / 2;
306 double* values = new double[nPoints];
307 double* deltas = new double[nPoints];
308 for(int i=0; i<nPoints; i++) {
309 values[i] = naNumValue(naVec_get(curve, 2*i)).num;
310 deltas[i] = naNumValue(naVec_get(curve, 2*i+1)).num;
313 ((SGInterpolator*)globals->get_subsystem_mgr()
314 ->get_group(SGSubsystemMgr::INIT)->get_subsystem("interpolator"))
315 ->interpolate(node, nPoints, values, deltas);
320 // This is a better RNG than the one in the default Nasal distribution
321 // (which is based on the C library rand() implementation). It will
323 static naRef f_rand(naContext c, naRef me, int argc, naRef* args)
325 return naNum(sg_random());
328 static naRef f_srand(naContext c, naRef me, int argc, naRef* args)
334 // Return an array listing of all files in a directory
335 static naRef f_directory(naContext c, naRef me, int argc, naRef* args)
337 if(argc != 1 || !naIsString(args[0]))
338 naRuntimeError(c, "bad arguments to directory()");
339 naRef ldir = args[0];
340 ulDir* dir = ulOpenDir(naStr_data(args[0]));
341 if(!dir) return naNil();
342 naRef result = naNewVector(c);
344 while((dent = ulReadDir(dir)))
345 naVec_append(result, naStr_fromdata(naNewString(c), dent->d_name,
346 strlen(dent->d_name)));
351 // Return UNIX epoch time in seconds.
352 static naRef f_systime(naContext c, naRef me, int argc, naRef* args)
356 GetSystemTimeAsFileTime(&ft);
357 double t = (4294967296.0 * ft.dwHighDateTime + ft.dwLowDateTime);
358 // Converts from 100ns units in 1601 epoch to unix epoch in sec
359 return naNum((t * 1e-7) - 11644473600.0);
363 do { t = time(0); gettimeofday(&td, 0); } while(t != time(0));
364 return naNum(t + 1e-6 * td.tv_usec);
369 // Table of extension functions. Terminate with zeros.
370 static struct { char* name; naCFunction func; } funcs[] = {
371 { "getprop", f_getprop },
372 { "setprop", f_setprop },
373 { "print", f_print },
374 { "_fgcommand", f_fgcommand },
375 { "settimer", f_settimer },
376 { "_setlistener", f_setlistener },
377 { "removelistener", f_removelistener },
378 { "_cmdarg", f_cmdarg },
379 { "_interpolate", f_interpolate },
381 { "srand", f_srand },
382 { "directory", f_directory },
383 { "systime", f_systime },
387 naRef FGNasalSys::cmdArgGhost()
389 return propNodeGhost(_cmdArg);
392 void FGNasalSys::init()
396 _context = naNewContext();
398 // Start with globals. Add it to itself as a recursive
399 // sub-reference under the name "globals". This gives client-code
400 // write access to the namespace if someone wants to do something
402 _globals = naInit_std(_context);
403 naSave(_context, _globals);
404 hashset(_globals, "globals", _globals);
406 hashset(_globals, "math", naInit_math(_context));
407 hashset(_globals, "bits", naInit_bits(_context));
408 hashset(_globals, "io", naInit_io(_context));
409 hashset(_globals, "thread", naInit_thread(_context));
410 hashset(_globals, "utf8", naInit_utf8(_context));
412 // Add our custom extension functions:
413 for(i=0; funcs[i].name; i++)
414 hashset(_globals, funcs[i].name,
415 naNewFunc(_context, naNewCCode(_context, funcs[i].func)));
417 // And our SGPropertyNode wrapper
418 hashset(_globals, "props", genPropsModule());
420 // Make a "__gcsave" hash to hold the naRef objects which get
421 // passed to handles outside the interpreter (to protect them from
422 // begin garbage-collected).
423 _gcHash = naNewHash(_context);
424 hashset(_globals, "__gcsave", _gcHash);
426 // Now load the various source files in the Nasal directory
427 SGPath p(globals->get_fg_root());
430 ulDir* dir = ulOpenDir(p.c_str());
431 while(dir && (dent = ulReadDir(dir)) != 0) {
433 fullpath.append(dent->d_name);
434 SGPath file(dent->d_name);
435 if(file.extension() != "nas") continue;
436 loadModule(fullpath, file.base().c_str());
440 // set signal and remove node to avoid restoring at reinit
441 const char *s = "nasal-dir-initialized";
442 SGPropertyNode *signal = fgGetNode("/sim/signals", true);
443 signal->setBoolValue(s, true);
444 signal->removeChildren(s);
446 // Pull scripts out of the property tree, too
447 loadPropertyScripts();
450 void FGNasalSys::update(double)
452 if(_purgeListeners) {
453 _purgeListeners = false;
454 map<int, FGNasalListener *>::iterator it;
455 for(it = _listener.begin(); it != _listener.end();) {
456 FGNasalListener *nl = it->second;
458 _listener.erase(it++);
467 // Loads the scripts found under /nasal in the global tree
468 void FGNasalSys::loadPropertyScripts()
470 SGPropertyNode* nasal = globals->get_props()->getNode("nasal");
473 for(int i=0; i<nasal->nChildren(); i++) {
474 SGPropertyNode* n = nasal->getChild(i);
476 const char* module = n->getName();
477 if(n->hasChild("module"))
478 module = n->getStringValue("module");
480 // allow multiple files to be specified within in a single
484 bool file_specified = false;
485 while ( (fn = n->getChild("file", j)) != NULL ) {
486 file_specified = true;
487 const char* file = fn->getStringValue();
488 SGPath p(globals->get_fg_root());
490 loadModule(p, module);
494 // Old code which only allowed a single file to be specified per module
496 const char* file = n->getStringValue("file");
497 if(!n->hasChild("file")) file = 0; // Hrm...
499 SGPath p(globals->get_fg_root());
501 loadModule(p, module);
505 const char* src = n->getStringValue("script");
506 if(!n->hasChild("script")) src = 0; // Hrm...
508 createModule(module, n->getPath(), src, strlen(src));
510 if(!file_specified && !src)
511 SG_LOG(SG_NASAL, SG_ALERT, "Nasal error: " <<
512 "no <file> or <script> defined in " <<
513 "/nasal/" << module);
517 // Logs a runtime error, with stack trace, to the FlightGear log stream
518 void FGNasalSys::logError(naContext context)
520 SG_LOG(SG_NASAL, SG_ALERT,
521 "Nasal runtime error: " << naGetError(context));
522 SG_LOG(SG_NASAL, SG_ALERT,
523 " at " << naStr_data(naGetSourceFile(context, 0)) <<
524 ", line " << naGetLine(context, 0));
525 for(int i=1; i<naStackDepth(context); i++)
526 SG_LOG(SG_NASAL, SG_ALERT,
527 " called from: " << naStr_data(naGetSourceFile(context, i)) <<
528 ", line " << naGetLine(context, i));
531 // Reads a script file, executes it, and places the resulting
532 // namespace into the global namespace under the specified module
534 void FGNasalSys::loadModule(SGPath file, const char* module)
537 char* buf = readfile(file.c_str(), &len);
539 SG_LOG(SG_NASAL, SG_ALERT,
540 "Nasal error: could not read script file " << file.c_str()
541 << " into module " << module);
545 createModule(module, file.c_str(), buf, len);
549 // Parse and run. Save the local variables namespace, as it will
550 // become a sub-object of globals. The optional "arg" argument can be
551 // used to pass an associated property node to the module, which can then
552 // be accessed via cmdarg(). (This is, for example, used by XML dialogs.)
553 void FGNasalSys::createModule(const char* moduleName, const char* fileName,
554 const char* src, int len, const SGPropertyNode* arg)
556 naRef code = parse(fileName, src, len);
560 // See if we already have a module hash to use. This allows the
561 // user to, for example, add functions to the built-in math
562 // module. Make a new one if necessary.
564 naRef modname = naNewString(_context);
565 naStr_fromdata(modname, (char*)moduleName, strlen(moduleName));
566 if(!naHash_get(_globals, modname, &locals))
567 locals = naNewHash(_context);
569 _cmdArg = (SGPropertyNode*)arg;
572 hashset(_globals, moduleName, locals);
575 void FGNasalSys::deleteModule(const char* moduleName)
577 naRef modname = naNewString(_context);
578 naStr_fromdata(modname, (char*)moduleName, strlen(moduleName));
579 naHash_delete(_globals, modname);
582 naRef FGNasalSys::parse(const char* filename, const char* buf, int len)
585 naRef srcfile = naNewString(_context);
586 naStr_fromdata(srcfile, (char*)filename, strlen(filename));
587 naRef code = naParseCode(_context, srcfile, 1, (char*)buf, len, &errLine);
589 SG_LOG(SG_NASAL, SG_ALERT,
590 "Nasal parse error: " << naGetError(_context) <<
591 " in "<< filename <<", line " << errLine);
595 // Bind to the global namespace before returning
596 return naBindFunction(_context, code, _globals);
599 bool FGNasalSys::handleCommand(const SGPropertyNode* arg)
601 const char* nasal = arg->getStringValue("script");
602 const char* moduleName = arg->getStringValue("module");
603 naRef code = parse(arg->getPath(true), nasal, strlen(nasal));
604 if(naIsNil(code)) return false;
606 // Commands can be run "in" a module. Make sure that module
607 // exists, and set it up as the local variables hash for the
609 naRef locals = naNil();
611 naRef modname = naNewString(_context);
612 naStr_fromdata(modname, (char*)moduleName, strlen(moduleName));
613 if(!naHash_get(_globals, modname, &locals)) {
614 locals = naNewHash(_context);
615 naHash_set(_globals, modname, locals);
619 // Cache this command's argument for inspection via cmdarg(). For
620 // performance reasons, we won't bother with it if the invoked
621 // code doesn't need it.
622 _cmdArg = (SGPropertyNode*)arg;
628 // settimer(func, dt, simtime) extension function. The first argument
629 // is a Nasal function to call, the second is a delta time (from now),
630 // in seconds. The third, if present, is a boolean value indicating
631 // that "real world" time (rather than simulator time) is to be used.
633 // Implementation note: the FGTimer objects don't live inside the
634 // garbage collector, so the Nasal handler functions have to be
635 // "saved" somehow lest they be inadvertently cleaned. In this case,
636 // they are inserted into a globals.__gcsave hash and removed on
638 void FGNasalSys::setTimer(naContext c, int argc, naRef* args)
640 // Extract the handler, delta, and simtime arguments:
641 naRef handler = argc > 0 ? args[0] : naNil();
642 if(!(naIsCode(handler) || naIsCCode(handler) || naIsFunc(handler))) {
643 naRuntimeError(c, "settimer() with invalid function argument");
647 naRef delta = argc > 1 ? args[1] : naNil();
649 naRuntimeError(c, "settimer() with invalid time argument");
653 bool simtime = (argc > 2 && naTrue(args[2])) ? false : true;
655 // Generate and register a C++ timer handler
656 NasalTimer* t = new NasalTimer;
657 t->handler = handler;
658 t->gcKey = gcSave(handler);
661 globals->get_event_mgr()->addEvent("NasalTimer",
662 t, &NasalTimer::timerExpired,
666 void FGNasalSys::handleTimer(NasalTimer* t)
668 call(t->handler, naNil());
672 int FGNasalSys::gcSave(naRef r)
674 int key = _nextGCKey++;
675 naHash_set(_gcHash, naNum(key), r);
679 void FGNasalSys::gcRelease(int key)
681 naHash_delete(_gcHash, naNum(key));
684 void FGNasalSys::NasalTimer::timerExpired()
686 nasal->handleTimer(this);
690 int FGNasalSys::_listenerId = 0;
692 // setlistener(property, func, bool) extension function. The first argument
693 // is either a ghost (SGPropertyNode_ptr*) or a string (global property
694 // path), the second is a Nasal function, the optional third one a bool.
695 // If the bool is true, then the listener is executed initially. The
696 // setlistener() function returns a unique id number, that can be used
697 // as argument to the removelistener() function.
698 naRef FGNasalSys::setListener(naContext c, int argc, naRef* args)
700 SGPropertyNode_ptr node;
701 naRef prop = argc > 0 ? args[0] : naNil();
702 if(naIsString(prop)) node = fgGetNode(naStr_data(prop), true);
703 else if(naIsGhost(prop)) node = *(SGPropertyNode_ptr*)naGhost_ptr(prop);
705 naRuntimeError(c, "setlistener() with invalid property argument");
710 SG_LOG(SG_NASAL, SG_DEBUG, "Attaching listener to tied property " <<
713 naRef handler = argc > 1 ? args[1] : naNil();
714 if(!(naIsCode(handler) || naIsCCode(handler) || naIsFunc(handler))) {
715 naRuntimeError(c, "setlistener() with invalid function argument");
719 bool initial = argc > 2 && naTrue(args[2]);
721 FGNasalListener *nl = new FGNasalListener(node, handler, this,
722 gcSave(handler), _listenerId);
723 node->addChangeListener(nl, initial);
725 _listener[_listenerId] = nl;
726 return naNum(_listenerId++);
729 // removelistener(int) extension function. The argument is the id of
730 // a listener as returned by the setlistener() function.
731 naRef FGNasalSys::removeListener(naContext c, int argc, naRef* args)
733 naRef id = argc > 0 ? args[0] : naNil();
734 map<int, FGNasalListener *>::iterator it = _listener.find(int(id.num));
736 if(!naIsNum(id) || it == _listener.end() || it->second->_dead) {
737 naRuntimeError(c, "removelistener() with invalid listener id");
741 FGNasalListener *nl = it->second;
744 _purgeListeners = true;
750 return naNum(_listener.size());
755 // FGNasalListener class.
757 FGNasalListener::FGNasalListener(SGPropertyNode_ptr node, naRef handler,
758 FGNasalSys* nasal, int key, int id) :
769 FGNasalListener::~FGNasalListener()
771 _node->removeChangeListener(this);
772 _nas->gcRelease(_gcKey);
775 void FGNasalListener::valueChanged(SGPropertyNode* node)
777 // drop recursive listener calls
781 SG_LOG(SG_NASAL, SG_DEBUG, "trigger listener #" << _id);
783 _nas->_cmdArg = node;
784 _nas->call(_handler, naNil());
791 // FGNasalModelData class. If sgLoad3DModel() is called with a pointer to
792 // such a class, then it lets modelLoaded() run the <load> script, and the
793 // destructor the <unload> script. The latter happens when the model branch
794 // is removed from the scene graph.
796 void FGNasalModelData::modelLoaded(const string& path, SGPropertyNode *prop,
799 SGPropertyNode *n = prop->getNode("nasal"), *load;
803 load = n->getNode("load");
804 _unload = n->getNode("unload");
805 if(!load && !_unload)
809 const char *s = load ? load->getStringValue() : "";
810 nasalSys->createModule(_module.c_str(), _module.c_str(), s, strlen(s), _props);
813 FGNasalModelData::~FGNasalModelData()
819 SG_LOG(SG_NASAL, SG_ALERT, "Trying to run an <unload> script "
820 "without Nasal subsystem present.");
825 const char *s = _unload->getStringValue();
826 nasalSys->createModule(_module.c_str(), _module.c_str(), s, strlen(s), _props);
828 nasalSys->deleteModule(_module.c_str());