/* GDK - The GIMP Drawing Kit
 * Copyright (C) 1995-1997 Peter Mattis, Spencer Kimball and Josh MacDonald
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

/*
 * Modified by the GTK+ Team and others 1997-1999.  See the AUTHORS
 * file for a list of people on the GTK+ Team.  See the ChangeLog
 * files for a list of changes.  These files are distributed with
 * GTK+ at ftp://ftp.gtk.org/pub/gtk/. 
 */

#include <time.h>
#include <sys/ioctl.h>
#include <string.h>
#include <stdlib.h>

#include "gdkcolor.h"
#include "gdkprivate-fb.h"

static gint  gdk_colormap_match_color (GdkColormap *cmap,
				       GdkColor    *color,
				       const gchar *available);
GdkColormap*
gdk_colormap_new (GdkVisual *visual,
		  gint       private_cmap)
{
  GdkColormap *colormap;
  GdkColormapPrivateFB *private;
  GdkFBDisplay *fbd;
  int i;

  g_return_val_if_fail (visual != NULL, NULL);

  private = g_new (GdkColormapPrivateFB, 1);
  colormap = (GdkColormap*) private;

  private->base.visual = visual;
  private->base.ref_count = 1;
  fbd = gdk_display;

  private->hash = NULL;
  
  colormap->size = visual->colormap_size;
  colormap->colors = NULL;

  switch (visual->type)
    {
    case GDK_VISUAL_STATIC_GRAY:
    case GDK_VISUAL_STATIC_COLOR:
    case GDK_VISUAL_GRAYSCALE:
    case GDK_VISUAL_PSEUDO_COLOR:
      private->info = g_new0 (GdkColorInfo, colormap->size);
      colormap->colors = g_new (GdkColor, colormap->size);
      
      private->hash = g_hash_table_new ((GHashFunc) gdk_color_hash,
					(GCompareFunc) gdk_color_equal);
      
      if (private_cmap)
	{
	  guint16 red[256], green[256], blue[256];
	  struct fb_cmap fbc = {0, 256};

	  fbc.red = red;
	  fbc.green = green;
	  fbc.blue = blue;

	  if(ioctl(fbd->fd, FBIOGETCMAP, &fbc))
	    g_error("ioctl(FBIOGETCMAP) failed");

	  for (i = 0; i < colormap->size; i++)
	    {
	      colormap->colors[i].pixel = i;
	      colormap->colors[i].red = red[i];
	      colormap->colors[i].green = green[i];
	      colormap->colors[i].blue = blue[i];
	    }

	  gdk_colormap_change (colormap, colormap->size);
	}
      break;

    case GDK_VISUAL_DIRECT_COLOR:
      g_error("NYI");
#if 0
      colormap->colors = g_new (GdkColor, colormap->size);

      size = 1 << visual->red_prec;
      for (i = 0; i < size; i++)
	colormap->colors[i].red = i * 65535 / (size - 1);

      size = 1 << visual->green_prec;
      for (i = 0; i < size; i++)
	colormap->colors[i].green = i * 65535 / (size - 1);

      size = 1 << visual->blue_prec;
      for (i = 0; i < size; i++)
	colormap->colors[i].blue = i * 65535 / (size - 1);

      gdk_colormap_change (colormap, colormap->size);
#endif
      break;

    default:
      g_assert_not_reached();

    case GDK_VISUAL_TRUE_COLOR:
      break;
    }

  return colormap;
}

void
_gdk_colormap_real_destroy (GdkColormap *colormap)
{
  GdkColormapPrivateFB *private = (GdkColormapPrivateFB*) colormap;

  if (private->hash)
    g_hash_table_destroy (private->hash);
  
  g_free (private->info);
  g_free (colormap->colors);
  g_free (colormap);
}

#define MIN_SYNC_TIME 2

void
gdk_colormap_sync (GdkColormap *colormap,
		   gboolean     force)
{
  
}		   

