/*

* This is a curses module for Python.

*

* Based on prior work by Lance Ellinghaus and Oliver Andrich

* Version 1.2 of this module: Copyright 1994 by Lance Ellinghouse,

* Cathedral City, California Republic, United States of America.

*

* Version 1.5b1, heavily extended for ncurses by Oliver Andrich:

* Copyright 1996,1997 by Oliver Andrich, Koblenz, Germany.

*

* Tidied for Python 1.6, and currently maintained by <amk@amk.ca>.

*

* Permission is hereby granted, free of charge, to any person obtaining

* a copy of this source file to use, copy, modify, merge, or publish it

* subject to the following conditions:

*

* The above copyright notice and this permission notice shall be included

* in all copies or in any new file that contains a substantial portion of

* this file.

*

* THE AUTHOR MAKES NO REPRESENTATIONS ABOUT THE SUITABILITY OF

* THE SOFTWARE FOR ANY PURPOSE. IT IS PROVIDED "AS IS" WITHOUT

* EXPRESS OR IMPLIED WARRANTY. THE AUTHOR DISCLAIMS ALL WARRANTIES

* WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES

* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

* NON-INFRINGEMENT OF THIRD PARTY RIGHTS. IN NO EVENT SHALL THE

* AUTHOR BE LIABLE TO YOU OR ANY OTHER PARTY FOR ANY SPECIAL,

* INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER

* WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE, STRICT LIABILITY OR

* ANY OTHER ACTION ARISING OUT OF OR IN CONNECTION WITH THE USE OR

* PERFORMANCE OF THIS SOFTWARE.

*/

/*

A number of SysV or ncurses functions don't have wrappers yet; if you

need a given function, add it and send a patch. See

http://www.python.org/dev/patches/ for instructions on how to submit

patches to Python.

Here's a list of currently unsupported functions:

addchnstr addchstr color_set define_key

del_curterm delscreen dupwin inchnstr inchstr innstr keyok

mcprint mvaddchnstr mvaddchstr mvcur mvinchnstr

mvinchstr mvinnstr mmvwaddchnstr mvwaddchstr

mvwinchnstr mvwinchstr mvwinnstr newterm

restartterm ripoffline scr_dump

scr_init scr_restore scr_set scrl set_curterm set_term setterm

tgetent tgetflag tgetnum tgetstr tgoto timeout tputs

vidattr vidputs waddchnstr waddchstr

wcolor_set winchnstr winchstr winnstr wmouse_trafo wscrl

Low-priority:

slk_attr slk_attr_off slk_attr_on slk_attr_set slk_attroff

slk_attron slk_attrset slk_clear slk_color slk_init slk_label

slk_noutrefresh slk_refresh slk_restore slk_set slk_touch

Menu extension (ncurses and probably SYSV):

current_item free_item free_menu item_count item_description

item_index item_init item_name item_opts item_opts_off

item_opts_on item_term item_userptr item_value item_visible

menu_back menu_driver menu_fore menu_format menu_grey

menu_init menu_items menu_mark menu_opts menu_opts_off

menu_opts_on menu_pad menu_pattern menu_request_by_name

menu_request_name menu_spacing menu_sub menu_term menu_userptr

menu_win new_item new_menu pos_menu_cursor post_menu

scale_menu set_current_item set_item_init set_item_opts

set_item_term set_item_userptr set_item_value set_menu_back

set_menu_fore set_menu_format set_menu_grey set_menu_init

set_menu_items set_menu_mark set_menu_opts set_menu_pad

set_menu_pattern set_menu_spacing set_menu_sub set_menu_term

set_menu_userptr set_menu_win set_top_row top_row unpost_menu

Form extension (ncurses and probably SYSV):

current_field data_ahead data_behind dup_field

dynamic_fieldinfo field_arg field_back field_buffer

field_count field_fore field_index field_info field_init

field_just field_opts field_opts_off field_opts_on field_pad

field_status field_term field_type field_userptr form_driver

form_fields form_init form_opts form_opts_off form_opts_on

form_page form_request_by_name form_request_name form_sub

form_term form_userptr form_win free_field free_form

link_field link_fieldtype move_field new_field new_form

new_page pos_form_cursor post_form scale_form

set_current_field set_field_back set_field_buffer

set_field_fore set_field_init set_field_just set_field_opts

set_field_pad set_field_status set_field_term set_field_type

set_field_userptr set_fieldtype_arg set_fieldtype_choice

set_form_fields set_form_init set_form_opts set_form_page

set_form_sub set_form_term set_form_userptr set_form_win

set_max_field set_new_page unpost_form

*/

