blob: 04107ac299a7f690c63d405f01437219ec60462b [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. Kuchling22b88ce2000-05-23 16:18:03 +000034/* CVS: $Id$ */
Guido van Rossumf6971e21994-08-30 12:25:20 +000035
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000036/*
37
Victor Stinnerc22fd672010-05-15 22:30:53 +000038 A number of SysV or ncurses functions don't have wrappers yet; if you
39 need a given function, add it and send a patch. See
40 http://www.python.org/dev/patches/ for instructions on how to submit
41 patches to Python.
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000042
Victor Stinnerc22fd672010-05-15 22:30:53 +000043 Here's a list of currently unsupported functions:
Guido van Rossumd8faa362007-04-27 19:54:29 +000044
Victor Stinnerc22fd672010-05-15 22:30:53 +000045 addchnstr addchstr color_set define_key
46 del_curterm delscreen dupwin inchnstr inchstr innstr keyok
47 mcprint mvaddchnstr mvaddchstr mvcur mvinchnstr
48 mvinchstr mvinnstr mmvwaddchnstr mvwaddchstr
49 mvwinchnstr mvwinchstr mvwinnstr newterm
50 restartterm ripoffline scr_dump
51 scr_init scr_restore scr_set scrl set_curterm set_term setterm
52 tgetent tgetflag tgetnum tgetstr tgoto timeout tputs
53 vidattr vidputs waddchnstr waddchstr
54 wcolor_set winchnstr winchstr winnstr wmouse_trafo wscrl
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000055
Victor Stinnerc22fd672010-05-15 22:30:53 +000056 Low-priority:
57 slk_attr slk_attr_off slk_attr_on slk_attr_set slk_attroff
58 slk_attron slk_attrset slk_clear slk_color slk_init slk_label
59 slk_noutrefresh slk_refresh slk_restore slk_set slk_touch
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000060
Victor Stinnerc22fd672010-05-15 22:30:53 +000061 Menu extension (ncurses and probably SYSV):
62 current_item free_item free_menu item_count item_description
63 item_index item_init item_name item_opts item_opts_off
64 item_opts_on item_term item_userptr item_value item_visible
65 menu_back menu_driver menu_fore menu_format menu_grey
66 menu_init menu_items menu_mark menu_opts menu_opts_off
67 menu_opts_on menu_pad menu_pattern menu_request_by_name
68 menu_request_name menu_spacing menu_sub menu_term menu_userptr
69 menu_win new_item new_menu pos_menu_cursor post_menu
70 scale_menu set_current_item set_item_init set_item_opts
71 set_item_term set_item_userptr set_item_value set_menu_back
72 set_menu_fore set_menu_format set_menu_grey set_menu_init
73 set_menu_items set_menu_mark set_menu_opts set_menu_pad
74 set_menu_pattern set_menu_spacing set_menu_sub set_menu_term
75 set_menu_userptr set_menu_win set_top_row top_row unpost_menu
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +000076
Victor Stinnerc22fd672010-05-15 22:30:53 +000077 Form extension (ncurses and probably SYSV):
78 current_field data_ahead data_behind dup_field
79 dynamic_fieldinfo field_arg field_back field_buffer
80 field_count field_fore field_index field_info field_init
81 field_just field_opts field_opts_off field_opts_on field_pad
82 field_status field_term field_type field_userptr form_driver
83 form_fields form_init form_opts form_opts_off form_opts_on
84 form_page form_request_by_name form_request_name form_sub
85 form_term form_userptr form_win free_field free_form
86 link_field link_fieldtype move_field new_field new_form
87 new_page pos_form_cursor post_form scale_form
88 set_current_field set_field_back set_field_buffer
89 set_field_fore set_field_init set_field_just set_field_opts
90 set_field_pad set_field_status set_field_term set_field_type
91 set_field_userptr set_fieldtype_arg set_fieldtype_choice
92 set_form_fields set_form_init set_form_opts set_form_page
93 set_form_sub set_form_term set_form_userptr set_form_win
94 set_max_field set_new_page unpost_form
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +000095
96
Victor Stinnerc22fd672010-05-15 22:30:53 +000097*/
Andrew M. Kuchlingfb6bbf52000-07-27 11:58:01 +000098
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +000099/* Release Number */
Guido van Rossumf6971e21994-08-30 12:25:20 +0000100
Andrew M. Kuchling16e65a02001-10-20 16:05:52 +0000101char *PyCursesVersion = "2.2";
Guido van Rossumf6971e21994-08-30 12:25:20 +0000102
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000103/* Includes */
Guido van Rossumf6971e21994-08-30 12:25:20 +0000104
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
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +0000108#ifdef __osf__
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +0000109#define STRICT_SYSV_CURSES /* Don't use ncurses extensions */
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +0000110#endif
111
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000112#ifdef __hpux
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000113#define STRICT_SYSV_CURSES
114#endif
115
Guido van Rossum858ca0f2001-04-10 19:53:37 +0000116#define CURSES_MODULE
117#include "py_curses.h"
118
Victor Stinnerc22fd672010-05-15 22:30:53 +0000119/* These prototypes are in <term.h>, but including this header
120 #defines many common symbols (such as "lines") which breaks the
121 curses module in other ways. So the code will just specify
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +0000122 explicit prototypes here. */
123extern int setupterm(char *,int,int *);
Fred Drake0368bc42001-07-19 20:48:32 +0000124#ifdef __sgi
125#include <term.h>
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +0000126#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
150/* Utility Macros */
Victor Stinnerc22fd672010-05-15 22:30:53 +0000151#define PyCursesSetupTermCalled \
152 if (initialised_setupterm != TRUE) { \
153 PyErr_SetString(PyCursesError, \
154 "must call (at least) setupterm() first"); \
155 return 0; }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +0000156
Victor Stinnerc22fd672010-05-15 22:30:53 +0000157#define PyCursesInitialised \
158 if (initialised != TRUE) { \
159 PyErr_SetString(PyCursesError, \
160 "must call initscr() first"); \
161 return 0; }
Guido van Rossumfbea2f31994-08-31 22:05:27 +0000162
Victor Stinnerc22fd672010-05-15 22:30:53 +0000163#define PyCursesInitialisedColor \
164 if (initialisedcolors != TRUE) { \
165 PyErr_SetString(PyCursesError, \
166 "must call start_color() first"); \
167 return 0; }
Guido van Rossumfbea2f31994-08-31 22:05:27 +0000168
Andrew M. Kuchling52a14c32005-06-09 17:53:27 +0000169#ifndef MIN
170#define MIN(x,y) ((x) < (y) ? (x) : (y))
171#endif
172
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000173/* Utility Functions */
Guido van Rossumfbea2f31994-08-31 22:05:27 +0000174
Guido van Rossum85738471995-02-17 13:50:17 +0000175/*
Victor Stinnerc22fd672010-05-15 22:30:53 +0000176 * Check the return code from a curses function and return None
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000177 * or raise an exception as appropriate. These are exported using the
Victor Stinnerc22fd672010-05-15 22:30:53 +0000178 * capsule API.
Guido van Rossum85738471995-02-17 13:50:17 +0000179 */
180
181static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000182PyCursesCheckERR(int code, char *fname)
Guido van Rossum85738471995-02-17 13:50:17 +0000183{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000184 if (code != ERR) {
185 Py_INCREF(Py_None);
186 return Py_None;
Guido van Rossum85738471995-02-17 13:50:17 +0000187 } else {
Victor Stinnerc22fd672010-05-15 22:30:53 +0000188 if (fname == NULL) {
189 PyErr_SetString(PyCursesError, catchall_ERR);
190 } else {
191 PyErr_Format(PyCursesError, "%s() returned ERR", fname);
192 }
193 return NULL;
Guido van Rossum85738471995-02-17 13:50:17 +0000194 }
Guido van Rossum85738471995-02-17 13:50:17 +0000195}
196
Victor Stinnerc22fd672010-05-15 22:30:53 +0000197static int
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000198PyCurses_ConvertToChtype(PyObject *obj, chtype *ch)
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000199{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000200 if (PyLong_CheckExact(obj)) {
201 int overflow;
202 /* XXX should the truncation by the cast also be reported
203 as an error? */
204 *ch = (chtype) PyLong_AsLongAndOverflow(obj, &overflow);
205 if (overflow)
206 return 0;
207 } else if(PyBytes_Check(obj)
208 && (PyBytes_Size(obj) == 1)) {
209 *ch = (chtype) *PyBytes_AsString(obj);
210 } else if (PyUnicode_Check(obj) && PyUnicode_GetSize(obj) == 1) {
211 *ch = (chtype) *PyUnicode_AS_UNICODE(obj);
212 } else {
213 return 0;
214 }
215 return 1;
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000216}
217
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000218/* Function versions of the 3 functions for tested whether curses has been
219 initialised or not. */
Victor Stinnerc22fd672010-05-15 22:30:53 +0000220
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000221static int func_PyCursesSetupTermCalled(void)
222{
223 PyCursesSetupTermCalled;
224 return 1;
225}
226
227static int func_PyCursesInitialised(void)
228{
229 PyCursesInitialised;
230 return 1;
231}
232
233static int func_PyCursesInitialisedColor(void)
234{
235 PyCursesInitialisedColor;
236 return 1;
237}
238
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000239/*****************************************************************************
240 The Window Object
241******************************************************************************/
Guido van Rossum585c6dc1995-02-24 13:45:43 +0000242
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000243/* Definition of the window type */
Guido van Rossum85738471995-02-17 13:50:17 +0000244
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000245PyTypeObject PyCursesWindow_Type;
246
Andrew M. Kuchling32552682000-12-22 21:52:27 +0000247/* Function prototype macros for Window object
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000248
249 X - function name
250 TYPE - parameter Type
251 ERGSTR - format string for construction of the return value
252 PARSESTR - format string for argument parsing
Victor Stinnerc22fd672010-05-15 22:30:53 +0000253*/
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000254
Victor Stinnerc22fd672010-05-15 22:30:53 +0000255#define Window_NoArgNoReturnFunction(X) \
256 static PyObject *PyCursesWindow_ ## X (PyCursesWindowObject *self, PyObject *args) \
257 { return PyCursesCheckERR(X(self->win), # X); }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000258
Victor Stinnerc22fd672010-05-15 22:30:53 +0000259#define Window_NoArgTrueFalseFunction(X) \
260 static PyObject * PyCursesWindow_ ## X (PyCursesWindowObject *self) \
261 { \
262 if (X (self->win) == FALSE) { Py_INCREF(Py_False); return Py_False; } \
263 else { Py_INCREF(Py_True); return Py_True; } }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000264
Victor Stinnerc22fd672010-05-15 22:30:53 +0000265#define Window_NoArgNoReturnVoidFunction(X) \
266 static PyObject * PyCursesWindow_ ## X (PyCursesWindowObject *self) \
267 { \
268 X(self->win); Py_INCREF(Py_None); return Py_None; }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000269
Victor Stinnerc22fd672010-05-15 22:30:53 +0000270#define Window_NoArg2TupleReturnFunction(X, TYPE, ERGSTR) \
271 static PyObject * PyCursesWindow_ ## X (PyCursesWindowObject *self) \
272 { \
273 TYPE arg1, arg2; \
274 X(self->win,arg1,arg2); return Py_BuildValue(ERGSTR, arg1, arg2); }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000275
Victor Stinnerc22fd672010-05-15 22:30:53 +0000276#define Window_OneArgNoReturnVoidFunction(X, TYPE, PARSESTR) \
277 static PyObject * PyCursesWindow_ ## X (PyCursesWindowObject *self, PyObject *args) \
278 { \
279 TYPE arg1; \
280 if (!PyArg_ParseTuple(args, PARSESTR, &arg1)) return NULL; \
281 X(self->win,arg1); Py_INCREF(Py_None); return Py_None; }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000282
Victor Stinnerc22fd672010-05-15 22:30:53 +0000283#define Window_OneArgNoReturnFunction(X, TYPE, PARSESTR) \
284 static PyObject * PyCursesWindow_ ## X (PyCursesWindowObject *self, PyObject *args) \
285 { \
286 TYPE arg1; \
287 if (!PyArg_ParseTuple(args,PARSESTR, &arg1)) return NULL; \
288 return PyCursesCheckERR(X(self->win, arg1), # X); }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000289
Victor Stinnerc22fd672010-05-15 22:30:53 +0000290#define Window_TwoArgNoReturnFunction(X, TYPE, PARSESTR) \
291 static PyObject * PyCursesWindow_ ## X (PyCursesWindowObject *self, PyObject *args) \
292 { \
293 TYPE arg1, arg2; \
294 if (!PyArg_ParseTuple(args,PARSESTR, &arg1, &arg2)) return NULL; \
295 return PyCursesCheckERR(X(self->win, arg1, arg2), # X); }
Guido van Rossum85738471995-02-17 13:50:17 +0000296
Guido van Rossumf6971e21994-08-30 12:25:20 +0000297/* ------------- WINDOW routines --------------- */
Guido van Rossum85738471995-02-17 13:50:17 +0000298
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000299Window_NoArgNoReturnFunction(untouchwin)
300Window_NoArgNoReturnFunction(touchwin)
301Window_NoArgNoReturnFunction(redrawwin)
302Window_NoArgNoReturnFunction(winsertln)
303Window_NoArgNoReturnFunction(werase)
304Window_NoArgNoReturnFunction(wdeleteln)
305
306Window_NoArgTrueFalseFunction(is_wintouched)
307
308Window_NoArgNoReturnVoidFunction(wsyncup)
309Window_NoArgNoReturnVoidFunction(wsyncdown)
310Window_NoArgNoReturnVoidFunction(wstandend)
311Window_NoArgNoReturnVoidFunction(wstandout)
312Window_NoArgNoReturnVoidFunction(wcursyncup)
313Window_NoArgNoReturnVoidFunction(wclrtoeol)
314Window_NoArgNoReturnVoidFunction(wclrtobot)
315Window_NoArgNoReturnVoidFunction(wclear)
316
317Window_OneArgNoReturnVoidFunction(idcok, int, "i;True(1) or False(0)")
318Window_OneArgNoReturnVoidFunction(immedok, int, "i;True(1) or False(0)")
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +0000319Window_OneArgNoReturnVoidFunction(wtimeout, int, "i;delay")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000320
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000321Window_NoArg2TupleReturnFunction(getyx, int, "ii")
322Window_NoArg2TupleReturnFunction(getbegyx, int, "ii")
323Window_NoArg2TupleReturnFunction(getmaxyx, int, "ii")
324Window_NoArg2TupleReturnFunction(getparyx, int, "ii")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000325
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000326Window_OneArgNoReturnFunction(clearok, int, "i;True(1) or False(0)")
327Window_OneArgNoReturnFunction(idlok, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000328#if defined(__NetBSD__)
329Window_OneArgNoReturnVoidFunction(keypad, int, "i;True(1) or False(0)")
330#else
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000331Window_OneArgNoReturnFunction(keypad, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000332#endif
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000333Window_OneArgNoReturnFunction(leaveok, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000334#if defined(__NetBSD__)
335Window_OneArgNoReturnVoidFunction(nodelay, int, "i;True(1) or False(0)")
336#else
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000337Window_OneArgNoReturnFunction(nodelay, int, "i;True(1) or False(0)")
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000338#endif
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000339Window_OneArgNoReturnFunction(notimeout, int, "i;True(1) or False(0)")
340Window_OneArgNoReturnFunction(scrollok, int, "i;True(1) or False(0)")
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +0000341Window_OneArgNoReturnFunction(winsdelln, int, "i;nlines")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000342Window_OneArgNoReturnFunction(syncok, int, "i;True(1) or False(0)")
343
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000344Window_TwoArgNoReturnFunction(mvwin, int, "ii;y,x")
345Window_TwoArgNoReturnFunction(mvderwin, int, "ii;y,x")
346Window_TwoArgNoReturnFunction(wmove, int, "ii;y,x")
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +0000347#ifndef STRICT_SYSV_CURSES
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000348Window_TwoArgNoReturnFunction(wresize, int, "ii;lines,columns")
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000349#endif
350
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +0000351/* Allocation and deallocation of Window Objects */
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000352
Guido van Rossumf6971e21994-08-30 12:25:20 +0000353static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000354PyCursesWindow_New(WINDOW *win)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000355{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000356 PyCursesWindowObject *wo;
Guido van Rossum85738471995-02-17 13:50:17 +0000357
Victor Stinnerc22fd672010-05-15 22:30:53 +0000358 wo = PyObject_NEW(PyCursesWindowObject, &PyCursesWindow_Type);
359 if (wo == NULL) return NULL;
360 wo->win = win;
361 return (PyObject *)wo;
Guido van Rossumf6971e21994-08-30 12:25:20 +0000362}
363
364static void
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000365PyCursesWindow_Dealloc(PyCursesWindowObject *wo)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000366{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000367 if (wo->win != stdscr) delwin(wo->win);
368 PyObject_DEL(wo);
Guido van Rossumf6971e21994-08-30 12:25:20 +0000369}
370
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000371/* Addch, Addstr, Addnstr */
Guido van Rossumf6971e21994-08-30 12:25:20 +0000372
373static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000374PyCursesWindow_AddCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000375{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000376 int rtn, x, y, use_xy = FALSE;
377 PyObject *temp;
378 chtype ch = 0;
379 attr_t attr = A_NORMAL;
380 long lattr;
Guido van Rossum85738471995-02-17 13:50:17 +0000381
Victor Stinnerc22fd672010-05-15 22:30:53 +0000382 switch (PyTuple_Size(args)) {
383 case 1:
384 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
385 return NULL;
386 break;
387 case 2:
388 if (!PyArg_ParseTuple(args, "Ol;ch or int,attr", &temp, &lattr))
389 return NULL;
390 attr = lattr;
391 break;
392 case 3:
393 if (!PyArg_ParseTuple(args,"iiO;y,x,ch or int", &y, &x, &temp))
394 return NULL;
395 use_xy = TRUE;
396 break;
397 case 4:
398 if (!PyArg_ParseTuple(args,"iiOl;y,x,ch or int, attr",
399 &y, &x, &temp, &lattr))
400 return NULL;
401 attr = lattr;
402 use_xy = TRUE;
403 break;
404 default:
405 PyErr_SetString(PyExc_TypeError, "addch requires 1 to 4 arguments");
406 return NULL;
407 }
408
409 if (!PyCurses_ConvertToChtype(temp, &ch)) {
410 PyErr_SetString(PyExc_TypeError, "argument 1 or 3 must be a ch or an int");
411 return NULL;
412 }
413
414 if (use_xy == TRUE)
415 rtn = mvwaddch(self->win,y,x, ch | attr);
416 else {
417 rtn = waddch(self->win, ch | attr);
418 }
419 return PyCursesCheckERR(rtn, "addch");
Guido van Rossumf6971e21994-08-30 12:25:20 +0000420}
421
422static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000423PyCursesWindow_AddStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000424{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000425 int rtn;
426 int x, y;
427 char *str;
428 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
429 long lattr;
430 int use_xy = FALSE, use_attr = FALSE;
Guido van Rossum85738471995-02-17 13:50:17 +0000431
Victor Stinnerc22fd672010-05-15 22:30:53 +0000432 switch (PyTuple_Size(args)) {
433 case 1:
434 if (!PyArg_ParseTuple(args,"s;str", &str))
435 return NULL;
436 break;
437 case 2:
438 if (!PyArg_ParseTuple(args,"sl;str,attr", &str, &lattr))
439 return NULL;
440 attr = lattr;
441 use_attr = TRUE;
442 break;
443 case 3:
444 if (!PyArg_ParseTuple(args,"iis;int,int,str", &y, &x, &str))
445 return NULL;
446 use_xy = TRUE;
447 break;
448 case 4:
449 if (!PyArg_ParseTuple(args,"iisl;int,int,str,attr", &y, &x, &str, &lattr))
450 return NULL;
451 attr = lattr;
452 use_xy = use_attr = TRUE;
453 break;
454 default:
455 PyErr_SetString(PyExc_TypeError, "addstr requires 1 to 4 arguments");
456 return NULL;
457 }
Guido van Rossum85738471995-02-17 13:50:17 +0000458
Victor Stinnerc22fd672010-05-15 22:30:53 +0000459 if (use_attr == TRUE) {
460 attr_old = getattrs(self->win);
461 (void)wattrset(self->win,attr);
462 }
463 if (use_xy == TRUE)
464 rtn = mvwaddstr(self->win,y,x,str);
465 else
466 rtn = waddstr(self->win,str);
467 if (use_attr == TRUE)
468 (void)wattrset(self->win,attr_old);
469 return PyCursesCheckERR(rtn, "addstr");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000470}
Guido van Rossum85738471995-02-17 13:50:17 +0000471
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000472static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000473PyCursesWindow_AddNStr(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000474{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000475 int rtn, x, y, n;
476 char *str;
477 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
478 long lattr;
479 int use_xy = FALSE, use_attr = FALSE;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000480
Victor Stinnerc22fd672010-05-15 22:30:53 +0000481 switch (PyTuple_Size(args)) {
482 case 2:
483 if (!PyArg_ParseTuple(args,"si;str,n", &str, &n))
484 return NULL;
485 break;
486 case 3:
487 if (!PyArg_ParseTuple(args,"sil;str,n,attr", &str, &n, &lattr))
488 return NULL;
489 attr = lattr;
490 use_attr = TRUE;
491 break;
492 case 4:
493 if (!PyArg_ParseTuple(args,"iisi;y,x,str,n", &y, &x, &str, &n))
494 return NULL;
495 use_xy = TRUE;
496 break;
497 case 5:
498 if (!PyArg_ParseTuple(args,"iisil;y,x,str,n,attr", &y, &x, &str, &n, &lattr))
499 return NULL;
500 attr = lattr;
501 use_xy = use_attr = TRUE;
502 break;
503 default:
504 PyErr_SetString(PyExc_TypeError, "addnstr requires 2 to 5 arguments");
505 return NULL;
506 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000507
Victor Stinnerc22fd672010-05-15 22:30:53 +0000508 if (use_attr == TRUE) {
509 attr_old = getattrs(self->win);
510 (void)wattrset(self->win,attr);
511 }
512 if (use_xy == TRUE)
513 rtn = mvwaddnstr(self->win,y,x,str,n);
514 else
515 rtn = waddnstr(self->win,str,n);
516 if (use_attr == TRUE)
517 (void)wattrset(self->win,attr_old);
518 return PyCursesCheckERR(rtn, "addnstr");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000519}
520
521static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000522PyCursesWindow_Bkgd(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000523{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000524 PyObject *temp;
525 chtype bkgd;
526 attr_t attr = A_NORMAL;
527 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000528
Victor Stinnerc22fd672010-05-15 22:30:53 +0000529 switch (PyTuple_Size(args)) {
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000530 case 1:
Victor Stinnerc22fd672010-05-15 22:30:53 +0000531 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
532 return NULL;
533 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000534 case 2:
Victor Stinnerc22fd672010-05-15 22:30:53 +0000535 if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))
536 return NULL;
537 attr = lattr;
538 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000539 default:
Victor Stinnerc22fd672010-05-15 22:30:53 +0000540 PyErr_SetString(PyExc_TypeError, "bkgd requires 1 or 2 arguments");
541 return NULL;
542 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000543
Victor Stinnerc22fd672010-05-15 22:30:53 +0000544 if (!PyCurses_ConvertToChtype(temp, &bkgd)) {
545 PyErr_SetString(PyExc_TypeError, "argument 1 or 3 must be a ch or an int");
546 return NULL;
547 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000548
Victor Stinnerc22fd672010-05-15 22:30:53 +0000549 return PyCursesCheckERR(wbkgd(self->win, bkgd | attr), "bkgd");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000550}
551
552static PyObject *
Christian Heimes2380ac72008-01-09 00:17:24 +0000553PyCursesWindow_AttrOff(PyCursesWindowObject *self, PyObject *args)
554{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000555 long lattr;
556 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
557 return NULL;
558 return PyCursesCheckERR(wattroff(self->win, (attr_t)lattr), "attroff");
Christian Heimes2380ac72008-01-09 00:17:24 +0000559}
560
561static PyObject *
562PyCursesWindow_AttrOn(PyCursesWindowObject *self, PyObject *args)
563{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000564 long lattr;
565 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
566 return NULL;
567 return PyCursesCheckERR(wattron(self->win, (attr_t)lattr), "attron");
Christian Heimes2380ac72008-01-09 00:17:24 +0000568}
569
570static PyObject *
571PyCursesWindow_AttrSet(PyCursesWindowObject *self, PyObject *args)
572{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000573 long lattr;
574 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
575 return NULL;
576 return PyCursesCheckERR(wattrset(self->win, (attr_t)lattr), "attrset");
Christian Heimes2380ac72008-01-09 00:17:24 +0000577}
578
579static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000580PyCursesWindow_BkgdSet(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000581{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000582 PyObject *temp;
583 chtype bkgd;
584 attr_t attr = A_NORMAL;
585 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000586
Victor Stinnerc22fd672010-05-15 22:30:53 +0000587 switch (PyTuple_Size(args)) {
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000588 case 1:
Victor Stinnerc22fd672010-05-15 22:30:53 +0000589 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
590 return NULL;
591 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000592 case 2:
Victor Stinnerc22fd672010-05-15 22:30:53 +0000593 if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))
594 return NULL;
595 attr = lattr;
596 break;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000597 default:
Victor Stinnerc22fd672010-05-15 22:30:53 +0000598 PyErr_SetString(PyExc_TypeError, "bkgdset requires 1 or 2 arguments");
599 return NULL;
600 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000601
Victor Stinnerc22fd672010-05-15 22:30:53 +0000602 if (!PyCurses_ConvertToChtype(temp, &bkgd)) {
603 PyErr_SetString(PyExc_TypeError, "argument 1 must be a ch or an int");
604 return NULL;
605 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000606
Victor Stinnerc22fd672010-05-15 22:30:53 +0000607 wbkgdset(self->win, bkgd | attr);
608 return PyCursesCheckERR(0, "bkgdset");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000609}
610
611static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000612PyCursesWindow_Border(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000613{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000614 PyObject *temp[8];
615 chtype ch[8];
616 int i;
Andrew M. Kuchling16e65a02001-10-20 16:05:52 +0000617
Victor Stinnerc22fd672010-05-15 22:30:53 +0000618 /* Clear the array of parameters */
619 for(i=0; i<8; i++) {
620 temp[i] = NULL;
621 ch[i] = 0;
622 }
Andrew M. Kuchling16e65a02001-10-20 16:05:52 +0000623
Victor Stinnerc22fd672010-05-15 22:30:53 +0000624 if (!PyArg_ParseTuple(args,"|OOOOOOOO;ls,rs,ts,bs,tl,tr,bl,br",
625 &temp[0], &temp[1], &temp[2], &temp[3],
626 &temp[4], &temp[5], &temp[6], &temp[7]))
627 return NULL;
628
629 for(i=0; i<8; i++) {
630 if (temp[i] != NULL && !PyCurses_ConvertToChtype(temp[i], &ch[i])) {
631 PyErr_Format(PyExc_TypeError,
632 "argument %i must be a ch or an int", i+1);
633 return NULL;
634 }
635 }
636
637 wborder(self->win,
638 ch[0], ch[1], ch[2], ch[3],
639 ch[4], ch[5], ch[6], ch[7]);
640 Py_INCREF(Py_None);
641 return Py_None;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000642}
643
644static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000645PyCursesWindow_Box(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000646{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000647 chtype ch1=0,ch2=0;
648 switch(PyTuple_Size(args)){
649 case 0: break;
650 default:
651 if (!PyArg_ParseTuple(args,"ll;vertint,horint", &ch1, &ch2))
652 return NULL;
653 }
654 box(self->win,ch1,ch2);
655 Py_INCREF(Py_None);
656 return Py_None;
Guido van Rossumf6971e21994-08-30 12:25:20 +0000657}
658
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000659#if defined(HAVE_NCURSES_H) || defined(MVWDELCH_IS_EXPRESSION)
660#define py_mvwdelch mvwdelch
661#else
662int py_mvwdelch(WINDOW *w, int y, int x)
663{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000664 mvwdelch(w,y,x);
665 /* On HP/UX, mvwdelch already returns. On other systems,
666 we may well run into this return statement. */
667 return 0;
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000668}
669#endif
670
Guido van Rossumd8faa362007-04-27 19:54:29 +0000671/* chgat, added by Fabian Kreutz <fabian.kreutz at gmx.net> */
672
673static PyObject *
674PyCursesWindow_ChgAt(PyCursesWindowObject *self, PyObject *args)
675{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000676 int rtn;
677 int x, y;
678 int num = -1;
679 short color;
680 attr_t attr = A_NORMAL;
681 long lattr;
682 int use_xy = FALSE;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000683
Victor Stinnerc22fd672010-05-15 22:30:53 +0000684 switch (PyTuple_Size(args)) {
685 case 1:
686 if (!PyArg_ParseTuple(args,"l;attr", &lattr))
687 return NULL;
688 attr = lattr;
689 break;
690 case 2:
691 if (!PyArg_ParseTuple(args,"il;n,attr", &num, &lattr))
692 return NULL;
693 attr = lattr;
694 break;
695 case 3:
696 if (!PyArg_ParseTuple(args,"iil;int,int,attr", &y, &x, &lattr))
697 return NULL;
698 attr = lattr;
699 use_xy = TRUE;
700 break;
701 case 4:
702 if (!PyArg_ParseTuple(args,"iiil;int,int,n,attr", &y, &x, &num, &lattr))
703 return NULL;
704 attr = lattr;
705 use_xy = TRUE;
706 break;
707 default:
708 PyErr_SetString(PyExc_TypeError, "chgat requires 1 to 4 arguments");
709 return NULL;
710 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000711
Victor Stinnerc22fd672010-05-15 22:30:53 +0000712 color = (short)((attr >> 8) & 0xff);
713 attr = attr - (color << 8);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000714
Victor Stinnerc22fd672010-05-15 22:30:53 +0000715 if (use_xy == TRUE) {
716 rtn = mvwchgat(self->win,y,x,num,attr,color,NULL);
717 touchline(self->win,y,1);
718 } else {
719 getyx(self->win,y,x);
720 rtn = wchgat(self->win,num,attr,color,NULL);
721 touchline(self->win,y,1);
722 }
723 return PyCursesCheckERR(rtn, "chgat");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000724}
725
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000726
Guido van Rossumf6971e21994-08-30 12:25:20 +0000727static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000728PyCursesWindow_DelCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000729{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000730 int rtn;
731 int x, y;
Guido van Rossum85738471995-02-17 13:50:17 +0000732
Victor Stinnerc22fd672010-05-15 22:30:53 +0000733 switch (PyTuple_Size(args)) {
734 case 0:
735 rtn = wdelch(self->win);
736 break;
737 case 2:
738 if (!PyArg_ParseTuple(args,"ii;y,x", &y, &x))
739 return NULL;
740 rtn = py_mvwdelch(self->win,y,x);
741 break;
742 default:
743 PyErr_SetString(PyExc_TypeError, "delch requires 0 or 2 arguments");
744 return NULL;
745 }
746 return PyCursesCheckERR(rtn, "[mv]wdelch");
Guido van Rossumf6971e21994-08-30 12:25:20 +0000747}
748
749static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000750PyCursesWindow_DerWin(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000751{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000752 WINDOW *win;
753 int nlines, ncols, begin_y, begin_x;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000754
Victor Stinnerc22fd672010-05-15 22:30:53 +0000755 nlines = 0;
756 ncols = 0;
757 switch (PyTuple_Size(args)) {
758 case 2:
759 if (!PyArg_ParseTuple(args,"ii;begin_y,begin_x",&begin_y,&begin_x))
760 return NULL;
761 break;
762 case 4:
763 if (!PyArg_ParseTuple(args, "iiii;nlines,ncols,begin_y,begin_x",
764 &nlines,&ncols,&begin_y,&begin_x))
765 return NULL;
766 break;
767 default:
768 PyErr_SetString(PyExc_TypeError, "derwin requires 2 or 4 arguments");
769 return NULL;
770 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000771
Victor Stinnerc22fd672010-05-15 22:30:53 +0000772 win = derwin(self->win,nlines,ncols,begin_y,begin_x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000773
Victor Stinnerc22fd672010-05-15 22:30:53 +0000774 if (win == NULL) {
775 PyErr_SetString(PyCursesError, catchall_NULL);
776 return NULL;
777 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000778
Victor Stinnerc22fd672010-05-15 22:30:53 +0000779 return (PyObject *)PyCursesWindow_New(win);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000780}
781
782static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000783PyCursesWindow_EchoChar(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000784{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000785 PyObject *temp;
786 chtype ch;
787 attr_t attr = A_NORMAL;
788 long lattr;
Guido van Rossum85738471995-02-17 13:50:17 +0000789
Victor Stinnerc22fd672010-05-15 22:30:53 +0000790 switch (PyTuple_Size(args)) {
791 case 1:
792 if (!PyArg_ParseTuple(args,"O;ch or int", &temp))
793 return NULL;
794 break;
795 case 2:
796 if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))
797 return NULL;
798 attr = lattr;
799 break;
800 default:
801 PyErr_SetString(PyExc_TypeError, "echochar requires 1 or 2 arguments");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000802
803
Victor Stinnerc22fd672010-05-15 22:30:53 +0000804 return NULL;
805 }
Guido van Rossum85738471995-02-17 13:50:17 +0000806
Victor Stinnerc22fd672010-05-15 22:30:53 +0000807 if (!PyCurses_ConvertToChtype(temp, &ch)) {
808 PyErr_SetString(PyExc_TypeError, "argument 1 must be a ch or an int");
809 return NULL;
810 }
811
Martin v. Löwiseb9b1032001-10-24 17:10:49 +0000812#ifdef WINDOW_HAS_FLAGS
Victor Stinnerc22fd672010-05-15 22:30:53 +0000813 if (self->win->_flags & _ISPAD)
814 return PyCursesCheckERR(pechochar(self->win, ch | attr),
815 "echochar");
816 else
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000817#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +0000818 return PyCursesCheckERR(wechochar(self->win, ch | attr),
819 "echochar");
Guido van Rossumf6971e21994-08-30 12:25:20 +0000820}
821
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +0000822#ifdef NCURSES_MOUSE_VERSION
823static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000824PyCursesWindow_Enclose(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +0000825{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000826 int x, y;
827 if (!PyArg_ParseTuple(args,"ii;y,x", &y, &x))
828 return NULL;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +0000829
Victor Stinnerc22fd672010-05-15 22:30:53 +0000830 return PyLong_FromLong( wenclose(self->win,y,x) );
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +0000831}
832#endif
833
Guido van Rossumf6971e21994-08-30 12:25:20 +0000834static PyObject *
Martin v. Löwisc0e16712002-01-17 23:08:27 +0000835PyCursesWindow_GetBkgd(PyCursesWindowObject *self)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000836{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000837 return PyLong_FromLong((long) getbkgd(self->win));
Guido van Rossumf6971e21994-08-30 12:25:20 +0000838}
839
840static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000841PyCursesWindow_GetCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000842{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000843 int x, y;
844 int rtn;
Guido van Rossum85738471995-02-17 13:50:17 +0000845
Victor Stinnerc22fd672010-05-15 22:30:53 +0000846 switch (PyTuple_Size(args)) {
847 case 0:
848 Py_BEGIN_ALLOW_THREADS
849 rtn = wgetch(self->win);
850 Py_END_ALLOW_THREADS
851 break;
852 case 2:
853 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
854 return NULL;
855 Py_BEGIN_ALLOW_THREADS
856 rtn = mvwgetch(self->win,y,x);
857 Py_END_ALLOW_THREADS
858 break;
859 default:
860 PyErr_SetString(PyExc_TypeError, "getch requires 0 or 2 arguments");
861 return NULL;
862 }
863 return PyLong_FromLong((long)rtn);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000864}
Guido van Rossum85738471995-02-17 13:50:17 +0000865
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000866static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000867PyCursesWindow_GetKey(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000868{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000869 int x, y;
870 int rtn;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000871
Victor Stinnerc22fd672010-05-15 22:30:53 +0000872 switch (PyTuple_Size(args)) {
873 case 0:
874 Py_BEGIN_ALLOW_THREADS
875 rtn = wgetch(self->win);
876 Py_END_ALLOW_THREADS
877 break;
878 case 2:
879 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
880 return NULL;
881 Py_BEGIN_ALLOW_THREADS
882 rtn = mvwgetch(self->win,y,x);
883 Py_END_ALLOW_THREADS
884 break;
885 default:
886 PyErr_SetString(PyExc_TypeError, "getkey requires 0 or 2 arguments");
887 return NULL;
888 }
889 if (rtn == ERR) {
890 /* getch() returns ERR in nodelay mode */
891 PyErr_SetString(PyCursesError, "no input");
892 return NULL;
893 } else if (rtn<=255) {
894 return Py_BuildValue("C", rtn);
895 } else {
896 const char *knp;
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000897#if defined(__NetBSD__)
Victor Stinnerc22fd672010-05-15 22:30:53 +0000898 knp = unctrl(rtn);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000899#else
Victor Stinnerc22fd672010-05-15 22:30:53 +0000900 knp = keyname(rtn);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +0000901#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +0000902 return PyUnicode_FromString((knp == NULL) ? "" : knp);
903 }
Guido van Rossumf6971e21994-08-30 12:25:20 +0000904}
905
906static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000907PyCursesWindow_GetStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +0000908{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000909 int x, y, n;
910 char rtn[1024]; /* This should be big enough.. I hope */
911 int rtn2;
Guido van Rossum85738471995-02-17 13:50:17 +0000912
Victor Stinnerc22fd672010-05-15 22:30:53 +0000913 switch (PyTuple_Size(args)) {
914 case 0:
915 Py_BEGIN_ALLOW_THREADS
916 rtn2 = wgetnstr(self->win,rtn, 1023);
917 Py_END_ALLOW_THREADS
918 break;
919 case 1:
920 if (!PyArg_ParseTuple(args,"i;n", &n))
921 return NULL;
922 Py_BEGIN_ALLOW_THREADS
923 rtn2 = wgetnstr(self->win,rtn,MIN(n, 1023));
924 Py_END_ALLOW_THREADS
925 break;
926 case 2:
927 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
928 return NULL;
929 Py_BEGIN_ALLOW_THREADS
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000930#ifdef STRICT_SYSV_CURSES
Victor Stinnerc22fd672010-05-15 22:30:53 +0000931 rtn2 = wmove(self->win,y,x)==ERR ? ERR : wgetnstr(self->win, rtn, 1023);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000932#else
Victor Stinnerc22fd672010-05-15 22:30:53 +0000933 rtn2 = mvwgetnstr(self->win,y,x,rtn, 1023);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000934#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +0000935 Py_END_ALLOW_THREADS
936 break;
937 case 3:
938 if (!PyArg_ParseTuple(args,"iii;y,x,n", &y, &x, &n))
939 return NULL;
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +0000940#ifdef STRICT_SYSV_CURSES
Victor Stinnerc22fd672010-05-15 22:30:53 +0000941 Py_BEGIN_ALLOW_THREADS
942 rtn2 = wmove(self->win,y,x)==ERR ? ERR :
943 wgetnstr(self->win, rtn, MIN(n, 1023));
944 Py_END_ALLOW_THREADS
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000945#else
Victor Stinnerc22fd672010-05-15 22:30:53 +0000946 Py_BEGIN_ALLOW_THREADS
947 rtn2 = mvwgetnstr(self->win, y, x, rtn, MIN(n, 1023));
948 Py_END_ALLOW_THREADS
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000949#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +0000950 break;
951 default:
952 PyErr_SetString(PyExc_TypeError, "getstr requires 0 to 3 arguments");
953 return NULL;
954 }
955 if (rtn2 == ERR)
956 rtn[0] = 0;
957 return PyBytes_FromString(rtn);
Guido van Rossumf6971e21994-08-30 12:25:20 +0000958}
959
960static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +0000961PyCursesWindow_Hline(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000962{
Victor Stinnerc22fd672010-05-15 22:30:53 +0000963 PyObject *temp;
964 chtype ch;
965 int n, x, y, code = OK;
966 attr_t attr = A_NORMAL;
967 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000968
Victor Stinnerc22fd672010-05-15 22:30:53 +0000969 switch (PyTuple_Size(args)) {
970 case 2:
971 if (!PyArg_ParseTuple(args, "Oi;ch or int,n", &temp, &n))
972 return NULL;
973 break;
974 case 3:
975 if (!PyArg_ParseTuple(args, "Oil;ch or int,n,attr", &temp, &n, &lattr))
976 return NULL;
977 attr = lattr;
978 break;
979 case 4:
980 if (!PyArg_ParseTuple(args, "iiOi;y,x,ch or int,n", &y, &x, &temp, &n))
981 return NULL;
982 code = wmove(self->win, y, x);
983 break;
984 case 5:
985 if (!PyArg_ParseTuple(args, "iiOil; y,x,ch or int,n,attr",
986 &y, &x, &temp, &n, &lattr))
987 return NULL;
988 attr = lattr;
989 code = wmove(self->win, y, x);
990 break;
991 default:
992 PyErr_SetString(PyExc_TypeError, "hline requires 2 to 5 arguments");
993 return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +0000994 }
Victor Stinnerc22fd672010-05-15 22:30:53 +0000995
996 if (code != ERR) {
997 if (!PyCurses_ConvertToChtype(temp, &ch)) {
998 PyErr_SetString(PyExc_TypeError,
999 "argument 1 or 3 must be a ch or an int");
1000 return NULL;
1001 }
1002 return PyCursesCheckERR(whline(self->win, ch | attr, n), "hline");
1003 } else
1004 return PyCursesCheckERR(code, "wmove");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001005}
1006
1007static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001008PyCursesWindow_InsCh(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001009{
Victor Stinnerc22fd672010-05-15 22:30:53 +00001010 int rtn, x, y, use_xy = FALSE;
1011 PyObject *temp;
1012 chtype ch = 0;
1013 attr_t attr = A_NORMAL;
1014 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001015
Victor Stinnerc22fd672010-05-15 22:30:53 +00001016 switch (PyTuple_Size(args)) {
1017 case 1:
1018 if (!PyArg_ParseTuple(args, "O;ch or int", &temp))
1019 return NULL;
1020 break;
1021 case 2:
1022 if (!PyArg_ParseTuple(args, "Ol;ch or int,attr", &temp, &lattr))
1023 return NULL;
1024 attr = lattr;
1025 break;
1026 case 3:
1027 if (!PyArg_ParseTuple(args,"iiO;y,x,ch or int", &y, &x, &temp))
1028 return NULL;
1029 use_xy = TRUE;
1030 break;
1031 case 4:
1032 if (!PyArg_ParseTuple(args,"iiOl;y,x,ch or int, attr", &y, &x, &temp, &lattr))
1033 return NULL;
1034 attr = lattr;
1035 use_xy = TRUE;
1036 break;
1037 default:
1038 PyErr_SetString(PyExc_TypeError, "insch requires 1 or 4 arguments");
1039 return NULL;
1040 }
1041
1042 if (!PyCurses_ConvertToChtype(temp, &ch)) {
1043 PyErr_SetString(PyExc_TypeError,
1044 "argument 1 or 3 must be a ch or an int");
1045 return NULL;
1046 }
1047
1048 if (use_xy == TRUE)
1049 rtn = mvwinsch(self->win,y,x, ch | attr);
1050 else {
1051 rtn = winsch(self->win, ch | attr);
1052 }
1053 return PyCursesCheckERR(rtn, "insch");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001054}
1055
1056static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001057PyCursesWindow_InCh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001058{
Victor Stinnerc22fd672010-05-15 22:30:53 +00001059 int x, y, rtn;
Guido van Rossum85738471995-02-17 13:50:17 +00001060
Victor Stinnerc22fd672010-05-15 22:30:53 +00001061 switch (PyTuple_Size(args)) {
1062 case 0:
1063 rtn = winch(self->win);
1064 break;
1065 case 2:
1066 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1067 return NULL;
1068 rtn = mvwinch(self->win,y,x);
1069 break;
1070 default:
1071 PyErr_SetString(PyExc_TypeError, "inch requires 0 or 2 arguments");
1072 return NULL;
1073 }
1074 return PyLong_FromLong((long) rtn);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001075}
1076
1077static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001078PyCursesWindow_InStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001079{
Victor Stinnerc22fd672010-05-15 22:30:53 +00001080 int x, y, n;
1081 char rtn[1024]; /* This should be big enough.. I hope */
1082 int rtn2;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001083
Victor Stinnerc22fd672010-05-15 22:30:53 +00001084 switch (PyTuple_Size(args)) {
1085 case 0:
1086 rtn2 = winnstr(self->win,rtn, 1023);
1087 break;
1088 case 1:
1089 if (!PyArg_ParseTuple(args,"i;n", &n))
1090 return NULL;
1091 rtn2 = winnstr(self->win,rtn,MIN(n,1023));
1092 break;
1093 case 2:
1094 if (!PyArg_ParseTuple(args,"ii;y,x",&y,&x))
1095 return NULL;
1096 rtn2 = mvwinnstr(self->win,y,x,rtn,1023);
1097 break;
1098 case 3:
1099 if (!PyArg_ParseTuple(args, "iii;y,x,n", &y, &x, &n))
1100 return NULL;
1101 rtn2 = mvwinnstr(self->win, y, x, rtn, MIN(n,1023));
1102 break;
1103 default:
1104 PyErr_SetString(PyExc_TypeError, "instr requires 0 or 3 arguments");
1105 return NULL;
1106 }
1107 if (rtn2 == ERR)
1108 rtn[0] = 0;
1109 return PyBytes_FromString(rtn);
Guido van Rossumf6971e21994-08-30 12:25:20 +00001110}
1111
1112static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001113PyCursesWindow_InsStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001114{
Victor Stinnerc22fd672010-05-15 22:30:53 +00001115 int rtn;
1116 int x, y;
1117 char *str;
1118 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
1119 long lattr;
1120 int use_xy = FALSE, use_attr = FALSE;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001121
Victor Stinnerc22fd672010-05-15 22:30:53 +00001122 switch (PyTuple_Size(args)) {
1123 case 1:
1124 if (!PyArg_ParseTuple(args,"s;str", &str))
1125 return NULL;
1126 break;
1127 case 2:
1128 if (!PyArg_ParseTuple(args,"sl;str,attr", &str, &lattr))
1129 return NULL;
1130 attr = lattr;
1131 use_attr = TRUE;
1132 break;
1133 case 3:
1134 if (!PyArg_ParseTuple(args,"iis;y,x,str", &y, &x, &str))
1135 return NULL;
1136 use_xy = TRUE;
1137 break;
1138 case 4:
1139 if (!PyArg_ParseTuple(args,"iisl;y,x,str,attr", &y, &x, &str, &lattr))
1140 return NULL;
1141 attr = lattr;
1142 use_xy = use_attr = TRUE;
1143 break;
1144 default:
1145 PyErr_SetString(PyExc_TypeError, "insstr requires 1 to 4 arguments");
1146 return NULL;
1147 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001148
Victor Stinnerc22fd672010-05-15 22:30:53 +00001149 if (use_attr == TRUE) {
1150 attr_old = getattrs(self->win);
1151 (void)wattrset(self->win,attr);
1152 }
1153 if (use_xy == TRUE)
1154 rtn = mvwinsstr(self->win,y,x,str);
1155 else
1156 rtn = winsstr(self->win,str);
1157 if (use_attr == TRUE)
1158 (void)wattrset(self->win,attr_old);
1159 return PyCursesCheckERR(rtn, "insstr");
Guido van Rossumf6971e21994-08-30 12:25:20 +00001160}
1161
1162static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001163PyCursesWindow_InsNStr(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001164{
Victor Stinnerc22fd672010-05-15 22:30:53 +00001165 int rtn, x, y, n;
1166 char *str;
1167 attr_t attr = A_NORMAL , attr_old = A_NORMAL;
1168 long lattr;
1169 int use_xy = FALSE, use_attr = FALSE;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001170
Victor Stinnerc22fd672010-05-15 22:30:53 +00001171 switch (PyTuple_Size(args)) {
1172 case 2:
1173 if (!PyArg_ParseTuple(args,"si;str,n", &str, &n))
1174 return NULL;
1175 break;
1176 case 3:
1177 if (!PyArg_ParseTuple(args,"sil;str,n,attr", &str, &n, &lattr))
1178 return NULL;
1179 attr = lattr;
1180 use_attr = TRUE;
1181 break;
1182 case 4:
1183 if (!PyArg_ParseTuple(args,"iisi;y,x,str,n", &y, &x, &str, &n))
1184 return NULL;
1185 use_xy = TRUE;
1186 break;
1187 case 5:
1188 if (!PyArg_ParseTuple(args,"iisil;y,x,str,n,attr", &y, &x, &str, &n, &lattr))
1189 return NULL;
1190 attr = lattr;
1191 use_xy = use_attr = TRUE;
1192 break;
1193 default:
1194 PyErr_SetString(PyExc_TypeError, "insnstr requires 2 to 5 arguments");
1195 return NULL;
1196 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001197
Victor Stinnerc22fd672010-05-15 22:30:53 +00001198 if (use_attr == TRUE) {
1199 attr_old = getattrs(self->win);
1200 (void)wattrset(self->win,attr);
1201 }
1202 if (use_xy == TRUE)
1203 rtn = mvwinsnstr(self->win,y,x,str,n);
1204 else
1205 rtn = winsnstr(self->win,str,n);
1206 if (use_attr == TRUE)
1207 (void)wattrset(self->win,attr_old);
1208 return PyCursesCheckERR(rtn, "insnstr");
Guido van Rossumf6971e21994-08-30 12:25:20 +00001209}
1210
1211static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001212PyCursesWindow_Is_LineTouched(PyCursesWindowObject *self, PyObject *args)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001213{
Victor Stinnerc22fd672010-05-15 22:30:53 +00001214 int line, erg;
1215 if (!PyArg_ParseTuple(args,"i;line", &line))
1216 return NULL;
1217 erg = is_linetouched(self->win, line);
1218 if (erg == ERR) {
1219 PyErr_SetString(PyExc_TypeError,
1220 "is_linetouched: line number outside of boundaries");
1221 return NULL;
1222 } else
1223 if (erg == FALSE) {
1224 Py_INCREF(Py_False);
1225 return Py_False;
1226 } else {
1227 Py_INCREF(Py_True);
1228 return Py_True;
1229 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001230}
1231
1232static PyObject *
Andrew M. Kuchlingad9b1f82000-07-09 14:35:00 +00001233PyCursesWindow_NoOutRefresh(PyCursesWindowObject *self, PyObject *args)
Guido van Rossumf6971e21994-08-30 12:25:20 +00001234{
Victor Stinnerc22fd672010-05-15 22:30:53 +00001235 int pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol;
1236 int rtn;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001237
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00001238#ifndef WINDOW_HAS_FLAGS
Victor Stinnerc22fd672010-05-15 22:30:53 +00001239 if (0) {
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001240#else
Victor Stinnerc22fd672010-05-15 22:30:53 +00001241 if (self->win->_flags & _ISPAD) {
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001242#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00001243 switch(PyTuple_Size(args)) {
1244 case 6:
1245 if (!PyArg_ParseTuple(args,
1246 "iiiiii;" \
1247 "pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol",
1248 &pminrow, &pmincol, &sminrow,
1249 &smincol, &smaxrow, &smaxcol))
1250 return NULL;
1251 Py_BEGIN_ALLOW_THREADS
1252 rtn = pnoutrefresh(self->win,
1253 pminrow, pmincol, sminrow,
1254 smincol, smaxrow, smaxcol);
1255 Py_END_ALLOW_THREADS
1256 return PyCursesCheckERR(rtn, "pnoutrefresh");
1257 default:
1258 PyErr_SetString(PyCursesError,
1259 "noutrefresh() called for a pad "
1260 "requires 6 arguments");
1261 return NULL;
1262 }
1263 } else {
1264 if (!PyArg_ParseTuple(args, ":noutrefresh"))
1265 return NULL;
Andrew M. Kuchlingf4195722000-06-23 01:36:21 +00001266
Victor Stinnerc22fd672010-05-15 22:30:53 +00001267 Py_BEGIN_ALLOW_THREADS
1268 rtn = wnoutrefresh(self->win);
1269 Py_END_ALLOW_THREADS
1270 return PyCursesCheckERR(rtn, "wnoutrefresh");
1271 }
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001272 }
1273
Victor Stinnerc22fd672010-05-15 22:30:53 +00001274 static PyObject *
1275 PyCursesWindow_Overlay(PyCursesWindowObject *self, PyObject *args)
1276 {
1277 PyCursesWindowObject *temp;
1278 int use_copywin = FALSE;
1279 int sminrow, smincol, dminrow, dmincol, dmaxrow, dmaxcol;
1280 int rtn;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001281
Victor Stinnerc22fd672010-05-15 22:30:53 +00001282 switch (PyTuple_Size(args)) {
1283 case 1:
1284 if (!PyArg_ParseTuple(args, "O!;window object",
1285 &PyCursesWindow_Type, &temp))
1286 return NULL;
1287 break;
1288 case 7:
1289 if (!PyArg_ParseTuple(args, "O!iiiiii;window object, int, int, int, int, int, int",
1290 &PyCursesWindow_Type, &temp, &sminrow, &smincol,
1291 &dminrow, &dmincol, &dmaxrow, &dmaxcol))
1292 return NULL;
1293 use_copywin = TRUE;
1294 break;
1295 default:
1296 PyErr_SetString(PyExc_TypeError,
1297 "overlay requires one or seven arguments");
1298 return NULL;
1299 }
1300
1301 if (use_copywin == TRUE) {
1302 rtn = copywin(self->win, temp->win, sminrow, smincol,
1303 dminrow, dmincol, dmaxrow, dmaxcol, TRUE);
1304 return PyCursesCheckERR(rtn, "copywin");
1305 }
1306 else {
1307 rtn = overlay(self->win, temp->win);
1308 return PyCursesCheckERR(rtn, "overlay");
1309 }
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001310 }
1311
Victor Stinnerc22fd672010-05-15 22:30:53 +00001312 static PyObject *
1313 PyCursesWindow_Overwrite(PyCursesWindowObject *self, PyObject *args)
1314 {
1315 PyCursesWindowObject *temp;
1316 int use_copywin = FALSE;
1317 int sminrow, smincol, dminrow, dmincol, dmaxrow, dmaxcol;
1318 int rtn;
1319
1320 switch (PyTuple_Size(args)) {
1321 case 1:
1322 if (!PyArg_ParseTuple(args, "O!;window object",
1323 &PyCursesWindow_Type, &temp))
1324 return NULL;
1325 break;
1326 case 7:
1327 if (!PyArg_ParseTuple(args, "O!iiiiii;window object, int, int, int, int, int, int",
1328 &PyCursesWindow_Type, &temp, &sminrow, &smincol,
1329 &dminrow, &dmincol, &dmaxrow, &dmaxcol))
1330 return NULL;
1331 use_copywin = TRUE;
1332 break;
1333 default:
1334 PyErr_SetString(PyExc_TypeError,
1335 "overwrite requires one or seven arguments");
1336 return NULL;
1337 }
1338
1339 if (use_copywin == TRUE) {
1340 rtn = copywin(self->win, temp->win, sminrow, smincol,
1341 dminrow, dmincol, dmaxrow, dmaxcol, FALSE);
1342 return PyCursesCheckERR(rtn, "copywin");
1343 }
1344 else {
1345 rtn = overwrite(self->win, temp->win);
1346 return PyCursesCheckERR(rtn, "overwrite");
1347 }
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001348 }
Victor Stinnerc22fd672010-05-15 22:30:53 +00001349
1350 static PyObject *
1351 PyCursesWindow_PutWin(PyCursesWindowObject *self, PyObject *stream)
1352 {
1353 /* We have to simulate this by writing to a temporary FILE*,
1354 then reading back, then writing to the argument stream. */
1355 char fn[100];
1356 int fd;
1357 FILE *fp;
1358 PyObject *res;
1359
1360 strcpy(fn, "/tmp/py.curses.putwin.XXXXXX");
1361 fd = mkstemp(fn);
1362 if (fd < 0)
1363 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
1364 fp = fdopen(fd, "wb+");
1365 if (fp == NULL) {
1366 close(fd);
1367 remove(fn);
1368 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
1369 }
1370 res = PyCursesCheckERR(putwin(self->win, fp), "putwin");
1371 if (res == NULL) {
1372 fclose(fp);
1373 remove(fn);
1374 return res;
1375 }
1376 fseek(fp, 0, 0);
1377 while (1) {
1378 char buf[BUFSIZ];
1379 int n = fread(buf, 1, BUFSIZ, fp);
1380 if (n <= 0)
1381 break;
1382 Py_DECREF(res);
1383 res = PyObject_CallMethod(stream, "write", "y#", buf, n);
1384 if (res == NULL)
1385 break;
1386 }
1387 fclose(fp);
1388 remove(fn);
1389 return res;
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001390 }
Andrew M. Kuchling9de6ffa2000-12-21 16:22:22 +00001391
Victor Stinnerc22fd672010-05-15 22:30:53 +00001392 static PyObject *
1393 PyCursesWindow_RedrawLine(PyCursesWindowObject *self, PyObject *args)
1394 {
1395 int beg, num;
1396 if (!PyArg_ParseTuple(args, "ii;beg,num", &beg, &num))
1397 return NULL;
1398 return PyCursesCheckERR(wredrawln(self->win,beg,num), "redrawln");
1399 }
Guido van Rossumda5b8f22007-06-12 23:30:11 +00001400
Victor Stinnerc22fd672010-05-15 22:30:53 +00001401 static PyObject *
1402 PyCursesWindow_Refresh(PyCursesWindowObject *self, PyObject *args)
1403 {
1404 int pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol;
1405 int rtn;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001406
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00001407#ifndef WINDOW_HAS_FLAGS
Victor Stinnerc22fd672010-05-15 22:30:53 +00001408 if (0) {
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001409#else
Victor Stinnerc22fd672010-05-15 22:30:53 +00001410 if (self->win->_flags & _ISPAD) {
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001411#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00001412 switch(PyTuple_Size(args)) {
1413 case 6:
1414 if (!PyArg_ParseTuple(args,
1415 "iiiiii;" \
1416 "pminrow,pmincol,sminrow,smincol,smaxrow,smaxcol",
1417 &pminrow, &pmincol, &sminrow,
1418 &smincol, &smaxrow, &smaxcol))
1419 return NULL;
Andrew M. Kuchlingf4195722000-06-23 01:36:21 +00001420
Victor Stinnerc22fd672010-05-15 22:30:53 +00001421 Py_BEGIN_ALLOW_THREADS
1422 rtn = prefresh(self->win,
1423 pminrow, pmincol, sminrow,
1424 smincol, smaxrow, smaxcol);
1425 Py_END_ALLOW_THREADS
1426 return PyCursesCheckERR(rtn, "prefresh");
1427 default:
1428 PyErr_SetString(PyCursesError,
1429 "refresh() for a pad requires 6 arguments");
1430 return NULL;
1431 }
1432 } else {
1433 if (!PyArg_ParseTuple(args, ":refresh"))
1434 return NULL;
1435 Py_BEGIN_ALLOW_THREADS
1436 rtn = wrefresh(self->win);
1437 Py_END_ALLOW_THREADS
1438 return PyCursesCheckERR(rtn, "prefresh");
1439 }
1440 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001441
Victor Stinnerc22fd672010-05-15 22:30:53 +00001442 static PyObject *
1443 PyCursesWindow_SetScrollRegion(PyCursesWindowObject *self, PyObject *args)
1444 {
1445 int x, y;
1446 if (!PyArg_ParseTuple(args,"ii;top, bottom",&y,&x))
1447 return NULL;
1448 return PyCursesCheckERR(wsetscrreg(self->win,y,x), "wsetscrreg");
1449 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001450
Victor Stinnerc22fd672010-05-15 22:30:53 +00001451 static PyObject *
1452 PyCursesWindow_SubWin(PyCursesWindowObject *self, PyObject *args)
1453 {
1454 WINDOW *win;
1455 int nlines, ncols, begin_y, begin_x;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001456
Victor Stinnerc22fd672010-05-15 22:30:53 +00001457 nlines = 0;
1458 ncols = 0;
1459 switch (PyTuple_Size(args)) {
1460 case 2:
1461 if (!PyArg_ParseTuple(args,"ii;begin_y,begin_x",&begin_y,&begin_x))
1462 return NULL;
1463 break;
1464 case 4:
1465 if (!PyArg_ParseTuple(args, "iiii;nlines,ncols,begin_y,begin_x",
1466 &nlines,&ncols,&begin_y,&begin_x))
1467 return NULL;
1468 break;
1469 default:
1470 PyErr_SetString(PyExc_TypeError, "subwin requires 2 or 4 arguments");
1471 return NULL;
1472 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001473
Victor Stinnerc22fd672010-05-15 22:30:53 +00001474 /* printf("Subwin: %i %i %i %i \n", nlines, ncols, begin_y, begin_x); */
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00001475#ifdef WINDOW_HAS_FLAGS
Victor Stinnerc22fd672010-05-15 22:30:53 +00001476 if (self->win->_flags & _ISPAD)
1477 win = subpad(self->win, nlines, ncols, begin_y, begin_x);
1478 else
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00001479#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00001480 win = subwin(self->win, nlines, ncols, begin_y, begin_x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001481
Victor Stinnerc22fd672010-05-15 22:30:53 +00001482 if (win == NULL) {
1483 PyErr_SetString(PyCursesError, catchall_NULL);
1484 return NULL;
1485 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001486
Victor Stinnerc22fd672010-05-15 22:30:53 +00001487 return (PyObject *)PyCursesWindow_New(win);
1488 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001489
Victor Stinnerc22fd672010-05-15 22:30:53 +00001490 static PyObject *
1491 PyCursesWindow_Scroll(PyCursesWindowObject *self, PyObject *args)
1492 {
1493 int nlines;
1494 switch(PyTuple_Size(args)) {
1495 case 0:
1496 return PyCursesCheckERR(scroll(self->win), "scroll");
1497 case 1:
1498 if (!PyArg_ParseTuple(args, "i;nlines", &nlines))
1499 return NULL;
1500 return PyCursesCheckERR(wscrl(self->win, nlines), "scroll");
1501 default:
1502 PyErr_SetString(PyExc_TypeError, "scroll requires 0 or 1 arguments");
1503 return NULL;
1504 }
1505 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001506
Victor Stinnerc22fd672010-05-15 22:30:53 +00001507 static PyObject *
1508 PyCursesWindow_TouchLine(PyCursesWindowObject *self, PyObject *args)
1509 {
1510 int st, cnt, val;
1511 switch (PyTuple_Size(args)) {
1512 case 2:
1513 if (!PyArg_ParseTuple(args,"ii;start,count",&st,&cnt))
1514 return NULL;
1515 return PyCursesCheckERR(touchline(self->win,st,cnt), "touchline");
1516 case 3:
1517 if (!PyArg_ParseTuple(args, "iii;start,count,val", &st, &cnt, &val))
1518 return NULL;
1519 return PyCursesCheckERR(wtouchln(self->win, st, cnt, val), "touchline");
1520 default:
1521 PyErr_SetString(PyExc_TypeError, "touchline requires 2 or 3 arguments");
1522 return NULL;
1523 }
1524 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001525
Victor Stinnerc22fd672010-05-15 22:30:53 +00001526 static PyObject *
1527 PyCursesWindow_Vline(PyCursesWindowObject *self, PyObject *args)
1528 {
1529 PyObject *temp;
1530 chtype ch;
1531 int n, x, y, code = OK;
1532 attr_t attr = A_NORMAL;
1533 long lattr;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001534
Victor Stinnerc22fd672010-05-15 22:30:53 +00001535 switch (PyTuple_Size(args)) {
1536 case 2:
1537 if (!PyArg_ParseTuple(args, "Oi;ch or int,n", &temp, &n))
1538 return NULL;
1539 break;
1540 case 3:
1541 if (!PyArg_ParseTuple(args, "Oil;ch or int,n,attr", &temp, &n, &lattr))
1542 return NULL;
1543 attr = lattr;
1544 break;
1545 case 4:
1546 if (!PyArg_ParseTuple(args, "iiOi;y,x,ch or int,n", &y, &x, &temp, &n))
1547 return NULL;
1548 code = wmove(self->win, y, x);
1549 break;
1550 case 5:
1551 if (!PyArg_ParseTuple(args, "iiOil; y,x,ch or int,n,attr",
1552 &y, &x, &temp, &n, &lattr))
1553 return NULL;
1554 attr = lattr;
1555 code = wmove(self->win, y, x);
1556 break;
1557 default:
1558 PyErr_SetString(PyExc_TypeError, "vline requires 2 to 5 arguments");
1559 return NULL;
1560 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00001561
Victor Stinnerc22fd672010-05-15 22:30:53 +00001562 if (code != ERR) {
1563 if (!PyCurses_ConvertToChtype(temp, &ch)) {
1564 PyErr_SetString(PyExc_TypeError,
1565 "argument 1 or 3 must be a ch or an int");
1566 return NULL;
1567 }
1568 return PyCursesCheckERR(wvline(self->win, ch | attr, n), "vline");
1569 } else
1570 return PyCursesCheckERR(code, "wmove");
1571 }
1572
1573 static PyMethodDef PyCursesWindow_Methods[] = {
1574 {"addch", (PyCFunction)PyCursesWindow_AddCh, METH_VARARGS},
1575 {"addnstr", (PyCFunction)PyCursesWindow_AddNStr, METH_VARARGS},
1576 {"addstr", (PyCFunction)PyCursesWindow_AddStr, METH_VARARGS},
1577 {"attroff", (PyCFunction)PyCursesWindow_AttrOff, METH_VARARGS},
1578 {"attron", (PyCFunction)PyCursesWindow_AttrOn, METH_VARARGS},
1579 {"attrset", (PyCFunction)PyCursesWindow_AttrSet, METH_VARARGS},
1580 {"bkgd", (PyCFunction)PyCursesWindow_Bkgd, METH_VARARGS},
1581 {"chgat", (PyCFunction)PyCursesWindow_ChgAt, METH_VARARGS},
1582 {"bkgdset", (PyCFunction)PyCursesWindow_BkgdSet, METH_VARARGS},
1583 {"border", (PyCFunction)PyCursesWindow_Border, METH_VARARGS},
1584 {"box", (PyCFunction)PyCursesWindow_Box, METH_VARARGS},
1585 {"clear", (PyCFunction)PyCursesWindow_wclear, METH_NOARGS},
1586 {"clearok", (PyCFunction)PyCursesWindow_clearok, METH_VARARGS},
1587 {"clrtobot", (PyCFunction)PyCursesWindow_wclrtobot, METH_NOARGS},
1588 {"clrtoeol", (PyCFunction)PyCursesWindow_wclrtoeol, METH_NOARGS},
1589 {"cursyncup", (PyCFunction)PyCursesWindow_wcursyncup, METH_NOARGS},
1590 {"delch", (PyCFunction)PyCursesWindow_DelCh, METH_VARARGS},
1591 {"deleteln", (PyCFunction)PyCursesWindow_wdeleteln, METH_NOARGS},
1592 {"derwin", (PyCFunction)PyCursesWindow_DerWin, METH_VARARGS},
1593 {"echochar", (PyCFunction)PyCursesWindow_EchoChar, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001594#ifdef NCURSES_MOUSE_VERSION
Victor Stinnerc22fd672010-05-15 22:30:53 +00001595 {"enclose", (PyCFunction)PyCursesWindow_Enclose, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001596#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00001597 {"erase", (PyCFunction)PyCursesWindow_werase, METH_NOARGS},
1598 {"getbegyx", (PyCFunction)PyCursesWindow_getbegyx, METH_NOARGS},
1599 {"getbkgd", (PyCFunction)PyCursesWindow_GetBkgd, METH_NOARGS},
1600 {"getch", (PyCFunction)PyCursesWindow_GetCh, METH_VARARGS},
1601 {"getkey", (PyCFunction)PyCursesWindow_GetKey, METH_VARARGS},
1602 {"getmaxyx", (PyCFunction)PyCursesWindow_getmaxyx, METH_NOARGS},
1603 {"getparyx", (PyCFunction)PyCursesWindow_getparyx, METH_NOARGS},
1604 {"getstr", (PyCFunction)PyCursesWindow_GetStr, METH_VARARGS},
1605 {"getyx", (PyCFunction)PyCursesWindow_getyx, METH_NOARGS},
1606 {"hline", (PyCFunction)PyCursesWindow_Hline, METH_VARARGS},
1607 {"idcok", (PyCFunction)PyCursesWindow_idcok, METH_VARARGS},
1608 {"idlok", (PyCFunction)PyCursesWindow_idlok, METH_VARARGS},
1609 {"immedok", (PyCFunction)PyCursesWindow_immedok, METH_VARARGS},
1610 {"inch", (PyCFunction)PyCursesWindow_InCh, METH_VARARGS},
1611 {"insch", (PyCFunction)PyCursesWindow_InsCh, METH_VARARGS},
1612 {"insdelln", (PyCFunction)PyCursesWindow_winsdelln, METH_VARARGS},
1613 {"insertln", (PyCFunction)PyCursesWindow_winsertln, METH_NOARGS},
1614 {"insnstr", (PyCFunction)PyCursesWindow_InsNStr, METH_VARARGS},
1615 {"insstr", (PyCFunction)PyCursesWindow_InsStr, METH_VARARGS},
1616 {"instr", (PyCFunction)PyCursesWindow_InStr, METH_VARARGS},
1617 {"is_linetouched", (PyCFunction)PyCursesWindow_Is_LineTouched, METH_VARARGS},
1618 {"is_wintouched", (PyCFunction)PyCursesWindow_is_wintouched, METH_NOARGS},
1619 {"keypad", (PyCFunction)PyCursesWindow_keypad, METH_VARARGS},
1620 {"leaveok", (PyCFunction)PyCursesWindow_leaveok, METH_VARARGS},
1621 {"move", (PyCFunction)PyCursesWindow_wmove, METH_VARARGS},
1622 {"mvderwin", (PyCFunction)PyCursesWindow_mvderwin, METH_VARARGS},
1623 {"mvwin", (PyCFunction)PyCursesWindow_mvwin, METH_VARARGS},
1624 {"nodelay", (PyCFunction)PyCursesWindow_nodelay, METH_VARARGS},
1625 {"notimeout", (PyCFunction)PyCursesWindow_notimeout, METH_VARARGS},
1626 {"noutrefresh", (PyCFunction)PyCursesWindow_NoOutRefresh, METH_VARARGS},
1627 /* Backward compatibility alias -- remove in Python 2.3 */
1628 {"nooutrefresh", (PyCFunction)PyCursesWindow_NoOutRefresh, METH_VARARGS},
1629 {"overlay", (PyCFunction)PyCursesWindow_Overlay, METH_VARARGS},
1630 {"overwrite", (PyCFunction)PyCursesWindow_Overwrite,
1631 METH_VARARGS},
1632 {"putwin", (PyCFunction)PyCursesWindow_PutWin, METH_O},
1633 {"redrawln", (PyCFunction)PyCursesWindow_RedrawLine, METH_VARARGS},
1634 {"redrawwin", (PyCFunction)PyCursesWindow_redrawwin, METH_NOARGS},
1635 {"refresh", (PyCFunction)PyCursesWindow_Refresh, METH_VARARGS},
Andrew M. Kuchlinga1e4b052000-06-27 21:49:47 +00001636#ifndef STRICT_SYSV_CURSES
Victor Stinnerc22fd672010-05-15 22:30:53 +00001637 {"resize", (PyCFunction)PyCursesWindow_wresize, METH_VARARGS},
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001638#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00001639 {"scroll", (PyCFunction)PyCursesWindow_Scroll, METH_VARARGS},
1640 {"scrollok", (PyCFunction)PyCursesWindow_scrollok, METH_VARARGS},
1641 {"setscrreg", (PyCFunction)PyCursesWindow_SetScrollRegion, METH_VARARGS},
1642 {"standend", (PyCFunction)PyCursesWindow_wstandend, METH_NOARGS},
1643 {"standout", (PyCFunction)PyCursesWindow_wstandout, METH_NOARGS},
1644 {"subpad", (PyCFunction)PyCursesWindow_SubWin, METH_VARARGS},
1645 {"subwin", (PyCFunction)PyCursesWindow_SubWin, METH_VARARGS},
1646 {"syncdown", (PyCFunction)PyCursesWindow_wsyncdown, METH_NOARGS},
1647 {"syncok", (PyCFunction)PyCursesWindow_syncok, METH_VARARGS},
1648 {"syncup", (PyCFunction)PyCursesWindow_wsyncup, METH_NOARGS},
1649 {"timeout", (PyCFunction)PyCursesWindow_wtimeout, METH_VARARGS},
1650 {"touchline", (PyCFunction)PyCursesWindow_TouchLine, METH_VARARGS},
1651 {"touchwin", (PyCFunction)PyCursesWindow_touchwin, METH_NOARGS},
1652 {"untouchwin", (PyCFunction)PyCursesWindow_untouchwin, METH_NOARGS},
1653 {"vline", (PyCFunction)PyCursesWindow_Vline, METH_VARARGS},
1654 {NULL, NULL} /* sentinel */
1655 };
Guido van Rossumf6971e21994-08-30 12:25:20 +00001656
Guido van Rossumf6971e21994-08-30 12:25:20 +00001657/* -------------------------------------------------------*/
Guido van Rossum85738471995-02-17 13:50:17 +00001658
Victor Stinnerc22fd672010-05-15 22:30:53 +00001659 PyTypeObject PyCursesWindow_Type = {
1660 PyVarObject_HEAD_INIT(NULL, 0)
1661 "_curses.curses window", /*tp_name*/
1662 sizeof(PyCursesWindowObject), /*tp_basicsize*/
1663 0, /*tp_itemsize*/
1664 /* methods */
1665 (destructor)PyCursesWindow_Dealloc, /*tp_dealloc*/
1666 0, /*tp_print*/
1667 (getattrfunc)0, /*tp_getattr*/
1668 (setattrfunc)0, /*tp_setattr*/
1669 0, /*tp_reserved*/
1670 0, /*tp_repr*/
1671 0, /*tp_as_number*/
1672 0, /*tp_as_sequence*/
1673 0, /*tp_as_mapping*/
1674 0, /*tp_hash*/
1675 0, /*tp_call*/
1676 0, /*tp_str*/
1677 0, /*tp_getattro*/
1678 0, /*tp_setattro*/
1679 0, /*tp_as_buffer*/
1680 Py_TPFLAGS_DEFAULT, /*tp_flags*/
1681 0, /*tp_doc*/
1682 0, /*tp_traverse*/
1683 0, /*tp_clear*/
1684 0, /*tp_richcompare*/
1685 0, /*tp_weaklistoffset*/
1686 0, /*tp_iter*/
1687 0, /*tp_iternext*/
1688 PyCursesWindow_Methods, /*tp_methods*/
1689 };
Guido van Rossumf6971e21994-08-30 12:25:20 +00001690
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001691/*********************************************************************
1692 Global Functions
1693**********************************************************************/
Guido van Rossumf6971e21994-08-30 12:25:20 +00001694
Victor Stinnerc22fd672010-05-15 22:30:53 +00001695 NoArgNoReturnFunction(beep)
1696 NoArgNoReturnFunction(def_prog_mode)
1697 NoArgNoReturnFunction(def_shell_mode)
1698 NoArgNoReturnFunction(doupdate)
1699 NoArgNoReturnFunction(endwin)
1700 NoArgNoReturnFunction(flash)
1701 NoArgNoReturnFunction(nocbreak)
1702 NoArgNoReturnFunction(noecho)
1703 NoArgNoReturnFunction(nonl)
1704 NoArgNoReturnFunction(noraw)
1705 NoArgNoReturnFunction(reset_prog_mode)
1706 NoArgNoReturnFunction(reset_shell_mode)
1707 NoArgNoReturnFunction(resetty)
1708 NoArgNoReturnFunction(savetty)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001709
Victor Stinnerc22fd672010-05-15 22:30:53 +00001710 NoArgOrFlagNoReturnFunction(cbreak)
1711 NoArgOrFlagNoReturnFunction(echo)
1712 NoArgOrFlagNoReturnFunction(nl)
1713 NoArgOrFlagNoReturnFunction(raw)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001714
Victor Stinnerc22fd672010-05-15 22:30:53 +00001715 NoArgReturnIntFunction(baudrate)
1716 NoArgReturnIntFunction(termattrs)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001717
Victor Stinnerc22fd672010-05-15 22:30:53 +00001718 NoArgReturnStringFunction(termname)
1719 NoArgReturnStringFunction(longname)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001720
Victor Stinnerc22fd672010-05-15 22:30:53 +00001721 NoArgTrueFalseFunction(can_change_color)
1722 NoArgTrueFalseFunction(has_colors)
1723 NoArgTrueFalseFunction(has_ic)
1724 NoArgTrueFalseFunction(has_il)
1725 NoArgTrueFalseFunction(isendwin)
1726 NoArgNoReturnVoidFunction(flushinp)
1727 NoArgNoReturnVoidFunction(noqiflush)
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001728
Victor Stinnerc22fd672010-05-15 22:30:53 +00001729 static PyObject *
1730 PyCurses_filter(PyObject *self)
1731 {
1732 /* not checking for PyCursesInitialised here since filter() must
1733 be called before initscr() */
1734 filter();
1735 Py_INCREF(Py_None);
1736 return Py_None;
1737 }
Christian Heimesaf98da12008-01-27 15:18:18 +00001738
Victor Stinnerc22fd672010-05-15 22:30:53 +00001739 static PyObject *
1740 PyCurses_Color_Content(PyObject *self, PyObject *args)
1741 {
1742 short color,r,g,b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001743
Victor Stinnerc22fd672010-05-15 22:30:53 +00001744 PyCursesInitialised
1745 PyCursesInitialisedColor
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001746
Victor Stinnerc22fd672010-05-15 22:30:53 +00001747 if (!PyArg_ParseTuple(args, "h:color_content", &color)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001748
Victor Stinnerc22fd672010-05-15 22:30:53 +00001749 if (color_content(color, &r, &g, &b) != ERR)
1750 return Py_BuildValue("(iii)", r, g, b);
1751 else {
1752 PyErr_SetString(PyCursesError,
1753 "Argument 1 was out of range. Check value of COLORS.");
1754 return NULL;
1755 }
1756 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001757
Victor Stinnerc22fd672010-05-15 22:30:53 +00001758 static PyObject *
1759 PyCurses_color_pair(PyObject *self, PyObject *args)
1760 {
1761 int n;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001762
Victor Stinnerc22fd672010-05-15 22:30:53 +00001763 PyCursesInitialised
1764 PyCursesInitialisedColor
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001765
Victor Stinnerc22fd672010-05-15 22:30:53 +00001766 if (!PyArg_ParseTuple(args, "i:color_pair", &n)) return NULL;
1767 return PyLong_FromLong((long) (n << 8));
1768 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001769
Victor Stinnerc22fd672010-05-15 22:30:53 +00001770 static PyObject *
1771 PyCurses_Curs_Set(PyObject *self, PyObject *args)
1772 {
1773 int vis,erg;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001774
Victor Stinnerc22fd672010-05-15 22:30:53 +00001775 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001776
Victor Stinnerc22fd672010-05-15 22:30:53 +00001777 if (!PyArg_ParseTuple(args, "i:curs_set", &vis)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001778
Victor Stinnerc22fd672010-05-15 22:30:53 +00001779 erg = curs_set(vis);
1780 if (erg == ERR) return PyCursesCheckERR(erg, "curs_set");
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001781
Victor Stinnerc22fd672010-05-15 22:30:53 +00001782 return PyLong_FromLong((long) erg);
1783 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001784
Victor Stinnerc22fd672010-05-15 22:30:53 +00001785 static PyObject *
1786 PyCurses_Delay_Output(PyObject *self, PyObject *args)
1787 {
1788 int ms;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001789
Victor Stinnerc22fd672010-05-15 22:30:53 +00001790 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001791
Victor Stinnerc22fd672010-05-15 22:30:53 +00001792 if (!PyArg_ParseTuple(args, "i:delay_output", &ms)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001793
Victor Stinnerc22fd672010-05-15 22:30:53 +00001794 return PyCursesCheckERR(delay_output(ms), "delay_output");
1795 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001796
Victor Stinnerc22fd672010-05-15 22:30:53 +00001797 static PyObject *
1798 PyCurses_EraseChar(PyObject *self)
1799 {
1800 char ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001801
Victor Stinnerc22fd672010-05-15 22:30:53 +00001802 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001803
Victor Stinnerc22fd672010-05-15 22:30:53 +00001804 ch = erasechar();
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001805
Victor Stinnerc22fd672010-05-15 22:30:53 +00001806 return PyBytes_FromStringAndSize(&ch, 1);
1807 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001808
Victor Stinnerc22fd672010-05-15 22:30:53 +00001809 static PyObject *
1810 PyCurses_getsyx(PyObject *self)
1811 {
1812 int x = 0;
1813 int y = 0;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001814
Victor Stinnerc22fd672010-05-15 22:30:53 +00001815 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001816
Victor Stinnerc22fd672010-05-15 22:30:53 +00001817 getsyx(y, x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001818
Victor Stinnerc22fd672010-05-15 22:30:53 +00001819 return Py_BuildValue("(ii)", y, x);
1820 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001821
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001822#ifdef NCURSES_MOUSE_VERSION
Victor Stinnerc22fd672010-05-15 22:30:53 +00001823 static PyObject *
1824 PyCurses_GetMouse(PyObject *self)
1825 {
1826 int rtn;
1827 MEVENT event;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001828
Victor Stinnerc22fd672010-05-15 22:30:53 +00001829 PyCursesInitialised
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001830
Victor Stinnerc22fd672010-05-15 22:30:53 +00001831 rtn = getmouse( &event );
1832 if (rtn == ERR) {
1833 PyErr_SetString(PyCursesError, "getmouse() returned ERR");
1834 return NULL;
1835 }
1836 return Py_BuildValue("(hiiil)",
1837 (short)event.id,
1838 event.x, event.y, event.z,
1839 (long) event.bstate);
1840 }
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001841
Victor Stinnerc22fd672010-05-15 22:30:53 +00001842 static PyObject *
1843 PyCurses_UngetMouse(PyObject *self, PyObject *args)
1844 {
1845 MEVENT event;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001846
Victor Stinnerc22fd672010-05-15 22:30:53 +00001847 PyCursesInitialised
1848 if (!PyArg_ParseTuple(args, "hiiil",
1849 &event.id,
1850 &event.x, &event.y, &event.z,
1851 (int *) &event.bstate))
1852 return NULL;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001853
Victor Stinnerc22fd672010-05-15 22:30:53 +00001854 return PyCursesCheckERR(ungetmouse(&event), "ungetmouse");
1855 }
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00001856#endif
1857
Victor Stinnerc22fd672010-05-15 22:30:53 +00001858 static PyObject *
1859 PyCurses_GetWin(PyCursesWindowObject *self, PyObject *stream)
1860 {
1861 char fn[100];
1862 int fd;
1863 FILE *fp;
1864 PyObject *data;
1865 size_t datalen;
1866 WINDOW *win;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001867
Victor Stinnerc22fd672010-05-15 22:30:53 +00001868 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001869
Victor Stinnerc22fd672010-05-15 22:30:53 +00001870 strcpy(fn, "/tmp/py.curses.getwin.XXXXXX");
1871 fd = mkstemp(fn);
1872 if (fd < 0)
1873 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
1874 fp = fdopen(fd, "wb+");
1875 if (fp == NULL) {
1876 close(fd);
1877 remove(fn);
1878 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
1879 }
1880 data = PyObject_CallMethod(stream, "read", "");
1881 if (data == NULL) {
1882 fclose(fp);
1883 remove(fn);
1884 return NULL;
1885 }
1886 if (!PyBytes_Check(data)) {
1887 PyErr_Format(PyExc_TypeError,
1888 "f.read() returned %.100s instead of bytes",
1889 data->ob_type->tp_name);
1890 Py_DECREF(data);
1891 fclose(fp);
1892 remove(fn);
1893 return NULL;
1894 }
1895 datalen = PyBytes_GET_SIZE(data);
1896 if (fwrite(PyBytes_AS_STRING(data), 1, datalen, fp) != datalen) {
1897 Py_DECREF(data);
1898 fclose(fp);
1899 remove(fn);
1900 return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
1901 }
1902 Py_DECREF(data);
1903 fseek(fp, 0, 0);
1904 win = getwin(fp);
1905 fclose(fp);
1906 remove(fn);
1907 if (win == NULL) {
1908 PyErr_SetString(PyCursesError, catchall_NULL);
1909 return NULL;
1910 }
1911 return PyCursesWindow_New(win);
1912 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001913
Victor Stinnerc22fd672010-05-15 22:30:53 +00001914 static PyObject *
1915 PyCurses_HalfDelay(PyObject *self, PyObject *args)
1916 {
1917 unsigned char tenths;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001918
Victor Stinnerc22fd672010-05-15 22:30:53 +00001919 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001920
Victor Stinnerc22fd672010-05-15 22:30:53 +00001921 if (!PyArg_ParseTuple(args, "b:halfdelay", &tenths)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001922
Victor Stinnerc22fd672010-05-15 22:30:53 +00001923 return PyCursesCheckERR(halfdelay(tenths), "halfdelay");
1924 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001925
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +00001926#ifndef STRICT_SYSV_CURSES
Victor Stinnerc22fd672010-05-15 22:30:53 +00001927 /* No has_key! */
1928 static PyObject * PyCurses_has_key(PyObject *self, PyObject *args)
1929 {
1930 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001931
Victor Stinnerc22fd672010-05-15 22:30:53 +00001932 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001933
Victor Stinnerc22fd672010-05-15 22:30:53 +00001934 if (!PyArg_ParseTuple(args,"i",&ch)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001935
Victor Stinnerc22fd672010-05-15 22:30:53 +00001936 if (has_key(ch) == FALSE) {
1937 Py_INCREF(Py_False);
1938 return Py_False;
1939 }
1940 Py_INCREF(Py_True);
1941 return Py_True;
1942 }
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +00001943#endif /* STRICT_SYSV_CURSES */
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001944
Victor Stinnerc22fd672010-05-15 22:30:53 +00001945 static PyObject *
1946 PyCurses_Init_Color(PyObject *self, PyObject *args)
1947 {
1948 short color, r, g, b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001949
Victor Stinnerc22fd672010-05-15 22:30:53 +00001950 PyCursesInitialised
1951 PyCursesInitialisedColor
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001952
Victor Stinnerc22fd672010-05-15 22:30:53 +00001953 switch(PyTuple_Size(args)) {
1954 case 4:
1955 if (!PyArg_ParseTuple(args, "hhhh;color,r,g,b", &color, &r, &g, &b)) return NULL;
1956 break;
1957 default:
1958 PyErr_SetString(PyExc_TypeError, "init_color requires 4 arguments");
1959 return NULL;
1960 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001961
Victor Stinnerc22fd672010-05-15 22:30:53 +00001962 return PyCursesCheckERR(init_color(color, r, g, b), "init_color");
1963 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001964
Victor Stinnerc22fd672010-05-15 22:30:53 +00001965 static PyObject *
1966 PyCurses_Init_Pair(PyObject *self, PyObject *args)
1967 {
1968 short pair, f, b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001969
Victor Stinnerc22fd672010-05-15 22:30:53 +00001970 PyCursesInitialised
1971 PyCursesInitialisedColor
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001972
Victor Stinnerc22fd672010-05-15 22:30:53 +00001973 if (PyTuple_Size(args) != 3) {
1974 PyErr_SetString(PyExc_TypeError, "init_pair requires 3 arguments");
1975 return NULL;
1976 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001977
Victor Stinnerc22fd672010-05-15 22:30:53 +00001978 if (!PyArg_ParseTuple(args, "hhh;pair, f, b", &pair, &f, &b)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001979
Victor Stinnerc22fd672010-05-15 22:30:53 +00001980 return PyCursesCheckERR(init_pair(pair, f, b), "init_pair");
1981 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001982
Victor Stinnerc22fd672010-05-15 22:30:53 +00001983 static PyObject *ModDict;
Andrew M. Kuchling32552682000-12-22 21:52:27 +00001984
Victor Stinnerc22fd672010-05-15 22:30:53 +00001985 static PyObject *
1986 PyCurses_InitScr(PyObject *self)
1987 {
1988 WINDOW *win;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001989
Victor Stinnerc22fd672010-05-15 22:30:53 +00001990 if (initialised == TRUE) {
1991 wrefresh(stdscr);
1992 return (PyObject *)PyCursesWindow_New(stdscr);
1993 }
Guido van Rossum56bf2351994-08-31 22:06:24 +00001994
Victor Stinnerc22fd672010-05-15 22:30:53 +00001995 win = initscr();
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00001996
Victor Stinnerc22fd672010-05-15 22:30:53 +00001997 if (win == NULL) {
1998 PyErr_SetString(PyCursesError, catchall_NULL);
1999 return NULL;
2000 }
Guido van Rossum85738471995-02-17 13:50:17 +00002001
Victor Stinnerc22fd672010-05-15 22:30:53 +00002002 initialised = initialised_setupterm = TRUE;
Guido van Rossum56bf2351994-08-31 22:06:24 +00002003
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002004/* This was moved from initcurses() because it core dumped on SGI,
2005 where they're not defined until you've called initscr() */
Victor Stinnerc22fd672010-05-15 22:30:53 +00002006#define SetDictInt(string,ch) \
2007 do { \
2008 PyObject *o = PyLong_FromLong((long) (ch)); \
2009 if (o && PyDict_SetItemString(ModDict, string, o) == 0) { \
2010 Py_DECREF(o); \
2011 } \
2012 } while (0)
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002013
Victor Stinnerc22fd672010-05-15 22:30:53 +00002014 /* Here are some graphic symbols you can use */
2015 SetDictInt("ACS_ULCORNER", (ACS_ULCORNER));
2016 SetDictInt("ACS_LLCORNER", (ACS_LLCORNER));
2017 SetDictInt("ACS_URCORNER", (ACS_URCORNER));
2018 SetDictInt("ACS_LRCORNER", (ACS_LRCORNER));
2019 SetDictInt("ACS_LTEE", (ACS_LTEE));
2020 SetDictInt("ACS_RTEE", (ACS_RTEE));
2021 SetDictInt("ACS_BTEE", (ACS_BTEE));
2022 SetDictInt("ACS_TTEE", (ACS_TTEE));
2023 SetDictInt("ACS_HLINE", (ACS_HLINE));
2024 SetDictInt("ACS_VLINE", (ACS_VLINE));
2025 SetDictInt("ACS_PLUS", (ACS_PLUS));
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00002026#if !defined(__hpux) || defined(HAVE_NCURSES_H)
Victor Stinnerc22fd672010-05-15 22:30:53 +00002027 /* On HP/UX 11, these are of type cchar_t, which is not an
2028 integral type. If this is a problem on more platforms, a
2029 configure test should be added to determine whether ACS_S1
2030 is of integral type. */
2031 SetDictInt("ACS_S1", (ACS_S1));
2032 SetDictInt("ACS_S9", (ACS_S9));
2033 SetDictInt("ACS_DIAMOND", (ACS_DIAMOND));
2034 SetDictInt("ACS_CKBOARD", (ACS_CKBOARD));
2035 SetDictInt("ACS_DEGREE", (ACS_DEGREE));
2036 SetDictInt("ACS_PLMINUS", (ACS_PLMINUS));
2037 SetDictInt("ACS_BULLET", (ACS_BULLET));
2038 SetDictInt("ACS_LARROW", (ACS_LARROW));
2039 SetDictInt("ACS_RARROW", (ACS_RARROW));
2040 SetDictInt("ACS_DARROW", (ACS_DARROW));
2041 SetDictInt("ACS_UARROW", (ACS_UARROW));
2042 SetDictInt("ACS_BOARD", (ACS_BOARD));
2043 SetDictInt("ACS_LANTERN", (ACS_LANTERN));
2044 SetDictInt("ACS_BLOCK", (ACS_BLOCK));
Martin v. Löwiseb9b1032001-10-24 17:10:49 +00002045#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002046 SetDictInt("ACS_BSSB", (ACS_ULCORNER));
2047 SetDictInt("ACS_SSBB", (ACS_LLCORNER));
2048 SetDictInt("ACS_BBSS", (ACS_URCORNER));
2049 SetDictInt("ACS_SBBS", (ACS_LRCORNER));
2050 SetDictInt("ACS_SBSS", (ACS_RTEE));
2051 SetDictInt("ACS_SSSB", (ACS_LTEE));
2052 SetDictInt("ACS_SSBS", (ACS_BTEE));
2053 SetDictInt("ACS_BSSS", (ACS_TTEE));
2054 SetDictInt("ACS_BSBS", (ACS_HLINE));
2055 SetDictInt("ACS_SBSB", (ACS_VLINE));
2056 SetDictInt("ACS_SSSS", (ACS_PLUS));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002057
Victor Stinnerc22fd672010-05-15 22:30:53 +00002058 /* The following are never available with strict SYSV curses */
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002059#ifdef ACS_S3
Victor Stinnerc22fd672010-05-15 22:30:53 +00002060 SetDictInt("ACS_S3", (ACS_S3));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002061#endif
2062#ifdef ACS_S7
Victor Stinnerc22fd672010-05-15 22:30:53 +00002063 SetDictInt("ACS_S7", (ACS_S7));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002064#endif
2065#ifdef ACS_LEQUAL
Victor Stinnerc22fd672010-05-15 22:30:53 +00002066 SetDictInt("ACS_LEQUAL", (ACS_LEQUAL));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002067#endif
2068#ifdef ACS_GEQUAL
Victor Stinnerc22fd672010-05-15 22:30:53 +00002069 SetDictInt("ACS_GEQUAL", (ACS_GEQUAL));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002070#endif
2071#ifdef ACS_PI
Victor Stinnerc22fd672010-05-15 22:30:53 +00002072 SetDictInt("ACS_PI", (ACS_PI));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002073#endif
2074#ifdef ACS_NEQUAL
Victor Stinnerc22fd672010-05-15 22:30:53 +00002075 SetDictInt("ACS_NEQUAL", (ACS_NEQUAL));
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002076#endif
2077#ifdef ACS_STERLING
Victor Stinnerc22fd672010-05-15 22:30:53 +00002078 SetDictInt("ACS_STERLING", (ACS_STERLING));
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002079#endif
2080
Victor Stinnerc22fd672010-05-15 22:30:53 +00002081 SetDictInt("LINES", LINES);
2082 SetDictInt("COLS", COLS);
Guido van Rossum56bf2351994-08-31 22:06:24 +00002083
Victor Stinnerc22fd672010-05-15 22:30:53 +00002084 return (PyObject *)PyCursesWindow_New(win);
2085 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00002086
Victor Stinnerc22fd672010-05-15 22:30:53 +00002087 static PyObject *
2088 PyCurses_setupterm(PyObject* self, PyObject *args, PyObject* keywds)
2089 {
2090 int fd = -1;
2091 int err;
2092 char* termstr = NULL;
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002093
Victor Stinnerc22fd672010-05-15 22:30:53 +00002094 static char *kwlist[] = {"term", "fd", NULL};
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002095
Victor Stinnerc22fd672010-05-15 22:30:53 +00002096 if (!PyArg_ParseTupleAndKeywords(
2097 args, keywds, "|zi:setupterm", kwlist, &termstr, &fd)) {
2098 return NULL;
2099 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002100
Victor Stinnerc22fd672010-05-15 22:30:53 +00002101 if (fd == -1) {
2102 PyObject* sys_stdout;
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002103
Victor Stinnerc22fd672010-05-15 22:30:53 +00002104 sys_stdout = PySys_GetObject("stdout");
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002105
Victor Stinnerc22fd672010-05-15 22:30:53 +00002106 if (sys_stdout == NULL || sys_stdout == Py_None) {
2107 PyErr_SetString(
2108 PyCursesError,
2109 "lost sys.stdout");
2110 return NULL;
2111 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002112
Victor Stinnerc22fd672010-05-15 22:30:53 +00002113 fd = PyObject_AsFileDescriptor(sys_stdout);
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002114
Victor Stinnerc22fd672010-05-15 22:30:53 +00002115 if (fd == -1) {
2116 return NULL;
2117 }
2118 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002119
Victor Stinnerc22fd672010-05-15 22:30:53 +00002120 if (setupterm(termstr,fd,&err) == ERR) {
2121 char* s = "setupterm: unknown error";
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002122
Victor Stinnerc22fd672010-05-15 22:30:53 +00002123 if (err == 0) {
2124 s = "setupterm: could not find terminal";
2125 } else if (err == -1) {
2126 s = "setupterm: could not find terminfo database";
2127 }
Andrew M. Kuchling26cc66f2000-11-18 17:45:59 +00002128
Victor Stinnerc22fd672010-05-15 22:30:53 +00002129 PyErr_SetString(PyCursesError,s);
2130 return NULL;
2131 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00002132
Victor Stinnerc22fd672010-05-15 22:30:53 +00002133 initialised_setupterm = TRUE;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002134
Victor Stinnerc22fd672010-05-15 22:30:53 +00002135 Py_INCREF(Py_None);
2136 return Py_None;
2137 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002138
Victor Stinnerc22fd672010-05-15 22:30:53 +00002139 static PyObject *
2140 PyCurses_IntrFlush(PyObject *self, PyObject *args)
2141 {
2142 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002143
Victor Stinnerc22fd672010-05-15 22:30:53 +00002144 PyCursesInitialised
2145
2146 switch(PyTuple_Size(args)) {
2147 case 1:
2148 if (!PyArg_ParseTuple(args,"i;True(1), False(0)",&ch)) return NULL;
2149 break;
2150 default:
2151 PyErr_SetString(PyExc_TypeError, "intrflush requires 1 argument");
2152 return NULL;
2153 }
2154
2155 return PyCursesCheckERR(intrflush(NULL,ch), "intrflush");
2156 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002157
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002158#ifdef HAVE_CURSES_IS_TERM_RESIZED
Victor Stinnerc22fd672010-05-15 22:30:53 +00002159 static PyObject *
2160 PyCurses_Is_Term_Resized(PyObject *self, PyObject *args)
2161 {
2162 int lines;
2163 int columns;
2164 int result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002165
Victor Stinnerc22fd672010-05-15 22:30:53 +00002166 PyCursesInitialised
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002167
Victor Stinnerc22fd672010-05-15 22:30:53 +00002168 if (!PyArg_ParseTuple(args,"ii:is_term_resized", &lines, &columns))
2169 return NULL;
2170 result = is_term_resized(lines, columns);
2171 if (result == TRUE) {
2172 Py_INCREF(Py_True);
2173 return Py_True;
2174 } else {
2175 Py_INCREF(Py_False);
2176 return Py_False;
2177 }
2178 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002179#endif /* HAVE_CURSES_IS_TERM_RESIZED */
2180
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002181#if !defined(__NetBSD__)
Victor Stinnerc22fd672010-05-15 22:30:53 +00002182 static PyObject *
2183 PyCurses_KeyName(PyObject *self, PyObject *args)
2184 {
2185 const char *knp;
2186 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002187
Victor Stinnerc22fd672010-05-15 22:30:53 +00002188 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002189
Victor Stinnerc22fd672010-05-15 22:30:53 +00002190 if (!PyArg_ParseTuple(args,"i",&ch)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002191
Victor Stinnerc22fd672010-05-15 22:30:53 +00002192 if (ch < 0) {
2193 PyErr_SetString(PyExc_ValueError, "invalid key number");
2194 return NULL;
2195 }
2196 knp = keyname(ch);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002197
Victor Stinnerc22fd672010-05-15 22:30:53 +00002198 return PyBytes_FromString((knp == NULL) ? "" : (char *)knp);
2199 }
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002200#endif
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002201
Victor Stinnerc22fd672010-05-15 22:30:53 +00002202 static PyObject *
2203 PyCurses_KillChar(PyObject *self)
2204 {
2205 char ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002206
Victor Stinnerc22fd672010-05-15 22:30:53 +00002207 ch = killchar();
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002208
Victor Stinnerc22fd672010-05-15 22:30:53 +00002209 return PyBytes_FromStringAndSize(&ch, 1);
2210 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002211
Victor Stinnerc22fd672010-05-15 22:30:53 +00002212 static PyObject *
2213 PyCurses_Meta(PyObject *self, PyObject *args)
2214 {
2215 int ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002216
Victor Stinnerc22fd672010-05-15 22:30:53 +00002217 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002218
Victor Stinnerc22fd672010-05-15 22:30:53 +00002219 switch(PyTuple_Size(args)) {
2220 case 1:
2221 if (!PyArg_ParseTuple(args,"i;True(1), False(0)",&ch)) return NULL;
2222 break;
2223 default:
2224 PyErr_SetString(PyExc_TypeError, "meta requires 1 argument");
2225 return NULL;
2226 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002227
Victor Stinnerc22fd672010-05-15 22:30:53 +00002228 return PyCursesCheckERR(meta(stdscr, ch), "meta");
2229 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002230
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002231#ifdef NCURSES_MOUSE_VERSION
Victor Stinnerc22fd672010-05-15 22:30:53 +00002232 static PyObject *
2233 PyCurses_MouseInterval(PyObject *self, PyObject *args)
2234 {
2235 int interval;
2236 PyCursesInitialised
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002237
Victor Stinnerc22fd672010-05-15 22:30:53 +00002238 if (!PyArg_ParseTuple(args,"i;interval",&interval))
2239 return NULL;
2240 return PyCursesCheckERR(mouseinterval(interval), "mouseinterval");
2241 }
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002242
Victor Stinnerc22fd672010-05-15 22:30:53 +00002243 static PyObject *
2244 PyCurses_MouseMask(PyObject *self, PyObject *args)
2245 {
2246 int newmask;
2247 mmask_t oldmask, availmask;
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002248
Victor Stinnerc22fd672010-05-15 22:30:53 +00002249 PyCursesInitialised
2250 if (!PyArg_ParseTuple(args,"i;mousemask",&newmask))
2251 return NULL;
2252 availmask = mousemask(newmask, &oldmask);
2253 return Py_BuildValue("(ll)", (long)availmask, (long)oldmask);
2254 }
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002255#endif
2256
Victor Stinnerc22fd672010-05-15 22:30:53 +00002257 static PyObject *
2258 PyCurses_Napms(PyObject *self, PyObject *args)
2259 {
2260 int ms;
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +00002261
Victor Stinnerc22fd672010-05-15 22:30:53 +00002262 PyCursesInitialised
2263 if (!PyArg_ParseTuple(args, "i;ms", &ms)) return NULL;
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +00002264
Victor Stinnerc22fd672010-05-15 22:30:53 +00002265 return Py_BuildValue("i", napms(ms));
2266 }
Andrew M. Kuchling2824d7f2000-12-15 00:44:02 +00002267
2268
Victor Stinnerc22fd672010-05-15 22:30:53 +00002269 static PyObject *
2270 PyCurses_NewPad(PyObject *self, PyObject *args)
2271 {
2272 WINDOW *win;
2273 int nlines, ncols;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002274
Victor Stinnerc22fd672010-05-15 22:30:53 +00002275 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002276
Victor Stinnerc22fd672010-05-15 22:30:53 +00002277 if (!PyArg_ParseTuple(args,"ii;nlines,ncols",&nlines,&ncols)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002278
Victor Stinnerc22fd672010-05-15 22:30:53 +00002279 win = newpad(nlines, ncols);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002280
Victor Stinnerc22fd672010-05-15 22:30:53 +00002281 if (win == NULL) {
2282 PyErr_SetString(PyCursesError, catchall_NULL);
2283 return NULL;
2284 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00002285
Victor Stinnerc22fd672010-05-15 22:30:53 +00002286 return (PyObject *)PyCursesWindow_New(win);
2287 }
Guido van Rossum85738471995-02-17 13:50:17 +00002288
Victor Stinnerc22fd672010-05-15 22:30:53 +00002289 static PyObject *
2290 PyCurses_NewWindow(PyObject *self, PyObject *args)
2291 {
2292 WINDOW *win;
2293 int nlines, ncols, begin_y=0, begin_x=0;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002294
Victor Stinnerc22fd672010-05-15 22:30:53 +00002295 PyCursesInitialised
Guido van Rossum85738471995-02-17 13:50:17 +00002296
Victor Stinnerc22fd672010-05-15 22:30:53 +00002297 switch (PyTuple_Size(args)) {
2298 case 2:
2299 if (!PyArg_ParseTuple(args,"ii;nlines,ncols",&nlines,&ncols))
2300 return NULL;
2301 break;
2302 case 4:
2303 if (!PyArg_ParseTuple(args, "iiii;nlines,ncols,begin_y,begin_x",
2304 &nlines,&ncols,&begin_y,&begin_x))
2305 return NULL;
2306 break;
2307 default:
2308 PyErr_SetString(PyExc_TypeError, "newwin requires 2 or 4 arguments");
2309 return NULL;
2310 }
Guido van Rossum85738471995-02-17 13:50:17 +00002311
Victor Stinnerc22fd672010-05-15 22:30:53 +00002312 win = newwin(nlines,ncols,begin_y,begin_x);
2313 if (win == NULL) {
2314 PyErr_SetString(PyCursesError, catchall_NULL);
2315 return NULL;
2316 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00002317
Victor Stinnerc22fd672010-05-15 22:30:53 +00002318 return (PyObject *)PyCursesWindow_New(win);
2319 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002320
Victor Stinnerc22fd672010-05-15 22:30:53 +00002321 static PyObject *
2322 PyCurses_Pair_Content(PyObject *self, PyObject *args)
2323 {
2324 short pair,f,b;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002325
Victor Stinnerc22fd672010-05-15 22:30:53 +00002326 PyCursesInitialised
2327 PyCursesInitialisedColor
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002328
Victor Stinnerc22fd672010-05-15 22:30:53 +00002329 switch(PyTuple_Size(args)) {
2330 case 1:
2331 if (!PyArg_ParseTuple(args, "h;pair", &pair)) return NULL;
2332 break;
2333 default:
2334 PyErr_SetString(PyExc_TypeError, "pair_content requires 1 argument");
2335 return NULL;
2336 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002337
Victor Stinnerc22fd672010-05-15 22:30:53 +00002338 if (pair_content(pair, &f, &b)==ERR) {
2339 PyErr_SetString(PyCursesError,
2340 "Argument 1 was out of range. (1..COLOR_PAIRS-1)");
2341 return NULL;
2342 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002343
Victor Stinnerc22fd672010-05-15 22:30:53 +00002344 return Py_BuildValue("(ii)", f, b);
2345 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002346
Victor Stinnerc22fd672010-05-15 22:30:53 +00002347 static PyObject *
2348 PyCurses_pair_number(PyObject *self, PyObject *args)
2349 {
2350 int n;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002351
Victor Stinnerc22fd672010-05-15 22:30:53 +00002352 PyCursesInitialised
2353 PyCursesInitialisedColor
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002354
Victor Stinnerc22fd672010-05-15 22:30:53 +00002355 switch(PyTuple_Size(args)) {
2356 case 1:
2357 if (!PyArg_ParseTuple(args, "i;pairvalue", &n)) return NULL;
2358 break;
2359 default:
2360 PyErr_SetString(PyExc_TypeError,
2361 "pair_number requires 1 argument");
2362 return NULL;
2363 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002364
Victor Stinnerc22fd672010-05-15 22:30:53 +00002365 return PyLong_FromLong((long) ((n & A_COLOR) >> 8));
2366 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002367
Victor Stinnerc22fd672010-05-15 22:30:53 +00002368 static PyObject *
2369 PyCurses_Putp(PyObject *self, PyObject *args)
2370 {
2371 char *str;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002372
Victor Stinnerc22fd672010-05-15 22:30:53 +00002373 if (!PyArg_ParseTuple(args,"s;str", &str)) return NULL;
2374 return PyCursesCheckERR(putp(str), "putp");
2375 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002376
Victor Stinnerc22fd672010-05-15 22:30:53 +00002377 static PyObject *
2378 PyCurses_QiFlush(PyObject *self, PyObject *args)
2379 {
2380 int flag = 0;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002381
Victor Stinnerc22fd672010-05-15 22:30:53 +00002382 PyCursesInitialised
2383
2384 switch(PyTuple_Size(args)) {
2385 case 0:
2386 qiflush();
2387 Py_INCREF(Py_None);
2388 return Py_None;
2389 case 1:
2390 if (!PyArg_ParseTuple(args, "i;True(1) or False(0)", &flag)) return NULL;
2391 if (flag) qiflush();
2392 else noqiflush();
2393 Py_INCREF(Py_None);
2394 return Py_None;
2395 default:
2396 PyErr_SetString(PyExc_TypeError, "qiflush requires 0 or 1 arguments");
2397 return NULL;
2398 }
2399 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002400
Guido van Rossumd8faa362007-04-27 19:54:29 +00002401/* Internal helper used for updating curses.LINES, curses.COLS, _curses.LINES
2402 * and _curses.COLS */
Benjamin Petersonfea6a942008-07-02 16:11:42 +00002403#if defined(HAVE_CURSES_RESIZETERM) || defined(HAVE_CURSES_RESIZE_TERM)
Victor Stinnerc22fd672010-05-15 22:30:53 +00002404 static int
2405 update_lines_cols(void)
2406 {
2407 PyObject *o;
2408 PyObject *m = PyImport_ImportModuleNoBlock("curses");
Guido van Rossumd8faa362007-04-27 19:54:29 +00002409
Victor Stinnerc22fd672010-05-15 22:30:53 +00002410 if (!m)
2411 return 0;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002412
Victor Stinnerc22fd672010-05-15 22:30:53 +00002413 o = PyLong_FromLong(LINES);
2414 if (!o) {
2415 Py_DECREF(m);
2416 return 0;
2417 }
2418 if (PyObject_SetAttrString(m, "LINES", o)) {
2419 Py_DECREF(m);
2420 Py_DECREF(o);
2421 return 0;
2422 }
2423 if (PyDict_SetItemString(ModDict, "LINES", o)) {
2424 Py_DECREF(m);
2425 Py_DECREF(o);
2426 return 0;
2427 }
2428 Py_DECREF(o);
2429 o = PyLong_FromLong(COLS);
2430 if (!o) {
2431 Py_DECREF(m);
2432 return 0;
2433 }
2434 if (PyObject_SetAttrString(m, "COLS", o)) {
2435 Py_DECREF(m);
2436 Py_DECREF(o);
2437 return 0;
2438 }
2439 if (PyDict_SetItemString(ModDict, "COLS", o)) {
2440 Py_DECREF(m);
2441 Py_DECREF(o);
2442 return 0;
2443 }
2444 Py_DECREF(o);
2445 Py_DECREF(m);
2446 return 1;
2447 }
Benjamin Petersonfea6a942008-07-02 16:11:42 +00002448#endif
Guido van Rossumd8faa362007-04-27 19:54:29 +00002449
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002450#ifdef HAVE_CURSES_RESIZETERM
Victor Stinnerc22fd672010-05-15 22:30:53 +00002451 static PyObject *
2452 PyCurses_ResizeTerm(PyObject *self, PyObject *args)
2453 {
2454 int lines;
2455 int columns;
2456 PyObject *result;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002457
Victor Stinnerc22fd672010-05-15 22:30:53 +00002458 PyCursesInitialised
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002459
Victor Stinnerc22fd672010-05-15 22:30:53 +00002460 if (!PyArg_ParseTuple(args,"ii:resizeterm", &lines, &columns))
2461 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002462
Victor Stinnerc22fd672010-05-15 22:30:53 +00002463 result = PyCursesCheckERR(resizeterm(lines, columns), "resizeterm");
2464 if (!result)
2465 return NULL;
2466 if (!update_lines_cols())
2467 return NULL;
2468 return result;
2469 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002470
2471#endif
2472
2473#ifdef HAVE_CURSES_RESIZE_TERM
Victor Stinnerc22fd672010-05-15 22:30:53 +00002474 static PyObject *
2475 PyCurses_Resize_Term(PyObject *self, PyObject *args)
2476 {
2477 int lines;
2478 int columns;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002479
Victor Stinnerc22fd672010-05-15 22:30:53 +00002480 PyObject *result;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002481
Victor Stinnerc22fd672010-05-15 22:30:53 +00002482 PyCursesInitialised
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002483
Victor Stinnerc22fd672010-05-15 22:30:53 +00002484 if (!PyArg_ParseTuple(args,"ii:resize_term", &lines, &columns))
2485 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002486
Victor Stinnerc22fd672010-05-15 22:30:53 +00002487 result = PyCursesCheckERR(resize_term(lines, columns), "resize_term");
2488 if (!result)
2489 return NULL;
2490 if (!update_lines_cols())
2491 return NULL;
2492 return result;
2493 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002494#endif /* HAVE_CURSES_RESIZE_TERM */
2495
Victor Stinnerc22fd672010-05-15 22:30:53 +00002496 static PyObject *
2497 PyCurses_setsyx(PyObject *self, PyObject *args)
2498 {
2499 int y,x;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002500
Victor Stinnerc22fd672010-05-15 22:30:53 +00002501 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002502
Victor Stinnerc22fd672010-05-15 22:30:53 +00002503 if (PyTuple_Size(args)!=2) {
2504 PyErr_SetString(PyExc_TypeError, "setsyx requires 2 arguments");
2505 return NULL;
2506 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002507
Victor Stinnerc22fd672010-05-15 22:30:53 +00002508 if (!PyArg_ParseTuple(args, "ii;y, x", &y, &x)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002509
Victor Stinnerc22fd672010-05-15 22:30:53 +00002510 setsyx(y,x);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002511
Victor Stinnerc22fd672010-05-15 22:30:53 +00002512 Py_INCREF(Py_None);
2513 return Py_None;
2514 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00002515
Victor Stinnerc22fd672010-05-15 22:30:53 +00002516 static PyObject *
2517 PyCurses_Start_Color(PyObject *self)
2518 {
2519 int code;
2520 PyObject *c, *cp;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002521
Victor Stinnerc22fd672010-05-15 22:30:53 +00002522 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002523
Victor Stinnerc22fd672010-05-15 22:30:53 +00002524 code = start_color();
2525 if (code != ERR) {
2526 initialisedcolors = TRUE;
2527 c = PyLong_FromLong((long) COLORS);
2528 PyDict_SetItemString(ModDict, "COLORS", c);
2529 Py_DECREF(c);
2530 cp = PyLong_FromLong((long) COLOR_PAIRS);
2531 PyDict_SetItemString(ModDict, "COLOR_PAIRS", cp);
2532 Py_DECREF(cp);
2533 Py_INCREF(Py_None);
2534 return Py_None;
2535 } else {
2536 PyErr_SetString(PyCursesError, "start_color() returned ERR");
2537 return NULL;
2538 }
2539 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002540
Victor Stinnerc22fd672010-05-15 22:30:53 +00002541 static PyObject *
2542 PyCurses_tigetflag(PyObject *self, PyObject *args)
2543 {
2544 char *capname;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002545
Victor Stinnerc22fd672010-05-15 22:30:53 +00002546 PyCursesSetupTermCalled;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002547
Victor Stinnerc22fd672010-05-15 22:30:53 +00002548 if (!PyArg_ParseTuple(args, "s", &capname))
2549 return NULL;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002550
Victor Stinnerc22fd672010-05-15 22:30:53 +00002551 return PyLong_FromLong( (long) tigetflag( capname ) );
2552 }
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002553
Victor Stinnerc22fd672010-05-15 22:30:53 +00002554 static PyObject *
2555 PyCurses_tigetnum(PyObject *self, PyObject *args)
2556 {
2557 char *capname;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002558
Victor Stinnerc22fd672010-05-15 22:30:53 +00002559 PyCursesSetupTermCalled;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002560
Victor Stinnerc22fd672010-05-15 22:30:53 +00002561 if (!PyArg_ParseTuple(args, "s", &capname))
2562 return NULL;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002563
Victor Stinnerc22fd672010-05-15 22:30:53 +00002564 return PyLong_FromLong( (long) tigetnum( capname ) );
2565 }
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002566
Victor Stinnerc22fd672010-05-15 22:30:53 +00002567 static PyObject *
2568 PyCurses_tigetstr(PyObject *self, PyObject *args)
2569 {
2570 char *capname;
Andrew M. Kuchling70e89c12000-07-26 02:57:20 +00002571
Victor Stinnerc22fd672010-05-15 22:30:53 +00002572 PyCursesSetupTermCalled;
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00002573
Victor Stinnerc22fd672010-05-15 22:30:53 +00002574 if (!PyArg_ParseTuple(args, "s", &capname))
2575 return NULL;
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00002576
Victor Stinnerc22fd672010-05-15 22:30:53 +00002577 capname = tigetstr( capname );
2578 if (capname == 0 || capname == (char*) -1) {
2579 Py_INCREF(Py_None);
2580 return Py_None;
2581 }
2582 return PyBytes_FromString( capname );
2583 }
Michael W. Hudson21007982002-11-21 14:17:51 +00002584
Victor Stinnerc22fd672010-05-15 22:30:53 +00002585 static PyObject *
2586 PyCurses_tparm(PyObject *self, PyObject *args)
2587 {
2588 char* fmt;
2589 char* result = NULL;
2590 int i1=0,i2=0,i3=0,i4=0,i5=0,i6=0,i7=0,i8=0,i9=0;
Michael W. Hudson21007982002-11-21 14:17:51 +00002591
Victor Stinnerc22fd672010-05-15 22:30:53 +00002592 PyCursesSetupTermCalled;
Andrew M. Kuchlingf16e0ed2000-11-07 03:35:24 +00002593
Victor Stinnerc22fd672010-05-15 22:30:53 +00002594 if (!PyArg_ParseTuple(args, "s|iiiiiiiii:tparm",
2595 &fmt, &i1, &i2, &i3, &i4,
2596 &i5, &i6, &i7, &i8, &i9)) {
2597 return NULL;
2598 }
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00002599
Victor Stinnerc22fd672010-05-15 22:30:53 +00002600 result = tparm(fmt,i1,i2,i3,i4,i5,i6,i7,i8,i9);
2601 if (!result) {
2602 PyErr_SetString(PyCursesError, "tparm() returned NULL");
2603 return NULL;
2604 }
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00002605
Victor Stinnerc22fd672010-05-15 22:30:53 +00002606 return PyBytes_FromString(result);
2607 }
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00002608
Victor Stinnerc22fd672010-05-15 22:30:53 +00002609 static PyObject *
2610 PyCurses_TypeAhead(PyObject *self, PyObject *args)
2611 {
2612 int fd;
Andrew M. Kuchling97311bb2000-06-21 01:41:48 +00002613
Victor Stinnerc22fd672010-05-15 22:30:53 +00002614 PyCursesInitialised
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002615
Victor Stinnerc22fd672010-05-15 22:30:53 +00002616 if (!PyArg_ParseTuple(args,"i;fd",&fd)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002617
Victor Stinnerc22fd672010-05-15 22:30:53 +00002618 return PyCursesCheckERR(typeahead( fd ), "typeahead");
2619 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002620
Victor Stinnerc22fd672010-05-15 22:30:53 +00002621 static PyObject *
2622 PyCurses_UnCtrl(PyObject *self, PyObject *args)
2623 {
2624 PyObject *temp;
2625 chtype ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002626
Victor Stinnerc22fd672010-05-15 22:30:53 +00002627 PyCursesInitialised
Guido van Rossumf6971e21994-08-30 12:25:20 +00002628
Victor Stinnerc22fd672010-05-15 22:30:53 +00002629 if (!PyArg_ParseTuple(args,"O;ch or int",&temp)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002630
Victor Stinnerc22fd672010-05-15 22:30:53 +00002631 if (!PyCurses_ConvertToChtype(temp, &ch)) {
2632 PyErr_SetString(PyExc_TypeError, "argument must be a ch or an int");
2633 return NULL;
2634 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002635
Victor Stinnerc22fd672010-05-15 22:30:53 +00002636 return PyBytes_FromString(unctrl(ch));
2637 }
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002638
Victor Stinnerc22fd672010-05-15 22:30:53 +00002639 static PyObject *
2640 PyCurses_UngetCh(PyObject *self, PyObject *args)
2641 {
2642 PyObject *temp;
2643 chtype ch;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002644
Victor Stinnerc22fd672010-05-15 22:30:53 +00002645 PyCursesInitialised
Guido van Rossumf6971e21994-08-30 12:25:20 +00002646
Victor Stinnerc22fd672010-05-15 22:30:53 +00002647 if (!PyArg_ParseTuple(args,"O;ch or int",&temp)) return NULL;
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002648
Victor Stinnerc22fd672010-05-15 22:30:53 +00002649 if (!PyCurses_ConvertToChtype(temp, &ch)) {
2650 PyErr_SetString(PyExc_TypeError, "argument must be a ch or an int");
2651 return NULL;
2652 }
2653
2654 return PyCursesCheckERR(ungetch(ch), "ungetch");
2655 }
2656
2657 static PyObject *
2658 PyCurses_Use_Env(PyObject *self, PyObject *args)
2659 {
2660 int flag;
2661
2662 switch(PyTuple_Size(args)) {
2663 case 1:
2664 if (!PyArg_ParseTuple(args,"i;True(1), False(0)",&flag))
2665 return NULL;
2666 break;
2667 default:
2668 PyErr_SetString(PyExc_TypeError, "use_env requires 1 argument");
2669 return NULL;
2670 }
2671 use_env(flag);
2672 Py_INCREF(Py_None);
2673 return Py_None;
2674 }
Guido van Rossumf6971e21994-08-30 12:25:20 +00002675
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00002676#ifndef STRICT_SYSV_CURSES
Victor Stinnerc22fd672010-05-15 22:30:53 +00002677 static PyObject *
2678 PyCurses_Use_Default_Colors(PyObject *self)
2679 {
2680 int code;
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00002681
Victor Stinnerc22fd672010-05-15 22:30:53 +00002682 PyCursesInitialised
2683 PyCursesInitialisedColor
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00002684
Victor Stinnerc22fd672010-05-15 22:30:53 +00002685 code = use_default_colors();
2686 if (code != ERR) {
2687 Py_INCREF(Py_None);
2688 return Py_None;
2689 } else {
2690 PyErr_SetString(PyCursesError, "use_default_colors() returned ERR");
2691 return NULL;
2692 }
2693 }
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00002694#endif /* STRICT_SYSV_CURSES */
2695
Guido van Rossumf6971e21994-08-30 12:25:20 +00002696/* List of functions defined in the module */
2697
Victor Stinnerc22fd672010-05-15 22:30:53 +00002698 static PyMethodDef PyCurses_methods[] = {
2699 {"baudrate", (PyCFunction)PyCurses_baudrate, METH_NOARGS},
2700 {"beep", (PyCFunction)PyCurses_beep, METH_NOARGS},
2701 {"can_change_color", (PyCFunction)PyCurses_can_change_color, METH_NOARGS},
2702 {"cbreak", (PyCFunction)PyCurses_cbreak, METH_VARARGS},
2703 {"color_content", (PyCFunction)PyCurses_Color_Content, METH_VARARGS},
2704 {"color_pair", (PyCFunction)PyCurses_color_pair, METH_VARARGS},
2705 {"curs_set", (PyCFunction)PyCurses_Curs_Set, METH_VARARGS},
2706 {"def_prog_mode", (PyCFunction)PyCurses_def_prog_mode, METH_NOARGS},
2707 {"def_shell_mode", (PyCFunction)PyCurses_def_shell_mode, METH_NOARGS},
2708 {"delay_output", (PyCFunction)PyCurses_Delay_Output, METH_VARARGS},
2709 {"doupdate", (PyCFunction)PyCurses_doupdate, METH_NOARGS},
2710 {"echo", (PyCFunction)PyCurses_echo, METH_VARARGS},
2711 {"endwin", (PyCFunction)PyCurses_endwin, METH_NOARGS},
2712 {"erasechar", (PyCFunction)PyCurses_EraseChar, METH_NOARGS},
2713 {"filter", (PyCFunction)PyCurses_filter, METH_NOARGS},
2714 {"flash", (PyCFunction)PyCurses_flash, METH_NOARGS},
2715 {"flushinp", (PyCFunction)PyCurses_flushinp, METH_NOARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002716#ifdef NCURSES_MOUSE_VERSION
Victor Stinnerc22fd672010-05-15 22:30:53 +00002717 {"getmouse", (PyCFunction)PyCurses_GetMouse, METH_NOARGS},
2718 {"ungetmouse", (PyCFunction)PyCurses_UngetMouse, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002719#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002720 {"getsyx", (PyCFunction)PyCurses_getsyx, METH_NOARGS},
2721 {"getwin", (PyCFunction)PyCurses_GetWin, METH_O},
2722 {"has_colors", (PyCFunction)PyCurses_has_colors, METH_NOARGS},
2723 {"has_ic", (PyCFunction)PyCurses_has_ic, METH_NOARGS},
2724 {"has_il", (PyCFunction)PyCurses_has_il, METH_NOARGS},
Andrew M. Kuchlingbf680262000-07-25 12:37:41 +00002725#ifndef STRICT_SYSV_CURSES
Victor Stinnerc22fd672010-05-15 22:30:53 +00002726 {"has_key", (PyCFunction)PyCurses_has_key, METH_VARARGS},
Guido van Rossumf6971e21994-08-30 12:25:20 +00002727#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002728 {"halfdelay", (PyCFunction)PyCurses_HalfDelay, METH_VARARGS},
2729 {"init_color", (PyCFunction)PyCurses_Init_Color, METH_VARARGS},
2730 {"init_pair", (PyCFunction)PyCurses_Init_Pair, METH_VARARGS},
2731 {"initscr", (PyCFunction)PyCurses_InitScr, METH_NOARGS},
2732 {"intrflush", (PyCFunction)PyCurses_IntrFlush, METH_VARARGS},
2733 {"isendwin", (PyCFunction)PyCurses_isendwin, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002734#ifdef HAVE_CURSES_IS_TERM_RESIZED
Victor Stinnerc22fd672010-05-15 22:30:53 +00002735 {"is_term_resized", (PyCFunction)PyCurses_Is_Term_Resized, METH_VARARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002736#endif
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002737#if !defined(__NetBSD__)
Victor Stinnerc22fd672010-05-15 22:30:53 +00002738 {"keyname", (PyCFunction)PyCurses_KeyName, METH_VARARGS},
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002739#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002740 {"killchar", (PyCFunction)PyCurses_KillChar, METH_NOARGS},
2741 {"longname", (PyCFunction)PyCurses_longname, METH_NOARGS},
2742 {"meta", (PyCFunction)PyCurses_Meta, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002743#ifdef NCURSES_MOUSE_VERSION
Victor Stinnerc22fd672010-05-15 22:30:53 +00002744 {"mouseinterval", (PyCFunction)PyCurses_MouseInterval, METH_VARARGS},
2745 {"mousemask", (PyCFunction)PyCurses_MouseMask, METH_VARARGS},
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002746#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002747 {"napms", (PyCFunction)PyCurses_Napms, METH_VARARGS},
2748 {"newpad", (PyCFunction)PyCurses_NewPad, METH_VARARGS},
2749 {"newwin", (PyCFunction)PyCurses_NewWindow, METH_VARARGS},
2750 {"nl", (PyCFunction)PyCurses_nl, METH_VARARGS},
2751 {"nocbreak", (PyCFunction)PyCurses_nocbreak, METH_NOARGS},
2752 {"noecho", (PyCFunction)PyCurses_noecho, METH_NOARGS},
2753 {"nonl", (PyCFunction)PyCurses_nonl, METH_NOARGS},
2754 {"noqiflush", (PyCFunction)PyCurses_noqiflush, METH_NOARGS},
2755 {"noraw", (PyCFunction)PyCurses_noraw, METH_NOARGS},
2756 {"pair_content", (PyCFunction)PyCurses_Pair_Content, METH_VARARGS},
2757 {"pair_number", (PyCFunction)PyCurses_pair_number, METH_VARARGS},
2758 {"putp", (PyCFunction)PyCurses_Putp, METH_VARARGS},
2759 {"qiflush", (PyCFunction)PyCurses_QiFlush, METH_VARARGS},
2760 {"raw", (PyCFunction)PyCurses_raw, METH_VARARGS},
2761 {"reset_prog_mode", (PyCFunction)PyCurses_reset_prog_mode, METH_NOARGS},
2762 {"reset_shell_mode", (PyCFunction)PyCurses_reset_shell_mode, METH_NOARGS},
2763 {"resetty", (PyCFunction)PyCurses_resetty, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002764#ifdef HAVE_CURSES_RESIZETERM
Victor Stinnerc22fd672010-05-15 22:30:53 +00002765 {"resizeterm", (PyCFunction)PyCurses_ResizeTerm, METH_VARARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002766#endif
2767#ifdef HAVE_CURSES_RESIZE_TERM
Victor Stinnerc22fd672010-05-15 22:30:53 +00002768 {"resize_term", (PyCFunction)PyCurses_Resize_Term, METH_VARARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002769#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002770 {"savetty", (PyCFunction)PyCurses_savetty, METH_NOARGS},
2771 {"setsyx", (PyCFunction)PyCurses_setsyx, METH_VARARGS},
2772 {"setupterm", (PyCFunction)PyCurses_setupterm,
2773 METH_VARARGS|METH_KEYWORDS},
2774 {"start_color", (PyCFunction)PyCurses_Start_Color, METH_NOARGS},
2775 {"termattrs", (PyCFunction)PyCurses_termattrs, METH_NOARGS},
2776 {"termname", (PyCFunction)PyCurses_termname, METH_NOARGS},
2777 {"tigetflag", (PyCFunction)PyCurses_tigetflag, METH_VARARGS},
2778 {"tigetnum", (PyCFunction)PyCurses_tigetnum, METH_VARARGS},
2779 {"tigetstr", (PyCFunction)PyCurses_tigetstr, METH_VARARGS},
2780 {"tparm", (PyCFunction)PyCurses_tparm, METH_VARARGS},
2781 {"typeahead", (PyCFunction)PyCurses_TypeAhead, METH_VARARGS},
2782 {"unctrl", (PyCFunction)PyCurses_UnCtrl, METH_VARARGS},
2783 {"ungetch", (PyCFunction)PyCurses_UngetCh, METH_VARARGS},
2784 {"use_env", (PyCFunction)PyCurses_Use_Env, METH_VARARGS},
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00002785#ifndef STRICT_SYSV_CURSES
Victor Stinnerc22fd672010-05-15 22:30:53 +00002786 {"use_default_colors", (PyCFunction)PyCurses_Use_Default_Colors, METH_NOARGS},
Andrew M. Kuchling69f31eb2003-08-13 23:11:04 +00002787#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002788 {NULL, NULL} /* sentinel */
2789 };
Guido van Rossumf6971e21994-08-30 12:25:20 +00002790
2791/* Initialization function for the module */
2792
Martin v. Löwis1a214512008-06-11 05:26:20 +00002793
Victor Stinnerc22fd672010-05-15 22:30:53 +00002794 static struct PyModuleDef _cursesmodule = {
2795 PyModuleDef_HEAD_INIT,
2796 "_curses",
2797 NULL,
2798 -1,
2799 PyCurses_methods,
2800 NULL,
2801 NULL,
2802 NULL,
2803 NULL
2804 };
Martin v. Löwis1a214512008-06-11 05:26:20 +00002805
Victor Stinnerc22fd672010-05-15 22:30:53 +00002806 PyMODINIT_FUNC
2807 PyInit__curses(void)
2808 {
2809 PyObject *m, *d, *v, *c_api_object;
2810 static void *PyCurses_API[PyCurses_API_pointers];
Andrew M. Kuchling32552682000-12-22 21:52:27 +00002811
Victor Stinnerc22fd672010-05-15 22:30:53 +00002812 /* Initialize object type */
2813 if (PyType_Ready(&PyCursesWindow_Type) < 0)
2814 return NULL;
Guido van Rossum6915c4d2001-01-19 00:28:08 +00002815
Victor Stinnerc22fd672010-05-15 22:30:53 +00002816 /* Initialize the C API pointer array */
2817 PyCurses_API[0] = (void *)&PyCursesWindow_Type;
2818 PyCurses_API[1] = (void *)func_PyCursesSetupTermCalled;
2819 PyCurses_API[2] = (void *)func_PyCursesInitialised;
2820 PyCurses_API[3] = (void *)func_PyCursesInitialisedColor;
Guido van Rossumf6971e21994-08-30 12:25:20 +00002821
Victor Stinnerc22fd672010-05-15 22:30:53 +00002822 /* Create the module and add the functions */
2823 m = PyModule_Create(&_cursesmodule);
2824 if (m == NULL)
2825 return NULL;
Guido van Rossumf6971e21994-08-30 12:25:20 +00002826
Victor Stinnerc22fd672010-05-15 22:30:53 +00002827 /* Add some symbolic constants to the module */
2828 d = PyModule_GetDict(m);
2829 if (d == NULL)
2830 return NULL;
2831 ModDict = d; /* For PyCurses_InitScr to use later */
Andrew M. Kuchling32552682000-12-22 21:52:27 +00002832
Victor Stinnerc22fd672010-05-15 22:30:53 +00002833 /* Add a capsule for the C API */
2834 c_api_object = PyCapsule_New(PyCurses_API, PyCurses_CAPSULE_NAME, NULL);
2835 PyDict_SetItemString(d, "_C_API", c_api_object);
2836 Py_DECREF(c_api_object);
Guido van Rossumfbea2f31994-08-31 22:05:27 +00002837
Victor Stinnerc22fd672010-05-15 22:30:53 +00002838 /* For exception curses.error */
2839 PyCursesError = PyErr_NewException("_curses.error", NULL, NULL);
2840 PyDict_SetItemString(d, "error", PyCursesError);
Guido van Rossumfbea2f31994-08-31 22:05:27 +00002841
Victor Stinnerc22fd672010-05-15 22:30:53 +00002842 /* Make the version available */
2843 v = PyBytes_FromString(PyCursesVersion);
2844 PyDict_SetItemString(d, "version", v);
2845 PyDict_SetItemString(d, "__version__", v);
2846 Py_DECREF(v);
Guido van Rossumf6971e21994-08-30 12:25:20 +00002847
Victor Stinnerc22fd672010-05-15 22:30:53 +00002848 SetDictInt("ERR", ERR);
2849 SetDictInt("OK", OK);
Andrew M. Kuchlingcaefb372000-12-26 15:57:01 +00002850
Victor Stinnerc22fd672010-05-15 22:30:53 +00002851 /* Here are some attributes you can add to chars to print */
2852
2853 SetDictInt("A_ATTRIBUTES", A_ATTRIBUTES);
2854 SetDictInt("A_NORMAL", A_NORMAL);
2855 SetDictInt("A_STANDOUT", A_STANDOUT);
2856 SetDictInt("A_UNDERLINE", A_UNDERLINE);
2857 SetDictInt("A_REVERSE", A_REVERSE);
2858 SetDictInt("A_BLINK", A_BLINK);
2859 SetDictInt("A_DIM", A_DIM);
2860 SetDictInt("A_BOLD", A_BOLD);
2861 SetDictInt("A_ALTCHARSET", A_ALTCHARSET);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002862#if !defined(__NetBSD__)
Victor Stinnerc22fd672010-05-15 22:30:53 +00002863 SetDictInt("A_INVIS", A_INVIS);
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002864#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002865 SetDictInt("A_PROTECT", A_PROTECT);
2866 SetDictInt("A_CHARTEXT", A_CHARTEXT);
2867 SetDictInt("A_COLOR", A_COLOR);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002868
Victor Stinnerc22fd672010-05-15 22:30:53 +00002869 /* The following are never available with strict SYSV curses */
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002870#ifdef A_HORIZONTAL
Victor Stinnerc22fd672010-05-15 22:30:53 +00002871 SetDictInt("A_HORIZONTAL", A_HORIZONTAL);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002872#endif
2873#ifdef A_LEFT
Victor Stinnerc22fd672010-05-15 22:30:53 +00002874 SetDictInt("A_LEFT", A_LEFT);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002875#endif
2876#ifdef A_LOW
Victor Stinnerc22fd672010-05-15 22:30:53 +00002877 SetDictInt("A_LOW", A_LOW);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002878#endif
2879#ifdef A_RIGHT
Victor Stinnerc22fd672010-05-15 22:30:53 +00002880 SetDictInt("A_RIGHT", A_RIGHT);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002881#endif
2882#ifdef A_TOP
Victor Stinnerc22fd672010-05-15 22:30:53 +00002883 SetDictInt("A_TOP", A_TOP);
Andrew M. Kuchlinge7d36162000-11-01 19:59:12 +00002884#endif
2885#ifdef A_VERTICAL
Victor Stinnerc22fd672010-05-15 22:30:53 +00002886 SetDictInt("A_VERTICAL", A_VERTICAL);
Andrew M. Kuchling22b88ce2000-05-23 16:18:03 +00002887#endif
Andrew M. Kuchling03e644b2000-05-23 16:24:54 +00002888
Victor Stinnerc22fd672010-05-15 22:30:53 +00002889 SetDictInt("COLOR_BLACK", COLOR_BLACK);
2890 SetDictInt("COLOR_RED", COLOR_RED);
2891 SetDictInt("COLOR_GREEN", COLOR_GREEN);
2892 SetDictInt("COLOR_YELLOW", COLOR_YELLOW);
2893 SetDictInt("COLOR_BLUE", COLOR_BLUE);
2894 SetDictInt("COLOR_MAGENTA", COLOR_MAGENTA);
2895 SetDictInt("COLOR_CYAN", COLOR_CYAN);
2896 SetDictInt("COLOR_WHITE", COLOR_WHITE);
Guido van Rossumf6971e21994-08-30 12:25:20 +00002897
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002898#ifdef NCURSES_MOUSE_VERSION
Victor Stinnerc22fd672010-05-15 22:30:53 +00002899 /* Mouse-related constants */
2900 SetDictInt("BUTTON1_PRESSED", BUTTON1_PRESSED);
2901 SetDictInt("BUTTON1_RELEASED", BUTTON1_RELEASED);
2902 SetDictInt("BUTTON1_CLICKED", BUTTON1_CLICKED);
2903 SetDictInt("BUTTON1_DOUBLE_CLICKED", BUTTON1_DOUBLE_CLICKED);
2904 SetDictInt("BUTTON1_TRIPLE_CLICKED", BUTTON1_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002905
Victor Stinnerc22fd672010-05-15 22:30:53 +00002906 SetDictInt("BUTTON2_PRESSED", BUTTON2_PRESSED);
2907 SetDictInt("BUTTON2_RELEASED", BUTTON2_RELEASED);
2908 SetDictInt("BUTTON2_CLICKED", BUTTON2_CLICKED);
2909 SetDictInt("BUTTON2_DOUBLE_CLICKED", BUTTON2_DOUBLE_CLICKED);
2910 SetDictInt("BUTTON2_TRIPLE_CLICKED", BUTTON2_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002911
Victor Stinnerc22fd672010-05-15 22:30:53 +00002912 SetDictInt("BUTTON3_PRESSED", BUTTON3_PRESSED);
2913 SetDictInt("BUTTON3_RELEASED", BUTTON3_RELEASED);
2914 SetDictInt("BUTTON3_CLICKED", BUTTON3_CLICKED);
2915 SetDictInt("BUTTON3_DOUBLE_CLICKED", BUTTON3_DOUBLE_CLICKED);
2916 SetDictInt("BUTTON3_TRIPLE_CLICKED", BUTTON3_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002917
Victor Stinnerc22fd672010-05-15 22:30:53 +00002918 SetDictInt("BUTTON4_PRESSED", BUTTON4_PRESSED);
2919 SetDictInt("BUTTON4_RELEASED", BUTTON4_RELEASED);
2920 SetDictInt("BUTTON4_CLICKED", BUTTON4_CLICKED);
2921 SetDictInt("BUTTON4_DOUBLE_CLICKED", BUTTON4_DOUBLE_CLICKED);
2922 SetDictInt("BUTTON4_TRIPLE_CLICKED", BUTTON4_TRIPLE_CLICKED);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002923
Victor Stinnerc22fd672010-05-15 22:30:53 +00002924 SetDictInt("BUTTON_SHIFT", BUTTON_SHIFT);
2925 SetDictInt("BUTTON_CTRL", BUTTON_CTRL);
2926 SetDictInt("BUTTON_ALT", BUTTON_ALT);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002927
Victor Stinnerc22fd672010-05-15 22:30:53 +00002928 SetDictInt("ALL_MOUSE_EVENTS", ALL_MOUSE_EVENTS);
2929 SetDictInt("REPORT_MOUSE_POSITION", REPORT_MOUSE_POSITION);
Andrew M. Kuchling8e9d23b2000-06-27 03:10:38 +00002930#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002931 /* Now set everything up for KEY_ variables */
2932 {
2933 int key;
2934 char *key_n;
2935 char *key_n2;
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002936#if !defined(__NetBSD__)
Victor Stinnerc22fd672010-05-15 22:30:53 +00002937 for (key=KEY_MIN;key < KEY_MAX; key++) {
2938 key_n = (char *)keyname(key);
2939 if (key_n == NULL || strcmp(key_n,"UNKNOWN KEY")==0)
2940 continue;
2941 if (strncmp(key_n,"KEY_F(",6)==0) {
2942 char *p1, *p2;
2943 key_n2 = malloc(strlen(key_n)+1);
2944 if (!key_n2) {
2945 PyErr_NoMemory();
2946 break;
2947 }
2948 p1 = key_n;
2949 p2 = key_n2;
2950 while (*p1) {
2951 if (*p1 != '(' && *p1 != ')') {
2952 *p2 = *p1;
2953 p2++;
2954 }
2955 p1++;
2956 }
2957 *p2 = (char)0;
2958 } else
2959 key_n2 = key_n;
2960 SetDictInt(key_n2,key);
2961 if (key_n2 != key_n)
2962 free(key_n2);
2963 }
Andrew M. Kuchling2d339f92001-01-29 20:47:59 +00002964#endif
Victor Stinnerc22fd672010-05-15 22:30:53 +00002965 SetDictInt("KEY_MIN", KEY_MIN);
2966 SetDictInt("KEY_MAX", KEY_MAX);
2967 }
2968 return m;
2969 }