GdkColormap*
gdk_colormap_get_system (void)
{
  static GdkColormap *colormap = NULL;

  if (!colormap)
    {
      guint16 red[256], green[256], blue[256];
      struct fb_cmap fbc = {0, 256};
      int i, r, g, b;
      GdkVisual *visual = gdk_visual_get_system();

      if(visual->type == GDK_VISUAL_GRAYSCALE
	 || visual->type ==  GDK_VISUAL_PSEUDO_COLOR)
	{
	  fbc.red = red;
	  fbc.green = green;
	  fbc.blue = blue;
	  switch(visual->type)
	    {
	    case GDK_VISUAL_GRAYSCALE:
	      for(i = 0; i < 256; i++)
		red[i] = green[i] = blue[i] = i << 8;
	      i--;
	      red[i] = green[i] = blue[i] = 65535; /* Make it a true white */
	      break;
	    case GDK_VISUAL_PSEUDO_COLOR:
	      /* Color cube stolen from gdkrgb upon advice from Owen */
	      for(i = r = 0; r < 6; r++)
		for(g = 0; g < 6; g++)
		  for(b = 0; b < 6; b++)
		    {
		      red[i] = r * 65535 / 5;
		      green[i] = g * 65535 / 5;
		      blue[i] = b * 65535 / 5;
		      i++;
		    }
	      g_assert(i == 216);

	  /* Fill in remaining space with grays */
	      for(i = 216; i < 256; i++)
		{
		  red[i] = green[i] = blue[i] =
		    (i - 216) * 40;
		}
	      /* Real white */
	      red[255] = green[255] = blue[255] = 65535;
	      break;
	    default:
	      break;
	    }

	  ioctl(gdk_display->fd, FBIOPUTCMAP, &fbc);
	}

      colormap = gdk_colormap_new(visual, TRUE);
    }

  return colormap;
}

gint
gdk_colormap_get_system_size (void)
{
  return 1 << (gdk_display->modeinfo.bits_per_pixel);
}

void
gdk_colormap_change (GdkColormap *colormap,
		     gint         ncolors)
{
  guint16 red[256], green[256], blue[256];
  struct fb_cmap fbc = {0,256};
  GdkColormapPrivateFB *private;
  int i;

  g_return_if_fail (colormap != NULL);

  fbc.red = red;
  fbc.green = green;
  fbc.blue = blue;

  private = (GdkColormapPrivateFB*) colormap;
  switch (private->base.visual->type)
    {
    case GDK_VISUAL_GRAYSCALE:
      for(i = 0; i < ncolors; i++)
	{
	  red[i] = green[i] = blue[i] =
	    (colormap->colors[i].red +
	     colormap->colors[i].green +
	     colormap->colors[i].blue)/3;
	}
      ioctl(gdk_display->fd, FBIOPUTCMAP, &fbc);
      break;

    case GDK_VISUAL_PSEUDO_COLOR:
      for (i = 0; i < ncolors; i++)
	{
	  red[i] = colormap->colors[i].red;
	  green[i] = colormap->colors[i].green;
	  blue[i] = colormap->colors[i].blue;
	}
      ioctl(gdk_display->fd, FBIOPUTCMAP, &fbc);
      break;

    default:
      break;
    }
}