/* Release Number */

char *PyCursesVersion = "2.2";

/* Includes */

#define PY_SSIZE_T_CLEAN

#include "Python.h"

#ifdef __hpux

#define STRICT_SYSV_CURSES

#endif

#define CURSES_MODULE

#include "py_curses.h"

/* These prototypes are in <term.h>, but including this header

#defines many common symbols (such as "lines") which breaks the

curses module in other ways. So the code will just specify

explicit prototypes here. */

extern int setupterm(char *,int,int *);

#ifdef __sgi

#include <term.h>

#endif

#ifdef HAVE_LANGINFO_H

#include <langinfo.h>

#endif

#if !defined(HAVE_NCURSES_H) && (defined(sgi) || defined(__sun) || defined(SCO5))

#define STRICT_SYSV_CURSES /* Don't use ncurses extensions */

typedef chtype attr_t; /* No attr_t type is available */

#endif

#if defined(_AIX)

#define STRICT_SYSV_CURSES

#endif

/*[clinic input]

module curses

class curses.window "PyCursesWindowObject *" "&PyCursesWindow_Type"

[clinic start generated code]*/

/*[clinic end generated code: output=da39a3ee5e6b4b0d input=88c860abdbb50e0c]*/

#include "clinic/_cursesmodule.c.h"

/* Definition of exception curses.error */

static PyObject *PyCursesError;

/* Tells whether setupterm() has been called to initialise terminfo. */

static int initialised_setupterm = FALSE;

/* Tells whether initscr() has been called to initialise curses. */

static int initialised = FALSE;

/* Tells whether start_color() has been called to initialise color usage. */

static int initialisedcolors = FALSE;

static char *screen_encoding = NULL;

/* Utility Macros */

#define PyCursesSetupTermCalled \

if (initialised_setupterm != TRUE) { \

PyErr_SetString(PyCursesError, \

"must call (at least) setupterm() first"); \

return 0; }

#define PyCursesInitialised \

if (initialised != TRUE) { \

PyErr_SetString(PyCursesError, \

"must call initscr() first"); \

return 0; }

#define PyCursesInitialisedColor \

if (initialisedcolors != TRUE) { \

PyErr_SetString(PyCursesError, \

"must call start_color() first"); \

return 0; }

/* Utility Functions */

/*

* Check the return code from a curses function and return None

* or raise an exception as appropriate. These are exported using the

* capsule API.

*/

static PyObject *

PyCursesCheckERR(int code, const char *fname)

{

if (code != ERR) {

Py_RETURN_NONE;

} else {

if (fname == NULL) {

PyErr_SetString(PyCursesError, catchall_ERR);

} else {

PyErr_Format(PyCursesError, "%s() returned ERR", fname);

}

return NULL;

}

}

/* Convert an object to a byte (an integer of type chtype):

- int

- bytes of length 1

- str of length 1

Return 1 on success, 0 on error (invalid type or integer overflow). */

static int

PyCurses_ConvertToChtype(PyCursesWindowObject *win, PyObject *obj, chtype *ch)

