Trim trailing white spaces.
[geeqie.git] / src / histogram.c
index fe393c1..8e379fb 100644 (file)
@@ -1,5 +1,6 @@
 /*
  * Geeqie
+ * Copyright (C) 2008 - 2012 The Geeqie Team
  *
  * Author: Vladimir Nadvornik
  * based on a patch by Uwe Ohse
@@ -13,6 +14,7 @@
 #include "histogram.h"
 
 #include "pixbuf_util.h"
+#include "filedata.h"
 
 #include <math.h>
 
  *----------------------------------------------------------------------------
  */
 
-#define HISTOGRAM_SIZE 256
-struct _Histogram {
-       gulong histmap[HISTOGRAM_SIZE*4];
-       gint histogram_chan;
-       gint histogram_logmode;
+#define HISTMAP_SIZE 256
+
+struct _HistMap {
+       gulong r[HISTMAP_SIZE];
+       gulong g[HISTMAP_SIZE];
+       gulong b[HISTMAP_SIZE];
+       gulong max[HISTMAP_SIZE];
+       
+       guint idle_id; /* event source id */
+       GdkPixbuf *pixbuf;
+       gint y;
 };
 
 
-Histogram *histogram_new()
+Histogram *histogram_new(void)
 {
        Histogram *histogram;
 
-        histogram = g_new0(Histogram, 1);
-       histogram->histogram_chan = HCHAN_RGB;
-       histogram->histogram_logmode = 1;
+       histogram = g_new0(Histogram, 1);
+       histogram->histogram_channel = HCHAN_DEFAULT;
+       histogram->histogram_mode = 0;
+
+       /* grid */
+       histogram->vgrid = 5;
+       histogram->hgrid = 3;
+       histogram->grid_color.R = 160;
+       histogram->grid_color.G = 160;
+       histogram->grid_color.B = 160;
+       histogram->grid_color.A = 250;
 
        return histogram;
 }
@@ -51,188 +66,326 @@ void histogram_free(Histogram *histogram)
 gint histogram_set_channel(Histogram *histogram, gint chan)
 {
        if (!histogram) return 0;
-       histogram->histogram_chan = chan;
+       histogram->histogram_channel = chan;
        return chan;
 }
 
 gint histogram_get_channel(Histogram *histogram)
 {
        if (!histogram) return 0;
-       return histogram->histogram_chan;
+       return histogram->histogram_channel;
 }
 
 gint histogram_set_mode(Histogram *histogram, gint mode)
 {
        if (!histogram) return 0;
-       histogram->histogram_logmode = mode;
+       histogram->histogram_mode = mode;
        return mode;
 }
 
 gint histogram_get_mode(Histogram *histogram)
 {
        if (!histogram) return 0;
-       return histogram->histogram_logmode;
+       return histogram->histogram_mode;
+}
+
+gint histogram_toggle_channel(Histogram *histogram)
+{
+       if (!histogram) return 0;
+       return histogram_set_channel(histogram, (histogram_get_channel(histogram)+1)%HCHAN_COUNT);
+}
+
+gint histogram_toggle_mode(Histogram *histogram)
+{
+       if (!histogram) return 0;
+       return histogram_set_mode(histogram, !histogram_get_mode(histogram));
 }
 
 const gchar *histogram_label(Histogram *histogram)
 {
        const gchar *t1 = "";
+       
        if (!histogram) return NULL;
 
-       if (histogram->histogram_logmode)
-                       switch (histogram->histogram_chan) {
-                       case HCHAN_R:   t1 = _("logarithmical histogram on red"); break;
-                        case HCHAN_G:   t1 = _("logarithmical histogram on green"); break;
-                        case HCHAN_B:   t1 = _("logarithmical histogram on blue"); break;
-                        case HCHAN_VAL: t1 = _("logarithmical histogram on value"); break;
-                        case HCHAN_RGB: t1 = _("logarithmical histogram on RGB"); break;
-                        case HCHAN_MAX: t1 = _("logarithmical histogram on max value"); break;
-                        }
-                else
-                        switch (histogram->histogram_chan) {
-                        case HCHAN_R:   t1 = _("linear histogram on red"); break;
-                        case HCHAN_G:   t1 = _("linear histogram on green"); break;
-                        case HCHAN_B:   t1 = _("linear histogram on blue"); break;
-                        case HCHAN_VAL: t1 = _("linear histogram on value"); break;
-                        case HCHAN_RGB: t1 = _("linear histogram on RGB"); break;
-                        case HCHAN_MAX: t1 = _("linear histogram on max value"); break;
-                        }
+       if (histogram->histogram_mode)
+               switch (histogram->histogram_channel)
+                       {
+                       case HCHAN_R:   t1 = _("Log Histogram on Red"); break;
+                       case HCHAN_G:   t1 = _("Log Histogram on Green"); break;
+                       case HCHAN_B:   t1 = _("Log Histogram on Blue"); break;
+                       case HCHAN_RGB: t1 = _("Log Histogram on RGB"); break;
+                       case HCHAN_MAX: t1 = _("Log Histogram on value"); break;
+                       }
+       else
+               switch (histogram->histogram_channel)
+                       {
+                       case HCHAN_R:   t1 = _("Linear Histogram on Red"); break;
+                       case HCHAN_G:   t1 = _("Linear Histogram on Green"); break;
+                       case HCHAN_B:   t1 = _("Linear Histogram on Blue"); break;
+                       case HCHAN_RGB: t1 = _("Linear Histogram on RGB"); break;
+                       case HCHAN_MAX: t1 = _("Linear Histogram on value"); break;
+                       }
        return t1;
 }
 
-gulong histogram_read(Histogram *histogram, GdkPixbuf *imgpixbuf)
+static HistMap *histmap_new(void)
 {
-       gint w, h, i, j, srs, has_alpha;
-       guchar *s_pix;
+       HistMap *histmap = g_new0(HistMap, 1);
+       return histmap;
+}
 
-       if (!histogram) return 0;
+void histmap_free(HistMap *histmap)
+{
+       if (!histmap) return;
+       if (histmap->idle_id) g_source_remove(histmap->idle_id);
+       if (histmap->pixbuf) g_object_unref(histmap->pixbuf);
+       g_free(histmap);
+}
 
+static gboolean histmap_read(HistMap *histmap, gboolean whole)
+{
+       gint w, h, i, j, srs, has_alpha, step, end_line;
+       guchar *s_pix;
+       GdkPixbuf *imgpixbuf = histmap->pixbuf;
+       
        w = gdk_pixbuf_get_width(imgpixbuf);
        h = gdk_pixbuf_get_height(imgpixbuf);
        srs = gdk_pixbuf_get_rowstride(imgpixbuf);
        s_pix = gdk_pixbuf_get_pixels(imgpixbuf);
        has_alpha = gdk_pixbuf_get_has_alpha(imgpixbuf);
+       
+       if (whole)
+               {
+               end_line = h;
+               }
+       else
+               {
+               gint lines = 1 + 16384 / w;
+               end_line = histmap->y + lines;
+               if (end_line > h) end_line = h;
+               }
 
-       for (i = 0; i < HISTOGRAM_SIZE*4; i++) histogram->histmap[i] = 0;
-
-       for (i = 0; i < h; i++)
+       step = 3 + !!(has_alpha);
+       for (i = histmap->y; i < end_line; i++)
                {
                guchar *sp = s_pix + (i * srs); /* 8bit */
                for (j = 0; j < w; j++)
                        {
-                       histogram->histmap[sp[0] + 0 * HISTOGRAM_SIZE]++;
-                       histogram->histmap[sp[1] + 1 * HISTOGRAM_SIZE]++;
-                       histogram->histmap[sp[2] + 2 * HISTOGRAM_SIZE]++;
-                       if (histogram->histogram_chan == HCHAN_MAX)
-                               {
-                               guchar t = sp[0];
-                               if (sp[1]>t) t = sp[1];
-                               if (sp[2]>t) t = sp[2];
-                               histogram->histmap[t + 3 * HISTOGRAM_SIZE]++;
-                               }
-                       else
-                               histogram->histmap[3 * HISTOGRAM_SIZE + (sp[0]+sp[1]+sp[2])/3]++;
-                       sp += 3;
-                       if (has_alpha) sp++;
+                       guint max = sp[0];
+                       if (sp[1] > max) max = sp[1];
+                       if (sp[2] > max) max = sp[2];
+               
+                       histmap->r[sp[0]]++;
+                       histmap->g[sp[1]]++;
+                       histmap->b[sp[2]]++;
+                       histmap->max[max]++;
+
+                       sp += step;
                        }
                }
+       histmap->y = end_line;
+       return end_line >= h;
+}
+
+const HistMap *histmap_get(FileData *fd)
+{
+       if (fd->histmap && !fd->histmap->idle_id) return fd->histmap; /* histmap exists and is finished */
+       
+       return NULL;
+}
+
+static gboolean histmap_idle_cb(gpointer data)
+{
+       FileData *fd = data;
+       if (histmap_read(fd->histmap, FALSE))
+               {
+               /* finished */
+               g_object_unref(fd->histmap->pixbuf); /*pixbuf is no longer needed */
+               fd->histmap->pixbuf = NULL;
+               fd->histmap->idle_id = 0;
+               file_data_send_notification(fd, NOTIFY_HISTMAP);
+               return FALSE;
+               }
+       return TRUE;
+}
+
+gboolean histmap_start_idle(FileData *fd)
+{
+       if (fd->histmap || !fd->pixbuf) return FALSE;
+
+       fd->histmap = histmap_new();
+       fd->histmap->pixbuf = fd->pixbuf;
+       g_object_ref(fd->histmap->pixbuf);
+
+       fd->histmap->idle_id = g_idle_add_full(G_PRIORITY_DEFAULT_IDLE, histmap_idle_cb, fd, NULL);
+       return TRUE;
+}
+
+
+static void histogram_vgrid(Histogram *histogram, GdkPixbuf *pixbuf, gint x, gint y, gint width, gint height)
+{
+       guint i;
+       float add;
+       
+       if (histogram->vgrid == 0) return;
+
+       add = width / (float)histogram->vgrid;
+
+       for (i = 1; i < histogram->vgrid; i++)
+               {
+               gint xpos = x + (int)(i * add + 0.5);
 
-       return w*h;
+               pixbuf_draw_line(pixbuf, x, y, width, height, xpos, y, xpos, y + height,
+                                histogram->grid_color.R,
+                                histogram->grid_color.G,
+                                histogram->grid_color.B,
+                                histogram->grid_color.A);
+               }
 }
 
+static void histogram_hgrid(Histogram *histogram, GdkPixbuf *pixbuf, gint x, gint y, gint width, gint height)
+{
+       guint i;
+       float add;
+       
+       if (histogram->hgrid == 0) return;
+
+       add = height / (float)histogram->hgrid;
+
+       for (i = 1; i < histogram->hgrid; i++)
+               {
+               gint ypos = y + (int)(i * add + 0.5);
+       
+               pixbuf_draw_line(pixbuf, x, y, width, height, x, ypos, x + width, ypos,
+                                histogram->grid_color.R,
+                                histogram->grid_color.G,
+                                histogram->grid_color.B,
+                                histogram->grid_color.A);
+               }
+}
 
-gint histogram_draw(Histogram *histogram, GdkPixbuf *pixbuf, gint x, gint y, gint width, gint height)
+gboolean histogram_draw(Histogram *histogram, const HistMap *histmap, GdkPixbuf *pixbuf, gint x, gint y, gint width, gint height)
 {
        /* FIXME: use the coordinates correctly */
        gint i;
        gulong max = 0;
-       double logmax;
-
-       if (!histogram) return 0;
-
-       for (i=0; i<1024; i++) {
-               gint flag = 0;
+       gdouble logmax;
+       gint combine = (HISTMAP_SIZE - 1) / width + 1;
+       gint ypos = y + height;
+       
+       if (!histogram || !histmap) return FALSE;
+       
+       /* Draw the grid */
+       histogram_vgrid(histogram, pixbuf, x, y, width, height);
+       histogram_hgrid(histogram, pixbuf, x, y, width, height);
 
-               switch (histogram->histogram_chan)
+       /* exclude overexposed and underexposed */
+       for (i = 1; i < HISTMAP_SIZE - 1; i++)
                {
-               case HCHAN_RGB: if ((i%4) != 3 ) flag = 1; break;
-               case HCHAN_R: if ((i%4) == 0) flag = 1; break;
-               case HCHAN_G: if ((i%4) == 1) flag = 1; break;
-               case HCHAN_B: if ((i%4) == 2) flag = 1; break;
-               case HCHAN_VAL: if ((i%4) == 3) flag = 1; break;
-               case HCHAN_MAX: if ((i%4) == 3) flag = 1; break;
+               if (histmap->r[i] > max) max = histmap->r[i];
+               if (histmap->g[i] > max) max = histmap->g[i];
+               if (histmap->b[i] > max) max = histmap->b[i];
+               if (histmap->max[i] > max) max = histmap->max[i];
                }
-               if (flag && histogram->histmap[i] > max) max = histogram->histmap[i];
-       }
 
-       logmax = log(max);
-       for (i=0; i<width; i++)
+       if (max > 0)
+               logmax = log(max);
+       else
+               logmax = 1.0;
+
+       for (i = 0; i < width; i++)
                {
                gint j;
                glong v[4] = {0, 0, 0, 0};
                gint rplus = 0;
                gint gplus = 0;
                gint bplus = 0;
-               gint ii = i * HISTOGRAM_SIZE / width;
-               gint combine  = (HISTOGRAM_SIZE - 1) / width + 1;
+               gint ii = i * HISTMAP_SIZE / width;
+               gint xpos = x + i;
+               gint num_chan;
 
                for (j = 0; j < combine; j++)
                        {
-                       v[0] += histogram->histmap[ii + j + 0*HISTOGRAM_SIZE]; // r
-                       v[1] += histogram->histmap[ii + j + 1*HISTOGRAM_SIZE]; // g
-                       v[2] += histogram->histmap[ii + j + 2*HISTOGRAM_SIZE]; // b
-                       v[3] += histogram->histmap[ii + j + 3*HISTOGRAM_SIZE]; // value, max
+                       guint p = ii + j;
+                       v[0] += histmap->r[p];
+                       v[1] += histmap->g[p];
+                       v[2] += histmap->b[p];
+                       v[3] += histmap->max[p];
                        }
-               
-               for (j=0; j<4; j++)
-                       {
-                       gint r = rplus;
-                       gint g = gplus;
-                       gint b = bplus;
-                       gint max2 = 0;
-                       gint k;
-                       gulong pt;
-
-                       for (k=1; k<4; k++)
-                               if (v[k] > v[max2]) max2 = k;
        
-                       switch (max2)
-                       {
-                       case HCHAN_R: rplus = r = 255; break;
-                       case HCHAN_G: gplus = g = 255; break;
-                       case HCHAN_B: bplus = b = 255; break;
-                       }
-
-                       switch(histogram->histogram_chan)
+               for (j = 0; combine > 1 && j < 4; j++)
+                       v[j] /= combine;
+               
+               num_chan = (histogram->histogram_channel == HCHAN_RGB) ? 3 : 1;
+               for (j = 0; j < num_chan; j++)
                        {
-                       case HCHAN_MAX: r = 0; b = 0; g = 0; break;
-                       case HCHAN_VAL: r = 0; b = 0; g = 0; break;
-                       case HCHAN_R: g = 0; b = 0; break;
-                       case HCHAN_G: r = 0; b = 0; break;
-                       case HCHAN_B: r = 0; g = 0; break;
-                       case HCHAN_RGB: 
-                               if (r == 255 && g == 255 && b == 255) {
-                                       r = 0; 
-                                       g = 0; 
-                                       b = 0; 
+                       gint chanmax;
+                       if (histogram->histogram_channel == HCHAN_RGB)
+                               {
+                               chanmax = HCHAN_R;
+                               if (v[HCHAN_G] > v[HCHAN_R]) chanmax = HCHAN_G;
+                               if (v[HCHAN_B] > v[chanmax]) chanmax = HCHAN_B;
                                }
-                               break;
-                       }
-
-                       if (v[max2] == 0)
-                               pt = 0;
-                       else if (histogram->histogram_logmode)
-                               pt = ((float)log(v[max2])) / logmax * (height - 1);
                        else
-                               pt = ((float)v[max2])/ max * (height - 1);
-                       if (histogram->histogram_chan >= HCHAN_RGB 
-                           || max2 == histogram->histogram_chan)
-                               pixbuf_draw_line(pixbuf, 
+                               {
+                               chanmax = histogram->histogram_channel;
+                               }
+                       
+                               {
+                               gulong pt;
+                               gint r = rplus;
+                               gint g = gplus;
+                               gint b = bplus;
+
+                               switch (chanmax)
+                                       {
+                                       case HCHAN_R: rplus = r = 255; break;
+                                       case HCHAN_G: gplus = g = 255; break;
+                                       case HCHAN_B: bplus = b = 255; break;
+                                       }
+
+                               switch (histogram->histogram_channel)
+                                       {
+                                       case HCHAN_RGB:
+                                               if (r == 255 && g == 255 && b == 255)
+                                                       {
+                                                       r = 0;  b = 0;  g = 0;
+                                                       }
+                                               break;
+                                       case HCHAN_R:           b = 0;  g = 0;  break;
+                                       case HCHAN_G:   r = 0;  b = 0;          break;
+                                       case HCHAN_B:   r = 0;          g = 0;  break;
+                                       case HCHAN_MAX: r = 0;  b = 0;  g = 0;  break;
+                                       }
+                               
+                               if (v[chanmax] == 0)
+                                       pt = 0;
+                               else if (histogram->histogram_mode)
+                                       pt = ((gdouble)log(v[chanmax])) / logmax * (height - 1);
+                               else
+                                       pt = ((gdouble)v[chanmax]) / max * (height - 1);
+
+                               pixbuf_draw_line(pixbuf,
                                        x, y, width, height,
-                                       x + i, y + height, x + i, y + height-pt, 
+                                       xpos, ypos, xpos, ypos - pt,
                                        r, g, b, 255);
-                       v[max2] = -1;
+                               }
+
+                       v[chanmax] = -1;
                        }
                }
+
        return TRUE;
 }
+
+void histogram_notify_cb(FileData *fd, NotifyType type, gpointer data)
+{
+       if ((type & NOTIFY_REREAD) && fd->histmap)
+               {
+               DEBUG_1("Notify histogram: %s %04x", fd->path, type);
+               histmap_free(fd->histmap);
+               fd->histmap = NULL;
+               }
+}
+
+/* vim: set shiftwidth=8 softtabstop=0 cindent cinoptions={1s: */