gboolean
gdk_color_parse (const gchar *spec,
		 GdkColor *color)
{
  char aline[512];
  FILE *fh;

  g_return_val_if_fail(spec, FALSE);
  g_return_val_if_fail(color, FALSE);

  if(spec[0] == '#')
    {
      if(strlen(spec) == 7)
	{
	  guint num;

	  sscanf(spec + 1, "%x", &num);
	  color->red = (num & 0xFF0000) >> 8;
	  color->green = (num & 0xFF00);
	  color->blue = (num & 0xFF) << 8;
	}
      else if(strlen(spec) == 13)
	{
	  char s1[5], s2[5], s3[5];
	  g_snprintf(s1, sizeof(s1), spec + 1);
	  g_snprintf(s2, sizeof(s2), spec + 5);
	  g_snprintf(s3, sizeof(s3), spec + 9);

	  if(!sscanf(s1, "%hx", &color->red))
	    g_error("sscanf failed");
	  if(!sscanf(s2, "%hx", &color->green))
	    g_error("sscanf failed");
	  if(!sscanf(s3, "%hx", &color->blue))
	    g_error("sscanf failed");
	}
      else
	{
	  g_warning("Couldn't parse color specifier `%s'", spec);
	  return FALSE;
	}

      return TRUE;
    }
  else
    {
      fh = fopen("/usr/lib/X11/rgb.txt", "r");
      if(!fh)
	return FALSE;

      while(fgets(aline, sizeof(aline), fh))
	{
	  int red, green, blue;
	  char *ctmp;

	  g_strstrip(aline);
	  if(!aline[0] || aline[0] == '#' || aline[0] == '!')
	    continue;

	  ctmp = strtok(aline, " \t");
	  if(!ctmp)
	    continue;
	  red = atoi(ctmp);

	  ctmp = strtok(NULL, " \t");
	  if(!ctmp)
	    continue;
	  green = atoi(ctmp);

	  ctmp = strtok(NULL, " \t");
	  if(!ctmp)
	    continue;
	  blue = atoi(ctmp);

	  ctmp = strtok(NULL, " \t");
	  if(!ctmp || strcmp(ctmp, spec))
	    continue;

	  color->red = red << 8;
	  color->green = green << 8;
	  color->blue = blue << 8;
	  return TRUE;
	}
      fclose(fh);
    }

  return FALSE;
}

void
gdk_colormap_free_colors (GdkColormap *colormap,
			  GdkColor    *colors,
			  gint         ncolors)
{
  GdkColormapPrivateFB *private;
  gint i;

  g_return_if_fail (colormap != NULL);
  g_return_if_fail (colors != NULL);

  private = (GdkColormapPrivateFB*) colormap;

  if ((private->base.visual->type != GDK_VISUAL_PSEUDO_COLOR) &&
      (private->base.visual->type != GDK_VISUAL_GRAYSCALE))
    return;

  for (i=0; i<ncolors; i++)
    {
      gulong pixel = colors[i].pixel;
      
      if (private->info[pixel].ref_count)
	{
	  private->info[pixel].ref_count--;

	  if (private->info[pixel].ref_count == 0)
	    {
	      if (!(private->info[pixel].flags & GDK_COLOR_WRITEABLE))
		g_hash_table_remove (private->hash, &colormap->colors[pixel]);
	      private->info[pixel].flags = 0;
	    }
	}
    }
}

/********************
 * Color allocation *
 ********************/

/* Try to allocate a single color using XAllocColor. If it succeeds,
 * cache the result in our colormap, and store in ret.
 */
static gboolean 
gdk_colormap_alloc1 (GdkColormap *colormap,
		     GdkColor    *color,
		     GdkColor    *ret)
{
  GdkColormapPrivateFB *private;
  int i;

  private = (GdkColormapPrivateFB*) colormap;

  if(private->base.visual->type != GDK_VISUAL_GRAYSCALE
     && private->base.visual->type != GDK_VISUAL_PSEUDO_COLOR)
    return FALSE;

  *ret = *color;
  if(!color->red && !color->green && !color->blue) /* black */
    {
      ret->pixel = 0;
      private->info[ret->pixel].ref_count++;
      return TRUE;
    }

  if(color->red == 65535 && color->green == 65535 && color->blue == 65535) /* white */
    {
      ret->pixel = 255;
      private->info[ret->pixel].ref_count++;
      return TRUE;
    }

  for(i = 1; i < (colormap->size - 1); i++)
    {
      if(!private->info[i].ref_count)
	{
	  guint16 red = color->red, green = color->green, blue = color->blue;
	  struct fb_cmap fbc;
	  fbc.len = 1;
	  fbc.start = i;
	  fbc.red = &red;
	  fbc.green = &green;
	  fbc.blue = &blue;

	  ioctl(gdk_display->fd, FBIOPUTCMAP, &fbc);

	  ret->pixel = i;
	  colormap->colors[ret->pixel] = *ret;
	  private->info[ret->pixel].ref_count = 1;
	  g_hash_table_insert (private->hash,
			       &colormap->colors[ret->pixel],
			       &colormap->colors[ret->pixel]);
	  return TRUE;
	}
    }

  return FALSE;
}