{

long value;

if(PyBytes_Check(obj) && PyBytes_Size(obj) == 1) {

value = (unsigned char)PyBytes_AsString(obj)[0];

}

else if (PyUnicode_Check(obj)) {

if (PyUnicode_GetLength(obj) != 1) {

PyErr_Format(PyExc_TypeError,

"expect bytes or str of length 1, or int, "

"got a str of length %zi",

PyUnicode_GET_LENGTH(obj));

return 0;

}

value = PyUnicode_READ_CHAR(obj, 0);

if (128 < value) {

PyObject *bytes;

const char *encoding;

if (win)

encoding = win->encoding;

else

encoding = screen_encoding;

bytes = PyUnicode_AsEncodedString(obj, encoding, NULL);

if (bytes == NULL)

return 0;

if (PyBytes_GET_SIZE(bytes) == 1)

value = (unsigned char)PyBytes_AS_STRING(bytes)[0];

else

value = -1;

Py_DECREF(bytes);

if (value < 0)

goto overflow;

}

}

else if (PyLong_CheckExact(obj)) {

int long_overflow;

value = PyLong_AsLongAndOverflow(obj, &long_overflow);

if (long_overflow)

goto overflow;

}

else {

PyErr_Format(PyExc_TypeError,

"expect bytes or str of length 1, or int, got %s",

Py_TYPE(obj)->tp_name);

return 0;

}

*ch = (chtype)value;

if ((long)*ch != value)

goto overflow;

return 1;

overflow:

PyErr_SetString(PyExc_OverflowError,

"byte doesn't fit in chtype");

return 0;

}

/* Convert an object to a byte (chtype) or a character (cchar_t):

- int

- bytes of length 1

- str of length 1

Return:

- 2 if obj is a character (written into *wch)

- 1 if obj is a byte (written into *ch)

- 0 on error: raise an exception */

static int

PyCurses_ConvertToCchar_t(PyCursesWindowObject *win, PyObject *obj,

chtype *ch

#ifdef HAVE_NCURSESW

, wchar_t *wch

#endif

)

{

long value;

#ifdef HAVE_NCURSESW

wchar_t buffer[2];

#endif

if (PyUnicode_Check(obj)) {

#ifdef HAVE_NCURSESW

if (PyUnicode_AsWideChar(obj, buffer, 2) != 1) {

PyErr_Format(PyExc_TypeError,

"expect bytes or str of length 1, or int, "

"got a str of length %zi",

PyUnicode_GET_LENGTH(obj));

return 0;

}

*wch = buffer[0];

return 2;

#else

return PyCurses_ConvertToChtype(win, obj, ch);

#endif

}

else if(PyBytes_Check(obj) && PyBytes_Size(obj) == 1) {

value = (unsigned char)PyBytes_AsString(obj)[0];

}

else if (PyLong_CheckExact(obj)) {

int overflow;

value = PyLong_AsLongAndOverflow(obj, &overflow);

if (overflow) {

PyErr_SetString(PyExc_OverflowError,

"int doesn't fit in long");

return 0;

}

}

else {

PyErr_Format(PyExc_TypeError,

"expect bytes or str of length 1, or int, got %s",

Py_TYPE(obj)->tp_name);

return 0;

}

*ch = (chtype)value;

if ((long)*ch != value) {

PyErr_Format(PyExc_OverflowError,

"byte doesn't fit in chtype");

return 0;

}

return 1;

}

/* Convert an object to a byte string (char*) or a wide character string

(wchar_t*). Return:

- 2 if obj is a character string (written into *wch)

- 1 if obj is a byte string (written into *bytes)

- 0 on error: raise an exception */

static int

PyCurses_ConvertToString(PyCursesWindowObject *win, PyObject *obj,

PyObject **bytes, wchar_t **wstr)

{

if (PyUnicode_Check(obj)) {

#ifdef HAVE_NCURSESW

assert (wstr != NULL);

*wstr = PyUnicode_AsWideCharString(obj, NULL);

if (*wstr == NULL)

return 0;

return 2;

#else

assert (wstr == NULL);

*bytes = PyUnicode_AsEncodedString(obj, win->encoding, NULL);

if (*bytes == NULL)

return 0;

return 1;

#endif

}

else if (PyBytes_Check(obj)) {

Py_INCREF(obj);

*bytes = obj;

return 1;

}

PyErr_Format(PyExc_TypeError, "expect bytes or str, got %s",

Py_TYPE(obj)->tp_name);

return 0;

}

