diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/blueshift_randr.pyx | 11 | ||||
| -rw-r--r-- | src/blueshift_vidmode.pyx | 12 |
2 files changed, 12 insertions, 11 deletions
diff --git a/src/blueshift_randr.pyx b/src/blueshift_randr.pyx index e845f65..80caab1 100644 --- a/src/blueshift_randr.pyx +++ b/src/blueshift_randr.pyx @@ -32,11 +32,6 @@ cdef extern void blueshift_randr_close() cdef unsigned short int* r_c cdef unsigned short int* g_c cdef unsigned short int* b_c -r_c = <unsigned short int*>malloc(256 * 2) -g_c = <unsigned short int*>malloc(256 * 2) -b_c = <unsigned short int*>malloc(256 * 2) -if (r_c is NULL) or (g_c is NULL) or (b_c is NULL): - raise MemoryError() @@ -48,9 +43,15 @@ def randr_open(int use_screen, display): @param display:bytes? The display to use, `None` for the current @return :int Zero on success ''' + global r_c, g_c, b_c cdef char* display_ = NULL if display is not None: display_ = display + r_c = <unsigned short int*>malloc(256 * 2) + g_c = <unsigned short int*>malloc(256 * 2) + b_c = <unsigned short int*>malloc(256 * 2) + if (r_c is NULL) or (g_c is NULL) or (b_c is NULL): + raise MemoryError() return blueshift_randr_open(use_screen, display_) diff --git a/src/blueshift_vidmode.pyx b/src/blueshift_vidmode.pyx index 7ad7a4c..c591db5 100644 --- a/src/blueshift_vidmode.pyx +++ b/src/blueshift_vidmode.pyx @@ -38,11 +38,6 @@ vidmode_gamma_size = 0 cdef unsigned short int* r_c cdef unsigned short int* g_c cdef unsigned short int* b_c -r_c = <unsigned short int*>malloc(256 * 2) -g_c = <unsigned short int*>malloc(256 * 2) -b_c = <unsigned short int*>malloc(256 * 2) -if (r_c is NULL) or (g_c is NULL) or (b_c is NULL): - raise MemoryError() @@ -54,10 +49,15 @@ def vidmode_open(int use_screen, display): @param display:bytes? The display to use, `None` for the current @return :bool Whether call was successful ''' - global vidmode_gamma_size + global vidmode_gamma_size, r_c, g_c, b_c cdef char* display_ = NULL if display is not None: display_ = display + r_c = <unsigned short int*>malloc(256 * 2) + g_c = <unsigned short int*>malloc(256 * 2) + b_c = <unsigned short int*>malloc(256 * 2) + if (r_c is NULL) or (g_c is NULL) or (b_c is NULL): + raise MemoryError() vidmode_gamma_size = blueshift_vidmode_open(use_screen, display_) return vidmode_gamma_size > 1 |