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