]> git.saurik.com Git - wxWidgets.git/blame_incremental - wxPython/src/helpers.cpp
fixed memory leaks when reading invalid GIFs
[wxWidgets.git] / wxPython / src / helpers.cpp
... / ...
CommitLineData
1/////////////////////////////////////////////////////////////////////////////
2// Name: helpers.cpp
3// Purpose: Helper functions/classes for the wxPython extension module
4//
5// Author: Robin Dunn
6//
7// Created: 1-July-1997
8// RCS-ID: $Id$
9// Copyright: (c) 1998 by Total Control Software
10// Licence: wxWindows license
11/////////////////////////////////////////////////////////////////////////////
12
13
14#undef DEBUG
15#include <Python.h>
16#include "wx/wxPython/wxPython_int.h"
17#include "wx/wxPython/pyistream.h"
18#include "wx/wxPython/swigver.h"
19#include "wx/wxPython/twoitem.h"
20
21#ifdef __WXMSW__
22#include <wx/msw/private.h>
23#include <wx/msw/winundef.h>
24#include <wx/msw/msvcrt.h>
25#endif
26
27#ifdef __WXGTK__
28#include <gdk/gdk.h>
29#include <gdk/gdkx.h>
30#include <gtk/gtk.h>
31#include <gdk/gdkprivate.h>
32#ifdef __WXGTK20__
33#include <wx/gtk/win_gtk.h>
34#else
35#include <wx/gtk1/win_gtk.h>
36#endif
37#define GetXWindow(wxwin) (wxwin)->m_wxwindow ? \
38 GDK_WINDOW_XWINDOW(GTK_PIZZA((wxwin)->m_wxwindow)->bin_window) : \
39 GDK_WINDOW_XWINDOW((wxwin)->m_widget->window)
40#include <locale.h>
41#endif
42
43#ifdef __WXX11__
44#include "wx/x11/privx.h"
45#define GetXWindow(wxwin) ((Window)(wxwin)->GetHandle())
46#endif
47
48#ifdef __WXMAC__
49#include <wx/mac/private.h>
50#endif
51
52#include <wx/clipbrd.h>
53#include <wx/mimetype.h>
54#include <wx/image.h>
55
56//----------------------------------------------------------------------
57
58#if PYTHON_API_VERSION < 1009 && wxUSE_UNICODE
59#error Python must support Unicode to use wxWindows Unicode
60#endif
61
62//----------------------------------------------------------------------
63
64wxPyApp* wxPythonApp = NULL; // Global instance of application object
65bool wxPyDoCleanup = false;
66bool wxPyDoingCleanup = false;
67
68
69#ifdef WXP_WITH_THREAD
70#if !wxPyUSE_GIL_STATE
71struct wxPyThreadState {
72 unsigned long tid;
73 PyThreadState* tstate;
74
75 wxPyThreadState(unsigned long _tid=0, PyThreadState* _tstate=NULL)
76 : tid(_tid), tstate(_tstate) {}
77};
78
79#include <wx/dynarray.h>
80WX_DECLARE_OBJARRAY(wxPyThreadState, wxPyThreadStateArray);
81#include <wx/arrimpl.cpp>
82WX_DEFINE_OBJARRAY(wxPyThreadStateArray);
83
84wxPyThreadStateArray* wxPyTStates = NULL;
85wxMutex* wxPyTMutex = NULL;
86
87#endif
88#endif
89
90
91#define DEFAULTENCODING_SIZE 64
92static char wxPyDefaultEncoding[DEFAULTENCODING_SIZE] = "ascii";
93
94static PyObject* wxPython_dict = NULL;
95static PyObject* wxPyAssertionError = NULL;
96static PyObject* wxPyNoAppError = NULL;
97
98PyObject* wxPyPtrTypeMap = NULL;
99
100
101#ifdef __WXMSW__ // If building for win32...
102//----------------------------------------------------------------------
103// This gets run when the DLL is loaded. We just need to save a handle.
104//----------------------------------------------------------------------
105
106extern "C"
107BOOL WINAPI DllMain(
108 HINSTANCE hinstDLL, // handle to DLL module
109 DWORD fdwReason, // reason for calling function
110 LPVOID lpvReserved // reserved
111 )
112{
113 // If wxPython is embedded in another wxWidgets app then
114 // the instance has already been set.
115 if (! wxGetInstance())
116 wxSetInstance(hinstDLL);
117 return true;
118}
119#endif
120
121//----------------------------------------------------------------------
122// Classes for implementing the wxp main application shell.
123//----------------------------------------------------------------------
124
125IMPLEMENT_ABSTRACT_CLASS(wxPyApp, wxApp);
126
127
128wxPyApp::wxPyApp() {
129 m_assertMode = wxPYAPP_ASSERT_EXCEPTION;
130 m_startupComplete = false;
131}
132
133
134wxPyApp::~wxPyApp() {
135 wxPythonApp = NULL;
136 wxApp::SetInstance(NULL);
137}
138
139
140// This one isn't acutally called... We fake it with _BootstrapApp
141bool wxPyApp::OnInit() {
142 return false;
143}
144
145
146int wxPyApp::MainLoop() {
147 int retval = 0;
148
149 DeletePendingObjects();
150 bool initialized = wxTopLevelWindows.GetCount() != 0;
151 if (initialized) {
152 if ( m_exitOnFrameDelete == Later ) {
153 m_exitOnFrameDelete = Yes;
154 }
155
156 retval = wxApp::MainLoop();
157 OnExit();
158 }
159 return retval;
160}
161
162
163bool wxPyApp::OnInitGui() {
164 bool rval=true;
165 wxApp::OnInitGui(); // in this case always call the base class version
166 wxPyBlock_t blocked = wxPyBeginBlockThreads();
167 if (wxPyCBH_findCallback(m_myInst, "OnInitGui"))
168 rval = wxPyCBH_callCallback(m_myInst, Py_BuildValue("()"));
169 wxPyEndBlockThreads(blocked);
170 return rval;
171}
172
173
174int wxPyApp::OnExit() {
175 int rval=0;
176 wxPyBlock_t blocked = wxPyBeginBlockThreads();
177 if (wxPyCBH_findCallback(m_myInst, "OnExit"))
178 rval = wxPyCBH_callCallback(m_myInst, Py_BuildValue("()"));
179 wxPyEndBlockThreads(blocked);
180 wxApp::OnExit(); // in this case always call the base class version
181 return rval;
182}
183
184
185
186void wxPyApp::ExitMainLoop() {
187 bool found;
188 wxPyBlock_t blocked = wxPyBeginBlockThreads();
189 if ((found = wxPyCBH_findCallback(m_myInst, "ExitMainLoop")))
190 wxPyCBH_callCallback(m_myInst, Py_BuildValue("()"));
191 wxPyEndBlockThreads(blocked);
192 if (! found)
193 wxApp::ExitMainLoop();
194}
195
196
197#ifdef __WXDEBUG__
198void wxPyApp::OnAssertFailure(const wxChar *file,
199 int line,
200 const wxChar *func,
201 const wxChar *cond,
202 const wxChar *msg)
203{
204 // if we're not fully initialized then just log the error
205 if (! m_startupComplete) {
206 wxString buf;
207 buf.Alloc(4096);
208 buf.Printf(wxT("%s(%d): assert \"%s\" failed"),
209 file, line, cond);
210 if ( func && *func )
211 buf << wxT(" in ") << func << wxT("()");
212 if (msg != NULL)
213 buf << wxT(": ") << msg;
214
215 wxLogDebug(buf);
216 return;
217 }
218
219 // If the OnAssert is overloaded in the Python class then call it...
220 bool found;
221 wxPyBlock_t blocked = wxPyBeginBlockThreads();
222 if ((found = wxPyCBH_findCallback(m_myInst, "OnAssert"))) {
223 PyObject* fso = wx2PyString(file);
224 PyObject* cso = wx2PyString(file);
225 PyObject* mso;
226 if (msg != NULL)
227 mso = wx2PyString(file);
228 else {
229 mso = Py_None; Py_INCREF(Py_None);
230 }
231 wxPyCBH_callCallback(m_myInst, Py_BuildValue("(OiOO)", fso, line, cso, mso));
232 Py_DECREF(fso);
233 Py_DECREF(cso);
234 Py_DECREF(mso);
235 }
236 wxPyEndBlockThreads(blocked);
237
238 // ...otherwise do our own thing with it
239 if (! found) {
240 // ignore it?
241 if (m_assertMode & wxPYAPP_ASSERT_SUPPRESS)
242 return;
243
244 // turn it into a Python exception?
245 if (m_assertMode & wxPYAPP_ASSERT_EXCEPTION) {
246 wxString buf;
247 buf.Alloc(4096);
248 buf.Printf(wxT("C++ assertion \"%s\" failed at %s(%d)"), cond, file, line);
249 if ( func && *func )
250 buf << wxT(" in ") << func << wxT("()");
251 if (msg != NULL)
252 buf << wxT(": ") << msg;
253
254
255 // set the exception
256 wxPyBlock_t blocked = wxPyBeginBlockThreads();
257 PyObject* s = wx2PyString(buf);
258 PyErr_SetObject(wxPyAssertionError, s);
259 Py_DECREF(s);
260 wxPyEndBlockThreads(blocked);
261
262 // Now when control returns to whatever API wrapper was called from
263 // Python it should detect that an exception is set and will return
264 // NULL, signalling the exception to Python.
265 }
266
267 // Send it to the normal log destination, but only if
268 // not _DIALOG because it will call this too
269 if ( (m_assertMode & wxPYAPP_ASSERT_LOG) && !(m_assertMode & wxPYAPP_ASSERT_DIALOG)) {
270 wxString buf;
271 buf.Alloc(4096);
272 buf.Printf(wxT("%s(%d): assert \"%s\" failed"),
273 file, line, cond);
274 if ( func && *func )
275 buf << wxT(" in ") << func << wxT("()");
276 if (msg != NULL)
277 buf << wxT(": ") << msg;
278 wxLogDebug(buf);
279 }
280
281 // do the normal wx assert dialog?
282 if (m_assertMode & wxPYAPP_ASSERT_DIALOG)
283 wxApp::OnAssertFailure(file, line, func, cond, msg);
284 }
285}
286#endif
287
288 // For catching Apple Events
289void wxPyApp::MacOpenFile(const wxString &fileName)
290{
291 wxPyBlock_t blocked = wxPyBeginBlockThreads();
292 if (wxPyCBH_findCallback(m_myInst, "MacOpenFile")) {
293 PyObject* s = wx2PyString(fileName);
294 wxPyCBH_callCallback(m_myInst, Py_BuildValue("(O)", s));
295 Py_DECREF(s);
296 }
297 wxPyEndBlockThreads(blocked);
298}
299
300void wxPyApp::MacPrintFile(const wxString &fileName)
301{
302 wxPyBlock_t blocked = wxPyBeginBlockThreads();
303 if (wxPyCBH_findCallback(m_myInst, "MacPrintFile")) {
304 PyObject* s = wx2PyString(fileName);
305 wxPyCBH_callCallback(m_myInst, Py_BuildValue("(O)", s));
306 Py_DECREF(s);
307 }
308 wxPyEndBlockThreads(blocked);
309}
310
311void wxPyApp::MacNewFile()
312{
313 wxPyBlock_t blocked = wxPyBeginBlockThreads();
314 if (wxPyCBH_findCallback(m_myInst, "MacNewFile"))
315 wxPyCBH_callCallback(m_myInst, Py_BuildValue("()"));
316 wxPyEndBlockThreads(blocked);
317}
318
319void wxPyApp::MacReopenApp()
320{
321 wxPyBlock_t blocked = wxPyBeginBlockThreads();
322 if (wxPyCBH_findCallback(m_myInst, "MacReopenApp"))
323 wxPyCBH_callCallback(m_myInst, Py_BuildValue("()"));
324 wxPyEndBlockThreads(blocked);
325}
326
327
328/*static*/
329bool wxPyApp::GetMacSupportPCMenuShortcuts() {
330 return 0;
331}
332
333/*static*/
334long wxPyApp::GetMacAboutMenuItemId() {
335#ifdef __WXMAC__
336 return s_macAboutMenuItemId;
337#else
338 return 0;
339#endif
340}
341
342/*static*/
343long wxPyApp::GetMacPreferencesMenuItemId() {
344#ifdef __WXMAC__
345 return s_macPreferencesMenuItemId;
346#else
347 return 0;
348#endif
349}
350
351/*static*/
352long wxPyApp::GetMacExitMenuItemId() {
353#ifdef __WXMAC__
354 return s_macExitMenuItemId;
355#else
356 return 0;
357#endif
358}
359
360/*static*/
361wxString wxPyApp::GetMacHelpMenuTitleName() {
362#ifdef __WXMAC__
363 return s_macHelpMenuTitleName;
364#else
365 return wxEmptyString;
366#endif
367}
368
369/*static*/
370void wxPyApp::SetMacSupportPCMenuShortcuts(bool) {
371}
372
373/*static*/
374void wxPyApp::SetMacAboutMenuItemId(long val) {
375#ifdef __WXMAC__
376 s_macAboutMenuItemId = val;
377#endif
378}
379
380/*static*/
381void wxPyApp::SetMacPreferencesMenuItemId(long val) {
382#ifdef __WXMAC__
383 s_macPreferencesMenuItemId = val;
384#endif
385}
386
387/*static*/
388void wxPyApp::SetMacExitMenuItemId(long val) {
389#ifdef __WXMAC__
390 s_macExitMenuItemId = val;
391#endif
392}
393
394/*static*/
395void wxPyApp::SetMacHelpMenuTitleName(const wxString& val) {
396#ifdef __WXMAC__
397 s_macHelpMenuTitleName = val;
398#endif
399}
400
401
402// This finishes the initialization of wxWindows and then calls the OnInit
403// that should be present in the derived (Python) class.
404void wxPyApp::_BootstrapApp()
405{
406 static bool haveInitialized = false;
407 bool result;
408 wxPyBlock_t blocked;
409 PyObject* retval = NULL;
410 PyObject* pyint = NULL;
411
412
413 // Only initialize wxWidgets once
414 if (! haveInitialized) {
415
416 // Get any command-line args passed to this program from the sys module
417 int argc = 0;
418 char** argv = NULL;
419 blocked = wxPyBeginBlockThreads();
420
421 PyObject* sysargv = PySys_GetObject("argv");
422 PyObject* executable = PySys_GetObject("executable");
423
424 if (sysargv != NULL && executable != NULL) {
425 argc = PyList_Size(sysargv) + 1;
426 argv = new char*[argc+1];
427 argv[0] = strdup(PyString_AsString(executable));
428 int x;
429 for(x=1; x<argc; x++) {
430 PyObject *pyArg = PyList_GetItem(sysargv, x-1);
431 argv[x] = strdup(PyString_AsString(pyArg));
432 }
433 argv[argc] = NULL;
434 }
435 wxPyEndBlockThreads(blocked);
436
437 // Initialize wxWidgets
438 result = wxEntryStart(argc, argv);
439 // wxApp takes ownership of the argv array, don't delete it here
440
441 blocked = wxPyBeginBlockThreads();
442 if (! result) {
443 PyErr_SetString(PyExc_SystemError,
444 "wxEntryStart failed, unable to initialize wxWidgets!"
445#ifdef __WXGTK__
446 " (Is DISPLAY set properly?)"
447#endif
448 );
449 goto error;
450 }
451
452 // On wxGTK the locale will be changed to match the system settings,
453 // but Python before 2.4 needs to have LC_NUMERIC set to "C" in order
454 // for the floating point conversions and such to work right.
455#if defined(__WXGTK__) && PY_VERSION_HEX < 0x02040000
456 setlocale(LC_NUMERIC, "C");
457#endif
458
459// wxSystemOptions::SetOption(wxT("mac.textcontrol-use-mlte"), 1);
460
461 wxPyEndBlockThreads(blocked);
462 haveInitialized = true;
463 }
464 else {
465 this->argc = 0;
466 this->argv = NULL;
467 }
468
469
470 // It's now ok to generate exceptions for assertion errors.
471 wxPythonApp->SetStartupComplete(true);
472
473
474 // Call the Python wxApp's OnPreInit and OnInit functions
475 blocked = wxPyBeginBlockThreads();
476 if (wxPyCBH_findCallback(m_myInst, "OnPreInit")) {
477 PyObject* method = m_myInst.GetLastFound();
478 PyObject* argTuple = PyTuple_New(0);
479 retval = PyEval_CallObject(method, argTuple);
480 m_myInst.clearRecursionGuard(method);
481 Py_DECREF(argTuple);
482 Py_DECREF(method);
483 if (retval == NULL)
484 goto error;
485 }
486 if (wxPyCBH_findCallback(m_myInst, "OnInit")) {
487
488 PyObject* method = m_myInst.GetLastFound();
489 PyObject* argTuple = PyTuple_New(0);
490 retval = PyEval_CallObject(method, argTuple);
491 m_myInst.clearRecursionGuard(method);
492 Py_DECREF(argTuple);
493 Py_DECREF(method);
494 if (retval == NULL)
495 // Don't PyErr_Print here, let the exception in this case go back
496 // up to the wx.PyApp.__init__ scope.
497 goto error;
498
499 pyint = PyNumber_Int(retval);
500 if (! pyint) {
501 PyErr_SetString(PyExc_TypeError, "OnInit should return a boolean value");
502 goto error;
503 }
504 result = PyInt_AS_LONG(pyint);
505 }
506 else {
507 // Is it okay if there is no OnInit? Probably so...
508 result = true;
509 }
510
511 if (! result) {
512 PyErr_SetString(PyExc_SystemExit, "OnInit returned false, exiting...");
513 }
514
515 error:
516 Py_XDECREF(retval);
517 Py_XDECREF(pyint);
518
519 wxPyEndBlockThreads(blocked);
520};
521
522//---------------------------------------------------------------------
523//----------------------------------------------------------------------
524
525
526#if 0
527static char* wxPyCopyCString(const wxChar* src)
528{
529 wxWX2MBbuf buff = (wxWX2MBbuf)wxConvCurrent->cWX2MB(src);
530 size_t len = strlen(buff);
531 char* dest = new char[len+1];
532 strcpy(dest, buff);
533 return dest;
534}
535
536#if wxUSE_UNICODE
537static char* wxPyCopyCString(const char* src) // we need a char version too
538{
539 size_t len = strlen(src);
540 char* dest = new char[len+1];
541 strcpy(dest, src);
542 return dest;
543}
544#endif
545
546static wxChar* wxPyCopyWString(const char *src)
547{
548 //wxMB2WXbuf buff = wxConvCurrent->cMB2WX(src);
549 wxString str(src, *wxConvCurrent);
550 return copystring(str);
551}
552
553#if wxUSE_UNICODE
554static wxChar* wxPyCopyWString(const wxChar *src)
555{
556 return copystring(src);
557}
558#endif
559#endif
560
561
562inline const char* dropwx(const char* name) {
563 if (name[0] == 'w' && name[1] == 'x')
564 return name+2;
565 else
566 return name;
567}
568
569//----------------------------------------------------------------------
570
571// This function is called when the wx._core_ module is imported to do some
572// initial setup. (Before there is a wxApp object.) The rest happens in
573// wxPyApp::_BootstrapApp
574void __wxPyPreStart(PyObject* moduleDict)
575{
576
577#ifdef __WXMSW__
578// wxCrtSetDbgFlag(_CRTDBG_LEAK_CHECK_DF
579// | _CRTDBG_CHECK_ALWAYS_DF
580// | _CRTDBG_DELAY_FREE_MEM_DF
581// );
582#endif
583
584#ifdef WXP_WITH_THREAD
585#if wxPyUSE_GIL_STATE
586 PyEval_InitThreads();
587#else
588 PyEval_InitThreads();
589 wxPyTStates = new wxPyThreadStateArray;
590 wxPyTMutex = new wxMutex;
591
592 // Save the current (main) thread state in our array
593 PyThreadState* tstate = wxPyBeginAllowThreads();
594 wxPyEndAllowThreads(tstate);
595#endif
596#endif
597
598 // Ensure that the build options in the DLL (or whatever) match this build
599 wxApp::CheckBuildOptions(WX_BUILD_OPTIONS_SIGNATURE, "wxPython");
600
601 wxInitAllImageHandlers();
602}
603
604
605
606void __wxPyCleanup() {
607 wxPyDoingCleanup = true;
608 if (wxPyDoCleanup) {
609 wxPyDoCleanup = false;
610 wxEntryCleanup();
611 }
612#ifdef WXP_WITH_THREAD
613#if !wxPyUSE_GIL_STATE
614 delete wxPyTMutex;
615 wxPyTMutex = NULL;
616 wxPyTStates->Empty();
617 delete wxPyTStates;
618 wxPyTStates = NULL;
619#endif
620#endif
621}
622
623
624// Save a reference to the dictionary of the wx._core module, and inject
625// a few more things into it.
626PyObject* __wxPySetDictionary(PyObject* /* self */, PyObject* args)
627{
628
629 if (!PyArg_ParseTuple(args, "O", &wxPython_dict))
630 return NULL;
631
632 if (!PyDict_Check(wxPython_dict)) {
633 PyErr_SetString(PyExc_TypeError,
634 "_wxPySetDictionary must have dictionary object!");
635 return NULL;
636 }
637
638 if (! wxPyPtrTypeMap)
639 wxPyPtrTypeMap = PyDict_New();
640 PyDict_SetItemString(wxPython_dict, "__wxPyPtrTypeMap", wxPyPtrTypeMap);
641
642 // Create an exception object to use for wxASSERTions
643 wxPyAssertionError = PyErr_NewException("wx._core.PyAssertionError",
644 PyExc_AssertionError, NULL);
645 PyDict_SetItemString(wxPython_dict, "PyAssertionError", wxPyAssertionError);
646
647 // Create an exception object to use when the app object hasn't been created yet
648 wxPyNoAppError = PyErr_NewException("wx._core.PyNoAppError",
649 PyExc_RuntimeError, NULL);
650 PyDict_SetItemString(wxPython_dict, "PyNoAppError", wxPyNoAppError);
651
652
653
654#ifdef __WXMOTIF__
655#define wxPlatform "__WXMOTIF__"
656#define wxPlatName "wxMotif"
657#endif
658#ifdef __WXX11__
659#define wxPlatform "__WXX11__"
660#define wxPlatName "wxX11"
661#endif
662#ifdef __WXGTK__
663#define wxPlatform "__WXGTK__"
664#define wxPlatName "wxGTK"
665#endif
666#ifdef __WXMSW__
667#define wxPlatform "__WXMSW__"
668#define wxPlatName "wxMSW"
669#endif
670#ifdef __WXMAC__
671#define wxPlatform "__WXMAC__"
672#define wxPlatName "wxMac"
673#endif
674
675#ifdef __WXDEBUG__
676 int wxdebug = 1;
677#else
678 int wxdebug = 0;
679#endif
680
681 // These should be deprecated in favor of the PlatformInfo tuple built below...
682 PyDict_SetItemString(wxPython_dict, "Platform", PyString_FromString(wxPlatform));
683 PyDict_SetItemString(wxPython_dict, "USE_UNICODE", PyInt_FromLong(wxUSE_UNICODE));
684 PyDict_SetItemString(wxPython_dict, "__WXDEBUG__", PyInt_FromLong(wxdebug));
685
686 // Make a tuple of strings that gives more info about the platform.
687 PyObject* PlatInfo = PyList_New(0);
688 PyObject* obj;
689
690#define _AddInfoString(st) \
691 obj = PyString_FromString(st); \
692 PyList_Append(PlatInfo, obj); \
693 Py_DECREF(obj)
694
695 _AddInfoString(wxPlatform);
696 _AddInfoString(wxPlatName);
697#if wxUSE_UNICODE
698 _AddInfoString("unicode");
699#else
700 _AddInfoString("ansi");
701#endif
702#ifdef __WXGTK__
703#ifdef __WXGTK20__
704 _AddInfoString("gtk2");
705#else
706 _AddInfoString("gtk1");
707#endif
708#endif
709#ifdef __WXDEBUG__
710 _AddInfoString("wx-assertions-on");
711#else
712 _AddInfoString("wx-assertions-off");
713#endif
714 _AddInfoString(wxPy_SWIG_VERSION);
715#ifdef __WXMAC__
716 #if wxMAC_USE_CORE_GRAPHICS
717 _AddInfoString("mac-cg");
718 #else
719 _AddInfoString("mac-qd");
720 #endif
721 #if wxMAC_USE_NATIVE_TOOLBAR
722 _AddInfoString("mac-native-tb");
723 #else
724 _AddInfoString("mac-no-native-tb");
725 #endif
726#endif
727
728#undef _AddInfoString
729
730 PyObject* PlatInfoTuple = PyList_AsTuple(PlatInfo);
731 Py_DECREF(PlatInfo);
732 PyDict_SetItemString(wxPython_dict, "PlatformInfo", PlatInfoTuple);
733
734 RETURN_NONE();
735}
736
737
738
739//---------------------------------------------------------------------------
740
741// Check for existence of a wxApp, setting an exception if there isn't one.
742// This doesn't need to aquire the GIL because it should only be called from
743// an %exception before the lock is released.
744
745bool wxPyCheckForApp() {
746 if (wxTheApp != NULL)
747 return true;
748 else {
749 PyErr_SetString(wxPyNoAppError, "The wx.App object must be created first!");
750 return false;
751 }
752}
753
754//---------------------------------------------------------------------------
755
756void wxPyUserData_dtor(wxPyUserData* self) {
757 if (! wxPyDoingCleanup) {
758 wxPyBlock_t blocked = wxPyBeginBlockThreads();
759 Py_DECREF(self->m_obj);
760 self->m_obj = NULL;
761 wxPyEndBlockThreads(blocked);
762 }
763}
764
765
766void wxPyClientData_dtor(wxPyClientData* self) {
767 if (! wxPyDoingCleanup) { // Don't do it during cleanup as Python
768 // may have already garbage collected the object...
769 if (self->m_incRef) {
770 wxPyBlock_t blocked = wxPyBeginBlockThreads();
771 Py_DECREF(self->m_obj);
772 wxPyEndBlockThreads(blocked);
773 }
774 self->m_obj = NULL;
775 }
776}
777
778
779
780// This is called when an OOR controled object is being destroyed. Although
781// the C++ object is going away there is no way to force the Python object
782// (and all references to it) to die too. This causes problems (crashes) in
783// wxPython when a python shadow object attempts to call a C++ method using
784// the now bogus pointer... So to try and prevent this we'll do a little black
785// magic and change the class of the python instance to a class that will
786// raise an exception for any attempt to call methods with it. See
787// _wxPyDeadObject in _core_ex.py for the implementation of this class.
788void wxPyOORClientData_dtor(wxPyOORClientData* self) {
789
790 static PyObject* deadObjectClass = NULL;
791
792 wxPyBlock_t blocked = wxPyBeginBlockThreads();
793 if (deadObjectClass == NULL) {
794 deadObjectClass = PyDict_GetItemString(wxPython_dict, "_wxPyDeadObject");
795 // TODO: Can not wxASSERT here because inside a wxPyBeginBlock Threads,
796 // will lead to a deadlock when it tries to aquire the GIL again.
797 //wxASSERT_MSG(deadObjectClass != NULL, wxT("Can't get _wxPyDeadObject class!"));
798 Py_INCREF(deadObjectClass);
799 }
800
801
802 // Only if there is more than one reference to the object and we are
803 // holding the OOR reference:
804 if ( !wxPyDoingCleanup && self->m_obj->ob_refcnt > 1 && self->m_incRef) {
805 // bool isInstance = wxPyInstance_Check(self->m_obj);
806 // TODO same here
807 //wxASSERT_MSG(isInstance, wxT("m_obj not an instance!?!?!"));
808
809 // Call __del__, if there is one.
810 PyObject* func = PyObject_GetAttrString(self->m_obj, "__del__");
811 if (func) {
812 PyObject* rv = PyObject_CallMethod(self->m_obj, "__del__", NULL);
813 Py_XDECREF(rv);
814 Py_DECREF(func);
815 }
816 if (PyErr_Occurred())
817 PyErr_Clear(); // just ignore it for now
818
819
820 PyObject* dict = PyObject_GetAttrString(self->m_obj, "__dict__");
821 if (dict) {
822 // Clear the instance's dictionary
823 PyDict_Clear(dict);
824
825 // put the name of the old class into the instance, and then reset the
826 // class to be the dead class.
827 PyObject* klass = PyObject_GetAttrString(self->m_obj, "__class__");
828 PyObject* name = PyObject_GetAttrString(klass, "__name__");
829 PyDict_SetItemString(dict, "_name", name);
830 PyObject_SetAttrString(self->m_obj, "__class__", deadObjectClass);
831 //Py_INCREF(deadObjectClass);
832 Py_DECREF(klass);
833 Py_DECREF(name);
834 }
835 }
836
837 // m_obj is DECREF'd in the base class dtor...
838 wxPyEndBlockThreads(blocked);
839}
840
841
842//---------------------------------------------------------------------------
843// Stuff used by OOR to find the right wxPython class type to return and to
844// build it.
845
846
847// The pointer type map is used when the "pointer" type name generated by SWIG
848// is not the same as the shadow class name, for example wxPyTreeCtrl
849// vs. wxTreeCtrl. It needs to be referenced in Python as well as from C++,
850// so we'll just make it a Python dictionary in the wx module's namespace.
851// (See __wxSetDictionary)
852void wxPyPtrTypeMap_Add(const char* commonName, const char* ptrName) {
853 if (! wxPyPtrTypeMap)
854 wxPyPtrTypeMap = PyDict_New();
855 PyDict_SetItemString(wxPyPtrTypeMap,
856 (char*)commonName,
857 PyString_FromString((char*)ptrName));
858}
859
860
861
862
863PyObject* wxPyMake_wxObject(wxObject* source, bool setThisOwn, bool checkEvtHandler) {
864 PyObject* target = NULL;
865 bool isEvtHandler = false;
866 bool isSizer = false;
867
868 if (source) {
869 // If it's derived from wxEvtHandler then there may
870 // already be a pointer to a Python object that we can use
871 // in the OOR data.
872 if (checkEvtHandler && wxIsKindOf(source, wxEvtHandler)) {
873 isEvtHandler = true;
874 wxEvtHandler* eh = (wxEvtHandler*)source;
875 wxPyOORClientData* data = (wxPyOORClientData*)eh->GetClientObject();
876 if (data) {
877 target = data->m_obj;
878 if (target)
879 Py_INCREF(target);
880 }
881 }
882
883 // Also check for wxSizer
884 if (!target && wxIsKindOf(source, wxSizer)) {
885 isSizer = true;
886 wxSizer* sz = (wxSizer*)source;
887 wxPyOORClientData* data = (wxPyOORClientData*)sz->GetClientObject();
888 if (data) {
889 target = data->m_obj;
890 if (target)
891 Py_INCREF(target);
892 }
893 }
894
895 if (! target) {
896 // Otherwise make it the old fashioned way by making a new shadow
897 // object and putting this pointer in it. Look up the class
898 // heirarchy until we find a class name that is located in the
899 // python module.
900 const wxClassInfo* info = source->GetClassInfo();
901 wxString name = info->GetClassName();
902 bool exists = wxPyCheckSwigType(name);
903 while (info && !exists) {
904 info = info->GetBaseClass1();
905 name = info->GetClassName();
906 exists = wxPyCheckSwigType(name);
907 }
908 if (info) {
909 target = wxPyConstructObject((void*)source, name, setThisOwn);
910 if (target && isEvtHandler)
911 ((wxEvtHandler*)source)->SetClientObject(new wxPyOORClientData(target));
912 if (target && isSizer)
913 ((wxSizer*)source)->SetClientObject(new wxPyOORClientData(target));
914 } else {
915 wxString msg(wxT("wxPython class not found for "));
916 msg += source->GetClassInfo()->GetClassName();
917 PyErr_SetString(PyExc_NameError, msg.mbc_str());
918 target = NULL;
919 }
920 }
921 } else { // source was NULL so return None.
922 Py_INCREF(Py_None); target = Py_None;
923 }
924 return target;
925}
926
927
928PyObject* wxPyMake_wxSizer(wxSizer* source, bool setThisOwn) {
929
930 return wxPyMake_wxObject(source, setThisOwn);
931}
932
933
934//---------------------------------------------------------------------------
935
936
937#ifdef WXP_WITH_THREAD
938#if !wxPyUSE_GIL_STATE
939
940inline
941unsigned long wxPyGetCurrentThreadId() {
942 return wxThread::GetCurrentId();
943}
944
945static wxPyThreadState gs_shutdownTState;
946
947static
948wxPyThreadState* wxPyGetThreadState() {
949 if (wxPyTMutex == NULL) // Python is shutting down...
950 return &gs_shutdownTState;
951
952 unsigned long ctid = wxPyGetCurrentThreadId();
953 wxPyThreadState* tstate = NULL;
954
955 wxPyTMutex->Lock();
956 for(size_t i=0; i < wxPyTStates->GetCount(); i++) {
957 wxPyThreadState& info = wxPyTStates->Item(i);
958 if (info.tid == ctid) {
959 tstate = &info;
960 break;
961 }
962 }
963 wxPyTMutex->Unlock();
964 wxASSERT_MSG(tstate, wxT("PyThreadState should not be NULL!"));
965 return tstate;
966}
967
968
969static
970void wxPySaveThreadState(PyThreadState* tstate) {
971 if (wxPyTMutex == NULL) { // Python is shutting down, assume a single thread...
972 gs_shutdownTState.tstate = tstate;
973 return;
974 }
975 unsigned long ctid = wxPyGetCurrentThreadId();
976 wxPyTMutex->Lock();
977 for(size_t i=0; i < wxPyTStates->GetCount(); i++) {
978 wxPyThreadState& info = wxPyTStates->Item(i);
979 if (info.tid == ctid) {
980#if 0
981 if (info.tstate != tstate)
982 wxLogMessage("*** tstate mismatch!???");
983#endif
984 info.tstate = tstate; // allow for transient tstates
985 // Normally it will never change, but apparently COM callbacks
986 // (i.e. ActiveX controls) will (incorrectly IMHO) use a transient
987 // tstate which will then be garbage the next time we try to use
988 // it...
989
990 wxPyTMutex->Unlock();
991 return;
992 }
993 }
994 // not found, so add it...
995 wxPyTStates->Add(new wxPyThreadState(ctid, tstate));
996 wxPyTMutex->Unlock();
997}
998
999#endif
1000#endif
1001
1002
1003
1004// Calls from Python to wxWindows code are wrapped in calls to these
1005// functions:
1006
1007PyThreadState* wxPyBeginAllowThreads() {
1008#ifdef WXP_WITH_THREAD
1009 PyThreadState* saved = PyEval_SaveThread(); // Py_BEGIN_ALLOW_THREADS;
1010#if !wxPyUSE_GIL_STATE
1011 wxPySaveThreadState(saved);
1012#endif
1013 return saved;
1014#else
1015 return NULL;
1016#endif
1017}
1018
1019void wxPyEndAllowThreads(PyThreadState* saved) {
1020#ifdef WXP_WITH_THREAD
1021 PyEval_RestoreThread(saved); // Py_END_ALLOW_THREADS;
1022#endif
1023}
1024
1025
1026
1027// Calls from wxWindows back to Python code, or even any PyObject
1028// manipulations, PyDECREF's and etc. are wrapped in calls to these functions:
1029
1030wxPyBlock_t wxPyBeginBlockThreads() {
1031#ifdef WXP_WITH_THREAD
1032 if (! Py_IsInitialized()) {
1033 return (wxPyBlock_t)0;
1034 }
1035#if wxPyUSE_GIL_STATE
1036 PyGILState_STATE state = PyGILState_Ensure();
1037 return state;
1038#else
1039 PyThreadState *current = _PyThreadState_Current;
1040
1041 // Only block if there wasn't already a tstate, or if the current one is
1042 // not the one we are wanting to change to. This should prevent deadlock
1043 // if there are nested calls to wxPyBeginBlockThreads
1044 wxPyBlock_t blocked = false;
1045 wxPyThreadState* tstate = wxPyGetThreadState();
1046 if (current != tstate->tstate) {
1047 PyEval_RestoreThread(tstate->tstate);
1048 blocked = true;
1049 }
1050 return blocked;
1051#endif
1052#else
1053 return (wxPyBlock_t)0;
1054#endif
1055}
1056
1057
1058void wxPyEndBlockThreads(wxPyBlock_t blocked) {
1059#ifdef WXP_WITH_THREAD
1060 if (! Py_IsInitialized()) {
1061 return;
1062 }
1063#if wxPyUSE_GIL_STATE
1064 PyGILState_Release(blocked);
1065#else
1066 // Only unblock if we blocked in the last call to wxPyBeginBlockThreads.
1067 // The value of blocked passed in needs to be the same as that returned
1068 // from wxPyBeginBlockThreads at the same nesting level.
1069 if ( blocked ) {
1070 PyEval_SaveThread();
1071 }
1072#endif
1073#endif
1074}
1075
1076
1077//---------------------------------------------------------------------------
1078// wxPyInputStream and wxPyCBInputStream methods
1079
1080
1081void wxPyInputStream::close() {
1082 /* do nothing for now */
1083}
1084
1085void wxPyInputStream::flush() {
1086 /* do nothing for now */
1087}
1088
1089bool wxPyInputStream::eof() {
1090 if (m_wxis)
1091 return m_wxis->Eof();
1092 else
1093 return true;
1094}
1095
1096wxPyInputStream::~wxPyInputStream() {
1097 if (m_wxis)
1098 delete m_wxis;
1099}
1100
1101
1102
1103
1104PyObject* wxPyInputStream::read(int size) {
1105 PyObject* obj = NULL;
1106 wxMemoryBuffer buf;
1107 const int BUFSIZE = 1024;
1108
1109 // check if we have a real wxInputStream to work with
1110 if (!m_wxis) {
1111 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1112 PyErr_SetString(PyExc_IOError, "no valid C-wxInputStream");
1113 wxPyEndBlockThreads(blocked);
1114 return NULL;
1115 }
1116
1117 if (size < 0) {
1118 // read while bytes are available on the stream
1119 while ( m_wxis->CanRead() ) {
1120 m_wxis->Read(buf.GetAppendBuf(BUFSIZE), BUFSIZE);
1121 buf.UngetAppendBuf(m_wxis->LastRead());
1122 }
1123
1124 } else { // Read only size number of characters
1125 m_wxis->Read(buf.GetWriteBuf(size), size);
1126 buf.UngetWriteBuf(m_wxis->LastRead());
1127 }
1128
1129 // error check
1130 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1131 wxStreamError err = m_wxis->GetLastError();
1132 if (err != wxSTREAM_NO_ERROR && err != wxSTREAM_EOF) {
1133 PyErr_SetString(PyExc_IOError,"IOError in wxInputStream");
1134 }
1135 else {
1136 // We use only strings for the streams, not unicode
1137 obj = PyString_FromStringAndSize(buf, buf.GetDataLen());
1138 }
1139 wxPyEndBlockThreads(blocked);
1140 return obj;
1141}
1142
1143
1144PyObject* wxPyInputStream::readline(int size) {
1145 PyObject* obj = NULL;
1146 wxMemoryBuffer buf;
1147 int i;
1148 char ch;
1149
1150 // check if we have a real wxInputStream to work with
1151 if (!m_wxis) {
1152 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1153 PyErr_SetString(PyExc_IOError,"no valid C-wxInputStream");
1154 wxPyEndBlockThreads(blocked);
1155 return NULL;
1156 }
1157
1158 // read until \n or byte limit reached
1159 for (i=ch=0; (ch != '\n') && (m_wxis->CanRead()) && ((size < 0) || (i < size)); i++) {
1160 ch = m_wxis->GetC();
1161 buf.AppendByte(ch);
1162 }
1163
1164 // errorcheck
1165 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1166 wxStreamError err = m_wxis->GetLastError();
1167 if (err != wxSTREAM_NO_ERROR && err != wxSTREAM_EOF) {
1168 PyErr_SetString(PyExc_IOError,"IOError in wxInputStream");
1169 }
1170 else {
1171 // We use only strings for the streams, not unicode
1172 obj = PyString_FromStringAndSize((char*)buf.GetData(), buf.GetDataLen());
1173 }
1174 wxPyEndBlockThreads(blocked);
1175 return obj;
1176}
1177
1178
1179PyObject* wxPyInputStream::readlines(int sizehint) {
1180 PyObject* pylist;
1181
1182 // check if we have a real wxInputStream to work with
1183 if (!m_wxis) {
1184 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1185 PyErr_SetString(PyExc_IOError,"no valid C-wxInputStream");
1186 wxPyEndBlockThreads(blocked);
1187 return NULL;
1188 }
1189
1190 // init list
1191 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1192 pylist = PyList_New(0);
1193 wxPyEndBlockThreads(blocked);
1194
1195 if (!pylist) {
1196 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1197 PyErr_NoMemory();
1198 wxPyEndBlockThreads(blocked);
1199 return NULL;
1200 }
1201
1202 // read sizehint bytes or until EOF
1203 int i;
1204 for (i=0; (m_wxis->CanRead()) && ((sizehint < 0) || (i < sizehint));) {
1205 PyObject* s = this->readline();
1206 if (s == NULL) {
1207 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1208 Py_DECREF(pylist);
1209 wxPyEndBlockThreads(blocked);
1210 return NULL;
1211 }
1212 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1213 PyList_Append(pylist, s);
1214 i += PyString_Size(s);
1215 wxPyEndBlockThreads(blocked);
1216 }
1217
1218 // error check
1219 wxStreamError err = m_wxis->GetLastError();
1220 if (err != wxSTREAM_NO_ERROR && err != wxSTREAM_EOF) {
1221 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1222 Py_DECREF(pylist);
1223 PyErr_SetString(PyExc_IOError,"IOError in wxInputStream");
1224 wxPyEndBlockThreads(blocked);
1225 return NULL;
1226 }
1227
1228 return pylist;
1229}
1230
1231
1232void wxPyInputStream::seek(int offset, int whence) {
1233 if (m_wxis)
1234 m_wxis->SeekI(offset, wxSeekMode(whence));
1235}
1236
1237int wxPyInputStream::tell(){
1238 if (m_wxis)
1239 return m_wxis->TellI();
1240 else return 0;
1241}
1242
1243
1244
1245
1246wxPyCBInputStream::wxPyCBInputStream(PyObject *r, PyObject *s, PyObject *t, bool block)
1247 : wxInputStream(), m_read(r), m_seek(s), m_tell(t), m_block(block)
1248{}
1249
1250wxPyCBInputStream::wxPyCBInputStream(const wxPyCBInputStream& other)
1251{
1252 m_read = other.m_read;
1253 m_seek = other.m_seek;
1254 m_tell = other.m_tell;
1255 m_block = other.m_block;
1256 Py_INCREF(m_read);
1257 Py_INCREF(m_seek);
1258 Py_INCREF(m_tell);
1259}
1260
1261
1262wxPyCBInputStream::~wxPyCBInputStream() {
1263 wxPyBlock_t blocked = wxPyBlock_t_default;
1264 if (m_block) blocked = wxPyBeginBlockThreads();
1265 Py_XDECREF(m_read);
1266 Py_XDECREF(m_seek);
1267 Py_XDECREF(m_tell);
1268 if (m_block) wxPyEndBlockThreads(blocked);
1269}
1270
1271
1272wxPyCBInputStream* wxPyCBInputStream::create(PyObject *py, bool block) {
1273 wxPyBlock_t blocked = wxPyBlock_t_default;
1274 if (block) blocked = wxPyBeginBlockThreads();
1275
1276 PyObject* read = getMethod(py, "read");
1277 PyObject* seek = getMethod(py, "seek");
1278 PyObject* tell = getMethod(py, "tell");
1279
1280 if (!read) {
1281 PyErr_SetString(PyExc_TypeError, "Not a file-like object");
1282 Py_XDECREF(read);
1283 Py_XDECREF(seek);
1284 Py_XDECREF(tell);
1285 if (block) wxPyEndBlockThreads(blocked);
1286 return NULL;
1287 }
1288
1289 if (block) wxPyEndBlockThreads(blocked);
1290 return new wxPyCBInputStream(read, seek, tell, block);
1291}
1292
1293
1294wxPyCBInputStream* wxPyCBInputStream_create(PyObject *py, bool block) {
1295 return wxPyCBInputStream::create(py, block);
1296}
1297
1298wxPyCBInputStream* wxPyCBInputStream_copy(wxPyCBInputStream* other) {
1299 return new wxPyCBInputStream(*other);
1300}
1301
1302PyObject* wxPyCBInputStream::getMethod(PyObject* py, char* name) {
1303 if (!PyObject_HasAttrString(py, name))
1304 return NULL;
1305 PyObject* o = PyObject_GetAttrString(py, name);
1306 if (!PyMethod_Check(o) && !PyCFunction_Check(o)) {
1307 Py_DECREF(o);
1308 return NULL;
1309 }
1310 return o;
1311}
1312
1313
1314wxFileOffset wxPyCBInputStream::GetLength() const {
1315 wxPyCBInputStream* self = (wxPyCBInputStream*)this; // cast off const
1316 if (m_seek && m_tell) {
1317 wxFileOffset temp = self->OnSysTell();
1318 wxFileOffset ret = self->OnSysSeek(0, wxFromEnd);
1319 self->OnSysSeek(temp, wxFromStart);
1320 return ret;
1321 }
1322 else
1323 return wxInvalidOffset;
1324}
1325
1326
1327size_t wxPyCBInputStream::OnSysRead(void *buffer, size_t bufsize) {
1328 if (bufsize == 0)
1329 return 0;
1330
1331 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1332 PyObject* arglist = Py_BuildValue("(i)", bufsize);
1333 PyObject* result = PyEval_CallObject(m_read, arglist);
1334 Py_DECREF(arglist);
1335
1336 size_t o = 0;
1337 if ((result != NULL) && PyString_Check(result)) {
1338 o = PyString_Size(result);
1339 if (o == 0)
1340 m_lasterror = wxSTREAM_EOF;
1341 if (o > bufsize)
1342 o = bufsize;
1343 memcpy((char*)buffer, PyString_AsString(result), o); // strings only, not unicode...
1344 Py_DECREF(result);
1345
1346 }
1347 else
1348 m_lasterror = wxSTREAM_READ_ERROR;
1349 wxPyEndBlockThreads(blocked);
1350 return o;
1351}
1352
1353size_t wxPyCBInputStream::OnSysWrite(const void *buffer, size_t bufsize) {
1354 m_lasterror = wxSTREAM_WRITE_ERROR;
1355 return 0;
1356}
1357
1358
1359wxFileOffset wxPyCBInputStream::OnSysSeek(wxFileOffset off, wxSeekMode mode) {
1360 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1361 PyObject* arglist = PyTuple_New(2);
1362
1363 if (sizeof(wxFileOffset) > sizeof(long))
1364 // wxFileOffset is a 64-bit value...
1365 PyTuple_SET_ITEM(arglist, 0, PyLong_FromLongLong(off));
1366 else
1367 PyTuple_SET_ITEM(arglist, 0, PyInt_FromLong(off));
1368
1369 PyTuple_SET_ITEM(arglist, 1, PyInt_FromLong(mode));
1370
1371
1372 PyObject* result = PyEval_CallObject(m_seek, arglist);
1373 Py_DECREF(arglist);
1374 Py_XDECREF(result);
1375 wxPyEndBlockThreads(blocked);
1376 return OnSysTell();
1377}
1378
1379
1380wxFileOffset wxPyCBInputStream::OnSysTell() const {
1381 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1382 PyObject* arglist = Py_BuildValue("()");
1383 PyObject* result = PyEval_CallObject(m_tell, arglist);
1384 Py_DECREF(arglist);
1385 wxFileOffset o = 0;
1386 if (result != NULL) {
1387 if (PyLong_Check(result))
1388 o = PyLong_AsLongLong(result);
1389 else
1390 o = PyInt_AsLong(result);
1391 Py_DECREF(result);
1392 };
1393 wxPyEndBlockThreads(blocked);
1394 return o;
1395}
1396
1397//----------------------------------------------------------------------
1398
1399IMPLEMENT_ABSTRACT_CLASS(wxPyCallback, wxObject);
1400
1401wxPyCallback::wxPyCallback(PyObject* func) {
1402 m_func = func;
1403 Py_INCREF(m_func);
1404}
1405
1406wxPyCallback::wxPyCallback(const wxPyCallback& other) {
1407 m_func = other.m_func;
1408 Py_INCREF(m_func);
1409}
1410
1411wxPyCallback::~wxPyCallback() {
1412 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1413 Py_DECREF(m_func);
1414 wxPyEndBlockThreads(blocked);
1415}
1416
1417
1418#define wxPy_PRECALLINIT "_preCallInit"
1419#define wxPy_POSTCALLCLEANUP "_postCallCleanup"
1420
1421// This function is used for all events destined for Python event handlers.
1422void wxPyCallback::EventThunker(wxEvent& event) {
1423 wxPyCallback* cb = (wxPyCallback*)event.m_callbackUserData;
1424 PyObject* func = cb->m_func;
1425 PyObject* result;
1426 PyObject* arg;
1427 PyObject* tuple;
1428 bool checkSkip = false;
1429
1430 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1431 wxString className = event.GetClassInfo()->GetClassName();
1432
1433 // If the event is one of these types then pass the original
1434 // event object instead of the one passed to us.
1435 if ( className == wxT("wxPyEvent") ) {
1436 arg = ((wxPyEvent*)&event)->GetSelf();
1437 checkSkip = ((wxPyEvent*)&event)->GetCloned();
1438 }
1439 else if ( className == wxT("wxPyCommandEvent") ) {
1440 arg = ((wxPyCommandEvent*)&event)->GetSelf();
1441 checkSkip = ((wxPyCommandEvent*)&event)->GetCloned();
1442 }
1443 else {
1444 arg = wxPyConstructObject((void*)&event, className);
1445 }
1446
1447 if (!arg) {
1448 PyErr_Print();
1449 } else {
1450 // "intern" the pre/post method names to speed up the HasAttr
1451 static PyObject* s_preName = NULL;
1452 static PyObject* s_postName = NULL;
1453 if (s_preName == NULL) {
1454 s_preName = PyString_FromString(wxPy_PRECALLINIT);
1455 s_postName = PyString_FromString(wxPy_POSTCALLCLEANUP);
1456 }
1457
1458 // Check if the event object needs some preinitialization
1459 if (PyObject_HasAttr(arg, s_preName)) {
1460 result = PyObject_CallMethodObjArgs(arg, s_preName, arg, NULL);
1461 if ( result ) {
1462 Py_DECREF(result); // result is ignored, but we still need to decref it
1463 PyErr_Clear(); // Just in case...
1464 } else {
1465 PyErr_Print();
1466 }
1467 }
1468
1469 // Call the event handler, passing the event object
1470 tuple = PyTuple_New(1);
1471 PyTuple_SET_ITEM(tuple, 0, arg); // steals ref to arg
1472 result = PyEval_CallObject(func, tuple);
1473 if ( result ) {
1474 Py_DECREF(result); // result is ignored, but we still need to decref it
1475 PyErr_Clear(); // Just in case...
1476 } else {
1477 PyErr_Print();
1478 }
1479
1480 // Check if the event object needs some post cleanup
1481 if (PyObject_HasAttr(arg, s_postName)) {
1482 result = PyObject_CallMethodObjArgs(arg, s_postName, arg, NULL);
1483 if ( result ) {
1484 Py_DECREF(result); // result is ignored, but we still need to decref it
1485 PyErr_Clear(); // Just in case...
1486 } else {
1487 PyErr_Print();
1488 }
1489 }
1490
1491 if ( checkSkip ) {
1492 // if the event object was one of our special types and
1493 // it had been cloned, then we need to extract the Skipped
1494 // value from the original and set it in the clone.
1495 result = PyObject_CallMethod(arg, "GetSkipped", "");
1496 if ( result ) {
1497 event.Skip(PyInt_AsLong(result));
1498 Py_DECREF(result);
1499 } else {
1500 PyErr_Print();
1501 }
1502 }
1503 Py_DECREF(tuple);
1504 }
1505 wxPyEndBlockThreads(blocked);
1506}
1507
1508
1509//----------------------------------------------------------------------
1510
1511wxPyCallbackHelper::wxPyCallbackHelper(const wxPyCallbackHelper& other) {
1512 m_lastFound = NULL;
1513 m_self = other.m_self;
1514 m_class = other.m_class;
1515 if (m_self) {
1516 Py_INCREF(m_self);
1517 Py_INCREF(m_class);
1518 }
1519}
1520
1521
1522void wxPyCallbackHelper::setSelf(PyObject* self, PyObject* klass, int incref) {
1523 m_self = self;
1524 m_class = klass;
1525 m_incRef = incref;
1526 if (incref) {
1527 Py_INCREF(m_self);
1528 Py_INCREF(m_class);
1529 }
1530}
1531
1532
1533#if PYTHON_API_VERSION >= 1011
1534
1535// Prior to Python 2.2 PyMethod_GetClass returned the class object
1536// in which the method was defined. Starting with 2.2 it returns
1537// "class that asked for the method" which seems totally bogus to me
1538// but apprently it fixes some obscure problem waiting to happen in
1539// Python. Since the API was not documented Guido and the gang felt
1540// safe in changing it. Needless to say that totally screwed up the
1541// logic below in wxPyCallbackHelper::findCallback, hence this icky
1542// code to find the class where the method is actually defined...
1543
1544static
1545PyObject* PyFindClassWithAttr(PyObject *klass, PyObject *name)
1546{
1547 int i, n;
1548
1549 if (PyType_Check(klass)) { // new style classes
1550 // This code is borrowed/adapted from _PyType_Lookup in typeobject.c
1551 PyTypeObject* type = (PyTypeObject*)klass;
1552 PyObject *mro, *res, *base, *dict;
1553 /* Look in tp_dict of types in MRO */
1554 mro = type->tp_mro;
1555 assert(PyTuple_Check(mro));
1556 n = PyTuple_GET_SIZE(mro);
1557 for (i = 0; i < n; i++) {
1558 base = PyTuple_GET_ITEM(mro, i);
1559 if (PyClass_Check(base))
1560 dict = ((PyClassObject *)base)->cl_dict;
1561 else {
1562 assert(PyType_Check(base));
1563 dict = ((PyTypeObject *)base)->tp_dict;
1564 }
1565 assert(dict && PyDict_Check(dict));
1566 res = PyDict_GetItem(dict, name);
1567 if (res != NULL)
1568 return base;
1569 }
1570 return NULL;
1571 }
1572
1573 else if (PyClass_Check(klass)) { // old style classes
1574 // This code is borrowed/adapted from class_lookup in classobject.c
1575 PyClassObject* cp = (PyClassObject*)klass;
1576 PyObject *value = PyDict_GetItem(cp->cl_dict, name);
1577 if (value != NULL) {
1578 return (PyObject*)cp;
1579 }
1580 n = PyTuple_Size(cp->cl_bases);
1581 for (i = 0; i < n; i++) {
1582 PyObject* base = PyTuple_GetItem(cp->cl_bases, i);
1583 PyObject *v = PyFindClassWithAttr(base, name);
1584 if (v != NULL)
1585 return v;
1586 }
1587 return NULL;
1588 }
1589 return NULL;
1590}
1591#endif
1592
1593
1594static
1595PyObject* PyMethod_GetDefiningClass(PyObject* method, PyObject* nameo)
1596{
1597 PyObject* mgc = PyMethod_GET_CLASS(method);
1598
1599#if PYTHON_API_VERSION <= 1010 // prior to Python 2.2, the easy way
1600 return mgc;
1601#else // 2.2 and after, the hard way...
1602 return PyFindClassWithAttr(mgc, nameo);
1603#endif
1604}
1605
1606
1607
1608// To avoid recursion when an overridden virtual method wants to call the base
1609// class version, temporarily set an attribute in the instance with the same
1610// name as the method. Then the PyObject_GetAttr in the next findCallback
1611// will return this attribute and the PyMethod_Check will fail.
1612
1613void wxPyCallbackHelper::setRecursionGuard(PyObject* method) const
1614{
1615 PyFunctionObject* func = (PyFunctionObject*)PyMethod_Function(method);
1616 PyObject_SetAttr(m_self, func->func_name, Py_None);
1617}
1618
1619void wxPyCallbackHelper::clearRecursionGuard(PyObject* method) const
1620{
1621 PyFunctionObject* func = (PyFunctionObject*)PyMethod_Function(method);
1622 if (PyObject_HasAttr(m_self, func->func_name)) {
1623 PyObject_DelAttr(m_self, func->func_name);
1624 }
1625}
1626
1627// bool wxPyCallbackHelper::hasRecursionGuard(PyObject* method) const
1628// {
1629// PyFunctionObject* func = (PyFunctionObject*)PyMethod_Function(method);
1630// if (PyObject_HasAttr(m_self, func->func_name)) {
1631// PyObject* attr = PyObject_GetAttr(m_self, func->func_name);
1632// bool retval = (attr == Py_None);
1633// Py_DECREF(attr);
1634// return retval;
1635// }
1636// return false;
1637// }
1638
1639
1640bool wxPyCallbackHelper::findCallback(const char* name, bool setGuard) const {
1641 wxPyCallbackHelper* self = (wxPyCallbackHelper*)this; // cast away const
1642 PyObject *method, *klass;
1643 PyObject* nameo = PyString_FromString(name);
1644 self->m_lastFound = NULL;
1645
1646 // If the object (m_self) has an attibute of the given name...
1647 if (m_self && PyObject_HasAttr(m_self, nameo)) {
1648 method = PyObject_GetAttr(m_self, nameo);
1649
1650 // ...and if that attribute is a method, and if that method's class is
1651 // not from the registered class or a base class...
1652 if (PyMethod_Check(method) &&
1653 (klass = PyMethod_GetDefiningClass(method, nameo)) != NULL &&
1654 (klass != m_class) &&
1655 PyObject_IsSubclass(klass, m_class)) {
1656
1657 // ...then we'll save a pointer to the method so callCallback can
1658 // call it. But first, set a recursion guard in case the
1659 // overridden method wants to call the base class version.
1660 if (setGuard)
1661 setRecursionGuard(method);
1662 self->m_lastFound = method;
1663 }
1664 else {
1665 Py_DECREF(method);
1666 }
1667 }
1668
1669 Py_DECREF(nameo);
1670 return m_lastFound != NULL;
1671}
1672
1673
1674int wxPyCallbackHelper::callCallback(PyObject* argTuple) const {
1675 PyObject* result;
1676 int retval = false;
1677
1678 result = callCallbackObj(argTuple);
1679 if (result) { // Assumes an integer return type...
1680 retval = PyInt_AsLong(result);
1681 Py_DECREF(result);
1682 PyErr_Clear(); // forget about it if it's not...
1683 }
1684 return retval;
1685}
1686
1687// Invoke the Python callable object, returning the raw PyObject return
1688// value. Caller should DECREF the return value and also manage the GIL.
1689PyObject* wxPyCallbackHelper::callCallbackObj(PyObject* argTuple) const {
1690 PyObject* result;
1691
1692 // Save a copy of the pointer in case the callback generates another
1693 // callback. In that case m_lastFound will have a different value when
1694 // it gets back here...
1695 PyObject* method = m_lastFound;
1696
1697 result = PyEval_CallObject(method, argTuple);
1698 clearRecursionGuard(method);
1699
1700 Py_DECREF(argTuple);
1701 Py_DECREF(method);
1702 if (!result) {
1703 PyErr_Print();
1704 }
1705 return result;
1706}
1707
1708
1709void wxPyCBH_setCallbackInfo(wxPyCallbackHelper& cbh, PyObject* self, PyObject* klass, int incref) {
1710 cbh.setSelf(self, klass, incref);
1711}
1712
1713bool wxPyCBH_findCallback(const wxPyCallbackHelper& cbh, const char* name, bool setGuard) {
1714 return cbh.findCallback(name, setGuard);
1715}
1716
1717int wxPyCBH_callCallback(const wxPyCallbackHelper& cbh, PyObject* argTuple) {
1718 return cbh.callCallback(argTuple);
1719}
1720
1721PyObject* wxPyCBH_callCallbackObj(const wxPyCallbackHelper& cbh, PyObject* argTuple) {
1722 return cbh.callCallbackObj(argTuple);
1723}
1724
1725
1726void wxPyCBH_delete(wxPyCallbackHelper* cbh) {
1727 if (cbh->m_incRef && Py_IsInitialized()) {
1728 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1729 Py_XDECREF(cbh->m_self);
1730 Py_XDECREF(cbh->m_class);
1731 wxPyEndBlockThreads(blocked);
1732 }
1733}
1734
1735//---------------------------------------------------------------------------
1736//---------------------------------------------------------------------------
1737// These event classes can be derived from in Python and passed through the event
1738// system without losing anything. They do this by keeping a reference to
1739// themselves and some special case handling in wxPyCallback::EventThunker.
1740
1741
1742wxPyEvtSelfRef::wxPyEvtSelfRef() {
1743 //m_self = Py_None; // **** We don't do normal ref counting to prevent
1744 //Py_INCREF(m_self); // circular loops...
1745 m_cloned = false;
1746}
1747
1748wxPyEvtSelfRef::~wxPyEvtSelfRef() {
1749 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1750 if (m_cloned)
1751 Py_DECREF(m_self);
1752 wxPyEndBlockThreads(blocked);
1753}
1754
1755void wxPyEvtSelfRef::SetSelf(PyObject* self, bool clone) {
1756 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1757 if (m_cloned)
1758 Py_DECREF(m_self);
1759 m_self = self;
1760 if (clone) {
1761 Py_INCREF(m_self);
1762 m_cloned = true;
1763 }
1764 wxPyEndBlockThreads(blocked);
1765}
1766
1767PyObject* wxPyEvtSelfRef::GetSelf() const {
1768 Py_INCREF(m_self);
1769 return m_self;
1770}
1771
1772
1773IMPLEMENT_ABSTRACT_CLASS(wxPyEvent, wxEvent);
1774IMPLEMENT_ABSTRACT_CLASS(wxPyCommandEvent, wxCommandEvent);
1775
1776
1777wxPyEvent::wxPyEvent(int winid, wxEventType commandType)
1778 : wxEvent(winid, commandType) {
1779}
1780
1781
1782wxPyEvent::wxPyEvent(const wxPyEvent& evt)
1783 : wxEvent(evt)
1784{
1785 SetSelf(evt.m_self, true);
1786}
1787
1788
1789wxPyEvent::~wxPyEvent() {
1790}
1791
1792
1793wxPyCommandEvent::wxPyCommandEvent(wxEventType commandType, int id)
1794 : wxCommandEvent(commandType, id) {
1795}
1796
1797
1798wxPyCommandEvent::wxPyCommandEvent(const wxPyCommandEvent& evt)
1799 : wxCommandEvent(evt)
1800{
1801 SetSelf(evt.m_self, true);
1802}
1803
1804
1805wxPyCommandEvent::~wxPyCommandEvent() {
1806}
1807
1808
1809
1810
1811
1812//---------------------------------------------------------------------------
1813//---------------------------------------------------------------------------
1814// Convert a wxList to a Python List, only works for lists of wxObjects
1815
1816PyObject* wxPy_ConvertList(wxListBase* listbase) {
1817 wxList* list = (wxList*)listbase; // this is probably bad...
1818 PyObject* pyList;
1819 PyObject* pyObj;
1820 wxObject* wxObj;
1821 wxNode* node = list->GetFirst();
1822
1823 wxPyBlock_t blocked = wxPyBeginBlockThreads();
1824 pyList = PyList_New(0);
1825 while (node) {
1826 wxObj = node->GetData();
1827 pyObj = wxPyMake_wxObject(wxObj,false);
1828 PyList_Append(pyList, pyObj);
1829 Py_DECREF(pyObj); // the Append also does an INCREF, that's one more than we need.
1830 node = node->GetNext();
1831 }
1832 wxPyEndBlockThreads(blocked);
1833 return pyList;
1834}
1835
1836//----------------------------------------------------------------------
1837
1838long wxPyGetWinHandle(wxWindow* win) {
1839
1840#ifdef __WXMSW__
1841 return (long)win->GetHandle();
1842#endif
1843
1844#if defined(__WXGTK__) || defined(__WXX11)
1845 return (long)GetXWindow(win);
1846#endif
1847
1848#ifdef __WXMAC__
1849 //return (long)MAC_WXHWND(win->MacGetTopLevelWindowRef());
1850 return (long)win->GetHandle();
1851#endif
1852
1853 return 0;
1854}
1855
1856//----------------------------------------------------------------------
1857// Some helper functions for typemaps in my_typemaps.i, so they won't be
1858// included in every file over and over again...
1859
1860wxString* wxString_in_helper(PyObject* source) {
1861 wxString* target = NULL;
1862
1863 if (!PyString_Check(source) && !PyUnicode_Check(source)) {
1864 PyErr_SetString(PyExc_TypeError, "String or Unicode type required");
1865 return NULL;
1866 }
1867#if wxUSE_UNICODE
1868 PyObject* uni = source;
1869 if (PyString_Check(source)) {
1870 uni = PyUnicode_FromEncodedObject(source, wxPyDefaultEncoding, "strict");
1871 if (PyErr_Occurred()) return NULL;
1872 }
1873 target = new wxString();
1874 size_t len = PyUnicode_GET_SIZE(uni);
1875 if (len) {
1876 PyUnicode_AsWideChar((PyUnicodeObject*)uni, wxStringBuffer(*target, len), len);
1877 }
1878
1879 if (PyString_Check(source))
1880 Py_DECREF(uni);
1881#else
1882 // Convert to a string object if it isn't already, then to wxString
1883 PyObject* str = source;
1884 if (PyUnicode_Check(source)) {
1885 str = PyUnicode_AsEncodedString(source, wxPyDefaultEncoding, "strict");
1886 if (PyErr_Occurred()) return NULL;
1887 }
1888 else if (!PyString_Check(source)) {
1889 str = PyObject_Str(source);
1890 if (PyErr_Occurred()) return NULL;
1891 }
1892 char* tmpPtr; Py_ssize_t tmpSize;
1893 PyString_AsStringAndSize(str, &tmpPtr, &tmpSize);
1894 target = new wxString(tmpPtr, tmpSize);
1895
1896 if (!PyString_Check(source))
1897 Py_DECREF(str);
1898#endif // wxUSE_UNICODE
1899
1900 return target;
1901}
1902
1903
1904// Similar to above except doesn't use "new" and doesn't set an exception
1905wxString Py2wxString(PyObject* source)
1906{
1907 wxString target;
1908
1909#if wxUSE_UNICODE
1910 // Convert to a unicode object, if not already, then to a wxString
1911 PyObject* uni = source;
1912 if (!PyUnicode_Check(source)) {
1913 uni = PyUnicode_FromEncodedObject(source, wxPyDefaultEncoding, "strict");
1914 if (PyErr_Occurred()) return wxEmptyString; // TODO: should we PyErr_Clear?
1915 }
1916 size_t len = PyUnicode_GET_SIZE(uni);
1917 if (len) {
1918 PyUnicode_AsWideChar((PyUnicodeObject*)uni, wxStringBuffer(target, len), len);
1919 }
1920
1921 if (!PyUnicode_Check(source))
1922 Py_DECREF(uni);
1923#else
1924 // Convert to a string object if it isn't already, then to wxString
1925 PyObject* str = source;
1926 if (PyUnicode_Check(source)) {
1927 str = PyUnicode_AsEncodedString(source, wxPyDefaultEncoding, "strict");
1928 if (PyErr_Occurred()) return wxEmptyString; // TODO: should we PyErr_Clear?
1929 }
1930 else if (!PyString_Check(source)) {
1931 str = PyObject_Str(source);
1932 if (PyErr_Occurred()) return wxEmptyString; // TODO: should we PyErr_Clear?
1933 }
1934 char* tmpPtr; Py_ssize_t tmpSize;
1935 PyString_AsStringAndSize(str, &tmpPtr, &tmpSize);
1936 target = wxString(tmpPtr, tmpSize);
1937
1938 if (!PyString_Check(source))
1939 Py_DECREF(str);
1940#endif // wxUSE_UNICODE
1941
1942 return target;
1943}
1944
1945
1946// Make either a Python String or Unicode object, depending on build mode
1947PyObject* wx2PyString(const wxString& src)
1948{
1949 PyObject* str;
1950#if wxUSE_UNICODE
1951 str = PyUnicode_FromWideChar(src.c_str(), src.Len());
1952#else
1953 str = PyString_FromStringAndSize(src.c_str(), src.Len());
1954#endif
1955 return str;
1956}
1957
1958
1959
1960void wxSetDefaultPyEncoding(const char* encoding)
1961{
1962 strncpy(wxPyDefaultEncoding, encoding, DEFAULTENCODING_SIZE);
1963}
1964
1965const char* wxGetDefaultPyEncoding()
1966{
1967 return wxPyDefaultEncoding;
1968}
1969
1970//----------------------------------------------------------------------
1971
1972
1973byte* byte_LIST_helper(PyObject* source) {
1974 if (!PyList_Check(source)) {
1975 PyErr_SetString(PyExc_TypeError, "Expected a list object.");
1976 return NULL;
1977 }
1978 int count = PyList_Size(source);
1979 byte* temp = new byte[count];
1980 if (! temp) {
1981 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
1982 return NULL;
1983 }
1984 for (int x=0; x<count; x++) {
1985 PyObject* o = PyList_GetItem(source, x);
1986 if (! PyInt_Check(o)) {
1987 PyErr_SetString(PyExc_TypeError, "Expected a list of integers.");
1988 return NULL;
1989 }
1990 temp[x] = (byte)PyInt_AsLong(o);
1991 }
1992 return temp;
1993}
1994
1995
1996int* int_LIST_helper(PyObject* source) {
1997 if (!PyList_Check(source)) {
1998 PyErr_SetString(PyExc_TypeError, "Expected a list object.");
1999 return NULL;
2000 }
2001 int count = PyList_Size(source);
2002 int* temp = new int[count];
2003 if (! temp) {
2004 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
2005 return NULL;
2006 }
2007 for (int x=0; x<count; x++) {
2008 PyObject* o = PyList_GetItem(source, x);
2009 if (! PyInt_Check(o)) {
2010 PyErr_SetString(PyExc_TypeError, "Expected a list of integers.");
2011 return NULL;
2012 }
2013 temp[x] = PyInt_AsLong(o);
2014 }
2015 return temp;
2016}
2017
2018
2019long* long_LIST_helper(PyObject* source) {
2020 if (!PyList_Check(source)) {
2021 PyErr_SetString(PyExc_TypeError, "Expected a list object.");
2022 return NULL;
2023 }
2024 int count = PyList_Size(source);
2025 long* temp = new long[count];
2026 if (! temp) {
2027 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
2028 return NULL;
2029 }
2030 for (int x=0; x<count; x++) {
2031 PyObject* o = PyList_GetItem(source, x);
2032 if (! PyInt_Check(o)) {
2033 PyErr_SetString(PyExc_TypeError, "Expected a list of integers.");
2034 return NULL;
2035 }
2036 temp[x] = PyInt_AsLong(o);
2037 }
2038 return temp;
2039}
2040
2041
2042char** string_LIST_helper(PyObject* source) {
2043 if (!PyList_Check(source)) {
2044 PyErr_SetString(PyExc_TypeError, "Expected a list object.");
2045 return NULL;
2046 }
2047 int count = PyList_Size(source);
2048 char** temp = new char*[count];
2049 if (! temp) {
2050 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
2051 return NULL;
2052 }
2053 for (int x=0; x<count; x++) {
2054 PyObject* o = PyList_GetItem(source, x);
2055 if (! PyString_Check(o)) {
2056 PyErr_SetString(PyExc_TypeError, "Expected a list of strings.");
2057 return NULL;
2058 }
2059 temp[x] = PyString_AsString(o);
2060 }
2061 return temp;
2062}
2063
2064//---------------------------------------------------------------------------
2065// NOTE: The following functions could really use some refactoring using
2066// templates to make things somewhat simpler...
2067
2068inline bool wxPointFromObjects(PyObject* o1, PyObject* o2, wxPoint* point)
2069{
2070 // get the x value
2071 if (PyInt_Check(o1))
2072 point->x = (int)PyInt_AS_LONG(o1);
2073 else if (PyFloat_Check(o1))
2074 point->x = (int)PyFloat_AS_DOUBLE(o1);
2075 else if (PyNumber_Check(o1))
2076 point->x = (int)PyInt_AsLong(o1);
2077 else
2078 return false;
2079
2080 // get the y value
2081 if (PyInt_Check(o2))
2082 point->y = (int)PyInt_AS_LONG(o2);
2083 else if (PyFloat_Check(o2))
2084 point->y = (int)PyFloat_AS_DOUBLE(o2);
2085 else if (PyNumber_Check(o2))
2086 point->y = (int)PyInt_AsLong(o2);
2087 else
2088 return false;
2089
2090 return true;
2091
2092// NOTE: This function used to have this code in it, but I don't know why it
2093// is a problem nor what havok it will cause, so removing for now...
2094// if (wxPySwigInstance_Check(o1) || wxPySwigInstance_Check(o2)) {
2095// // Disallow instances because they can cause havok
2096// return false;
2097// }
2098}
2099
2100
2101inline bool wxPoint2DFromObjects(PyObject* o1, PyObject* o2, wxPoint2D* point)
2102{
2103 // get the x value
2104 if (PyInt_Check(o1))
2105 point->m_x = (double)PyInt_AS_LONG(o1);
2106 else if (PyFloat_Check(o1))
2107 point->m_x = (double)PyFloat_AS_DOUBLE(o1);
2108 else if (PyNumber_Check(o1))
2109 point->m_x = (double)PyFloat_AsDouble(o1);
2110 else
2111 return false;
2112
2113 // get the y value
2114 if (PyInt_Check(o2))
2115 point->m_y = (double)PyInt_AS_LONG(o2);
2116 else if (PyFloat_Check(o2))
2117 point->m_y = (double)PyFloat_AS_DOUBLE(o2);
2118 else if (PyNumber_Check(o2))
2119 point->m_y = (double)PyFloat_AsDouble(o2);
2120 else
2121 return false;
2122
2123 return true;
2124}
2125
2126
2127
2128wxPoint* wxPoint_LIST_helper(PyObject* source, int *count)
2129{
2130 int idx;
2131 wxPoint* temp;
2132 PyObject *o, *o1, *o2;
2133 bool isFast = PyList_Check(source) || PyTuple_Check(source);
2134
2135 if (!PySequence_Check(source)) {
2136 goto error0;
2137 }
2138
2139 // The length of the sequence is returned in count.
2140 *count = PySequence_Length(source);
2141 if (*count < 0) {
2142 goto error0;
2143 }
2144
2145 temp = new wxPoint[*count];
2146 if (!temp) {
2147 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
2148 return NULL;
2149 }
2150 for (idx=0; idx<*count; idx++) {
2151 // Get an item: try fast way first.
2152 if (isFast) {
2153 o = PySequence_Fast_GET_ITEM(source, idx);
2154 }
2155 else {
2156 o = PySequence_GetItem(source, idx);
2157 if (o == NULL) {
2158 goto error1;
2159 }
2160 }
2161
2162 // Convert o to wxPoint.
2163 if ((PyTuple_Check(o) && PyTuple_GET_SIZE(o) == 2) ||
2164 (PyList_Check(o) && PyList_GET_SIZE(o) == 2)) {
2165 o1 = PySequence_Fast_GET_ITEM(o, 0);
2166 o2 = PySequence_Fast_GET_ITEM(o, 1);
2167 if (!wxPointFromObjects(o1, o2, &temp[idx])) {
2168 goto error2;
2169 }
2170 }
2171 else if (wxPySwigInstance_Check(o)) {
2172 wxPoint* pt;
2173 if (! wxPyConvertSwigPtr(o, (void **)&pt, wxT("wxPoint"))) {
2174 goto error2;
2175 }
2176 temp[idx] = *pt;
2177 }
2178 else if (PySequence_Check(o) && PySequence_Length(o) == 2) {
2179 o1 = PySequence_GetItem(o, 0);
2180 o2 = PySequence_GetItem(o, 1);
2181 if (!wxPointFromObjects(o1, o2, &temp[idx])) {
2182 goto error3;
2183 }
2184 Py_DECREF(o1);
2185 Py_DECREF(o2);
2186 }
2187 else {
2188 goto error2;
2189 }
2190 // Clean up.
2191 if (!isFast)
2192 Py_DECREF(o);
2193 }
2194 return temp;
2195
2196error3:
2197 Py_DECREF(o1);
2198 Py_DECREF(o2);
2199error2:
2200 if (!isFast)
2201 Py_DECREF(o);
2202error1:
2203 delete [] temp;
2204error0:
2205 PyErr_SetString(PyExc_TypeError, "Expected a sequence of length-2 sequences or wxPoints.");
2206 return NULL;
2207}
2208
2209
2210
2211wxPoint2D* wxPoint2D_LIST_helper(PyObject* source, size_t *count)
2212{
2213 size_t idx;
2214 wxPoint2D* temp;
2215 PyObject *o, *o1, *o2;
2216 bool isFast = PyList_Check(source) || PyTuple_Check(source);
2217
2218 if (!PySequence_Check(source)) {
2219 goto error0;
2220 }
2221
2222 // The length of the sequence is returned in count.
2223 *count = PySequence_Length(source);
2224 if (*count < 0) {
2225 goto error0;
2226 }
2227
2228 temp = new wxPoint2D[*count];
2229 if (!temp) {
2230 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
2231 return NULL;
2232 }
2233 for (idx=0; idx<*count; idx++) {
2234 // Get an item: try fast way first.
2235 if (isFast) {
2236 o = PySequence_Fast_GET_ITEM(source, idx);
2237 }
2238 else {
2239 o = PySequence_GetItem(source, idx);
2240 if (o == NULL) {
2241 goto error1;
2242 }
2243 }
2244
2245 // Convert o to wxPoint.
2246 if ((PyTuple_Check(o) && PyTuple_GET_SIZE(o) == 2) ||
2247 (PyList_Check(o) && PyList_GET_SIZE(o) == 2)) {
2248 o1 = PySequence_Fast_GET_ITEM(o, 0);
2249 o2 = PySequence_Fast_GET_ITEM(o, 1);
2250 if (!wxPoint2DFromObjects(o1, o2, &temp[idx])) {
2251 goto error2;
2252 }
2253 }
2254 else if (wxPySwigInstance_Check(o)) {
2255 wxPoint2D* pt;
2256 if (! wxPyConvertSwigPtr(o, (void **)&pt, wxT("wxPoint2D"))) {
2257 goto error2;
2258 }
2259 temp[idx] = *pt;
2260 }
2261 else if (PySequence_Check(o) && PySequence_Length(o) == 2) {
2262 o1 = PySequence_GetItem(o, 0);
2263 o2 = PySequence_GetItem(o, 1);
2264 if (!wxPoint2DFromObjects(o1, o2, &temp[idx])) {
2265 goto error3;
2266 }
2267 Py_DECREF(o1);
2268 Py_DECREF(o2);
2269 }
2270 else {
2271 goto error2;
2272 }
2273 // Clean up.
2274 if (!isFast)
2275 Py_DECREF(o);
2276 }
2277 return temp;
2278
2279error3:
2280 Py_DECREF(o1);
2281 Py_DECREF(o2);
2282error2:
2283 if (!isFast)
2284 Py_DECREF(o);
2285error1:
2286 delete [] temp;
2287error0:
2288 PyErr_SetString(PyExc_TypeError, "Expected a sequence of length-2 sequences or wxPoint2Ds.");
2289 return NULL;
2290}
2291
2292//---------------------------------------------------------------------------
2293
2294
2295wxBitmap** wxBitmap_LIST_helper(PyObject* source) {
2296 if (!PyList_Check(source)) {
2297 PyErr_SetString(PyExc_TypeError, "Expected a list object.");
2298 return NULL;
2299 }
2300 int count = PyList_Size(source);
2301 wxBitmap** temp = new wxBitmap*[count];
2302 if (! temp) {
2303 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
2304 return NULL;
2305 }
2306 for (int x=0; x<count; x++) {
2307 PyObject* o = PyList_GetItem(source, x);
2308 if (wxPySwigInstance_Check(o)) {
2309 wxBitmap* pt;
2310 if (! wxPyConvertSwigPtr(o, (void **) &pt, wxT("wxBitmap"))) {
2311 PyErr_SetString(PyExc_TypeError,"Expected wxBitmap.");
2312 return NULL;
2313 }
2314 temp[x] = pt;
2315 }
2316 else {
2317 PyErr_SetString(PyExc_TypeError, "Expected a list of wxBitmaps.");
2318 return NULL;
2319 }
2320 }
2321 return temp;
2322}
2323
2324
2325
2326wxString* wxString_LIST_helper(PyObject* source) {
2327 if (!PyList_Check(source)) {
2328 PyErr_SetString(PyExc_TypeError, "Expected a list object.");
2329 return NULL;
2330 }
2331 int count = PyList_Size(source);
2332 wxString* temp = new wxString[count];
2333 if (! temp) {
2334 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
2335 return NULL;
2336 }
2337 for (int x=0; x<count; x++) {
2338 PyObject* o = PyList_GetItem(source, x);
2339#if PYTHON_API_VERSION >= 1009
2340 if (! PyString_Check(o) && ! PyUnicode_Check(o)) {
2341 PyErr_SetString(PyExc_TypeError, "Expected a list of string or unicode objects.");
2342 return NULL;
2343 }
2344#else
2345 if (! PyString_Check(o)) {
2346 PyErr_SetString(PyExc_TypeError, "Expected a list of strings.");
2347 return NULL;
2348 }
2349#endif
2350
2351 wxString* pStr = wxString_in_helper(o);
2352 temp[x] = *pStr;
2353 delete pStr;
2354 }
2355 return temp;
2356}
2357
2358
2359wxAcceleratorEntry* wxAcceleratorEntry_LIST_helper(PyObject* source) {
2360 if (!PyList_Check(source)) {
2361 PyErr_SetString(PyExc_TypeError, "Expected a list object.");
2362 return NULL;
2363 }
2364 int count = PyList_Size(source);
2365 wxAcceleratorEntry* temp = new wxAcceleratorEntry[count];
2366 if (! temp) {
2367 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
2368 return NULL;
2369 }
2370 for (int x=0; x<count; x++) {
2371 PyObject* o = PyList_GetItem(source, x);
2372 if (wxPySwigInstance_Check(o)) {
2373 wxAcceleratorEntry* ae;
2374 if (! wxPyConvertSwigPtr(o, (void **) &ae, wxT("wxAcceleratorEntry"))) {
2375 PyErr_SetString(PyExc_TypeError,"Expected wxAcceleratorEntry.");
2376 return NULL;
2377 }
2378 temp[x] = *ae;
2379 }
2380 else if (PyTuple_Check(o)) {
2381 PyObject* o1 = PyTuple_GetItem(o, 0);
2382 PyObject* o2 = PyTuple_GetItem(o, 1);
2383 PyObject* o3 = PyTuple_GetItem(o, 2);
2384 temp[x].Set(PyInt_AsLong(o1), PyInt_AsLong(o2), PyInt_AsLong(o3));
2385 }
2386 else {
2387 PyErr_SetString(PyExc_TypeError, "Expected a list of 3-tuples or wxAcceleratorEntry objects.");
2388 return NULL;
2389 }
2390 }
2391 return temp;
2392}
2393
2394
2395wxPen** wxPen_LIST_helper(PyObject* source) {
2396 if (!PyList_Check(source)) {
2397 PyErr_SetString(PyExc_TypeError, "Expected a list object.");
2398 return NULL;
2399 }
2400 int count = PyList_Size(source);
2401 wxPen** temp = new wxPen*[count];
2402 if (!temp) {
2403 PyErr_SetString(PyExc_MemoryError, "Unable to allocate temporary array");
2404 return NULL;
2405 }
2406 for (int x=0; x<count; x++) {
2407 PyObject* o = PyList_GetItem(source, x);
2408 if (wxPySwigInstance_Check(o)) {
2409 wxPen* pt;
2410 if (! wxPyConvertSwigPtr(o, (void **)&pt, wxT("wxPen"))) {
2411 delete temp;
2412 PyErr_SetString(PyExc_TypeError,"Expected wxPen.");
2413 return NULL;
2414 }
2415 temp[x] = pt;
2416 }
2417 else {
2418 delete temp;
2419 PyErr_SetString(PyExc_TypeError, "Expected a list of wxPens.");
2420 return NULL;
2421 }
2422 }
2423 return temp;
2424}
2425
2426
2427bool wxPy2int_seq_helper(PyObject* source, int* i1, int* i2) {
2428 bool isFast = PyList_Check(source) || PyTuple_Check(source);
2429 PyObject *o1, *o2;
2430
2431 if (!PySequence_Check(source) || PySequence_Length(source) != 2)
2432 return false;
2433
2434 if (isFast) {
2435 o1 = PySequence_Fast_GET_ITEM(source, 0);
2436 o2 = PySequence_Fast_GET_ITEM(source, 1);
2437 }
2438 else {
2439 o1 = PySequence_GetItem(source, 0);
2440 o2 = PySequence_GetItem(source, 1);
2441 }
2442
2443 *i1 = PyInt_AsLong(o1);
2444 *i2 = PyInt_AsLong(o2);
2445
2446 if (! isFast) {
2447 Py_DECREF(o1);
2448 Py_DECREF(o2);
2449 }
2450 return true;
2451}
2452
2453
2454bool wxPy4int_seq_helper(PyObject* source, int* i1, int* i2, int* i3, int* i4) {
2455 bool isFast = PyList_Check(source) || PyTuple_Check(source);
2456 PyObject *o1, *o2, *o3, *o4;
2457
2458 if (!PySequence_Check(source) || PySequence_Length(source) != 4)
2459 return false;
2460
2461 if (isFast) {
2462 o1 = PySequence_Fast_GET_ITEM(source, 0);
2463 o2 = PySequence_Fast_GET_ITEM(source, 1);
2464 o3 = PySequence_Fast_GET_ITEM(source, 2);
2465 o4 = PySequence_Fast_GET_ITEM(source, 3);
2466 }
2467 else {
2468 o1 = PySequence_GetItem(source, 0);
2469 o2 = PySequence_GetItem(source, 1);
2470 o3 = PySequence_GetItem(source, 2);
2471 o4 = PySequence_GetItem(source, 3);
2472 }
2473
2474 *i1 = PyInt_AsLong(o1);
2475 *i2 = PyInt_AsLong(o2);
2476 *i3 = PyInt_AsLong(o3);
2477 *i4 = PyInt_AsLong(o4);
2478
2479 if (! isFast) {
2480 Py_DECREF(o1);
2481 Py_DECREF(o2);
2482 Py_DECREF(o3);
2483 Py_DECREF(o4);
2484 }
2485 return true;
2486}
2487
2488
2489//----------------------------------------------------------------------
2490
2491bool wxPySimple_typecheck(PyObject* source, const wxChar* classname, int seqLen)
2492{
2493 void* ptr;
2494
2495 if (wxPySwigInstance_Check(source) &&
2496 wxPyConvertSwigPtr(source, (void **)&ptr, classname))
2497 return true;
2498
2499 PyErr_Clear();
2500 if (PySequence_Check(source) && PySequence_Length(source) == seqLen)
2501 return true;
2502
2503 return false;
2504}
2505
2506bool wxSize_helper(PyObject* source, wxSize** obj)
2507{
2508 if (source == Py_None) {
2509 **obj = wxSize(-1,-1);
2510 return true;
2511 }
2512 return wxPyTwoIntItem_helper(source, obj, wxT("wxSize"));
2513}
2514
2515
2516bool wxPoint_helper(PyObject* source, wxPoint** obj)
2517{
2518 if (source == Py_None) {
2519 **obj = wxPoint(-1,-1);
2520 return true;
2521 }
2522 return wxPyTwoIntItem_helper(source, obj, wxT("wxPoint"));
2523}
2524
2525
2526
2527bool wxRealPoint_helper(PyObject* source, wxRealPoint** obj) {
2528
2529 if (source == Py_None) {
2530 **obj = wxRealPoint(-1,-1);
2531 return true;
2532 }
2533
2534 // If source is an object instance then it may already be the right type
2535 if (wxPySwigInstance_Check(source)) {
2536 wxRealPoint* ptr;
2537 if (! wxPyConvertSwigPtr(source, (void **)&ptr, wxT("wxRealPoint")))
2538 goto error;
2539 *obj = ptr;
2540 return true;
2541 }
2542 // otherwise a 2-tuple of floats is expected
2543 else if (PySequence_Check(source) && PyObject_Length(source) == 2) {
2544 PyObject* o1 = PySequence_GetItem(source, 0);
2545 PyObject* o2 = PySequence_GetItem(source, 1);
2546 if (!PyNumber_Check(o1) || !PyNumber_Check(o2)) {
2547 Py_DECREF(o1);
2548 Py_DECREF(o2);
2549 goto error;
2550 }
2551 **obj = wxRealPoint(PyFloat_AsDouble(o1), PyFloat_AsDouble(o2));
2552 Py_DECREF(o1);
2553 Py_DECREF(o2);
2554 return true;
2555 }
2556
2557 error:
2558 PyErr_SetString(PyExc_TypeError, "Expected a 2-tuple of floats or a wxRealPoint object.");
2559 return false;
2560}
2561
2562
2563
2564bool wxRect_helper(PyObject* source, wxRect** obj) {
2565
2566 if (source == Py_None) {
2567 **obj = wxRect(-1,-1,-1,-1);
2568 return true;
2569 }
2570
2571 // If source is an object instance then it may already be the right type
2572 if (wxPySwigInstance_Check(source)) {
2573 wxRect* ptr;
2574 if (! wxPyConvertSwigPtr(source, (void **)&ptr, wxT("wxRect")))
2575 goto error;
2576 *obj = ptr;
2577 return true;
2578 }
2579 // otherwise a 4-tuple of integers is expected
2580 else if (PySequence_Check(source) && PyObject_Length(source) == 4) {
2581 PyObject* o1 = PySequence_GetItem(source, 0);
2582 PyObject* o2 = PySequence_GetItem(source, 1);
2583 PyObject* o3 = PySequence_GetItem(source, 2);
2584 PyObject* o4 = PySequence_GetItem(source, 3);
2585 if (!PyNumber_Check(o1) || !PyNumber_Check(o2) ||
2586 !PyNumber_Check(o3) || !PyNumber_Check(o4)) {
2587 Py_DECREF(o1);
2588 Py_DECREF(o2);
2589 Py_DECREF(o3);
2590 Py_DECREF(o4);
2591 goto error;
2592 }
2593 **obj = wxRect(PyInt_AsLong(o1), PyInt_AsLong(o2),
2594 PyInt_AsLong(o3), PyInt_AsLong(o4));
2595 Py_DECREF(o1);
2596 Py_DECREF(o2);
2597 Py_DECREF(o3);
2598 Py_DECREF(o4);
2599 return true;
2600 }
2601
2602 error:
2603 PyErr_SetString(PyExc_TypeError, "Expected a 4-tuple of integers or a wxRect object.");
2604 return false;
2605}
2606
2607
2608
2609bool wxColour_helper(PyObject* source, wxColour** obj) {
2610
2611 if (source == Py_None) {
2612 **obj = wxNullColour;
2613 return true;
2614 }
2615
2616 // If source is an object instance then it may already be the right type
2617 if (wxPySwigInstance_Check(source)) {
2618 wxColour* ptr;
2619 if (! wxPyConvertSwigPtr(source, (void **)&ptr, wxT("wxColour")))
2620 goto error;
2621 *obj = ptr;
2622 return true;
2623 }
2624 // otherwise check for a string
2625 else if (PyString_Check(source) || PyUnicode_Check(source)) {
2626 wxString spec = Py2wxString(source);
2627 if (spec.GetChar(0) == '#' && spec.Length() == 7) { // It's #RRGGBB
2628 long red, green, blue;
2629 red = green = blue = 0;
2630 spec.Mid(1,2).ToLong(&red, 16);
2631 spec.Mid(3,2).ToLong(&green, 16);
2632 spec.Mid(5,2).ToLong(&blue, 16);
2633
2634 **obj = wxColour(red, green, blue);
2635 return true;
2636 }
2637 else { // it's a colour name
2638 **obj = wxColour(spec);
2639 return true;
2640 }
2641 }
2642 // last chance: 3-tuple or 4-tuple of integers is expected
2643 else if (PySequence_Check(source) && PyObject_Length(source) == 3) {
2644 PyObject* o1 = PySequence_GetItem(source, 0);
2645 PyObject* o2 = PySequence_GetItem(source, 1);
2646 PyObject* o3 = PySequence_GetItem(source, 2);
2647 if (!PyNumber_Check(o1) || !PyNumber_Check(o2) || !PyNumber_Check(o3)) {
2648 Py_DECREF(o1);
2649 Py_DECREF(o2);
2650 Py_DECREF(o3);
2651 goto error;
2652 }
2653 **obj = wxColour(PyInt_AsLong(o1), PyInt_AsLong(o2), PyInt_AsLong(o3));
2654 Py_DECREF(o1);
2655 Py_DECREF(o2);
2656 Py_DECREF(o3);
2657 return true;
2658 }
2659 else if (PySequence_Check(source) && PyObject_Length(source) == 4) {
2660 PyObject* o1 = PySequence_GetItem(source, 0);
2661 PyObject* o2 = PySequence_GetItem(source, 1);
2662 PyObject* o3 = PySequence_GetItem(source, 2);
2663 PyObject* o4 = PySequence_GetItem(source, 3);
2664 if (!PyNumber_Check(o1) || !PyNumber_Check(o2) || !PyNumber_Check(o3) || !PyNumber_Check(o4)) {
2665 Py_DECREF(o1);
2666 Py_DECREF(o2);
2667 Py_DECREF(o3);
2668 Py_DECREF(o4);
2669 goto error;
2670 }
2671 **obj = wxColour(PyInt_AsLong(o1), PyInt_AsLong(o2), PyInt_AsLong(o3), PyInt_AsLong(o4));
2672 Py_DECREF(o1);
2673 Py_DECREF(o2);
2674 Py_DECREF(o3);
2675 Py_DECREF(o4);
2676 return true;
2677 }
2678
2679 error:
2680 PyErr_SetString(PyExc_TypeError,
2681 "Expected a wxColour object, a string containing a colour name or '#RRGGBB', or a 3- or 4-tuple of integers.");
2682 return false;
2683}
2684
2685
2686bool wxColour_typecheck(PyObject* source) {
2687
2688 if (wxPySimple_typecheck(source, wxT("wxColour"), 3))
2689 return true;
2690
2691 if (PyString_Check(source) || PyUnicode_Check(source))
2692 return true;
2693
2694 return false;
2695}
2696
2697
2698
2699bool wxPoint2D_helper(PyObject* source, wxPoint2D** obj) {
2700
2701 if (source == Py_None) {
2702 **obj = wxPoint2D(-1,-1);
2703 return true;
2704 }
2705
2706 // If source is an object instance then it may already be the right type
2707 if (wxPySwigInstance_Check(source)) {
2708 wxPoint2D* ptr;
2709 if (! wxPyConvertSwigPtr(source, (void **)&ptr, wxT("wxPoint2D")))
2710 goto error;
2711 *obj = ptr;
2712 return true;
2713 }
2714 // otherwise a length-2 sequence of floats is expected
2715 if (PySequence_Check(source) && PySequence_Length(source) == 2) {
2716 PyObject* o1 = PySequence_GetItem(source, 0);
2717 PyObject* o2 = PySequence_GetItem(source, 1);
2718 // This should really check for floats, not numbers -- but that would break code.
2719 if (!PyNumber_Check(o1) || !PyNumber_Check(o2)) {
2720 Py_DECREF(o1);
2721 Py_DECREF(o2);
2722 goto error;
2723 }
2724 **obj = wxPoint2D(PyFloat_AsDouble(o1), PyFloat_AsDouble(o2));
2725 Py_DECREF(o1);
2726 Py_DECREF(o2);
2727 return true;
2728 }
2729 error:
2730 PyErr_SetString(PyExc_TypeError, "Expected a 2-tuple of floats or a wx.Point2D object.");
2731 return false;
2732}
2733
2734
2735
2736bool wxRect2D_helper(PyObject* source, wxRect2D** obj) {
2737
2738 if (source == Py_None) {
2739 **obj = wxRect2D(-1,-1,-1,-1);
2740 return true;
2741 }
2742
2743 // If source is an object instance then it may already be the right type
2744 if (wxPySwigInstance_Check(source)) {
2745 wxRect2D* ptr;
2746 if (! wxPyConvertSwigPtr(source, (void **)&ptr, wxT("wxRect2D")))
2747 goto error;
2748 *obj = ptr;
2749 return true;
2750 }
2751 // otherwise a length-4 sequence of floats is expected
2752 if (PySequence_Check(source) && PySequence_Length(source) == 4) {
2753 PyObject* o1 = PySequence_GetItem(source, 0);
2754 PyObject* o2 = PySequence_GetItem(source, 1);
2755 PyObject* o3 = PySequence_GetItem(source, 2);
2756 PyObject* o4 = PySequence_GetItem(source, 3);
2757 // This should really check for floats, not numbers -- but that would break code.
2758 if (!PyNumber_Check(o1) || !PyNumber_Check(o2) ||
2759 !PyNumber_Check(o3) || !PyNumber_Check(o4)) {
2760 Py_DECREF(o1);
2761 Py_DECREF(o2);
2762 Py_DECREF(o3);
2763 Py_DECREF(o4);
2764 goto error;
2765 }
2766 **obj = wxRect2D(PyFloat_AsDouble(o1), PyFloat_AsDouble(o2),
2767 PyFloat_AsDouble(o3), PyFloat_AsDouble(o4));
2768 Py_DECREF(o1);
2769 Py_DECREF(o2);
2770 Py_DECREF(o3);
2771 Py_DECREF(o4);
2772 return true;
2773 }
2774 error:
2775 PyErr_SetString(PyExc_TypeError, "Expected a 4-tuple of floats or a wx.Rect2D object.");
2776 return false;
2777}
2778
2779
2780//----------------------------------------------------------------------
2781
2782PyObject* wxArrayString2PyList_helper(const wxArrayString& arr)
2783{
2784 PyObject* list = PyList_New(0);
2785 for (size_t i=0; i < arr.GetCount(); i++) {
2786#if wxUSE_UNICODE
2787 PyObject* str = PyUnicode_FromWideChar(arr[i].c_str(), arr[i].Len());
2788#else
2789 PyObject* str = PyString_FromStringAndSize(arr[i].c_str(), arr[i].Len());
2790#endif
2791 PyList_Append(list, str);
2792 Py_DECREF(str);
2793 }
2794 return list;
2795}
2796
2797
2798PyObject* wxArrayInt2PyList_helper(const wxArrayInt& arr)
2799{
2800 PyObject* list = PyList_New(0);
2801 for (size_t i=0; i < arr.GetCount(); i++) {
2802 PyObject* number = PyInt_FromLong(arr[i]);
2803 PyList_Append(list, number);
2804 Py_DECREF(number);
2805 }
2806 return list;
2807}
2808
2809
2810PyObject* wxArrayDouble2PyList_helper(const wxArrayDouble& arr)
2811{
2812 PyObject* list = PyList_New(0);
2813 for (size_t i=0; i < arr.GetCount(); i++) {
2814 PyObject* number = PyFloat_FromDouble(arr[i]);
2815 PyList_Append(list, number);
2816 Py_DECREF(number);
2817 }
2818 return list;
2819}
2820
2821
2822//----------------------------------------------------------------------
2823// wxPyImageHandler methods
2824//
2825// TODO: Switch these to use wxPython's standard macros and helper classes
2826// for calling callbacks.
2827
2828PyObject* wxPyImageHandler::m_DoCanRead_Name = NULL;
2829PyObject* wxPyImageHandler::m_GetImageCount_Name = NULL;
2830PyObject* wxPyImageHandler::m_LoadFile_Name = NULL;
2831PyObject* wxPyImageHandler::m_SaveFile_Name = NULL;
2832
2833PyObject* wxPyImageHandler::py_InputStream(wxInputStream* stream) {
2834 return wxPyConstructObject(new wxPyInputStream(stream),
2835 wxT("wxPyInputStream"), 0);
2836}
2837
2838PyObject* wxPyImageHandler::py_Image(wxImage* image) {
2839 return wxPyConstructObject(image, wxT("wxImage"), 0);
2840}
2841
2842PyObject* wxPyImageHandler::py_OutputStream(wxOutputStream* stream) {
2843 return wxPyConstructObject(stream, wxT("wxOutputStream"), 0);
2844}
2845
2846wxPyImageHandler::wxPyImageHandler():
2847 m_self(NULL)
2848{
2849 if (!m_DoCanRead_Name) {
2850 m_DoCanRead_Name = PyString_FromString("DoCanRead");
2851 m_GetImageCount_Name = PyString_FromString("GetImageCount");
2852 m_LoadFile_Name = PyString_FromString("LoadFile");
2853 m_SaveFile_Name = PyString_FromString("SaveFile");
2854 }
2855}
2856
2857wxPyImageHandler::~wxPyImageHandler() {
2858 if (m_self) {
2859 Py_DECREF(m_self);
2860 m_self = NULL;
2861 }
2862}
2863
2864void wxPyImageHandler::_SetSelf(PyObject *self) {
2865 // should check here for isinstance(PyImageHandler) ??
2866 m_self = self;
2867 Py_INCREF(m_self);
2868}
2869
2870bool wxPyImageHandler::DoCanRead(wxInputStream& stream) {
2871 // check if our object has this method
2872 wxPyBlock_t blocked = wxPyBeginBlockThreads();
2873 if (!m_self || !PyObject_HasAttr(m_self, m_DoCanRead_Name)) {
2874 wxPyEndBlockThreads(blocked);
2875 return false;
2876 }
2877
2878 PyObject* res = PyObject_CallMethodObjArgs(m_self, m_DoCanRead_Name,
2879 py_InputStream(&stream), NULL);
2880 bool retval = false;
2881 if (res) {
2882 retval = PyInt_AsLong(res);
2883 Py_DECREF(res);
2884 PyErr_Clear();
2885 }
2886 else
2887 PyErr_Print();
2888 wxPyEndBlockThreads(blocked);
2889 return retval;
2890}
2891
2892bool wxPyImageHandler::LoadFile( wxImage* image, wxInputStream& stream,
2893 bool verbose, int index ) {
2894 // check if our object has this method
2895 wxPyBlock_t blocked = wxPyBeginBlockThreads();
2896 if (!m_self || !PyObject_HasAttr(m_self, m_LoadFile_Name)) {
2897 wxPyEndBlockThreads(blocked);
2898 return false;
2899 }
2900 PyObject* res = PyObject_CallMethodObjArgs(m_self, m_LoadFile_Name,
2901 py_Image(image),
2902 py_InputStream(&stream),
2903 PyInt_FromLong(verbose),
2904 PyInt_FromLong(index),
2905 NULL);
2906 bool retval = false;
2907 if (res) {
2908 retval = PyInt_AsLong(res);
2909 Py_DECREF(res);
2910 PyErr_Clear();
2911 } else
2912 PyErr_Print();
2913 wxPyEndBlockThreads(blocked);
2914 return retval;
2915}
2916
2917bool wxPyImageHandler::SaveFile( wxImage* image, wxOutputStream& stream,
2918 bool verbose ) {
2919 wxPyBlock_t blocked = wxPyBeginBlockThreads();
2920 if (!m_self || !PyObject_HasAttr(m_self, m_SaveFile_Name)) {
2921 wxPyEndBlockThreads(blocked);
2922 return false;
2923 }
2924 PyObject* res = PyObject_CallMethodObjArgs(m_self, m_SaveFile_Name,
2925 py_Image(image),
2926 py_OutputStream(&stream),
2927 PyInt_FromLong(verbose),
2928 NULL);
2929 bool retval = false;
2930 if(res) {
2931 retval=PyInt_AsLong(res);
2932 Py_DECREF(res);
2933 PyErr_Clear();
2934 } else
2935 PyErr_Print();
2936 wxPyEndBlockThreads(blocked);
2937 return retval;
2938}
2939
2940int wxPyImageHandler::GetImageCount( wxInputStream& stream ) {
2941 wxPyBlock_t blocked = wxPyBeginBlockThreads();
2942 if (!m_self || !PyObject_HasAttr(m_self, m_GetImageCount_Name)) {
2943 wxPyEndBlockThreads(blocked);
2944 return 1;
2945 }
2946 PyObject *res=PyObject_CallMethodObjArgs(m_self, m_GetImageCount_Name,
2947 py_InputStream(&stream),
2948 NULL);
2949 int retval = 1;
2950 if(res) {
2951 retval=PyInt_AsLong(res);
2952 Py_DECREF(res);
2953 PyErr_Clear();
2954 } else
2955 PyErr_Print();
2956 wxPyEndBlockThreads(blocked);
2957 return retval;
2958}
2959
2960
2961//----------------------------------------------------------------------
2962// Function to test if the Display (or whatever is the platform equivallent)
2963// can be connected to. This is accessable from wxPython as a staticmethod of
2964// wx.App called DisplayAvailable().
2965
2966
2967bool wxPyTestDisplayAvailable()
2968{
2969#ifdef __WXGTK__
2970 Display* display;
2971 display = XOpenDisplay(NULL);
2972 if (display == NULL)
2973 return false;
2974 XCloseDisplay(display);
2975 return true;
2976#endif
2977
2978#ifdef __WXMAC__
2979 // This is adapted from Python's Mac/Modules/MacOS.c in the
2980 // MacOS_WMAvailable function.
2981 bool rv;
2982 ProcessSerialNumber psn;
2983
2984 /*
2985 ** This is a fairly innocuous call to make if we don't have a window
2986 ** manager, or if we have no permission to talk to it. It will print
2987 ** a message on stderr, but at least it won't abort the process.
2988 ** It appears the function caches the result itself, and it's cheap, so
2989 ** no need for us to cache.
2990 */
2991#ifdef kCGNullDirectDisplay
2992 /* On 10.1 CGMainDisplayID() isn't available, and
2993 ** kCGNullDirectDisplay isn't defined.
2994 */
2995 if (CGMainDisplayID() == 0) {
2996 rv = false;
2997 } else
2998#endif
2999 {
3000 // Also foreground the application on the first call as a side-effect.
3001 if (GetCurrentProcess(&psn) < 0 || SetFrontProcess(&psn) < 0) {
3002 rv = false;
3003 } else {
3004 rv = true;
3005 }
3006 }
3007 return rv;
3008#endif
3009
3010#ifdef __WXMSW__
3011 // TODO...
3012 return true;
3013#endif
3014}
3015
3016
3017//----------------------------------------------------------------------
3018//----------------------------------------------------------------------
3019
3020
3021
3022