/**
 * crt-calibrator – Calibration utility for CRT monitors
 * Copyright © 2014  Mattias Andrée (maandree@member.fsf.org)
 * 
 * This program 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 program 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 program.  If not, see <http://www.gnu.org/licenses/>.
 */
#include "state.h"

#include <stdlib.h>


/**
 * The framebuffers on the system
 */
framebuffer_t* restrict framebuffers = NULL;

/**
 * The number of elements in `framebuffers`
 */
size_t framebuffer_count = 0;

/**
 * The graphics cards on the system
 */
drm_card_t* restrict cards = NULL;

/**
 * The number of elements in `cards`
 */
size_t card_count = 0;

/**
 * The connected CRT controllers on the system
 */
drm_crtc_t* restrict crtcs = NULL;

/**
 * The software brightness setting on each connected CRT controller, on each channel
 */
double* restrict brightnesses[3];

/**
 * The software contrast setting on each connected CRT controller, on each channel
 */
double* restrict contrasts[3];

/**
 * The gamma correction on each connected CRT controller, on each channel
 */
double* restrict gammas[3];

/**
 * The number of elements in `crtcs`, `brightnesses[]`, `contrasts[]` and `gammas[]`
 */
size_t crtc_count = 0;



/**
 * Acquire video control
 * 
 * @return  Zero on success, -1 on error
 */
int acquire_video(void)
{
  size_t f, c, i, fn = fb_count(), cn = drm_card_count();
  drm_crtc_t* restrict old_crtcs;
  
  framebuffers = malloc(fn * sizeof(framebuffer_t));
  if (framebuffers == NULL)
    return -1;
  
  for (f = 0; f < fn; f++)
    {
      framebuffer_t fb;
      if (fb_open(f, &fb) < 0)
	return -1;
      framebuffers[framebuffer_count++] = fb;
    }
  
  cards = malloc(cn * sizeof(drm_card_t));
  if (cards == NULL)
    return -1;
  
  for (c = 0; c < cn; c++)
    {
      drm_card_t card;
      if (drm_card_open(c, &card) < 0)
	return -1;
      cards[card_count++] = card;
      
      old_crtcs = crtcs;
      crtcs = realloc(crtcs, (crtc_count + card.crtc_count) * sizeof(drm_crtc_t));
      if (crtcs == NULL)
	{
	  crtcs = old_crtcs;
	  return -1;
	}
      
      for (i = 0; i < card.crtc_count; i++)
	{
	  drm_crtc_t crtc;
	  if (drm_crtc_open(i, cards + c, &crtc) < 0)
	    return -1;
	  if (crtc.connected)
	    crtcs[crtc_count++] = crtc;
	  else
	    drm_crtc_close(&crtc);
	}
    }
  
  for (c = 0; c < 3; c++)
    {
      brightnesses[c] = malloc(crtc_count * sizeof(double));
      if (brightnesses[c] == NULL)
	return -1;
      
      contrasts[c] = malloc(crtc_count * sizeof(double));
      if (contrasts[c] == NULL)
	return -1;
      
      gammas[c] = malloc(crtc_count * sizeof(double));
      if (gammas[c] == NULL)
	return -1;
    }
  
  return 0;
}


/**
 * Release video control
 */
void release_video(void)
{
  size_t i;
  
  for (i = 0; i < crtc_count; i++)
    drm_crtc_close(crtcs + i);
  crtc_count = 0;
  
  for (i = 0; i < card_count; i++)
    drm_card_close(cards + i);
  card_count = 0;
  
  for (i = 0; i < framebuffer_count; i++)
    fb_close(framebuffers + i);
  framebuffer_count = 0;
  
  for (i = 0; i < 3; i++)
    {
      free(brightnesses[i]), brightnesses[i] = NULL;
      free(contrasts[i]), contrasts[i] = NULL;
      free(gammas[i]), gammas[i] = NULL;
    }
  
  free(crtcs), crtcs = NULL;
  free(cards), cards = NULL;
  free(framebuffers), framebuffers = NULL;
}