static gint
gdk_colormap_alloc_colors_shared (GdkColormap *colormap,
				  GdkColor    *colors,
				  gint         ncolors,
				  gboolean     writeable,
				  gboolean     best_match,
				  gboolean    *success)
{
  GdkColormapPrivateFB *private;
  gint i, index;
  gint nremaining = 0;
  gint nfailed = 0;

  private = (GdkColormapPrivateFB*) colormap;
  index = -1;

  for (i=0; i<ncolors; i++)
    {
      if (!success[i])
	{
	  if (gdk_colormap_alloc1 (colormap, &colors[i], &colors[i]))
	    success[i] = TRUE;
	  else
	    nremaining++;
	}
    }


  if (nremaining > 0 && best_match)
    {
      gchar *available = g_new (gchar, colormap->size);

      for (i = 0; i < colormap->size; i++)
	available[i] = ((private->info[i].ref_count == 0) ||
			!(private->info[i].flags && GDK_COLOR_WRITEABLE));
      
      while (nremaining > 0)
	{
	  for (i=0; i<ncolors; i++)
	    {
	      if (!success[i])
		{
		  index = gdk_colormap_match_color (colormap, &colors[i], available);
		  if (index != -1)
		    {
		      if (private->info[index].ref_count)
			{
			  private->info[index].ref_count++;
			  colors[i] = colormap->colors[index];
			  success[i] = TRUE;
			  nremaining--;
			}
		      else
			{
			  if (gdk_colormap_alloc1 (colormap, 
						   &colormap->colors[index],
						   &colors[i]))
			    {
			      success[i] = TRUE;
			      nremaining--;
			      break;
			    }
			  else
			    {
			      available[index] = FALSE;
			    }
			}
		    }
		  else
		    {
		      nfailed++;
		      nremaining--;
		      success[i] = 2; /* flag as permanent failure */
		    }
		}
	    }
	}
      g_free (available);
    }

  /* Change back the values we flagged as permanent failures */
  if (nfailed > 0)
    {
      for (i=0; i<ncolors; i++)
	if (success[i] == 2)
	  success[i] = FALSE;
      nremaining = nfailed;
    }
  
  return (ncolors - nremaining);
}

static gint
gdk_colormap_alloc_colors_pseudocolor (GdkColormap *colormap,
				       GdkColor    *colors,
				       gint         ncolors,
				       gboolean     writeable,
				       gboolean     best_match,
				       gboolean    *success)
{
  GdkColormapPrivateFB *private;
  GdkColor *lookup_color;
  gint i;
  gint nremaining = 0;

  private = (GdkColormapPrivateFB*) colormap;

  /* Check for an exact match among previously allocated colors */

  for (i=0; i<ncolors; i++)
    {
      if (!success[i])
	{
	  lookup_color = g_hash_table_lookup (private->hash, &colors[i]);
	  if (lookup_color)
	    {
	      private->info[lookup_color->pixel].ref_count++;
	      colors[i].pixel = lookup_color->pixel;
	      success[i] = TRUE;
	    }
	  else
	    nremaining++;
	}
    }

  /* If that failed, we try to allocate a new color, or approxmiate
   * with what we can get if best_match is TRUE.
   */
  if (nremaining > 0)
    return gdk_colormap_alloc_colors_shared (colormap, colors, ncolors, writeable, best_match, success);
  else
    return 0;
}

