blob: b5af867acf906c6da3813c52026119f4ece64335 [file] [log] [blame]
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001/*
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002 * This is a curses module for Python.
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003 *
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00004 * Based on prior work by Lance Ellinghaus and Oliver Andrich
5 * Version 1.2 of this module: Copyright 1994 by Lance Ellinghouse,
6 * Cathedral City, California Republic, United States of America.
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00007 *
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00008 * Version 1.5b1, heavily extended for ncurses by Oliver Andrich:
9 * Copyright 1996,1997 by Oliver Andrich, Koblenz, Germany.
10 *
Andrew M. Kuchling3adefcc2002-10-30 21:08:34 +000011 * Tidied for Python 1.6, and currently maintained by <amk@amk.ca>.
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +000012 *
13 * Permission is hereby granted, free of charge, to any person obtaining
14 * a copy of this source file to use, copy, modify, merge, or publish it
15 * subject to the following conditions:
16 *
17 * The above copyright notice and this permission notice shall be included
18 * in all copies or in any new file that contains a substantial portion of
19 * this file.
20 *
21 * THE AUTHOR MAKES NO REPRESENTATIONS ABOUT THE SUITABILITY OF
22 * THE SOFTWARE FOR ANY PURPOSE. IT IS PROVIDED "AS IS" WITHOUT
23 * EXPRESS OR IMPLIED WARRANTY. THE AUTHOR DISCLAIMS ALL WARRANTIES
24 * WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
25 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
26 * NON-INFRINGEMENT OF THIRD PARTY RIGHTS. IN NO EVENT SHALL THE
27 * AUTHOR BE LIABLE TO YOU OR ANY OTHER PARTY FOR ANY SPECIAL,
28 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
29 * WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE, STRICT LIABILITY OR
30 * ANY OTHER ACTION ARISING OUT OF OR IN CONNECTION WITH THE USE OR
31 * PERFORMANCE OF THIS SOFTWARE.
32 */
Guido van Rossumf6971e21994-08-30 12:25:20 +000033
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000034/*
35
Victor Stinner26486ea2010-05-15 22:23:53 +000036 A number of SysV or ncurses functions don't have wrappers yet; if you
37 need a given function, add it and send a patch. See
38 http://www.python.org/dev/patches/ for instructions on how to submit
39 patches to Python.
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000040
Victor Stinner26486ea2010-05-15 22:23:53 +000041 Here's a list of currently unsupported functions:
Guido van Rossumd8faa362007-04-27 19:54:29 +000042
Victor Stinner26486ea2010-05-15 22:23:53 +000043 addchnstr addchstr color_set define_key
44 del_curterm delscreen dupwin inchnstr inchstr innstr keyok
45 mcprint mvaddchnstr mvaddchstr mvcur mvinchnstr
46 mvinchstr mvinnstr mmvwaddchnstr mvwaddchstr
47 mvwinchnstr mvwinchstr mvwinnstr newterm
48 restartterm ripoffline scr_dump
49 scr_init scr_restore scr_set scrl set_curterm set_term setterm
50 tgetent tgetflag tgetnum tgetstr tgoto timeout tputs
51 vidattr vidputs waddchnstr waddchstr
52 wcolor_set winchnstr winchstr winnstr wmouse_trafo wscrl
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000053
Victor Stinner26486ea2010-05-15 22:23:53 +000054 Low-priority:
55 slk_attr slk_attr_off slk_attr_on slk_attr_set slk_attroff
56 slk_attron slk_attrset slk_clear slk_color slk_init slk_label
57 slk_noutrefresh slk_refresh slk_restore slk_set slk_touch
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000058
Victor Stinner26486ea2010-05-15 22:23:53 +000059 Menu extension (ncurses and probably SYSV):
60 current_item free_item free_menu item_count item_description
61 item_index item_init item_name item_opts item_opts_off
62 item_opts_on item_term item_userptr item_value item_visible
63 menu_back menu_driver menu_fore menu_format menu_grey
64 menu_init menu_items menu_mark menu_opts menu_opts_off
65 menu_opts_on menu_pad menu_pattern menu_request_by_name
66 menu_request_name menu_spacing menu_sub menu_term menu_userptr
67 menu_win new_item new_menu pos_menu_cursor post_menu
68 scale_menu set_current_item set_item_init set_item_opts
69 set_item_term set_item_userptr set_item_value set_menu_back
70 set_menu_fore set_menu_format set_menu_grey set_menu_init
71 set_menu_items set_menu_mark set_menu_opts set_menu_pad
72 set_menu_pattern set_menu_spacing set_menu_sub set_menu_term
73 set_menu_userptr set_menu_win set_top_row top_row unpost_menu
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +000074
Victor Stinner26486ea2010-05-15 22:23:53 +000075 Form extension (ncurses and probably SYSV):
76 current_field data_ahead data_behind dup_field
77 dynamic_fieldinfo field_arg field_back field_buffer
78 field_count field_fore field_index field_info field_init
79 field_just field_opts field_opts_off field_opts_on field_pad
80 field_status field_term field_type field_userptr form_driver
81 form_fields form_init form_opts form_opts_off form_opts_on
82 form_page form_request_by_name form_request_name form_sub
83 form_term form_userptr form_win free_field free_form
84 link_field link_fieldtype move_field new_field new_form
85 new_page pos_form_cursor post_form scale_form
86 set_current_field set_field_back set_field_buffer
87 set_field_fore set_field_init set_field_just set_field_opts
88 set_field_pad set_field_status set_field_term set_field_type
89 set_field_userptr set_fieldtype_arg set_fieldtype_choice
90 set_form_fields set_form_init set_form_opts set_form_page
91 set_form_sub set_form_term set_form_userptr set_form_win
92 set_max_field set_new_page unpost_form
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +000093
94
Victor Stinner26486ea2010-05-15 22:23:53 +000095*/
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000096
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +000097/* Release Number */
Guido van Rossumf6971e21994-08-30 12:25:20 +000098
Andrew M. Kuchling16e65a02001-10-20 16:05:52 +000099char *PyCursesVersion = "2.2";
Guido van Rossumf6971e21994-08-30 12:25:20 +0000100
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000101/* Includes */
Guido van Rossumf6971e21994-08-30 12:25:20 +0000102
Mark Dickinsone047cfa2010-05-11 17:57:09 +0000103#define PY_SSIZE_T_CLEAN
104
Guido van Rossum602099a1994-09-14 13:32:22 +0000105#include "Python.h"
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000106
Benjamin Petersonb173f782009-05-05 22:31:58 +0000107
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000108#ifdef __hpux
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000109#define STRICT_SYSV_CURSES
110#endif
111
Guido van Rossum858ca0f2001-04-10 19:53:37 +0000112#define CURSES_MODULE
113#include "py_curses.h"
114
Victor Stinner26486ea2010-05-15 22:23:53 +0000115/* These prototypes are in <term.h>, but including this header
116 #defines many common symbols (such as "lines") which breaks the
117 curses module in other ways. So the code will just specify
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +0000118 explicit prototypes here. */
119extern int setupterm(char *,int,int *);
Fred Drake0368bc42001-07-19 20:48:32 +0000120#ifdef __sgi
121#include <term.h>
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +0000122#endif
123
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100124#ifdef HAVE_LANGINFO_H
125#include <langinfo.h>
126#endif
127
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000128#if !defined(HAVE_NCURSES_H) && (defined(sgi) || defined(__sun) || defined(SCO5))
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +0000129#define STRICT_SYSV_CURSES /* Don't use ncurses extensions */
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +0000130typedef chtype attr_t; /* No attr_t type is available */
Guido van Rossum1266a011996-02-25 04:50:31 +0000131#endif
Guido van Rossumf6971e21994-08-30 12:25:20 +0000132
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000133#if defined(_AIX)
134#define STRICT_SYSV_CURSES
135#endif
136
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000137/* Definition of exception curses.error */
Guido van Rossumf6971e21994-08-30 12:25:20 +0000138
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000139static PyObject *PyCursesError;
Guido van Rossumf6971e21994-08-30 12:25:20 +0000140
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +0000141/* Tells whether setupterm() has been called to initialise terminfo. */
142static int initialised_setupterm = FALSE;
143
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000144/* Tells whether initscr() has been called to initialise curses. */
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000145static int initialised = FALSE;
146
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000147/* Tells whether start_color() has been called to initialise color usage. */
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000148static int initialisedcolors = FALSE;
149
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100150static char *screen_encoding = NULL;
151
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000152/* Utility Macros */
Victor Stinner26486ea2010-05-15 22:23:53 +0000153#define PyCursesSetupTermCalled \
154 if (initialised_setupterm != TRUE) { \
155 PyErr_SetString(PyCursesError, \
156 "must call (at least) setupterm() first"); \
157 return 0; }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +0000158
Victor Stinner26486ea2010-05-15 22:23:53 +0000159#define PyCursesInitialised \
160 if (initialised != TRUE) { \
161 PyErr_SetString(PyCursesError, \
162 "must call initscr() first"); \
163 return 0; }
Guido van Rossumfbea2f31994-08-31 22:05:27 +0000164
Victor Stinner26486ea2010-05-15 22:23:53 +0000165#define PyCursesInitialisedColor \
166 if (initialisedcolors != TRUE) { \
167 PyErr_SetString(PyCursesError, \
168 "must call start_color() first"); \
169 return 0; }
Guido van Rossumfbea2f31994-08-31 22:05:27 +0000170
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000171/* Utility Functions */
Guido van Rossumfbea2f31994-08-31 22:05:27 +0000172
Guido van Rossum85738471995-02-17 13:50:17 +0000173/*
Victor Stinner26486ea2010-05-15 22:23:53 +0000174 * Check the return code from a curses function and return None
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000175 * or raise an exception as appropriate. These are exported using the
Victor Stinner26486ea2010-05-15 22:23:53 +0000176 * capsule API.
Guido van Rossum85738471995-02-17 13:50:17 +0000177 */
178
179static PyObject *
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100180PyCursesCheckERR(int code, const char *fname)
Guido van Rossum85738471995-02-17 13:50:17 +0000181{
Victor Stinner26486ea2010-05-15 22:23:53 +0000182 if (code != ERR) {
183 Py_INCREF(Py_None);
184 return Py_None;
Guido van Rossum85738471995-02-17 13:50:17 +0000185 } else {
Victor Stinner26486ea2010-05-15 22:23:53 +0000186 if (fname == NULL) {
187 PyErr_SetString(PyCursesError, catchall_ERR);
188 } else {
189 PyErr_Format(PyCursesError, "%s() returned ERR", fname);
190 }
191 return NULL;
Guido van Rossum85738471995-02-17 13:50:17 +0000192 }
Guido van Rossum85738471995-02-17 13:50:17 +0000193}
194
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100195/* Convert an object to a byte (an integer of type chtype):
196
197 - int
198 - bytes of length 1
199 - str of length 1
200
201 Return 1 on success, 0 on error (invalid type or integer overflow). */
Victor Stinner26486ea2010-05-15 22:23:53 +0000202static int
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100203PyCurses_ConvertToChtype(PyCursesWindowObject *win, PyObject *obj, chtype *ch)
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000204{
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100205 long value;
206 if(PyBytes_Check(obj) && PyBytes_Size(obj) == 1) {
207 value = (unsigned char)PyBytes_AsString(obj)[0];
208 }
209 else if (PyUnicode_Check(obj)) {
210 if (PyUnicode_GetLength(obj) != 1) {
211 PyErr_Format(PyExc_TypeError,
212 "expect bytes or str of length 1, or int, "
213 "got a str of length %zi",
214 PyUnicode_GET_LENGTH(obj));
Victor Stinner26486ea2010-05-15 22:23:53 +0000215 return 0;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100216 }
217 value = PyUnicode_READ_CHAR(obj, 0);
218 if (128 < value) {
219 PyObject *bytes;
220 const char *encoding;
221 if (win)
222 encoding = win->encoding;
223 else
224 encoding = screen_encoding;
225 bytes = PyUnicode_AsEncodedObject(obj, encoding, NULL);
226 if (bytes == NULL)
227 return 0;
228 if (PyBytes_GET_SIZE(bytes) == 1)
229 value = (unsigned char)PyBytes_AS_STRING(bytes)[0];
230 else
231 value = -1;
232 Py_DECREF(bytes);
233 if (value < 0)
234 goto overflow;
235 }
236 }
237 else if (PyLong_CheckExact(obj)) {
238 int long_overflow;
239 value = PyLong_AsLongAndOverflow(obj, &long_overflow);
240 if (long_overflow)
241 goto overflow;
242 }
243 else {
244 PyErr_Format(PyExc_TypeError,
245 "expect bytes or str of length 1, or int, got %s",
246 Py_TYPE(obj)->tp_name);
Victor Stinner26486ea2010-05-15 22:23:53 +0000247 return 0;
248 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100249 *ch = (chtype)value;
250 if ((long)*ch != value)
251 goto overflow;
Victor Stinner26486ea2010-05-15 22:23:53 +0000252 return 1;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100253
254overflow:
255 PyErr_SetString(PyExc_OverflowError,
256 "byte doesn't fit in chtype");
257 return 0;
258}
259
260/* Convert an object to a byte (chtype) or a character (cchar_t):
261
262 - int
263 - bytes of length 1
264 - str of length 1
265
266 Return:
267
268 - 2 if obj is a character (written into *wch)
269 - 1 if obj is a byte (written into *ch)
270 - 0 on error: raise an exception */
271static int
272PyCurses_ConvertToCchar_t(PyCursesWindowObject *win, PyObject *obj,
273 chtype *ch
274#ifdef HAVE_NCURSESW
275 , cchar_t *wch
276#endif
277 )
278{
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100279 long value;
280#ifdef HAVE_NCURSESW
281 wchar_t buffer[2];
282#endif
283
284 if (PyUnicode_Check(obj)) {
285#ifdef HAVE_NCURSESW
286 if (PyUnicode_AsWideChar(obj, buffer, 2) != 1) {
287 PyErr_Format(PyExc_TypeError,
288 "expect bytes or str of length 1, or int, "
289 "got a str of length %zi",
290 PyUnicode_GET_LENGTH(obj));
291 return 0;
292 }
293 memset(wch->chars, 0, sizeof(wch->chars));
294 wch->chars[0] = buffer[0];
295 return 2;
296#else
297 return PyCurses_ConvertToChtype(win, obj, ch);
298#endif
299 }
300 else if(PyBytes_Check(obj) && PyBytes_Size(obj) == 1) {
301 value = (unsigned char)PyBytes_AsString(obj)[0];
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100302 }
303 else if (PyLong_CheckExact(obj)) {
304 int overflow;
305 value = PyLong_AsLongAndOverflow(obj, &overflow);
306 if (overflow) {
307 PyErr_SetString(PyExc_OverflowError,
308 "int doesn't fit in long");
309 return 0;
310 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100311 }
312 else {
313 PyErr_Format(PyExc_TypeError,
314 "expect bytes or str of length 1, or int, got %s",
315 Py_TYPE(obj)->tp_name);
316 return 0;
317 }
Victor Stinner9f16e442012-09-01 15:00:34 +0200318
319 *ch = (chtype)value;
320 if ((long)*ch != value) {
321 PyErr_Format(PyExc_OverflowError,
322 "byte doesn't fit in chtype");
323 return 0;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100324 }
Victor Stinner9f16e442012-09-01 15:00:34 +0200325 return 1;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100326}
327
328/* Convert an object to a byte string (char*) or a wide character string
329 (wchar_t*). Return:
330
331 - 2 if obj is a character string (written into *wch)
332 - 1 if obj is a byte string (written into *bytes)
333 - 0 on error: raise an exception */
334static int
335PyCurses_ConvertToString(PyCursesWindowObject *win, PyObject *obj,
336 PyObject **bytes, wchar_t **wstr)
337{
338 if (PyUnicode_Check(obj)) {
339#ifdef HAVE_NCURSESW
340 assert (wstr != NULL);
341 *wstr = PyUnicode_AsWideCharString(obj, NULL);
342 if (*wstr == NULL)
343 return 0;
344 return 2;
345#else
346 assert (wstr == NULL);
347 *bytes = PyUnicode_AsEncodedObject(obj, win->encoding, NULL);
348 if (*bytes == NULL)
349 return 0;
350 return 1;
351#endif
352 }
353 else if (PyBytes_Check(obj)) {
354 Py_INCREF(obj);
355 *bytes = obj;
356 return 1;
357 }
358
359 PyErr_Format(PyExc_TypeError, "expect bytes or str, got %s",
360 Py_TYPE(obj)->tp_name);
361 return 0;
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000362}
363
Benjamin Peterson21896a32010-03-21 22:03:03 +0000364/* Function versions of the 3 functions for testing whether curses has been
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000365 initialised or not. */
Victor Stinner26486ea2010-05-15 22:23:53 +0000366
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000367static int func_PyCursesSetupTermCalled(void)
368{
369 PyCursesSetupTermCalled;
370 return 1;
371}
372
373static int func_PyCursesInitialised(void)
374{
375 PyCursesInitialised;
376 return 1;
377}
378
379static int func_PyCursesInitialisedColor(void)
380{
381 PyCursesInitialisedColor;
382 return 1;
383}
384
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000385/*****************************************************************************
386 The Window Object
387******************************************************************************/
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000388
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000389/* Definition of the window type */
Guido van Rossum85738471995-02-17 13:50:17 +0000390
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000391PyTypeObject PyCursesWindow_Type;
392
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000393/* Function prototype macros for Window object
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000394
395 X - function name
396 TYPE - parameter Type
397 ERGSTR - format string for construction of the return value
398 PARSESTR - format string for argument parsing
Victor Stinner26486ea2010-05-15 22:23:53 +0000399*/
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000400
Victor Stinner26486ea2010-05-15 22:23:53 +0000401#define Window_NoArgNoReturnFunction(X) \
402 static PyObject *PyCursesWindow_ ## X \
403 (PyCursesWindowObject *self, PyObject *args) \
404 { return PyCursesCheckERR(X(self->win), # X); }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000405
Victor Stinner26486ea2010-05-15 22:23:53 +0000406#define Window_NoArgTrueFalseFunction(X) \
407 static PyObject * PyCursesWindow_ ## X \
408 (PyCursesWindowObject *self) \
409 { \
410 if (X (self->win) == FALSE) { Py_INCREF(Py_False); return Py_False; } \
411 else { Py_INCREF(Py_True); return Py_True; } }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000412
Victor Stinner26486ea2010-05-15 22:23:53 +0000413#define Window_NoArgNoReturnVoidFunction(X) \
414 static PyObject * PyCursesWindow_ ## X \
415 (PyCursesWindowObject *self) \
416 { \
417 X(self->win); Py_INCREF(Py_None); return Py_None; }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000418
Victor Stinner26486ea2010-05-15 22:23:53 +0000419#define Window_NoArg2TupleReturnFunction(X, TYPE, ERGSTR) \
420 static PyObject * PyCursesWindow_ ## X \
421 (PyCursesWindowObject *self) \
422 { \
423 TYPE arg1, arg2; \
424 X(self->win,arg1,arg2); return Py_BuildValue(ERGSTR, arg1, arg2); }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000425
Victor Stinner26486ea2010-05-15 22:23:53 +0000426#define Window_OneArgNoReturnVoidFunction(X, TYPE, PARSESTR) \
427 static PyObject * PyCursesWindow_ ## X \
428 (PyCursesWindowObject *self, PyObject *args) \
429 { \
430 TYPE arg1; \
431 if (!PyArg_ParseTuple(args, PARSESTR, &arg1)) return NULL; \
432 X(self->win,arg1); Py_INCREF(Py_None); return Py_None; }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000433
Victor Stinner26486ea2010-05-15 22:23:53 +0000434#define Window_OneArgNoReturnFunction(X, TYPE, PARSESTR) \
435 static PyObject * PyCursesWindow_ ## X \
436 (PyCursesWindowObject *self, PyObject *args) \
437 { \
438 TYPE arg1; \
439 if (!PyArg_ParseTuple(args,PARSESTR, &arg1)) return NULL; \
440 return PyCursesCheckERR(X(self->win, arg1), # X); }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000441
Victor Stinner26486ea2010-05-15 22:23:53 +0000442#define Window_TwoArgNoReturnFunction(X, TYPE, PARSESTR) \
443 static PyObject * PyCursesWindow_ ## X \
444 (PyCursesWindowObject *self, PyObject *args) \
445 { \
446 TYPE arg1, arg2; \
447 if (!PyArg_ParseTuple(args,PARSESTR, &arg1, &arg2)) return NULL; \
448 return PyCursesCheckERR(X(self->win, arg1, arg2), # X); }
Guido van Rossum85738471995-02-17 13:50:17 +0000449
Guido van Rossumf6971e21994-08-30 12:25:20 +0000450/* ------------- WINDOW routines --------------- */
Guido van Rossum85738471995-02-17 13:50:17 +0000451
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000452Window_NoArgNoReturnFunction(untouchwin)
453Window_NoArgNoReturnFunction(touchwin)
454Window_NoArgNoReturnFunction(redrawwin)
455Window_NoArgNoReturnFunction(winsertln)
456Window_NoArgNoReturnFunction(werase)
457Window_NoArgNoReturnFunction(wdeleteln)
458
459Window_NoArgTrueFalseFunction(is_wintouched)
460
461Window_NoArgNoReturnVoidFunction(wsyncup)
462Window_NoArgNoReturnVoidFunction(wsyncdown)
463Window_NoArgNoReturnVoidFunction(wstandend)
464Window_NoArgNoReturnVoidFunction(wstandout)
465Window_NoArgNoReturnVoidFunction(wcursyncup)
466Window_NoArgNoReturnVoidFunction(wclrtoeol)
467Window_NoArgNoReturnVoidFunction(wclrtobot)
468Window_NoArgNoReturnVoidFunction(wclear)
469
470Window_OneArgNoReturnVoidFunction(idcok, int, "i;True(1) or False(0)")
471Window_OneArgNoReturnVoidFunction(immedok, int, "i;True(1) or False(0)")
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +0000472Window_OneArgNoReturnVoidFunction(wtimeout, int, "i;delay")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000473
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000474Window_NoArg2TupleReturnFunction(getyx, int, "ii")
475Window_NoArg2TupleReturnFunction(getbegyx, int, "ii")
476Window_NoArg2TupleReturnFunction(getmaxyx, int, "ii")
477Window_NoArg2TupleReturnFunction(getparyx, int, "ii")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000478
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000479Window_OneArgNoReturnFunction(clearok, int, "i;True(1) or False(0)")
480Window_OneArgNoReturnFunction(idlok, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000481#if defined(__NetBSD__)
482Window_OneArgNoReturnVoidFunction(keypad, int, "i;True(1) or False(0)")
483#else
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000484Window_OneArgNoReturnFunction(keypad, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000485#endif
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000486Window_OneArgNoReturnFunction(leaveok, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000487#if defined(__NetBSD__)
488Window_OneArgNoReturnVoidFunction(nodelay, int, "i;True(1) or False(0)")
489#else
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000490Window_OneArgNoReturnFunction(nodelay, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000491#endif
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000492Window_OneArgNoReturnFunction(notimeout, int, "i;True(1) or False(0)")
493Window_OneArgNoReturnFunction(scrollok, int, "i;True(1) or False(0)")
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +0000494Window_OneArgNoReturnFunction(winsdelln, int, "i;nlines")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000495Window_OneArgNoReturnFunction(syncok, int, "i;True(1) or False(0)")
496
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000497Window_TwoArgNoReturnFunction(mvwin, int, "ii;y,x")
498Window_TwoArgNoReturnFunction(mvderwin, int, "ii;y,x")
499Window_TwoArgNoReturnFunction(wmove, int, "ii;y,x")
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +0000500#ifndef STRICT_SYSV_CURSES
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000501Window_TwoArgNoReturnFunction(wresize, int, "ii;lines,columns")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000502#endif
503
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +0000504/* Allocation and deallocation of Window Objects */
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000505
Guido van Rossumf6971e21994-08-30 12:25:20 +0000506static PyObject *
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100507PyCursesWindow_New(WINDOW *win, const char *encoding)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000508{
Victor Stinner26486ea2010-05-15 22:23:53 +0000509 PyCursesWindowObject *wo;
Guido van Rossum85738471995-02-17 13:50:17 +0000510
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100511 if (encoding == NULL) {
512#if defined(MS_WINDOWS)
513 char *buffer[100];
514 UINT cp;
515 cp = GetConsoleOutputCP();
516 if (cp != 0) {
517 PyOS_snprintf(buffer, sizeof(buffer), "cp%u", cp);
518 encoding = buffer;
519 }
520#elif defined(CODESET)
521 const char *codeset = nl_langinfo(CODESET);
522 if (codeset != NULL && codeset[0] != 0)
523 encoding = codeset;
524#endif
525 if (encoding == NULL)
526 encoding = "utf-8";
527 }
528
Victor Stinner26486ea2010-05-15 22:23:53 +0000529 wo = PyObject_NEW(PyCursesWindowObject, &PyCursesWindow_Type);
530 if (wo == NULL) return NULL;
531 wo->win = win;
Victor Stinner49fc8ec2013-07-07 23:30:24 +0200532 wo->encoding = _PyMem_Strdup(encoding);
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100533 if (wo->encoding == NULL) {
534 Py_DECREF(wo);
535 PyErr_NoMemory();
536 return NULL;
537 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000538 return (PyObject *)wo;
Guido van Rossumf6971e21994-08-30 12:25:20 +0000539}
540
541static void
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000542PyCursesWindow_Dealloc(PyCursesWindowObject *wo)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000543{
Victor Stinner26486ea2010-05-15 22:23:53 +0000544 if (wo->win != stdscr) delwin(wo->win);
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100545 if (wo->encoding != NULL)
Victor Stinner49fc8ec2013-07-07 23:30:24 +0200546 PyMem_Free(wo->encoding);
Victor Stinner26486ea2010-05-15 22:23:53 +0000547 PyObject_DEL(wo);
Guido van Rossumf6971e21994-08-30 12:25:20 +0000548}
549
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000550/* Addch, Addstr, Addnstr */
Guido van Rossumf6971e21994-08-30 12:25:20 +0000551
552static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000553PyCursesWindow_AddCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000554{
Victor Stinner26486ea2010-05-15 22:23:53 +0000555 int rtn, x, y, use_xy = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100556 PyObject *chobj;
557 int type;
558 chtype ch;
559#ifdef HAVE_NCURSESW
560 cchar_t wch;
561#endif
Victor Stinner26486ea2010-05-15 22:23:53 +0000562 attr_t attr = A_NORMAL;
563 long lattr;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100564 const char *funcname;
Guido van Rossum85738471995-02-17 13:50:17 +0000565
Victor Stinner26486ea2010-05-15 22:23:53 +0000566 switch (PyTuple_Size(args)) {
567 case 1:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100568 if (!PyArg_ParseTuple(args, "O;ch or int", &chobj))
Victor Stinner26486ea2010-05-15 22:23:53 +0000569 return NULL;
570 break;
571 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100572 if (!PyArg_ParseTuple(args, "Ol;ch or int,attr", &chobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000573 return NULL;
574 attr = lattr;
575 break;
576 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100577 if (!PyArg_ParseTuple(args,"iiO;y,x,ch or int", &y, &x, &chobj))
Victor Stinner26486ea2010-05-15 22:23:53 +0000578 return NULL;
579 use_xy = TRUE;
580 break;
581 case 4:
582 if (!PyArg_ParseTuple(args,"iiOl;y,x,ch or int, attr",
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100583 &y, &x, &chobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000584 return NULL;
585 attr = lattr;
586 use_xy = TRUE;
587 break;
588 default:
589 PyErr_SetString(PyExc_TypeError, "addch requires 1 to 4 arguments");
590 return NULL;
591 }
592
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100593#ifdef HAVE_NCURSESW
594 type = PyCurses_ConvertToCchar_t(self, chobj, &ch, &wch);
595 if (type == 2) {
596 funcname = "add_wch";
597 wch.attr = attr;
598 if (use_xy == TRUE)
599 rtn = mvwadd_wch(self->win,y,x, &wch);
600 else {
601 rtn = wadd_wch(self->win, &wch);
602 }
603 }
604 else
605#else
606 type = PyCurses_ConvertToCchar_t(self, chobj, &ch);
607#endif
608 if (type == 1) {
609 funcname = "addch";
610 if (use_xy == TRUE)
611 rtn = mvwaddch(self->win,y,x, ch | attr);
612 else {
613 rtn = waddch(self->win, ch | attr);
614 }
615 }
616 else {
Victor Stinner26486ea2010-05-15 22:23:53 +0000617 return NULL;
618 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100619 return PyCursesCheckERR(rtn, funcname);
Guido van Rossumf6971e21994-08-30 12:25:20 +0000620}
621
622static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000623PyCursesWindow_AddStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000624{
Victor Stinner26486ea2010-05-15 22:23:53 +0000625 int rtn;
626 int x, y;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100627 int strtype;
Benjamin Peterson5e545192012-03-14 18:21:35 -0500628 PyObject *strobj, *bytesobj = NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100629#ifdef HAVE_NCURSESW
630 wchar_t *wstr = NULL;
631#endif
Victor Stinner26486ea2010-05-15 22:23:53 +0000632 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
633 long lattr;
634 int use_xy = FALSE, use_attr = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100635 const char *funcname;
Guido van Rossum85738471995-02-17 13:50:17 +0000636
Victor Stinner26486ea2010-05-15 22:23:53 +0000637 switch (PyTuple_Size(args)) {
638 case 1:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100639 if (!PyArg_ParseTuple(args,"O;str", &strobj))
Victor Stinner26486ea2010-05-15 22:23:53 +0000640 return NULL;
641 break;
642 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100643 if (!PyArg_ParseTuple(args,"Ol;str,attr", &strobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000644 return NULL;
645 attr = lattr;
646 use_attr = TRUE;
647 break;
648 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100649 if (!PyArg_ParseTuple(args,"iiO;int,int,str", &y, &x, &strobj))
Victor Stinner26486ea2010-05-15 22:23:53 +0000650 return NULL;
651 use_xy = TRUE;
652 break;
653 case 4:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100654 if (!PyArg_ParseTuple(args,"iiOl;int,int,str,attr", &y, &x, &strobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000655 return NULL;
656 attr = lattr;
657 use_xy = use_attr = TRUE;
658 break;
659 default:
660 PyErr_SetString(PyExc_TypeError, "addstr requires 1 to 4 arguments");
661 return NULL;
662 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100663#ifdef HAVE_NCURSESW
664 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);
665#else
666 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);
667#endif
668 if (strtype == 0)
669 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +0000670 if (use_attr == TRUE) {
671 attr_old = getattrs(self->win);
672 (void)wattrset(self->win,attr);
673 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100674#ifdef HAVE_NCURSESW
675 if (strtype == 2) {
676 funcname = "addwstr";
677 if (use_xy == TRUE)
678 rtn = mvwaddwstr(self->win,y,x,wstr);
679 else
680 rtn = waddwstr(self->win,wstr);
681 PyMem_Free(wstr);
682 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000683 else
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100684#endif
685 {
686 char *str = PyBytes_AS_STRING(bytesobj);
687 funcname = "addstr";
688 if (use_xy == TRUE)
689 rtn = mvwaddstr(self->win,y,x,str);
690 else
691 rtn = waddstr(self->win,str);
692 Py_DECREF(bytesobj);
693 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000694 if (use_attr == TRUE)
695 (void)wattrset(self->win,attr_old);
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100696 return PyCursesCheckERR(rtn, funcname);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000697}
Guido van Rossum85738471995-02-17 13:50:17 +0000698
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000699static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000700PyCursesWindow_AddNStr(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000701{
Victor Stinner26486ea2010-05-15 22:23:53 +0000702 int rtn, x, y, n;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100703 int strtype;
Benjamin Peterson5e545192012-03-14 18:21:35 -0500704 PyObject *strobj, *bytesobj = NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100705#ifdef HAVE_NCURSESW
706 wchar_t *wstr = NULL;
707#endif
Victor Stinner26486ea2010-05-15 22:23:53 +0000708 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
709 long lattr;
710 int use_xy = FALSE, use_attr = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100711 const char *funcname;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000712
Victor Stinner26486ea2010-05-15 22:23:53 +0000713 switch (PyTuple_Size(args)) {
714 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100715 if (!PyArg_ParseTuple(args,"Oi;str,n", &strobj, &n))
Victor Stinner26486ea2010-05-15 22:23:53 +0000716 return NULL;
717 break;
718 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100719 if (!PyArg_ParseTuple(args,"Oil;str,n,attr", &strobj, &n, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000720 return NULL;
721 attr = lattr;
722 use_attr = TRUE;
723 break;
724 case 4:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100725 if (!PyArg_ParseTuple(args,"iiOi;y,x,str,n", &y, &x, &strobj, &n))
Victor Stinner26486ea2010-05-15 22:23:53 +0000726 return NULL;
727 use_xy = TRUE;
728 break;
729 case 5:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100730 if (!PyArg_ParseTuple(args,"iiOil;y,x,str,n,attr", &y, &x, &strobj, &n, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000731 return NULL;
732 attr = lattr;
733 use_xy = use_attr = TRUE;
734 break;
735 default:
736 PyErr_SetString(PyExc_TypeError, "addnstr requires 2 to 5 arguments");
737 return NULL;
738 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100739#ifdef HAVE_NCURSESW
740 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);
741#else
742 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);
743#endif
744 if (strtype == 0)
745 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000746
Victor Stinner26486ea2010-05-15 22:23:53 +0000747 if (use_attr == TRUE) {
748 attr_old = getattrs(self->win);
749 (void)wattrset(self->win,attr);
750 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100751#ifdef HAVE_NCURSESW
752 if (strtype == 2) {
753 funcname = "addnwstr";
754 if (use_xy == TRUE)
755 rtn = mvwaddnwstr(self->win,y,x,wstr,n);
756 else
757 rtn = waddnwstr(self->win,wstr,n);
758 PyMem_Free(wstr);
759 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000760 else
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100761#endif
762 {
763 char *str = PyBytes_AS_STRING(bytesobj);
764 funcname = "addnstr";
765 if (use_xy == TRUE)
766 rtn = mvwaddnstr(self->win,y,x,str,n);
767 else
768 rtn = waddnstr(self->win,str,n);
769 Py_DECREF(bytesobj);
770 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000771 if (use_attr == TRUE)
772 (void)wattrset(self->win,attr_old);
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100773 return PyCursesCheckERR(rtn, funcname);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000774}
775
776static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000777PyCursesWindow_Bkgd(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000778{
Victor Stinner26486ea2010-05-15 22:23:53 +0000779 PyObject *temp;
780 chtype bkgd;
781 attr_t attr = A_NORMAL;
782 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000783
Victor Stinner26486ea2010-05-15 22:23:53 +0000784 switch (PyTuple_Size(args)) {
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000785 case 1:
Victor Stinner26486ea2010-05-15 22:23:53 +0000786 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
787 return NULL;
788 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000789 case 2:
Victor Stinner26486ea2010-05-15 22:23:53 +0000790 if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))
791 return NULL;
792 attr = lattr;
793 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000794 default:
Victor Stinner26486ea2010-05-15 22:23:53 +0000795 PyErr_SetString(PyExc_TypeError, "bkgd requires 1 or 2 arguments");
796 return NULL;
797 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000798
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100799 if (!PyCurses_ConvertToChtype(self, temp, &bkgd))
Victor Stinner26486ea2010-05-15 22:23:53 +0000800 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000801
Victor Stinner26486ea2010-05-15 22:23:53 +0000802 return PyCursesCheckERR(wbkgd(self->win, bkgd | attr), "bkgd");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000803}
804
805static PyObject *
Christian Heimes2380ac72008-01-09 00:17:24 +0000806PyCursesWindow_AttrOff(PyCursesWindowObject *self, PyObject *args)
807{
Victor Stinner26486ea2010-05-15 22:23:53 +0000808 long lattr;
809 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
810 return NULL;
811 return PyCursesCheckERR(wattroff(self->win, (attr_t)lattr), "attroff");
Christian Heimes2380ac72008-01-09 00:17:24 +0000812}
813
814static PyObject *
815PyCursesWindow_AttrOn(PyCursesWindowObject *self, PyObject *args)
816{
Victor Stinner26486ea2010-05-15 22:23:53 +0000817 long lattr;
818 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
819 return NULL;
820 return PyCursesCheckERR(wattron(self->win, (attr_t)lattr), "attron");
Christian Heimes2380ac72008-01-09 00:17:24 +0000821}
822
823static PyObject *
824PyCursesWindow_AttrSet(PyCursesWindowObject *self, PyObject *args)
825{
Victor Stinner26486ea2010-05-15 22:23:53 +0000826 long lattr;
827 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
828 return NULL;
829 return PyCursesCheckERR(wattrset(self->win, (attr_t)lattr), "attrset");
Christian Heimes2380ac72008-01-09 00:17:24 +0000830}
831
832static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000833PyCursesWindow_BkgdSet(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000834{
Victor Stinner26486ea2010-05-15 22:23:53 +0000835 PyObject *temp;
836 chtype bkgd;
837 attr_t attr = A_NORMAL;
838 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000839
Victor Stinner26486ea2010-05-15 22:23:53 +0000840 switch (PyTuple_Size(args)) {
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000841 case 1:
Victor Stinner26486ea2010-05-15 22:23:53 +0000842 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
843 return NULL;
844 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000845 case 2:
Victor Stinner26486ea2010-05-15 22:23:53 +0000846 if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))
847 return NULL;
848 attr = lattr;
849 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000850 default:
Victor Stinner26486ea2010-05-15 22:23:53 +0000851 PyErr_SetString(PyExc_TypeError, "bkgdset requires 1 or 2 arguments");
852 return NULL;
853 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000854
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100855 if (!PyCurses_ConvertToChtype(self, temp, &bkgd))
Victor Stinner26486ea2010-05-15 22:23:53 +0000856 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000857
Victor Stinner26486ea2010-05-15 22:23:53 +0000858 wbkgdset(self->win, bkgd | attr);
859 return PyCursesCheckERR(0, "bkgdset");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000860}
861
862static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000863PyCursesWindow_Border(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000864{
Victor Stinner26486ea2010-05-15 22:23:53 +0000865 PyObject *temp[8];
866 chtype ch[8];
867 int i;
Andrew M. Kuchling16e65a02001-10-20 16:05:52 +0000868
Victor Stinner26486ea2010-05-15 22:23:53 +0000869 /* Clear the array of parameters */
870 for(i=0; i<8; i++) {
871 temp[i] = NULL;
872 ch[i] = 0;
873 }
Andrew M. Kuchling16e65a02001-10-20 16:05:52 +0000874
Victor Stinner26486ea2010-05-15 22:23:53 +0000875 if (!PyArg_ParseTuple(args,"|OOOOOOOO;ls,rs,ts,bs,tl,tr,bl,br",
876 &temp[0], &temp[1], &temp[2], &temp[3],
877 &temp[4], &temp[5], &temp[6], &temp[7]))
878 return NULL;
879
880 for(i=0; i<8; i++) {
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100881 if (temp[i] != NULL && !PyCurses_ConvertToChtype(self, temp[i], &ch[i]))
Victor Stinner26486ea2010-05-15 22:23:53 +0000882 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +0000883 }
884
885 wborder(self->win,
886 ch[0], ch[1], ch[2], ch[3],
887 ch[4], ch[5], ch[6], ch[7]);
888 Py_INCREF(Py_None);
889 return Py_None;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000890}
891
892static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000893PyCursesWindow_Box(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000894{
Victor Stinner26486ea2010-05-15 22:23:53 +0000895 chtype ch1=0,ch2=0;
896 switch(PyTuple_Size(args)){
897 case 0: break;
898 default:
899 if (!PyArg_ParseTuple(args,"ll;vertint,horint", &ch1, &ch2))
900 return NULL;
901 }
902 box(self->win,ch1,ch2);
903 Py_INCREF(Py_None);
904 return Py_None;
Guido van Rossumf6971e21994-08-30 12:25:20 +0000905}
906
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000907#if defined(HAVE_NCURSES_H) || defined(MVWDELCH_IS_EXPRESSION)
908#define py_mvwdelch mvwdelch
909#else
910int py_mvwdelch(WINDOW *w, int y, int x)
911{
Victor Stinner26486ea2010-05-15 22:23:53 +0000912 mvwdelch(w,y,x);
913 /* On HP/UX, mvwdelch already returns. On other systems,
914 we may well run into this return statement. */
915 return 0;
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000916}
917#endif
918
Guido van Rossumd8faa362007-04-27 19:54:29 +0000919/* chgat, added by Fabian Kreutz <fabian.kreutz at gmx.net> */
920
921static PyObject *
922PyCursesWindow_ChgAt(PyCursesWindowObject *self, PyObject *args)
923{
Victor Stinner26486ea2010-05-15 22:23:53 +0000924 int rtn;
925 int x, y;
926 int num = -1;
927 short color;
928 attr_t attr = A_NORMAL;
929 long lattr;
930 int use_xy = FALSE;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000931
Victor Stinner26486ea2010-05-15 22:23:53 +0000932 switch (PyTuple_Size(args)) {
933 case 1:
934 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
935 return NULL;
936 attr = lattr;
937 break;
938 case 2:
939 if (!PyArg_ParseTuple(args,"il;n,attr", &num, &lattr))
940 return NULL;
941 attr = lattr;
942 break;
943 case 3:
944 if (!PyArg_ParseTuple(args,"iil;int,int,attr", &y, &x, &lattr))
945 return NULL;
946 attr = lattr;
947 use_xy = TRUE;
948 break;
949 case 4:
950 if (!PyArg_ParseTuple(args,"iiil;int,int,n,attr", &y, &x, &num, &lattr))
951 return NULL;
952 attr = lattr;
953 use_xy = TRUE;
954 break;
955 default:
956 PyErr_SetString(PyExc_TypeError, "chgat requires 1 to 4 arguments");
957 return NULL;
958 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000959
Victor Stinner26486ea2010-05-15 22:23:53 +0000960 color = (short)((attr >> 8) & 0xff);
961 attr = attr - (color << 8);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000962
Victor Stinner26486ea2010-05-15 22:23:53 +0000963 if (use_xy == TRUE) {
964 rtn = mvwchgat(self->win,y,x,num,attr,color,NULL);
965 touchline(self->win,y,1);
966 } else {
967 getyx(self->win,y,x);
968 rtn = wchgat(self->win,num,attr,color,NULL);
969 touchline(self->win,y,1);
970 }
971 return PyCursesCheckERR(rtn, "chgat");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000972}
973
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000974
Guido van Rossumf6971e21994-08-30 12:25:20 +0000975static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000976PyCursesWindow_DelCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000977{
Victor Stinner26486ea2010-05-15 22:23:53 +0000978 int rtn;
979 int x, y;
Guido van Rossum85738471995-02-17 13:50:17 +0000980
Victor Stinner26486ea2010-05-15 22:23:53 +0000981 switch (PyTuple_Size(args)) {
982 case 0:
983 rtn = wdelch(self->win);
984 break;
985 case 2:
986 if (!PyArg_ParseTuple(args,"ii;y,x", &y, &x))
987 return NULL;
988 rtn = py_mvwdelch(self->win,y,x);
989 break;
990 default:
991 PyErr_SetString(PyExc_TypeError, "delch requires 0 or 2 arguments");
992 return NULL;
993 }
994 return PyCursesCheckERR(rtn, "[mv]wdelch");
Guido van Rossumf6971e21994-08-30 12:25:20 +0000995}
996
997static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000998PyCursesWindow_DerWin(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000999{
Victor Stinner26486ea2010-05-15 22:23:53 +00001000 WINDOW *win;
1001 int nlines, ncols, begin_y, begin_x;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001002
Victor Stinner26486ea2010-05-15 22:23:53 +00001003 nlines = 0;
1004 ncols = 0;
1005 switch (PyTuple_Size(args)) {
1006 case 2:
1007 if (!PyArg_ParseTuple(args,"ii;begin_y,begin_x",&begin_y,&begin_x))
1008 return NULL;
1009 break;
1010 case 4:
1011 if (!PyArg_ParseTuple(args, "iiii;nlines,ncols,begin_y,begin_x",
1012 &nlines,&ncols,&begin_y,&begin_x))
1013 return NULL;
1014 break;
1015 default:
1016 PyErr_SetString(PyExc_TypeError, "derwin requires 2 or 4 arguments");
1017 return NULL;
1018 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001019
Victor Stinner26486ea2010-05-15 22:23:53 +00001020 win = derwin(self->win,nlines,ncols,begin_y,begin_x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001021
Victor Stinner26486ea2010-05-15 22:23:53 +00001022 if (win == NULL) {
1023 PyErr_SetString(PyCursesError, catchall_NULL);
1024 return NULL;
1025 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001026
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001027 return (PyObject *)PyCursesWindow_New(win, NULL);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001028}
1029
1030static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001031PyCursesWindow_EchoChar(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001032{
Victor Stinner26486ea2010-05-15 22:23:53 +00001033 PyObject *temp;
1034 chtype ch;
1035 attr_t attr = A_NORMAL;
1036 long lattr;
Guido van Rossum85738471995-02-17 13:50:17 +00001037
Victor Stinner26486ea2010-05-15 22:23:53 +00001038 switch (PyTuple_Size(args)) {
1039 case 1:
1040 if (!PyArg_ParseTuple(args,"O;ch or int", &temp))
1041 return NULL;
1042 break;
1043 case 2:
1044 if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))
1045 return NULL;
1046 attr = lattr;
1047 break;
1048 default:
1049 PyErr_SetString(PyExc_TypeError, "echochar requires 1 or 2 arguments");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001050
1051
Victor Stinner26486ea2010-05-15 22:23:53 +00001052 return NULL;
1053 }
Guido van Rossum85738471995-02-17 13:50:17 +00001054
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001055 if (!PyCurses_ConvertToChtype(self, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00001056 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00001057
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00001058#ifdef WINDOW_HAS_FLAGS
Victor Stinner26486ea2010-05-15 22:23:53 +00001059 if (self->win->_flags & _ISPAD)
1060 return PyCursesCheckERR(pechochar(self->win, ch | attr),
1061 "echochar");
1062 else
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001063#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001064 return PyCursesCheckERR(wechochar(self->win, ch | attr),
1065 "echochar");
Guido van Rossumf6971e21994-08-30 12:25:20 +00001066}
1067
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001068#ifdef NCURSES_MOUSE_VERSION
1069static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001070PyCursesWindow_Enclose(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001071{
Victor Stinner26486ea2010-05-15 22:23:53 +00001072 int x, y;
1073 if (!PyArg_ParseTuple(args,"ii;y,x", &y, &x))
1074 return NULL;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001075
Victor Stinner26486ea2010-05-15 22:23:53 +00001076 return PyLong_FromLong( wenclose(self->win,y,x) );
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001077}
1078#endif
1079
Guido van Rossumf6971e21994-08-30 12:25:20 +00001080static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00001081PyCursesWindow_GetBkgd(PyCursesWindowObject *self)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001082{
Victor Stinner26486ea2010-05-15 22:23:53 +00001083 return PyLong_FromLong((long) getbkgd(self->win));
Guido van Rossumf6971e21994-08-30 12:25:20 +00001084}
1085
1086static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001087PyCursesWindow_GetCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001088{
Victor Stinner26486ea2010-05-15 22:23:53 +00001089 int x, y;
1090 int rtn;
Guido van Rossum85738471995-02-17 13:50:17 +00001091
Victor Stinner26486ea2010-05-15 22:23:53 +00001092 switch (PyTuple_Size(args)) {
1093 case 0:
1094 Py_BEGIN_ALLOW_THREADS
1095 rtn = wgetch(self->win);
1096 Py_END_ALLOW_THREADS
1097 break;
1098 case 2:
1099 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1100 return NULL;
1101 Py_BEGIN_ALLOW_THREADS
1102 rtn = mvwgetch(self->win,y,x);
1103 Py_END_ALLOW_THREADS
1104 break;
1105 default:
1106 PyErr_SetString(PyExc_TypeError, "getch requires 0 or 2 arguments");
1107 return NULL;
1108 }
1109 return PyLong_FromLong((long)rtn);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001110}
Guido van Rossum85738471995-02-17 13:50:17 +00001111
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001112static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001113PyCursesWindow_GetKey(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001114{
Victor Stinner26486ea2010-05-15 22:23:53 +00001115 int x, y;
1116 int rtn;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001117
Victor Stinner26486ea2010-05-15 22:23:53 +00001118 switch (PyTuple_Size(args)) {
1119 case 0:
1120 Py_BEGIN_ALLOW_THREADS
1121 rtn = wgetch(self->win);
1122 Py_END_ALLOW_THREADS
1123 break;
1124 case 2:
1125 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1126 return NULL;
1127 Py_BEGIN_ALLOW_THREADS
1128 rtn = mvwgetch(self->win,y,x);
1129 Py_END_ALLOW_THREADS
1130 break;
1131 default:
1132 PyErr_SetString(PyExc_TypeError, "getkey requires 0 or 2 arguments");
1133 return NULL;
1134 }
1135 if (rtn == ERR) {
1136 /* getch() returns ERR in nodelay mode */
R David Murrayf5d7cc22013-03-19 16:23:09 -04001137 PyErr_CheckSignals();
1138 if (!PyErr_Occurred())
1139 PyErr_SetString(PyCursesError, "no input");
Victor Stinner26486ea2010-05-15 22:23:53 +00001140 return NULL;
1141 } else if (rtn<=255) {
1142 return Py_BuildValue("C", rtn);
1143 } else {
1144 const char *knp;
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001145#if defined(__NetBSD__)
Victor Stinner26486ea2010-05-15 22:23:53 +00001146 knp = unctrl(rtn);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001147#else
Victor Stinner26486ea2010-05-15 22:23:53 +00001148 knp = keyname(rtn);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001149#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001150 return PyUnicode_FromString((knp == NULL) ? "" : knp);
1151 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001152}
1153
Nadeem Vawda9e2e9902011-07-31 15:01:11 +02001154#ifdef HAVE_NCURSESW
Guido van Rossumf6971e21994-08-30 12:25:20 +00001155static PyObject *
Victor Stinnera7878b72011-07-14 23:07:44 +02001156PyCursesWindow_Get_WCh(PyCursesWindowObject *self, PyObject *args)
1157{
1158 int x, y;
1159 int ct;
1160 wint_t rtn;
1161
1162 switch (PyTuple_Size(args)) {
1163 case 0:
1164 Py_BEGIN_ALLOW_THREADS
1165 ct = wget_wch(self->win,&rtn);
1166 Py_END_ALLOW_THREADS
1167 break;
1168 case 2:
1169 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1170 return NULL;
1171 Py_BEGIN_ALLOW_THREADS
1172 ct = mvwget_wch(self->win,y,x,&rtn);
1173 Py_END_ALLOW_THREADS
1174 break;
1175 default:
1176 PyErr_SetString(PyExc_TypeError, "get_wch requires 0 or 2 arguments");
1177 return NULL;
1178 }
1179 if (ct == ERR) {
Victor Stinnerbd2d30c2013-03-21 12:21:06 +01001180 if (PyErr_CheckSignals())
1181 return NULL;
1182
Victor Stinnera7878b72011-07-14 23:07:44 +02001183 /* get_wch() returns ERR in nodelay mode */
1184 PyErr_SetString(PyCursesError, "no input");
1185 return NULL;
1186 }
Victor Stinner1d39cde2012-08-29 01:40:57 +02001187 if (ct == KEY_CODE_YES)
1188 return PyLong_FromLong(rtn);
1189 else
1190 return PyUnicode_FromOrdinal(rtn);
Victor Stinnera7878b72011-07-14 23:07:44 +02001191}
Nadeem Vawda9e2e9902011-07-31 15:01:11 +02001192#endif
Victor Stinnera7878b72011-07-14 23:07:44 +02001193
1194static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001195PyCursesWindow_GetStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001196{
Victor Stinner26486ea2010-05-15 22:23:53 +00001197 int x, y, n;
1198 char rtn[1024]; /* This should be big enough.. I hope */
1199 int rtn2;
Guido van Rossum85738471995-02-17 13:50:17 +00001200
Victor Stinner26486ea2010-05-15 22:23:53 +00001201 switch (PyTuple_Size(args)) {
1202 case 0:
1203 Py_BEGIN_ALLOW_THREADS
1204 rtn2 = wgetnstr(self->win,rtn, 1023);
1205 Py_END_ALLOW_THREADS
1206 break;
1207 case 1:
1208 if (!PyArg_ParseTuple(args,"i;n", &n))
1209 return NULL;
1210 Py_BEGIN_ALLOW_THREADS
Victor Stinner640c35c2013-06-04 23:14:37 +02001211 rtn2 = wgetnstr(self->win, rtn, Py_MIN(n, 1023));
Victor Stinner26486ea2010-05-15 22:23:53 +00001212 Py_END_ALLOW_THREADS
1213 break;
1214 case 2:
1215 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1216 return NULL;
1217 Py_BEGIN_ALLOW_THREADS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001218#ifdef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00001219 rtn2 = wmove(self->win,y,x)==ERR ? ERR : wgetnstr(self->win, rtn, 1023);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001220#else
Victor Stinner26486ea2010-05-15 22:23:53 +00001221 rtn2 = mvwgetnstr(self->win,y,x,rtn, 1023);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001222#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001223 Py_END_ALLOW_THREADS
1224 break;
1225 case 3:
1226 if (!PyArg_ParseTuple(args,"iii;y,x,n", &y, &x, &n))
1227 return NULL;
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +00001228#ifdef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00001229 Py_BEGIN_ALLOW_THREADS
1230 rtn2 = wmove(self->win,y,x)==ERR ? ERR :
Victor Stinner640c35c2013-06-04 23:14:37 +02001231 wgetnstr(self->win, rtn, Py_MIN(n, 1023));
Victor Stinner26486ea2010-05-15 22:23:53 +00001232 Py_END_ALLOW_THREADS
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001233#else
Victor Stinner26486ea2010-05-15 22:23:53 +00001234 Py_BEGIN_ALLOW_THREADS
Victor Stinner640c35c2013-06-04 23:14:37 +02001235 rtn2 = mvwgetnstr(self->win, y, x, rtn, Py_MIN(n, 1023));
Victor Stinner26486ea2010-05-15 22:23:53 +00001236 Py_END_ALLOW_THREADS
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001237#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001238 break;
1239 default:
1240 PyErr_SetString(PyExc_TypeError, "getstr requires 0 to 3 arguments");
1241 return NULL;
1242 }
1243 if (rtn2 == ERR)
1244 rtn[0] = 0;
1245 return PyBytes_FromString(rtn);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001246}
1247
1248static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001249PyCursesWindow_Hline(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001250{
Victor Stinner26486ea2010-05-15 22:23:53 +00001251 PyObject *temp;
1252 chtype ch;
1253 int n, x, y, code = OK;
1254 attr_t attr = A_NORMAL;
1255 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001256
Victor Stinner26486ea2010-05-15 22:23:53 +00001257 switch (PyTuple_Size(args)) {
1258 case 2:
1259 if (!PyArg_ParseTuple(args, "Oi;ch or int,n", &temp, &n))
1260 return NULL;
1261 break;
1262 case 3:
1263 if (!PyArg_ParseTuple(args, "Oil;ch or int,n,attr", &temp, &n, &lattr))
1264 return NULL;
1265 attr = lattr;
1266 break;
1267 case 4:
1268 if (!PyArg_ParseTuple(args, "iiOi;y,x,ch or int,n", &y, &x, &temp, &n))
1269 return NULL;
1270 code = wmove(self->win, y, x);
1271 break;
1272 case 5:
1273 if (!PyArg_ParseTuple(args, "iiOil; y,x,ch or int,n,attr",
1274 &y, &x, &temp, &n, &lattr))
1275 return NULL;
1276 attr = lattr;
1277 code = wmove(self->win, y, x);
1278 break;
1279 default:
1280 PyErr_SetString(PyExc_TypeError, "hline requires 2 to 5 arguments");
1281 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001282 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001283
1284 if (code != ERR) {
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001285 if (!PyCurses_ConvertToChtype(self, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00001286 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00001287 return PyCursesCheckERR(whline(self->win, ch | attr, n), "hline");
1288 } else
1289 return PyCursesCheckERR(code, "wmove");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001290}
1291
1292static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001293PyCursesWindow_InsCh(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001294{
Victor Stinner26486ea2010-05-15 22:23:53 +00001295 int rtn, x, y, use_xy = FALSE;
1296 PyObject *temp;
1297 chtype ch = 0;
1298 attr_t attr = A_NORMAL;
1299 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001300
Victor Stinner26486ea2010-05-15 22:23:53 +00001301 switch (PyTuple_Size(args)) {
1302 case 1:
1303 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
1304 return NULL;
1305 break;
1306 case 2:
1307 if (!PyArg_ParseTuple(args, "Ol;ch or int,attr", &temp, &lattr))
1308 return NULL;
1309 attr = lattr;
1310 break;
1311 case 3:
1312 if (!PyArg_ParseTuple(args,"iiO;y,x,ch or int", &y, &x, &temp))
1313 return NULL;
1314 use_xy = TRUE;
1315 break;
1316 case 4:
1317 if (!PyArg_ParseTuple(args,"iiOl;y,x,ch or int, attr", &y, &x, &temp, &lattr))
1318 return NULL;
1319 attr = lattr;
1320 use_xy = TRUE;
1321 break;
1322 default:
1323 PyErr_SetString(PyExc_TypeError, "insch requires 1 or 4 arguments");
1324 return NULL;
1325 }
1326
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001327 if (!PyCurses_ConvertToChtype(self, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00001328 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00001329
1330 if (use_xy == TRUE)
1331 rtn = mvwinsch(self->win,y,x, ch | attr);
1332 else {
1333 rtn = winsch(self->win, ch | attr);
1334 }
1335 return PyCursesCheckERR(rtn, "insch");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001336}
1337
1338static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001339PyCursesWindow_InCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001340{
Victor Stinner26486ea2010-05-15 22:23:53 +00001341 int x, y, rtn;
Guido van Rossum85738471995-02-17 13:50:17 +00001342
Victor Stinner26486ea2010-05-15 22:23:53 +00001343 switch (PyTuple_Size(args)) {
1344 case 0:
1345 rtn = winch(self->win);
1346 break;
1347 case 2:
1348 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1349 return NULL;
1350 rtn = mvwinch(self->win,y,x);
1351 break;
1352 default:
1353 PyErr_SetString(PyExc_TypeError, "inch requires 0 or 2 arguments");
1354 return NULL;
1355 }
1356 return PyLong_FromLong((long) rtn);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001357}
1358
1359static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001360PyCursesWindow_InStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001361{
Victor Stinner26486ea2010-05-15 22:23:53 +00001362 int x, y, n;
1363 char rtn[1024]; /* This should be big enough.. I hope */
1364 int rtn2;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001365
Victor Stinner26486ea2010-05-15 22:23:53 +00001366 switch (PyTuple_Size(args)) {
1367 case 0:
1368 rtn2 = winnstr(self->win,rtn, 1023);
1369 break;
1370 case 1:
1371 if (!PyArg_ParseTuple(args,"i;n", &n))
1372 return NULL;
Victor Stinner640c35c2013-06-04 23:14:37 +02001373 rtn2 = winnstr(self->win, rtn, Py_MIN(n, 1023));
Victor Stinner26486ea2010-05-15 22:23:53 +00001374 break;
1375 case 2:
1376 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1377 return NULL;
1378 rtn2 = mvwinnstr(self->win,y,x,rtn,1023);
1379 break;
1380 case 3:
1381 if (!PyArg_ParseTuple(args, "iii;y,x,n", &y, &x, &n))
1382 return NULL;
Victor Stinner640c35c2013-06-04 23:14:37 +02001383 rtn2 = mvwinnstr(self->win, y, x, rtn, Py_MIN(n,1023));
Victor Stinner26486ea2010-05-15 22:23:53 +00001384 break;
1385 default:
1386 PyErr_SetString(PyExc_TypeError, "instr requires 0 or 3 arguments");
1387 return NULL;
1388 }
1389 if (rtn2 == ERR)
1390 rtn[0] = 0;
1391 return PyBytes_FromString(rtn);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001392}
1393
1394static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001395PyCursesWindow_InsStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001396{
Victor Stinner26486ea2010-05-15 22:23:53 +00001397 int rtn;
1398 int x, y;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001399 int strtype;
Benjamin Peterson5e545192012-03-14 18:21:35 -05001400 PyObject *strobj, *bytesobj = NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001401#ifdef HAVE_NCURSESW
1402 wchar_t *wstr = NULL;
1403#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001404 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
1405 long lattr;
1406 int use_xy = FALSE, use_attr = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001407 const char *funcname;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001408
Victor Stinner26486ea2010-05-15 22:23:53 +00001409 switch (PyTuple_Size(args)) {
1410 case 1:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001411 if (!PyArg_ParseTuple(args,"O;str", &strobj))
Victor Stinner26486ea2010-05-15 22:23:53 +00001412 return NULL;
1413 break;
1414 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001415 if (!PyArg_ParseTuple(args,"Ol;str,attr", &strobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +00001416 return NULL;
1417 attr = lattr;
1418 use_attr = TRUE;
1419 break;
1420 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001421 if (!PyArg_ParseTuple(args,"iiO;y,x,str", &y, &x, &strobj))
Victor Stinner26486ea2010-05-15 22:23:53 +00001422 return NULL;
1423 use_xy = TRUE;
1424 break;
1425 case 4:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001426 if (!PyArg_ParseTuple(args,"iiOl;y,x,str,attr", &y, &x, &strobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +00001427 return NULL;
1428 attr = lattr;
1429 use_xy = use_attr = TRUE;
1430 break;
1431 default:
1432 PyErr_SetString(PyExc_TypeError, "insstr requires 1 to 4 arguments");
1433 return NULL;
1434 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001435
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001436#ifdef HAVE_NCURSESW
1437 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);
1438#else
1439 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);
1440#endif
1441 if (strtype == 0)
1442 return NULL;
1443
Victor Stinner26486ea2010-05-15 22:23:53 +00001444 if (use_attr == TRUE) {
1445 attr_old = getattrs(self->win);
1446 (void)wattrset(self->win,attr);
1447 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001448#ifdef HAVE_NCURSESW
1449 if (strtype == 2) {
1450 funcname = "inswstr";
1451 if (use_xy == TRUE)
1452 rtn = mvwins_wstr(self->win,y,x,wstr);
1453 else
1454 rtn = wins_wstr(self->win,wstr);
1455 PyMem_Free(wstr);
1456 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001457 else
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001458#endif
1459 {
1460 char *str = PyBytes_AS_STRING(bytesobj);
1461 funcname = "insstr";
1462 if (use_xy == TRUE)
1463 rtn = mvwinsstr(self->win,y,x,str);
1464 else
1465 rtn = winsstr(self->win,str);
1466 Py_DECREF(bytesobj);
1467 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001468 if (use_attr == TRUE)
1469 (void)wattrset(self->win,attr_old);
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001470 return PyCursesCheckERR(rtn, funcname);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001471}
1472
1473static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001474PyCursesWindow_InsNStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001475{
Victor Stinner26486ea2010-05-15 22:23:53 +00001476 int rtn, x, y, n;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001477 int strtype;
Benjamin Peterson5e545192012-03-14 18:21:35 -05001478 PyObject *strobj, *bytesobj = NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001479#ifdef HAVE_NCURSESW
1480 wchar_t *wstr = NULL;
1481#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001482 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
1483 long lattr;
1484 int use_xy = FALSE, use_attr = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001485 const char *funcname;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001486
Victor Stinner26486ea2010-05-15 22:23:53 +00001487 switch (PyTuple_Size(args)) {
1488 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001489 if (!PyArg_ParseTuple(args,"Oi;str,n", &strobj, &n))
Victor Stinner26486ea2010-05-15 22:23:53 +00001490 return NULL;
1491 break;
1492 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001493 if (!PyArg_ParseTuple(args,"Oil;str,n,attr", &strobj, &n, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +00001494 return NULL;
1495 attr = lattr;
1496 use_attr = TRUE;
1497 break;
1498 case 4:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001499 if (!PyArg_ParseTuple(args,"iiOi;y,x,str,n", &y, &x, &strobj, &n))
Victor Stinner26486ea2010-05-15 22:23:53 +00001500 return NULL;
1501 use_xy = TRUE;
1502 break;
1503 case 5:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001504 if (!PyArg_ParseTuple(args,"iiOil;y,x,str,n,attr", &y, &x, &strobj, &n, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +00001505 return NULL;
1506 attr = lattr;
1507 use_xy = use_attr = TRUE;
1508 break;
1509 default:
1510 PyErr_SetString(PyExc_TypeError, "insnstr requires 2 to 5 arguments");
1511 return NULL;
1512 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001513
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001514#ifdef HAVE_NCURSESW
1515 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);
1516#else
1517 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);
1518#endif
1519 if (strtype == 0)
1520 return NULL;
1521
Victor Stinner26486ea2010-05-15 22:23:53 +00001522 if (use_attr == TRUE) {
1523 attr_old = getattrs(self->win);
1524 (void)wattrset(self->win,attr);
1525 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001526#ifdef HAVE_NCURSESW
1527 if (strtype == 2) {
1528 funcname = "insn_wstr";
1529 if (use_xy == TRUE)
1530 rtn = mvwins_nwstr(self->win,y,x,wstr,n);
1531 else
1532 rtn = wins_nwstr(self->win,wstr,n);
1533 PyMem_Free(wstr);
1534 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001535 else
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001536#endif
1537 {
1538 char *str = PyBytes_AS_STRING(bytesobj);
1539 funcname = "insnstr";
1540 if (use_xy == TRUE)
1541 rtn = mvwinsnstr(self->win,y,x,str,n);
1542 else
1543 rtn = winsnstr(self->win,str,n);
1544 Py_DECREF(bytesobj);
1545 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001546 if (use_attr == TRUE)
1547 (void)wattrset(self->win,attr_old);
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001548 return PyCursesCheckERR(rtn, funcname);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001549}
1550
1551static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001552PyCursesWindow_Is_LineTouched(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001553{
Victor Stinner26486ea2010-05-15 22:23:53 +00001554 int line, erg;
1555 if (!PyArg_ParseTuple(args,"i;line", &line))
1556 return NULL;
1557 erg = is_linetouched(self->win, line);
1558 if (erg == ERR) {
1559 PyErr_SetString(PyExc_TypeError,
1560 "is_linetouched: line number outside of boundaries");
1561 return NULL;
1562 } else
1563 if (erg == FALSE) {
1564 Py_INCREF(Py_False);
1565 return Py_False;
1566 } else {
1567 Py_INCREF(Py_True);
1568 return Py_True;
1569 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001570}
1571
1572static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001573PyCursesWindow_NoOutRefresh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001574{
Victor Stinner26486ea2010-05-15 22:23:53 +00001575 int pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol;
1576 int rtn;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001577
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00001578#ifndef WINDOW_HAS_FLAGS
Victor Stinner26486ea2010-05-15 22:23:53 +00001579 if (0)
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001580#else
Victor Stinner26486ea2010-05-15 22:23:53 +00001581 if (self->win->_flags & _ISPAD)
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001582#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001583 {
1584 switch(PyTuple_Size(args)) {
1585 case 6:
1586 if (!PyArg_ParseTuple(args,
1587 "iiiiii;" \
1588 "pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol",
1589 &pminrow, &pmincol, &sminrow,
1590 &smincol, &smaxrow, &smaxcol))
1591 return NULL;
1592 Py_BEGIN_ALLOW_THREADS
1593 rtn = pnoutrefresh(self->win,
1594 pminrow, pmincol, sminrow,
1595 smincol, smaxrow, smaxcol);
1596 Py_END_ALLOW_THREADS
1597 return PyCursesCheckERR(rtn, "pnoutrefresh");
1598 default:
1599 PyErr_SetString(PyCursesError,
1600 "noutrefresh() called for a pad "
1601 "requires 6 arguments");
1602 return NULL;
1603 }
1604 } else {
1605 if (!PyArg_ParseTuple(args, ":noutrefresh"))
1606 return NULL;
Andrew M. Kuchlingf4195722000-06-23 01:36:21 +00001607
Victor Stinner26486ea2010-05-15 22:23:53 +00001608 Py_BEGIN_ALLOW_THREADS
1609 rtn = wnoutrefresh(self->win);
1610 Py_END_ALLOW_THREADS
1611 return PyCursesCheckERR(rtn, "wnoutrefresh");
1612 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001613}
1614
1615static PyObject *
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001616PyCursesWindow_Overlay(PyCursesWindowObject *self, PyObject *args)
1617{
1618 PyCursesWindowObject *temp;
1619 int use_copywin = FALSE;
1620 int sminrow, smincol, dminrow, dmincol, dmaxrow, dmaxcol;
1621 int rtn;
Victor Stinner26486ea2010-05-15 22:23:53 +00001622
Martin v. Löwisc0e16712002-01-17 23:08:27 +00001623 switch (PyTuple_Size(args)) {
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001624 case 1:
Victor Stinner26486ea2010-05-15 22:23:53 +00001625 if (!PyArg_ParseTuple(args, "O!;window object",
1626 &PyCursesWindow_Type, &temp))
1627 return NULL;
1628 break;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001629 case 7:
Victor Stinner26486ea2010-05-15 22:23:53 +00001630 if (!PyArg_ParseTuple(args, "O!iiiiii;window object, int, int, int, int, int, int",
1631 &PyCursesWindow_Type, &temp, &sminrow, &smincol,
1632 &dminrow, &dmincol, &dmaxrow, &dmaxcol))
1633 return NULL;
1634 use_copywin = TRUE;
1635 break;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001636 default:
Victor Stinner26486ea2010-05-15 22:23:53 +00001637 PyErr_SetString(PyExc_TypeError,
1638 "overlay requires one or seven arguments");
1639 return NULL;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001640 }
1641
1642 if (use_copywin == TRUE) {
Victor Stinner26486ea2010-05-15 22:23:53 +00001643 rtn = copywin(self->win, temp->win, sminrow, smincol,
1644 dminrow, dmincol, dmaxrow, dmaxcol, TRUE);
1645 return PyCursesCheckERR(rtn, "copywin");
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001646 }
1647 else {
Victor Stinner26486ea2010-05-15 22:23:53 +00001648 rtn = overlay(self->win, temp->win);
1649 return PyCursesCheckERR(rtn, "overlay");
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001650 }
1651}
1652
1653static PyObject *
1654PyCursesWindow_Overwrite(PyCursesWindowObject *self, PyObject *args)
1655{
1656 PyCursesWindowObject *temp;
1657 int use_copywin = FALSE;
1658 int sminrow, smincol, dminrow, dmincol, dmaxrow, dmaxcol;
1659 int rtn;
Victor Stinner26486ea2010-05-15 22:23:53 +00001660
Martin v. Löwisc0e16712002-01-17 23:08:27 +00001661 switch (PyTuple_Size(args)) {
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001662 case 1:
Victor Stinner26486ea2010-05-15 22:23:53 +00001663 if (!PyArg_ParseTuple(args, "O!;window object",
1664 &PyCursesWindow_Type, &temp))
1665 return NULL;
1666 break;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001667 case 7:
Victor Stinner26486ea2010-05-15 22:23:53 +00001668 if (!PyArg_ParseTuple(args, "O!iiiiii;window object, int, int, int, int, int, int",
1669 &PyCursesWindow_Type, &temp, &sminrow, &smincol,
1670 &dminrow, &dmincol, &dmaxrow, &dmaxcol))
1671 return NULL;
1672 use_copywin = TRUE;
1673 break;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001674 default:
Victor Stinner26486ea2010-05-15 22:23:53 +00001675 PyErr_SetString(PyExc_TypeError,
1676 "overwrite requires one or seven arguments");
1677 return NULL;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001678 }
1679
1680 if (use_copywin == TRUE) {
Victor Stinner26486ea2010-05-15 22:23:53 +00001681 rtn = copywin(self->win, temp->win, sminrow, smincol,
1682 dminrow, dmincol, dmaxrow, dmaxcol, FALSE);
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001683 return PyCursesCheckERR(rtn, "copywin");
1684 }
1685 else {
Victor Stinner26486ea2010-05-15 22:23:53 +00001686 rtn = overwrite(self->win, temp->win);
1687 return PyCursesCheckERR(rtn, "overwrite");
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001688 }
1689}
1690
1691static PyObject *
Guido van Rossum6c95da32007-07-24 00:16:38 +00001692PyCursesWindow_PutWin(PyCursesWindowObject *self, PyObject *stream)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001693{
Victor Stinner26486ea2010-05-15 22:23:53 +00001694 /* We have to simulate this by writing to a temporary FILE*,
1695 then reading back, then writing to the argument stream. */
1696 char fn[100];
Victor Stinnerdaf45552013-08-28 00:53:59 +02001697 int fd = -1;
1698 FILE *fp = NULL;
1699 PyObject *res = NULL;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00001700
Victor Stinner26486ea2010-05-15 22:23:53 +00001701 strcpy(fn, "/tmp/py.curses.putwin.XXXXXX");
1702 fd = mkstemp(fn);
1703 if (fd < 0)
1704 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
Victor Stinnerdaf45552013-08-28 00:53:59 +02001705 if (_Py_set_inheritable(fd, 0, NULL) < 0)
1706 goto exit;
Victor Stinner26486ea2010-05-15 22:23:53 +00001707 fp = fdopen(fd, "wb+");
1708 if (fp == NULL) {
Victor Stinnerdaf45552013-08-28 00:53:59 +02001709 PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
1710 goto exit;
Victor Stinner26486ea2010-05-15 22:23:53 +00001711 }
1712 res = PyCursesCheckERR(putwin(self->win, fp), "putwin");
Victor Stinnerdaf45552013-08-28 00:53:59 +02001713 if (res == NULL)
1714 goto exit;
Victor Stinner26486ea2010-05-15 22:23:53 +00001715 fseek(fp, 0, 0);
1716 while (1) {
1717 char buf[BUFSIZ];
1718 Py_ssize_t n = fread(buf, 1, BUFSIZ, fp);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02001719 _Py_IDENTIFIER(write);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001720
Victor Stinner26486ea2010-05-15 22:23:53 +00001721 if (n <= 0)
1722 break;
1723 Py_DECREF(res);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001724 res = _PyObject_CallMethodId(stream, &PyId_write, "y#", buf, n);
Victor Stinner26486ea2010-05-15 22:23:53 +00001725 if (res == NULL)
1726 break;
1727 }
Victor Stinnerdaf45552013-08-28 00:53:59 +02001728
1729exit:
1730 if (fp != NULL)
1731 fclose(fp);
1732 else if (fd != -1)
1733 close(fd);
Guido van Rossum150b7d72007-08-30 23:34:01 +00001734 remove(fn);
Guido van Rossum6c95da32007-07-24 00:16:38 +00001735 return res;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001736}
1737
1738static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001739PyCursesWindow_RedrawLine(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001740{
Victor Stinner26486ea2010-05-15 22:23:53 +00001741 int beg, num;
1742 if (!PyArg_ParseTuple(args, "ii;beg,num", &beg, &num))
1743 return NULL;
1744 return PyCursesCheckERR(wredrawln(self->win,beg,num), "redrawln");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001745}
1746
1747static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001748PyCursesWindow_Refresh(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001749{
Victor Stinner26486ea2010-05-15 22:23:53 +00001750 int pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol;
1751 int rtn;
Andrew M. Kuchlingf4195722000-06-23 01:36:21 +00001752
Victor Stinner26486ea2010-05-15 22:23:53 +00001753#ifndef WINDOW_HAS_FLAGS
1754 if (0)
1755#else
1756 if (self->win->_flags & _ISPAD)
1757#endif
1758 {
1759 switch(PyTuple_Size(args)) {
1760 case 6:
1761 if (!PyArg_ParseTuple(args,
1762 "iiiiii;" \
1763 "pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol",
1764 &pminrow, &pmincol, &sminrow,
1765 &smincol, &smaxrow, &smaxcol))
1766 return NULL;
1767
1768 Py_BEGIN_ALLOW_THREADS
1769 rtn = prefresh(self->win,
1770 pminrow, pmincol, sminrow,
1771 smincol, smaxrow, smaxcol);
1772 Py_END_ALLOW_THREADS
1773 return PyCursesCheckERR(rtn, "prefresh");
1774 default:
1775 PyErr_SetString(PyCursesError,
1776 "refresh() for a pad requires 6 arguments");
1777 return NULL;
1778 }
1779 } else {
1780 if (!PyArg_ParseTuple(args, ":refresh"))
1781 return NULL;
1782 Py_BEGIN_ALLOW_THREADS
1783 rtn = wrefresh(self->win);
1784 Py_END_ALLOW_THREADS
1785 return PyCursesCheckERR(rtn, "prefresh");
1786 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001787}
1788
1789static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001790PyCursesWindow_SetScrollRegion(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001791{
Victor Stinner26486ea2010-05-15 22:23:53 +00001792 int x, y;
1793 if (!PyArg_ParseTuple(args,"ii;top, bottom",&y,&x))
1794 return NULL;
1795 return PyCursesCheckERR(wsetscrreg(self->win,y,x), "wsetscrreg");
Guido van Rossumf6971e21994-08-30 12:25:20 +00001796}
1797
1798static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001799PyCursesWindow_SubWin(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001800{
Victor Stinner26486ea2010-05-15 22:23:53 +00001801 WINDOW *win;
1802 int nlines, ncols, begin_y, begin_x;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001803
Victor Stinner26486ea2010-05-15 22:23:53 +00001804 nlines = 0;
1805 ncols = 0;
1806 switch (PyTuple_Size(args)) {
1807 case 2:
1808 if (!PyArg_ParseTuple(args,"ii;begin_y,begin_x",&begin_y,&begin_x))
1809 return NULL;
1810 break;
1811 case 4:
1812 if (!PyArg_ParseTuple(args, "iiii;nlines,ncols,begin_y,begin_x",
1813 &nlines,&ncols,&begin_y,&begin_x))
1814 return NULL;
1815 break;
1816 default:
1817 PyErr_SetString(PyExc_TypeError, "subwin requires 2 or 4 arguments");
1818 return NULL;
1819 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001820
Victor Stinner26486ea2010-05-15 22:23:53 +00001821 /* printf("Subwin: %i %i %i %i \n", nlines, ncols, begin_y, begin_x); */
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00001822#ifdef WINDOW_HAS_FLAGS
Victor Stinner26486ea2010-05-15 22:23:53 +00001823 if (self->win->_flags & _ISPAD)
1824 win = subpad(self->win, nlines, ncols, begin_y, begin_x);
1825 else
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001826#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001827 win = subwin(self->win, nlines, ncols, begin_y, begin_x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001828
Victor Stinner26486ea2010-05-15 22:23:53 +00001829 if (win == NULL) {
1830 PyErr_SetString(PyCursesError, catchall_NULL);
1831 return NULL;
1832 }
1833
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001834 return (PyObject *)PyCursesWindow_New(win, self->encoding);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001835}
1836
1837static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001838PyCursesWindow_Scroll(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001839{
Victor Stinner26486ea2010-05-15 22:23:53 +00001840 int nlines;
1841 switch(PyTuple_Size(args)) {
1842 case 0:
1843 return PyCursesCheckERR(scroll(self->win), "scroll");
1844 case 1:
1845 if (!PyArg_ParseTuple(args, "i;nlines", &nlines))
1846 return NULL;
1847 return PyCursesCheckERR(wscrl(self->win, nlines), "scroll");
1848 default:
1849 PyErr_SetString(PyExc_TypeError, "scroll requires 0 or 1 arguments");
1850 return NULL;
1851 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001852}
1853
1854static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001855PyCursesWindow_TouchLine(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001856{
Victor Stinner26486ea2010-05-15 22:23:53 +00001857 int st, cnt, val;
1858 switch (PyTuple_Size(args)) {
1859 case 2:
1860 if (!PyArg_ParseTuple(args,"ii;start,count",&st,&cnt))
1861 return NULL;
1862 return PyCursesCheckERR(touchline(self->win,st,cnt), "touchline");
1863 case 3:
1864 if (!PyArg_ParseTuple(args, "iii;start,count,val", &st, &cnt, &val))
1865 return NULL;
1866 return PyCursesCheckERR(wtouchln(self->win, st, cnt, val), "touchline");
1867 default:
1868 PyErr_SetString(PyExc_TypeError, "touchline requires 2 or 3 arguments");
1869 return NULL;
1870 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001871}
1872
1873static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001874PyCursesWindow_Vline(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001875{
Victor Stinner26486ea2010-05-15 22:23:53 +00001876 PyObject *temp;
1877 chtype ch;
1878 int n, x, y, code = OK;
1879 attr_t attr = A_NORMAL;
1880 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001881
Victor Stinner26486ea2010-05-15 22:23:53 +00001882 switch (PyTuple_Size(args)) {
1883 case 2:
1884 if (!PyArg_ParseTuple(args, "Oi;ch or int,n", &temp, &n))
1885 return NULL;
1886 break;
1887 case 3:
1888 if (!PyArg_ParseTuple(args, "Oil;ch or int,n,attr", &temp, &n, &lattr))
1889 return NULL;
1890 attr = lattr;
1891 break;
1892 case 4:
1893 if (!PyArg_ParseTuple(args, "iiOi;y,x,ch or int,n", &y, &x, &temp, &n))
1894 return NULL;
1895 code = wmove(self->win, y, x);
1896 break;
1897 case 5:
1898 if (!PyArg_ParseTuple(args, "iiOil; y,x,ch or int,n,attr",
1899 &y, &x, &temp, &n, &lattr))
1900 return NULL;
1901 attr = lattr;
1902 code = wmove(self->win, y, x);
1903 break;
1904 default:
1905 PyErr_SetString(PyExc_TypeError, "vline requires 2 to 5 arguments");
1906 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001907 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001908
1909 if (code != ERR) {
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001910 if (!PyCurses_ConvertToChtype(self, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00001911 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00001912 return PyCursesCheckERR(wvline(self->win, ch | attr, n), "vline");
1913 } else
1914 return PyCursesCheckERR(code, "wmove");
Guido van Rossumf6971e21994-08-30 12:25:20 +00001915}
1916
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001917static PyObject *
1918PyCursesWindow_get_encoding(PyCursesWindowObject *self, void *closure)
1919{
1920 return PyUnicode_FromString(self->encoding);
1921}
1922
1923static int
1924PyCursesWindow_set_encoding(PyCursesWindowObject *self, PyObject *value)
1925{
1926 PyObject *ascii;
1927 char *encoding;
1928
1929 /* It is illegal to del win.encoding */
1930 if (value == NULL) {
1931 PyErr_SetString(PyExc_TypeError,
1932 "encoding may not be deleted");
1933 return -1;
1934 }
1935
1936 if (!PyUnicode_Check(value)) {
1937 PyErr_SetString(PyExc_TypeError,
1938 "setting encoding to a non-string");
1939 return -1;
1940 }
1941 ascii = PyUnicode_AsASCIIString(value);
1942 if (ascii == NULL)
1943 return -1;
Victor Stinner49fc8ec2013-07-07 23:30:24 +02001944 encoding = _PyMem_Strdup(PyBytes_AS_STRING(ascii));
Ross Lagerwall2dabaf62012-09-07 08:34:23 +02001945 Py_DECREF(ascii);
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001946 if (encoding == NULL) {
1947 PyErr_NoMemory();
1948 return -1;
1949 }
Victor Stinner49fc8ec2013-07-07 23:30:24 +02001950 PyMem_Free(self->encoding);
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001951 self->encoding = encoding;
1952 return 0;
1953}
1954
1955
Guido van Rossumf6971e21994-08-30 12:25:20 +00001956static PyMethodDef PyCursesWindow_Methods[] = {
Victor Stinner26486ea2010-05-15 22:23:53 +00001957 {"addch", (PyCFunction)PyCursesWindow_AddCh, METH_VARARGS},
1958 {"addnstr", (PyCFunction)PyCursesWindow_AddNStr, METH_VARARGS},
1959 {"addstr", (PyCFunction)PyCursesWindow_AddStr, METH_VARARGS},
1960 {"attroff", (PyCFunction)PyCursesWindow_AttrOff, METH_VARARGS},
1961 {"attron", (PyCFunction)PyCursesWindow_AttrOn, METH_VARARGS},
1962 {"attrset", (PyCFunction)PyCursesWindow_AttrSet, METH_VARARGS},
1963 {"bkgd", (PyCFunction)PyCursesWindow_Bkgd, METH_VARARGS},
1964 {"chgat", (PyCFunction)PyCursesWindow_ChgAt, METH_VARARGS},
1965 {"bkgdset", (PyCFunction)PyCursesWindow_BkgdSet, METH_VARARGS},
1966 {"border", (PyCFunction)PyCursesWindow_Border, METH_VARARGS},
1967 {"box", (PyCFunction)PyCursesWindow_Box, METH_VARARGS},
1968 {"clear", (PyCFunction)PyCursesWindow_wclear, METH_NOARGS},
1969 {"clearok", (PyCFunction)PyCursesWindow_clearok, METH_VARARGS},
1970 {"clrtobot", (PyCFunction)PyCursesWindow_wclrtobot, METH_NOARGS},
1971 {"clrtoeol", (PyCFunction)PyCursesWindow_wclrtoeol, METH_NOARGS},
1972 {"cursyncup", (PyCFunction)PyCursesWindow_wcursyncup, METH_NOARGS},
1973 {"delch", (PyCFunction)PyCursesWindow_DelCh, METH_VARARGS},
1974 {"deleteln", (PyCFunction)PyCursesWindow_wdeleteln, METH_NOARGS},
1975 {"derwin", (PyCFunction)PyCursesWindow_DerWin, METH_VARARGS},
1976 {"echochar", (PyCFunction)PyCursesWindow_EchoChar, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001977#ifdef NCURSES_MOUSE_VERSION
Victor Stinner26486ea2010-05-15 22:23:53 +00001978 {"enclose", (PyCFunction)PyCursesWindow_Enclose, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001979#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001980 {"erase", (PyCFunction)PyCursesWindow_werase, METH_NOARGS},
1981 {"getbegyx", (PyCFunction)PyCursesWindow_getbegyx, METH_NOARGS},
1982 {"getbkgd", (PyCFunction)PyCursesWindow_GetBkgd, METH_NOARGS},
1983 {"getch", (PyCFunction)PyCursesWindow_GetCh, METH_VARARGS},
1984 {"getkey", (PyCFunction)PyCursesWindow_GetKey, METH_VARARGS},
Nadeem Vawda9e2e9902011-07-31 15:01:11 +02001985#ifdef HAVE_NCURSESW
Victor Stinnera7878b72011-07-14 23:07:44 +02001986 {"get_wch", (PyCFunction)PyCursesWindow_Get_WCh, METH_VARARGS},
Nadeem Vawda9e2e9902011-07-31 15:01:11 +02001987#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001988 {"getmaxyx", (PyCFunction)PyCursesWindow_getmaxyx, METH_NOARGS},
1989 {"getparyx", (PyCFunction)PyCursesWindow_getparyx, METH_NOARGS},
1990 {"getstr", (PyCFunction)PyCursesWindow_GetStr, METH_VARARGS},
1991 {"getyx", (PyCFunction)PyCursesWindow_getyx, METH_NOARGS},
1992 {"hline", (PyCFunction)PyCursesWindow_Hline, METH_VARARGS},
1993 {"idcok", (PyCFunction)PyCursesWindow_idcok, METH_VARARGS},
1994 {"idlok", (PyCFunction)PyCursesWindow_idlok, METH_VARARGS},
1995 {"immedok", (PyCFunction)PyCursesWindow_immedok, METH_VARARGS},
1996 {"inch", (PyCFunction)PyCursesWindow_InCh, METH_VARARGS},
1997 {"insch", (PyCFunction)PyCursesWindow_InsCh, METH_VARARGS},
1998 {"insdelln", (PyCFunction)PyCursesWindow_winsdelln, METH_VARARGS},
1999 {"insertln", (PyCFunction)PyCursesWindow_winsertln, METH_NOARGS},
2000 {"insnstr", (PyCFunction)PyCursesWindow_InsNStr, METH_VARARGS},
2001 {"insstr", (PyCFunction)PyCursesWindow_InsStr, METH_VARARGS},
2002 {"instr", (PyCFunction)PyCursesWindow_InStr, METH_VARARGS},
2003 {"is_linetouched", (PyCFunction)PyCursesWindow_Is_LineTouched, METH_VARARGS},
2004 {"is_wintouched", (PyCFunction)PyCursesWindow_is_wintouched, METH_NOARGS},
2005 {"keypad", (PyCFunction)PyCursesWindow_keypad, METH_VARARGS},
2006 {"leaveok", (PyCFunction)PyCursesWindow_leaveok, METH_VARARGS},
2007 {"move", (PyCFunction)PyCursesWindow_wmove, METH_VARARGS},
2008 {"mvderwin", (PyCFunction)PyCursesWindow_mvderwin, METH_VARARGS},
2009 {"mvwin", (PyCFunction)PyCursesWindow_mvwin, METH_VARARGS},
2010 {"nodelay", (PyCFunction)PyCursesWindow_nodelay, METH_VARARGS},
2011 {"notimeout", (PyCFunction)PyCursesWindow_notimeout, METH_VARARGS},
2012 {"noutrefresh", (PyCFunction)PyCursesWindow_NoOutRefresh, METH_VARARGS},
2013 /* Backward compatibility alias -- remove in Python 2.3 */
2014 {"nooutrefresh", (PyCFunction)PyCursesWindow_NoOutRefresh, METH_VARARGS},
2015 {"overlay", (PyCFunction)PyCursesWindow_Overlay, METH_VARARGS},
2016 {"overwrite", (PyCFunction)PyCursesWindow_Overwrite,
2017 METH_VARARGS},
2018 {"putwin", (PyCFunction)PyCursesWindow_PutWin, METH_O},
2019 {"redrawln", (PyCFunction)PyCursesWindow_RedrawLine, METH_VARARGS},
2020 {"redrawwin", (PyCFunction)PyCursesWindow_redrawwin, METH_NOARGS},
2021 {"refresh", (PyCFunction)PyCursesWindow_Refresh, METH_VARARGS},
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +00002022#ifndef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00002023 {"resize", (PyCFunction)PyCursesWindow_wresize, METH_VARARGS},
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002024#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00002025 {"scroll", (PyCFunction)PyCursesWindow_Scroll, METH_VARARGS},
2026 {"scrollok", (PyCFunction)PyCursesWindow_scrollok, METH_VARARGS},
2027 {"setscrreg", (PyCFunction)PyCursesWindow_SetScrollRegion, METH_VARARGS},
2028 {"standend", (PyCFunction)PyCursesWindow_wstandend, METH_NOARGS},
2029 {"standout", (PyCFunction)PyCursesWindow_wstandout, METH_NOARGS},
2030 {"subpad", (PyCFunction)PyCursesWindow_SubWin, METH_VARARGS},
2031 {"subwin", (PyCFunction)PyCursesWindow_SubWin, METH_VARARGS},
2032 {"syncdown", (PyCFunction)PyCursesWindow_wsyncdown, METH_NOARGS},
2033 {"syncok", (PyCFunction)PyCursesWindow_syncok, METH_VARARGS},
2034 {"syncup", (PyCFunction)PyCursesWindow_wsyncup, METH_NOARGS},
2035 {"timeout", (PyCFunction)PyCursesWindow_wtimeout, METH_VARARGS},
2036 {"touchline", (PyCFunction)PyCursesWindow_TouchLine, METH_VARARGS},
2037 {"touchwin", (PyCFunction)PyCursesWindow_touchwin, METH_NOARGS},
2038 {"untouchwin", (PyCFunction)PyCursesWindow_untouchwin, METH_NOARGS},
2039 {"vline", (PyCFunction)PyCursesWindow_Vline, METH_VARARGS},
2040 {NULL, NULL} /* sentinel */
Guido van Rossumf6971e21994-08-30 12:25:20 +00002041};
2042
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002043static PyGetSetDef PyCursesWindow_getsets[] = {
2044 {"encoding",
2045 (getter)PyCursesWindow_get_encoding,
2046 (setter)PyCursesWindow_set_encoding,
Ronald Oussoren2a7fe032012-01-17 16:53:20 +01002047 "the typecode character used to create the array"},
2048 {NULL, NULL, NULL, NULL } /* sentinel */
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002049};
2050
Guido van Rossumf6971e21994-08-30 12:25:20 +00002051/* -------------------------------------------------------*/
Guido van Rossum85738471995-02-17 13:50:17 +00002052
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002053PyTypeObject PyCursesWindow_Type = {
Victor Stinner26486ea2010-05-15 22:23:53 +00002054 PyVarObject_HEAD_INIT(NULL, 0)
2055 "_curses.curses window", /*tp_name*/
2056 sizeof(PyCursesWindowObject), /*tp_basicsize*/
2057 0, /*tp_itemsize*/
2058 /* methods */
2059 (destructor)PyCursesWindow_Dealloc, /*tp_dealloc*/
2060 0, /*tp_print*/
2061 (getattrfunc)0, /*tp_getattr*/
2062 (setattrfunc)0, /*tp_setattr*/
2063 0, /*tp_reserved*/
2064 0, /*tp_repr*/
2065 0, /*tp_as_number*/
2066 0, /*tp_as_sequence*/
2067 0, /*tp_as_mapping*/
2068 0, /*tp_hash*/
2069 0, /*tp_call*/
2070 0, /*tp_str*/
2071 0, /*tp_getattro*/
2072 0, /*tp_setattro*/
2073 0, /*tp_as_buffer*/
2074 Py_TPFLAGS_DEFAULT, /*tp_flags*/
2075 0, /*tp_doc*/
2076 0, /*tp_traverse*/
2077 0, /*tp_clear*/
2078 0, /*tp_richcompare*/
2079 0, /*tp_weaklistoffset*/
2080 0, /*tp_iter*/
2081 0, /*tp_iternext*/
2082 PyCursesWindow_Methods, /*tp_methods*/
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002083 0, /* tp_members */
2084 PyCursesWindow_getsets, /* tp_getset */
Guido van Rossumf6971e21994-08-30 12:25:20 +00002085};
2086
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002087/*********************************************************************
2088 Global Functions
2089**********************************************************************/
Guido van Rossumf6971e21994-08-30 12:25:20 +00002090
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002091NoArgNoReturnFunction(beep)
2092NoArgNoReturnFunction(def_prog_mode)
2093NoArgNoReturnFunction(def_shell_mode)
2094NoArgNoReturnFunction(doupdate)
2095NoArgNoReturnFunction(endwin)
2096NoArgNoReturnFunction(flash)
2097NoArgNoReturnFunction(nocbreak)
2098NoArgNoReturnFunction(noecho)
2099NoArgNoReturnFunction(nonl)
2100NoArgNoReturnFunction(noraw)
2101NoArgNoReturnFunction(reset_prog_mode)
2102NoArgNoReturnFunction(reset_shell_mode)
2103NoArgNoReturnFunction(resetty)
2104NoArgNoReturnFunction(savetty)
2105
2106NoArgOrFlagNoReturnFunction(cbreak)
2107NoArgOrFlagNoReturnFunction(echo)
2108NoArgOrFlagNoReturnFunction(nl)
2109NoArgOrFlagNoReturnFunction(raw)
2110
2111NoArgReturnIntFunction(baudrate)
2112NoArgReturnIntFunction(termattrs)
2113
2114NoArgReturnStringFunction(termname)
2115NoArgReturnStringFunction(longname)
2116
2117NoArgTrueFalseFunction(can_change_color)
2118NoArgTrueFalseFunction(has_colors)
2119NoArgTrueFalseFunction(has_ic)
2120NoArgTrueFalseFunction(has_il)
2121NoArgTrueFalseFunction(isendwin)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002122NoArgNoReturnVoidFunction(flushinp)
2123NoArgNoReturnVoidFunction(noqiflush)
2124
2125static PyObject *
Christian Heimesaf98da12008-01-27 15:18:18 +00002126PyCurses_filter(PyObject *self)
2127{
Victor Stinner26486ea2010-05-15 22:23:53 +00002128 /* not checking for PyCursesInitialised here since filter() must
2129 be called before initscr() */
2130 filter();
2131 Py_INCREF(Py_None);
2132 return Py_None;
Christian Heimesaf98da12008-01-27 15:18:18 +00002133}
2134
2135static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002136PyCurses_Color_Content(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002137{
Victor Stinner26486ea2010-05-15 22:23:53 +00002138 short color,r,g,b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002139
Victor Stinner26486ea2010-05-15 22:23:53 +00002140 PyCursesInitialised;
2141 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002142
Victor Stinner26486ea2010-05-15 22:23:53 +00002143 if (!PyArg_ParseTuple(args, "h:color_content", &color)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002144
Victor Stinner26486ea2010-05-15 22:23:53 +00002145 if (color_content(color, &r, &g, &b) != ERR)
2146 return Py_BuildValue("(iii)", r, g, b);
2147 else {
2148 PyErr_SetString(PyCursesError,
2149 "Argument 1 was out of range. Check value of COLORS.");
2150 return NULL;
2151 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002152}
2153
2154static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002155PyCurses_color_pair(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002156{
Victor Stinner26486ea2010-05-15 22:23:53 +00002157 int n;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002158
Victor Stinner26486ea2010-05-15 22:23:53 +00002159 PyCursesInitialised;
2160 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002161
Victor Stinner26486ea2010-05-15 22:23:53 +00002162 if (!PyArg_ParseTuple(args, "i:color_pair", &n)) return NULL;
2163 return PyLong_FromLong((long) (n << 8));
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002164}
2165
2166static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002167PyCurses_Curs_Set(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002168{
Victor Stinner26486ea2010-05-15 22:23:53 +00002169 int vis,erg;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002170
Victor Stinner26486ea2010-05-15 22:23:53 +00002171 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002172
Victor Stinner26486ea2010-05-15 22:23:53 +00002173 if (!PyArg_ParseTuple(args, "i:curs_set", &vis)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002174
Victor Stinner26486ea2010-05-15 22:23:53 +00002175 erg = curs_set(vis);
2176 if (erg == ERR) return PyCursesCheckERR(erg, "curs_set");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002177
Victor Stinner26486ea2010-05-15 22:23:53 +00002178 return PyLong_FromLong((long) erg);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002179}
2180
2181static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002182PyCurses_Delay_Output(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002183{
Victor Stinner26486ea2010-05-15 22:23:53 +00002184 int ms;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002185
Victor Stinner26486ea2010-05-15 22:23:53 +00002186 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002187
Victor Stinner26486ea2010-05-15 22:23:53 +00002188 if (!PyArg_ParseTuple(args, "i:delay_output", &ms)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002189
Victor Stinner26486ea2010-05-15 22:23:53 +00002190 return PyCursesCheckERR(delay_output(ms), "delay_output");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002191}
2192
2193static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002194PyCurses_EraseChar(PyObject *self)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002195{
Victor Stinner26486ea2010-05-15 22:23:53 +00002196 char ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002197
Victor Stinner26486ea2010-05-15 22:23:53 +00002198 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002199
Victor Stinner26486ea2010-05-15 22:23:53 +00002200 ch = erasechar();
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002201
Victor Stinner26486ea2010-05-15 22:23:53 +00002202 return PyBytes_FromStringAndSize(&ch, 1);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002203}
2204
2205static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002206PyCurses_getsyx(PyObject *self)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002207{
Victor Stinner26486ea2010-05-15 22:23:53 +00002208 int x = 0;
2209 int y = 0;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002210
Victor Stinner26486ea2010-05-15 22:23:53 +00002211 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002212
Victor Stinner26486ea2010-05-15 22:23:53 +00002213 getsyx(y, x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002214
Victor Stinner26486ea2010-05-15 22:23:53 +00002215 return Py_BuildValue("(ii)", y, x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002216}
2217
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002218#ifdef NCURSES_MOUSE_VERSION
2219static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002220PyCurses_GetMouse(PyObject *self)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002221{
Victor Stinner26486ea2010-05-15 22:23:53 +00002222 int rtn;
2223 MEVENT event;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002224
Victor Stinner26486ea2010-05-15 22:23:53 +00002225 PyCursesInitialised;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002226
Victor Stinner26486ea2010-05-15 22:23:53 +00002227 rtn = getmouse( &event );
2228 if (rtn == ERR) {
2229 PyErr_SetString(PyCursesError, "getmouse() returned ERR");
2230 return NULL;
2231 }
2232 return Py_BuildValue("(hiiil)",
2233 (short)event.id,
2234 event.x, event.y, event.z,
2235 (long) event.bstate);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002236}
2237
2238static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002239PyCurses_UngetMouse(PyObject *self, PyObject *args)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002240{
Victor Stinner26486ea2010-05-15 22:23:53 +00002241 MEVENT event;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002242
Victor Stinner26486ea2010-05-15 22:23:53 +00002243 PyCursesInitialised;
2244 if (!PyArg_ParseTuple(args, "hiiil",
2245 &event.id,
2246 &event.x, &event.y, &event.z,
2247 (int *) &event.bstate))
2248 return NULL;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002249
Victor Stinner26486ea2010-05-15 22:23:53 +00002250 return PyCursesCheckERR(ungetmouse(&event), "ungetmouse");
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002251}
2252#endif
2253
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002254static PyObject *
Guido van Rossum6c95da32007-07-24 00:16:38 +00002255PyCurses_GetWin(PyCursesWindowObject *self, PyObject *stream)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002256{
Victor Stinner26486ea2010-05-15 22:23:53 +00002257 char fn[100];
Victor Stinnerdaf45552013-08-28 00:53:59 +02002258 int fd = -1;
2259 FILE *fp = NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00002260 PyObject *data;
2261 size_t datalen;
2262 WINDOW *win;
Benjamin Peterson18b71912013-05-16 15:29:44 -05002263 _Py_IDENTIFIER(read);
Victor Stinnerdaf45552013-08-28 00:53:59 +02002264 PyObject *res = NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002265
Victor Stinner26486ea2010-05-15 22:23:53 +00002266 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002267
Victor Stinner26486ea2010-05-15 22:23:53 +00002268 strcpy(fn, "/tmp/py.curses.getwin.XXXXXX");
2269 fd = mkstemp(fn);
2270 if (fd < 0)
2271 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
Victor Stinnerdaf45552013-08-28 00:53:59 +02002272 if (_Py_set_inheritable(fd, 0, NULL) < 0)
2273 goto error;
Victor Stinner26486ea2010-05-15 22:23:53 +00002274 fp = fdopen(fd, "wb+");
2275 if (fp == NULL) {
Victor Stinnerdaf45552013-08-28 00:53:59 +02002276 PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
2277 goto error;
Victor Stinner26486ea2010-05-15 22:23:53 +00002278 }
Victor Stinnerdaf45552013-08-28 00:53:59 +02002279
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002280 data = _PyObject_CallMethodId(stream, &PyId_read, "");
Victor Stinnerdaf45552013-08-28 00:53:59 +02002281 if (data == NULL)
2282 goto error;
Victor Stinner26486ea2010-05-15 22:23:53 +00002283 if (!PyBytes_Check(data)) {
2284 PyErr_Format(PyExc_TypeError,
2285 "f.read() returned %.100s instead of bytes",
2286 data->ob_type->tp_name);
2287 Py_DECREF(data);
Victor Stinnerdaf45552013-08-28 00:53:59 +02002288 goto error;
Victor Stinner26486ea2010-05-15 22:23:53 +00002289 }
2290 datalen = PyBytes_GET_SIZE(data);
2291 if (fwrite(PyBytes_AS_STRING(data), 1, datalen, fp) != datalen) {
2292 Py_DECREF(data);
Victor Stinnerdaf45552013-08-28 00:53:59 +02002293 PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
2294 goto error;
Victor Stinner26486ea2010-05-15 22:23:53 +00002295 }
Guido van Rossum6c95da32007-07-24 00:16:38 +00002296 Py_DECREF(data);
Victor Stinnerdaf45552013-08-28 00:53:59 +02002297
Victor Stinner26486ea2010-05-15 22:23:53 +00002298 fseek(fp, 0, 0);
2299 win = getwin(fp);
Victor Stinner26486ea2010-05-15 22:23:53 +00002300 if (win == NULL) {
2301 PyErr_SetString(PyCursesError, catchall_NULL);
Victor Stinnerdaf45552013-08-28 00:53:59 +02002302 goto error;
Victor Stinner26486ea2010-05-15 22:23:53 +00002303 }
Victor Stinnerdaf45552013-08-28 00:53:59 +02002304 res = PyCursesWindow_New(win, NULL);
2305
2306error:
2307 if (fp != NULL)
2308 fclose(fp);
2309 else if (fd != -1)
2310 close(fd);
2311 remove(fn);
2312 return res;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002313}
2314
2315static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002316PyCurses_HalfDelay(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002317{
Victor Stinner26486ea2010-05-15 22:23:53 +00002318 unsigned char tenths;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002319
Victor Stinner26486ea2010-05-15 22:23:53 +00002320 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002321
Victor Stinner26486ea2010-05-15 22:23:53 +00002322 if (!PyArg_ParseTuple(args, "b:halfdelay", &tenths)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002323
Victor Stinner26486ea2010-05-15 22:23:53 +00002324 return PyCursesCheckERR(halfdelay(tenths), "halfdelay");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002325}
2326
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +00002327#ifndef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00002328/* No has_key! */
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002329static PyObject * PyCurses_has_key(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002330{
Victor Stinner26486ea2010-05-15 22:23:53 +00002331 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002332
Victor Stinner26486ea2010-05-15 22:23:53 +00002333 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002334
Victor Stinner26486ea2010-05-15 22:23:53 +00002335 if (!PyArg_ParseTuple(args,"i",&ch)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002336
Victor Stinner26486ea2010-05-15 22:23:53 +00002337 if (has_key(ch) == FALSE) {
2338 Py_INCREF(Py_False);
2339 return Py_False;
2340 }
2341 Py_INCREF(Py_True);
2342 return Py_True;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002343}
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +00002344#endif /* STRICT_SYSV_CURSES */
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002345
2346static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002347PyCurses_Init_Color(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002348{
Victor Stinner26486ea2010-05-15 22:23:53 +00002349 short color, r, g, b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002350
Victor Stinner26486ea2010-05-15 22:23:53 +00002351 PyCursesInitialised;
2352 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002353
Victor Stinner26486ea2010-05-15 22:23:53 +00002354 switch(PyTuple_Size(args)) {
2355 case 4:
2356 if (!PyArg_ParseTuple(args, "hhhh;color,r,g,b", &color, &r, &g, &b)) return NULL;
2357 break;
2358 default:
2359 PyErr_SetString(PyExc_TypeError, "init_color requires 4 arguments");
2360 return NULL;
2361 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002362
Victor Stinner26486ea2010-05-15 22:23:53 +00002363 return PyCursesCheckERR(init_color(color, r, g, b), "init_color");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002364}
2365
2366static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002367PyCurses_Init_Pair(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002368{
Victor Stinner26486ea2010-05-15 22:23:53 +00002369 short pair, f, b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002370
Victor Stinner26486ea2010-05-15 22:23:53 +00002371 PyCursesInitialised;
2372 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002373
Victor Stinner26486ea2010-05-15 22:23:53 +00002374 if (PyTuple_Size(args) != 3) {
2375 PyErr_SetString(PyExc_TypeError, "init_pair requires 3 arguments");
2376 return NULL;
2377 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002378
Victor Stinner26486ea2010-05-15 22:23:53 +00002379 if (!PyArg_ParseTuple(args, "hhh;pair, f, b", &pair, &f, &b)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002380
Victor Stinner26486ea2010-05-15 22:23:53 +00002381 return PyCursesCheckERR(init_pair(pair, f, b), "init_pair");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002382}
2383
Andrew M. Kuchling32552682000-12-22 21:52:27 +00002384static PyObject *ModDict;
2385
Victor Stinner26486ea2010-05-15 22:23:53 +00002386static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002387PyCurses_InitScr(PyObject *self)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002388{
Victor Stinner26486ea2010-05-15 22:23:53 +00002389 WINDOW *win;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002390 PyCursesWindowObject *winobj;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002391
Victor Stinner26486ea2010-05-15 22:23:53 +00002392 if (initialised == TRUE) {
2393 wrefresh(stdscr);
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002394 return (PyObject *)PyCursesWindow_New(stdscr, NULL);
Victor Stinner26486ea2010-05-15 22:23:53 +00002395 }
Guido van Rossum56bf2351994-08-31 22:06:24 +00002396
Victor Stinner26486ea2010-05-15 22:23:53 +00002397 win = initscr();
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002398
Victor Stinner26486ea2010-05-15 22:23:53 +00002399 if (win == NULL) {
2400 PyErr_SetString(PyCursesError, catchall_NULL);
2401 return NULL;
2402 }
Guido van Rossum85738471995-02-17 13:50:17 +00002403
Victor Stinner26486ea2010-05-15 22:23:53 +00002404 initialised = initialised_setupterm = TRUE;
Guido van Rossum56bf2351994-08-31 22:06:24 +00002405
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002406/* This was moved from initcurses() because it core dumped on SGI,
2407 where they're not defined until you've called initscr() */
Victor Stinner26486ea2010-05-15 22:23:53 +00002408#define SetDictInt(string,ch) \
2409 do { \
2410 PyObject *o = PyLong_FromLong((long) (ch)); \
2411 if (o && PyDict_SetItemString(ModDict, string, o) == 0) { \
2412 Py_DECREF(o); \
2413 } \
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00002414 } while (0)
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002415
Victor Stinner26486ea2010-05-15 22:23:53 +00002416 /* Here are some graphic symbols you can use */
2417 SetDictInt("ACS_ULCORNER", (ACS_ULCORNER));
2418 SetDictInt("ACS_LLCORNER", (ACS_LLCORNER));
2419 SetDictInt("ACS_URCORNER", (ACS_URCORNER));
2420 SetDictInt("ACS_LRCORNER", (ACS_LRCORNER));
2421 SetDictInt("ACS_LTEE", (ACS_LTEE));
2422 SetDictInt("ACS_RTEE", (ACS_RTEE));
2423 SetDictInt("ACS_BTEE", (ACS_BTEE));
2424 SetDictInt("ACS_TTEE", (ACS_TTEE));
2425 SetDictInt("ACS_HLINE", (ACS_HLINE));
2426 SetDictInt("ACS_VLINE", (ACS_VLINE));
2427 SetDictInt("ACS_PLUS", (ACS_PLUS));
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00002428#if !defined(__hpux) || defined(HAVE_NCURSES_H)
Victor Stinner26486ea2010-05-15 22:23:53 +00002429 /* On HP/UX 11, these are of type cchar_t, which is not an
2430 integral type. If this is a problem on more platforms, a
2431 configure test should be added to determine whether ACS_S1
2432 is of integral type. */
2433 SetDictInt("ACS_S1", (ACS_S1));
2434 SetDictInt("ACS_S9", (ACS_S9));
2435 SetDictInt("ACS_DIAMOND", (ACS_DIAMOND));
2436 SetDictInt("ACS_CKBOARD", (ACS_CKBOARD));
2437 SetDictInt("ACS_DEGREE", (ACS_DEGREE));
2438 SetDictInt("ACS_PLMINUS", (ACS_PLMINUS));
2439 SetDictInt("ACS_BULLET", (ACS_BULLET));
2440 SetDictInt("ACS_LARROW", (ACS_LARROW));
2441 SetDictInt("ACS_RARROW", (ACS_RARROW));
2442 SetDictInt("ACS_DARROW", (ACS_DARROW));
2443 SetDictInt("ACS_UARROW", (ACS_UARROW));
2444 SetDictInt("ACS_BOARD", (ACS_BOARD));
2445 SetDictInt("ACS_LANTERN", (ACS_LANTERN));
2446 SetDictInt("ACS_BLOCK", (ACS_BLOCK));
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00002447#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00002448 SetDictInt("ACS_BSSB", (ACS_ULCORNER));
2449 SetDictInt("ACS_SSBB", (ACS_LLCORNER));
2450 SetDictInt("ACS_BBSS", (ACS_URCORNER));
2451 SetDictInt("ACS_SBBS", (ACS_LRCORNER));
2452 SetDictInt("ACS_SBSS", (ACS_RTEE));
2453 SetDictInt("ACS_SSSB", (ACS_LTEE));
2454 SetDictInt("ACS_SSBS", (ACS_BTEE));
2455 SetDictInt("ACS_BSSS", (ACS_TTEE));
2456 SetDictInt("ACS_BSBS", (ACS_HLINE));
2457 SetDictInt("ACS_SBSB", (ACS_VLINE));
2458 SetDictInt("ACS_SSSS", (ACS_PLUS));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002459
Victor Stinner26486ea2010-05-15 22:23:53 +00002460 /* The following are never available with strict SYSV curses */
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002461#ifdef ACS_S3
Victor Stinner26486ea2010-05-15 22:23:53 +00002462 SetDictInt("ACS_S3", (ACS_S3));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002463#endif
2464#ifdef ACS_S7
Victor Stinner26486ea2010-05-15 22:23:53 +00002465 SetDictInt("ACS_S7", (ACS_S7));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002466#endif
2467#ifdef ACS_LEQUAL
Victor Stinner26486ea2010-05-15 22:23:53 +00002468 SetDictInt("ACS_LEQUAL", (ACS_LEQUAL));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002469#endif
2470#ifdef ACS_GEQUAL
Victor Stinner26486ea2010-05-15 22:23:53 +00002471 SetDictInt("ACS_GEQUAL", (ACS_GEQUAL));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002472#endif
2473#ifdef ACS_PI
Victor Stinner26486ea2010-05-15 22:23:53 +00002474 SetDictInt("ACS_PI", (ACS_PI));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002475#endif
2476#ifdef ACS_NEQUAL
Victor Stinner26486ea2010-05-15 22:23:53 +00002477 SetDictInt("ACS_NEQUAL", (ACS_NEQUAL));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002478#endif
2479#ifdef ACS_STERLING
Victor Stinner26486ea2010-05-15 22:23:53 +00002480 SetDictInt("ACS_STERLING", (ACS_STERLING));
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002481#endif
2482
Victor Stinner26486ea2010-05-15 22:23:53 +00002483 SetDictInt("LINES", LINES);
2484 SetDictInt("COLS", COLS);
Guido van Rossum56bf2351994-08-31 22:06:24 +00002485
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002486 winobj = (PyCursesWindowObject *)PyCursesWindow_New(win, NULL);
2487 screen_encoding = winobj->encoding;
2488 return (PyObject *)winobj;
Guido van Rossumf6971e21994-08-30 12:25:20 +00002489}
2490
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002491static PyObject *
2492PyCurses_setupterm(PyObject* self, PyObject *args, PyObject* keywds)
2493{
Victor Stinner26486ea2010-05-15 22:23:53 +00002494 int fd = -1;
2495 int err;
2496 char* termstr = NULL;
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002497
Victor Stinner26486ea2010-05-15 22:23:53 +00002498 static char *kwlist[] = {"term", "fd", NULL};
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002499
Victor Stinner26486ea2010-05-15 22:23:53 +00002500 if (!PyArg_ParseTupleAndKeywords(
2501 args, keywds, "|zi:setupterm", kwlist, &termstr, &fd)) {
2502 return NULL;
2503 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002504
Victor Stinner26486ea2010-05-15 22:23:53 +00002505 if (fd == -1) {
2506 PyObject* sys_stdout;
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002507
Victor Stinner26486ea2010-05-15 22:23:53 +00002508 sys_stdout = PySys_GetObject("stdout");
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002509
Victor Stinner26486ea2010-05-15 22:23:53 +00002510 if (sys_stdout == NULL || sys_stdout == Py_None) {
2511 PyErr_SetString(
2512 PyCursesError,
2513 "lost sys.stdout");
2514 return NULL;
2515 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002516
Victor Stinner26486ea2010-05-15 22:23:53 +00002517 fd = PyObject_AsFileDescriptor(sys_stdout);
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002518
Victor Stinner26486ea2010-05-15 22:23:53 +00002519 if (fd == -1) {
2520 return NULL;
2521 }
2522 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002523
Matthias Klose635edd12010-07-30 21:40:57 +00002524 if (!initialised_setupterm && setupterm(termstr,fd,&err) == ERR) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002525 char* s = "setupterm: unknown error";
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002526
Victor Stinner26486ea2010-05-15 22:23:53 +00002527 if (err == 0) {
2528 s = "setupterm: could not find terminal";
2529 } else if (err == -1) {
2530 s = "setupterm: could not find terminfo database";
2531 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002532
Victor Stinner26486ea2010-05-15 22:23:53 +00002533 PyErr_SetString(PyCursesError,s);
2534 return NULL;
2535 }
2536
2537 initialised_setupterm = TRUE;
2538
2539 Py_INCREF(Py_None);
2540 return Py_None;
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002541}
Guido van Rossumf6971e21994-08-30 12:25:20 +00002542
2543static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002544PyCurses_IntrFlush(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002545{
Victor Stinner26486ea2010-05-15 22:23:53 +00002546 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002547
Victor Stinner26486ea2010-05-15 22:23:53 +00002548 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002549
Victor Stinner26486ea2010-05-15 22:23:53 +00002550 switch(PyTuple_Size(args)) {
2551 case 1:
2552 if (!PyArg_ParseTuple(args,"i;True(1), False(0)",&ch)) return NULL;
2553 break;
2554 default:
2555 PyErr_SetString(PyExc_TypeError, "intrflush requires 1 argument");
2556 return NULL;
2557 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002558
Victor Stinner26486ea2010-05-15 22:23:53 +00002559 return PyCursesCheckERR(intrflush(NULL,ch), "intrflush");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002560}
2561
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002562#ifdef HAVE_CURSES_IS_TERM_RESIZED
2563static PyObject *
2564PyCurses_Is_Term_Resized(PyObject *self, PyObject *args)
2565{
Victor Stinner26486ea2010-05-15 22:23:53 +00002566 int lines;
2567 int columns;
2568 int result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002569
Victor Stinner26486ea2010-05-15 22:23:53 +00002570 PyCursesInitialised;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002571
Victor Stinner26486ea2010-05-15 22:23:53 +00002572 if (!PyArg_ParseTuple(args,"ii:is_term_resized", &lines, &columns))
2573 return NULL;
2574 result = is_term_resized(lines, columns);
2575 if (result == TRUE) {
2576 Py_INCREF(Py_True);
2577 return Py_True;
2578 } else {
2579 Py_INCREF(Py_False);
2580 return Py_False;
2581 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002582}
2583#endif /* HAVE_CURSES_IS_TERM_RESIZED */
2584
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002585#if !defined(__NetBSD__)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002586static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002587PyCurses_KeyName(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002588{
Victor Stinner26486ea2010-05-15 22:23:53 +00002589 const char *knp;
2590 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002591
Victor Stinner26486ea2010-05-15 22:23:53 +00002592 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002593
Victor Stinner26486ea2010-05-15 22:23:53 +00002594 if (!PyArg_ParseTuple(args,"i",&ch)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002595
Victor Stinner26486ea2010-05-15 22:23:53 +00002596 if (ch < 0) {
2597 PyErr_SetString(PyExc_ValueError, "invalid key number");
2598 return NULL;
2599 }
2600 knp = keyname(ch);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002601
Victor Stinner26486ea2010-05-15 22:23:53 +00002602 return PyBytes_FromString((knp == NULL) ? "" : (char *)knp);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002603}
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002604#endif
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002605
Victor Stinner26486ea2010-05-15 22:23:53 +00002606static PyObject *
2607PyCurses_KillChar(PyObject *self)
2608{
2609 char ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002610
Victor Stinner26486ea2010-05-15 22:23:53 +00002611 ch = killchar();
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002612
Victor Stinner26486ea2010-05-15 22:23:53 +00002613 return PyBytes_FromStringAndSize(&ch, 1);
2614}
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002615
2616static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002617PyCurses_Meta(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002618{
Victor Stinner26486ea2010-05-15 22:23:53 +00002619 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002620
Victor Stinner26486ea2010-05-15 22:23:53 +00002621 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002622
Victor Stinner26486ea2010-05-15 22:23:53 +00002623 switch(PyTuple_Size(args)) {
2624 case 1:
2625 if (!PyArg_ParseTuple(args,"i;True(1), False(0)",&ch)) return NULL;
2626 break;
2627 default:
2628 PyErr_SetString(PyExc_TypeError, "meta requires 1 argument");
2629 return NULL;
2630 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002631
Victor Stinner26486ea2010-05-15 22:23:53 +00002632 return PyCursesCheckERR(meta(stdscr, ch), "meta");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002633}
2634
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002635#ifdef NCURSES_MOUSE_VERSION
2636static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002637PyCurses_MouseInterval(PyObject *self, PyObject *args)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002638{
Victor Stinner26486ea2010-05-15 22:23:53 +00002639 int interval;
2640 PyCursesInitialised;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002641
Victor Stinner26486ea2010-05-15 22:23:53 +00002642 if (!PyArg_ParseTuple(args,"i;interval",&interval))
2643 return NULL;
2644 return PyCursesCheckERR(mouseinterval(interval), "mouseinterval");
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002645}
2646
2647static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002648PyCurses_MouseMask(PyObject *self, PyObject *args)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002649{
Victor Stinner26486ea2010-05-15 22:23:53 +00002650 int newmask;
2651 mmask_t oldmask, availmask;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002652
Victor Stinner26486ea2010-05-15 22:23:53 +00002653 PyCursesInitialised;
2654 if (!PyArg_ParseTuple(args,"i;mousemask",&newmask))
2655 return NULL;
2656 availmask = mousemask(newmask, &oldmask);
2657 return Py_BuildValue("(ll)", (long)availmask, (long)oldmask);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002658}
2659#endif
2660
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002661static PyObject *
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +00002662PyCurses_Napms(PyObject *self, PyObject *args)
2663{
2664 int ms;
2665
Andrew M. Kuchling2b146762010-05-10 23:30:25 +00002666 PyCursesInitialised;
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002667 if (!PyArg_ParseTuple(args, "i;ms", &ms)) return NULL;
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +00002668
2669 return Py_BuildValue("i", napms(ms));
2670}
2671
2672
2673static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002674PyCurses_NewPad(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002675{
Victor Stinner26486ea2010-05-15 22:23:53 +00002676 WINDOW *win;
2677 int nlines, ncols;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002678
Victor Stinner26486ea2010-05-15 22:23:53 +00002679 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002680
Victor Stinner26486ea2010-05-15 22:23:53 +00002681 if (!PyArg_ParseTuple(args,"ii;nlines,ncols",&nlines,&ncols)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002682
Victor Stinner26486ea2010-05-15 22:23:53 +00002683 win = newpad(nlines, ncols);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002684
Victor Stinner26486ea2010-05-15 22:23:53 +00002685 if (win == NULL) {
2686 PyErr_SetString(PyCursesError, catchall_NULL);
2687 return NULL;
2688 }
2689
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002690 return (PyObject *)PyCursesWindow_New(win, NULL);
Guido van Rossumf6971e21994-08-30 12:25:20 +00002691}
2692
2693static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002694PyCurses_NewWindow(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002695{
Victor Stinner26486ea2010-05-15 22:23:53 +00002696 WINDOW *win;
2697 int nlines, ncols, begin_y=0, begin_x=0;
Guido van Rossum85738471995-02-17 13:50:17 +00002698
Victor Stinner26486ea2010-05-15 22:23:53 +00002699 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002700
Victor Stinner26486ea2010-05-15 22:23:53 +00002701 switch (PyTuple_Size(args)) {
2702 case 2:
2703 if (!PyArg_ParseTuple(args,"ii;nlines,ncols",&nlines,&ncols))
2704 return NULL;
2705 break;
2706 case 4:
2707 if (!PyArg_ParseTuple(args, "iiii;nlines,ncols,begin_y,begin_x",
2708 &nlines,&ncols,&begin_y,&begin_x))
2709 return NULL;
2710 break;
2711 default:
2712 PyErr_SetString(PyExc_TypeError, "newwin requires 2 or 4 arguments");
2713 return NULL;
2714 }
Guido van Rossum85738471995-02-17 13:50:17 +00002715
Victor Stinner26486ea2010-05-15 22:23:53 +00002716 win = newwin(nlines,ncols,begin_y,begin_x);
2717 if (win == NULL) {
2718 PyErr_SetString(PyCursesError, catchall_NULL);
2719 return NULL;
2720 }
Guido van Rossum85738471995-02-17 13:50:17 +00002721
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002722 return (PyObject *)PyCursesWindow_New(win, NULL);
Guido van Rossumf6971e21994-08-30 12:25:20 +00002723}
2724
2725static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002726PyCurses_Pair_Content(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002727{
Victor Stinner26486ea2010-05-15 22:23:53 +00002728 short pair,f,b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002729
Victor Stinner26486ea2010-05-15 22:23:53 +00002730 PyCursesInitialised;
2731 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002732
Victor Stinner26486ea2010-05-15 22:23:53 +00002733 switch(PyTuple_Size(args)) {
2734 case 1:
2735 if (!PyArg_ParseTuple(args, "h;pair", &pair)) return NULL;
2736 break;
2737 default:
2738 PyErr_SetString(PyExc_TypeError, "pair_content requires 1 argument");
2739 return NULL;
2740 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002741
Victor Stinner26486ea2010-05-15 22:23:53 +00002742 if (pair_content(pair, &f, &b)==ERR) {
2743 PyErr_SetString(PyCursesError,
2744 "Argument 1 was out of range. (1..COLOR_PAIRS-1)");
2745 return NULL;
2746 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002747
Victor Stinner26486ea2010-05-15 22:23:53 +00002748 return Py_BuildValue("(ii)", f, b);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002749}
2750
2751static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002752PyCurses_pair_number(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002753{
Victor Stinner26486ea2010-05-15 22:23:53 +00002754 int n;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002755
Victor Stinner26486ea2010-05-15 22:23:53 +00002756 PyCursesInitialised;
2757 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002758
Victor Stinner26486ea2010-05-15 22:23:53 +00002759 switch(PyTuple_Size(args)) {
2760 case 1:
2761 if (!PyArg_ParseTuple(args, "i;pairvalue", &n)) return NULL;
2762 break;
2763 default:
2764 PyErr_SetString(PyExc_TypeError,
2765 "pair_number requires 1 argument");
2766 return NULL;
2767 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002768
Victor Stinner26486ea2010-05-15 22:23:53 +00002769 return PyLong_FromLong((long) ((n & A_COLOR) >> 8));
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002770}
2771
2772static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002773PyCurses_Putp(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002774{
Victor Stinner26486ea2010-05-15 22:23:53 +00002775 char *str;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002776
Victor Stinnerb3bc7e72011-11-03 20:35:40 +01002777 if (!PyArg_ParseTuple(args,"y;str", &str))
2778 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00002779 return PyCursesCheckERR(putp(str), "putp");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002780}
2781
2782static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002783PyCurses_QiFlush(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002784{
Victor Stinner26486ea2010-05-15 22:23:53 +00002785 int flag = 0;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002786
Victor Stinner26486ea2010-05-15 22:23:53 +00002787 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002788
Victor Stinner26486ea2010-05-15 22:23:53 +00002789 switch(PyTuple_Size(args)) {
2790 case 0:
2791 qiflush();
2792 Py_INCREF(Py_None);
2793 return Py_None;
2794 case 1:
2795 if (!PyArg_ParseTuple(args, "i;True(1) or False(0)", &flag)) return NULL;
2796 if (flag) qiflush();
2797 else noqiflush();
2798 Py_INCREF(Py_None);
2799 return Py_None;
2800 default:
2801 PyErr_SetString(PyExc_TypeError, "qiflush requires 0 or 1 arguments");
2802 return NULL;
2803 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002804}
2805
Guido van Rossumd8faa362007-04-27 19:54:29 +00002806/* Internal helper used for updating curses.LINES, curses.COLS, _curses.LINES
2807 * and _curses.COLS */
Benjamin Petersonfea6a942008-07-02 16:11:42 +00002808#if defined(HAVE_CURSES_RESIZETERM) || defined(HAVE_CURSES_RESIZE_TERM)
Guido van Rossumd8faa362007-04-27 19:54:29 +00002809static int
2810update_lines_cols(void)
2811{
Victor Stinner26486ea2010-05-15 22:23:53 +00002812 PyObject *o;
2813 PyObject *m = PyImport_ImportModuleNoBlock("curses");
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002814 _Py_IDENTIFIER(LINES);
2815 _Py_IDENTIFIER(COLS);
Guido van Rossumd8faa362007-04-27 19:54:29 +00002816
Victor Stinner26486ea2010-05-15 22:23:53 +00002817 if (!m)
2818 return 0;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002819
Victor Stinner26486ea2010-05-15 22:23:53 +00002820 o = PyLong_FromLong(LINES);
2821 if (!o) {
2822 Py_DECREF(m);
2823 return 0;
2824 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002825 if (_PyObject_SetAttrId(m, &PyId_LINES, o)) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002826 Py_DECREF(m);
2827 Py_DECREF(o);
2828 return 0;
2829 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002830 /* PyId_LINES.object will be initialized here. */
2831 if (PyDict_SetItem(ModDict, PyId_LINES.object, o)) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002832 Py_DECREF(m);
2833 Py_DECREF(o);
2834 return 0;
2835 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002836 Py_DECREF(o);
Victor Stinner26486ea2010-05-15 22:23:53 +00002837 o = PyLong_FromLong(COLS);
2838 if (!o) {
2839 Py_DECREF(m);
2840 return 0;
2841 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002842 if (_PyObject_SetAttrId(m, &PyId_COLS, o)) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002843 Py_DECREF(m);
2844 Py_DECREF(o);
2845 return 0;
2846 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002847 if (PyDict_SetItem(ModDict, PyId_COLS.object, o)) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002848 Py_DECREF(m);
2849 Py_DECREF(o);
2850 return 0;
2851 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002852 Py_DECREF(o);
Guido van Rossumd8faa362007-04-27 19:54:29 +00002853 Py_DECREF(m);
Victor Stinner26486ea2010-05-15 22:23:53 +00002854 return 1;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002855}
Benjamin Petersonfea6a942008-07-02 16:11:42 +00002856#endif
Guido van Rossumd8faa362007-04-27 19:54:29 +00002857
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002858#ifdef HAVE_CURSES_RESIZETERM
2859static PyObject *
2860PyCurses_ResizeTerm(PyObject *self, PyObject *args)
2861{
Victor Stinner26486ea2010-05-15 22:23:53 +00002862 int lines;
2863 int columns;
2864 PyObject *result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002865
Victor Stinner26486ea2010-05-15 22:23:53 +00002866 PyCursesInitialised;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002867
Victor Stinner26486ea2010-05-15 22:23:53 +00002868 if (!PyArg_ParseTuple(args,"ii:resizeterm", &lines, &columns))
2869 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002870
Victor Stinner26486ea2010-05-15 22:23:53 +00002871 result = PyCursesCheckERR(resizeterm(lines, columns), "resizeterm");
2872 if (!result)
2873 return NULL;
2874 if (!update_lines_cols())
2875 return NULL;
2876 return result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002877}
2878
2879#endif
2880
2881#ifdef HAVE_CURSES_RESIZE_TERM
2882static PyObject *
2883PyCurses_Resize_Term(PyObject *self, PyObject *args)
2884{
Victor Stinner26486ea2010-05-15 22:23:53 +00002885 int lines;
2886 int columns;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002887
Victor Stinner26486ea2010-05-15 22:23:53 +00002888 PyObject *result;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002889
Victor Stinner26486ea2010-05-15 22:23:53 +00002890 PyCursesInitialised;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002891
Victor Stinner26486ea2010-05-15 22:23:53 +00002892 if (!PyArg_ParseTuple(args,"ii:resize_term", &lines, &columns))
2893 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002894
Victor Stinner26486ea2010-05-15 22:23:53 +00002895 result = PyCursesCheckERR(resize_term(lines, columns), "resize_term");
2896 if (!result)
2897 return NULL;
2898 if (!update_lines_cols())
2899 return NULL;
2900 return result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002901}
2902#endif /* HAVE_CURSES_RESIZE_TERM */
2903
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002904static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002905PyCurses_setsyx(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002906{
Victor Stinner26486ea2010-05-15 22:23:53 +00002907 int y,x;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002908
Victor Stinner26486ea2010-05-15 22:23:53 +00002909 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002910
Victor Stinner26486ea2010-05-15 22:23:53 +00002911 if (PyTuple_Size(args)!=2) {
2912 PyErr_SetString(PyExc_TypeError, "setsyx requires 2 arguments");
2913 return NULL;
2914 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002915
Victor Stinner26486ea2010-05-15 22:23:53 +00002916 if (!PyArg_ParseTuple(args, "ii;y, x", &y, &x)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002917
Victor Stinner26486ea2010-05-15 22:23:53 +00002918 setsyx(y,x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002919
Victor Stinner26486ea2010-05-15 22:23:53 +00002920 Py_INCREF(Py_None);
2921 return Py_None;
Guido van Rossumf6971e21994-08-30 12:25:20 +00002922}
2923
2924static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002925PyCurses_Start_Color(PyObject *self)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002926{
Victor Stinner26486ea2010-05-15 22:23:53 +00002927 int code;
2928 PyObject *c, *cp;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002929
Victor Stinner26486ea2010-05-15 22:23:53 +00002930 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002931
Victor Stinner26486ea2010-05-15 22:23:53 +00002932 code = start_color();
2933 if (code != ERR) {
2934 initialisedcolors = TRUE;
2935 c = PyLong_FromLong((long) COLORS);
Christian Heimesa956e642013-07-26 14:45:37 +02002936 if (c == NULL)
2937 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00002938 PyDict_SetItemString(ModDict, "COLORS", c);
2939 Py_DECREF(c);
2940 cp = PyLong_FromLong((long) COLOR_PAIRS);
Christian Heimesa956e642013-07-26 14:45:37 +02002941 if (cp == NULL)
2942 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00002943 PyDict_SetItemString(ModDict, "COLOR_PAIRS", cp);
2944 Py_DECREF(cp);
2945 Py_INCREF(Py_None);
2946 return Py_None;
2947 } else {
2948 PyErr_SetString(PyCursesError, "start_color() returned ERR");
2949 return NULL;
2950 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002951}
2952
2953static PyObject *
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002954PyCurses_tigetflag(PyObject *self, PyObject *args)
2955{
Victor Stinner26486ea2010-05-15 22:23:53 +00002956 char *capname;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002957
Victor Stinner26486ea2010-05-15 22:23:53 +00002958 PyCursesSetupTermCalled;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002959
Victor Stinner26486ea2010-05-15 22:23:53 +00002960 if (!PyArg_ParseTuple(args, "s", &capname))
2961 return NULL;
2962
2963 return PyLong_FromLong( (long) tigetflag( capname ) );
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002964}
2965
2966static PyObject *
2967PyCurses_tigetnum(PyObject *self, PyObject *args)
2968{
Victor Stinner26486ea2010-05-15 22:23:53 +00002969 char *capname;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002970
Victor Stinner26486ea2010-05-15 22:23:53 +00002971 PyCursesSetupTermCalled;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002972
Victor Stinner26486ea2010-05-15 22:23:53 +00002973 if (!PyArg_ParseTuple(args, "s", &capname))
2974 return NULL;
2975
2976 return PyLong_FromLong( (long) tigetnum( capname ) );
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002977}
2978
2979static PyObject *
2980PyCurses_tigetstr(PyObject *self, PyObject *args)
2981{
Victor Stinner26486ea2010-05-15 22:23:53 +00002982 char *capname;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002983
Victor Stinner26486ea2010-05-15 22:23:53 +00002984 PyCursesSetupTermCalled;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002985
Victor Stinner26486ea2010-05-15 22:23:53 +00002986 if (!PyArg_ParseTuple(args, "s", &capname))
2987 return NULL;
2988
2989 capname = tigetstr( capname );
2990 if (capname == 0 || capname == (char*) -1) {
2991 Py_INCREF(Py_None);
2992 return Py_None;
2993 }
2994 return PyBytes_FromString( capname );
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002995}
2996
2997static PyObject *
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00002998PyCurses_tparm(PyObject *self, PyObject *args)
2999{
Victor Stinner26486ea2010-05-15 22:23:53 +00003000 char* fmt;
3001 char* result = NULL;
3002 int i1=0,i2=0,i3=0,i4=0,i5=0,i6=0,i7=0,i8=0,i9=0;
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00003003
Victor Stinner26486ea2010-05-15 22:23:53 +00003004 PyCursesSetupTermCalled;
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00003005
Victor Stinner26621332011-11-02 23:45:29 +01003006 if (!PyArg_ParseTuple(args, "y|iiiiiiiii:tparm",
Victor Stinner26486ea2010-05-15 22:23:53 +00003007 &fmt, &i1, &i2, &i3, &i4,
3008 &i5, &i6, &i7, &i8, &i9)) {
3009 return NULL;
3010 }
Michael W. Hudson21007982002-11-21 14:17:51 +00003011
Victor Stinner26486ea2010-05-15 22:23:53 +00003012 result = tparm(fmt,i1,i2,i3,i4,i5,i6,i7,i8,i9);
3013 if (!result) {
3014 PyErr_SetString(PyCursesError, "tparm() returned NULL");
3015 return NULL;
3016 }
Michael W. Hudson21007982002-11-21 14:17:51 +00003017
Victor Stinner26486ea2010-05-15 22:23:53 +00003018 return PyBytes_FromString(result);
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00003019}
3020
3021static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00003022PyCurses_TypeAhead(PyObject *self, PyObject *args)
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003023{
Victor Stinner26486ea2010-05-15 22:23:53 +00003024 int fd;
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003025
Victor Stinner26486ea2010-05-15 22:23:53 +00003026 PyCursesInitialised;
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003027
Victor Stinner26486ea2010-05-15 22:23:53 +00003028 if (!PyArg_ParseTuple(args,"i;fd",&fd)) return NULL;
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003029
Victor Stinner26486ea2010-05-15 22:23:53 +00003030 return PyCursesCheckERR(typeahead( fd ), "typeahead");
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003031}
3032
3033static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00003034PyCurses_UnCtrl(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003035{
Victor Stinner26486ea2010-05-15 22:23:53 +00003036 PyObject *temp;
3037 chtype ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003038
Victor Stinner26486ea2010-05-15 22:23:53 +00003039 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003040
Victor Stinner26486ea2010-05-15 22:23:53 +00003041 if (!PyArg_ParseTuple(args,"O;ch or int",&temp)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003042
Victor Stinner0fdfceb2011-11-25 22:10:02 +01003043 if (!PyCurses_ConvertToChtype(NULL, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00003044 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003045
Victor Stinner26486ea2010-05-15 22:23:53 +00003046 return PyBytes_FromString(unctrl(ch));
Guido van Rossumf6971e21994-08-30 12:25:20 +00003047}
3048
3049static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00003050PyCurses_UngetCh(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00003051{
Victor Stinner26486ea2010-05-15 22:23:53 +00003052 PyObject *temp;
3053 chtype ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003054
Victor Stinner26486ea2010-05-15 22:23:53 +00003055 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003056
Victor Stinner0fdfceb2011-11-25 22:10:02 +01003057 if (!PyArg_ParseTuple(args,"O;ch or int",&temp))
Victor Stinner26486ea2010-05-15 22:23:53 +00003058 return NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01003059
3060 if (!PyCurses_ConvertToChtype(NULL, temp, &ch))
3061 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003062
Victor Stinner26486ea2010-05-15 22:23:53 +00003063 return PyCursesCheckERR(ungetch(ch), "ungetch");
Guido van Rossumf6971e21994-08-30 12:25:20 +00003064}
3065
Victor Stinner71e44cb2011-09-06 01:53:03 +02003066#ifdef HAVE_NCURSESW
3067/* Convert an object to a character (wchar_t):
3068
3069 - int
3070 - str of length 1
3071
3072 Return 1 on success, 0 on error. */
3073static int
3074PyCurses_ConvertToWchar_t(PyObject *obj,
3075 wchar_t *wch)
3076{
3077 if (PyUnicode_Check(obj)) {
3078 wchar_t buffer[2];
3079 if (PyUnicode_AsWideChar(obj, buffer, 2) != 1) {
3080 PyErr_Format(PyExc_TypeError,
3081 "expect bytes or str of length 1, or int, "
3082 "got a str of length %zi",
Victor Stinnerc4f281e2011-10-11 22:11:42 +02003083 PyUnicode_GET_LENGTH(obj));
Victor Stinner71e44cb2011-09-06 01:53:03 +02003084 return 0;
3085 }
3086 *wch = buffer[0];
3087 return 2;
3088 }
3089 else if (PyLong_CheckExact(obj)) {
3090 long value;
3091 int overflow;
3092 value = PyLong_AsLongAndOverflow(obj, &overflow);
3093 if (overflow) {
3094 PyErr_SetString(PyExc_OverflowError,
3095 "int doesn't fit in long");
3096 return 0;
3097 }
3098 *wch = (wchar_t)value;
3099 if ((long)*wch != value) {
3100 PyErr_Format(PyExc_OverflowError,
3101 "character doesn't fit in wchar_t");
3102 return 0;
3103 }
3104 return 1;
3105 }
3106 else {
3107 PyErr_Format(PyExc_TypeError,
3108 "expect bytes or str of length 1, or int, got %s",
3109 Py_TYPE(obj)->tp_name);
3110 return 0;
3111 }
3112}
3113
3114static PyObject *
3115PyCurses_Unget_Wch(PyObject *self, PyObject *args)
3116{
3117 PyObject *obj;
3118 wchar_t wch;
3119
3120 PyCursesInitialised;
3121
3122 if (!PyArg_ParseTuple(args,"O", &obj))
3123 return NULL;
3124
3125 if (!PyCurses_ConvertToWchar_t(obj, &wch))
3126 return NULL;
3127 return PyCursesCheckERR(unget_wch(wch), "unget_wch");
3128}
3129#endif
3130
Guido van Rossumf6971e21994-08-30 12:25:20 +00003131static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00003132PyCurses_Use_Env(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00003133{
Victor Stinner26486ea2010-05-15 22:23:53 +00003134 int flag;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003135
Victor Stinner26486ea2010-05-15 22:23:53 +00003136 switch(PyTuple_Size(args)) {
3137 case 1:
3138 if (!PyArg_ParseTuple(args,"i;True(1), False(0)",&flag))
3139 return NULL;
3140 break;
3141 default:
3142 PyErr_SetString(PyExc_TypeError, "use_env requires 1 argument");
3143 return NULL;
3144 }
3145 use_env(flag);
3146 Py_INCREF(Py_None);
3147 return Py_None;
Guido van Rossumf6971e21994-08-30 12:25:20 +00003148}
3149
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003150#ifndef STRICT_SYSV_CURSES
3151static PyObject *
3152PyCurses_Use_Default_Colors(PyObject *self)
3153{
Victor Stinner26486ea2010-05-15 22:23:53 +00003154 int code;
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003155
Victor Stinner26486ea2010-05-15 22:23:53 +00003156 PyCursesInitialised;
3157 PyCursesInitialisedColor;
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003158
Victor Stinner26486ea2010-05-15 22:23:53 +00003159 code = use_default_colors();
3160 if (code != ERR) {
3161 Py_INCREF(Py_None);
3162 return Py_None;
3163 } else {
3164 PyErr_SetString(PyCursesError, "use_default_colors() returned ERR");
3165 return NULL;
3166 }
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003167}
3168#endif /* STRICT_SYSV_CURSES */
3169
Guido van Rossumf6971e21994-08-30 12:25:20 +00003170/* List of functions defined in the module */
3171
3172static PyMethodDef PyCurses_methods[] = {
Victor Stinner26486ea2010-05-15 22:23:53 +00003173 {"baudrate", (PyCFunction)PyCurses_baudrate, METH_NOARGS},
3174 {"beep", (PyCFunction)PyCurses_beep, METH_NOARGS},
3175 {"can_change_color", (PyCFunction)PyCurses_can_change_color, METH_NOARGS},
3176 {"cbreak", (PyCFunction)PyCurses_cbreak, METH_VARARGS},
3177 {"color_content", (PyCFunction)PyCurses_Color_Content, METH_VARARGS},
3178 {"color_pair", (PyCFunction)PyCurses_color_pair, METH_VARARGS},
3179 {"curs_set", (PyCFunction)PyCurses_Curs_Set, METH_VARARGS},
3180 {"def_prog_mode", (PyCFunction)PyCurses_def_prog_mode, METH_NOARGS},
3181 {"def_shell_mode", (PyCFunction)PyCurses_def_shell_mode, METH_NOARGS},
3182 {"delay_output", (PyCFunction)PyCurses_Delay_Output, METH_VARARGS},
3183 {"doupdate", (PyCFunction)PyCurses_doupdate, METH_NOARGS},
3184 {"echo", (PyCFunction)PyCurses_echo, METH_VARARGS},
3185 {"endwin", (PyCFunction)PyCurses_endwin, METH_NOARGS},
3186 {"erasechar", (PyCFunction)PyCurses_EraseChar, METH_NOARGS},
3187 {"filter", (PyCFunction)PyCurses_filter, METH_NOARGS},
3188 {"flash", (PyCFunction)PyCurses_flash, METH_NOARGS},
3189 {"flushinp", (PyCFunction)PyCurses_flushinp, METH_NOARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003190#ifdef NCURSES_MOUSE_VERSION
Victor Stinner26486ea2010-05-15 22:23:53 +00003191 {"getmouse", (PyCFunction)PyCurses_GetMouse, METH_NOARGS},
3192 {"ungetmouse", (PyCFunction)PyCurses_UngetMouse, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003193#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003194 {"getsyx", (PyCFunction)PyCurses_getsyx, METH_NOARGS},
3195 {"getwin", (PyCFunction)PyCurses_GetWin, METH_O},
3196 {"has_colors", (PyCFunction)PyCurses_has_colors, METH_NOARGS},
3197 {"has_ic", (PyCFunction)PyCurses_has_ic, METH_NOARGS},
3198 {"has_il", (PyCFunction)PyCurses_has_il, METH_NOARGS},
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +00003199#ifndef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00003200 {"has_key", (PyCFunction)PyCurses_has_key, METH_VARARGS},
Guido van Rossumf6971e21994-08-30 12:25:20 +00003201#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003202 {"halfdelay", (PyCFunction)PyCurses_HalfDelay, METH_VARARGS},
3203 {"init_color", (PyCFunction)PyCurses_Init_Color, METH_VARARGS},
3204 {"init_pair", (PyCFunction)PyCurses_Init_Pair, METH_VARARGS},
3205 {"initscr", (PyCFunction)PyCurses_InitScr, METH_NOARGS},
3206 {"intrflush", (PyCFunction)PyCurses_IntrFlush, METH_VARARGS},
3207 {"isendwin", (PyCFunction)PyCurses_isendwin, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003208#ifdef HAVE_CURSES_IS_TERM_RESIZED
Victor Stinner26486ea2010-05-15 22:23:53 +00003209 {"is_term_resized", (PyCFunction)PyCurses_Is_Term_Resized, METH_VARARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003210#endif
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003211#if !defined(__NetBSD__)
Victor Stinner26486ea2010-05-15 22:23:53 +00003212 {"keyname", (PyCFunction)PyCurses_KeyName, METH_VARARGS},
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003213#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003214 {"killchar", (PyCFunction)PyCurses_KillChar, METH_NOARGS},
3215 {"longname", (PyCFunction)PyCurses_longname, METH_NOARGS},
3216 {"meta", (PyCFunction)PyCurses_Meta, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003217#ifdef NCURSES_MOUSE_VERSION
Victor Stinner26486ea2010-05-15 22:23:53 +00003218 {"mouseinterval", (PyCFunction)PyCurses_MouseInterval, METH_VARARGS},
3219 {"mousemask", (PyCFunction)PyCurses_MouseMask, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003220#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003221 {"napms", (PyCFunction)PyCurses_Napms, METH_VARARGS},
3222 {"newpad", (PyCFunction)PyCurses_NewPad, METH_VARARGS},
3223 {"newwin", (PyCFunction)PyCurses_NewWindow, METH_VARARGS},
3224 {"nl", (PyCFunction)PyCurses_nl, METH_VARARGS},
3225 {"nocbreak", (PyCFunction)PyCurses_nocbreak, METH_NOARGS},
3226 {"noecho", (PyCFunction)PyCurses_noecho, METH_NOARGS},
3227 {"nonl", (PyCFunction)PyCurses_nonl, METH_NOARGS},
3228 {"noqiflush", (PyCFunction)PyCurses_noqiflush, METH_NOARGS},
3229 {"noraw", (PyCFunction)PyCurses_noraw, METH_NOARGS},
3230 {"pair_content", (PyCFunction)PyCurses_Pair_Content, METH_VARARGS},
3231 {"pair_number", (PyCFunction)PyCurses_pair_number, METH_VARARGS},
3232 {"putp", (PyCFunction)PyCurses_Putp, METH_VARARGS},
3233 {"qiflush", (PyCFunction)PyCurses_QiFlush, METH_VARARGS},
3234 {"raw", (PyCFunction)PyCurses_raw, METH_VARARGS},
3235 {"reset_prog_mode", (PyCFunction)PyCurses_reset_prog_mode, METH_NOARGS},
3236 {"reset_shell_mode", (PyCFunction)PyCurses_reset_shell_mode, METH_NOARGS},
3237 {"resetty", (PyCFunction)PyCurses_resetty, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003238#ifdef HAVE_CURSES_RESIZETERM
Victor Stinner26486ea2010-05-15 22:23:53 +00003239 {"resizeterm", (PyCFunction)PyCurses_ResizeTerm, METH_VARARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003240#endif
3241#ifdef HAVE_CURSES_RESIZE_TERM
Victor Stinner26486ea2010-05-15 22:23:53 +00003242 {"resize_term", (PyCFunction)PyCurses_Resize_Term, METH_VARARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003243#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003244 {"savetty", (PyCFunction)PyCurses_savetty, METH_NOARGS},
3245 {"setsyx", (PyCFunction)PyCurses_setsyx, METH_VARARGS},
3246 {"setupterm", (PyCFunction)PyCurses_setupterm,
3247 METH_VARARGS|METH_KEYWORDS},
3248 {"start_color", (PyCFunction)PyCurses_Start_Color, METH_NOARGS},
3249 {"termattrs", (PyCFunction)PyCurses_termattrs, METH_NOARGS},
3250 {"termname", (PyCFunction)PyCurses_termname, METH_NOARGS},
3251 {"tigetflag", (PyCFunction)PyCurses_tigetflag, METH_VARARGS},
3252 {"tigetnum", (PyCFunction)PyCurses_tigetnum, METH_VARARGS},
3253 {"tigetstr", (PyCFunction)PyCurses_tigetstr, METH_VARARGS},
3254 {"tparm", (PyCFunction)PyCurses_tparm, METH_VARARGS},
3255 {"typeahead", (PyCFunction)PyCurses_TypeAhead, METH_VARARGS},
3256 {"unctrl", (PyCFunction)PyCurses_UnCtrl, METH_VARARGS},
3257 {"ungetch", (PyCFunction)PyCurses_UngetCh, METH_VARARGS},
Victor Stinner71e44cb2011-09-06 01:53:03 +02003258#ifdef HAVE_NCURSESW
3259 {"unget_wch", (PyCFunction)PyCurses_Unget_Wch, METH_VARARGS},
3260#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003261 {"use_env", (PyCFunction)PyCurses_Use_Env, METH_VARARGS},
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003262#ifndef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00003263 {"use_default_colors", (PyCFunction)PyCurses_Use_Default_Colors, METH_NOARGS},
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003264#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003265 {NULL, NULL} /* sentinel */
Guido van Rossumf6971e21994-08-30 12:25:20 +00003266};
3267
3268/* Initialization function for the module */
3269
Martin v. Löwis1a214512008-06-11 05:26:20 +00003270
3271static struct PyModuleDef _cursesmodule = {
Victor Stinner26486ea2010-05-15 22:23:53 +00003272 PyModuleDef_HEAD_INIT,
3273 "_curses",
3274 NULL,
3275 -1,
3276 PyCurses_methods,
3277 NULL,
3278 NULL,
3279 NULL,
3280 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00003281};
3282
Mark Hammondfe51c6d2002-08-02 02:27:13 +00003283PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00003284PyInit__curses(void)
Guido van Rossumf6971e21994-08-30 12:25:20 +00003285{
Victor Stinner26486ea2010-05-15 22:23:53 +00003286 PyObject *m, *d, *v, *c_api_object;
3287 static void *PyCurses_API[PyCurses_API_pointers];
Andrew M. Kuchling32552682000-12-22 21:52:27 +00003288
Victor Stinner26486ea2010-05-15 22:23:53 +00003289 /* Initialize object type */
3290 if (PyType_Ready(&PyCursesWindow_Type) < 0)
3291 return NULL;
Guido van Rossum6915c4d2001-01-19 00:28:08 +00003292
Victor Stinner26486ea2010-05-15 22:23:53 +00003293 /* Initialize the C API pointer array */
3294 PyCurses_API[0] = (void *)&PyCursesWindow_Type;
3295 PyCurses_API[1] = (void *)func_PyCursesSetupTermCalled;
3296 PyCurses_API[2] = (void *)func_PyCursesInitialised;
3297 PyCurses_API[3] = (void *)func_PyCursesInitialisedColor;
Guido van Rossumf6971e21994-08-30 12:25:20 +00003298
Victor Stinner26486ea2010-05-15 22:23:53 +00003299 /* Create the module and add the functions */
3300 m = PyModule_Create(&_cursesmodule);
3301 if (m == NULL)
3302 return NULL;
Guido van Rossumf6971e21994-08-30 12:25:20 +00003303
Victor Stinner26486ea2010-05-15 22:23:53 +00003304 /* Add some symbolic constants to the module */
3305 d = PyModule_GetDict(m);
3306 if (d == NULL)
3307 return NULL;
3308 ModDict = d; /* For PyCurses_InitScr to use later */
Andrew M. Kuchling32552682000-12-22 21:52:27 +00003309
Victor Stinner26486ea2010-05-15 22:23:53 +00003310 /* Add a capsule for the C API */
3311 c_api_object = PyCapsule_New(PyCurses_API, PyCurses_CAPSULE_NAME, NULL);
3312 PyDict_SetItemString(d, "_C_API", c_api_object);
3313 Py_DECREF(c_api_object);
Guido van Rossumfbea2f31994-08-31 22:05:27 +00003314
Victor Stinner26486ea2010-05-15 22:23:53 +00003315 /* For exception curses.error */
3316 PyCursesError = PyErr_NewException("_curses.error", NULL, NULL);
3317 PyDict_SetItemString(d, "error", PyCursesError);
Guido van Rossumfbea2f31994-08-31 22:05:27 +00003318
Victor Stinner26486ea2010-05-15 22:23:53 +00003319 /* Make the version available */
3320 v = PyBytes_FromString(PyCursesVersion);
3321 PyDict_SetItemString(d, "version", v);
3322 PyDict_SetItemString(d, "__version__", v);
3323 Py_DECREF(v);
Guido van Rossumf6971e21994-08-30 12:25:20 +00003324
Victor Stinner26486ea2010-05-15 22:23:53 +00003325 SetDictInt("ERR", ERR);
3326 SetDictInt("OK", OK);
Andrew M. Kuchlingcaefb372000-12-26 15:57:01 +00003327
Victor Stinner26486ea2010-05-15 22:23:53 +00003328 /* Here are some attributes you can add to chars to print */
3329
3330 SetDictInt("A_ATTRIBUTES", A_ATTRIBUTES);
3331 SetDictInt("A_NORMAL", A_NORMAL);
3332 SetDictInt("A_STANDOUT", A_STANDOUT);
3333 SetDictInt("A_UNDERLINE", A_UNDERLINE);
3334 SetDictInt("A_REVERSE", A_REVERSE);
3335 SetDictInt("A_BLINK", A_BLINK);
3336 SetDictInt("A_DIM", A_DIM);
3337 SetDictInt("A_BOLD", A_BOLD);
3338 SetDictInt("A_ALTCHARSET", A_ALTCHARSET);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003339#if !defined(__NetBSD__)
Victor Stinner26486ea2010-05-15 22:23:53 +00003340 SetDictInt("A_INVIS", A_INVIS);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003341#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003342 SetDictInt("A_PROTECT", A_PROTECT);
3343 SetDictInt("A_CHARTEXT", A_CHARTEXT);
3344 SetDictInt("A_COLOR", A_COLOR);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003345
Victor Stinner26486ea2010-05-15 22:23:53 +00003346 /* The following are never available with strict SYSV curses */
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003347#ifdef A_HORIZONTAL
Victor Stinner26486ea2010-05-15 22:23:53 +00003348 SetDictInt("A_HORIZONTAL", A_HORIZONTAL);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003349#endif
3350#ifdef A_LEFT
Victor Stinner26486ea2010-05-15 22:23:53 +00003351 SetDictInt("A_LEFT", A_LEFT);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003352#endif
3353#ifdef A_LOW
Victor Stinner26486ea2010-05-15 22:23:53 +00003354 SetDictInt("A_LOW", A_LOW);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003355#endif
3356#ifdef A_RIGHT
Victor Stinner26486ea2010-05-15 22:23:53 +00003357 SetDictInt("A_RIGHT", A_RIGHT);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003358#endif
3359#ifdef A_TOP
Victor Stinner26486ea2010-05-15 22:23:53 +00003360 SetDictInt("A_TOP", A_TOP);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003361#endif
3362#ifdef A_VERTICAL
Victor Stinner26486ea2010-05-15 22:23:53 +00003363 SetDictInt("A_VERTICAL", A_VERTICAL);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003364#endif
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00003365
Victor Stinner26486ea2010-05-15 22:23:53 +00003366 SetDictInt("COLOR_BLACK", COLOR_BLACK);
3367 SetDictInt("COLOR_RED", COLOR_RED);
3368 SetDictInt("COLOR_GREEN", COLOR_GREEN);
3369 SetDictInt("COLOR_YELLOW", COLOR_YELLOW);
3370 SetDictInt("COLOR_BLUE", COLOR_BLUE);
3371 SetDictInt("COLOR_MAGENTA", COLOR_MAGENTA);
3372 SetDictInt("COLOR_CYAN", COLOR_CYAN);
3373 SetDictInt("COLOR_WHITE", COLOR_WHITE);
Guido van Rossumf6971e21994-08-30 12:25:20 +00003374
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003375#ifdef NCURSES_MOUSE_VERSION
Victor Stinner26486ea2010-05-15 22:23:53 +00003376 /* Mouse-related constants */
3377 SetDictInt("BUTTON1_PRESSED", BUTTON1_PRESSED);
3378 SetDictInt("BUTTON1_RELEASED", BUTTON1_RELEASED);
3379 SetDictInt("BUTTON1_CLICKED", BUTTON1_CLICKED);
3380 SetDictInt("BUTTON1_DOUBLE_CLICKED", BUTTON1_DOUBLE_CLICKED);
3381 SetDictInt("BUTTON1_TRIPLE_CLICKED", BUTTON1_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003382
Victor Stinner26486ea2010-05-15 22:23:53 +00003383 SetDictInt("BUTTON2_PRESSED", BUTTON2_PRESSED);
3384 SetDictInt("BUTTON2_RELEASED", BUTTON2_RELEASED);
3385 SetDictInt("BUTTON2_CLICKED", BUTTON2_CLICKED);
3386 SetDictInt("BUTTON2_DOUBLE_CLICKED", BUTTON2_DOUBLE_CLICKED);
3387 SetDictInt("BUTTON2_TRIPLE_CLICKED", BUTTON2_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003388
Victor Stinner26486ea2010-05-15 22:23:53 +00003389 SetDictInt("BUTTON3_PRESSED", BUTTON3_PRESSED);
3390 SetDictInt("BUTTON3_RELEASED", BUTTON3_RELEASED);
3391 SetDictInt("BUTTON3_CLICKED", BUTTON3_CLICKED);
3392 SetDictInt("BUTTON3_DOUBLE_CLICKED", BUTTON3_DOUBLE_CLICKED);
3393 SetDictInt("BUTTON3_TRIPLE_CLICKED", BUTTON3_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003394
Victor Stinner26486ea2010-05-15 22:23:53 +00003395 SetDictInt("BUTTON4_PRESSED", BUTTON4_PRESSED);
3396 SetDictInt("BUTTON4_RELEASED", BUTTON4_RELEASED);
3397 SetDictInt("BUTTON4_CLICKED", BUTTON4_CLICKED);
3398 SetDictInt("BUTTON4_DOUBLE_CLICKED", BUTTON4_DOUBLE_CLICKED);
3399 SetDictInt("BUTTON4_TRIPLE_CLICKED", BUTTON4_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003400
Victor Stinner26486ea2010-05-15 22:23:53 +00003401 SetDictInt("BUTTON_SHIFT", BUTTON_SHIFT);
3402 SetDictInt("BUTTON_CTRL", BUTTON_CTRL);
3403 SetDictInt("BUTTON_ALT", BUTTON_ALT);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003404
Victor Stinner26486ea2010-05-15 22:23:53 +00003405 SetDictInt("ALL_MOUSE_EVENTS", ALL_MOUSE_EVENTS);
3406 SetDictInt("REPORT_MOUSE_POSITION", REPORT_MOUSE_POSITION);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003407#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003408 /* Now set everything up for KEY_ variables */
3409 {
3410 int key;
3411 char *key_n;
3412 char *key_n2;
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003413#if !defined(__NetBSD__)
Victor Stinner26486ea2010-05-15 22:23:53 +00003414 for (key=KEY_MIN;key < KEY_MAX; key++) {
3415 key_n = (char *)keyname(key);
3416 if (key_n == NULL || strcmp(key_n,"UNKNOWN KEY")==0)
3417 continue;
3418 if (strncmp(key_n,"KEY_F(",6)==0) {
3419 char *p1, *p2;
Victor Stinnerb6404912013-07-07 16:21:41 +02003420 key_n2 = PyMem_Malloc(strlen(key_n)+1);
Victor Stinner26486ea2010-05-15 22:23:53 +00003421 if (!key_n2) {
3422 PyErr_NoMemory();
3423 break;
3424 }
3425 p1 = key_n;
3426 p2 = key_n2;
3427 while (*p1) {
3428 if (*p1 != '(' && *p1 != ')') {
3429 *p2 = *p1;
3430 p2++;
3431 }
3432 p1++;
3433 }
3434 *p2 = (char)0;
3435 } else
3436 key_n2 = key_n;
3437 SetDictInt(key_n2,key);
3438 if (key_n2 != key_n)
Victor Stinnerb6404912013-07-07 16:21:41 +02003439 PyMem_Free(key_n2);
Victor Stinner26486ea2010-05-15 22:23:53 +00003440 }
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003441#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003442 SetDictInt("KEY_MIN", KEY_MIN);
3443 SetDictInt("KEY_MAX", KEY_MAX);
3444 }
3445 return m;
Guido van Rossumf6971e21994-08-30 12:25:20 +00003446}