better paths
[mikachu/openbox.git] / otk_c / display.c
1 // -*- mode: C; indent-tabs-mode: nil; -*-
2
3 #include "display.h"
4
5 #include <X11/keysym.h>
6
7 #ifdef    SHAPE
8 #include <X11/extensions/shape.h>
9 #endif // SHAPE
10
11 #ifdef    HAVE_STDIO_H
12 #  include <stdio.h>
13 #endif // HAVE_STDIO_H
14
15 #ifdef    HAVE_STDLIB_H
16 #  include <stdlib.h>
17 #endif // HAVE_STDLIB_H
18
19 #ifdef    HAVE_FCNTL_H
20 #  include <fcntl.h>
21 #endif // HAVE_FCNTL_H
22
23 #ifdef    HAVE_UNISTD_H
24 #  include <sys/types.h>
25 #  include <unistd.h>
26 #endif // HAVE_UNISTD_H
27
28 #include "../src/gettext.h"
29
30 extern PyTypeObject OtkDisplay_Type;
31
32 static int xerrorHandler(Display *d, XErrorEvent *e);
33
34 PyObject *OtkDisplay_New(char *name)
35 {
36   OtkDisplay* self;
37   PyObject *disp_env;
38   XModifierKeymap *modmap;
39   unsigned int NumLockMask = 0, ScrollLockMask = 0;
40   size_t cnt;
41   int i;
42   int junk;
43   (void) junk;
44
45   self = PyObject_New(OtkDisplay, &OtkDisplay_Type);
46
47   // Open the X display
48   if (!(self->display = XOpenDisplay(name))) {
49     printf(_("Unable to open connection to the X server. Please set the \n\
50 DISPLAY environment variable approriately, or use the '-display' command \n\
51 line argument.\n\n"));
52     exit(1);
53   }
54   if (fcntl(ConnectionNumber(self->display), F_SETFD, 1) == -1) {
55     printf(_("Couldn't mark display connection as close-on-exec.\n\n"));
56     exit(1);
57   }
58
59   XSetErrorHandler(xerrorHandler);
60
61   // set the DISPLAY environment variable for any lauched children, to the
62   // display we're using, so they open in the right place.
63   disp_env = PyString_FromFormat("DISPLAY=%s", DisplayString(self->display));
64   if (putenv(PyString_AsString(disp_env))) {
65     printf(_("warning: couldn't set environment variable 'DISPLAY'\n"));
66     perror("putenv()");
67   }
68   Py_DECREF(disp_env);
69   
70   // find the availability of X extensions we like to use
71 #ifdef SHAPE
72   self->shape = XShapeQueryExtension(self->display,
73                                      &self->shape_event_basep, &junk);
74 #endif
75
76 #ifdef XINERAMA
77   self->xinerama = XineramaQueryExtension(self->display,
78                                           &self->xinerama_event_basep, &junk);
79 #endif // XINERAMA
80
81   // get lock masks that are defined by the display (not constant)
82   modmap = XGetModifierMapping(self->display);
83   if (modmap && modmap->max_keypermod > 0) {
84     const int mask_table[] = {
85       ShiftMask, LockMask, ControlMask, Mod1Mask,
86       Mod2Mask, Mod3Mask, Mod4Mask, Mod5Mask
87     };
88     const size_t size = (sizeof(mask_table) / sizeof(mask_table[0])) *
89       modmap->max_keypermod;
90     // get the values of the keyboard lock modifiers
91     // Note: Caps lock is not retrieved the same way as Scroll and Num lock
92     // since it doesn't need to be.
93     const KeyCode num_lock = XKeysymToKeycode(self->display, XK_Num_Lock);
94     const KeyCode scroll_lock = XKeysymToKeycode(self->display,
95                                                  XK_Scroll_Lock);
96
97     for (cnt = 0; cnt < size; ++cnt) {
98       if (! modmap->modifiermap[cnt]) continue;
99
100       if (num_lock == modmap->modifiermap[cnt])
101         NumLockMask = mask_table[cnt / modmap->max_keypermod];
102       if (scroll_lock == modmap->modifiermap[cnt])
103         ScrollLockMask = mask_table[cnt / modmap->max_keypermod];
104     }
105   }
106
107   if (modmap) XFreeModifiermap(modmap);
108
109   self->mask_list[0] = 0;
110   self->mask_list[1] = LockMask;
111   self->mask_list[2] = NumLockMask;
112   self->mask_list[3] = LockMask | NumLockMask;
113   self->mask_list[4] = ScrollLockMask;
114   self->mask_list[5] = ScrollLockMask | LockMask;
115   self->mask_list[6] = ScrollLockMask | NumLockMask;
116   self->mask_list[7] = ScrollLockMask | LockMask | NumLockMask;
117
118   // Get information on all the screens which are available.
119   self->screenInfoList = PyList_New(ScreenCount(self->display));
120   for (i = 0; i < ScreenCount(self->display); ++i)
121     PyList_Append(self->screenInfoList, OtkScreenInfo_New(i));
122
123   self->gccache = OtkGCCache_New(PyList_Size(self->screenInfoList));
124   
125   return (PyObject*)self;
126 }
127
128 void OtkDisplay_Grab(OtkDisplay *self)
129 {
130   assert(self);
131   if (self->grab_count == 0)
132     XGrabServer(self->display);
133   self->grab_count++;
134 }
135
136 void OtkDisplay_Ungrab(OtkDisplay *self)
137 {
138   if (self->grab_count == 0) return;
139   self->grab_count--;
140   if (self->grab_count == 0)
141     XUngrabServer(self->display);
142 }
143
144
145
146 static PyObject *otkdisplay_grab(OtkDisplay* self, PyObject* args)
147 {
148   if (!PyArg_ParseTuple(args, ":grab"))
149     return NULL;
150   OtkDisplay_Grab(self);
151   return Py_None;
152 }
153
154 static PyObject *otkdisplay_ungrab(OtkDisplay* self, PyObject* args)
155 {
156   if (!PyArg_ParseTuple(args, ":ungrab"))
157     return NULL;
158   OtkDisplay_Ungrab(self);
159   return Py_None;
160 }
161
162 static PyMethodDef get_methods[] = {
163   {"grab", (PyCFunction)otkdisplay_grab, METH_VARARGS,
164    "Grab the X display."},
165   {"ungrab", (PyCFunction)otkdisplay_ungrab, METH_VARARGS,
166    "Ungrab/Release the X display."},
167   {NULL, NULL, 0, NULL}
168 };
169
170
171
172 static void otkdisplay_dealloc(PyObject* self)
173 {
174   XCloseDisplay(((OtkDisplay*) self)->display);
175   PyObject_Del(((OtkDisplay*) self)->screenInfoList);
176   PyObject_Del(self);
177 }
178
179 static PyObject *otkdisplay_getattr(PyObject *obj, char *name)
180 {
181   return Py_FindMethod(get_methods, obj, name);
182 }
183
184
185 PyTypeObject OtkDisplay_Type = {
186   PyObject_HEAD_INIT(NULL)
187   0,
188   "OtkDisplay",
189   sizeof(OtkDisplay),
190   0,
191   otkdisplay_dealloc, /*tp_dealloc*/
192   0,          /*tp_print*/
193   otkdisplay_getattr, /*tp_getattr*/
194   0,          /*tp_setattr*/
195   0,          /*tp_compare*/
196   0,          /*tp_repr*/
197   0,          /*tp_as_number*/
198   0,          /*tp_as_sequence*/
199   0,          /*tp_as_mapping*/
200   0,          /*tp_hash */
201 };
202
203
204 static int xerrorHandler(Display *d, XErrorEvent *e)
205 {
206 #ifdef DEBUG
207   char errtxt[128];
208
209   //if (e->error_code != BadWindow)
210   {
211     XGetErrorText(d, e->error_code, errtxt, 128);
212     printf("X Error: %s\n", errtxt);
213   }
214 #else
215   (void)d;
216   (void)e;
217 #endif
218
219   return False;
220 }