/**
 * 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/>.
 */
#include "gamma-helper.h"

#include "libgamma-method.h"
#include "libgamma-error.h"

#include <errno.h>
#include <stdlib.h>
#include <stdint.h>


/**
 * Just an arbitrary version
 */
#define ANY  bits64


#define __translate(instruction)  for (i = 0; i < n; i++)  instruction;  break


/**
 * Convert any set of gamma ramps into a 64-bit integer array with all channels.
 * 
 * @param  depth  The depth of the gamma ramp, `-1` for `float`, `-2` for `double`.
 * @param  n      The grand size of gamma ramps (sum of all channels' sizes.)
 * @param  out    Output array.
 * @param  in     Input gamma ramps.
 */
static void translate_to_64(signed depth, size_t n, uint64_t* restrict out, libgamma_gamma_ramps_any_t in)
{
  size_t i;
  switch (depth)
    {
      /* Translate integer. */
    case 16:  __translate(out[i] = (uint64_t)(in.bits16.red[i]) * 0x0001000100010001ULL);
    case 32:  __translate(out[i] = (uint64_t)(in.bits32.red[i]) * 0x0000000100000001ULL);
      /* Identity translation. */
    case 64:  __translate(out[i] = in.bits64.red[i]);
      /* Translate floating point. */
    case -1:  __translate(out[i] = (uint64_t)(in.float_single.red[i] * (float)UINT64_MAX));
    case -2:  __translate(out[i] = (uint64_t)(in.float_double.red[i] * (double)UINT64_MAX));
    default:
      /* This is not possible. */
      abort();
      break;
    }
}


/**
 * Undo the actions of `translate_to_64`.
 * 
 * @param  depth  The depth of the gamma ramp, `-1` for `float`, `-2` for `double`.
 * @param  n      The grand size of gamma ramps (sum of all channels' sizes.)
 * @param  out    Output gamma ramps.
 * @param  in     Input array, may be modified.
 */
static void translate_from_64(signed depth, size_t n, libgamma_gamma_ramps_any_t out, uint64_t* restrict in)
{
  size_t i;
  switch (depth)
    {
      /* Translate integer. */
    case 16:  __translate(out.bits16.red[i] = (uint16_t)(in[i] / 0x0001000100010001ULL));
    case 32:  __translate(out.bits32.red[i] = (uint32_t)(in[i] / 0x0000000100000001ULL));
      /* Identity translation. */
    case 64:  __translate(out.bits64.red[i] = in[i]);
      /* Translate floating point. */
    case -1:  __translate(out.float_single.red[i] = (float)(in[i]) / (float)UINT64_MAX);
    case -2:  __translate(out.float_double.red[i] = (double)(in[i]) / (double)UINT64_MAX);
    default:
      /* This is not possible. */
      abort();
      break;
    }
}


/**
 * Allocate and initalise a gamma ramp with any depth.
 * 
 * @param   ramps_sys  Output gamma ramps.
 * @param   ramps      The gamma ramps whose sizes should be duplicated.
 * @param   depth      The depth of the gamma ramps to allocate,
 *                     `-1` for `float`, `-2` for `double`.
 * @param   elements   Output reference for the grand size of the gamma ramps.
 * @return             Zero on success, otherwise (negative) the value of an
 *                     error identifier provided by this library.
 */
static int allocated_any_ramp(libgamma_gamma_ramps_any_t* restrict ramps_sys,
			      libgamma_gamma_ramps_any_t ramps, signed depth, size_t* restrict elements)
{
  size_t d, n = ramps.ANY.red_size + ramps.ANY.green_size + ramps.ANY.blue_size;
  
  switch (depth)
    {
    case 16:  d = sizeof(uint16_t);  break;
    case 32:  d = sizeof(uint32_t);  break;
    case 64:  d = sizeof(uint64_t);  break;
    case -1:  d = sizeof(float);     break;
    case -2:  d = sizeof(double);    break;
    default:
      return errno = EINVAL, LIBGAMMA_ERRNO_SET;
    }
  
  ramps_sys->ANY = ramps.ANY;
  ramps_sys->ANY.red   = malloc(n * d);
  ramps_sys->ANY.green = (void*)(((char*)(ramps_sys->ANY.  red)) + ramps.ANY.  red_size * d / sizeof(char));
  ramps_sys->ANY.blue  = (void*)(((char*)(ramps_sys->ANY.green)) + ramps.ANY.green_size * d / sizeof(char));
  
  *elements = n;
  return ramps_sys->ANY.red == NULL ? LIBGAMMA_ERRNO_SET : 0;
}