/* Function versions of the 3 functions for testing whether curses has been

initialised or not. */

static int func_PyCursesSetupTermCalled(void)

{

PyCursesSetupTermCalled;

return 1;

}

static int func_PyCursesInitialised(void)

{

PyCursesInitialised;

return 1;

}

static int func_PyCursesInitialisedColor(void)

{

PyCursesInitialisedColor;

return 1;

}

/*****************************************************************************

The Window Object

******************************************************************************/

/* Definition of the window type */

PyTypeObject PyCursesWindow_Type;

/* Function prototype macros for Window object

X - function name

TYPE - parameter Type

ERGSTR - format string for construction of the return value

PARSESTR - format string for argument parsing

*/

#define Window_NoArgNoReturnFunction(X) \

static PyObject *PyCursesWindow_ ## X \

(PyCursesWindowObject *self, PyObject *args) \

{ return PyCursesCheckERR(X(self->win), # X); }

#define Window_NoArgTrueFalseFunction(X) \

static PyObject * PyCursesWindow_ ## X \

(PyCursesWindowObject *self) \

{ \

if (X (self->win) == FALSE) { Py_RETURN_FALSE; } \

else { Py_RETURN_TRUE; } }

#define Window_NoArgNoReturnVoidFunction(X) \

static PyObject * PyCursesWindow_ ## X \

(PyCursesWindowObject *self) \

{ \

X(self->win); Py_RETURN_NONE; }

#define Window_NoArg2TupleReturnFunction(X, TYPE, ERGSTR) \

static PyObject * PyCursesWindow_ ## X \

(PyCursesWindowObject *self) \

{ \

TYPE arg1, arg2; \

X(self->win,arg1,arg2); return Py_BuildValue(ERGSTR, arg1, arg2); }

#define Window_OneArgNoReturnVoidFunction(X, TYPE, PARSESTR) \

static PyObject * PyCursesWindow_ ## X \

(PyCursesWindowObject *self, PyObject *args) \

{ \

TYPE arg1; \

if (!PyArg_ParseTuple(args, PARSESTR, &arg1)) return NULL; \

X(self->win,arg1); Py_RETURN_NONE; }

#define Window_OneArgNoReturnFunction(X, TYPE, PARSESTR) \

static PyObject * PyCursesWindow_ ## X \

(PyCursesWindowObject *self, PyObject *args) \

{ \

TYPE arg1; \

if (!PyArg_ParseTuple(args,PARSESTR, &arg1)) return NULL; \

return PyCursesCheckERR(X(self->win, arg1), # X); }

#define Window_TwoArgNoReturnFunction(X, TYPE, PARSESTR) \

static PyObject * PyCursesWindow_ ## X \

(PyCursesWindowObject *self, PyObject *args) \

{ \

TYPE arg1, arg2; \

if (!PyArg_ParseTuple(args,PARSESTR, &arg1, &arg2)) return NULL; \

return PyCursesCheckERR(X(self->win, arg1, arg2), # X); }

/* ------------- WINDOW routines --------------- */

Window_NoArgNoReturnFunction(untouchwin)

Window_NoArgNoReturnFunction(touchwin)

Window_NoArgNoReturnFunction(redrawwin)

Window_NoArgNoReturnFunction(winsertln)

Window_NoArgNoReturnFunction(werase)

Window_NoArgNoReturnFunction(wdeleteln)

Window_NoArgTrueFalseFunction(is_wintouched)

Window_NoArgNoReturnVoidFunction(wsyncup)

Window_NoArgNoReturnVoidFunction(wsyncdown)

Window_NoArgNoReturnVoidFunction(wstandend)

Window_NoArgNoReturnVoidFunction(wstandout)

Window_NoArgNoReturnVoidFunction(wcursyncup)

Window_NoArgNoReturnVoidFunction(wclrtoeol)

Window_NoArgNoReturnVoidFunction(wclrtobot)

Window_NoArgNoReturnVoidFunction(wclear)

Window_OneArgNoReturnVoidFunction(idcok, int, "i;True(1) or False(0)")

Window_OneArgNoReturnVoidFunction(immedok, int, "i;True(1) or False(0)")

Window_OneArgNoReturnVoidFunction(wtimeout, int, "i;delay")

Window_NoArg2TupleReturnFunction(getyx, int, "ii")

Window_NoArg2TupleReturnFunction(getbegyx, int, "ii")

Window_NoArg2TupleReturnFunction(getmaxyx, int, "ii")

Window_NoArg2TupleReturnFunction(getparyx, int, "ii")

Window_OneArgNoReturnFunction(clearok, int, "i;True(1) or False(0)")

Window_OneArgNoReturnFunction(idlok, int, "i;True(1) or False(0)")

#if defined(__NetBSD__)

Window_OneArgNoReturnVoidFunction(keypad, int, "i;True(1) or False(0)")

#else

Window_OneArgNoReturnFunction(keypad, int, "i;True(1) or False(0)")

#endif

Window_OneArgNoReturnFunction(leaveok, int, "i;True(1) or False(0)")

#if defined(__NetBSD__)

Window_OneArgNoReturnVoidFunction(nodelay, int, "i;True(1) or False(0)")

#else

Window_OneArgNoReturnFunction(nodelay, int, "i;True(1) or False(0)")

#endif

Window_OneArgNoReturnFunction(notimeout, int, "i;True(1) or False(0)")

Window_OneArgNoReturnFunction(scrollok, int, "i;True(1) or False(0)")

Window_OneArgNoReturnFunction(winsdelln, int, "i;nlines")

Window_OneArgNoReturnFunction(syncok, int, "i;True(1) or False(0)")

Window_TwoArgNoReturnFunction(mvwin, int, "ii;y,x")

Window_TwoArgNoReturnFunction(mvderwin, int, "ii;y,x")

Window_TwoArgNoReturnFunction(wmove, int, "ii;y,x")

#ifndef STRICT_SYSV_CURSES

Window_TwoArgNoReturnFunction(wresize, int, "ii;lines,columns")

#endif

/* Allocation and deallocation of Window Objects */

static PyObject *

PyCursesWindow_New(WINDOW *win, const char *encoding)

{

PyCursesWindowObject *wo;

if (encoding == NULL) {

#if defined(MS_WINDOWS)

char *buffer[100];

UINT cp;

cp = GetConsoleOutputCP();

if (cp != 0) {

PyOS_snprintf(buffer, sizeof(buffer), "cp%u", cp);

encoding = buffer;

}

#elif defined(CODESET)

const char *codeset = nl_langinfo(CODESET);

if (codeset != NULL && codeset[0] != 0)

encoding = codeset;

#endif

if (encoding == NULL)

encoding = "utf-8";

}

wo = PyObject_NEW(PyCursesWindowObject, &PyCursesWindow_Type);

if (wo == NULL) return NULL;

wo->win = win;

wo->encoding = _PyMem_Strdup(encoding);

if (wo->encoding == NULL) {

Py_DECREF(wo);

PyErr_NoMemory();

return NULL;

}

return (PyObject *)wo;

}

static void

PyCursesWindow_Dealloc(PyCursesWindowObject *wo)

{

if (wo->win != stdscr) delwin(wo->win);

if (wo->encoding != NULL)

PyMem_Free(wo->encoding);

PyObject_DEL(wo);

}

/* Addch, Addstr, Addnstr */

/*[clinic input]

curses.window.addch

[

y: int

Y-coordinate.

x: int

X-coordinate.

]

ch: object

Character to add.

[

attr: long

Attributes for the character.

]

/

Paint character ch at (y, x) with attributes attr.

Paint character ch at (y, x) with attributes attr,

overwriting any character previously painted at that location.

By default, the character position and attributes are the

current settings for the window object.

[clinic start generated code]*/

static PyObject *

curses_window_addch_impl(PyCursesWindowObject *self, int group_left_1, int y,

int x, PyObject *ch, int group_right_1, long attr)

/*[clinic end generated code: output=99f7f85078ec06c3 input=5a41efb34a2de338]*/

{

PyCursesWindowObject *cwself = (PyCursesWindowObject *)self;

int coordinates_group = group_left_1;

int attr_group = group_right_1;

int rtn;

int type;

chtype cch = 0;

#ifdef HAVE_NCURSESW

wchar_t wstr[2];

cchar_t wcval;

#endif

const char *funcname;

if (!attr_group)

attr = A_NORMAL;

#ifdef HAVE_NCURSESW

type = PyCurses_ConvertToCchar_t(cwself, ch, &cch, wstr);

if (type == 2) {

funcname = "add_wch";

wstr[1] = L'\0';

setcchar(&wcval, wstr, attr, 0, NULL);

if (coordinates_group)

rtn = mvwadd_wch(cwself->win,y,x, &wcval);

else {

rtn = wadd_wch(cwself->win, &wcval);

}

}

else

#else

type = PyCurses_ConvertToCchar_t(cwself, ch, &cch);

#endif

if (type == 1) {

funcname = "addch";

if (coordinates_group)

rtn = mvwaddch(cwself->win,y,x, cch | attr);

else {

rtn = waddch(cwself->win, cch | attr);

}

}

else {

return NULL;

}

return PyCursesCheckERR(rtn, funcname);

}

static PyObject *

PyCursesWindow_AddStr(PyCursesWindowObject *self, PyObject *args)

{

int rtn;

int x, y;

int strtype;

PyObject *strobj, *bytesobj = NULL;

#ifdef HAVE_NCURSESW

wchar_t *wstr = NULL;

#endif

attr_t attr = A_NORMAL , attr_old = A_NORMAL;

long lattr;

int use_xy = FALSE, use_attr = FALSE;

const char *funcname;

switch (PyTuple_Size(args)) {

case 1:

if (!PyArg_ParseTuple(args,"O;str", &strobj))

return NULL;

break;

case 2:

if (!PyArg_ParseTuple(args,"Ol;str,attr", &strobj, &lattr))

return NULL;

attr = lattr;

use_attr = TRUE;

break;

case 3:

if (!PyArg_ParseTuple(args,"iiO;int,int,str", &y, &x, &strobj))

return NULL;

use_xy = TRUE;

break;

case 4:

if (!PyArg_ParseTuple(args,"iiOl;int,int,str,attr", &y, &x, &strobj, &lattr))

return NULL;

attr = lattr;

use_xy = use_attr = TRUE;

break;

default:

PyErr_SetString(PyExc_TypeError, "addstr requires 1 to 4 arguments");

return NULL;

}

#ifdef HAVE_NCURSESW

strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);

#else

strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);

#endif

if (strtype == 0)

return NULL;

if (use_attr == TRUE) {

attr_old = getattrs(self->win);

(void)wattrset(self->win,attr);

}

#ifdef HAVE_NCURSESW

if (strtype == 2) {

funcname = "addwstr";

if (use_xy == TRUE)

rtn = mvwaddwstr(self->win,y,x,wstr);

else

rtn = waddwstr(self->win,wstr);

PyMem_Free(wstr);

}

else

#endif

{

char *str = PyBytes_AS_STRING(bytesobj);

funcname = "addstr";

if (use_xy == TRUE)

rtn = mvwaddstr(self->win,y,x,str);

else

rtn = waddstr(self->win,str);

Py_DECREF(bytesobj);

}

if (use_attr == TRUE)

(void)wattrset(self->win,attr_old);

return PyCursesCheckERR(rtn, funcname);

}

static PyObject *

PyCursesWindow_AddNStr(PyCursesWindowObject *self, PyObject *args)

{

int rtn, x, y, n;

int strtype;

PyObject *strobj, *bytesobj = NULL;

#ifdef HAVE_NCURSESW

wchar_t *wstr = NULL;

#endif

attr_t attr = A_NORMAL , attr_old = A_NORMAL;

long lattr;

int use_xy = FALSE, use_attr = FALSE;

const char *funcname;

switch (PyTuple_Size(args)) {

case 2:

if (!PyArg_ParseTuple(args,"Oi;str,n", &strobj, &n))

return NULL;

break;

case 3:

if (!PyArg_ParseTuple(args,"Oil;str,n,attr", &strobj, &n, &lattr))

return NULL;

attr = lattr;

use_attr = TRUE;

break;

case 4:

if (!PyArg_ParseTuple(args,"iiOi;y,x,str,n", &y, &x, &strobj, &n))

return NULL;

use_xy = TRUE;

break;

case 5:

if (!PyArg_ParseTuple(args,"iiOil;y,x,str,n,attr", &y, &x, &strobj, &n, &lattr))

return NULL;

attr = lattr;

use_xy = use_attr = TRUE;

break;

default:

PyErr_SetString(PyExc_TypeError, "addnstr requires 2 to 5 arguments");

return NULL;

}

#ifdef HAVE_NCURSESW

strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, &wstr);

#else

strtype = PyCurses_ConvertToString(self, strobj, &bytesobj, NULL);

#endif

if (strtype == 0)

return NULL;

if (use_attr == TRUE) {

attr_old = getattrs(self->win);

(void)wattrset(self->win,attr);

}

#ifdef HAVE_NCURSESW

if (strtype == 2) {

funcname = "addnwstr";

if (use_xy == TRUE)

rtn = mvwaddnwstr(self->win,y,x,wstr,n);

else

rtn = waddnwstr(self->win,wstr,n);

PyMem_Free(wstr);

}

else

#endif

{

char *str = PyBytes_AS_STRING(bytesobj);

funcname = "addnstr";

if (use_xy == TRUE)

rtn = mvwaddnstr(self->win,y,x,str,n);

else

rtn = waddnstr(self->win,str,n);

Py_DECREF(bytesobj);

}

if (use_attr == TRUE)

(void)wattrset(self->win,attr_old);

return PyCursesCheckERR(rtn, funcname);

}

static PyObject *

PyCursesWindow_Bkgd(PyCursesWindowObject *self, PyObject *args)

{

PyObject *temp;

chtype bkgd;

attr_t attr = A_NORMAL;

long lattr;

switch (PyTuple_Size(args)) {

case 1:

if (!PyArg_ParseTuple(args, "O;ch or int", &temp))

return NULL;

break;

case 2:

if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))

return NULL;

attr = lattr;

break;

default:

PyErr_SetString(PyExc_TypeError, "bkgd requires 1 or 2 arguments");

return NULL;

}

if (!PyCurses_ConvertToChtype(self, temp, &bkgd))

return NULL;

return PyCursesCheckERR(wbkgd(self->win, bkgd | attr), "bkgd");

}

static PyObject *

PyCursesWindow_AttrOff(PyCursesWindowObject *self, PyObject *args)

{

long lattr;

if (!PyArg_ParseTuple(args,"l;attr", &lattr))

return NULL;

return PyCursesCheckERR(wattroff(self->win, (attr_t)lattr), "attroff");

}

static PyObject *

PyCursesWindow_AttrOn(PyCursesWindowObject *self, PyObject *args)

{

long lattr;

if (!PyArg_ParseTuple(args,"l;attr", &lattr))

return NULL;

return PyCursesCheckERR(wattron(self->win, (attr_t)lattr), "attron");

}

static PyObject *

PyCursesWindow_AttrSet(PyCursesWindowObject *self, PyObject *args)

{

long lattr;

if (!PyArg_ParseTuple(args,"l;attr", &lattr))

return NULL;

return PyCursesCheckERR(wattrset(self->win, (attr_t)lattr), "attrset");

}

static PyObject *

PyCursesWindow_BkgdSet(PyCursesWindowObject *self, PyObject *args)

{

PyObject *temp;

chtype bkgd;

attr_t attr = A_NORMAL;

long lattr;

switch (PyTuple_Size(args)) {

case 1:

if (!PyArg_ParseTuple(args, "O;ch or int", &temp))

return NULL;

break;

case 2:

if (!PyArg_ParseTuple(args,"Ol;ch or int,attr", &temp, &lattr))

return NULL;

attr = lattr;

break;

default:

PyErr_SetString(PyExc_TypeError, "bkgdset requires 1 or 2 arguments");

return NULL;

}

if (!PyCurses_ConvertToChtype(self, temp, &bkgd))

return NULL;

wbkgdset(self->win, bkgd | attr);

return PyCursesCheckERR(0, "bkgdset");

}

static PyObject *

PyCursesWindow_Border(PyCursesWindowObject *self, PyObject *args)

{

PyObject *temp[8];

chtype ch[8];

int i;

/* Clear the array of parameters */

for(i=0; i<8; i++) {

temp[i] = NULL;

ch[i] = 0;

}

if (!PyArg_ParseTuple(args,"|OOOOOOOO;ls,rs,ts,bs,tl,tr,bl,br",

&temp[0], &temp[1], &temp[2], &temp[3],

&temp[4], &temp[5], &temp[6], &temp[7]))

return NULL;

for(i=0; i<8; i++) {

if (temp[i] != NULL && !PyCurses_ConvertToChtype(self, temp[i], &ch[i]))

return NULL;

}

wborder(self->win,

ch[0], ch[1], ch[2], ch[3],

ch[4], ch[5], ch[6], ch[7]);

Py_RETURN_NONE;

}

static PyObject *

PyCursesWindow_Box(PyCursesWindowObject *self, PyObject *args)

{

chtype ch1=0,ch2=0;

switch(PyTuple_Size(args)){

case 0: break;

default:

if (!PyArg_ParseTuple(args,"ll;vertint,horint", &ch1, &ch2))

return NULL;

}

box(self->win,ch1,ch2);

Py_RETURN_NONE;

}

#if defined(HAVE_NCURSES_H) || defined(MVWDELCH_IS_EXPRESSION)

#define py_mvwdelch mvwdelch

#else

int py_mvwdelch(WINDOW *w, int y, int x)

{

mvwdelch(w,y,x);

/* On HP/UX, mvwdelch already returns. On other systems,

we may well run into this return statement. */

return 0;

}

#endif

/* chgat, added by Fabian Kreutz <fabian.kreutz at gmx.net> */

static PyObject *

PyCursesWindow_ChgAt(PyCursesWindowObject *self, PyObject *args)

{

int rtn;

int x, y;

int num = -1;

short color;

attr_t attr = A_NORMAL;

long lattr;

int use_xy = FALSE;

switch (PyTuple_Size(args)) {

case 1:

if (!PyArg_ParseTuple(args,"l;attr", &lattr))

return NULL;

attr = lattr;

break;

case 2:

if (!PyArg_ParseTuple(args,"il;n,attr", &num, &lattr))

return NULL;

attr = lattr;

break;

case 3:

if (!PyArg_ParseTuple(args,"iil;int,int,attr", &y, &x, &lattr))

return NULL;

attr = lattr;

use_xy = TRUE;

break;

case 4:

if (!PyArg_ParseTuple(args,"iiil;int,int,n,attr", &y, &x, &num, &lattr))

return NULL;

attr = lattr;

use_xy = TRUE;

break;

default:

PyErr_SetString(PyExc_TypeError, "chgat requires 1 to 4 arguments");

return NULL;

}

color = (short)((attr >> 8) & 0xff);

attr = attr - (color << 8);

if (use_xy == TRUE) {

rtn = mvwchgat(self->win,y,x,num,attr,color,NULL);

touchline(self->win,y,1);

} else {

getyx(self->win,y,x);

rtn = wchgat(self->win,num,attr,color,NULL);

touchline(self->win,y,1);

}

return PyCursesCheckERR(rtn, "chgat");

}

static PyObject *

PyCursesWindow_DelCh(PyCursesWindowObject *self, PyObject *args)

{

int rtn;

int x, y;

switch (PyTuple_Size(args)) {

case 0:

rtn = wdelch(self->win);

break;

case 2:

if (!PyArg_ParseTuple(args,"ii;y,x", &y, &x))

return NULL;

rtn = py_mvwdelch(self->win,y,x);