1 files changed, 128 insertions, 0 deletions

diff --git a/src/gamma-quartz-cg.c b/src/gamma-quartz-cg.c
index 598780f..99d46e1 100644
--- a/src/gamma-quartz-cg.c
+++ b/src/gamma-quartz-cg.c
@@ -23,6 +23,14 @@
 
 #include "libgamma-error.h"
 
+#ifdef FAKE_GAMMA_METHOD_QUARTZ_CORE_GRAPHICS
+# include "fake-quartz-cg.h"
+#else
+# include <CoreGraphics/CGDirectDisplay.h>
+# define close_fake_quartz_cg() /* For compatibility with "fake-quartz-cg.h". */
+#endif
+
+#include <stdlib.h>
 #include <errno.h>
 
 
@@ -76,6 +84,11 @@ void libgamma_quartz_cg_method_capabilities(libgamma_method_capabilities_t* rest
 int libgamma_quartz_cg_site_initialise(libgamma_site_state_t* restrict this,
 				       char* restrict site)
 {
+  if (site != NULL)
+    return LIBGAMMA_NO_SUCH_SITE;
+  
+  this->partitions_available = 1;
+  return 0;
 }
 
 
@@ -86,6 +99,8 @@ int libgamma_quartz_cg_site_initialise(libgamma_site_state_t* restrict this,
  */
 void libgamma_quartz_cg_site_destroy(libgamma_site_state_t* restrict this)
 {
+  (void) this;
+  close_fake_quartz_cg();
 }
 
 
@@ -98,6 +113,9 @@ void libgamma_quartz_cg_site_destroy(libgamma_site_state_t* restrict this)
  */
 int libgamma_quartz_cg_site_restore(libgamma_site_state_t* restrict this)
 {
+  (void) this;
+  CGDisplayRestoreColorSyncSettings();
+  return 0;
 }
 
 
@@ -114,6 +132,49 @@ int libgamma_quartz_cg_site_restore(libgamma_site_state_t* restrict this)
 int libgamma_quartz_cg_partition_initialise(libgamma_partition_state_t* restrict this,
 					    libgamma_site_state_t* restrict site, size_t partition)
 {
+  uint32_t cap = 4, n;
+  CGDirectDisplayID* crtcs;
+  CGDirectDisplayID* crtcs_old;
+  CGError r;
+  
+  (void) site;
+  
+  this->data = NULL;
+  
+  if (partition != 0)
+    return LIBGAMMA_NO_SUCH_PARTITION;
+  
+  crtcs = malloc((size_t)cap * sizeof(CGDirectDisplayID));
+  if (crtcs == NULL)
+    return LIBGAMMA_ERRNO_SET;
+  
+  for (;;)
+    {
+      r = CGGetOnlineDisplayList(cap, crtcs, &n);
+      if (r != kCGErrorSuccess)
+	{
+	  free(crtcs);
+	  return LIBGAMMA_LIST_CRTCS_FAILED;
+	}
+      if (n < cap)
+	break;
+      if (cap == 0) /* We could also test ~0, but it is still too many. */
+	{
+	  free(crtcs);
+	  return LIBGAMMA_IMPOSSIBLE_AMOUNT;
+	}
+      crtcs_old = crtcs;
+      crtcs = realloc(crtcs, (size_t)cap * sizeof(CGDirectDisplayID));
+      if (crtcs)
+	{
+	  free(crtcs_old);
+	  return LIBGAMMA_ERRNO_SET;
+	}
+    }
+  
+  this->data = crtcs;
+  this->crtcs_available = (size_t)n;
+  return 0;
 }
 
 
@@ -124,6 +185,7 @@ int libgamma_quartz_cg_partition_initialise(libgamma_partition_state_t* restrict
  */
 void libgamma_quartz_cg_partition_destroy(libgamma_partition_state_t* restrict this)
 {
+  free(this->data);
 }
 
 
@@ -153,6 +215,10 @@ int libgamma_quartz_cg_partition_restore(libgamma_partition_state_t* restrict th
 int libgamma_quartz_cg_crtc_initialise(libgamma_crtc_state_t* restrict this,
 				       libgamma_partition_state_t* restrict partition, size_t crtc)
 {
+  (void) this;
+  if (crtc >= partition->crtcs_available)
+    return LIBGAMMA_NO_SUCH_CRTC;
+  return 0;
 }
 
 
@@ -163,6 +229,7 @@ int libgamma_quartz_cg_crtc_initialise(libgamma_crtc_state_t* restrict this,
  */
 void libgamma_quartz_cg_crtc_destroy(libgamma_crtc_state_t* restrict this)
 {
+  (void) this;
 }
 
 
@@ -192,6 +259,41 @@ int libgamma_quartz_cg_crtc_restore(libgamma_crtc_state_t* restrict this)
 int libgamma_quartz_cg_get_crtc_information(libgamma_crtc_information_t* restrict this,
 					    libgamma_crtc_state_t* restrict crtc, int32_t fields)
 {
+#define SUPPORTED_FIELDS  (CRTC_INFO_GAMMA_SIZE | CRTC_INFO_GAMMA_DEPTH)
+   
+#define _E(FIELD)  ((fields & FIELD) ? LIBGAMMA_CRTC_INFO_NOT_SUPPORTED : 0)
+  
+  this->edid_error = _E(CRTC_INFO_EDID);
+  this->width_mm_error = _E(CRTC_INFO_WIDTH_MM);
+  this->height_mm_error = _E(CRTC_INFO_HEIGHT_MM);
+  this->width_mm_edid_error = _E(CRTC_INFO_WIDTH_MM_EDID);
+  this->height_mm_edid_error = _E(CRTC_INFO_HEIGHT_MM_EDID);
+  if ((fields & CRTC_INFO_GAMMA_SIZE))
+    {
+      CGDirectDisplayID* crtcs = crtc->partition->data;
+      CGDirectDisplayID crtc_id = crtcs[crtc->crtc];
+      size_t gamma_size = CGDisplayGammaTableCapacity(crtc_id);;
+      this->red_gamma_size = (size_t)gamma_size;
+      this->green_gamma_size = this->red_gamma_size;
+      this->blue_gamma_size = this->red_gamma_size;
+      this->gamma_size_error = gamma_size < 2 ? LIBGAMMA_SINGLETON_GAMMA_RAMP : 0;
+    }
+  else
+    this->gamma_size_error = 0;
+  this->gamma_depth = -1;
+  this->gamma_depth_error = 0;
+  this->gamma_support_error = _E(CRTC_INFO_GAMMA_SUPPORT);
+  this->subpixel_order_error = _E(CRTC_INFO_SUBPIXEL_ORDER);
+  this->active_error = _E(CRTC_INFO_ACTIVE);
+  this->connector_name_error = _E(CRTC_INFO_CONNECTOR_NAME);
+  this->connector_type_error = _E(CRTC_INFO_CONNECTOR_TYPE);
+  this->gamma_error = _E(CRTC_INFO_GAMMA);
+  
+#undef _E
+  
+  return this->gamma_size_error ? -1 : (fields & ~SUPPORTED_FIELDS) ? -1 : 0;
+  
+#undef SUPPORTED_FIELDS
 }
 
 
@@ -206,6 +308,22 @@ int libgamma_quartz_cg_get_crtc_information(libgamma_crtc_information_t* restric
 int libgamma_quartz_cg_crtc_get_gamma_rampsf(libgamma_crtc_state_t* restrict this,
 					     libgamma_gamma_rampsf_t* restrict ramps)
 {
+  CGDirectDisplayID* crtcs = this->partition->data;
+  CGDirectDisplayID crtc_id = crtcs[this->crtc];
+  uint32_t gamma_size_out;
+  CGError r;
+#ifdef DEBUG
+  if ((ramps->red_size != ramps->green_size) ||
+      (ramps->red_size != ramps->blue_size))
+    return LIBGAMMA_MIXED_GAMMA_RAMP_SIZE;
+#endif
+  r = CGGetDisplayTransferByTable(crtc_id, (uint32_t)(ramps->red_size),
+				  ramps->red, ramps->green, ramps->blue, &gamma_size_out);
+  if (r != kCGErrorSuccess)
+    return LIBGAMMA_GAMMA_RAMP_READ_FAILED;
+  if (gamma_size_out != ramps->red_size)
+    return LIBGAMMA_GAMMA_RAMP_SIZE_CHANGED;
+  return 0;
 }
 
 
@@ -220,5 +338,15 @@ int libgamma_quartz_cg_crtc_get_gamma_rampsf(libgamma_crtc_state_t* restrict thi
 int libgamma_quartz_cg_crtc_set_gamma_rampsf(libgamma_crtc_state_t* restrict this,
 					     libgamma_gamma_rampsf_t ramps)
 {
+  CGDirectDisplayID* crtcs = this->partition->data;
+  CGDirectDisplayID crtc_id = crtcs[this->crtc];
+  CGError r;
+#ifdef DEBUG
+  if ((ramps.red_size != ramps.green_size) || (ramps.red_size != ramps.blue_size))
+    return LIBGAMMA_MIXED_GAMMA_RAMP_SIZE;
+#endif
+  r = CGSetDisplayTransferByTable(crtc_id, (uint32_t)(ramps.red_size),
+				  ramps.red, ramps.green, ramps.blue);
+  return r == kCGErrorSuccess ? 0 : LIBGAMMA_GAMMA_RAMP_WRITE_FAILED;
 }