/**
 * libgamma — Display server abstraction layer for gamma ramp adjustments
 * Copyright © 2014  Mattias Andrée (maandree@member.fsf.org)
 * 
 * This library 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 library 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 library.  If not, see <http://www.gnu.org/licenses/>.
 */
#ifndef HAVE_GAMMA_METHOD_QUARTZ_CORE_GRAPHICS
# error Compiling gamma-quartz-cg.c without HAVE_GAMMA_METHOD_QUARTZ_CORE_GRAPHICS
#endif

#include "gamma-quartz-cg.h"

#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>


/**
 * Return the capabilities of the adjustment method
 * 
 * @param  this  The data structure to fill with the method's capabilities
 */
void libgamma_quartz_cg_method_capabilities(libgamma_method_capabilities_t* restrict this)
{
  this->crtc_information = CRTC_INFO_GAMMA_SIZE
			 | CRTC_INFO_GAMMA_DEPTH;
  this->default_site_known = 1;
  this->multiple_sites = 0;
  this->multiple_partitions = 0;
  this->multiple_crtcs = 1;
  this->partitions_are_graphics_cards = 0;
  this->site_restore = 1;
  this->partition_restore = 1;
  this->crtc_restore = 0;
  this->identical_gamma_sizes = 1;
  this->fixed_gamma_size = 0;
  this->fixed_gamma_depth = 1;
#ifdef FAKE_GAMMA_METHOD_QUARTZ_CORE_GRAPHICS
  this->fake = 1;
# ifdef HAVE_GAMMA_METHOD_X_RANDR
  this->real = 1;
# else
  this->real = 0;
# endif
#else
  this->fake = 0;
  this->real = 1;
#endif
}


/**
 * Initialise an allocated site state
 * 
 * @param   this    The site state to initialise
 * @param   site    The site identifier, unless it is `NULL` it must a
 *                  `free`:able. One the state is destroyed the library
 *                  will attempt to free it. There you should not free
 *                  it yourself, and it must not be a string constant
 *                  or allocate on the stack. Note however that it will
 *                  not be free:d if this function fails.
 * @return          Zero on success, otherwise (negative) the value of an
 *                  error identifier provided by this library
 */
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;
}


/**
 * Release all resources held by a site state
 * 
 * @param  this  The site state
 */
void libgamma_quartz_cg_site_destroy(libgamma_site_state_t* restrict this)
{
  (void) this;
  close_fake_quartz_cg();
}


/**
 * Restore the gamma ramps all CRTCS with a site to the system settings
 * 
 * @param   this  The site state
 * @return        Zero on success, otherwise (negative) the value of an
 *                error identifier provided by this library
 */
int libgamma_quartz_cg_site_restore(libgamma_site_state_t* restrict this)
{
  (void) this;
  CGDisplayRestoreColorSyncSettings();
  return 0;
}



/**
 * Initialise an allocated partition state
 * 
 * @param   this       The partition state to initialise
 * @param   site       The site state for the site that the partition belongs to
 * @param   partition  The the index of the partition within the site
 * @return             Zero on success, otherwise (negative) the value of an
 *                     error identifier provided by this library
 */
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;
}


/**
 * Release all resources held by a partition state
 * 
 * @param  this  The partition state
 */
void libgamma_quartz_cg_partition_destroy(libgamma_partition_state_t* restrict this)
{
  free(this->data);
}


/**
 * Restore the gamma ramps all CRTCS with a partition to the system settings
 * 
 * @param   this  The partition state
 * @return        Zero on success, otherwise (negative) the value of an
 *                error identifier provided by this library
 */
int libgamma_quartz_cg_partition_restore(libgamma_partition_state_t* restrict this)
{
  return libgamma_quartz_cg_site_restore(this->site);
}



/**
 * Initialise an allocated CRTC state
 * 
 * @param   this       The CRTC state to initialise
 * @param   partition  The partition state for the partition that the CRTC belongs to
 * @param   crtc       The the index of the CRTC within the site
 * @return             Zero on success, otherwise (negative) the value of an
 *                     error identifier provided by this library
 */
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;
}


/**
 * Release all resources held by a CRTC state
 * 
 * @param  this  The CRTC state
 */
void libgamma_quartz_cg_crtc_destroy(libgamma_crtc_state_t* restrict this)
{
  (void) this;
}


/**
 * Restore the gamma ramps for a CRTC to the system settings for that CRTC
 * 
 * @param   this  The CRTC state
 * @return        Zero on success, otherwise (negative) the value of an
 *                error identifier provided by this library
 */
int libgamma_quartz_cg_crtc_restore(libgamma_crtc_state_t* restrict this)
{
  (void) this;
  return errno = ENOTSUP, LIBGAMMA_ERRNO_SET;
}



/**
 * Read information about a CRTC
 * 
 * @param   this    Instance of a data structure to fill with the information about the CRTC
 * @param   crtc    The state of the CRTC whose information should be read
 * @param   fields  OR:ed identifiers for the information about the CRTC that should be read
 * @return          Zero on success, -1 on error. On error refer to the error reports in `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
}


/**
 * Get current the gamma ramps for a CRTC, `float` version
 * 
 * @param   this   The CRTC state
 * @param   ramps  The gamma ramps to fill with the current values
 * @return         Zero on success, otherwise (negative) the value of an
 *                 error identifier provided by this library
 */
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;
}


/**
 * Set the gamma ramps for a CRTC, `float` version
 * 
 * @param   this   The CRTC state
 * @param   ramps  The gamma ramps to apply
 * @return         Zero on success, otherwise (negative) the value of an
 *                 error identifier provided by this library
 */
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;
}