2 files changed, 23 insertions, 20 deletions
diff --git a/src/_blueshift_randr.pyx b/src/_blueshift_randr.pyx
new file mode 100644
index 0000000..44fd4b6
--- /dev/null
+++ b/src/_blueshift_randr.pyx
@@ -0,0 +1,23 @@
+# -*- python -*-
+cdef extern int blueshift_randr_open(int use_screen)
+cdef extern int blueshift_randr_apply(unsigned long long int use_crtcs,
+                                      unsigned short int r_curve[256],
+                                      unsigned short int g_curve[256],
+                                      unsigned short int b_curve[256])
+cdef extern void blueshift_randr_close()
+
+
+def randr_open(int use_screen):
+    return blueshift_randr_open(use_screen)
+
+
+def randr_apply(unsigned long long int use_crtcs,
+                unsigned short int r_curve[256],
+                unsigned short int g_curve[256],
+                unsigned short int b_curve[256]):
+    return blueshift_randr_apply(use_crtcs, r_curve, g_curve, b_curve)
+
+
+def randr_close():
+    blueshift_randr_close()
+
diff --git a/src/blueshift_randr.c b/src/blueshift_randr_c.c
index 98ba2a3..bf2656e 100644
--- a/src/blueshift_randr.c
+++ b/src/blueshift_randr_c.c
@@ -290,23 +290,3 @@ void blueshift_randr_close(void)
   xcb_disconnect(connection);
 }
 
-
-
-int main(int argc, char** argv)
-{
-  long i;
-  
-  (void) argc;
-  (void) argv;
-  
-  if (blueshift_randr_open(0))
-    return 1;
-  
-  for (i = 0; i < 256; i++)
-    r_curve[i] = g_curve[i] = b_curve[i] = (int)((float)i / 255.f * (float)((1 << 16) - 1) + 0.f);
-  
-  i = blueshift_randr_apply(~0, r_curve, g_curve, b_curve);
-  blueshift_randr_close();
-  return i;
-}
-