/**
 * Get current the gamma ramps for a CRTC, re-encoding version.
 * 
 * @param   this          The CRTC state.
 * @param   ramps         The gamma ramps to fill with the current values.
 * @param   depth_user    The depth of the gamma ramps that are provided by the user,
 *                        `-1` for `float`, `-2` for `double`.
 * @param   depth_system  The depth of the gamma ramps as required by the adjustment method,
 *                        `-1` for `float`, `-2` for `double`.
 * @param   fun           Function that is to be used read the ramps, its parameters have
 *                        the same function as those of this function with the same names,
 *                        and the return value too is identical.
 * @return                Zero on success, otherwise (negative) the value of an
 *                        error identifier provided by this library.
 */
int libgamma_translated_ramp_get_(libgamma_crtc_state_t* restrict this,
				  libgamma_gamma_ramps_any_t* restrict ramps,
				  signed depth_user, signed depth_system,
				  libgamma_get_ramps_any_fun* fun)
{
  size_t n;
  int r;
  libgamma_gamma_ramps_any_t ramps_sys;
  uint64_t* restrict ramps_full;
  
  if ((r = allocated_any_ramp(&ramps_sys, *ramps, depth_system, &n)))
    return r;
  
  if ((r = fun(this, &ramps_sys)))
    return free(ramps_sys.ANY.red), r;
  
  if ((ramps_full = malloc(n * sizeof(uint64_t))) == NULL)
    return free(ramps_sys.ANY.red), LIBGAMMA_ERRNO_SET;
  
  translate_to_64(depth_system, n, ramps_full, ramps_sys);
  free(ramps_sys.ANY.red);
  
  translate_from_64(depth_user, n, *ramps, ramps_full);
  free(ramps_full);
  return 0;
}


/**
 * Set the gamma ramps for a CRTC, re-encoding version.
 * 
 * @param   this          The CRTC state.
 * @param   ramps         The gamma ramps to apply.
 * @param   depth_user    The depth of the gamma ramps that are provided by the user,
 *                        `-1` for `float`, `-2` for `double`.
 * @param   depth_system  The depth of the gamma ramps as required by the adjustment method,
 *                        `-1` for `float`, `-2` for `double`.
 * @param   fun           Function that is to be used write the ramps, its parameters have
 *                        the same function as those of this function with the same names,
 *                        and the return value too is identical.
 * @return                Zero on success, otherwise (negative) the value of an
 *                        error identifier provided by this library.
 */
int libgamma_translated_ramp_set_(libgamma_crtc_state_t* restrict this,
				  libgamma_gamma_ramps_any_t ramps,
				  signed depth_user, signed depth_system,
				  libgamma_set_ramps_any_fun* fun)
{
  size_t n;
  int r;
  libgamma_gamma_ramps_any_t ramps_sys;
  uint64_t* restrict ramps_full;
  
  if ((r = allocated_any_ramp(&ramps_sys, ramps, depth_system, &n)))
    return r;
  
  if ((ramps_full = malloc(n * sizeof(uint64_t))) == NULL)
    return free(ramps_sys.ANY.red), LIBGAMMA_ERRNO_SET;
  
  translate_to_64(depth_user, n, ramps_full, ramps);
  translate_from_64(depth_system, n, ramps_sys, ramps_full);
  free(ramps_full);
  
  r = fun(this, ramps_sys);
  
  free(ramps_sys.ANY.red);
  return r;
}


#undef __translate
#undef ANY