blob: de45f5189cbefb1dbc96eed4176ebf4a4406e662 [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. Kuchling52a14c32005-06-09 17:53:27 +0000171#ifndef MIN
172#define MIN(x,y) ((x) < (y) ? (x) : (y))
173#endif
174
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000175/* Utility Functions */
Guido van Rossumfbea2f31994-08-31 22:05:27 +0000176
Guido van Rossum85738471995-02-17 13:50:17 +0000177/*
Victor Stinner26486ea2010-05-15 22:23:53 +0000178 * Check the return code from a curses function and return None
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000179 * or raise an exception as appropriate. These are exported using the
Victor Stinner26486ea2010-05-15 22:23:53 +0000180 * capsule API.
Guido van Rossum85738471995-02-17 13:50:17 +0000181 */
182
183static PyObject *
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100184PyCursesCheckERR(int code, const char *fname)
Guido van Rossum85738471995-02-17 13:50:17 +0000185{
Victor Stinner26486ea2010-05-15 22:23:53 +0000186 if (code != ERR) {
187 Py_INCREF(Py_None);
188 return Py_None;
Guido van Rossum85738471995-02-17 13:50:17 +0000189 } else {
Victor Stinner26486ea2010-05-15 22:23:53 +0000190 if (fname == NULL) {
191 PyErr_SetString(PyCursesError, catchall_ERR);
192 } else {
193 PyErr_Format(PyCursesError, "%s() returned ERR", fname);
194 }
195 return NULL;
Guido van Rossum85738471995-02-17 13:50:17 +0000196 }
Guido van Rossum85738471995-02-17 13:50:17 +0000197}
198
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100199/* Convert an object to a byte (an integer of type chtype):
200
201 - int
202 - bytes of length 1
203 - str of length 1
204
205 Return 1 on success, 0 on error (invalid type or integer overflow). */
Victor Stinner26486ea2010-05-15 22:23:53 +0000206static int
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100207PyCurses_ConvertToChtype(PyCursesWindowObject *win, PyObject *obj, chtype *ch)
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000208{
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100209 long value;
210 if(PyBytes_Check(obj) && PyBytes_Size(obj) == 1) {
211 value = (unsigned char)PyBytes_AsString(obj)[0];
212 }
213 else if (PyUnicode_Check(obj)) {
214 if (PyUnicode_GetLength(obj) != 1) {
215 PyErr_Format(PyExc_TypeError,
216 "expect bytes or str of length 1, or int, "
217 "got a str of length %zi",
218 PyUnicode_GET_LENGTH(obj));
Victor Stinner26486ea2010-05-15 22:23:53 +0000219 return 0;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100220 }
221 value = PyUnicode_READ_CHAR(obj, 0);
222 if (128 < value) {
223 PyObject *bytes;
224 const char *encoding;
225 if (win)
226 encoding = win->encoding;
227 else
228 encoding = screen_encoding;
229 bytes = PyUnicode_AsEncodedObject(obj, encoding, NULL);
230 if (bytes == NULL)
231 return 0;
232 if (PyBytes_GET_SIZE(bytes) == 1)
233 value = (unsigned char)PyBytes_AS_STRING(bytes)[0];
234 else
235 value = -1;
236 Py_DECREF(bytes);
237 if (value < 0)
238 goto overflow;
239 }
240 }
241 else if (PyLong_CheckExact(obj)) {
242 int long_overflow;
243 value = PyLong_AsLongAndOverflow(obj, &long_overflow);
244 if (long_overflow)
245 goto overflow;
246 }
247 else {
248 PyErr_Format(PyExc_TypeError,
249 "expect bytes or str of length 1, or int, got %s",
250 Py_TYPE(obj)->tp_name);
Victor Stinner26486ea2010-05-15 22:23:53 +0000251 return 0;
252 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100253 *ch = (chtype)value;
254 if ((long)*ch != value)
255 goto overflow;
Victor Stinner26486ea2010-05-15 22:23:53 +0000256 return 1;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100257
258overflow:
259 PyErr_SetString(PyExc_OverflowError,
260 "byte doesn't fit in chtype");
261 return 0;
262}
263
264/* Convert an object to a byte (chtype) or a character (cchar_t):
265
266 - int
267 - bytes of length 1
268 - str of length 1
269
270 Return:
271
272 - 2 if obj is a character (written into *wch)
273 - 1 if obj is a byte (written into *ch)
274 - 0 on error: raise an exception */
275static int
276PyCurses_ConvertToCchar_t(PyCursesWindowObject *win, PyObject *obj,
277 chtype *ch
278#ifdef HAVE_NCURSESW
279 , cchar_t *wch
280#endif
281 )
282{
283 int ret = 0;
284 long value;
285#ifdef HAVE_NCURSESW
286 wchar_t buffer[2];
287#endif
288
289 if (PyUnicode_Check(obj)) {
290#ifdef HAVE_NCURSESW
291 if (PyUnicode_AsWideChar(obj, buffer, 2) != 1) {
292 PyErr_Format(PyExc_TypeError,
293 "expect bytes or str of length 1, or int, "
294 "got a str of length %zi",
295 PyUnicode_GET_LENGTH(obj));
296 return 0;
297 }
298 memset(wch->chars, 0, sizeof(wch->chars));
299 wch->chars[0] = buffer[0];
300 return 2;
301#else
302 return PyCurses_ConvertToChtype(win, obj, ch);
303#endif
304 }
305 else if(PyBytes_Check(obj) && PyBytes_Size(obj) == 1) {
306 value = (unsigned char)PyBytes_AsString(obj)[0];
307 ret = 1;
308 }
309 else if (PyLong_CheckExact(obj)) {
310 int overflow;
311 value = PyLong_AsLongAndOverflow(obj, &overflow);
312 if (overflow) {
313 PyErr_SetString(PyExc_OverflowError,
314 "int doesn't fit in long");
315 return 0;
316 }
317#ifdef HAVE_NCURSESW
318 ret = 2;
319#else
320 ret = 1;
321#endif
322 }
323 else {
324 PyErr_Format(PyExc_TypeError,
325 "expect bytes or str of length 1, or int, got %s",
326 Py_TYPE(obj)->tp_name);
327 return 0;
328 }
329#ifdef HAVE_NCURSESW
330 if (ret == 2) {
331 memset(wch->chars, 0, sizeof(wch->chars));
332 wch->chars[0] = (wchar_t)value;
333 if ((long)wch->chars[0] != value) {
334 PyErr_Format(PyExc_OverflowError,
335 "character doesn't fit in wchar_t");
336 return 0;
337 }
338 }
339 else
340#endif
341 {
342 *ch = (chtype)value;
Victor Stinner8f408602012-03-08 02:08:48 +0100343 if ((long)*ch != value) {
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100344 PyErr_Format(PyExc_OverflowError,
345 "byte doesn't fit in chtype");
346 return 0;
347 }
348 }
349 return ret;
350}
351
352/* Convert an object to a byte string (char*) or a wide character string
353 (wchar_t*). Return:
354
355 - 2 if obj is a character string (written into *wch)
356 - 1 if obj is a byte string (written into *bytes)
357 - 0 on error: raise an exception */
358static int
359PyCurses_ConvertToString(PyCursesWindowObject *win, PyObject *obj,
360 PyObject **bytes, wchar_t **wstr)
361{
362 if (PyUnicode_Check(obj)) {
363#ifdef HAVE_NCURSESW
364 assert (wstr != NULL);
365 *wstr = PyUnicode_AsWideCharString(obj, NULL);
366 if (*wstr == NULL)
367 return 0;
368 return 2;
369#else
370 assert (wstr == NULL);
371 *bytes = PyUnicode_AsEncodedObject(obj, win->encoding, NULL);
372 if (*bytes == NULL)
373 return 0;
374 return 1;
375#endif
376 }
377 else if (PyBytes_Check(obj)) {
378 Py_INCREF(obj);
379 *bytes = obj;
380 return 1;
381 }
382
383 PyErr_Format(PyExc_TypeError, "expect bytes or str, got %s",
384 Py_TYPE(obj)->tp_name);
385 return 0;
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000386}
387
Benjamin Peterson21896a32010-03-21 22:03:03 +0000388/* Function versions of the 3 functions for testing whether curses has been
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000389 initialised or not. */
Victor Stinner26486ea2010-05-15 22:23:53 +0000390
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000391static int func_PyCursesSetupTermCalled(void)
392{
393 PyCursesSetupTermCalled;
394 return 1;
395}
396
397static int func_PyCursesInitialised(void)
398{
399 PyCursesInitialised;
400 return 1;
401}
402
403static int func_PyCursesInitialisedColor(void)
404{
405 PyCursesInitialisedColor;
406 return 1;
407}
408
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000409/*****************************************************************************
410 The Window Object
411******************************************************************************/
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000412
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000413/* Definition of the window type */
Guido van Rossum85738471995-02-17 13:50:17 +0000414
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000415PyTypeObject PyCursesWindow_Type;
416
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000417/* Function prototype macros for Window object
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000418
419 X - function name
420 TYPE - parameter Type
421 ERGSTR - format string for construction of the return value
422 PARSESTR - format string for argument parsing
Victor Stinner26486ea2010-05-15 22:23:53 +0000423*/
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000424
Victor Stinner26486ea2010-05-15 22:23:53 +0000425#define Window_NoArgNoReturnFunction(X) \
426 static PyObject *PyCursesWindow_ ## X \
427 (PyCursesWindowObject *self, PyObject *args) \
428 { return PyCursesCheckERR(X(self->win), # X); }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000429
Victor Stinner26486ea2010-05-15 22:23:53 +0000430#define Window_NoArgTrueFalseFunction(X) \
431 static PyObject * PyCursesWindow_ ## X \
432 (PyCursesWindowObject *self) \
433 { \
434 if (X (self->win) == FALSE) { Py_INCREF(Py_False); return Py_False; } \
435 else { Py_INCREF(Py_True); return Py_True; } }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000436
Victor Stinner26486ea2010-05-15 22:23:53 +0000437#define Window_NoArgNoReturnVoidFunction(X) \
438 static PyObject * PyCursesWindow_ ## X \
439 (PyCursesWindowObject *self) \
440 { \
441 X(self->win); Py_INCREF(Py_None); return Py_None; }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000442
Victor Stinner26486ea2010-05-15 22:23:53 +0000443#define Window_NoArg2TupleReturnFunction(X, TYPE, ERGSTR) \
444 static PyObject * PyCursesWindow_ ## X \
445 (PyCursesWindowObject *self) \
446 { \
447 TYPE arg1, arg2; \
448 X(self->win,arg1,arg2); return Py_BuildValue(ERGSTR, arg1, arg2); }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000449
Victor Stinner26486ea2010-05-15 22:23:53 +0000450#define Window_OneArgNoReturnVoidFunction(X, TYPE, PARSESTR) \
451 static PyObject * PyCursesWindow_ ## X \
452 (PyCursesWindowObject *self, PyObject *args) \
453 { \
454 TYPE arg1; \
455 if (!PyArg_ParseTuple(args, PARSESTR, &arg1)) return NULL; \
456 X(self->win,arg1); Py_INCREF(Py_None); return Py_None; }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000457
Victor Stinner26486ea2010-05-15 22:23:53 +0000458#define Window_OneArgNoReturnFunction(X, TYPE, PARSESTR) \
459 static PyObject * PyCursesWindow_ ## X \
460 (PyCursesWindowObject *self, PyObject *args) \
461 { \
462 TYPE arg1; \
463 if (!PyArg_ParseTuple(args,PARSESTR, &arg1)) return NULL; \
464 return PyCursesCheckERR(X(self->win, arg1), # X); }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000465
Victor Stinner26486ea2010-05-15 22:23:53 +0000466#define Window_TwoArgNoReturnFunction(X, TYPE, PARSESTR) \
467 static PyObject * PyCursesWindow_ ## X \
468 (PyCursesWindowObject *self, PyObject *args) \
469 { \
470 TYPE arg1, arg2; \
471 if (!PyArg_ParseTuple(args,PARSESTR, &arg1, &arg2)) return NULL; \
472 return PyCursesCheckERR(X(self->win, arg1, arg2), # X); }
Guido van Rossum85738471995-02-17 13:50:17 +0000473
Guido van Rossumf6971e21994-08-30 12:25:20 +0000474/* ------------- WINDOW routines --------------- */
Guido van Rossum85738471995-02-17 13:50:17 +0000475
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000476Window_NoArgNoReturnFunction(untouchwin)
477Window_NoArgNoReturnFunction(touchwin)
478Window_NoArgNoReturnFunction(redrawwin)
479Window_NoArgNoReturnFunction(winsertln)
480Window_NoArgNoReturnFunction(werase)
481Window_NoArgNoReturnFunction(wdeleteln)
482
483Window_NoArgTrueFalseFunction(is_wintouched)
484
485Window_NoArgNoReturnVoidFunction(wsyncup)
486Window_NoArgNoReturnVoidFunction(wsyncdown)
487Window_NoArgNoReturnVoidFunction(wstandend)
488Window_NoArgNoReturnVoidFunction(wstandout)
489Window_NoArgNoReturnVoidFunction(wcursyncup)
490Window_NoArgNoReturnVoidFunction(wclrtoeol)
491Window_NoArgNoReturnVoidFunction(wclrtobot)
492Window_NoArgNoReturnVoidFunction(wclear)
493
494Window_OneArgNoReturnVoidFunction(idcok, int, "i;True(1) or False(0)")
495Window_OneArgNoReturnVoidFunction(immedok, int, "i;True(1) or False(0)")
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +0000496Window_OneArgNoReturnVoidFunction(wtimeout, int, "i;delay")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000497
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000498Window_NoArg2TupleReturnFunction(getyx, int, "ii")
499Window_NoArg2TupleReturnFunction(getbegyx, int, "ii")
500Window_NoArg2TupleReturnFunction(getmaxyx, int, "ii")
501Window_NoArg2TupleReturnFunction(getparyx, int, "ii")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000502
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000503Window_OneArgNoReturnFunction(clearok, int, "i;True(1) or False(0)")
504Window_OneArgNoReturnFunction(idlok, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000505#if defined(__NetBSD__)
506Window_OneArgNoReturnVoidFunction(keypad, int, "i;True(1) or False(0)")
507#else
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000508Window_OneArgNoReturnFunction(keypad, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000509#endif
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000510Window_OneArgNoReturnFunction(leaveok, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000511#if defined(__NetBSD__)
512Window_OneArgNoReturnVoidFunction(nodelay, int, "i;True(1) or False(0)")
513#else
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000514Window_OneArgNoReturnFunction(nodelay, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000515#endif
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000516Window_OneArgNoReturnFunction(notimeout, int, "i;True(1) or False(0)")
517Window_OneArgNoReturnFunction(scrollok, int, "i;True(1) or False(0)")
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +0000518Window_OneArgNoReturnFunction(winsdelln, int, "i;nlines")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000519Window_OneArgNoReturnFunction(syncok, int, "i;True(1) or False(0)")
520
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000521Window_TwoArgNoReturnFunction(mvwin, int, "ii;y,x")
522Window_TwoArgNoReturnFunction(mvderwin, int, "ii;y,x")
523Window_TwoArgNoReturnFunction(wmove, int, "ii;y,x")
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +0000524#ifndef STRICT_SYSV_CURSES
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000525Window_TwoArgNoReturnFunction(wresize, int, "ii;lines,columns")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000526#endif
527
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +0000528/* Allocation and deallocation of Window Objects */
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000529
Guido van Rossumf6971e21994-08-30 12:25:20 +0000530static PyObject *
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100531PyCursesWindow_New(WINDOW *win, const char *encoding)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000532{
Victor Stinner26486ea2010-05-15 22:23:53 +0000533 PyCursesWindowObject *wo;
Guido van Rossum85738471995-02-17 13:50:17 +0000534
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100535 if (encoding == NULL) {
536#if defined(MS_WINDOWS)
537 char *buffer[100];
538 UINT cp;
539 cp = GetConsoleOutputCP();
540 if (cp != 0) {
541 PyOS_snprintf(buffer, sizeof(buffer), "cp%u", cp);
542 encoding = buffer;
543 }
544#elif defined(CODESET)
545 const char *codeset = nl_langinfo(CODESET);
546 if (codeset != NULL && codeset[0] != 0)
547 encoding = codeset;
548#endif
549 if (encoding == NULL)
550 encoding = "utf-8";
551 }
552
Victor Stinner26486ea2010-05-15 22:23:53 +0000553 wo = PyObject_NEW(PyCursesWindowObject, &PyCursesWindow_Type);
554 if (wo == NULL) return NULL;
555 wo->win = win;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100556 wo->encoding = strdup(encoding);
557 if (wo->encoding == NULL) {
558 Py_DECREF(wo);
559 PyErr_NoMemory();
560 return NULL;
561 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000562 return (PyObject *)wo;
Guido van Rossumf6971e21994-08-30 12:25:20 +0000563}
564
565static void
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000566PyCursesWindow_Dealloc(PyCursesWindowObject *wo)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000567{
Victor Stinner26486ea2010-05-15 22:23:53 +0000568 if (wo->win != stdscr) delwin(wo->win);
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100569 if (wo->encoding != NULL)
570 free(wo->encoding);
Victor Stinner26486ea2010-05-15 22:23:53 +0000571 PyObject_DEL(wo);
Guido van Rossumf6971e21994-08-30 12:25:20 +0000572}
573
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000574/* Addch, Addstr, Addnstr */
Guido van Rossumf6971e21994-08-30 12:25:20 +0000575
576static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000577PyCursesWindow_AddCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000578{
Victor Stinner26486ea2010-05-15 22:23:53 +0000579 int rtn, x, y, use_xy = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100580 PyObject *chobj;
581 int type;
582 chtype ch;
583#ifdef HAVE_NCURSESW
584 cchar_t wch;
585#endif
Victor Stinner26486ea2010-05-15 22:23:53 +0000586 attr_t attr = A_NORMAL;
587 long lattr;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100588 const char *funcname;
Guido van Rossum85738471995-02-17 13:50:17 +0000589
Victor Stinner26486ea2010-05-15 22:23:53 +0000590 switch (PyTuple_Size(args)) {
591 case 1:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100592 if (!PyArg_ParseTuple(args, "O;ch or int", &chobj))
Victor Stinner26486ea2010-05-15 22:23:53 +0000593 return NULL;
594 break;
595 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100596 if (!PyArg_ParseTuple(args, "Ol;ch or int,attr", &chobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000597 return NULL;
598 attr = lattr;
599 break;
600 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100601 if (!PyArg_ParseTuple(args,"iiO;y,x,ch or int", &y, &x, &chobj))
Victor Stinner26486ea2010-05-15 22:23:53 +0000602 return NULL;
603 use_xy = TRUE;
604 break;
605 case 4:
606 if (!PyArg_ParseTuple(args,"iiOl;y,x,ch or int, attr",
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100607 &y, &x, &chobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000608 return NULL;
609 attr = lattr;
610 use_xy = TRUE;
611 break;
612 default:
613 PyErr_SetString(PyExc_TypeError, "addch requires 1 to 4 arguments");
614 return NULL;
615 }
616
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100617#ifdef HAVE_NCURSESW
618 type = PyCurses_ConvertToCchar_t(self, chobj, &ch, &wch);
619 if (type == 2) {
620 funcname = "add_wch";
621 wch.attr = attr;
622 if (use_xy == TRUE)
623 rtn = mvwadd_wch(self->win,y,x, &wch);
624 else {
625 rtn = wadd_wch(self->win, &wch);
626 }
627 }
628 else
629#else
630 type = PyCurses_ConvertToCchar_t(self, chobj, &ch);
631#endif
632 if (type == 1) {
633 funcname = "addch";
634 if (use_xy == TRUE)
635 rtn = mvwaddch(self->win,y,x, ch | attr);
636 else {
637 rtn = waddch(self->win, ch | attr);
638 }
639 }
640 else {
Victor Stinner26486ea2010-05-15 22:23:53 +0000641 return NULL;
642 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100643 return PyCursesCheckERR(rtn, funcname);
Guido van Rossumf6971e21994-08-30 12:25:20 +0000644}
645
646static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000647PyCursesWindow_AddStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000648{
Victor Stinner26486ea2010-05-15 22:23:53 +0000649 int rtn;
650 int x, y;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100651 int strtype;
Benjamin Peterson5e545192012-03-14 18:21:35 -0500652 PyObject *strobj, *bytesobj = NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100653#ifdef HAVE_NCURSESW
654 wchar_t *wstr = NULL;
655#endif
Victor Stinner26486ea2010-05-15 22:23:53 +0000656 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
657 long lattr;
658 int use_xy = FALSE, use_attr = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100659 const char *funcname;
Guido van Rossum85738471995-02-17 13:50:17 +0000660
Victor Stinner26486ea2010-05-15 22:23:53 +0000661 switch (PyTuple_Size(args)) {
662 case 1:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100663 if (!PyArg_ParseTuple(args,"O;str", &strobj))
Victor Stinner26486ea2010-05-15 22:23:53 +0000664 return NULL;
665 break;
666 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100667 if (!PyArg_ParseTuple(args,"Ol;str,attr", &strobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000668 return NULL;
669 attr = lattr;
670 use_attr = TRUE;
671 break;
672 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100673 if (!PyArg_ParseTuple(args,"iiO;int,int,str", &y, &x, &strobj))
Victor Stinner26486ea2010-05-15 22:23:53 +0000674 return NULL;
675 use_xy = TRUE;
676 break;
677 case 4:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100678 if (!PyArg_ParseTuple(args,"iiOl;int,int,str,attr", &y, &x, &strobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000679 return NULL;
680 attr = lattr;
681 use_xy = use_attr = TRUE;
682 break;
683 default:
684 PyErr_SetString(PyExc_TypeError, "addstr requires 1 to 4 arguments");
685 return NULL;
686 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100687#ifdef HAVE_NCURSESW
688 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);
689#else
690 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);
691#endif
692 if (strtype == 0)
693 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +0000694 if (use_attr == TRUE) {
695 attr_old = getattrs(self->win);
696 (void)wattrset(self->win,attr);
697 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100698#ifdef HAVE_NCURSESW
699 if (strtype == 2) {
700 funcname = "addwstr";
701 if (use_xy == TRUE)
702 rtn = mvwaddwstr(self->win,y,x,wstr);
703 else
704 rtn = waddwstr(self->win,wstr);
705 PyMem_Free(wstr);
706 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000707 else
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100708#endif
709 {
710 char *str = PyBytes_AS_STRING(bytesobj);
711 funcname = "addstr";
712 if (use_xy == TRUE)
713 rtn = mvwaddstr(self->win,y,x,str);
714 else
715 rtn = waddstr(self->win,str);
716 Py_DECREF(bytesobj);
717 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000718 if (use_attr == TRUE)
719 (void)wattrset(self->win,attr_old);
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100720 return PyCursesCheckERR(rtn, funcname);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000721}
Guido van Rossum85738471995-02-17 13:50:17 +0000722
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000723static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000724PyCursesWindow_AddNStr(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000725{
Victor Stinner26486ea2010-05-15 22:23:53 +0000726 int rtn, x, y, n;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100727 int strtype;
Benjamin Peterson5e545192012-03-14 18:21:35 -0500728 PyObject *strobj, *bytesobj = NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100729#ifdef HAVE_NCURSESW
730 wchar_t *wstr = NULL;
731#endif
Victor Stinner26486ea2010-05-15 22:23:53 +0000732 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
733 long lattr;
734 int use_xy = FALSE, use_attr = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100735 const char *funcname;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000736
Victor Stinner26486ea2010-05-15 22:23:53 +0000737 switch (PyTuple_Size(args)) {
738 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100739 if (!PyArg_ParseTuple(args,"Oi;str,n", &strobj, &n))
Victor Stinner26486ea2010-05-15 22:23:53 +0000740 return NULL;
741 break;
742 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100743 if (!PyArg_ParseTuple(args,"Oil;str,n,attr", &strobj, &n, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000744 return NULL;
745 attr = lattr;
746 use_attr = TRUE;
747 break;
748 case 4:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100749 if (!PyArg_ParseTuple(args,"iiOi;y,x,str,n", &y, &x, &strobj, &n))
Victor Stinner26486ea2010-05-15 22:23:53 +0000750 return NULL;
751 use_xy = TRUE;
752 break;
753 case 5:
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100754 if (!PyArg_ParseTuple(args,"iiOil;y,x,str,n,attr", &y, &x, &strobj, &n, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +0000755 return NULL;
756 attr = lattr;
757 use_xy = use_attr = TRUE;
758 break;
759 default:
760 PyErr_SetString(PyExc_TypeError, "addnstr requires 2 to 5 arguments");
761 return NULL;
762 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100763#ifdef HAVE_NCURSESW
764 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);
765#else
766 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);
767#endif
768 if (strtype == 0)
769 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000770
Victor Stinner26486ea2010-05-15 22:23:53 +0000771 if (use_attr == TRUE) {
772 attr_old = getattrs(self->win);
773 (void)wattrset(self->win,attr);
774 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100775#ifdef HAVE_NCURSESW
776 if (strtype == 2) {
777 funcname = "addnwstr";
778 if (use_xy == TRUE)
779 rtn = mvwaddnwstr(self->win,y,x,wstr,n);
780 else
781 rtn = waddnwstr(self->win,wstr,n);
782 PyMem_Free(wstr);
783 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000784 else
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100785#endif
786 {
787 char *str = PyBytes_AS_STRING(bytesobj);
788 funcname = "addnstr";
789 if (use_xy == TRUE)
790 rtn = mvwaddnstr(self->win,y,x,str,n);
791 else
792 rtn = waddnstr(self->win,str,n);
793 Py_DECREF(bytesobj);
794 }
Victor Stinner26486ea2010-05-15 22:23:53 +0000795 if (use_attr == TRUE)
796 (void)wattrset(self->win,attr_old);
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100797 return PyCursesCheckERR(rtn, funcname);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000798}
799
800static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000801PyCursesWindow_Bkgd(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000802{
Victor Stinner26486ea2010-05-15 22:23:53 +0000803 PyObject *temp;
804 chtype bkgd;
805 attr_t attr = A_NORMAL;
806 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000807
Victor Stinner26486ea2010-05-15 22:23:53 +0000808 switch (PyTuple_Size(args)) {
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000809 case 1:
Victor Stinner26486ea2010-05-15 22:23:53 +0000810 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
811 return NULL;
812 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000813 case 2:
Victor Stinner26486ea2010-05-15 22:23:53 +0000814 if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))
815 return NULL;
816 attr = lattr;
817 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000818 default:
Victor Stinner26486ea2010-05-15 22:23:53 +0000819 PyErr_SetString(PyExc_TypeError, "bkgd requires 1 or 2 arguments");
820 return NULL;
821 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000822
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100823 if (!PyCurses_ConvertToChtype(self, temp, &bkgd))
Victor Stinner26486ea2010-05-15 22:23:53 +0000824 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000825
Victor Stinner26486ea2010-05-15 22:23:53 +0000826 return PyCursesCheckERR(wbkgd(self->win, bkgd | attr), "bkgd");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000827}
828
829static PyObject *
Christian Heimes2380ac72008-01-09 00:17:24 +0000830PyCursesWindow_AttrOff(PyCursesWindowObject *self, PyObject *args)
831{
Victor Stinner26486ea2010-05-15 22:23:53 +0000832 long lattr;
833 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
834 return NULL;
835 return PyCursesCheckERR(wattroff(self->win, (attr_t)lattr), "attroff");
Christian Heimes2380ac72008-01-09 00:17:24 +0000836}
837
838static PyObject *
839PyCursesWindow_AttrOn(PyCursesWindowObject *self, PyObject *args)
840{
Victor Stinner26486ea2010-05-15 22:23:53 +0000841 long lattr;
842 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
843 return NULL;
844 return PyCursesCheckERR(wattron(self->win, (attr_t)lattr), "attron");
Christian Heimes2380ac72008-01-09 00:17:24 +0000845}
846
847static PyObject *
848PyCursesWindow_AttrSet(PyCursesWindowObject *self, PyObject *args)
849{
Victor Stinner26486ea2010-05-15 22:23:53 +0000850 long lattr;
851 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
852 return NULL;
853 return PyCursesCheckERR(wattrset(self->win, (attr_t)lattr), "attrset");
Christian Heimes2380ac72008-01-09 00:17:24 +0000854}
855
856static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000857PyCursesWindow_BkgdSet(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000858{
Victor Stinner26486ea2010-05-15 22:23:53 +0000859 PyObject *temp;
860 chtype bkgd;
861 attr_t attr = A_NORMAL;
862 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000863
Victor Stinner26486ea2010-05-15 22:23:53 +0000864 switch (PyTuple_Size(args)) {
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000865 case 1:
Victor Stinner26486ea2010-05-15 22:23:53 +0000866 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
867 return NULL;
868 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000869 case 2:
Victor Stinner26486ea2010-05-15 22:23:53 +0000870 if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))
871 return NULL;
872 attr = lattr;
873 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000874 default:
Victor Stinner26486ea2010-05-15 22:23:53 +0000875 PyErr_SetString(PyExc_TypeError, "bkgdset requires 1 or 2 arguments");
876 return NULL;
877 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000878
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100879 if (!PyCurses_ConvertToChtype(self, temp, &bkgd))
Victor Stinner26486ea2010-05-15 22:23:53 +0000880 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000881
Victor Stinner26486ea2010-05-15 22:23:53 +0000882 wbkgdset(self->win, bkgd | attr);
883 return PyCursesCheckERR(0, "bkgdset");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000884}
885
886static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000887PyCursesWindow_Border(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000888{
Victor Stinner26486ea2010-05-15 22:23:53 +0000889 PyObject *temp[8];
890 chtype ch[8];
891 int i;
Andrew M. Kuchling16e65a02001-10-20 16:05:52 +0000892
Victor Stinner26486ea2010-05-15 22:23:53 +0000893 /* Clear the array of parameters */
894 for(i=0; i<8; i++) {
895 temp[i] = NULL;
896 ch[i] = 0;
897 }
Andrew M. Kuchling16e65a02001-10-20 16:05:52 +0000898
Victor Stinner26486ea2010-05-15 22:23:53 +0000899 if (!PyArg_ParseTuple(args,"|OOOOOOOO;ls,rs,ts,bs,tl,tr,bl,br",
900 &temp[0], &temp[1], &temp[2], &temp[3],
901 &temp[4], &temp[5], &temp[6], &temp[7]))
902 return NULL;
903
904 for(i=0; i<8; i++) {
Victor Stinner0fdfceb2011-11-25 22:10:02 +0100905 if (temp[i] != NULL && !PyCurses_ConvertToChtype(self, temp[i], &ch[i]))
Victor Stinner26486ea2010-05-15 22:23:53 +0000906 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +0000907 }
908
909 wborder(self->win,
910 ch[0], ch[1], ch[2], ch[3],
911 ch[4], ch[5], ch[6], ch[7]);
912 Py_INCREF(Py_None);
913 return Py_None;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000914}
915
916static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000917PyCursesWindow_Box(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000918{
Victor Stinner26486ea2010-05-15 22:23:53 +0000919 chtype ch1=0,ch2=0;
920 switch(PyTuple_Size(args)){
921 case 0: break;
922 default:
923 if (!PyArg_ParseTuple(args,"ll;vertint,horint", &ch1, &ch2))
924 return NULL;
925 }
926 box(self->win,ch1,ch2);
927 Py_INCREF(Py_None);
928 return Py_None;
Guido van Rossumf6971e21994-08-30 12:25:20 +0000929}
930
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000931#if defined(HAVE_NCURSES_H) || defined(MVWDELCH_IS_EXPRESSION)
932#define py_mvwdelch mvwdelch
933#else
934int py_mvwdelch(WINDOW *w, int y, int x)
935{
Victor Stinner26486ea2010-05-15 22:23:53 +0000936 mvwdelch(w,y,x);
937 /* On HP/UX, mvwdelch already returns. On other systems,
938 we may well run into this return statement. */
939 return 0;
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000940}
941#endif
942
Guido van Rossumd8faa362007-04-27 19:54:29 +0000943/* chgat, added by Fabian Kreutz <fabian.kreutz at gmx.net> */
944
945static PyObject *
946PyCursesWindow_ChgAt(PyCursesWindowObject *self, PyObject *args)
947{
Victor Stinner26486ea2010-05-15 22:23:53 +0000948 int rtn;
949 int x, y;
950 int num = -1;
951 short color;
952 attr_t attr = A_NORMAL;
953 long lattr;
954 int use_xy = FALSE;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000955
Victor Stinner26486ea2010-05-15 22:23:53 +0000956 switch (PyTuple_Size(args)) {
957 case 1:
958 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
959 return NULL;
960 attr = lattr;
961 break;
962 case 2:
963 if (!PyArg_ParseTuple(args,"il;n,attr", &num, &lattr))
964 return NULL;
965 attr = lattr;
966 break;
967 case 3:
968 if (!PyArg_ParseTuple(args,"iil;int,int,attr", &y, &x, &lattr))
969 return NULL;
970 attr = lattr;
971 use_xy = TRUE;
972 break;
973 case 4:
974 if (!PyArg_ParseTuple(args,"iiil;int,int,n,attr", &y, &x, &num, &lattr))
975 return NULL;
976 attr = lattr;
977 use_xy = TRUE;
978 break;
979 default:
980 PyErr_SetString(PyExc_TypeError, "chgat requires 1 to 4 arguments");
981 return NULL;
982 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000983
Victor Stinner26486ea2010-05-15 22:23:53 +0000984 color = (short)((attr >> 8) & 0xff);
985 attr = attr - (color << 8);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000986
Victor Stinner26486ea2010-05-15 22:23:53 +0000987 if (use_xy == TRUE) {
988 rtn = mvwchgat(self->win,y,x,num,attr,color,NULL);
989 touchline(self->win,y,1);
990 } else {
991 getyx(self->win,y,x);
992 rtn = wchgat(self->win,num,attr,color,NULL);
993 touchline(self->win,y,1);
994 }
995 return PyCursesCheckERR(rtn, "chgat");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000996}
997
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000998
Guido van Rossumf6971e21994-08-30 12:25:20 +0000999static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001000PyCursesWindow_DelCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001001{
Victor Stinner26486ea2010-05-15 22:23:53 +00001002 int rtn;
1003 int x, y;
Guido van Rossum85738471995-02-17 13:50:17 +00001004
Victor Stinner26486ea2010-05-15 22:23:53 +00001005 switch (PyTuple_Size(args)) {
1006 case 0:
1007 rtn = wdelch(self->win);
1008 break;
1009 case 2:
1010 if (!PyArg_ParseTuple(args,"ii;y,x", &y, &x))
1011 return NULL;
1012 rtn = py_mvwdelch(self->win,y,x);
1013 break;
1014 default:
1015 PyErr_SetString(PyExc_TypeError, "delch requires 0 or 2 arguments");
1016 return NULL;
1017 }
1018 return PyCursesCheckERR(rtn, "[mv]wdelch");
Guido van Rossumf6971e21994-08-30 12:25:20 +00001019}
1020
1021static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001022PyCursesWindow_DerWin(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001023{
Victor Stinner26486ea2010-05-15 22:23:53 +00001024 WINDOW *win;
1025 int nlines, ncols, begin_y, begin_x;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001026
Victor Stinner26486ea2010-05-15 22:23:53 +00001027 nlines = 0;
1028 ncols = 0;
1029 switch (PyTuple_Size(args)) {
1030 case 2:
1031 if (!PyArg_ParseTuple(args,"ii;begin_y,begin_x",&begin_y,&begin_x))
1032 return NULL;
1033 break;
1034 case 4:
1035 if (!PyArg_ParseTuple(args, "iiii;nlines,ncols,begin_y,begin_x",
1036 &nlines,&ncols,&begin_y,&begin_x))
1037 return NULL;
1038 break;
1039 default:
1040 PyErr_SetString(PyExc_TypeError, "derwin requires 2 or 4 arguments");
1041 return NULL;
1042 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001043
Victor Stinner26486ea2010-05-15 22:23:53 +00001044 win = derwin(self->win,nlines,ncols,begin_y,begin_x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001045
Victor Stinner26486ea2010-05-15 22:23:53 +00001046 if (win == NULL) {
1047 PyErr_SetString(PyCursesError, catchall_NULL);
1048 return NULL;
1049 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001050
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001051 return (PyObject *)PyCursesWindow_New(win, NULL);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001052}
1053
1054static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001055PyCursesWindow_EchoChar(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001056{
Victor Stinner26486ea2010-05-15 22:23:53 +00001057 PyObject *temp;
1058 chtype ch;
1059 attr_t attr = A_NORMAL;
1060 long lattr;
Guido van Rossum85738471995-02-17 13:50:17 +00001061
Victor Stinner26486ea2010-05-15 22:23:53 +00001062 switch (PyTuple_Size(args)) {
1063 case 1:
1064 if (!PyArg_ParseTuple(args,"O;ch or int", &temp))
1065 return NULL;
1066 break;
1067 case 2:
1068 if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))
1069 return NULL;
1070 attr = lattr;
1071 break;
1072 default:
1073 PyErr_SetString(PyExc_TypeError, "echochar requires 1 or 2 arguments");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001074
1075
Victor Stinner26486ea2010-05-15 22:23:53 +00001076 return NULL;
1077 }
Guido van Rossum85738471995-02-17 13:50:17 +00001078
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001079 if (!PyCurses_ConvertToChtype(self, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00001080 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00001081
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00001082#ifdef WINDOW_HAS_FLAGS
Victor Stinner26486ea2010-05-15 22:23:53 +00001083 if (self->win->_flags & _ISPAD)
1084 return PyCursesCheckERR(pechochar(self->win, ch | attr),
1085 "echochar");
1086 else
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001087#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001088 return PyCursesCheckERR(wechochar(self->win, ch | attr),
1089 "echochar");
Guido van Rossumf6971e21994-08-30 12:25:20 +00001090}
1091
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001092#ifdef NCURSES_MOUSE_VERSION
1093static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001094PyCursesWindow_Enclose(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001095{
Victor Stinner26486ea2010-05-15 22:23:53 +00001096 int x, y;
1097 if (!PyArg_ParseTuple(args,"ii;y,x", &y, &x))
1098 return NULL;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001099
Victor Stinner26486ea2010-05-15 22:23:53 +00001100 return PyLong_FromLong( wenclose(self->win,y,x) );
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001101}
1102#endif
1103
Guido van Rossumf6971e21994-08-30 12:25:20 +00001104static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00001105PyCursesWindow_GetBkgd(PyCursesWindowObject *self)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001106{
Victor Stinner26486ea2010-05-15 22:23:53 +00001107 return PyLong_FromLong((long) getbkgd(self->win));
Guido van Rossumf6971e21994-08-30 12:25:20 +00001108}
1109
1110static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001111PyCursesWindow_GetCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001112{
Victor Stinner26486ea2010-05-15 22:23:53 +00001113 int x, y;
1114 int rtn;
Guido van Rossum85738471995-02-17 13:50:17 +00001115
Victor Stinner26486ea2010-05-15 22:23:53 +00001116 switch (PyTuple_Size(args)) {
1117 case 0:
1118 Py_BEGIN_ALLOW_THREADS
1119 rtn = wgetch(self->win);
1120 Py_END_ALLOW_THREADS
1121 break;
1122 case 2:
1123 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1124 return NULL;
1125 Py_BEGIN_ALLOW_THREADS
1126 rtn = mvwgetch(self->win,y,x);
1127 Py_END_ALLOW_THREADS
1128 break;
1129 default:
1130 PyErr_SetString(PyExc_TypeError, "getch requires 0 or 2 arguments");
1131 return NULL;
1132 }
1133 return PyLong_FromLong((long)rtn);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001134}
Guido van Rossum85738471995-02-17 13:50:17 +00001135
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001136static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001137PyCursesWindow_GetKey(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001138{
Victor Stinner26486ea2010-05-15 22:23:53 +00001139 int x, y;
1140 int rtn;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001141
Victor Stinner26486ea2010-05-15 22:23:53 +00001142 switch (PyTuple_Size(args)) {
1143 case 0:
1144 Py_BEGIN_ALLOW_THREADS
1145 rtn = wgetch(self->win);
1146 Py_END_ALLOW_THREADS
1147 break;
1148 case 2:
1149 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1150 return NULL;
1151 Py_BEGIN_ALLOW_THREADS
1152 rtn = mvwgetch(self->win,y,x);
1153 Py_END_ALLOW_THREADS
1154 break;
1155 default:
1156 PyErr_SetString(PyExc_TypeError, "getkey requires 0 or 2 arguments");
1157 return NULL;
1158 }
1159 if (rtn == ERR) {
1160 /* getch() returns ERR in nodelay mode */
1161 PyErr_SetString(PyCursesError, "no input");
1162 return NULL;
1163 } else if (rtn<=255) {
1164 return Py_BuildValue("C", rtn);
1165 } else {
1166 const char *knp;
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001167#if defined(__NetBSD__)
Victor Stinner26486ea2010-05-15 22:23:53 +00001168 knp = unctrl(rtn);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001169#else
Victor Stinner26486ea2010-05-15 22:23:53 +00001170 knp = keyname(rtn);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001171#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001172 return PyUnicode_FromString((knp == NULL) ? "" : knp);
1173 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001174}
1175
Nadeem Vawda9e2e9902011-07-31 15:01:11 +02001176#ifdef HAVE_NCURSESW
Guido van Rossumf6971e21994-08-30 12:25:20 +00001177static PyObject *
Victor Stinnera7878b72011-07-14 23:07:44 +02001178PyCursesWindow_Get_WCh(PyCursesWindowObject *self, PyObject *args)
1179{
1180 int x, y;
1181 int ct;
1182 wint_t rtn;
1183
1184 switch (PyTuple_Size(args)) {
1185 case 0:
1186 Py_BEGIN_ALLOW_THREADS
1187 ct = wget_wch(self->win,&rtn);
1188 Py_END_ALLOW_THREADS
1189 break;
1190 case 2:
1191 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1192 return NULL;
1193 Py_BEGIN_ALLOW_THREADS
1194 ct = mvwget_wch(self->win,y,x,&rtn);
1195 Py_END_ALLOW_THREADS
1196 break;
1197 default:
1198 PyErr_SetString(PyExc_TypeError, "get_wch requires 0 or 2 arguments");
1199 return NULL;
1200 }
1201 if (ct == ERR) {
1202 /* get_wch() returns ERR in nodelay mode */
1203 PyErr_SetString(PyCursesError, "no input");
1204 return NULL;
1205 }
1206 return PyLong_FromLong(rtn);
1207}
Nadeem Vawda9e2e9902011-07-31 15:01:11 +02001208#endif
Victor Stinnera7878b72011-07-14 23:07:44 +02001209
1210static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001211PyCursesWindow_GetStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001212{
Victor Stinner26486ea2010-05-15 22:23:53 +00001213 int x, y, n;
1214 char rtn[1024]; /* This should be big enough.. I hope */
1215 int rtn2;
Guido van Rossum85738471995-02-17 13:50:17 +00001216
Victor Stinner26486ea2010-05-15 22:23:53 +00001217 switch (PyTuple_Size(args)) {
1218 case 0:
1219 Py_BEGIN_ALLOW_THREADS
1220 rtn2 = wgetnstr(self->win,rtn, 1023);
1221 Py_END_ALLOW_THREADS
1222 break;
1223 case 1:
1224 if (!PyArg_ParseTuple(args,"i;n", &n))
1225 return NULL;
1226 Py_BEGIN_ALLOW_THREADS
1227 rtn2 = wgetnstr(self->win,rtn,MIN(n, 1023));
1228 Py_END_ALLOW_THREADS
1229 break;
1230 case 2:
1231 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1232 return NULL;
1233 Py_BEGIN_ALLOW_THREADS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001234#ifdef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00001235 rtn2 = wmove(self->win,y,x)==ERR ? ERR : wgetnstr(self->win, rtn, 1023);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001236#else
Victor Stinner26486ea2010-05-15 22:23:53 +00001237 rtn2 = mvwgetnstr(self->win,y,x,rtn, 1023);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001238#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001239 Py_END_ALLOW_THREADS
1240 break;
1241 case 3:
1242 if (!PyArg_ParseTuple(args,"iii;y,x,n", &y, &x, &n))
1243 return NULL;
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +00001244#ifdef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00001245 Py_BEGIN_ALLOW_THREADS
1246 rtn2 = wmove(self->win,y,x)==ERR ? ERR :
1247 wgetnstr(self->win, rtn, MIN(n, 1023));
1248 Py_END_ALLOW_THREADS
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001249#else
Victor Stinner26486ea2010-05-15 22:23:53 +00001250 Py_BEGIN_ALLOW_THREADS
1251 rtn2 = mvwgetnstr(self->win, y, x, rtn, MIN(n, 1023));
1252 Py_END_ALLOW_THREADS
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001253#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001254 break;
1255 default:
1256 PyErr_SetString(PyExc_TypeError, "getstr requires 0 to 3 arguments");
1257 return NULL;
1258 }
1259 if (rtn2 == ERR)
1260 rtn[0] = 0;
1261 return PyBytes_FromString(rtn);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001262}
1263
1264static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001265PyCursesWindow_Hline(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001266{
Victor Stinner26486ea2010-05-15 22:23:53 +00001267 PyObject *temp;
1268 chtype ch;
1269 int n, x, y, code = OK;
1270 attr_t attr = A_NORMAL;
1271 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001272
Victor Stinner26486ea2010-05-15 22:23:53 +00001273 switch (PyTuple_Size(args)) {
1274 case 2:
1275 if (!PyArg_ParseTuple(args, "Oi;ch or int,n", &temp, &n))
1276 return NULL;
1277 break;
1278 case 3:
1279 if (!PyArg_ParseTuple(args, "Oil;ch or int,n,attr", &temp, &n, &lattr))
1280 return NULL;
1281 attr = lattr;
1282 break;
1283 case 4:
1284 if (!PyArg_ParseTuple(args, "iiOi;y,x,ch or int,n", &y, &x, &temp, &n))
1285 return NULL;
1286 code = wmove(self->win, y, x);
1287 break;
1288 case 5:
1289 if (!PyArg_ParseTuple(args, "iiOil; y,x,ch or int,n,attr",
1290 &y, &x, &temp, &n, &lattr))
1291 return NULL;
1292 attr = lattr;
1293 code = wmove(self->win, y, x);
1294 break;
1295 default:
1296 PyErr_SetString(PyExc_TypeError, "hline requires 2 to 5 arguments");
1297 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001298 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001299
1300 if (code != ERR) {
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001301 if (!PyCurses_ConvertToChtype(self, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00001302 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00001303 return PyCursesCheckERR(whline(self->win, ch | attr, n), "hline");
1304 } else
1305 return PyCursesCheckERR(code, "wmove");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001306}
1307
1308static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001309PyCursesWindow_InsCh(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001310{
Victor Stinner26486ea2010-05-15 22:23:53 +00001311 int rtn, x, y, use_xy = FALSE;
1312 PyObject *temp;
1313 chtype ch = 0;
1314 attr_t attr = A_NORMAL;
1315 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001316
Victor Stinner26486ea2010-05-15 22:23:53 +00001317 switch (PyTuple_Size(args)) {
1318 case 1:
1319 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
1320 return NULL;
1321 break;
1322 case 2:
1323 if (!PyArg_ParseTuple(args, "Ol;ch or int,attr", &temp, &lattr))
1324 return NULL;
1325 attr = lattr;
1326 break;
1327 case 3:
1328 if (!PyArg_ParseTuple(args,"iiO;y,x,ch or int", &y, &x, &temp))
1329 return NULL;
1330 use_xy = TRUE;
1331 break;
1332 case 4:
1333 if (!PyArg_ParseTuple(args,"iiOl;y,x,ch or int, attr", &y, &x, &temp, &lattr))
1334 return NULL;
1335 attr = lattr;
1336 use_xy = TRUE;
1337 break;
1338 default:
1339 PyErr_SetString(PyExc_TypeError, "insch requires 1 or 4 arguments");
1340 return NULL;
1341 }
1342
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001343 if (!PyCurses_ConvertToChtype(self, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00001344 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00001345
1346 if (use_xy == TRUE)
1347 rtn = mvwinsch(self->win,y,x, ch | attr);
1348 else {
1349 rtn = winsch(self->win, ch | attr);
1350 }
1351 return PyCursesCheckERR(rtn, "insch");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001352}
1353
1354static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001355PyCursesWindow_InCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001356{
Victor Stinner26486ea2010-05-15 22:23:53 +00001357 int x, y, rtn;
Guido van Rossum85738471995-02-17 13:50:17 +00001358
Victor Stinner26486ea2010-05-15 22:23:53 +00001359 switch (PyTuple_Size(args)) {
1360 case 0:
1361 rtn = winch(self->win);
1362 break;
1363 case 2:
1364 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1365 return NULL;
1366 rtn = mvwinch(self->win,y,x);
1367 break;
1368 default:
1369 PyErr_SetString(PyExc_TypeError, "inch requires 0 or 2 arguments");
1370 return NULL;
1371 }
1372 return PyLong_FromLong((long) rtn);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001373}
1374
1375static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001376PyCursesWindow_InStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001377{
Victor Stinner26486ea2010-05-15 22:23:53 +00001378 int x, y, n;
1379 char rtn[1024]; /* This should be big enough.. I hope */
1380 int rtn2;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001381
Victor Stinner26486ea2010-05-15 22:23:53 +00001382 switch (PyTuple_Size(args)) {
1383 case 0:
1384 rtn2 = winnstr(self->win,rtn, 1023);
1385 break;
1386 case 1:
1387 if (!PyArg_ParseTuple(args,"i;n", &n))
1388 return NULL;
1389 rtn2 = winnstr(self->win,rtn,MIN(n,1023));
1390 break;
1391 case 2:
1392 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1393 return NULL;
1394 rtn2 = mvwinnstr(self->win,y,x,rtn,1023);
1395 break;
1396 case 3:
1397 if (!PyArg_ParseTuple(args, "iii;y,x,n", &y, &x, &n))
1398 return NULL;
1399 rtn2 = mvwinnstr(self->win, y, x, rtn, MIN(n,1023));
1400 break;
1401 default:
1402 PyErr_SetString(PyExc_TypeError, "instr requires 0 or 3 arguments");
1403 return NULL;
1404 }
1405 if (rtn2 == ERR)
1406 rtn[0] = 0;
1407 return PyBytes_FromString(rtn);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001408}
1409
1410static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001411PyCursesWindow_InsStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001412{
Victor Stinner26486ea2010-05-15 22:23:53 +00001413 int rtn;
1414 int x, y;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001415 int strtype;
Benjamin Peterson5e545192012-03-14 18:21:35 -05001416 PyObject *strobj, *bytesobj = NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001417#ifdef HAVE_NCURSESW
1418 wchar_t *wstr = NULL;
1419#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001420 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
1421 long lattr;
1422 int use_xy = FALSE, use_attr = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001423 const char *funcname;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001424
Victor Stinner26486ea2010-05-15 22:23:53 +00001425 switch (PyTuple_Size(args)) {
1426 case 1:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001427 if (!PyArg_ParseTuple(args,"O;str", &strobj))
Victor Stinner26486ea2010-05-15 22:23:53 +00001428 return NULL;
1429 break;
1430 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001431 if (!PyArg_ParseTuple(args,"Ol;str,attr", &strobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +00001432 return NULL;
1433 attr = lattr;
1434 use_attr = TRUE;
1435 break;
1436 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001437 if (!PyArg_ParseTuple(args,"iiO;y,x,str", &y, &x, &strobj))
Victor Stinner26486ea2010-05-15 22:23:53 +00001438 return NULL;
1439 use_xy = TRUE;
1440 break;
1441 case 4:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001442 if (!PyArg_ParseTuple(args,"iiOl;y,x,str,attr", &y, &x, &strobj, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +00001443 return NULL;
1444 attr = lattr;
1445 use_xy = use_attr = TRUE;
1446 break;
1447 default:
1448 PyErr_SetString(PyExc_TypeError, "insstr requires 1 to 4 arguments");
1449 return NULL;
1450 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001451
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001452#ifdef HAVE_NCURSESW
1453 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);
1454#else
1455 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);
1456#endif
1457 if (strtype == 0)
1458 return NULL;
1459
Victor Stinner26486ea2010-05-15 22:23:53 +00001460 if (use_attr == TRUE) {
1461 attr_old = getattrs(self->win);
1462 (void)wattrset(self->win,attr);
1463 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001464#ifdef HAVE_NCURSESW
1465 if (strtype == 2) {
1466 funcname = "inswstr";
1467 if (use_xy == TRUE)
1468 rtn = mvwins_wstr(self->win,y,x,wstr);
1469 else
1470 rtn = wins_wstr(self->win,wstr);
1471 PyMem_Free(wstr);
1472 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001473 else
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001474#endif
1475 {
1476 char *str = PyBytes_AS_STRING(bytesobj);
1477 funcname = "insstr";
1478 if (use_xy == TRUE)
1479 rtn = mvwinsstr(self->win,y,x,str);
1480 else
1481 rtn = winsstr(self->win,str);
1482 Py_DECREF(bytesobj);
1483 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001484 if (use_attr == TRUE)
1485 (void)wattrset(self->win,attr_old);
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001486 return PyCursesCheckERR(rtn, funcname);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001487}
1488
1489static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001490PyCursesWindow_InsNStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001491{
Victor Stinner26486ea2010-05-15 22:23:53 +00001492 int rtn, x, y, n;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001493 int strtype;
Benjamin Peterson5e545192012-03-14 18:21:35 -05001494 PyObject *strobj, *bytesobj = NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001495#ifdef HAVE_NCURSESW
1496 wchar_t *wstr = NULL;
1497#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001498 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
1499 long lattr;
1500 int use_xy = FALSE, use_attr = FALSE;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001501 const char *funcname;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001502
Victor Stinner26486ea2010-05-15 22:23:53 +00001503 switch (PyTuple_Size(args)) {
1504 case 2:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001505 if (!PyArg_ParseTuple(args,"Oi;str,n", &strobj, &n))
Victor Stinner26486ea2010-05-15 22:23:53 +00001506 return NULL;
1507 break;
1508 case 3:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001509 if (!PyArg_ParseTuple(args,"Oil;str,n,attr", &strobj, &n, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +00001510 return NULL;
1511 attr = lattr;
1512 use_attr = TRUE;
1513 break;
1514 case 4:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001515 if (!PyArg_ParseTuple(args,"iiOi;y,x,str,n", &y, &x, &strobj, &n))
Victor Stinner26486ea2010-05-15 22:23:53 +00001516 return NULL;
1517 use_xy = TRUE;
1518 break;
1519 case 5:
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001520 if (!PyArg_ParseTuple(args,"iiOil;y,x,str,n,attr", &y, &x, &strobj, &n, &lattr))
Victor Stinner26486ea2010-05-15 22:23:53 +00001521 return NULL;
1522 attr = lattr;
1523 use_xy = use_attr = TRUE;
1524 break;
1525 default:
1526 PyErr_SetString(PyExc_TypeError, "insnstr requires 2 to 5 arguments");
1527 return NULL;
1528 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001529
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001530#ifdef HAVE_NCURSESW
1531 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);
1532#else
1533 strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);
1534#endif
1535 if (strtype == 0)
1536 return NULL;
1537
Victor Stinner26486ea2010-05-15 22:23:53 +00001538 if (use_attr == TRUE) {
1539 attr_old = getattrs(self->win);
1540 (void)wattrset(self->win,attr);
1541 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001542#ifdef HAVE_NCURSESW
1543 if (strtype == 2) {
1544 funcname = "insn_wstr";
1545 if (use_xy == TRUE)
1546 rtn = mvwins_nwstr(self->win,y,x,wstr,n);
1547 else
1548 rtn = wins_nwstr(self->win,wstr,n);
1549 PyMem_Free(wstr);
1550 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001551 else
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001552#endif
1553 {
1554 char *str = PyBytes_AS_STRING(bytesobj);
1555 funcname = "insnstr";
1556 if (use_xy == TRUE)
1557 rtn = mvwinsnstr(self->win,y,x,str,n);
1558 else
1559 rtn = winsnstr(self->win,str,n);
1560 Py_DECREF(bytesobj);
1561 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001562 if (use_attr == TRUE)
1563 (void)wattrset(self->win,attr_old);
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001564 return PyCursesCheckERR(rtn, funcname);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001565}
1566
1567static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001568PyCursesWindow_Is_LineTouched(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001569{
Victor Stinner26486ea2010-05-15 22:23:53 +00001570 int line, erg;
1571 if (!PyArg_ParseTuple(args,"i;line", &line))
1572 return NULL;
1573 erg = is_linetouched(self->win, line);
1574 if (erg == ERR) {
1575 PyErr_SetString(PyExc_TypeError,
1576 "is_linetouched: line number outside of boundaries");
1577 return NULL;
1578 } else
1579 if (erg == FALSE) {
1580 Py_INCREF(Py_False);
1581 return Py_False;
1582 } else {
1583 Py_INCREF(Py_True);
1584 return Py_True;
1585 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001586}
1587
1588static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001589PyCursesWindow_NoOutRefresh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001590{
Victor Stinner26486ea2010-05-15 22:23:53 +00001591 int pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol;
1592 int rtn;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001593
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00001594#ifndef WINDOW_HAS_FLAGS
Victor Stinner26486ea2010-05-15 22:23:53 +00001595 if (0)
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001596#else
Victor Stinner26486ea2010-05-15 22:23:53 +00001597 if (self->win->_flags & _ISPAD)
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001598#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001599 {
1600 switch(PyTuple_Size(args)) {
1601 case 6:
1602 if (!PyArg_ParseTuple(args,
1603 "iiiiii;" \
1604 "pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol",
1605 &pminrow, &pmincol, &sminrow,
1606 &smincol, &smaxrow, &smaxcol))
1607 return NULL;
1608 Py_BEGIN_ALLOW_THREADS
1609 rtn = pnoutrefresh(self->win,
1610 pminrow, pmincol, sminrow,
1611 smincol, smaxrow, smaxcol);
1612 Py_END_ALLOW_THREADS
1613 return PyCursesCheckERR(rtn, "pnoutrefresh");
1614 default:
1615 PyErr_SetString(PyCursesError,
1616 "noutrefresh() called for a pad "
1617 "requires 6 arguments");
1618 return NULL;
1619 }
1620 } else {
1621 if (!PyArg_ParseTuple(args, ":noutrefresh"))
1622 return NULL;
Andrew M. Kuchlingf4195722000-06-23 01:36:21 +00001623
Victor Stinner26486ea2010-05-15 22:23:53 +00001624 Py_BEGIN_ALLOW_THREADS
1625 rtn = wnoutrefresh(self->win);
1626 Py_END_ALLOW_THREADS
1627 return PyCursesCheckERR(rtn, "wnoutrefresh");
1628 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001629}
1630
1631static PyObject *
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001632PyCursesWindow_Overlay(PyCursesWindowObject *self, PyObject *args)
1633{
1634 PyCursesWindowObject *temp;
1635 int use_copywin = FALSE;
1636 int sminrow, smincol, dminrow, dmincol, dmaxrow, dmaxcol;
1637 int rtn;
Victor Stinner26486ea2010-05-15 22:23:53 +00001638
Martin v. Löwisc0e16712002-01-17 23:08:27 +00001639 switch (PyTuple_Size(args)) {
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001640 case 1:
Victor Stinner26486ea2010-05-15 22:23:53 +00001641 if (!PyArg_ParseTuple(args, "O!;window object",
1642 &PyCursesWindow_Type, &temp))
1643 return NULL;
1644 break;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001645 case 7:
Victor Stinner26486ea2010-05-15 22:23:53 +00001646 if (!PyArg_ParseTuple(args, "O!iiiiii;window object, int, int, int, int, int, int",
1647 &PyCursesWindow_Type, &temp, &sminrow, &smincol,
1648 &dminrow, &dmincol, &dmaxrow, &dmaxcol))
1649 return NULL;
1650 use_copywin = TRUE;
1651 break;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001652 default:
Victor Stinner26486ea2010-05-15 22:23:53 +00001653 PyErr_SetString(PyExc_TypeError,
1654 "overlay requires one or seven arguments");
1655 return NULL;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001656 }
1657
1658 if (use_copywin == TRUE) {
Victor Stinner26486ea2010-05-15 22:23:53 +00001659 rtn = copywin(self->win, temp->win, sminrow, smincol,
1660 dminrow, dmincol, dmaxrow, dmaxcol, TRUE);
1661 return PyCursesCheckERR(rtn, "copywin");
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001662 }
1663 else {
Victor Stinner26486ea2010-05-15 22:23:53 +00001664 rtn = overlay(self->win, temp->win);
1665 return PyCursesCheckERR(rtn, "overlay");
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001666 }
1667}
1668
1669static PyObject *
1670PyCursesWindow_Overwrite(PyCursesWindowObject *self, PyObject *args)
1671{
1672 PyCursesWindowObject *temp;
1673 int use_copywin = FALSE;
1674 int sminrow, smincol, dminrow, dmincol, dmaxrow, dmaxcol;
1675 int rtn;
Victor Stinner26486ea2010-05-15 22:23:53 +00001676
Martin v. Löwisc0e16712002-01-17 23:08:27 +00001677 switch (PyTuple_Size(args)) {
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001678 case 1:
Victor Stinner26486ea2010-05-15 22:23:53 +00001679 if (!PyArg_ParseTuple(args, "O!;window object",
1680 &PyCursesWindow_Type, &temp))
1681 return NULL;
1682 break;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001683 case 7:
Victor Stinner26486ea2010-05-15 22:23:53 +00001684 if (!PyArg_ParseTuple(args, "O!iiiiii;window object, int, int, int, int, int, int",
1685 &PyCursesWindow_Type, &temp, &sminrow, &smincol,
1686 &dminrow, &dmincol, &dmaxrow, &dmaxcol))
1687 return NULL;
1688 use_copywin = TRUE;
1689 break;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001690 default:
Victor Stinner26486ea2010-05-15 22:23:53 +00001691 PyErr_SetString(PyExc_TypeError,
1692 "overwrite requires one or seven arguments");
1693 return NULL;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001694 }
1695
1696 if (use_copywin == TRUE) {
Victor Stinner26486ea2010-05-15 22:23:53 +00001697 rtn = copywin(self->win, temp->win, sminrow, smincol,
1698 dminrow, dmincol, dmaxrow, dmaxcol, FALSE);
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001699 return PyCursesCheckERR(rtn, "copywin");
1700 }
1701 else {
Victor Stinner26486ea2010-05-15 22:23:53 +00001702 rtn = overwrite(self->win, temp->win);
1703 return PyCursesCheckERR(rtn, "overwrite");
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001704 }
1705}
1706
1707static PyObject *
Guido van Rossum6c95da32007-07-24 00:16:38 +00001708PyCursesWindow_PutWin(PyCursesWindowObject *self, PyObject *stream)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001709{
Victor Stinner26486ea2010-05-15 22:23:53 +00001710 /* We have to simulate this by writing to a temporary FILE*,
1711 then reading back, then writing to the argument stream. */
1712 char fn[100];
1713 int fd;
1714 FILE *fp;
1715 PyObject *res;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00001716
Victor Stinner26486ea2010-05-15 22:23:53 +00001717 strcpy(fn, "/tmp/py.curses.putwin.XXXXXX");
1718 fd = mkstemp(fn);
1719 if (fd < 0)
1720 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
1721 fp = fdopen(fd, "wb+");
1722 if (fp == NULL) {
1723 close(fd);
1724 remove(fn);
1725 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
1726 }
1727 res = PyCursesCheckERR(putwin(self->win, fp), "putwin");
1728 if (res == NULL) {
1729 fclose(fp);
1730 remove(fn);
1731 return res;
1732 }
1733 fseek(fp, 0, 0);
1734 while (1) {
1735 char buf[BUFSIZ];
1736 Py_ssize_t n = fread(buf, 1, BUFSIZ, fp);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02001737 _Py_IDENTIFIER(write);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001738
Victor Stinner26486ea2010-05-15 22:23:53 +00001739 if (n <= 0)
1740 break;
1741 Py_DECREF(res);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001742 res = _PyObject_CallMethodId(stream, &PyId_write, "y#", buf, n);
Victor Stinner26486ea2010-05-15 22:23:53 +00001743 if (res == NULL)
1744 break;
1745 }
Guido van Rossum6c95da32007-07-24 00:16:38 +00001746 fclose(fp);
Guido van Rossum150b7d72007-08-30 23:34:01 +00001747 remove(fn);
Guido van Rossum6c95da32007-07-24 00:16:38 +00001748 return res;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001749}
1750
1751static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001752PyCursesWindow_RedrawLine(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001753{
Victor Stinner26486ea2010-05-15 22:23:53 +00001754 int beg, num;
1755 if (!PyArg_ParseTuple(args, "ii;beg,num", &beg, &num))
1756 return NULL;
1757 return PyCursesCheckERR(wredrawln(self->win,beg,num), "redrawln");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001758}
1759
1760static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001761PyCursesWindow_Refresh(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001762{
Victor Stinner26486ea2010-05-15 22:23:53 +00001763 int pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol;
1764 int rtn;
Andrew M. Kuchlingf4195722000-06-23 01:36:21 +00001765
Victor Stinner26486ea2010-05-15 22:23:53 +00001766#ifndef WINDOW_HAS_FLAGS
1767 if (0)
1768#else
1769 if (self->win->_flags & _ISPAD)
1770#endif
1771 {
1772 switch(PyTuple_Size(args)) {
1773 case 6:
1774 if (!PyArg_ParseTuple(args,
1775 "iiiiii;" \
1776 "pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol",
1777 &pminrow, &pmincol, &sminrow,
1778 &smincol, &smaxrow, &smaxcol))
1779 return NULL;
1780
1781 Py_BEGIN_ALLOW_THREADS
1782 rtn = prefresh(self->win,
1783 pminrow, pmincol, sminrow,
1784 smincol, smaxrow, smaxcol);
1785 Py_END_ALLOW_THREADS
1786 return PyCursesCheckERR(rtn, "prefresh");
1787 default:
1788 PyErr_SetString(PyCursesError,
1789 "refresh() for a pad requires 6 arguments");
1790 return NULL;
1791 }
1792 } else {
1793 if (!PyArg_ParseTuple(args, ":refresh"))
1794 return NULL;
1795 Py_BEGIN_ALLOW_THREADS
1796 rtn = wrefresh(self->win);
1797 Py_END_ALLOW_THREADS
1798 return PyCursesCheckERR(rtn, "prefresh");
1799 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001800}
1801
1802static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001803PyCursesWindow_SetScrollRegion(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001804{
Victor Stinner26486ea2010-05-15 22:23:53 +00001805 int x, y;
1806 if (!PyArg_ParseTuple(args,"ii;top, bottom",&y,&x))
1807 return NULL;
1808 return PyCursesCheckERR(wsetscrreg(self->win,y,x), "wsetscrreg");
Guido van Rossumf6971e21994-08-30 12:25:20 +00001809}
1810
1811static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001812PyCursesWindow_SubWin(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001813{
Victor Stinner26486ea2010-05-15 22:23:53 +00001814 WINDOW *win;
1815 int nlines, ncols, begin_y, begin_x;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001816
Victor Stinner26486ea2010-05-15 22:23:53 +00001817 nlines = 0;
1818 ncols = 0;
1819 switch (PyTuple_Size(args)) {
1820 case 2:
1821 if (!PyArg_ParseTuple(args,"ii;begin_y,begin_x",&begin_y,&begin_x))
1822 return NULL;
1823 break;
1824 case 4:
1825 if (!PyArg_ParseTuple(args, "iiii;nlines,ncols,begin_y,begin_x",
1826 &nlines,&ncols,&begin_y,&begin_x))
1827 return NULL;
1828 break;
1829 default:
1830 PyErr_SetString(PyExc_TypeError, "subwin requires 2 or 4 arguments");
1831 return NULL;
1832 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001833
Victor Stinner26486ea2010-05-15 22:23:53 +00001834 /* printf("Subwin: %i %i %i %i \n", nlines, ncols, begin_y, begin_x); */
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00001835#ifdef WINDOW_HAS_FLAGS
Victor Stinner26486ea2010-05-15 22:23:53 +00001836 if (self->win->_flags & _ISPAD)
1837 win = subpad(self->win, nlines, ncols, begin_y, begin_x);
1838 else
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001839#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001840 win = subwin(self->win, nlines, ncols, begin_y, begin_x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001841
Victor Stinner26486ea2010-05-15 22:23:53 +00001842 if (win == NULL) {
1843 PyErr_SetString(PyCursesError, catchall_NULL);
1844 return NULL;
1845 }
1846
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001847 return (PyObject *)PyCursesWindow_New(win, self->encoding);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001848}
1849
1850static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001851PyCursesWindow_Scroll(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001852{
Victor Stinner26486ea2010-05-15 22:23:53 +00001853 int nlines;
1854 switch(PyTuple_Size(args)) {
1855 case 0:
1856 return PyCursesCheckERR(scroll(self->win), "scroll");
1857 case 1:
1858 if (!PyArg_ParseTuple(args, "i;nlines", &nlines))
1859 return NULL;
1860 return PyCursesCheckERR(wscrl(self->win, nlines), "scroll");
1861 default:
1862 PyErr_SetString(PyExc_TypeError, "scroll requires 0 or 1 arguments");
1863 return NULL;
1864 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001865}
1866
1867static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001868PyCursesWindow_TouchLine(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001869{
Victor Stinner26486ea2010-05-15 22:23:53 +00001870 int st, cnt, val;
1871 switch (PyTuple_Size(args)) {
1872 case 2:
1873 if (!PyArg_ParseTuple(args,"ii;start,count",&st,&cnt))
1874 return NULL;
1875 return PyCursesCheckERR(touchline(self->win,st,cnt), "touchline");
1876 case 3:
1877 if (!PyArg_ParseTuple(args, "iii;start,count,val", &st, &cnt, &val))
1878 return NULL;
1879 return PyCursesCheckERR(wtouchln(self->win, st, cnt, val), "touchline");
1880 default:
1881 PyErr_SetString(PyExc_TypeError, "touchline requires 2 or 3 arguments");
1882 return NULL;
1883 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001884}
1885
1886static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001887PyCursesWindow_Vline(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001888{
Victor Stinner26486ea2010-05-15 22:23:53 +00001889 PyObject *temp;
1890 chtype ch;
1891 int n, x, y, code = OK;
1892 attr_t attr = A_NORMAL;
1893 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001894
Victor Stinner26486ea2010-05-15 22:23:53 +00001895 switch (PyTuple_Size(args)) {
1896 case 2:
1897 if (!PyArg_ParseTuple(args, "Oi;ch or int,n", &temp, &n))
1898 return NULL;
1899 break;
1900 case 3:
1901 if (!PyArg_ParseTuple(args, "Oil;ch or int,n,attr", &temp, &n, &lattr))
1902 return NULL;
1903 attr = lattr;
1904 break;
1905 case 4:
1906 if (!PyArg_ParseTuple(args, "iiOi;y,x,ch or int,n", &y, &x, &temp, &n))
1907 return NULL;
1908 code = wmove(self->win, y, x);
1909 break;
1910 case 5:
1911 if (!PyArg_ParseTuple(args, "iiOil; y,x,ch or int,n,attr",
1912 &y, &x, &temp, &n, &lattr))
1913 return NULL;
1914 attr = lattr;
1915 code = wmove(self->win, y, x);
1916 break;
1917 default:
1918 PyErr_SetString(PyExc_TypeError, "vline requires 2 to 5 arguments");
1919 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001920 }
Victor Stinner26486ea2010-05-15 22:23:53 +00001921
1922 if (code != ERR) {
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001923 if (!PyCurses_ConvertToChtype(self, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00001924 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00001925 return PyCursesCheckERR(wvline(self->win, ch | attr, n), "vline");
1926 } else
1927 return PyCursesCheckERR(code, "wmove");
Guido van Rossumf6971e21994-08-30 12:25:20 +00001928}
1929
Victor Stinner0fdfceb2011-11-25 22:10:02 +01001930static PyObject *
1931PyCursesWindow_get_encoding(PyCursesWindowObject *self, void *closure)
1932{
1933 return PyUnicode_FromString(self->encoding);
1934}
1935
1936static int
1937PyCursesWindow_set_encoding(PyCursesWindowObject *self, PyObject *value)
1938{
1939 PyObject *ascii;
1940 char *encoding;
1941
1942 /* It is illegal to del win.encoding */
1943 if (value == NULL) {
1944 PyErr_SetString(PyExc_TypeError,
1945 "encoding may not be deleted");
1946 return -1;
1947 }
1948
1949 if (!PyUnicode_Check(value)) {
1950 PyErr_SetString(PyExc_TypeError,
1951 "setting encoding to a non-string");
1952 return -1;
1953 }
1954 ascii = PyUnicode_AsASCIIString(value);
1955 if (ascii == NULL)
1956 return -1;
1957 encoding = strdup(PyBytes_AS_STRING(ascii));
1958 if (encoding == NULL) {
1959 PyErr_NoMemory();
1960 return -1;
1961 }
1962 free(self->encoding);
1963 self->encoding = encoding;
1964 return 0;
1965}
1966
1967
Guido van Rossumf6971e21994-08-30 12:25:20 +00001968static PyMethodDef PyCursesWindow_Methods[] = {
Victor Stinner26486ea2010-05-15 22:23:53 +00001969 {"addch", (PyCFunction)PyCursesWindow_AddCh, METH_VARARGS},
1970 {"addnstr", (PyCFunction)PyCursesWindow_AddNStr, METH_VARARGS},
1971 {"addstr", (PyCFunction)PyCursesWindow_AddStr, METH_VARARGS},
1972 {"attroff", (PyCFunction)PyCursesWindow_AttrOff, METH_VARARGS},
1973 {"attron", (PyCFunction)PyCursesWindow_AttrOn, METH_VARARGS},
1974 {"attrset", (PyCFunction)PyCursesWindow_AttrSet, METH_VARARGS},
1975 {"bkgd", (PyCFunction)PyCursesWindow_Bkgd, METH_VARARGS},
1976 {"chgat", (PyCFunction)PyCursesWindow_ChgAt, METH_VARARGS},
1977 {"bkgdset", (PyCFunction)PyCursesWindow_BkgdSet, METH_VARARGS},
1978 {"border", (PyCFunction)PyCursesWindow_Border, METH_VARARGS},
1979 {"box", (PyCFunction)PyCursesWindow_Box, METH_VARARGS},
1980 {"clear", (PyCFunction)PyCursesWindow_wclear, METH_NOARGS},
1981 {"clearok", (PyCFunction)PyCursesWindow_clearok, METH_VARARGS},
1982 {"clrtobot", (PyCFunction)PyCursesWindow_wclrtobot, METH_NOARGS},
1983 {"clrtoeol", (PyCFunction)PyCursesWindow_wclrtoeol, METH_NOARGS},
1984 {"cursyncup", (PyCFunction)PyCursesWindow_wcursyncup, METH_NOARGS},
1985 {"delch", (PyCFunction)PyCursesWindow_DelCh, METH_VARARGS},
1986 {"deleteln", (PyCFunction)PyCursesWindow_wdeleteln, METH_NOARGS},
1987 {"derwin", (PyCFunction)PyCursesWindow_DerWin, METH_VARARGS},
1988 {"echochar", (PyCFunction)PyCursesWindow_EchoChar, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001989#ifdef NCURSES_MOUSE_VERSION
Victor Stinner26486ea2010-05-15 22:23:53 +00001990 {"enclose", (PyCFunction)PyCursesWindow_Enclose, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001991#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00001992 {"erase", (PyCFunction)PyCursesWindow_werase, METH_NOARGS},
1993 {"getbegyx", (PyCFunction)PyCursesWindow_getbegyx, METH_NOARGS},
1994 {"getbkgd", (PyCFunction)PyCursesWindow_GetBkgd, METH_NOARGS},
1995 {"getch", (PyCFunction)PyCursesWindow_GetCh, METH_VARARGS},
1996 {"getkey", (PyCFunction)PyCursesWindow_GetKey, METH_VARARGS},
Nadeem Vawda9e2e9902011-07-31 15:01:11 +02001997#ifdef HAVE_NCURSESW
Victor Stinnera7878b72011-07-14 23:07:44 +02001998 {"get_wch", (PyCFunction)PyCursesWindow_Get_WCh, METH_VARARGS},
Nadeem Vawda9e2e9902011-07-31 15:01:11 +02001999#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00002000 {"getmaxyx", (PyCFunction)PyCursesWindow_getmaxyx, METH_NOARGS},
2001 {"getparyx", (PyCFunction)PyCursesWindow_getparyx, METH_NOARGS},
2002 {"getstr", (PyCFunction)PyCursesWindow_GetStr, METH_VARARGS},
2003 {"getyx", (PyCFunction)PyCursesWindow_getyx, METH_NOARGS},
2004 {"hline", (PyCFunction)PyCursesWindow_Hline, METH_VARARGS},
2005 {"idcok", (PyCFunction)PyCursesWindow_idcok, METH_VARARGS},
2006 {"idlok", (PyCFunction)PyCursesWindow_idlok, METH_VARARGS},
2007 {"immedok", (PyCFunction)PyCursesWindow_immedok, METH_VARARGS},
2008 {"inch", (PyCFunction)PyCursesWindow_InCh, METH_VARARGS},
2009 {"insch", (PyCFunction)PyCursesWindow_InsCh, METH_VARARGS},
2010 {"insdelln", (PyCFunction)PyCursesWindow_winsdelln, METH_VARARGS},
2011 {"insertln", (PyCFunction)PyCursesWindow_winsertln, METH_NOARGS},
2012 {"insnstr", (PyCFunction)PyCursesWindow_InsNStr, METH_VARARGS},
2013 {"insstr", (PyCFunction)PyCursesWindow_InsStr, METH_VARARGS},
2014 {"instr", (PyCFunction)PyCursesWindow_InStr, METH_VARARGS},
2015 {"is_linetouched", (PyCFunction)PyCursesWindow_Is_LineTouched, METH_VARARGS},
2016 {"is_wintouched", (PyCFunction)PyCursesWindow_is_wintouched, METH_NOARGS},
2017 {"keypad", (PyCFunction)PyCursesWindow_keypad, METH_VARARGS},
2018 {"leaveok", (PyCFunction)PyCursesWindow_leaveok, METH_VARARGS},
2019 {"move", (PyCFunction)PyCursesWindow_wmove, METH_VARARGS},
2020 {"mvderwin", (PyCFunction)PyCursesWindow_mvderwin, METH_VARARGS},
2021 {"mvwin", (PyCFunction)PyCursesWindow_mvwin, METH_VARARGS},
2022 {"nodelay", (PyCFunction)PyCursesWindow_nodelay, METH_VARARGS},
2023 {"notimeout", (PyCFunction)PyCursesWindow_notimeout, METH_VARARGS},
2024 {"noutrefresh", (PyCFunction)PyCursesWindow_NoOutRefresh, METH_VARARGS},
2025 /* Backward compatibility alias -- remove in Python 2.3 */
2026 {"nooutrefresh", (PyCFunction)PyCursesWindow_NoOutRefresh, METH_VARARGS},
2027 {"overlay", (PyCFunction)PyCursesWindow_Overlay, METH_VARARGS},
2028 {"overwrite", (PyCFunction)PyCursesWindow_Overwrite,
2029 METH_VARARGS},
2030 {"putwin", (PyCFunction)PyCursesWindow_PutWin, METH_O},
2031 {"redrawln", (PyCFunction)PyCursesWindow_RedrawLine, METH_VARARGS},
2032 {"redrawwin", (PyCFunction)PyCursesWindow_redrawwin, METH_NOARGS},
2033 {"refresh", (PyCFunction)PyCursesWindow_Refresh, METH_VARARGS},
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +00002034#ifndef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00002035 {"resize", (PyCFunction)PyCursesWindow_wresize, METH_VARARGS},
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002036#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00002037 {"scroll", (PyCFunction)PyCursesWindow_Scroll, METH_VARARGS},
2038 {"scrollok", (PyCFunction)PyCursesWindow_scrollok, METH_VARARGS},
2039 {"setscrreg", (PyCFunction)PyCursesWindow_SetScrollRegion, METH_VARARGS},
2040 {"standend", (PyCFunction)PyCursesWindow_wstandend, METH_NOARGS},
2041 {"standout", (PyCFunction)PyCursesWindow_wstandout, METH_NOARGS},
2042 {"subpad", (PyCFunction)PyCursesWindow_SubWin, METH_VARARGS},
2043 {"subwin", (PyCFunction)PyCursesWindow_SubWin, METH_VARARGS},
2044 {"syncdown", (PyCFunction)PyCursesWindow_wsyncdown, METH_NOARGS},
2045 {"syncok", (PyCFunction)PyCursesWindow_syncok, METH_VARARGS},
2046 {"syncup", (PyCFunction)PyCursesWindow_wsyncup, METH_NOARGS},
2047 {"timeout", (PyCFunction)PyCursesWindow_wtimeout, METH_VARARGS},
2048 {"touchline", (PyCFunction)PyCursesWindow_TouchLine, METH_VARARGS},
2049 {"touchwin", (PyCFunction)PyCursesWindow_touchwin, METH_NOARGS},
2050 {"untouchwin", (PyCFunction)PyCursesWindow_untouchwin, METH_NOARGS},
2051 {"vline", (PyCFunction)PyCursesWindow_Vline, METH_VARARGS},
2052 {NULL, NULL} /* sentinel */
Guido van Rossumf6971e21994-08-30 12:25:20 +00002053};
2054
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002055static PyGetSetDef PyCursesWindow_getsets[] = {
2056 {"encoding",
2057 (getter)PyCursesWindow_get_encoding,
2058 (setter)PyCursesWindow_set_encoding,
Ronald Oussoren2a7fe032012-01-17 16:53:20 +01002059 "the typecode character used to create the array"},
2060 {NULL, NULL, NULL, NULL } /* sentinel */
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002061};
2062
Guido van Rossumf6971e21994-08-30 12:25:20 +00002063/* -------------------------------------------------------*/
Guido van Rossum85738471995-02-17 13:50:17 +00002064
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002065PyTypeObject PyCursesWindow_Type = {
Victor Stinner26486ea2010-05-15 22:23:53 +00002066 PyVarObject_HEAD_INIT(NULL, 0)
2067 "_curses.curses window", /*tp_name*/
2068 sizeof(PyCursesWindowObject), /*tp_basicsize*/
2069 0, /*tp_itemsize*/
2070 /* methods */
2071 (destructor)PyCursesWindow_Dealloc, /*tp_dealloc*/
2072 0, /*tp_print*/
2073 (getattrfunc)0, /*tp_getattr*/
2074 (setattrfunc)0, /*tp_setattr*/
2075 0, /*tp_reserved*/
2076 0, /*tp_repr*/
2077 0, /*tp_as_number*/
2078 0, /*tp_as_sequence*/
2079 0, /*tp_as_mapping*/
2080 0, /*tp_hash*/
2081 0, /*tp_call*/
2082 0, /*tp_str*/
2083 0, /*tp_getattro*/
2084 0, /*tp_setattro*/
2085 0, /*tp_as_buffer*/
2086 Py_TPFLAGS_DEFAULT, /*tp_flags*/
2087 0, /*tp_doc*/
2088 0, /*tp_traverse*/
2089 0, /*tp_clear*/
2090 0, /*tp_richcompare*/
2091 0, /*tp_weaklistoffset*/
2092 0, /*tp_iter*/
2093 0, /*tp_iternext*/
2094 PyCursesWindow_Methods, /*tp_methods*/
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002095 0, /* tp_members */
2096 PyCursesWindow_getsets, /* tp_getset */
Guido van Rossumf6971e21994-08-30 12:25:20 +00002097};
2098
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002099/*********************************************************************
2100 Global Functions
2101**********************************************************************/
Guido van Rossumf6971e21994-08-30 12:25:20 +00002102
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002103NoArgNoReturnFunction(beep)
2104NoArgNoReturnFunction(def_prog_mode)
2105NoArgNoReturnFunction(def_shell_mode)
2106NoArgNoReturnFunction(doupdate)
2107NoArgNoReturnFunction(endwin)
2108NoArgNoReturnFunction(flash)
2109NoArgNoReturnFunction(nocbreak)
2110NoArgNoReturnFunction(noecho)
2111NoArgNoReturnFunction(nonl)
2112NoArgNoReturnFunction(noraw)
2113NoArgNoReturnFunction(reset_prog_mode)
2114NoArgNoReturnFunction(reset_shell_mode)
2115NoArgNoReturnFunction(resetty)
2116NoArgNoReturnFunction(savetty)
2117
2118NoArgOrFlagNoReturnFunction(cbreak)
2119NoArgOrFlagNoReturnFunction(echo)
2120NoArgOrFlagNoReturnFunction(nl)
2121NoArgOrFlagNoReturnFunction(raw)
2122
2123NoArgReturnIntFunction(baudrate)
2124NoArgReturnIntFunction(termattrs)
2125
2126NoArgReturnStringFunction(termname)
2127NoArgReturnStringFunction(longname)
2128
2129NoArgTrueFalseFunction(can_change_color)
2130NoArgTrueFalseFunction(has_colors)
2131NoArgTrueFalseFunction(has_ic)
2132NoArgTrueFalseFunction(has_il)
2133NoArgTrueFalseFunction(isendwin)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002134NoArgNoReturnVoidFunction(flushinp)
2135NoArgNoReturnVoidFunction(noqiflush)
2136
2137static PyObject *
Christian Heimesaf98da12008-01-27 15:18:18 +00002138PyCurses_filter(PyObject *self)
2139{
Victor Stinner26486ea2010-05-15 22:23:53 +00002140 /* not checking for PyCursesInitialised here since filter() must
2141 be called before initscr() */
2142 filter();
2143 Py_INCREF(Py_None);
2144 return Py_None;
Christian Heimesaf98da12008-01-27 15:18:18 +00002145}
2146
2147static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002148PyCurses_Color_Content(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002149{
Victor Stinner26486ea2010-05-15 22:23:53 +00002150 short color,r,g,b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002151
Victor Stinner26486ea2010-05-15 22:23:53 +00002152 PyCursesInitialised;
2153 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002154
Victor Stinner26486ea2010-05-15 22:23:53 +00002155 if (!PyArg_ParseTuple(args, "h:color_content", &color)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002156
Victor Stinner26486ea2010-05-15 22:23:53 +00002157 if (color_content(color, &r, &g, &b) != ERR)
2158 return Py_BuildValue("(iii)", r, g, b);
2159 else {
2160 PyErr_SetString(PyCursesError,
2161 "Argument 1 was out of range. Check value of COLORS.");
2162 return NULL;
2163 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002164}
2165
2166static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002167PyCurses_color_pair(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002168{
Victor Stinner26486ea2010-05-15 22:23:53 +00002169 int n;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002170
Victor Stinner26486ea2010-05-15 22:23:53 +00002171 PyCursesInitialised;
2172 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002173
Victor Stinner26486ea2010-05-15 22:23:53 +00002174 if (!PyArg_ParseTuple(args, "i:color_pair", &n)) return NULL;
2175 return PyLong_FromLong((long) (n << 8));
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002176}
2177
2178static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002179PyCurses_Curs_Set(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002180{
Victor Stinner26486ea2010-05-15 22:23:53 +00002181 int vis,erg;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002182
Victor Stinner26486ea2010-05-15 22:23:53 +00002183 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002184
Victor Stinner26486ea2010-05-15 22:23:53 +00002185 if (!PyArg_ParseTuple(args, "i:curs_set", &vis)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002186
Victor Stinner26486ea2010-05-15 22:23:53 +00002187 erg = curs_set(vis);
2188 if (erg == ERR) return PyCursesCheckERR(erg, "curs_set");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002189
Victor Stinner26486ea2010-05-15 22:23:53 +00002190 return PyLong_FromLong((long) erg);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002191}
2192
2193static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002194PyCurses_Delay_Output(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002195{
Victor Stinner26486ea2010-05-15 22:23:53 +00002196 int ms;
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 if (!PyArg_ParseTuple(args, "i:delay_output", &ms)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002201
Victor Stinner26486ea2010-05-15 22:23:53 +00002202 return PyCursesCheckERR(delay_output(ms), "delay_output");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002203}
2204
2205static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002206PyCurses_EraseChar(PyObject *self)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002207{
Victor Stinner26486ea2010-05-15 22:23:53 +00002208 char ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002209
Victor Stinner26486ea2010-05-15 22:23:53 +00002210 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002211
Victor Stinner26486ea2010-05-15 22:23:53 +00002212 ch = erasechar();
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002213
Victor Stinner26486ea2010-05-15 22:23:53 +00002214 return PyBytes_FromStringAndSize(&ch, 1);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002215}
2216
2217static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002218PyCurses_getsyx(PyObject *self)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002219{
Victor Stinner26486ea2010-05-15 22:23:53 +00002220 int x = 0;
2221 int y = 0;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002222
Victor Stinner26486ea2010-05-15 22:23:53 +00002223 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002224
Victor Stinner26486ea2010-05-15 22:23:53 +00002225 getsyx(y, x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002226
Victor Stinner26486ea2010-05-15 22:23:53 +00002227 return Py_BuildValue("(ii)", y, x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002228}
2229
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002230#ifdef NCURSES_MOUSE_VERSION
2231static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002232PyCurses_GetMouse(PyObject *self)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002233{
Victor Stinner26486ea2010-05-15 22:23:53 +00002234 int rtn;
2235 MEVENT event;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002236
Victor Stinner26486ea2010-05-15 22:23:53 +00002237 PyCursesInitialised;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002238
Victor Stinner26486ea2010-05-15 22:23:53 +00002239 rtn = getmouse( &event );
2240 if (rtn == ERR) {
2241 PyErr_SetString(PyCursesError, "getmouse() returned ERR");
2242 return NULL;
2243 }
2244 return Py_BuildValue("(hiiil)",
2245 (short)event.id,
2246 event.x, event.y, event.z,
2247 (long) event.bstate);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002248}
2249
2250static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002251PyCurses_UngetMouse(PyObject *self, PyObject *args)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002252{
Victor Stinner26486ea2010-05-15 22:23:53 +00002253 MEVENT event;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002254
Victor Stinner26486ea2010-05-15 22:23:53 +00002255 PyCursesInitialised;
2256 if (!PyArg_ParseTuple(args, "hiiil",
2257 &event.id,
2258 &event.x, &event.y, &event.z,
2259 (int *) &event.bstate))
2260 return NULL;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002261
Victor Stinner26486ea2010-05-15 22:23:53 +00002262 return PyCursesCheckERR(ungetmouse(&event), "ungetmouse");
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002263}
2264#endif
2265
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002266static PyObject *
Guido van Rossum6c95da32007-07-24 00:16:38 +00002267PyCurses_GetWin(PyCursesWindowObject *self, PyObject *stream)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002268{
Victor Stinner26486ea2010-05-15 22:23:53 +00002269 char fn[100];
2270 int fd;
2271 FILE *fp;
2272 PyObject *data;
2273 size_t datalen;
2274 WINDOW *win;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002275
Victor Stinner26486ea2010-05-15 22:23:53 +00002276 PyCursesInitialised;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02002277 _Py_IDENTIFIER(read);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002278
Victor Stinner26486ea2010-05-15 22:23:53 +00002279 strcpy(fn, "/tmp/py.curses.getwin.XXXXXX");
2280 fd = mkstemp(fn);
2281 if (fd < 0)
2282 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
2283 fp = fdopen(fd, "wb+");
2284 if (fp == NULL) {
2285 close(fd);
2286 remove(fn);
2287 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
2288 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002289 data = _PyObject_CallMethodId(stream, &PyId_read, "");
Victor Stinner26486ea2010-05-15 22:23:53 +00002290 if (data == NULL) {
2291 fclose(fp);
2292 remove(fn);
2293 return NULL;
2294 }
2295 if (!PyBytes_Check(data)) {
2296 PyErr_Format(PyExc_TypeError,
2297 "f.read() returned %.100s instead of bytes",
2298 data->ob_type->tp_name);
2299 Py_DECREF(data);
2300 fclose(fp);
2301 remove(fn);
2302 return NULL;
2303 }
2304 datalen = PyBytes_GET_SIZE(data);
2305 if (fwrite(PyBytes_AS_STRING(data), 1, datalen, fp) != datalen) {
2306 Py_DECREF(data);
2307 fclose(fp);
2308 remove(fn);
2309 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
2310 }
Guido van Rossum6c95da32007-07-24 00:16:38 +00002311 Py_DECREF(data);
Victor Stinner26486ea2010-05-15 22:23:53 +00002312 fseek(fp, 0, 0);
2313 win = getwin(fp);
Guido van Rossum6c95da32007-07-24 00:16:38 +00002314 fclose(fp);
Guido van Rossum150b7d72007-08-30 23:34:01 +00002315 remove(fn);
Victor Stinner26486ea2010-05-15 22:23:53 +00002316 if (win == NULL) {
2317 PyErr_SetString(PyCursesError, catchall_NULL);
2318 return NULL;
2319 }
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002320 return PyCursesWindow_New(win, NULL);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002321}
2322
2323static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002324PyCurses_HalfDelay(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002325{
Victor Stinner26486ea2010-05-15 22:23:53 +00002326 unsigned char tenths;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002327
Victor Stinner26486ea2010-05-15 22:23:53 +00002328 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002329
Victor Stinner26486ea2010-05-15 22:23:53 +00002330 if (!PyArg_ParseTuple(args, "b:halfdelay", &tenths)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002331
Victor Stinner26486ea2010-05-15 22:23:53 +00002332 return PyCursesCheckERR(halfdelay(tenths), "halfdelay");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002333}
2334
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +00002335#ifndef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00002336/* No has_key! */
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002337static PyObject * PyCurses_has_key(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002338{
Victor Stinner26486ea2010-05-15 22:23:53 +00002339 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002340
Victor Stinner26486ea2010-05-15 22:23:53 +00002341 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002342
Victor Stinner26486ea2010-05-15 22:23:53 +00002343 if (!PyArg_ParseTuple(args,"i",&ch)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002344
Victor Stinner26486ea2010-05-15 22:23:53 +00002345 if (has_key(ch) == FALSE) {
2346 Py_INCREF(Py_False);
2347 return Py_False;
2348 }
2349 Py_INCREF(Py_True);
2350 return Py_True;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002351}
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +00002352#endif /* STRICT_SYSV_CURSES */
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002353
2354static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002355PyCurses_Init_Color(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002356{
Victor Stinner26486ea2010-05-15 22:23:53 +00002357 short color, r, g, b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002358
Victor Stinner26486ea2010-05-15 22:23:53 +00002359 PyCursesInitialised;
2360 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002361
Victor Stinner26486ea2010-05-15 22:23:53 +00002362 switch(PyTuple_Size(args)) {
2363 case 4:
2364 if (!PyArg_ParseTuple(args, "hhhh;color,r,g,b", &color, &r, &g, &b)) return NULL;
2365 break;
2366 default:
2367 PyErr_SetString(PyExc_TypeError, "init_color requires 4 arguments");
2368 return NULL;
2369 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002370
Victor Stinner26486ea2010-05-15 22:23:53 +00002371 return PyCursesCheckERR(init_color(color, r, g, b), "init_color");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002372}
2373
2374static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002375PyCurses_Init_Pair(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002376{
Victor Stinner26486ea2010-05-15 22:23:53 +00002377 short pair, f, b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002378
Victor Stinner26486ea2010-05-15 22:23:53 +00002379 PyCursesInitialised;
2380 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002381
Victor Stinner26486ea2010-05-15 22:23:53 +00002382 if (PyTuple_Size(args) != 3) {
2383 PyErr_SetString(PyExc_TypeError, "init_pair requires 3 arguments");
2384 return NULL;
2385 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002386
Victor Stinner26486ea2010-05-15 22:23:53 +00002387 if (!PyArg_ParseTuple(args, "hhh;pair, f, b", &pair, &f, &b)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002388
Victor Stinner26486ea2010-05-15 22:23:53 +00002389 return PyCursesCheckERR(init_pair(pair, f, b), "init_pair");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002390}
2391
Andrew M. Kuchling32552682000-12-22 21:52:27 +00002392static PyObject *ModDict;
2393
Victor Stinner26486ea2010-05-15 22:23:53 +00002394static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002395PyCurses_InitScr(PyObject *self)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002396{
Victor Stinner26486ea2010-05-15 22:23:53 +00002397 WINDOW *win;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002398 PyCursesWindowObject *winobj;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002399
Victor Stinner26486ea2010-05-15 22:23:53 +00002400 if (initialised == TRUE) {
2401 wrefresh(stdscr);
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002402 return (PyObject *)PyCursesWindow_New(stdscr, NULL);
Victor Stinner26486ea2010-05-15 22:23:53 +00002403 }
Guido van Rossum56bf2351994-08-31 22:06:24 +00002404
Victor Stinner26486ea2010-05-15 22:23:53 +00002405 win = initscr();
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002406
Victor Stinner26486ea2010-05-15 22:23:53 +00002407 if (win == NULL) {
2408 PyErr_SetString(PyCursesError, catchall_NULL);
2409 return NULL;
2410 }
Guido van Rossum85738471995-02-17 13:50:17 +00002411
Victor Stinner26486ea2010-05-15 22:23:53 +00002412 initialised = initialised_setupterm = TRUE;
Guido van Rossum56bf2351994-08-31 22:06:24 +00002413
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002414/* This was moved from initcurses() because it core dumped on SGI,
2415 where they're not defined until you've called initscr() */
Victor Stinner26486ea2010-05-15 22:23:53 +00002416#define SetDictInt(string,ch) \
2417 do { \
2418 PyObject *o = PyLong_FromLong((long) (ch)); \
2419 if (o && PyDict_SetItemString(ModDict, string, o) == 0) { \
2420 Py_DECREF(o); \
2421 } \
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00002422 } while (0)
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002423
Victor Stinner26486ea2010-05-15 22:23:53 +00002424 /* Here are some graphic symbols you can use */
2425 SetDictInt("ACS_ULCORNER", (ACS_ULCORNER));
2426 SetDictInt("ACS_LLCORNER", (ACS_LLCORNER));
2427 SetDictInt("ACS_URCORNER", (ACS_URCORNER));
2428 SetDictInt("ACS_LRCORNER", (ACS_LRCORNER));
2429 SetDictInt("ACS_LTEE", (ACS_LTEE));
2430 SetDictInt("ACS_RTEE", (ACS_RTEE));
2431 SetDictInt("ACS_BTEE", (ACS_BTEE));
2432 SetDictInt("ACS_TTEE", (ACS_TTEE));
2433 SetDictInt("ACS_HLINE", (ACS_HLINE));
2434 SetDictInt("ACS_VLINE", (ACS_VLINE));
2435 SetDictInt("ACS_PLUS", (ACS_PLUS));
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00002436#if !defined(__hpux) || defined(HAVE_NCURSES_H)
Victor Stinner26486ea2010-05-15 22:23:53 +00002437 /* On HP/UX 11, these are of type cchar_t, which is not an
2438 integral type. If this is a problem on more platforms, a
2439 configure test should be added to determine whether ACS_S1
2440 is of integral type. */
2441 SetDictInt("ACS_S1", (ACS_S1));
2442 SetDictInt("ACS_S9", (ACS_S9));
2443 SetDictInt("ACS_DIAMOND", (ACS_DIAMOND));
2444 SetDictInt("ACS_CKBOARD", (ACS_CKBOARD));
2445 SetDictInt("ACS_DEGREE", (ACS_DEGREE));
2446 SetDictInt("ACS_PLMINUS", (ACS_PLMINUS));
2447 SetDictInt("ACS_BULLET", (ACS_BULLET));
2448 SetDictInt("ACS_LARROW", (ACS_LARROW));
2449 SetDictInt("ACS_RARROW", (ACS_RARROW));
2450 SetDictInt("ACS_DARROW", (ACS_DARROW));
2451 SetDictInt("ACS_UARROW", (ACS_UARROW));
2452 SetDictInt("ACS_BOARD", (ACS_BOARD));
2453 SetDictInt("ACS_LANTERN", (ACS_LANTERN));
2454 SetDictInt("ACS_BLOCK", (ACS_BLOCK));
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00002455#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00002456 SetDictInt("ACS_BSSB", (ACS_ULCORNER));
2457 SetDictInt("ACS_SSBB", (ACS_LLCORNER));
2458 SetDictInt("ACS_BBSS", (ACS_URCORNER));
2459 SetDictInt("ACS_SBBS", (ACS_LRCORNER));
2460 SetDictInt("ACS_SBSS", (ACS_RTEE));
2461 SetDictInt("ACS_SSSB", (ACS_LTEE));
2462 SetDictInt("ACS_SSBS", (ACS_BTEE));
2463 SetDictInt("ACS_BSSS", (ACS_TTEE));
2464 SetDictInt("ACS_BSBS", (ACS_HLINE));
2465 SetDictInt("ACS_SBSB", (ACS_VLINE));
2466 SetDictInt("ACS_SSSS", (ACS_PLUS));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002467
Victor Stinner26486ea2010-05-15 22:23:53 +00002468 /* The following are never available with strict SYSV curses */
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002469#ifdef ACS_S3
Victor Stinner26486ea2010-05-15 22:23:53 +00002470 SetDictInt("ACS_S3", (ACS_S3));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002471#endif
2472#ifdef ACS_S7
Victor Stinner26486ea2010-05-15 22:23:53 +00002473 SetDictInt("ACS_S7", (ACS_S7));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002474#endif
2475#ifdef ACS_LEQUAL
Victor Stinner26486ea2010-05-15 22:23:53 +00002476 SetDictInt("ACS_LEQUAL", (ACS_LEQUAL));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002477#endif
2478#ifdef ACS_GEQUAL
Victor Stinner26486ea2010-05-15 22:23:53 +00002479 SetDictInt("ACS_GEQUAL", (ACS_GEQUAL));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002480#endif
2481#ifdef ACS_PI
Victor Stinner26486ea2010-05-15 22:23:53 +00002482 SetDictInt("ACS_PI", (ACS_PI));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002483#endif
2484#ifdef ACS_NEQUAL
Victor Stinner26486ea2010-05-15 22:23:53 +00002485 SetDictInt("ACS_NEQUAL", (ACS_NEQUAL));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002486#endif
2487#ifdef ACS_STERLING
Victor Stinner26486ea2010-05-15 22:23:53 +00002488 SetDictInt("ACS_STERLING", (ACS_STERLING));
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002489#endif
2490
Victor Stinner26486ea2010-05-15 22:23:53 +00002491 SetDictInt("LINES", LINES);
2492 SetDictInt("COLS", COLS);
Guido van Rossum56bf2351994-08-31 22:06:24 +00002493
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002494 winobj = (PyCursesWindowObject *)PyCursesWindow_New(win, NULL);
2495 screen_encoding = winobj->encoding;
2496 return (PyObject *)winobj;
Guido van Rossumf6971e21994-08-30 12:25:20 +00002497}
2498
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002499static PyObject *
2500PyCurses_setupterm(PyObject* self, PyObject *args, PyObject* keywds)
2501{
Victor Stinner26486ea2010-05-15 22:23:53 +00002502 int fd = -1;
2503 int err;
2504 char* termstr = NULL;
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002505
Victor Stinner26486ea2010-05-15 22:23:53 +00002506 static char *kwlist[] = {"term", "fd", NULL};
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002507
Victor Stinner26486ea2010-05-15 22:23:53 +00002508 if (!PyArg_ParseTupleAndKeywords(
2509 args, keywds, "|zi:setupterm", kwlist, &termstr, &fd)) {
2510 return NULL;
2511 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002512
Victor Stinner26486ea2010-05-15 22:23:53 +00002513 if (fd == -1) {
2514 PyObject* sys_stdout;
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002515
Victor Stinner26486ea2010-05-15 22:23:53 +00002516 sys_stdout = PySys_GetObject("stdout");
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002517
Victor Stinner26486ea2010-05-15 22:23:53 +00002518 if (sys_stdout == NULL || sys_stdout == Py_None) {
2519 PyErr_SetString(
2520 PyCursesError,
2521 "lost sys.stdout");
2522 return NULL;
2523 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002524
Victor Stinner26486ea2010-05-15 22:23:53 +00002525 fd = PyObject_AsFileDescriptor(sys_stdout);
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002526
Victor Stinner26486ea2010-05-15 22:23:53 +00002527 if (fd == -1) {
2528 return NULL;
2529 }
2530 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002531
Matthias Klose635edd12010-07-30 21:40:57 +00002532 if (!initialised_setupterm && setupterm(termstr,fd,&err) == ERR) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002533 char* s = "setupterm: unknown error";
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002534
Victor Stinner26486ea2010-05-15 22:23:53 +00002535 if (err == 0) {
2536 s = "setupterm: could not find terminal";
2537 } else if (err == -1) {
2538 s = "setupterm: could not find terminfo database";
2539 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002540
Victor Stinner26486ea2010-05-15 22:23:53 +00002541 PyErr_SetString(PyCursesError,s);
2542 return NULL;
2543 }
2544
2545 initialised_setupterm = TRUE;
2546
2547 Py_INCREF(Py_None);
2548 return Py_None;
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002549}
Guido van Rossumf6971e21994-08-30 12:25:20 +00002550
2551static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002552PyCurses_IntrFlush(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002553{
Victor Stinner26486ea2010-05-15 22:23:53 +00002554 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002555
Victor Stinner26486ea2010-05-15 22:23:53 +00002556 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002557
Victor Stinner26486ea2010-05-15 22:23:53 +00002558 switch(PyTuple_Size(args)) {
2559 case 1:
2560 if (!PyArg_ParseTuple(args,"i;True(1), False(0)",&ch)) return NULL;
2561 break;
2562 default:
2563 PyErr_SetString(PyExc_TypeError, "intrflush requires 1 argument");
2564 return NULL;
2565 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002566
Victor Stinner26486ea2010-05-15 22:23:53 +00002567 return PyCursesCheckERR(intrflush(NULL,ch), "intrflush");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002568}
2569
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002570#ifdef HAVE_CURSES_IS_TERM_RESIZED
2571static PyObject *
2572PyCurses_Is_Term_Resized(PyObject *self, PyObject *args)
2573{
Victor Stinner26486ea2010-05-15 22:23:53 +00002574 int lines;
2575 int columns;
2576 int result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002577
Victor Stinner26486ea2010-05-15 22:23:53 +00002578 PyCursesInitialised;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002579
Victor Stinner26486ea2010-05-15 22:23:53 +00002580 if (!PyArg_ParseTuple(args,"ii:is_term_resized", &lines, &columns))
2581 return NULL;
2582 result = is_term_resized(lines, columns);
2583 if (result == TRUE) {
2584 Py_INCREF(Py_True);
2585 return Py_True;
2586 } else {
2587 Py_INCREF(Py_False);
2588 return Py_False;
2589 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002590}
2591#endif /* HAVE_CURSES_IS_TERM_RESIZED */
2592
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002593#if !defined(__NetBSD__)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002594static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002595PyCurses_KeyName(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002596{
Victor Stinner26486ea2010-05-15 22:23:53 +00002597 const char *knp;
2598 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002599
Victor Stinner26486ea2010-05-15 22:23:53 +00002600 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002601
Victor Stinner26486ea2010-05-15 22:23:53 +00002602 if (!PyArg_ParseTuple(args,"i",&ch)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002603
Victor Stinner26486ea2010-05-15 22:23:53 +00002604 if (ch < 0) {
2605 PyErr_SetString(PyExc_ValueError, "invalid key number");
2606 return NULL;
2607 }
2608 knp = keyname(ch);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002609
Victor Stinner26486ea2010-05-15 22:23:53 +00002610 return PyBytes_FromString((knp == NULL) ? "" : (char *)knp);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002611}
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002612#endif
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002613
Victor Stinner26486ea2010-05-15 22:23:53 +00002614static PyObject *
2615PyCurses_KillChar(PyObject *self)
2616{
2617 char ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002618
Victor Stinner26486ea2010-05-15 22:23:53 +00002619 ch = killchar();
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002620
Victor Stinner26486ea2010-05-15 22:23:53 +00002621 return PyBytes_FromStringAndSize(&ch, 1);
2622}
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002623
2624static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002625PyCurses_Meta(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002626{
Victor Stinner26486ea2010-05-15 22:23:53 +00002627 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002628
Victor Stinner26486ea2010-05-15 22:23:53 +00002629 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002630
Victor Stinner26486ea2010-05-15 22:23:53 +00002631 switch(PyTuple_Size(args)) {
2632 case 1:
2633 if (!PyArg_ParseTuple(args,"i;True(1), False(0)",&ch)) return NULL;
2634 break;
2635 default:
2636 PyErr_SetString(PyExc_TypeError, "meta requires 1 argument");
2637 return NULL;
2638 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002639
Victor Stinner26486ea2010-05-15 22:23:53 +00002640 return PyCursesCheckERR(meta(stdscr, ch), "meta");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002641}
2642
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002643#ifdef NCURSES_MOUSE_VERSION
2644static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002645PyCurses_MouseInterval(PyObject *self, PyObject *args)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002646{
Victor Stinner26486ea2010-05-15 22:23:53 +00002647 int interval;
2648 PyCursesInitialised;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002649
Victor Stinner26486ea2010-05-15 22:23:53 +00002650 if (!PyArg_ParseTuple(args,"i;interval",&interval))
2651 return NULL;
2652 return PyCursesCheckERR(mouseinterval(interval), "mouseinterval");
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002653}
2654
2655static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002656PyCurses_MouseMask(PyObject *self, PyObject *args)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002657{
Victor Stinner26486ea2010-05-15 22:23:53 +00002658 int newmask;
2659 mmask_t oldmask, availmask;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002660
Victor Stinner26486ea2010-05-15 22:23:53 +00002661 PyCursesInitialised;
2662 if (!PyArg_ParseTuple(args,"i;mousemask",&newmask))
2663 return NULL;
2664 availmask = mousemask(newmask, &oldmask);
2665 return Py_BuildValue("(ll)", (long)availmask, (long)oldmask);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002666}
2667#endif
2668
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002669static PyObject *
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +00002670PyCurses_Napms(PyObject *self, PyObject *args)
2671{
2672 int ms;
2673
Andrew M. Kuchling2b146762010-05-10 23:30:25 +00002674 PyCursesInitialised;
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002675 if (!PyArg_ParseTuple(args, "i;ms", &ms)) return NULL;
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +00002676
2677 return Py_BuildValue("i", napms(ms));
2678}
2679
2680
2681static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002682PyCurses_NewPad(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002683{
Victor Stinner26486ea2010-05-15 22:23:53 +00002684 WINDOW *win;
2685 int nlines, ncols;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002686
Victor Stinner26486ea2010-05-15 22:23:53 +00002687 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002688
Victor Stinner26486ea2010-05-15 22:23:53 +00002689 if (!PyArg_ParseTuple(args,"ii;nlines,ncols",&nlines,&ncols)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002690
Victor Stinner26486ea2010-05-15 22:23:53 +00002691 win = newpad(nlines, ncols);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002692
Victor Stinner26486ea2010-05-15 22:23:53 +00002693 if (win == NULL) {
2694 PyErr_SetString(PyCursesError, catchall_NULL);
2695 return NULL;
2696 }
2697
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002698 return (PyObject *)PyCursesWindow_New(win, NULL);
Guido van Rossumf6971e21994-08-30 12:25:20 +00002699}
2700
2701static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002702PyCurses_NewWindow(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002703{
Victor Stinner26486ea2010-05-15 22:23:53 +00002704 WINDOW *win;
2705 int nlines, ncols, begin_y=0, begin_x=0;
Guido van Rossum85738471995-02-17 13:50:17 +00002706
Victor Stinner26486ea2010-05-15 22:23:53 +00002707 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002708
Victor Stinner26486ea2010-05-15 22:23:53 +00002709 switch (PyTuple_Size(args)) {
2710 case 2:
2711 if (!PyArg_ParseTuple(args,"ii;nlines,ncols",&nlines,&ncols))
2712 return NULL;
2713 break;
2714 case 4:
2715 if (!PyArg_ParseTuple(args, "iiii;nlines,ncols,begin_y,begin_x",
2716 &nlines,&ncols,&begin_y,&begin_x))
2717 return NULL;
2718 break;
2719 default:
2720 PyErr_SetString(PyExc_TypeError, "newwin requires 2 or 4 arguments");
2721 return NULL;
2722 }
Guido van Rossum85738471995-02-17 13:50:17 +00002723
Victor Stinner26486ea2010-05-15 22:23:53 +00002724 win = newwin(nlines,ncols,begin_y,begin_x);
2725 if (win == NULL) {
2726 PyErr_SetString(PyCursesError, catchall_NULL);
2727 return NULL;
2728 }
Guido van Rossum85738471995-02-17 13:50:17 +00002729
Victor Stinner0fdfceb2011-11-25 22:10:02 +01002730 return (PyObject *)PyCursesWindow_New(win, NULL);
Guido van Rossumf6971e21994-08-30 12:25:20 +00002731}
2732
2733static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002734PyCurses_Pair_Content(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002735{
Victor Stinner26486ea2010-05-15 22:23:53 +00002736 short pair,f,b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002737
Victor Stinner26486ea2010-05-15 22:23:53 +00002738 PyCursesInitialised;
2739 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002740
Victor Stinner26486ea2010-05-15 22:23:53 +00002741 switch(PyTuple_Size(args)) {
2742 case 1:
2743 if (!PyArg_ParseTuple(args, "h;pair", &pair)) return NULL;
2744 break;
2745 default:
2746 PyErr_SetString(PyExc_TypeError, "pair_content requires 1 argument");
2747 return NULL;
2748 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002749
Victor Stinner26486ea2010-05-15 22:23:53 +00002750 if (pair_content(pair, &f, &b)==ERR) {
2751 PyErr_SetString(PyCursesError,
2752 "Argument 1 was out of range. (1..COLOR_PAIRS-1)");
2753 return NULL;
2754 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002755
Victor Stinner26486ea2010-05-15 22:23:53 +00002756 return Py_BuildValue("(ii)", f, b);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002757}
2758
2759static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002760PyCurses_pair_number(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002761{
Victor Stinner26486ea2010-05-15 22:23:53 +00002762 int n;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002763
Victor Stinner26486ea2010-05-15 22:23:53 +00002764 PyCursesInitialised;
2765 PyCursesInitialisedColor;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002766
Victor Stinner26486ea2010-05-15 22:23:53 +00002767 switch(PyTuple_Size(args)) {
2768 case 1:
2769 if (!PyArg_ParseTuple(args, "i;pairvalue", &n)) return NULL;
2770 break;
2771 default:
2772 PyErr_SetString(PyExc_TypeError,
2773 "pair_number requires 1 argument");
2774 return NULL;
2775 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002776
Victor Stinner26486ea2010-05-15 22:23:53 +00002777 return PyLong_FromLong((long) ((n & A_COLOR) >> 8));
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002778}
2779
2780static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002781PyCurses_Putp(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002782{
Victor Stinner26486ea2010-05-15 22:23:53 +00002783 char *str;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002784
Victor Stinnerb3bc7e72011-11-03 20:35:40 +01002785 if (!PyArg_ParseTuple(args,"y;str", &str))
2786 return NULL;
Victor Stinner26486ea2010-05-15 22:23:53 +00002787 return PyCursesCheckERR(putp(str), "putp");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002788}
2789
2790static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002791PyCurses_QiFlush(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002792{
Victor Stinner26486ea2010-05-15 22:23:53 +00002793 int flag = 0;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002794
Victor Stinner26486ea2010-05-15 22:23:53 +00002795 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002796
Victor Stinner26486ea2010-05-15 22:23:53 +00002797 switch(PyTuple_Size(args)) {
2798 case 0:
2799 qiflush();
2800 Py_INCREF(Py_None);
2801 return Py_None;
2802 case 1:
2803 if (!PyArg_ParseTuple(args, "i;True(1) or False(0)", &flag)) return NULL;
2804 if (flag) qiflush();
2805 else noqiflush();
2806 Py_INCREF(Py_None);
2807 return Py_None;
2808 default:
2809 PyErr_SetString(PyExc_TypeError, "qiflush requires 0 or 1 arguments");
2810 return NULL;
2811 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002812}
2813
Guido van Rossumd8faa362007-04-27 19:54:29 +00002814/* Internal helper used for updating curses.LINES, curses.COLS, _curses.LINES
2815 * and _curses.COLS */
Benjamin Petersonfea6a942008-07-02 16:11:42 +00002816#if defined(HAVE_CURSES_RESIZETERM) || defined(HAVE_CURSES_RESIZE_TERM)
Guido van Rossumd8faa362007-04-27 19:54:29 +00002817static int
2818update_lines_cols(void)
2819{
Victor Stinner26486ea2010-05-15 22:23:53 +00002820 PyObject *o;
2821 PyObject *m = PyImport_ImportModuleNoBlock("curses");
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002822 _Py_IDENTIFIER(LINES);
2823 _Py_IDENTIFIER(COLS);
Guido van Rossumd8faa362007-04-27 19:54:29 +00002824
Victor Stinner26486ea2010-05-15 22:23:53 +00002825 if (!m)
2826 return 0;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002827
Victor Stinner26486ea2010-05-15 22:23:53 +00002828 o = PyLong_FromLong(LINES);
2829 if (!o) {
2830 Py_DECREF(m);
2831 return 0;
2832 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002833 if (_PyObject_SetAttrId(m, &PyId_LINES, o)) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002834 Py_DECREF(m);
2835 Py_DECREF(o);
2836 return 0;
2837 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002838 /* PyId_LINES.object will be initialized here. */
2839 if (PyDict_SetItem(ModDict, PyId_LINES.object, o)) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002840 Py_DECREF(m);
2841 Py_DECREF(o);
2842 return 0;
2843 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002844 Py_DECREF(o);
Victor Stinner26486ea2010-05-15 22:23:53 +00002845 o = PyLong_FromLong(COLS);
2846 if (!o) {
2847 Py_DECREF(m);
2848 return 0;
2849 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002850 if (_PyObject_SetAttrId(m, &PyId_COLS, o)) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002851 Py_DECREF(m);
2852 Py_DECREF(o);
2853 return 0;
2854 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002855 if (PyDict_SetItem(ModDict, PyId_COLS.object, o)) {
Victor Stinner26486ea2010-05-15 22:23:53 +00002856 Py_DECREF(m);
2857 Py_DECREF(o);
2858 return 0;
2859 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002860 Py_DECREF(o);
Guido van Rossumd8faa362007-04-27 19:54:29 +00002861 Py_DECREF(m);
Victor Stinner26486ea2010-05-15 22:23:53 +00002862 return 1;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002863}
Benjamin Petersonfea6a942008-07-02 16:11:42 +00002864#endif
Guido van Rossumd8faa362007-04-27 19:54:29 +00002865
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002866#ifdef HAVE_CURSES_RESIZETERM
2867static PyObject *
2868PyCurses_ResizeTerm(PyObject *self, PyObject *args)
2869{
Victor Stinner26486ea2010-05-15 22:23:53 +00002870 int lines;
2871 int columns;
2872 PyObject *result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002873
Victor Stinner26486ea2010-05-15 22:23:53 +00002874 PyCursesInitialised;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002875
Victor Stinner26486ea2010-05-15 22:23:53 +00002876 if (!PyArg_ParseTuple(args,"ii:resizeterm", &lines, &columns))
2877 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002878
Victor Stinner26486ea2010-05-15 22:23:53 +00002879 result = PyCursesCheckERR(resizeterm(lines, columns), "resizeterm");
2880 if (!result)
2881 return NULL;
2882 if (!update_lines_cols())
2883 return NULL;
2884 return result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002885}
2886
2887#endif
2888
2889#ifdef HAVE_CURSES_RESIZE_TERM
2890static PyObject *
2891PyCurses_Resize_Term(PyObject *self, PyObject *args)
2892{
Victor Stinner26486ea2010-05-15 22:23:53 +00002893 int lines;
2894 int columns;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002895
Victor Stinner26486ea2010-05-15 22:23:53 +00002896 PyObject *result;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002897
Victor Stinner26486ea2010-05-15 22:23:53 +00002898 PyCursesInitialised;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002899
Victor Stinner26486ea2010-05-15 22:23:53 +00002900 if (!PyArg_ParseTuple(args,"ii:resize_term", &lines, &columns))
2901 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002902
Victor Stinner26486ea2010-05-15 22:23:53 +00002903 result = PyCursesCheckERR(resize_term(lines, columns), "resize_term");
2904 if (!result)
2905 return NULL;
2906 if (!update_lines_cols())
2907 return NULL;
2908 return result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002909}
2910#endif /* HAVE_CURSES_RESIZE_TERM */
2911
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002912static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00002913PyCurses_setsyx(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002914{
Victor Stinner26486ea2010-05-15 22:23:53 +00002915 int y,x;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002916
Victor Stinner26486ea2010-05-15 22:23:53 +00002917 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002918
Victor Stinner26486ea2010-05-15 22:23:53 +00002919 if (PyTuple_Size(args)!=2) {
2920 PyErr_SetString(PyExc_TypeError, "setsyx requires 2 arguments");
2921 return NULL;
2922 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002923
Victor Stinner26486ea2010-05-15 22:23:53 +00002924 if (!PyArg_ParseTuple(args, "ii;y, x", &y, &x)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002925
Victor Stinner26486ea2010-05-15 22:23:53 +00002926 setsyx(y,x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002927
Victor Stinner26486ea2010-05-15 22:23:53 +00002928 Py_INCREF(Py_None);
2929 return Py_None;
Guido van Rossumf6971e21994-08-30 12:25:20 +00002930}
2931
2932static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +00002933PyCurses_Start_Color(PyObject *self)
Guido van Rossumf6971e21994-08-30 12:25:20 +00002934{
Victor Stinner26486ea2010-05-15 22:23:53 +00002935 int code;
2936 PyObject *c, *cp;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002937
Victor Stinner26486ea2010-05-15 22:23:53 +00002938 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002939
Victor Stinner26486ea2010-05-15 22:23:53 +00002940 code = start_color();
2941 if (code != ERR) {
2942 initialisedcolors = TRUE;
2943 c = PyLong_FromLong((long) COLORS);
2944 PyDict_SetItemString(ModDict, "COLORS", c);
2945 Py_DECREF(c);
2946 cp = PyLong_FromLong((long) COLOR_PAIRS);
2947 PyDict_SetItemString(ModDict, "COLOR_PAIRS", cp);
2948 Py_DECREF(cp);
2949 Py_INCREF(Py_None);
2950 return Py_None;
2951 } else {
2952 PyErr_SetString(PyCursesError, "start_color() returned ERR");
2953 return NULL;
2954 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002955}
2956
2957static PyObject *
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002958PyCurses_tigetflag(PyObject *self, PyObject *args)
2959{
Victor Stinner26486ea2010-05-15 22:23:53 +00002960 char *capname;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002961
Victor Stinner26486ea2010-05-15 22:23:53 +00002962 PyCursesSetupTermCalled;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002963
Victor Stinner26486ea2010-05-15 22:23:53 +00002964 if (!PyArg_ParseTuple(args, "s", &capname))
2965 return NULL;
2966
2967 return PyLong_FromLong( (long) tigetflag( capname ) );
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002968}
2969
2970static PyObject *
2971PyCurses_tigetnum(PyObject *self, PyObject *args)
2972{
Victor Stinner26486ea2010-05-15 22:23:53 +00002973 char *capname;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002974
Victor Stinner26486ea2010-05-15 22:23:53 +00002975 PyCursesSetupTermCalled;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002976
Victor Stinner26486ea2010-05-15 22:23:53 +00002977 if (!PyArg_ParseTuple(args, "s", &capname))
2978 return NULL;
2979
2980 return PyLong_FromLong( (long) tigetnum( capname ) );
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002981}
2982
2983static PyObject *
2984PyCurses_tigetstr(PyObject *self, PyObject *args)
2985{
Victor Stinner26486ea2010-05-15 22:23:53 +00002986 char *capname;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002987
Victor Stinner26486ea2010-05-15 22:23:53 +00002988 PyCursesSetupTermCalled;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002989
Victor Stinner26486ea2010-05-15 22:23:53 +00002990 if (!PyArg_ParseTuple(args, "s", &capname))
2991 return NULL;
2992
2993 capname = tigetstr( capname );
2994 if (capname == 0 || capname == (char*) -1) {
2995 Py_INCREF(Py_None);
2996 return Py_None;
2997 }
2998 return PyBytes_FromString( capname );
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002999}
3000
3001static PyObject *
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00003002PyCurses_tparm(PyObject *self, PyObject *args)
3003{
Victor Stinner26486ea2010-05-15 22:23:53 +00003004 char* fmt;
3005 char* result = NULL;
3006 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 +00003007
Victor Stinner26486ea2010-05-15 22:23:53 +00003008 PyCursesSetupTermCalled;
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00003009
Victor Stinner26621332011-11-02 23:45:29 +01003010 if (!PyArg_ParseTuple(args, "y|iiiiiiiii:tparm",
Victor Stinner26486ea2010-05-15 22:23:53 +00003011 &fmt, &i1, &i2, &i3, &i4,
3012 &i5, &i6, &i7, &i8, &i9)) {
3013 return NULL;
3014 }
Michael W. Hudson21007982002-11-21 14:17:51 +00003015
Victor Stinner26486ea2010-05-15 22:23:53 +00003016 result = tparm(fmt,i1,i2,i3,i4,i5,i6,i7,i8,i9);
3017 if (!result) {
3018 PyErr_SetString(PyCursesError, "tparm() returned NULL");
3019 return NULL;
3020 }
Michael W. Hudson21007982002-11-21 14:17:51 +00003021
Victor Stinner26486ea2010-05-15 22:23:53 +00003022 return PyBytes_FromString(result);
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00003023}
3024
3025static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00003026PyCurses_TypeAhead(PyObject *self, PyObject *args)
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003027{
Victor Stinner26486ea2010-05-15 22:23:53 +00003028 int fd;
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003029
Victor Stinner26486ea2010-05-15 22:23:53 +00003030 PyCursesInitialised;
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003031
Victor Stinner26486ea2010-05-15 22:23:53 +00003032 if (!PyArg_ParseTuple(args,"i;fd",&fd)) return NULL;
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003033
Victor Stinner26486ea2010-05-15 22:23:53 +00003034 return PyCursesCheckERR(typeahead( fd ), "typeahead");
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00003035}
3036
3037static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00003038PyCurses_UnCtrl(PyObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003039{
Victor Stinner26486ea2010-05-15 22:23:53 +00003040 PyObject *temp;
3041 chtype ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003042
Victor Stinner26486ea2010-05-15 22:23:53 +00003043 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003044
Victor Stinner26486ea2010-05-15 22:23:53 +00003045 if (!PyArg_ParseTuple(args,"O;ch or int",&temp)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003046
Victor Stinner0fdfceb2011-11-25 22:10:02 +01003047 if (!PyCurses_ConvertToChtype(NULL, temp, &ch))
Victor Stinner26486ea2010-05-15 22:23:53 +00003048 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003049
Victor Stinner26486ea2010-05-15 22:23:53 +00003050 return PyBytes_FromString(unctrl(ch));
Guido van Rossumf6971e21994-08-30 12:25:20 +00003051}
3052
3053static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00003054PyCurses_UngetCh(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00003055{
Victor Stinner26486ea2010-05-15 22:23:53 +00003056 PyObject *temp;
3057 chtype ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003058
Victor Stinner26486ea2010-05-15 22:23:53 +00003059 PyCursesInitialised;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003060
Victor Stinner0fdfceb2011-11-25 22:10:02 +01003061 if (!PyArg_ParseTuple(args,"O;ch or int",&temp))
Victor Stinner26486ea2010-05-15 22:23:53 +00003062 return NULL;
Victor Stinner0fdfceb2011-11-25 22:10:02 +01003063
3064 if (!PyCurses_ConvertToChtype(NULL, temp, &ch))
3065 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003066
Victor Stinner26486ea2010-05-15 22:23:53 +00003067 return PyCursesCheckERR(ungetch(ch), "ungetch");
Guido van Rossumf6971e21994-08-30 12:25:20 +00003068}
3069
Victor Stinner71e44cb2011-09-06 01:53:03 +02003070#ifdef HAVE_NCURSESW
3071/* Convert an object to a character (wchar_t):
3072
3073 - int
3074 - str of length 1
3075
3076 Return 1 on success, 0 on error. */
3077static int
3078PyCurses_ConvertToWchar_t(PyObject *obj,
3079 wchar_t *wch)
3080{
3081 if (PyUnicode_Check(obj)) {
3082 wchar_t buffer[2];
3083 if (PyUnicode_AsWideChar(obj, buffer, 2) != 1) {
3084 PyErr_Format(PyExc_TypeError,
3085 "expect bytes or str of length 1, or int, "
3086 "got a str of length %zi",
Victor Stinnerc4f281e2011-10-11 22:11:42 +02003087 PyUnicode_GET_LENGTH(obj));
Victor Stinner71e44cb2011-09-06 01:53:03 +02003088 return 0;
3089 }
3090 *wch = buffer[0];
3091 return 2;
3092 }
3093 else if (PyLong_CheckExact(obj)) {
3094 long value;
3095 int overflow;
3096 value = PyLong_AsLongAndOverflow(obj, &overflow);
3097 if (overflow) {
3098 PyErr_SetString(PyExc_OverflowError,
3099 "int doesn't fit in long");
3100 return 0;
3101 }
3102 *wch = (wchar_t)value;
3103 if ((long)*wch != value) {
3104 PyErr_Format(PyExc_OverflowError,
3105 "character doesn't fit in wchar_t");
3106 return 0;
3107 }
3108 return 1;
3109 }
3110 else {
3111 PyErr_Format(PyExc_TypeError,
3112 "expect bytes or str of length 1, or int, got %s",
3113 Py_TYPE(obj)->tp_name);
3114 return 0;
3115 }
3116}
3117
3118static PyObject *
3119PyCurses_Unget_Wch(PyObject *self, PyObject *args)
3120{
3121 PyObject *obj;
3122 wchar_t wch;
3123
3124 PyCursesInitialised;
3125
3126 if (!PyArg_ParseTuple(args,"O", &obj))
3127 return NULL;
3128
3129 if (!PyCurses_ConvertToWchar_t(obj, &wch))
3130 return NULL;
3131 return PyCursesCheckERR(unget_wch(wch), "unget_wch");
3132}
3133#endif
3134
Guido van Rossumf6971e21994-08-30 12:25:20 +00003135static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00003136PyCurses_Use_Env(PyObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00003137{
Victor Stinner26486ea2010-05-15 22:23:53 +00003138 int flag;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003139
Victor Stinner26486ea2010-05-15 22:23:53 +00003140 switch(PyTuple_Size(args)) {
3141 case 1:
3142 if (!PyArg_ParseTuple(args,"i;True(1), False(0)",&flag))
3143 return NULL;
3144 break;
3145 default:
3146 PyErr_SetString(PyExc_TypeError, "use_env requires 1 argument");
3147 return NULL;
3148 }
3149 use_env(flag);
3150 Py_INCREF(Py_None);
3151 return Py_None;
Guido van Rossumf6971e21994-08-30 12:25:20 +00003152}
3153
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003154#ifndef STRICT_SYSV_CURSES
3155static PyObject *
3156PyCurses_Use_Default_Colors(PyObject *self)
3157{
Victor Stinner26486ea2010-05-15 22:23:53 +00003158 int code;
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003159
Victor Stinner26486ea2010-05-15 22:23:53 +00003160 PyCursesInitialised;
3161 PyCursesInitialisedColor;
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003162
Victor Stinner26486ea2010-05-15 22:23:53 +00003163 code = use_default_colors();
3164 if (code != ERR) {
3165 Py_INCREF(Py_None);
3166 return Py_None;
3167 } else {
3168 PyErr_SetString(PyCursesError, "use_default_colors() returned ERR");
3169 return NULL;
3170 }
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003171}
3172#endif /* STRICT_SYSV_CURSES */
3173
Guido van Rossumf6971e21994-08-30 12:25:20 +00003174/* List of functions defined in the module */
3175
3176static PyMethodDef PyCurses_methods[] = {
Victor Stinner26486ea2010-05-15 22:23:53 +00003177 {"baudrate", (PyCFunction)PyCurses_baudrate, METH_NOARGS},
3178 {"beep", (PyCFunction)PyCurses_beep, METH_NOARGS},
3179 {"can_change_color", (PyCFunction)PyCurses_can_change_color, METH_NOARGS},
3180 {"cbreak", (PyCFunction)PyCurses_cbreak, METH_VARARGS},
3181 {"color_content", (PyCFunction)PyCurses_Color_Content, METH_VARARGS},
3182 {"color_pair", (PyCFunction)PyCurses_color_pair, METH_VARARGS},
3183 {"curs_set", (PyCFunction)PyCurses_Curs_Set, METH_VARARGS},
3184 {"def_prog_mode", (PyCFunction)PyCurses_def_prog_mode, METH_NOARGS},
3185 {"def_shell_mode", (PyCFunction)PyCurses_def_shell_mode, METH_NOARGS},
3186 {"delay_output", (PyCFunction)PyCurses_Delay_Output, METH_VARARGS},
3187 {"doupdate", (PyCFunction)PyCurses_doupdate, METH_NOARGS},
3188 {"echo", (PyCFunction)PyCurses_echo, METH_VARARGS},
3189 {"endwin", (PyCFunction)PyCurses_endwin, METH_NOARGS},
3190 {"erasechar", (PyCFunction)PyCurses_EraseChar, METH_NOARGS},
3191 {"filter", (PyCFunction)PyCurses_filter, METH_NOARGS},
3192 {"flash", (PyCFunction)PyCurses_flash, METH_NOARGS},
3193 {"flushinp", (PyCFunction)PyCurses_flushinp, METH_NOARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003194#ifdef NCURSES_MOUSE_VERSION
Victor Stinner26486ea2010-05-15 22:23:53 +00003195 {"getmouse", (PyCFunction)PyCurses_GetMouse, METH_NOARGS},
3196 {"ungetmouse", (PyCFunction)PyCurses_UngetMouse, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003197#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003198 {"getsyx", (PyCFunction)PyCurses_getsyx, METH_NOARGS},
3199 {"getwin", (PyCFunction)PyCurses_GetWin, METH_O},
3200 {"has_colors", (PyCFunction)PyCurses_has_colors, METH_NOARGS},
3201 {"has_ic", (PyCFunction)PyCurses_has_ic, METH_NOARGS},
3202 {"has_il", (PyCFunction)PyCurses_has_il, METH_NOARGS},
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +00003203#ifndef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00003204 {"has_key", (PyCFunction)PyCurses_has_key, METH_VARARGS},
Guido van Rossumf6971e21994-08-30 12:25:20 +00003205#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003206 {"halfdelay", (PyCFunction)PyCurses_HalfDelay, METH_VARARGS},
3207 {"init_color", (PyCFunction)PyCurses_Init_Color, METH_VARARGS},
3208 {"init_pair", (PyCFunction)PyCurses_Init_Pair, METH_VARARGS},
3209 {"initscr", (PyCFunction)PyCurses_InitScr, METH_NOARGS},
3210 {"intrflush", (PyCFunction)PyCurses_IntrFlush, METH_VARARGS},
3211 {"isendwin", (PyCFunction)PyCurses_isendwin, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003212#ifdef HAVE_CURSES_IS_TERM_RESIZED
Victor Stinner26486ea2010-05-15 22:23:53 +00003213 {"is_term_resized", (PyCFunction)PyCurses_Is_Term_Resized, METH_VARARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003214#endif
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003215#if !defined(__NetBSD__)
Victor Stinner26486ea2010-05-15 22:23:53 +00003216 {"keyname", (PyCFunction)PyCurses_KeyName, METH_VARARGS},
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003217#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003218 {"killchar", (PyCFunction)PyCurses_KillChar, METH_NOARGS},
3219 {"longname", (PyCFunction)PyCurses_longname, METH_NOARGS},
3220 {"meta", (PyCFunction)PyCurses_Meta, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003221#ifdef NCURSES_MOUSE_VERSION
Victor Stinner26486ea2010-05-15 22:23:53 +00003222 {"mouseinterval", (PyCFunction)PyCurses_MouseInterval, METH_VARARGS},
3223 {"mousemask", (PyCFunction)PyCurses_MouseMask, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003224#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003225 {"napms", (PyCFunction)PyCurses_Napms, METH_VARARGS},
3226 {"newpad", (PyCFunction)PyCurses_NewPad, METH_VARARGS},
3227 {"newwin", (PyCFunction)PyCurses_NewWindow, METH_VARARGS},
3228 {"nl", (PyCFunction)PyCurses_nl, METH_VARARGS},
3229 {"nocbreak", (PyCFunction)PyCurses_nocbreak, METH_NOARGS},
3230 {"noecho", (PyCFunction)PyCurses_noecho, METH_NOARGS},
3231 {"nonl", (PyCFunction)PyCurses_nonl, METH_NOARGS},
3232 {"noqiflush", (PyCFunction)PyCurses_noqiflush, METH_NOARGS},
3233 {"noraw", (PyCFunction)PyCurses_noraw, METH_NOARGS},
3234 {"pair_content", (PyCFunction)PyCurses_Pair_Content, METH_VARARGS},
3235 {"pair_number", (PyCFunction)PyCurses_pair_number, METH_VARARGS},
3236 {"putp", (PyCFunction)PyCurses_Putp, METH_VARARGS},
3237 {"qiflush", (PyCFunction)PyCurses_QiFlush, METH_VARARGS},
3238 {"raw", (PyCFunction)PyCurses_raw, METH_VARARGS},
3239 {"reset_prog_mode", (PyCFunction)PyCurses_reset_prog_mode, METH_NOARGS},
3240 {"reset_shell_mode", (PyCFunction)PyCurses_reset_shell_mode, METH_NOARGS},
3241 {"resetty", (PyCFunction)PyCurses_resetty, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003242#ifdef HAVE_CURSES_RESIZETERM
Victor Stinner26486ea2010-05-15 22:23:53 +00003243 {"resizeterm", (PyCFunction)PyCurses_ResizeTerm, METH_VARARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003244#endif
3245#ifdef HAVE_CURSES_RESIZE_TERM
Victor Stinner26486ea2010-05-15 22:23:53 +00003246 {"resize_term", (PyCFunction)PyCurses_Resize_Term, METH_VARARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003247#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003248 {"savetty", (PyCFunction)PyCurses_savetty, METH_NOARGS},
3249 {"setsyx", (PyCFunction)PyCurses_setsyx, METH_VARARGS},
3250 {"setupterm", (PyCFunction)PyCurses_setupterm,
3251 METH_VARARGS|METH_KEYWORDS},
3252 {"start_color", (PyCFunction)PyCurses_Start_Color, METH_NOARGS},
3253 {"termattrs", (PyCFunction)PyCurses_termattrs, METH_NOARGS},
3254 {"termname", (PyCFunction)PyCurses_termname, METH_NOARGS},
3255 {"tigetflag", (PyCFunction)PyCurses_tigetflag, METH_VARARGS},
3256 {"tigetnum", (PyCFunction)PyCurses_tigetnum, METH_VARARGS},
3257 {"tigetstr", (PyCFunction)PyCurses_tigetstr, METH_VARARGS},
3258 {"tparm", (PyCFunction)PyCurses_tparm, METH_VARARGS},
3259 {"typeahead", (PyCFunction)PyCurses_TypeAhead, METH_VARARGS},
3260 {"unctrl", (PyCFunction)PyCurses_UnCtrl, METH_VARARGS},
3261 {"ungetch", (PyCFunction)PyCurses_UngetCh, METH_VARARGS},
Victor Stinner71e44cb2011-09-06 01:53:03 +02003262#ifdef HAVE_NCURSESW
3263 {"unget_wch", (PyCFunction)PyCurses_Unget_Wch, METH_VARARGS},
3264#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003265 {"use_env", (PyCFunction)PyCurses_Use_Env, METH_VARARGS},
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003266#ifndef STRICT_SYSV_CURSES
Victor Stinner26486ea2010-05-15 22:23:53 +00003267 {"use_default_colors", (PyCFunction)PyCurses_Use_Default_Colors, METH_NOARGS},
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00003268#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003269 {NULL, NULL} /* sentinel */
Guido van Rossumf6971e21994-08-30 12:25:20 +00003270};
3271
3272/* Initialization function for the module */
3273
Martin v. Löwis1a214512008-06-11 05:26:20 +00003274
3275static struct PyModuleDef _cursesmodule = {
Victor Stinner26486ea2010-05-15 22:23:53 +00003276 PyModuleDef_HEAD_INIT,
3277 "_curses",
3278 NULL,
3279 -1,
3280 PyCurses_methods,
3281 NULL,
3282 NULL,
3283 NULL,
3284 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00003285};
3286
Mark Hammondfe51c6d2002-08-02 02:27:13 +00003287PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00003288PyInit__curses(void)
Guido van Rossumf6971e21994-08-30 12:25:20 +00003289{
Victor Stinner26486ea2010-05-15 22:23:53 +00003290 PyObject *m, *d, *v, *c_api_object;
3291 static void *PyCurses_API[PyCurses_API_pointers];
Andrew M. Kuchling32552682000-12-22 21:52:27 +00003292
Victor Stinner26486ea2010-05-15 22:23:53 +00003293 /* Initialize object type */
3294 if (PyType_Ready(&PyCursesWindow_Type) < 0)
3295 return NULL;
Guido van Rossum6915c4d2001-01-19 00:28:08 +00003296
Victor Stinner26486ea2010-05-15 22:23:53 +00003297 /* Initialize the C API pointer array */
3298 PyCurses_API[0] = (void *)&PyCursesWindow_Type;
3299 PyCurses_API[1] = (void *)func_PyCursesSetupTermCalled;
3300 PyCurses_API[2] = (void *)func_PyCursesInitialised;
3301 PyCurses_API[3] = (void *)func_PyCursesInitialisedColor;
Guido van Rossumf6971e21994-08-30 12:25:20 +00003302
Victor Stinner26486ea2010-05-15 22:23:53 +00003303 /* Create the module and add the functions */
3304 m = PyModule_Create(&_cursesmodule);
3305 if (m == NULL)
3306 return NULL;
Guido van Rossumf6971e21994-08-30 12:25:20 +00003307
Victor Stinner26486ea2010-05-15 22:23:53 +00003308 /* Add some symbolic constants to the module */
3309 d = PyModule_GetDict(m);
3310 if (d == NULL)
3311 return NULL;
3312 ModDict = d; /* For PyCurses_InitScr to use later */
Andrew M. Kuchling32552682000-12-22 21:52:27 +00003313
Victor Stinner26486ea2010-05-15 22:23:53 +00003314 /* Add a capsule for the C API */
3315 c_api_object = PyCapsule_New(PyCurses_API, PyCurses_CAPSULE_NAME, NULL);
3316 PyDict_SetItemString(d, "_C_API", c_api_object);
3317 Py_DECREF(c_api_object);
Guido van Rossumfbea2f31994-08-31 22:05:27 +00003318
Victor Stinner26486ea2010-05-15 22:23:53 +00003319 /* For exception curses.error */
3320 PyCursesError = PyErr_NewException("_curses.error", NULL, NULL);
3321 PyDict_SetItemString(d, "error", PyCursesError);
Guido van Rossumfbea2f31994-08-31 22:05:27 +00003322
Victor Stinner26486ea2010-05-15 22:23:53 +00003323 /* Make the version available */
3324 v = PyBytes_FromString(PyCursesVersion);
3325 PyDict_SetItemString(d, "version", v);
3326 PyDict_SetItemString(d, "__version__", v);
3327 Py_DECREF(v);
Guido van Rossumf6971e21994-08-30 12:25:20 +00003328
Victor Stinner26486ea2010-05-15 22:23:53 +00003329 SetDictInt("ERR", ERR);
3330 SetDictInt("OK", OK);
Andrew M. Kuchlingcaefb372000-12-26 15:57:01 +00003331
Victor Stinner26486ea2010-05-15 22:23:53 +00003332 /* Here are some attributes you can add to chars to print */
3333
3334 SetDictInt("A_ATTRIBUTES", A_ATTRIBUTES);
3335 SetDictInt("A_NORMAL", A_NORMAL);
3336 SetDictInt("A_STANDOUT", A_STANDOUT);
3337 SetDictInt("A_UNDERLINE", A_UNDERLINE);
3338 SetDictInt("A_REVERSE", A_REVERSE);
3339 SetDictInt("A_BLINK", A_BLINK);
3340 SetDictInt("A_DIM", A_DIM);
3341 SetDictInt("A_BOLD", A_BOLD);
3342 SetDictInt("A_ALTCHARSET", A_ALTCHARSET);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003343#if !defined(__NetBSD__)
Victor Stinner26486ea2010-05-15 22:23:53 +00003344 SetDictInt("A_INVIS", A_INVIS);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003345#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003346 SetDictInt("A_PROTECT", A_PROTECT);
3347 SetDictInt("A_CHARTEXT", A_CHARTEXT);
3348 SetDictInt("A_COLOR", A_COLOR);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003349
Victor Stinner26486ea2010-05-15 22:23:53 +00003350 /* The following are never available with strict SYSV curses */
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003351#ifdef A_HORIZONTAL
Victor Stinner26486ea2010-05-15 22:23:53 +00003352 SetDictInt("A_HORIZONTAL", A_HORIZONTAL);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003353#endif
3354#ifdef A_LEFT
Victor Stinner26486ea2010-05-15 22:23:53 +00003355 SetDictInt("A_LEFT", A_LEFT);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003356#endif
3357#ifdef A_LOW
Victor Stinner26486ea2010-05-15 22:23:53 +00003358 SetDictInt("A_LOW", A_LOW);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003359#endif
3360#ifdef A_RIGHT
Victor Stinner26486ea2010-05-15 22:23:53 +00003361 SetDictInt("A_RIGHT", A_RIGHT);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003362#endif
3363#ifdef A_TOP
Victor Stinner26486ea2010-05-15 22:23:53 +00003364 SetDictInt("A_TOP", A_TOP);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00003365#endif
3366#ifdef A_VERTICAL
Victor Stinner26486ea2010-05-15 22:23:53 +00003367 SetDictInt("A_VERTICAL", A_VERTICAL);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00003368#endif
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00003369
Victor Stinner26486ea2010-05-15 22:23:53 +00003370 SetDictInt("COLOR_BLACK", COLOR_BLACK);
3371 SetDictInt("COLOR_RED", COLOR_RED);
3372 SetDictInt("COLOR_GREEN", COLOR_GREEN);
3373 SetDictInt("COLOR_YELLOW", COLOR_YELLOW);
3374 SetDictInt("COLOR_BLUE", COLOR_BLUE);
3375 SetDictInt("COLOR_MAGENTA", COLOR_MAGENTA);
3376 SetDictInt("COLOR_CYAN", COLOR_CYAN);
3377 SetDictInt("COLOR_WHITE", COLOR_WHITE);
Guido van Rossumf6971e21994-08-30 12:25:20 +00003378
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003379#ifdef NCURSES_MOUSE_VERSION
Victor Stinner26486ea2010-05-15 22:23:53 +00003380 /* Mouse-related constants */
3381 SetDictInt("BUTTON1_PRESSED", BUTTON1_PRESSED);
3382 SetDictInt("BUTTON1_RELEASED", BUTTON1_RELEASED);
3383 SetDictInt("BUTTON1_CLICKED", BUTTON1_CLICKED);
3384 SetDictInt("BUTTON1_DOUBLE_CLICKED", BUTTON1_DOUBLE_CLICKED);
3385 SetDictInt("BUTTON1_TRIPLE_CLICKED", BUTTON1_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003386
Victor Stinner26486ea2010-05-15 22:23:53 +00003387 SetDictInt("BUTTON2_PRESSED", BUTTON2_PRESSED);
3388 SetDictInt("BUTTON2_RELEASED", BUTTON2_RELEASED);
3389 SetDictInt("BUTTON2_CLICKED", BUTTON2_CLICKED);
3390 SetDictInt("BUTTON2_DOUBLE_CLICKED", BUTTON2_DOUBLE_CLICKED);
3391 SetDictInt("BUTTON2_TRIPLE_CLICKED", BUTTON2_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003392
Victor Stinner26486ea2010-05-15 22:23:53 +00003393 SetDictInt("BUTTON3_PRESSED", BUTTON3_PRESSED);
3394 SetDictInt("BUTTON3_RELEASED", BUTTON3_RELEASED);
3395 SetDictInt("BUTTON3_CLICKED", BUTTON3_CLICKED);
3396 SetDictInt("BUTTON3_DOUBLE_CLICKED", BUTTON3_DOUBLE_CLICKED);
3397 SetDictInt("BUTTON3_TRIPLE_CLICKED", BUTTON3_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003398
Victor Stinner26486ea2010-05-15 22:23:53 +00003399 SetDictInt("BUTTON4_PRESSED", BUTTON4_PRESSED);
3400 SetDictInt("BUTTON4_RELEASED", BUTTON4_RELEASED);
3401 SetDictInt("BUTTON4_CLICKED", BUTTON4_CLICKED);
3402 SetDictInt("BUTTON4_DOUBLE_CLICKED", BUTTON4_DOUBLE_CLICKED);
3403 SetDictInt("BUTTON4_TRIPLE_CLICKED", BUTTON4_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003404
Victor Stinner26486ea2010-05-15 22:23:53 +00003405 SetDictInt("BUTTON_SHIFT", BUTTON_SHIFT);
3406 SetDictInt("BUTTON_CTRL", BUTTON_CTRL);
3407 SetDictInt("BUTTON_ALT", BUTTON_ALT);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003408
Victor Stinner26486ea2010-05-15 22:23:53 +00003409 SetDictInt("ALL_MOUSE_EVENTS", ALL_MOUSE_EVENTS);
3410 SetDictInt("REPORT_MOUSE_POSITION", REPORT_MOUSE_POSITION);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00003411#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003412 /* Now set everything up for KEY_ variables */
3413 {
3414 int key;
3415 char *key_n;
3416 char *key_n2;
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003417#if !defined(__NetBSD__)
Victor Stinner26486ea2010-05-15 22:23:53 +00003418 for (key=KEY_MIN;key < KEY_MAX; key++) {
3419 key_n = (char *)keyname(key);
3420 if (key_n == NULL || strcmp(key_n,"UNKNOWN KEY")==0)
3421 continue;
3422 if (strncmp(key_n,"KEY_F(",6)==0) {
3423 char *p1, *p2;
3424 key_n2 = malloc(strlen(key_n)+1);
3425 if (!key_n2) {
3426 PyErr_NoMemory();
3427 break;
3428 }
3429 p1 = key_n;
3430 p2 = key_n2;
3431 while (*p1) {
3432 if (*p1 != '(' && *p1 != ')') {
3433 *p2 = *p1;
3434 p2++;
3435 }
3436 p1++;
3437 }
3438 *p2 = (char)0;
3439 } else
3440 key_n2 = key_n;
3441 SetDictInt(key_n2,key);
3442 if (key_n2 != key_n)
3443 free(key_n2);
3444 }
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00003445#endif
Victor Stinner26486ea2010-05-15 22:23:53 +00003446 SetDictInt("KEY_MIN", KEY_MIN);
3447 SetDictInt("KEY_MAX", KEY_MAX);
3448 }
3449 return m;
Guido van Rossumf6971e21994-08-30 12:25:20 +00003450}