| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928 |
- /**
- * PANDA 3D SOFTWARE
- * Copyright (c) Carnegie Mellon University. All rights reserved.
- *
- * All use of this software is subject to the terms of the revised BSD
- * license. You should have received a copy of this license along
- * with this source code in a file named "LICENSE."
- *
- * @file p3dPythonRun.cxx
- * @author drose
- * @date 2009-06-05
- */
- #include "p3dPythonRun.h"
- #include "asyncTaskManager.h"
- #include "binaryXml.h"
- #include "multifile.h"
- #include "virtualFileSystem.h"
- #include "nativeWindowHandle.h"
- #include "py_panda.h"
- extern "C" {
- // This has been compiled-in by the build system, if all is well.
- extern struct _frozen _PyImport_FrozenModules[];
- };
- // There is only one P3DPythonRun object in any given process space. Makes
- // the statics easier to deal with, and we don't need multiple instances of
- // this thing.
- P3DPythonRun *P3DPythonRun::_global_ptr = nullptr;
- TypeHandle P3DPythonRun::P3DWindowHandle::_type_handle;
- /**
- *
- */
- P3DPythonRun::
- P3DPythonRun(const char *program_name, const char *archive_file,
- FHandle input_handle, FHandle output_handle,
- const char *log_pathname, bool interactive_console) {
- P3DWindowHandle::init_type();
- init_xml();
- _read_thread_continue = false;
- _program_continue = true;
- _session_terminated = false;
- _taskMgr = nullptr;
- INIT_LOCK(_commands_lock);
- INIT_THREAD(_read_thread);
- _session_id = 0;
- _next_sent_id = 0;
- _interactive_console = interactive_console;
- if (program_name != nullptr) {
- #if PY_MAJOR_VERSION >= 3
- // Python 3 case: we have to convert it to a wstring.
- TextEncoder enc;
- enc.set_text(program_name);
- _program_name = enc.get_wtext();
- #else
- // Python 2 is happy with a regular string.
- _program_name = program_name;
- #endif
- }
- if (archive_file != nullptr) {
- _archive_file = Filename::from_os_specific(archive_file);
- }
- _py_argc = 1;
- #if PY_MAJOR_VERSION >= 3
- _py_argv[0] = (wchar_t *)_program_name.c_str();
- #else
- _py_argv[0] = (char *)_program_name.c_str();
- #endif
- _py_argv[1] = nullptr;
- #ifdef NDEBUG
- // In OPTIMIZE 4 compilation mode, run Python in optimized mode too.
- extern int Py_OptimizeFlag;
- Py_OptimizeFlag = 2;
- #endif
- // Turn off the automatic load of site.py at startup.
- extern int Py_NoSiteFlag;
- Py_NoSiteFlag = 1;
- Py_NoUserSiteDirectory = 1;
- // Tell Python not to write bytecode files for loaded modules.
- Py_DontWriteBytecodeFlag = 1;
- // Prevent Python from complaining about finding the standard modules.
- Py_FrozenFlag = 1;
- // This contains the modules we need in order to call Py_Initialize, as well
- // as the VFSImporter.
- PyImport_FrozenModules = _PyImport_FrozenModules;
- // Initialize Python. It appears to be important to do this before we open
- // the pipe streams and spawn the thread, below.
- #if PY_MAJOR_VERSION >= 3
- Py_SetProgramName((wchar_t *)_program_name.c_str());
- Py_SetPythonHome((wchar_t *)L"");
- #else
- Py_SetProgramName((char *)_program_name.c_str());
- Py_SetPythonHome((char *)"");
- #endif
- Py_Initialize();
- PyEval_InitThreads();
- PySys_SetArgvEx(_py_argc, _py_argv, 0);
- // Open the error output before we do too much more.
- if (log_pathname != nullptr && *log_pathname != '\0') {
- Filename f = Filename::from_os_specific(log_pathname);
- f.set_text();
- if (f.open_write(_error_log)) {
- // Set up the indicated error log as the Notify output.
- _error_log.setf(ios::unitbuf);
- Notify::ptr()->set_ostream_ptr(&_error_log, false);
- }
- }
- // Open the pipe streams with the input and output handles from the parent.
- _pipe_read.open_read(input_handle);
- _pipe_write.open_write(output_handle);
- if (!_pipe_read) {
- nout << "unable to open read pipe\n";
- }
- if (!_pipe_write) {
- nout << "unable to open write pipe\n";
- }
- spawn_read_thread();
- }
- /**
- *
- */
- P3DPythonRun::
- ~P3DPythonRun() {
- terminate_session();
- // Close the write pipe, so the parent process will terminate us.
- _pipe_write.close();
- // Shut down Python and Panda.
- Py_Finalize();
- join_read_thread();
- DESTROY_LOCK(_commands_lock);
- // Restore the notify stream in case it tries to write to anything else
- // after our shutdown.
- Notify::ptr()->set_ostream_ptr(&cerr, false);
- }
- /**
- * Runs the embedded Python process. This method does not return until the
- * plugin is ready to exit.
- *
- * Returns the exit status, which will be 0 on success.
- */
- int P3DPythonRun::
- run_python() {
- #if defined(_WIN32) && defined(USE_DEBUG_PYTHON)
- // On Windows, in a debug build, we have to preload sys.dll_suffix = "_d",
- // so that the Panda DLL preloader can import the correct filenames.
- PyRun_SimpleString("import sys; sys.dll_suffix = '_d'");
- #endif
- Filename dir = _archive_file.get_dirname();
- // We'll need to synthesize a 'panda3d' module before loading VFSImporter.
- // We could simply freeze it, but Python has a bug setting __path__ of
- // frozen modules properly.
- PyObject *panda3d_module = PyImport_AddModule("panda3d");
- if (panda3d_module == nullptr) {
- nout << "Failed to add panda3d module:\n";
- PyErr_Print();
- return 1;
- }
- // Set the __path__ such that it can find panda3dcore.pyd, etc.
- Filename panda3d_dir(dir, "panda3d");
- string dir_str = panda3d_dir.to_os_specific();
- PyModule_AddObject(panda3d_module, "__path__", Py_BuildValue("[s#]", dir_str.data(), dir_str.length()));
- PyModule_AddStringConstant(panda3d_module, "__package__", "panda3d");
- // Import the VFSImporter module that was frozen in.
- PyObject *vfsimporter_module = PyImport_ImportModule("direct.showbase.VFSImporter");
- if (vfsimporter_module == nullptr) {
- nout << "Failed to import VFSImporter:\n";
- PyErr_Print();
- return 1;
- }
- // Now repair the "direct" and "direct.showbase" trees, which were
- // presumably frozen along with the VFSImporter, by setting their __path__
- // such that we can still find the other direct modules.
- Filename direct_dir(dir, "direct");
- PyObject *direct_module = PyImport_AddModule("direct");
- if (direct_module != nullptr) {
- dir_str = direct_dir.to_os_specific();
- PyModule_AddObject(direct_module, "__path__", Py_BuildValue("[s#]", dir_str.data(), dir_str.length()));
- PyModule_AddStringConstant(direct_module, "__package__", "direct");
- }
- PyObject *showbase_module = PyImport_AddModule("direct.showbase");
- if (showbase_module != nullptr) {
- Filename showbase_dir(direct_dir, "showbase");
- dir_str = showbase_dir.to_os_specific();
- PyModule_AddObject(showbase_module, "__path__", Py_BuildValue("[s#]", dir_str.data(), dir_str.length()));
- PyModule_AddStringConstant(showbase_module, "__package__", "direct.showbase");
- }
- PyObject *stdpy_module = PyImport_AddModule("direct.stdpy");
- if (stdpy_module != nullptr) {
- Filename stdpy_dir(direct_dir, "stdpy");
- dir_str = stdpy_dir.to_os_specific();
- PyModule_AddObject(stdpy_module, "__path__", Py_BuildValue("[s#]", dir_str.data(), dir_str.length()));
- PyModule_AddStringConstant(stdpy_module, "__package__", "direct.stdpy");
- }
- // And the encodings package as well, since we've presumably picked up a
- // bunch of encodings modules as part of the frozen bundle.
- Filename encodings_dir(dir, "encodings");
- PyObject *encodings_module = PyImport_AddModule("encodings");
- if (encodings_module != nullptr) {
- dir_str = encodings_dir.to_os_specific();
- PyModule_AddObject(encodings_module, "__path__", Py_BuildValue("[s#]", dir_str.data(), dir_str.length()));
- PyModule_AddStringConstant(encodings_module, "__package__", "encodings");
- }
- // And register the VFSImporter.
- PyObject *result = PyObject_CallMethod(vfsimporter_module, (char *)"register", (char *)"");
- if (result == nullptr) {
- nout << "Failed to call VFSImporter.register():\n";
- PyErr_Print();
- return 1;
- }
- Py_DECREF(result);
- Py_DECREF(vfsimporter_module);
- // Now, the VFSImporter has been registered, which means we can start
- // importing the rest of the Python modules, which are all defined in the
- // multifile. First, we need to mount the multifile into the VFS.
- PT(Multifile) mf = new Multifile;
- if (!mf->open_read(_archive_file)) {
- nout << "Could not read " << _archive_file << "\n";
- return 1;
- }
- VirtualFileSystem *vfs = VirtualFileSystem::get_global_ptr();
- if (!vfs->mount(mf, dir, VirtualFileSystem::MF_read_only)) {
- nout << "Could not mount " << _archive_file << "\n";
- return 1;
- }
- // And finally, we can import the startup module.
- PyObject *app_runner_module = PyImport_ImportModule("direct.p3d.AppRunner");
- if (app_runner_module == nullptr) {
- nout << "Failed to import direct.p3d.AppRunner\n";
- PyErr_Print();
- return 1;
- }
- // Get the pointers to the objects needed within the module.
- PyObject *app_runner_class = PyObject_GetAttrString(app_runner_module, "AppRunner");
- if (app_runner_class == nullptr) {
- nout << "Failed to get AppRunner class\n";
- PyErr_Print();
- return 1;
- }
- // Construct an instance of AppRunner.
- _runner = PyObject_CallFunction(app_runner_class, (char *)"");
- if (_runner == nullptr) {
- nout << "Failed to construct AppRunner instance\n";
- PyErr_Print();
- return 1;
- }
- Py_DECREF(app_runner_class);
- // Import the JavaScript module.
- PyObject *javascript_module = PyImport_ImportModule("direct.p3d.JavaScript");
- if (javascript_module == nullptr) {
- nout << "Failed to import direct.p3d.JavaScript\n";
- PyErr_Print();
- return false;
- }
- // Get the UndefinedObject class.
- _undefined_object_class = PyObject_GetAttrString(javascript_module, "UndefinedObject");
- if (_undefined_object_class == nullptr) {
- PyErr_Print();
- return 1;
- }
- // And the "Undefined" instance.
- _undefined = PyObject_GetAttrString(javascript_module, "Undefined");
- if (_undefined == nullptr) {
- PyErr_Print();
- return 1;
- }
- // Get the ConcreteStruct class.
- _concrete_struct_class = PyObject_GetAttrString(javascript_module, "ConcreteStruct");
- if (_concrete_struct_class == nullptr) {
- PyErr_Print();
- return 1;
- }
- // Get the BrowserObject class.
- _browser_object_class = PyObject_GetAttrString(javascript_module, "BrowserObject");
- if (_browser_object_class == nullptr) {
- PyErr_Print();
- return 1;
- }
- // Get the global TaskManager.
- _taskMgr = PyObject_GetAttrString(app_runner_module, "taskMgr");
- if (_taskMgr == nullptr) {
- PyErr_Print();
- return 1;
- }
- Py_DECREF(app_runner_module);
- Py_DECREF(javascript_module);
- // Construct a Python wrapper around our methods we need to expose to
- // Python.
- static PyMethodDef p3dpython_methods[] = {
- { "check_comm", P3DPythonRun::st_check_comm, METH_VARARGS,
- "Poll for communications from the parent process" },
- { "request_func", P3DPythonRun::st_request_func, METH_VARARGS,
- "Send an asynchronous request to the plugin host" },
- { nullptr, nullptr, 0, nullptr } /* Sentinel */
- };
- #if PY_MAJOR_VERSION >= 3
- static PyModuleDef p3dpython_module = {
- PyModuleDef_HEAD_INIT,
- "p3dpython",
- nullptr,
- -1,
- p3dpython_methods,
- nullptr, nullptr, nullptr, nullptr
- };
- PyObject *p3dpython = PyModule_Create(&p3dpython_module);
- #else
- PyObject *p3dpython = Py_InitModule("p3dpython", p3dpython_methods);
- #endif
- if (p3dpython == nullptr) {
- PyErr_Print();
- return 1;
- }
- PyObject *request_func = PyObject_GetAttrString(p3dpython, "request_func");
- if (request_func == nullptr) {
- PyErr_Print();
- return 1;
- }
- // Now pass that func pointer back to our AppRunner instance, so it can call
- // up to us.
- result = PyObject_CallMethod(_runner, (char *)"setRequestFunc", (char *)"N", request_func);
- if (result == nullptr) {
- PyErr_Print();
- return 1;
- }
- Py_DECREF(result);
- // Now add check_comm() as a task. It can be a threaded task, but this does
- // mean that application programmers will have to be alert to asynchronous
- // calls coming in from JavaScript. We'll put it on its own task chain so
- // the application programmer can decide how it should be.
- AsyncTaskManager *task_mgr = AsyncTaskManager::get_global_ptr();
- PT(AsyncTaskChain) chain = task_mgr->make_task_chain("JavaScript");
- // The default is not threaded (num_threads == 0), but if the app programmer
- // decides to enable threads, the default is TP_low priority.
- chain->set_thread_priority(TP_low);
- PyObject *check_comm = PyObject_GetAttrString(p3dpython, "check_comm");
- if (check_comm == nullptr) {
- PyErr_Print();
- return 1;
- }
- // Add it to the task manager. We do this instead of constructing a
- // PythonTask because linking p3dpython with core.pyd is problematic.
- result = PyObject_CallMethod(_taskMgr, (char *)"add", (char *)"Ns", check_comm, "check_comm");
- if (result == nullptr) {
- PyErr_Print();
- return 1;
- }
- Py_DECREF(result);
- // Finally, get lost in AppRunner.run() (which is really a call to
- // taskMgr.run()).
- PyObject *done = PyObject_CallMethod(_runner, (char *)"run", (char *)"");
- if (done == nullptr) {
- int status = 1;
- // An uncaught application exception, and not handled by
- // appRunner.exceptionHandler. If it is a SystemExit, extract the exit
- // status that we should return.
- if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
- PyObject *ptype, *ptraceback;
- PyObject *value = nullptr;
- PyErr_Fetch(&ptype, &value, &ptraceback);
- if (value != nullptr && PyExceptionInstance_Check(value)) {
- PyObject *code = PyObject_GetAttrString(value, "code");
- if (code) {
- Py_DECREF(value);
- value = code;
- }
- }
- if (value == nullptr || value == Py_None) {
- status = 0;
- #if PY_MAJOR_VERSION >= 3
- } else if (PyLong_Check(value)) {
- status = (int)PyLong_AsLong(value);
- #else
- } else if (PyInt_Check(value)) {
- status = (int)PyInt_AsLong(value);
- #endif
- } else {
- status = 1;
- }
- } else {
- PyErr_Print();
- }
- if (_interactive_console) {
- // Give an interactive user a chance to explore the exception.
- run_interactive_console();
- return 0;
- }
- // We're done.
- return status;
- }
- // A normal exit from the taskManager. We're presumably done.
- Py_DECREF(done);
- if (_interactive_console) {
- run_interactive_console();
- }
- return 0;
- }
- /**
- * Called from low-level Panda (via the P3DWindowHandle object) when a child
- * window has been attached or detached from the browser window. This
- * triggers the "onwindowattach" and "onwindowdetach" JavaScript
- * notifications.
- */
- void P3DPythonRun::
- set_window_open(P3DCInstance *inst, bool is_open) {
- TiXmlDocument doc;
- TiXmlElement *xrequest = new TiXmlElement("request");
- xrequest->SetAttribute("instance_id", inst->get_instance_id());
- xrequest->SetAttribute("rtype", "notify");
- if (is_open) {
- xrequest->SetAttribute("message", "onwindowattach");
- } else {
- xrequest->SetAttribute("message", "onwindowdetach");
- }
- doc.LinkEndChild(xrequest);
- write_xml(_pipe_write, &doc, nout);
- }
- /**
- * Called from low-level Panda (via the P3DWindowHandle object) when its main
- * window requires keyboard focus, but is unable to assign it directly. This
- * is particularly necessary under Windows Vista, where a child window of the
- * browser is specifically disallowed from being given keyboard focus.
- *
- * This sends a notify request up to the parent process, to ask the parent to
- * manage keyboard events by proxy, and send them back down to Panda, again
- * via the P3DWindowHandle.
- */
- void P3DPythonRun::
- request_keyboard_focus(P3DCInstance *inst) {
- TiXmlDocument doc;
- TiXmlElement *xrequest = new TiXmlElement("request");
- xrequest->SetAttribute("instance_id", inst->get_instance_id());
- xrequest->SetAttribute("rtype", "notify");
- xrequest->SetAttribute("message", "keyboardfocus");
- doc.LinkEndChild(xrequest);
- write_xml(_pipe_write, &doc, nout);
- }
- /**
- * Gives the user a chance to type interactive Python commands, for easy
- * development of a p3d application. This method is only called if
- * "interactive_console=1" is set as a web token, and "allow_python_dev" is
- * set within the application itself.
- */
- void P3DPythonRun::
- run_interactive_console() {
- #ifdef _WIN32
- // Make sure that control-C support is enabled for the interpreter.
- SetConsoleCtrlHandler(nullptr, false);
- #endif
- // The "readline" module makes the Python prompt friendlier, with command
- // history and everything. Simply importing it is sufficient.
- PyObject *readline_module = PyImport_ImportModule("readline");
- if (readline_module == nullptr) {
- // But, the module might not exist on certain platforms. If not, no
- // sweat.
- PyErr_Clear();
- } else {
- Py_DECREF(readline_module);
- }
- PyRun_InteractiveLoop(stdin, "<stdin>");
- }
- /**
- * Handles a command received from the plugin host, via an XML syntax on the
- * wire. Ownership of the XML document object is passed into this method.
- *
- * It's important *not* to be holding _commands_lock when calling this method.
- */
- void P3DPythonRun::
- handle_command(TiXmlDocument *doc) {
- TiXmlElement *xcommand = doc->FirstChildElement("command");
- if (xcommand != nullptr) {
- bool needs_response = false;
- int want_response_id;
- if (xcommand->QueryIntAttribute("want_response_id", &want_response_id) == TIXML_SUCCESS) {
- // This command will be waiting for a response.
- needs_response = true;
- }
- const char *cmd = xcommand->Attribute("cmd");
- if (cmd != nullptr) {
- if (strcmp(cmd, "init") == 0) {
- assert(!needs_response);
- // The only purpose of the "init" command is to send us a unique
- // session ID, which in fact we don't do much with.
- xcommand->Attribute("session_id", &_session_id);
- // We do use it to initiate our object id sequence with a number at
- // least a little bit distinct from other sessions, though. No
- // technical requirement that we do this, but it does make debugging
- // the logs a bit easier.
- _next_sent_id = _session_id * 1000;
- PyObject *obj = PyObject_CallMethod(_runner, (char*)"setSessionId", (char *)"i", _session_id);
- Py_XDECREF(obj);
- } else if (strcmp(cmd, "start_instance") == 0) {
- assert(!needs_response);
- TiXmlElement *xinstance = xcommand->FirstChildElement("instance");
- if (xinstance != nullptr) {
- P3DCInstance *inst = new P3DCInstance(xinstance);
- start_instance(inst, xinstance);
- }
- } else if (strcmp(cmd, "terminate_instance") == 0) {
- assert(!needs_response);
- int instance_id;
- if (xcommand->QueryIntAttribute("instance_id", &instance_id) == TIXML_SUCCESS) {
- terminate_instance(instance_id);
- }
- } else if (strcmp(cmd, "setup_window") == 0) {
- assert(!needs_response);
- int instance_id;
- TiXmlElement *xwparams = xcommand->FirstChildElement("wparams");
- if (xwparams != nullptr &&
- xcommand->QueryIntAttribute("instance_id", &instance_id) == TIXML_SUCCESS) {
- setup_window(instance_id, xwparams);
- }
- } else if (strcmp(cmd, "windows_message") == 0) {
- assert(!needs_response);
- // This is a special message that we use to proxy keyboard events from
- // the parent process down into Panda, a necessary hack on Vista.
- int instance_id = 0, msg = 0, wparam = 0, lparam = 0;
- xcommand->Attribute("instance_id", &instance_id);
- xcommand->Attribute("msg", &msg);
- xcommand->Attribute("wparam", &wparam);
- xcommand->Attribute("lparam", &lparam);
- send_windows_message(instance_id, msg, wparam, lparam);
- } else if (strcmp(cmd, "exit") == 0) {
- assert(!needs_response);
- terminate_session();
- } else if (strcmp(cmd, "pyobj") == 0) {
- // Manipulate or query a python object.
- handle_pyobj_command(xcommand, needs_response, want_response_id);
- } else if (strcmp(cmd, "script_response") == 0) {
- // Response from a script request. In this case, we just store it
- // away instead of processing it immediately.
- MutexHolder holder(_responses_lock);
- _responses.push_back(doc);
- // And now we must return out, instead of deleting the document at the
- // bottom of this method.
- return;
- } else if (strcmp(cmd, "drop_pyobj") == 0) {
- int object_id;
- if (xcommand->QueryIntAttribute("object_id", &object_id) == TIXML_SUCCESS) {
- SentObjects::iterator si = _sent_objects.find(object_id);
- if (si != _sent_objects.end()) {
- PyObject *obj = (*si).second;
- Py_DECREF(obj);
- _sent_objects.erase(si);
- }
- }
- } else {
- nout << "Unhandled command " << cmd << "\n";
- if (needs_response) {
- // Better send a response.
- TiXmlDocument doc;
- TiXmlElement *xresponse = new TiXmlElement("response");
- xresponse->SetAttribute("response_id", want_response_id);
- doc.LinkEndChild(xresponse);
- write_xml(_pipe_write, &doc, nout);
- }
- }
- }
- }
- delete doc;
- }
- /**
- * Handles the pyobj command, which queries or modifies a Python object from
- * the browser scripts.
- */
- void P3DPythonRun::
- handle_pyobj_command(TiXmlElement *xcommand, bool needs_response,
- int want_response_id) {
- TiXmlDocument doc;
- TiXmlElement *xresponse = new TiXmlElement("response");
- xresponse->SetAttribute("response_id", want_response_id);
- doc.LinkEndChild(xresponse);
- const char *op = xcommand->Attribute("op");
- if (op != nullptr && !PyErr_Occurred()) {
- if (strcmp(op, "get_panda_script_object") == 0) {
- // Get Panda's toplevel Python object.
- PyObject *obj = PyObject_CallMethod(_runner, (char*)"getPandaScriptObject", (char *)"");
- if (obj != nullptr) {
- xresponse->LinkEndChild(pyobj_to_xml(obj));
- Py_DECREF(obj);
- }
- } else if (strcmp(op, "set_browser_script_object") == 0) {
- // Set the Browser's toplevel window object.
- PyObject *obj;
- TiXmlElement *xvalue = xcommand->FirstChildElement("value");
- if (xvalue != nullptr) {
- obj = xml_to_pyobj(xvalue);
- } else {
- obj = Py_None;
- Py_INCREF(obj);
- }
- Py_XDECREF(PyObject_CallMethod
- (_runner, (char *)"setBrowserScriptObject", (char *)"N", obj));
- } else if (strcmp(op, "call") == 0) {
- // Call the named method on the indicated object, or the object itself
- // if method_name isn't given.
- TiXmlElement *xobject = xcommand->FirstChildElement("object");
- if (xobject != nullptr) {
- PyObject *obj = xml_to_pyobj(xobject);
- const char *method_name = xcommand->Attribute("method_name");
- // Build up a list of params.
- PyObject *list = PyList_New(0);
- TiXmlElement *xchild = xcommand->FirstChildElement("value");
- while (xchild != nullptr) {
- PyObject *child = xml_to_pyobj(xchild);
- PyList_Append(list, child);
- Py_DECREF(child);
- xchild = xchild->NextSiblingElement("value");
- }
- // Convert the list to a tuple for the call.
- PyObject *params = PyList_AsTuple(list);
- Py_DECREF(list);
- // Now call the method.
- PyObject *result = nullptr;
- if (method_name == nullptr) {
- // No method name; call the object directly.
- result = PyObject_CallObject(obj, params);
- // Several special-case "method" names.
- } else if (strcmp(method_name, "__bool__") == 0) {
- result = PyBool_FromLong(PyObject_IsTrue(obj));
- } else if (strcmp(method_name, "__int__") == 0) {
- #if PY_MAJOR_VERSION >= 3
- result = PyNumber_Long(obj);
- #else
- result = PyNumber_Int(obj);
- #endif
- } else if (strcmp(method_name, "__float__") == 0) {
- result = PyNumber_Float(obj);
- } else if (strcmp(method_name, "__repr__") == 0) {
- result = PyObject_Repr(obj);
- } else if (strcmp(method_name, "__str__") == 0 ||
- strcmp(method_name, "toString") == 0) {
- result = PyObject_Str(obj);
- } else if (strcmp(method_name, "__set_property__") == 0) {
- // We call these methods __set_property__ et al instead of
- // __setattr__ et al, because they do not precisely duplicate the
- // Python semantics.
- char *property_name;
- PyObject *value;
- if (PyArg_ParseTuple(params, "sO", &property_name, &value)) {
- bool success = false;
- // If it already exists as an attribute, update it there.
- if (PyObject_HasAttrString(obj, property_name)) {
- if (PyObject_SetAttrString(obj, property_name, value) != -1) {
- success = true;
- } else {
- PyErr_Clear();
- }
- }
- // If the object supports the mapping protocol, store it in the
- // object's dictionary.
- if (!success && PyMapping_Check(obj)) {
- if (PyMapping_SetItemString(obj, property_name, value) != -1) {
- success = true;
- } else {
- PyErr_Clear();
- }
- }
- // Finally, try to store it on the object.
- if (!success) {
- if (PyObject_SetAttrString(obj, property_name, value) != -1) {
- success = true;
- } else {
- PyErr_Clear();
- }
- }
- if (success) {
- result = Py_True;
- } else {
- result = Py_False;
- }
- Py_INCREF(result);
- }
- } else if (strcmp(method_name, "__del_property__") == 0) {
- char *property_name;
- if (PyArg_ParseTuple(params, "s", &property_name)) {
- bool success = false;
- if (PyObject_HasAttrString(obj, property_name)) {
- if (PyObject_DelAttrString(obj, property_name) != -1) {
- success = true;
- } else {
- PyErr_Clear();
- }
- }
- if (!success) {
- if (PyObject_DelItemString(obj, property_name) != -1) {
- success = true;
- } else {
- PyErr_Clear();
- }
- }
- if (success) {
- result = Py_True;
- } else {
- result = Py_False;
- }
- Py_INCREF(result);
- }
- } else if (strcmp(method_name, "__get_property__") == 0) {
- char *property_name;
- if (PyArg_ParseTuple(params, "s", &property_name)) {
- bool success = false;
- if (PyObject_HasAttrString(obj, property_name)) {
- result = PyObject_GetAttrString(obj, property_name);
- if (result != nullptr) {
- success = true;
- } else {
- PyErr_Clear();
- }
- }
- if (!success) {
- if (PyMapping_HasKeyString(obj, property_name)) {
- result = PyMapping_GetItemString(obj, property_name);
- if (result != nullptr) {
- success = true;
- } else {
- PyErr_Clear();
- }
- }
- }
- if (!success) {
- result = nullptr;
- }
- }
- } else if (strcmp(method_name, "__has_method__") == 0) {
- char *property_name;
- result = Py_False;
- if (PyArg_ParseTuple(params, "s", &property_name)) {
- if (*property_name) {
- // Check for a callable method
- if (PyObject_HasAttrString(obj, property_name)) {
- PyObject *prop = PyObject_GetAttrString(obj, property_name);
- if (PyCallable_Check(prop)) {
- result = Py_True;
- }
- Py_DECREF(prop);
- }
- } else {
- // Check for the default method
- if (PyCallable_Check(obj)) {
- result = Py_True;
- }
- }
- }
- Py_INCREF(result);
- } else {
- // Not a special-case name. Call the named method.
- PyObject *method = PyObject_GetAttrString(obj, (char *)method_name);
- if (method != nullptr) {
- result = PyObject_CallObject(method, params);
- Py_DECREF(method);
- }
- }
- Py_DECREF(params);
- // Feed the return value back through the XML pipe to the caller.
- if (result != nullptr) {
- xresponse->LinkEndChild(pyobj_to_xml(result));
- Py_DECREF(result);
- } else {
- // Print the Python error message if there was one.
- PyErr_Print();
- }
- Py_DECREF(obj);
- }
- }
- }
- if (needs_response) {
- write_xml(_pipe_write, &doc, nout);
- }
- }
- /**
- * This method is added to the task manager (via st_check_comm, below) so that
- * it gets a call every frame. Its job is to check for commands received from
- * the plugin host in the parent process.
- */
- void P3DPythonRun::
- check_comm() {
- // nout << ":";
- ACQUIRE_LOCK(_commands_lock);
- while (!_commands.empty()) {
- TiXmlDocument *doc = _commands.front();
- _commands.pop_front();
- RELEASE_LOCK(_commands_lock);
- handle_command(doc);
- if (_session_terminated) {
- return;
- }
- ACQUIRE_LOCK(_commands_lock);
- }
- RELEASE_LOCK(_commands_lock);
- if (!_program_continue) {
- // The low-level thread detected an error, for instance pipe closed. We
- // should exit gracefully.
- terminate_session();
- return;
- }
- // Sleep to yield the timeslice, but only if we're not running in the main
- // thread.
- if (Thread::get_current_thread() != Thread::get_main_thread()) {
- Thread::sleep(0.001);
- }
- }
- /**
- * This is a static Python wrapper around py_check_comm, needed to add the
- * function to a PythonTask.
- */
- PyObject *P3DPythonRun::
- st_check_comm(PyObject *, PyObject *args) {
- P3DPythonRun::_global_ptr->check_comm();
- return Py_BuildValue("i", AsyncTask::DS_cont);
- }
- /**
- * This method is similar to check_comm(), above, but instead of handling all
- * events, it waits for a specific script_response ID to come back from the
- * browser, and leaves all other events in the queue.
- */
- TiXmlDocument *P3DPythonRun::
- wait_script_response(int response_id) {
- // nout << "waiting script_response " << response_id << "\n";
- while (true) {
- Commands::iterator ci;
- // First, walk through the _commands queue to see if there's anything that
- // needs immediate processing.
- ACQUIRE_LOCK(_commands_lock);
- for (ci = _commands.begin(); ci != _commands.end(); ++ci) {
- TiXmlDocument *doc = (*ci);
- TiXmlElement *xcommand = doc->FirstChildElement("command");
- if (xcommand != nullptr) {
- const char *cmd = xcommand->Attribute("cmd");
- if ((cmd != nullptr && strcmp(cmd, "script_response") == 0) ||
- xcommand->Attribute("want_response_id") != nullptr) {
- // This is either a response, or it's a command that will want a
- // response itself. In either case we should handle it right away.
- // ("handling" a response means moving it to the _responses queue.)
- _commands.erase(ci);
- RELEASE_LOCK(_commands_lock);
- handle_command(doc);
- if (_session_terminated) {
- return nullptr;
- }
- ACQUIRE_LOCK(_commands_lock);
- break;
- }
- }
- }
- RELEASE_LOCK(_commands_lock);
- // Now, walk through the _responses queue to look for the particular
- // response we're waiting for.
- _responses_lock.acquire();
- for (ci = _responses.begin(); ci != _responses.end(); ++ci) {
- TiXmlDocument *doc = (*ci);
- TiXmlElement *xcommand = doc->FirstChildElement("command");
- assert(xcommand != nullptr);
- const char *cmd = xcommand->Attribute("cmd");
- assert(cmd != nullptr && strcmp(cmd, "script_response") == 0);
- int unique_id;
- if (xcommand->QueryIntAttribute("unique_id", &unique_id) == TIXML_SUCCESS) {
- if (unique_id == response_id) {
- // This is the response we were waiting for.
- _responses.erase(ci);
- _responses_lock.release();
- // nout << "got script_response " << unique_id << "\n";
- return doc;
- }
- }
- }
- _responses_lock.release();
- if (!_program_continue) {
- terminate_session();
- return nullptr;
- }
- #ifdef _WIN32
- // Make sure we process the Windows event loop while we're waiting, or
- // everything that depends on Windows messages will starve.
- // We appear to be best off with just a single PeekMessage() call here;
- // the full message pump seems to cause problems.
- MSG msg;
- PeekMessage(&msg, nullptr, 0, 0, PM_NOREMOVE | PM_NOYIELD);
- #endif // _WIN32
- // nout << ".";
- // It hasn't shown up yet. Give the sub-thread a chance to process the
- // input and append it to the queue.
- Thread::force_yield();
- }
- assert(false);
- }
- /**
- * This method is a special Python function that is added as a callback to the
- * AppRunner class, to allow Python to upcall into this object.
- */
- PyObject *P3DPythonRun::
- py_request_func(PyObject *args) {
- int instance_id;
- const char *request_type;
- PyObject *extra_args;
- if (!PyArg_ParseTuple(args, "isO", &instance_id, &request_type, &extra_args)) {
- return nullptr;
- }
- if (strcmp(request_type, "wait_script_response") == 0) {
- // This is a special case. Instead of generating a new request, this
- // means to wait for a particular script_response to come in on the wire.
- int response_id;
- if (!PyArg_ParseTuple(extra_args, "i", &response_id)) {
- return nullptr;
- }
- TiXmlDocument *doc = wait_script_response(response_id);
- if (_session_terminated) {
- return Py_BuildValue("");
- }
- assert(doc != nullptr);
- TiXmlElement *xcommand = doc->FirstChildElement("command");
- assert(xcommand != nullptr);
- TiXmlElement *xvalue = xcommand->FirstChildElement("value");
- PyObject *value = nullptr;
- if (xvalue != nullptr) {
- value = xml_to_pyobj(xvalue);
- } else {
- // An absence of a <value> element is an exception. We will return NULL
- // from this function, but first set the error condition.
- PyErr_SetString(PyExc_EnvironmentError, "Error on script call");
- }
- delete doc;
- return value;
- }
- TiXmlDocument doc;
- TiXmlElement *xrequest = new TiXmlElement("request");
- xrequest->SetAttribute("instance_id", instance_id);
- xrequest->SetAttribute("rtype", request_type);
- doc.LinkEndChild(xrequest);
- if (strcmp(request_type, "notify") == 0) {
- // A general notification to be sent directly to the instance.
- const char *message;
- if (!PyArg_ParseTuple(extra_args, "s", &message)) {
- return nullptr;
- }
- xrequest->SetAttribute("message", message);
- write_xml(_pipe_write, &doc, nout);
- } else if (strcmp(request_type, "script") == 0) {
- // Meddling with a scripting variable on the browser side.
- const char *operation;
- PyObject *object;
- const char *property_name;
- PyObject *value;
- int needs_response;
- int unique_id;
- if (!PyArg_ParseTuple(extra_args, "sOsOii",
- &operation, &object, &property_name, &value,
- &needs_response, &unique_id)) {
- return nullptr;
- }
- xrequest->SetAttribute("operation", operation);
- xrequest->SetAttribute("property_name", property_name);
- xrequest->SetAttribute("needs_response", (int)(needs_response != 0));
- xrequest->SetAttribute("unique_id", unique_id);
- TiXmlElement *xobject = pyobj_to_xml(object);
- xobject->SetValue("object");
- xrequest->LinkEndChild(xobject);
- TiXmlElement *xvalue = pyobj_to_xml(value);
- xrequest->LinkEndChild(xvalue);
- write_xml(_pipe_write, &doc, nout);
- } else if (strcmp(request_type, "drop_p3dobj") == 0) {
- // Release a particular P3D_object that we were holding a reference to.
- int object_id;
- if (!PyArg_ParseTuple(extra_args, "i", &object_id)) {
- return nullptr;
- }
- xrequest->SetAttribute("object_id", object_id);
- write_xml(_pipe_write, &doc, nout);
- } else if (strcmp(request_type, "forget_package") == 0) {
- // A request to the instance to drop a particular package (or host) from
- // the cache.
- const char *host_url;
- const char *package_name;
- const char *package_version;
- if (!PyArg_ParseTuple(extra_args, "sss", &host_url, &package_name, &package_version)) {
- return nullptr;
- }
- xrequest->SetAttribute("host_url", host_url);
- if (*package_name) {
- xrequest->SetAttribute("package_name", package_name);
- if (*package_version) {
- xrequest->SetAttribute("package_version", package_version);
- }
- }
- write_xml(_pipe_write, &doc, nout);
- } else {
- string message = string("Unsupported request type: ") + string(request_type);
- PyErr_SetString(PyExc_ValueError, message.c_str());
- return nullptr;
- }
- return Py_BuildValue("");
- }
- /**
- * This is the static wrapper around py_request_func.
- */
- PyObject *P3DPythonRun::
- st_request_func(PyObject *, PyObject *args) {
- return P3DPythonRun::_global_ptr->py_request_func(args);
- }
- /**
- * Starts the read thread. This thread is responsible for reading the
- * standard input socket for XML commands and storing them in the _commands
- * queue.
- */
- void P3DPythonRun::
- spawn_read_thread() {
- assert(!_read_thread_continue);
- // We have to use direct OS calls to create the thread instead of Panda
- // constructs, because it has to be an actual thread, not necessarily a
- // Panda thread (we can't use Panda's simple threads implementation, because
- // we can't get overlapped IO on an anonymous pipe in Windows).
- _read_thread_continue = true;
- SPAWN_THREAD(_read_thread, rt_thread_run, this);
- }
- /**
- * Waits for the read thread to stop.
- */
- void P3DPythonRun::
- join_read_thread() {
- _read_thread_continue = false;
- JOIN_THREAD(_read_thread);
- }
- /**
- * Starts the indicated instance running within the Python process.
- */
- void P3DPythonRun::
- start_instance(P3DCInstance *inst, TiXmlElement *xinstance) {
- _instances[inst->get_instance_id()] = inst;
- set_instance_info(inst, xinstance);
- TiXmlElement *xpackage = xinstance->FirstChildElement("package");
- while (xpackage != nullptr) {
- add_package_info(inst, xpackage);
- xpackage = xpackage->NextSiblingElement("package");
- }
- TiXmlElement *xfparams = xinstance->FirstChildElement("fparams");
- if (xfparams != nullptr) {
- set_p3d_filename(inst, xfparams);
- }
- TiXmlElement *xwparams = xinstance->FirstChildElement("wparams");
- if (xwparams != nullptr) {
- setup_window(inst, xwparams);
- }
- }
- /**
- * Stops the instance with the indicated id.
- */
- void P3DPythonRun::
- terminate_instance(int id) {
- Instances::iterator ii = _instances.find(id);
- if (ii == _instances.end()) {
- nout << "Can't stop instance " << id << ": not started.\n";
- return;
- }
- P3DCInstance *inst = (*ii).second;
- _instances.erase(ii);
- delete inst;
- // TODO: we don't currently have any way to stop just one instance of a
- // multi-instance session. This will require a different Python interface
- // than ShowBase.
- terminate_session();
- }
- /**
- * Sets some global information about the instance.
- */
- void P3DPythonRun::
- set_instance_info(P3DCInstance *inst, TiXmlElement *xinstance) {
- const char *root_dir = xinstance->Attribute("root_dir");
- if (root_dir == nullptr) {
- root_dir = "";
- }
- const char *log_directory = xinstance->Attribute("log_directory");
- if (log_directory == nullptr) {
- log_directory = "";
- }
- int verify_contents = 0;
- xinstance->Attribute("verify_contents", &verify_contents);
- const char *super_mirror = xinstance->Attribute("super_mirror");
- if (super_mirror == nullptr) {
- super_mirror = "";
- }
- // Get the initial "main" object, if specified.
- PyObject *main;
- TiXmlElement *xmain = xinstance->FirstChildElement("main");
- if (xmain != nullptr) {
- main = xml_to_pyobj(xmain);
- } else {
- main = Py_None;
- Py_INCREF(main);
- }
- int respect_per_platform = 0;
- xinstance->Attribute("respect_per_platform", &respect_per_platform);
- PyObject *result = PyObject_CallMethod
- (_runner, (char *)"setInstanceInfo", (char *)"sssiNi", root_dir,
- log_directory, super_mirror, verify_contents, main, respect_per_platform);
- if (result == nullptr) {
- PyErr_Print();
- if (_interactive_console) {
- run_interactive_console();
- }
- exit(1);
- }
- Py_XDECREF(result);
- }
- /**
- * Adds some information about a pre-loaded package.
- */
- void P3DPythonRun::
- add_package_info(P3DCInstance *inst, TiXmlElement *xpackage) {
- const char *name = xpackage->Attribute("name");
- const char *platform = xpackage->Attribute("platform");
- const char *version = xpackage->Attribute("version");
- const char *host = xpackage->Attribute("host");
- const char *host_dir = xpackage->Attribute("host_dir");
- if (name == nullptr || host == nullptr) {
- return;
- }
- if (version == nullptr) {
- version = "";
- }
- if (platform == nullptr) {
- platform = "";
- }
- if (host_dir == nullptr) {
- host_dir = "";
- }
- PyObject *result = PyObject_CallMethod
- (_runner, (char *)"addPackageInfo", (char *)"sssss",
- name, platform, version, host, host_dir);
- if (result == nullptr) {
- PyErr_Print();
- if (_interactive_console) {
- run_interactive_console();
- }
- exit(1);
- }
- Py_XDECREF(result);
- }
- /**
- * Sets the startup filename and tokens for the indicated instance.
- */
- void P3DPythonRun::
- set_p3d_filename(P3DCInstance *inst, TiXmlElement *xfparams) {
- string p3d_filename;
- const char *p3d_filename_c = xfparams->Attribute("p3d_filename");
- if (p3d_filename_c != nullptr) {
- p3d_filename = p3d_filename_c;
- }
- int p3d_offset = 0;
- xfparams->Attribute("p3d_offset", &p3d_offset);
- string p3d_url;
- const char *p3d_url_c = xfparams->Attribute("p3d_url");
- if (p3d_url_c != nullptr) {
- p3d_url = p3d_url_c;
- }
- PyObject *token_list = PyList_New(0);
- TiXmlElement *xtoken = xfparams->FirstChildElement("token");
- while (xtoken != nullptr) {
- string keyword, value;
- const char *keyword_c = xtoken->Attribute("keyword");
- if (keyword_c != nullptr) {
- keyword = keyword_c;
- }
- const char *value_c = xtoken->Attribute("value");
- if (value_c != nullptr) {
- value = value_c;
- }
- PyObject *tuple = Py_BuildValue("(ss)", keyword.c_str(),
- value.c_str());
- PyList_Append(token_list, tuple);
- Py_DECREF(tuple);
- xtoken = xtoken->NextSiblingElement("token");
- }
- PyObject *arg_list = PyList_New(0);
- TiXmlElement *xarg = xfparams->FirstChildElement("arg");
- while (xarg != nullptr) {
- string value;
- const char *value_c = xarg->Attribute("value");
- if (value_c != nullptr) {
- value = value_c;
- }
- PyObject *str = Py_BuildValue("s", value.c_str());
- PyList_Append(arg_list, str);
- Py_DECREF(str);
- xarg = xarg->NextSiblingElement("arg");
- }
- PyObject *result = PyObject_CallMethod
- (_runner, (char *)"setP3DFilename", (char *)"sNNiiis", p3d_filename.c_str(),
- token_list, arg_list, inst->get_instance_id(), _interactive_console, p3d_offset,
- p3d_url.c_str());
- if (result == nullptr) {
- PyErr_Print();
- if (_interactive_console) {
- run_interactive_console();
- }
- exit(1);
- }
- Py_XDECREF(result);
- }
- /**
- * Sets the window parameters for the indicated instance.
- */
- void P3DPythonRun::
- setup_window(int id, TiXmlElement *xwparams) {
- Instances::iterator ii = _instances.find(id);
- if (ii == _instances.end()) {
- nout << "Can't setup window for " << id << ": not started.\n";
- return;
- }
- P3DCInstance *inst = (*ii).second;
- setup_window(inst, xwparams);
- }
- /**
- * Sets the window parameters for the indicated instance.
- */
- void P3DPythonRun::
- setup_window(P3DCInstance *inst, TiXmlElement *xwparams) {
- string window_type;
- const char *window_type_c = xwparams->Attribute("window_type");
- if (window_type_c != nullptr) {
- window_type = window_type_c;
- }
- int win_x, win_y, win_width, win_height;
- xwparams->Attribute("win_x", &win_x);
- xwparams->Attribute("win_y", &win_y);
- xwparams->Attribute("win_width", &win_width);
- xwparams->Attribute("win_height", &win_height);
- PT(WindowHandle) parent_window_handle;
- #ifdef _WIN32
- int hwnd;
- if (xwparams->Attribute("parent_hwnd", &hwnd)) {
- parent_window_handle = NativeWindowHandle::make_win((HWND)hwnd);
- }
- #elif __APPLE__
- // On Mac, we don't parent windows directly to the browser; instead, we have
- // to go through this subprocess-window nonsense.
- const char *subprocess_window = xwparams->Attribute("subprocess_window");
- if (subprocess_window != nullptr) {
- Filename filename = Filename::from_os_specific(subprocess_window);
- parent_window_handle = NativeWindowHandle::make_subprocess(filename);
- }
- #elif defined(HAVE_X11)
- // Use stringstream to decode the "long" attribute.
- const char *parent_cstr = xwparams->Attribute("parent_xwindow");
- if (parent_cstr != nullptr) {
- long window;
- istringstream strm(parent_cstr);
- strm >> window;
- parent_window_handle = NativeWindowHandle::make_x11((X11_Window)window);
- }
- #endif
- PyObject *py_handle = Py_None;
- if (parent_window_handle != nullptr) {
- // We have a valid parent WindowHandle, but replace it with a
- // P3DWindowHandle so we can get the callbacks.
- parent_window_handle = new P3DWindowHandle(this, inst, *parent_window_handle);
- inst->_parent_window_handle = parent_window_handle;
- // Also pass this P3DWindowHandle object down into Panda, via the
- // setupWindow() call. For this, we need to create a Python wrapper
- // objcet.
- parent_window_handle->ref();
- py_handle = DTool_CreatePyInstanceTyped(parent_window_handle.p(), true);
- }
- Py_INCREF(py_handle);
- // TODO: direct this into the particular instance. This will require a
- // specialized ShowBase replacement.
- PyObject *result = PyObject_CallMethod
- (_runner, (char *)"setupWindow", (char *)"siiiiN", window_type.c_str(),
- win_x, win_y, win_width, win_height, py_handle);
- if (result == nullptr) {
- PyErr_Print();
- if (_interactive_console) {
- run_interactive_console();
- }
- exit(1);
- }
- Py_XDECREF(result);
- }
- /**
- * This is used to deliver a windows keyboard message to the Panda process
- * from the parent process, a necessary hack on Vista.
- */
- void P3DPythonRun::
- send_windows_message(int id, unsigned int msg, int wparam, int lparam) {
- Instances::iterator ii = _instances.find(id);
- if (ii == _instances.end()) {
- return;
- }
- P3DCInstance *inst = (*ii).second;
- if (inst->_parent_window_handle != nullptr) {
- inst->_parent_window_handle->send_windows_message(msg, wparam, lparam);
- }
- }
- /**
- * Stops all currently-running instances.
- */
- void P3DPythonRun::
- terminate_session() {
- Instances::iterator ii;
- for (ii = _instances.begin(); ii != _instances.end(); ++ii) {
- P3DCInstance *inst = (*ii).second;
- delete inst;
- }
- _instances.clear();
- if (!_session_terminated) {
- if (_taskMgr != nullptr) {
- PyObject *result = PyObject_CallMethod(_taskMgr, (char *)"stop", (char *)"");
- if (result == nullptr) {
- PyErr_Print();
- } else {
- Py_DECREF(result);
- }
- }
- _session_terminated = true;
- }
- }
- /**
- * Converts the indicated PyObject to the appropriate XML representation of a
- * P3D_value type, and returns a freshly-allocated TiXmlElement.
- */
- TiXmlElement *P3DPythonRun::
- pyobj_to_xml(PyObject *value) {
- TiXmlElement *xvalue = new TiXmlElement("value");
- if (value == Py_None) {
- // None.
- xvalue->SetAttribute("type", "none");
- } else if (PyBool_Check(value)) {
- // A bool value.
- xvalue->SetAttribute("type", "bool");
- xvalue->SetAttribute("value", PyObject_IsTrue(value));
- #if PY_MAJOR_VERSION < 3
- } else if (PyInt_Check(value)) {
- // A plain integer value.
- xvalue->SetAttribute("type", "int");
- xvalue->SetAttribute("value", PyInt_AsLong(value));
- #endif
- } else if (PyLong_Check(value)) {
- // A long integer value. This gets converted either as an integer or as a
- // floating-point type, whichever fits.
- long lvalue = PyLong_AsLong(value);
- if (PyErr_Occurred()) {
- // It won't fit as an integer; make it a double.
- PyErr_Clear();
- xvalue->SetAttribute("type", "float");
- xvalue->SetDoubleAttribute("value", PyLong_AsDouble(value));
- } else {
- // It fits as an integer.
- xvalue->SetAttribute("type", "int");
- xvalue->SetAttribute("value", lvalue);
- }
- } else if (PyFloat_Check(value)) {
- // A floating-point value.
- xvalue->SetAttribute("type", "float");
- xvalue->SetDoubleAttribute("value", PyFloat_AsDouble(value));
- } else if (PyUnicode_Check(value)) {
- // A unicode value. Convert to utf-8 for the XML encoding.
- xvalue->SetAttribute("type", "string");
- #if PY_MAJOR_VERSION >= 3
- // In Python 3, there are only unicode strings, and there is a handy
- // function for getting the UTF-8 encoded version.
- Py_ssize_t length = 0;
- char *buffer = PyUnicode_AsUTF8AndSize(value, &length);
- if (buffer != nullptr) {
- string str(buffer, length);
- xvalue->SetAttribute("value", str);
- }
- #else
- PyObject *as_str = PyUnicode_AsUTF8String(value);
- if (as_str != nullptr) {
- char *buffer;
- Py_ssize_t length;
- if (PyString_AsStringAndSize(as_str, &buffer, &length) != -1) {
- string str(buffer, length);
- xvalue->SetAttribute("value", str);
- }
- Py_DECREF(as_str);
- }
- } else if (PyString_Check(value)) {
- // A string value. Insist that it is utf-8 encoded, by decoding it first
- // using the standard encoding, then re-encoding it.
- xvalue->SetAttribute("type", "string");
- PyObject *ustr = PyUnicode_FromEncodedObject(value, nullptr, nullptr);
- if (ustr == nullptr) {
- PyErr_Print();
- return xvalue;
- } else {
- PyObject *as_str = PyUnicode_AsUTF8String(ustr);
- if (as_str != nullptr) {
- char *buffer;
- Py_ssize_t length;
- if (PyString_AsStringAndSize(as_str, &buffer, &length) != -1) {
- string str(buffer, length);
- xvalue->SetAttribute("value", str);
- }
- Py_DECREF(as_str);
- }
- Py_DECREF(ustr);
- }
- #endif // PY_MAJOR_VERSION
- } else if (PyTuple_Check(value)) {
- // An immutable sequence. Pass it as a concrete.
- xvalue->SetAttribute("type", "concrete_sequence");
- Py_ssize_t length = PySequence_Length(value);
- for (Py_ssize_t i = 0; i < length; ++i) {
- PyObject *item = PySequence_GetItem(value, i);
- if (item != nullptr) {
- xvalue->LinkEndChild(pyobj_to_xml(item));
- Py_DECREF(item);
- }
- }
- } else if (PyObject_IsInstance(value, _concrete_struct_class)) {
- // This is a ConcreteStruct.
- xvalue->SetAttribute("type", "concrete_struct");
- PyObject *items = PyObject_CallMethod(value, (char *)"getConcreteProperties", (char *)"");
- if (items == nullptr) {
- PyErr_Print();
- return xvalue;
- }
- Py_ssize_t length = PySequence_Length(items);
- for (Py_ssize_t i = 0; i < length; ++i) {
- PyObject *item = PySequence_GetItem(items, i);
- if (item != nullptr) {
- PyObject *a = PySequence_GetItem(item, 0);
- if (a != nullptr) {
- PyObject *b = PySequence_GetItem(item, 1);
- if (b != nullptr) {
- TiXmlElement *xitem = pyobj_to_xml(b);
- Py_DECREF(b);
- PyObject *as_str;
- if (PyUnicode_Check(a)) {
- // The key is a unicode value.
- #if PY_MAJOR_VERSION >= 3
- as_str = a;
- #else
- as_str = PyUnicode_AsUTF8String(a);
- #endif
- } else {
- // The key is a string value or something else. Make it a
- // string.
- as_str = PyObject_Str(a);
- }
- char *buffer;
- Py_ssize_t length;
- #if PY_MAJOR_VERSION >= 3
- buffer = PyUnicode_AsUTF8AndSize(as_str, &length);
- if (buffer != nullptr) {
- #else
- if (PyString_AsStringAndSize(as_str, &buffer, &length) != -1) {
- #endif
- string str(buffer, length);
- xitem->SetAttribute("key", str);
- }
- Py_DECREF(as_str);
- xvalue->LinkEndChild(xitem);
- }
- Py_DECREF(a);
- }
- Py_DECREF(item);
- }
- }
- // We've already avoided errors in the above code; clear the error flag.
- PyErr_Clear();
- Py_DECREF(items);
- } else if (PyObject_IsInstance(value, _undefined_object_class)) {
- // This is an UndefinedObject.
- xvalue->SetAttribute("type", "undefined");
- } else if (PyObject_IsInstance(value, _browser_object_class)) {
- // This is a BrowserObject, a reference to an object that actually exists
- // in the host namespace. So, pass up the appropriate object ID.
- PyObject *objectId = PyObject_GetAttrString(value, (char *)"_BrowserObject__objectId");
- if (objectId != nullptr) {
- int object_id = PyInt_AsLong(objectId);
- xvalue->SetAttribute("type", "browser");
- xvalue->SetAttribute("object_id", object_id);
- Py_DECREF(objectId);
- }
- } else {
- // Some other kind of object. Make it a generic Python object. This is
- // more expensive for the caller to deal with--it requires a back-and-
- // forth across the XML pipe--but it's much more general.
- int object_id = _next_sent_id;
- ++_next_sent_id;
- bool inserted = _sent_objects.insert(SentObjects::value_type(object_id, value)).second;
- while (!inserted) {
- // Hmm, we must have cycled around the entire int space? Either that,
- // or there's a logic bug somewhere. Assume the former, and keep
- // looking for an empty slot.
- object_id = _next_sent_id;
- ++_next_sent_id;
- inserted = _sent_objects.insert(SentObjects::value_type(object_id, value)).second;
- }
- // Now that it's stored in the map, increment its reference count.
- Py_INCREF(value);
- xvalue->SetAttribute("type", "python");
- xvalue->SetAttribute("object_id", object_id);
- }
- return xvalue;
- }
- /**
- * Converts the XML representation of a P3D_value type into the equivalent
- * Python object and returns it.
- */
- PyObject *P3DPythonRun::
- xml_to_pyobj(TiXmlElement *xvalue) {
- const char *type = xvalue->Attribute("type");
- if (strcmp(type, "none") == 0) {
- return Py_BuildValue("");
- } else if (strcmp(type, "bool") == 0) {
- int value;
- if (xvalue->QueryIntAttribute("value", &value) == TIXML_SUCCESS) {
- return PyBool_FromLong(value);
- }
- } else if (strcmp(type, "int") == 0) {
- int value;
- if (xvalue->QueryIntAttribute("value", &value) == TIXML_SUCCESS) {
- #if PY_MAJOR_VERSION >= 3
- return PyLong_FromLong(value);
- #else
- return PyInt_FromLong(value);
- #endif
- }
- } else if (strcmp(type, "float") == 0) {
- double value;
- if (xvalue->QueryDoubleAttribute("value", &value) == TIXML_SUCCESS) {
- return PyFloat_FromDouble(value);
- }
- } else if (strcmp(type, "string") == 0) {
- // Using the string form here instead of the char * form, so we don't get
- // tripped up on embedded null characters.
- const string *value = xvalue->Attribute(string("value"));
- if (value != nullptr) {
- return PyUnicode_DecodeUTF8(value->data(), value->length(), nullptr);
- }
- } else if (strcmp(type, "undefined") == 0) {
- Py_INCREF(_undefined);
- return _undefined;
- } else if (strcmp(type, "browser") == 0) {
- int object_id;
- if (xvalue->QueryIntAttribute("object_id", &object_id) == TIXML_SUCCESS) {
- // Construct a new BrowserObject wrapper around this object.
- return PyObject_CallFunction(_browser_object_class, (char *)"Oi",
- _runner, object_id);
- }
- } else if (strcmp(type, "concrete_sequence") == 0) {
- // Receive a concrete sequence as a tuple.
- PyObject *list = PyList_New(0);
- TiXmlElement *xitem = xvalue->FirstChildElement("value");
- while (xitem != nullptr) {
- PyObject *item = xml_to_pyobj(xitem);
- if (item != nullptr) {
- PyList_Append(list, item);
- Py_DECREF(item);
- }
- xitem = xitem->NextSiblingElement("value");
- }
- PyObject *result = PyList_AsTuple(list);
- Py_DECREF(list);
- return result;
- } else if (strcmp(type, "concrete_struct") == 0) {
- // Receive a concrete struct as a new ConcreteStruct instance.
- PyObject *obj = PyObject_CallFunction(_concrete_struct_class, (char *)"");
- if (obj != nullptr) {
- TiXmlElement *xitem = xvalue->FirstChildElement("value");
- while (xitem != nullptr) {
- const char *key = xitem->Attribute("key");
- if (key != nullptr) {
- PyObject *item = xml_to_pyobj(xitem);
- if (item != nullptr) {
- PyObject_SetAttrString(obj, (char *)key, item);
- Py_DECREF(item);
- }
- }
- xitem = xitem->NextSiblingElement("value");
- }
- return obj;
- }
- } else if (strcmp(type, "python") == 0) {
- int object_id;
- if (xvalue->QueryIntAttribute("object_id", &object_id) == TIXML_SUCCESS) {
- SentObjects::iterator si = _sent_objects.find(object_id);
- PyObject *result;
- if (si == _sent_objects.end()) {
- // Hmm, the parent process gave us a bogus object ID.
- result = _undefined;
- } else {
- result = (*si).second;
- }
- Py_INCREF(result);
- return result;
- }
- }
- // Something went wrong in decoding.
- PyObject *result = _undefined;
- Py_INCREF(result);
- return result;
- }
- /**
- * The main function for the read thread.
- */
- void P3DPythonRun::
- rt_thread_run() {
- while (_read_thread_continue) {
- TiXmlDocument *doc = read_xml(_pipe_read, nout);
- if (doc == nullptr) {
- // Some error on reading. Abort.
- _program_continue = false;
- return;
- }
- // Successfully read an XML document.
- // Check for one special case: the "exit" command means we shut down the
- // read thread along with everything else.
- TiXmlElement *xcommand = doc->FirstChildElement("command");
- if (xcommand != nullptr) {
- const char *cmd = xcommand->Attribute("cmd");
- if (cmd != nullptr) {
- if (strcmp(cmd, "exit") == 0) {
- _read_thread_continue = false;
- }
- }
- }
- // Feed the command up to the parent.
- ACQUIRE_LOCK(_commands_lock);
- _commands.push_back(doc);
- RELEASE_LOCK(_commands_lock);
- }
- }
- /**
- *
- */
- P3DPythonRun::P3DWindowHandle::
- P3DWindowHandle(P3DPythonRun *p3dpython, P3DCInstance *inst,
- const WindowHandle ©) :
- WindowHandle(copy),
- _p3dpython(p3dpython),
- _inst(inst),
- _child_count(0)
- {
- }
- /**
- * Called on a parent handle to indicate a child window's intention to attach
- * itself.
- */
- void P3DPythonRun::P3DWindowHandle::
- attach_child(WindowHandle *child) {
- WindowHandle::attach_child(child);
- ++_child_count;
- if (_child_count == 1) {
- // When the first child window is attached, we tell the plugin.
- _p3dpython->set_window_open(_inst, true);
- }
- }
- /**
- * Called on a parent handle to indicate a child window's intention to detach
- * itself.
- */
- void P3DPythonRun::P3DWindowHandle::
- detach_child(WindowHandle *child) {
- WindowHandle::detach_child(child);
- --_child_count;
- if (_child_count == 0) {
- // When the last child window is detached, we tell the plugin.
- _p3dpython->set_window_open(_inst, false);
- }
- }
- /**
- * Called on a parent handle to indicate a child window's wish to receive
- * keyboard button events.
- */
- void P3DPythonRun::P3DWindowHandle::
- request_keyboard_focus(WindowHandle *child) {
- WindowHandle::request_keyboard_focus(child);
- _p3dpython->request_keyboard_focus(_inst);
- }
|