diff options
Diffstat (limited to 'src/blueshift_drm.pyx')
| -rw-r--r-- | src/blueshift_drm.pyx | 307 |
1 files changed, 307 insertions, 0 deletions
diff --git a/src/blueshift_drm.pyx b/src/blueshift_drm.pyx new file mode 100644 index 0000000..30f07c1 --- /dev/null +++ b/src/blueshift_drm.pyx @@ -0,0 +1,307 @@ +# -*- python -*- + +# Copyright © 2014 Mattias Andrée (maandree@member.fsf.org) +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +cimport cython +from libc.stdlib cimport malloc, free + + +cdef extern void blueshift_drm_close() +cdef extern int blueshift_drm_card_count() +cdef extern int blueshift_drm_open_card(int card_index) +cdef extern void blueshift_drm_update_card(int connection) +cdef extern void blueshift_drm_close_card(int connection) +cdef extern int blueshift_drm_crtc_count(int connection) +cdef extern int blueshift_drm_connector_count(int connection) +cdef extern int blueshift_drm_gamma_size(int connection, int crtc_index) +cdef extern int blueshift_drm_get_gamma_ramps(int connection, int crtc_index, int gamma_size, + unsigned short int* red, + unsigned short int* green, + unsigned short int* blue) +cdef extern int blueshift_drm_set_gamma_ramps(int connection, int crtc_index, int gamma_size, + unsigned short int* red, + unsigned short int* green, + unsigned short int* blue) +cdef extern void blueshift_drm_open_connector(int connection, int connector_index) +cdef extern void blueshift_drm_close_connector(int connection, int connector_index) +cdef extern int blueshift_drm_get_width(int connection, int connector_index) +cdef extern int blueshift_drm_get_height(int connection, int connector_index) +cdef extern int blueshift_drm_is_connected(int connection, int connector_index) +cdef extern int blueshift_drm_get_crtc(int connection, int connector_index) +cdef extern int blueshift_drm_get_connector_type_index(int connection, int connector_index) +cdef extern const char* blueshift_drm_get_connector_type_name(int connection, int connector_index) +cdef extern long int blueshift_drm_get_edid(int connection, int connector_index, char* edid, + long int size, int hexadecimal) + + +def drm_close(): + ''' + Free all resources, but you need to close all connections first + ''' + blueshift_drm_close() + + +def drm_card_count(): + ''' + Get the number of cards present on the system + + @return :int The number of cards present on the system + ''' + return blueshift_drm_card_count() + + +def drm_open_card(int card_index): + ''' + Open connection to a graphics card + + @param card_index The index of the graphics card + @return :int -1 on failure, otherwise an identifier for the connection to the card + ''' + return blueshift_drm_open_card(card_index) + + +def drm_update_card(int connection): + ''' + Update the resource, required after `blueshift_drm_open_card` + + @param connection The identifier for the connection to the card + ''' + blueshift_drm_update_card(connection) + + +def drm_close_card(int connection): + ''' + Close connection to the graphics card + + @param connection The identifier for the connection to the card + ''' + blueshift_drm_close_card(connection) + + +def drm_crtc_count(int connection): + ''' + Return the number of CRTC:s on the opened card + + @param connection The identifier for the connection to the card + @return :int The number of CRTC:s on the opened card + ''' + return blueshift_drm_crtc_count(connection) + + +def drm_connector_count(int connection): + ''' + Return the number of connectors on the opened card + + @param connection The identifier for the connection to the card + @return :int The number of connectors on the opened card + ''' + return blueshift_drm_connector_count(connection) + + +def drm_gamma_size(int connection, int crtc_index): + ''' + Return the size of the gamma ramps on a CRTC + + @param connection The identifier for the connection to the card + @param crtc_index The index of the CRTC + @return :int The size of the gamma ramps on a CRTC + ''' + return blueshift_drm_gamma_size(connection, crtc_index) + + +def drm_get_gamma_ramps(int connection, int crtc_index, int gamma_size): + ''' + Get the gamma ramps of the of a monitor + + @param connection The identifier for the connection to the card + @param crtc_index The index of the CRTC to read from + @param gamma_size The size a gamma ramp + @return :(r:list<int>, g:list<int>, b:list<int>)? The current red, green and blue colour curves + ''' + cdef unsigned short int* r + cdef unsigned short int* g + cdef unsigned short int* b + r = <unsigned short int*>malloc(gamma_size * 2) + g = <unsigned short int*>malloc(gamma_size * 2) + b = <unsigned short int*>malloc(gamma_size * 2) + if (r is NULL) or (g is NULL) or (b is NULL): + raise MemoryError() + rc = blueshift_drm_get_gamma_ramps(connection, crtc_index, gamma_size, r, g, b) + if rc == 0: + rc_r, rc_g, rc_b = [], [], [] + for i in range(gamma_size): + rc_r.append(r[i]) + rc_g.append(g[i]) + rc_b.append(b[i]) + free(r) + free(g) + free(b) + return (rc_r, rc_g, rc_b) + else: + free(r) + free(g) + free(b) + return None + + +def drm_set_gamma_ramps(int connection, int crtc_index, int gamma_size, r_curve, g_curve, b_curve): + ''' + Set the gamma ramps of the of a monitor + + @param connection The identifier for the connection to the card + @param crtc_index The index of the CRTC to read from + @param gamma_size The size a gamma ramp + @param r_curve:list<unsigned short int> The red gamma ramp + @param g_curve:list<unsigned short int> The green gamma ramp + @param b_curve:list<unsigned short int> The blue gamma ramp + @return :int Zero on success + ''' + cdef unsigned short int* r + cdef unsigned short int* g + cdef unsigned short int* b + r = <unsigned short int*>malloc(gamma_size * 2) + g = <unsigned short int*>malloc(gamma_size * 2) + b = <unsigned short int*>malloc(gamma_size * 2) + if (r is NULL) or (g is NULL) or (b is NULL): + raise MemoryError() + for i in range(gamma_size): + r[i] = r_curve[i] & 0xFFFF + g[i] = g_curve[i] & 0xFFFF + b[i] = b_curve[i] & 0xFFFF + rc = blueshift_drm_set_gamma_ramps(connection, crtc_index, gamma_size, r, g, b) + free(r) + free(g) + free(b) + return rc + + +def drm_open_connector(int connection, int connector_index): + ''' + Acquire information about a connector + + @param connection The identifier for the connection to the card + @param connector_index The index of the connector + ''' + blueshift_drm_open_connector(connection, connector_index) + + +def drm_close_connector(int connection, int connector_index): + ''' + Release information about a connector + + @param connection The identifier for the connection to the card + @param connector_index The index of the connector + ''' + blueshift_drm_close_connector(connection, connector_index) + + +def drm_get_width(int connection, int connector_index): + ''' + Get the physical width the monitor connected to a connector + + @param connection The identifier for the connection to the card + @param connector_index The index of the connector + @return :int The physical width of the monitor in millimetres, 0 if unknown or not connected + ''' + return blueshift_drm_get_width(connection, connector_index) + + +def drm_get_height(int connection, int connector_index): + ''' + Get the physical height the monitor connected to a connector + + @param connection The identifier for the connection to the card + @param connector_index The index of the connector + @return :int The physical height of the monitor in millimetres, 0 if unknown or not connected + ''' + return blueshift_drm_get_height(connection, connector_index) + + +def drm_is_connected(int connection, int connector_index): + ''' + Get whether a monitor is connected to a connector + + @param connection The identifier for the connection to the card + @param connector_index The index of the connector + @return :int 1 if there is a connection, 0 otherwise, -1 if unknown + ''' + return blueshift_drm_is_connected(connection, connector_index) + + +def drm_get_crtc(int connection, int connector_index): + ''' + Get the index of the CRTC of the monitor connected to a connector + + @param connection The identifier for the connection to the card + @param connector_index The index of the connector + @return :int The index of the CRTC + ''' + return blueshift_drm_get_crtc(connection, connector_index) + + +def drm_get_connector_type_index(int connection, int connector_index): + ''' + Get the index of the type of a connector + + @param connection The identifier for the connection to the card + @param connector_index The index of the connector + @return :int The connector type by index, 0 for unknown + ''' + return blueshift_drm_get_connector_type_index(connection, connector_index) + + +def drm_get_connector_type_name(int connection, int connector_index): + ''' + Get the name of the type of a connector + + @param connection The identifier for the connection to the card + @param connector_index The index of the connector + @return :str The connector type by name, "Unknown" if not identifiable, + "Unrecognised" if Blueshift does not recognise it. + ''' + return <bytes>blueshift_drm_get_connector_type_name(connection, connector_index) + + +def drm_get_edid(int connection, int connector_index): + ''' + Get the extended display identification data for the monitor connected to a connector + + @param connection The identifier for the connection to the card + @param connector_index The index of the connector + @return :str? The extended display identification data for the monitor + ''' + cdef long int size + cdef long int got + cdef char* edid + cdef bytes rc + + size = 256 + edid = <char*>malloc(size + 1) + got = blueshift_drm_get_edid(connection, connector_index, edid, size, 1) + + if got == 0: + free(edid) + return None + + if got * 2 > size: + size = got + blueshift_drm_get_edid(connection, connector_index, edid, size, 1) + + edid[got * 2] = 0 + rc = edid + free(edid) + return rc + |