gint
gdk_colormap_alloc_colors (GdkColormap *colormap,
			   GdkColor    *colors,
			   gint         ncolors,
			   gboolean     writeable,
			   gboolean     best_match,
			   gboolean    *success)
{
  GdkColormapPrivateFB *private;
  GdkVisual *visual;
  gint i;
  gint nremaining = 0;

  g_return_val_if_fail (colormap != NULL, FALSE);
  g_return_val_if_fail (colors != NULL, FALSE);

  private = (GdkColormapPrivateFB*) colormap;

  for (i=0; i<ncolors; i++)
    success[i] = FALSE;

  switch (private->base.visual->type)
    {
    case GDK_VISUAL_PSEUDO_COLOR:
    case GDK_VISUAL_GRAYSCALE:
    case GDK_VISUAL_STATIC_GRAY:
    case GDK_VISUAL_STATIC_COLOR:
      return gdk_colormap_alloc_colors_pseudocolor (colormap, colors, ncolors,
						    writeable, best_match, success);
      break;

    case GDK_VISUAL_DIRECT_COLOR:
    case GDK_VISUAL_TRUE_COLOR:
      visual = private->base.visual;

      for (i=0; i<ncolors; i++)
	{
	  colors[i].pixel = (((colors[i].red >> (16 - visual->red_prec)) << visual->red_shift) +
			     ((colors[i].green >> (16 - visual->green_prec)) << visual->green_shift) +
			     ((colors[i].blue >> (16 - visual->blue_prec)) << visual->blue_shift));
	  success[i] = TRUE;
	}
      break;
    }
  return nremaining;
}

gboolean
gdk_color_change (GdkColormap *colormap,
		  GdkColor    *color)
{
  GdkColormapPrivateFB *private;
  struct fb_cmap fbc = {0, 1};

  g_return_val_if_fail (colormap != NULL, FALSE);
  g_return_val_if_fail (color != NULL, FALSE);

  private = (GdkColormapPrivateFB*) colormap;

  switch(private->base.visual->type)
    {
    case GDK_VISUAL_GRAYSCALE:
      color->red = color->green = color->blue = (color->red + color->green + color->blue)/3;

    case GDK_VISUAL_PSEUDO_COLOR:
      fbc.start = color->pixel;
      fbc.red = &color->red;
      fbc.green = &color->green;
      fbc.blue = &color->blue;
      ioctl(gdk_display->fd, FBIOPUTCMAP, &fbc);
      break;

    default:
      break;
    }

  return TRUE;
}

static gint
gdk_colormap_match_color (GdkColormap *cmap,
			  GdkColor    *color,
			  const gchar *available)
{
  GdkColor *colors;
  guint sum, max;
  gint rdiff, gdiff, bdiff;
  gint i, index;

  g_return_val_if_fail (cmap != NULL, 0);
  g_return_val_if_fail (color != NULL, 0);

  colors = cmap->colors;
  max = 3 * (65536);
  index = -1;

  for (i = 0; i < cmap->size; i++)
    {
      if ((!available) || (available && available[i]))
	{
	  rdiff = (color->red - colors[i].red);
	  gdiff = (color->green - colors[i].green);
	  bdiff = (color->blue - colors[i].blue);

	  sum = ABS (rdiff) + ABS (gdiff) + ABS (bdiff);

	  if (sum < max)
	    {
	      index = i;
	      max = sum;
	    }
	}
    }

  return index;
}

gint gdk_colors_alloc	 (GdkColormap	*colormap,
			  gboolean	 contiguous,
			  gulong	*planes,
			  gint		 nplanes,
			  gulong	*pixels,
			  gint		 npixels)
{
  return 0;
}

void
gdk_colors_free	 (GdkColormap	*colormap,
		  gulong	*pixels,
		  gint		 npixels,
		  gulong	 planes)
{
}

gulong
gdk_color_context_get_pixel(GdkColorContext *cc,
			    gushort	    red,
			    gushort	    green,
			    gushort	    blue,
			    gint	   *failed)
{
  g_error("NYI");

  return 0;
}

GdkColorContext *
gdk_color_context_new(GdkVisual   *visual,
		      GdkColormap *colormap)
{
  g_error("NYI");

  return NULL;
}

GdkColorContext *
gdk_color_context_new_mono(GdkVisual   *visual,
			   GdkColormap *colormap)
{
  g_error("NYI");

  return NULL;
}

void
gdk_color_context_free(GdkColorContext *cc)
{
  g_error("NYI");
}

gint
gdk_color_context_query_color(GdkColorContext *cc,
			      GdkColor	   *color)
{
  g_error("NYI");

  return 0;
}