add function for applying curves with drm

Signed-off-by: Mattias Andrée <maandree@operamail.com>

1 files changed, 15 insertions, 0 deletions

diff --git a/src/monitor.py b/src/monitor.py
index 01c70d0..3947435 100644
--- a/src/monitor.py
+++ b/src/monitor.py
@@ -197,6 +197,21 @@ def vidmode(*crtcs, screen = 0):
         pass # Happens on exit by TERM signal
 
 
+def drm(*crtcs, card = 0):
+    '''
+    Applies colour curves using DRM
+    
+    @param  crtcs:*int  The CRT controllers to use, all are used if none are specified
+    @param  card:int    The card that the monitors belong to
+    '''
+    connection = drm_manager.open_card(card)
+    (R_curve, G_curve, B_curve) = translate_to_integers()
+    try:
+        drm_set_gamma_ramps(connection, list(crtcs), i_size, R_curve, G_curve, B_curve)
+    except OverflowError:
+        pass # Happens on exit by TERM signal
+
+
 def print_curves(*crtcs, screen = 0, compact = False):
     '''
     Prints the curves to stdout