/* scale.c - image scaling
 *
 * Raster graphics library
 *
 * Copyright (c) 1997-2003 Alfredo K. Kojima
 *
 *  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., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include <config.h>


#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <X11/Xlib.h>
#include <math.h>

#ifndef PI
#define PI 3.14159265358979323846
#endif

#include <assert.h>

#include "wraster.h"


/*
 *----------------------------------------------------------------------
 * RScaleImage--
 * 	Creates a scaled copy of an image.
 *
 * Returns:
 * 	The new scaled image.
 *
 *----------------------------------------------------------------------
 */
#ifndef broken_code
RImage*
RScaleImage(RImage *image, unsigned new_width, unsigned new_height)
{
    int ox;
    int px, py;
    register int x, y, t;
    int dx, dy;
    unsigned char *s;
    unsigned char *d;
    RImage *img;

    assert(new_width >= 0 && new_height >= 0);

    if (new_width == image->width && new_height == image->height)
        return RCloneImage(image);

    img = RCreateImage(new_width, new_height, image->format==RRGBAFormat);

    if (!img)
        return NULL;

    /* fixed point math idea taken from Imlib by
     * Carsten Haitzler (Rasterman) */
    dx = (image->width<<16)/new_width;
    dy = (image->height<<16)/new_height;

    py = 0;

    d = img->data;

    if (image->format == RRGBAFormat) {
        for (y=0; y<new_height; y++) {
            t = image->width*(py>>16);

            s = image->data+(t<<2); /* image->data+t*4 */

            ox = 0;
            px = 0;
            for (x=0; x<new_width; x++) {
                px += dx;

                *(d++) = *(s);
                *(d++) = *(s+1);
                *(d++) = *(s+2);
                *(d++) = *(s+3);

                t = (px - ox)>>16;
                ox += t<<16;

                s += t<<2; /* t*4 */
            }
            py += dy;
        }
    } else {
        for (y=0; y<new_height; y++) {
            t = image->width*(py>>16);

            s = image->data+(t<<1)+t; /* image->data+t*3 */

            ox = 0;
            px = 0;
            for (x=0; x<new_width; x++) {
                px += dx;

                *(d++) = *(s);
                *(d++) = *(s+1);
                *(d++) = *(s+2);

                t = (px - ox)>>16;
                ox += t<<16;

                s += (t<<1)+t; /* t*3 */
            }
            py += dy;
        }
    }

    return img;
}

#else
RImage*
RScaleImage(RImage *src, unsigned new_width, unsigned new_height)
{
    int ddy, ee;
    int h2;
    int yd;
    int xd, xs;
    RImage *dst;
    int e, xd2;
    unsigned char *sr, *sg, *sb, *sa;
    unsigned char *dr, *dg, *db, *da;
    int ys = 0;



    dst = RCreateImage(new_width, new_height, src->data[3]!=NULL);

    ddy = src->height/2;
    ee = (ddy/2) - dst->height;
    h2 = new_height/2;

    xd = dst->width;
    xs = src->width/2;
    e = (src->width/2)-xd;
    xd2 = xd/2;

    sr = src->data[0];
    sg = src->data[1];
    sb = src->data[2];
    sa = src->data[3];

    dr = dst->data[0];
    dg = dst->data[1];
    db = dst->data[2];
    da = dst->data[3];

    if (sa == NULL) {
        for (yd = 0; yd < new_height; yd++) {
            int x;

            sr = src->data[0] + ys * src->width;
            sg = src->data[1] + ys * src->width;
            sb = src->data[2] + ys * src->width;

            for (x = 0; x < xd; x++) {
                *(dr++) = *sr;
                *(dg++) = *sg;
                *(db++) = *sb;

                while (e >= 0) {
                    sr++;
                    sg++;
                    sb++;
                    e -= xd2;
                }
                e += xs;
            }
            while (ee >= 0) {
                ys++;
                ee -= h2;
            }
            ee += ddy;
        }
    } else {
        for (yd = 0; yd < new_height; yd++) {
            int x;

            sr = src->data[0] + ys * src->width;
            sg = src->data[1] + ys * src->width;
            sb = src->data[2] + ys * src->width;
            sa = src->data[3] + ys * src->width;

            for (x = 0; x < xd; x++) {
                *(dr++) = *sr;
                *(dg++) = *sg;
                *(db++) = *sb;
                *(da++) = *sa;

                while (e >= 0) {
                    sr++;
                    sg++;
                    sb++;
                    sa++;
                    e -= xd2;
                }
                e += xs;
            }
            while (ee >= 0) {
                ys++;
                ee -= h2;
            }
            ee += ddy;
        }
    }

    return dst;
}
#endif



/*
 * Filtered Image Rescaling code copy/pasted from
 * Graphics Gems III
 * Public Domain 1991 by Dale Schumacher
 */


/*
 *	filter function definitions
 */
#if 0
#define	filter_support		(1.0)

static double
filter(t)
double t;
{
    /* f(t) = 2|t|^3 - 3|t|^2 + 1, -1 <= t <= 1 */
    if(t < 0.0) t = -t;
    if(t < 1.0) return((2.0 * t - 3.0) * t * t + 1.0);
    return(0.0);
}
#endif
#define	box_support		(0.5)

static double
box_filter(t)
double t;
{
    if((t > -0.5) && (t <= 0.5)) return(1.0);
    return(0.0);
}

#define	triangle_support	(1.0)

static double
triangle_filter(t)
double t;
{
    if(t < 0.0) t = -t;
    if(t < 1.0) return(1.0 - t);
    return(0.0);
}

#define	bell_support		(1.5)

static double
bell_filter(t)		/* box (*) box (*) box */
double t;
{
    if(t < 0) t = -t;
    if(t < .5) return(.75 - (t * t));
    if(t < 1.5) {
        t = (t - 1.5);
        return(.5 * (t * t));
    }
    return(0.0);
}

#define	B_spline_support	(2.0)

static double
B_spline_filter(t)	/* box (*) box (*) box (*) box */
double t;
{
    double tt;

    if(t < 0) t = -t;
    if(t < 1) {
        tt = t * t;
        return((.5 * tt * t) - tt + (2.0 / 3.0));
    } else if(t < 2) {
        t = 2 - t;
        return((1.0 / 6.0) * (t * t * t));
    }
    return(0.0);
}

static double
sinc(x)
double x;
{
    x *= PI;
    if(x != 0) return(sin(x) / x);
    return(1.0);
}

#define	Lanczos3_support	(3.0)

static double
Lanczos3_filter(t)
double t;
{
    if(t < 0) t = -t;
    if(t < 3.0) return(sinc(t) * sinc(t/3.0));
    return(0.0);
}

#define	Mitchell_support	(2.0)

#define	B	(1.0 / 3.0)
#define	C	(1.0 / 3.0)

static double
Mitchell_filter(t)
double t;
{
    double tt;

    tt = t * t;
    if(t < 0) t = -t;
    if(t < 1.0) {
        t = (((12.0 - 9.0 * B - 6.0 * C) * (t * tt))
             + ((-18.0 + 12.0 * B + 6.0 * C) * tt)
             + (6.0 - 2 * B));
        return(t / 6.0);
    } else if(t < 2.0) {
        t = (((-1.0 * B - 6.0 * C) * (t * tt))
             + ((6.0 * B + 30.0 * C) * tt)
             + ((-12.0 * B - 48.0 * C) * t)
             + (8.0 * B + 24 * C));
        return(t / 6.0);
    }
    return(0.0);
}

static double (*filterf)() = Mitchell_filter;
static double fwidth = Mitchell_support;

void
_wraster_change_filter(int type)
{
    switch (type) {
    case RBoxFilter:
        filterf = box_filter;
        fwidth = box_support;
        break;
    case RTriangleFilter:
        filterf = triangle_filter;
        fwidth = triangle_support;
        break;
    case RBellFilter:
        filterf = bell_filter;
        fwidth = bell_support;
        break;
    case RBSplineFilter:
        filterf = B_spline_filter;
        fwidth = B_spline_support;
        break;
    case RLanczos3Filter:
        filterf = Lanczos3_filter;
        fwidth = Lanczos3_support;
        break;
    default:
    case RMitchellFilter:
        filterf = Mitchell_filter;
        fwidth = Mitchell_support;
        break;
    }
}


/*
 *	image rescaling routine
 */

typedef struct {
    int pixel;
    double	weight;
} CONTRIB;

typedef struct {
    int	n;		/* number of contributors */
    CONTRIB	*p;		/* pointer to list of contributions */
} CLIST;

CLIST	*contrib;		/* array of contribution lists */

/* clamp the input to the specified range */
#define CLAMP(v,l,h)    ((v)<(l) ? (l) : (v) > (h) ? (h) : v)


/* return of calloc is not checked if NULL in the function below! */
RImage*
RSmoothScaleImage(RImage *src, unsigned new_width, unsigned new_height)
{
    RImage *tmp;		       /* intermediate image */
    double xscale, yscale;	       /* zoom scale factors */
    int i, j, k;		       /* loop variables */
    int n;			       /* pixel number */
    double center, left, right;	       /* filter calculation variables */
    double width, fscale;	       /* filter calculation variables */
    double rweight, gweight, bweight;
    RImage *dst;
    unsigned char *p;
    unsigned char *sp;
    int sch = src->format == RRGBAFormat ? 4 : 3;


    dst = RCreateImage(new_width, new_height, False);

    /* create intermediate image to hold horizontal zoom */
    tmp = RCreateImage(dst->width, src->height, False);
    xscale = (double)new_width / (double)src->width;
    yscale = (double)new_height / (double)src->height;

    /* pre-calculate filter contributions for a row */
    contrib = (CLIST *)calloc(new_width, sizeof(CLIST));
    if (xscale < 1.0) {
        width = fwidth / xscale;
        fscale = 1.0 / xscale;
        for (i = 0; i < new_width; ++i) {
            contrib[i].n = 0;
            contrib[i].p = (CONTRIB *)calloc((int)(width * 2 + 1),
                                             sizeof(CONTRIB));
            center = (double) i / xscale;
            left = ceil(center - width);
            right = floor(center + width);
            for(j = left; j <= right; ++j) {
                rweight = center - (double) j;
                rweight = (*filterf)(rweight / fscale) / fscale;
                if(j < 0) {
                    n = -j;
                } else if(j >= src->width) {
                    n = (src->width - j) + src->width - 1;
                } else {
                    n = j;
                }
                k = contrib[i].n++;
                contrib[i].p[k].pixel = n*sch;
                contrib[i].p[k].weight = rweight;
            }
        }
    } else {

        for(i = 0; i < new_width; ++i) {
            contrib[i].n = 0;
            contrib[i].p = (CONTRIB *)calloc((int) (fwidth * 2 + 1),
                                             sizeof(CONTRIB));
            center = (double) i / xscale;
            left = ceil(center - fwidth);
            right = floor(center + fwidth);
            for(j = left; j <= right; ++j) {
                rweight = center - (double) j;
                rweight = (*filterf)(rweight);
                if(j < 0) {
                    n = -j;
                } else if(j >= src->width) {
                    n = (src->width - j) + src->width - 1;
                } else {
                    n = j;
                }
                k = contrib[i].n++;
                contrib[i].p[k].pixel = n*sch;
                contrib[i].p[k].weight = rweight;
            }
        }
    }

    /* apply filter to zoom horizontally from src to tmp */
    p = tmp->data;


    for(k = 0; k < tmp->height; ++k) {
        CONTRIB *pp;

        sp = src->data + src->width*k*sch;

        for(i = 0; i < tmp->width; ++i) {
            rweight = gweight = bweight = 0.0;

            pp = contrib[i].p;

            for(j = 0; j < contrib[i].n; ++j) {
                rweight += sp[pp[j].pixel] * pp[j].weight;
                gweight += sp[pp[j].pixel+1] * pp[j].weight;
                bweight += sp[pp[j].pixel+2] * pp[j].weight;
            }
            *p++ = CLAMP(rweight, 0, 255);
            *p++ = CLAMP(gweight, 0, 255);
            *p++ = CLAMP(bweight, 0, 255);
        }
    }

    /* free the memory allocated for horizontal filter weights */
    for(i = 0; i < tmp->width; ++i) {
        free(contrib[i].p);
    }
    free(contrib);

    /* pre-calculate filter contributions for a column */
    contrib = (CLIST *)calloc(dst->height, sizeof(CLIST));
    if(yscale < 1.0) {
        width = fwidth / yscale;
        fscale = 1.0 / yscale;
        for(i = 0; i < dst->height; ++i) {
            contrib[i].n = 0;
            contrib[i].p = (CONTRIB *)calloc((int) (width * 2 + 1),
                                             sizeof(CONTRIB));
            center = (double) i / yscale;
            left = ceil(center - width);
            right = floor(center + width);
            for(j = left; j <= right; ++j) {
                rweight = center - (double) j;
                rweight = (*filterf)(rweight / fscale) / fscale;
                if(j < 0) {
                    n = -j;
                } else if(j >= tmp->height) {
                    n = (tmp->height - j) + tmp->height - 1;
                } else {
                    n = j;
                }
                k = contrib[i].n++;
                contrib[i].p[k].pixel = n*3;
                contrib[i].p[k].weight = rweight;
            }
        }
    } else {
        for(i = 0; i < dst->height; ++i) {
            contrib[i].n = 0;
            contrib[i].p = (CONTRIB *)calloc((int) (fwidth * 2 + 1),
                                             sizeof(CONTRIB));
            center = (double) i / yscale;
            left = ceil(center - fwidth);
            right = floor(center + fwidth);
            for(j = left; j <= right; ++j) {
                rweight = center - (double) j;
                rweight = (*filterf)(rweight);
                if(j < 0) {
                    n = -j;
                } else if(j >= tmp->height) {
                    n = (tmp->height - j) + tmp->height - 1;
                } else {
                    n = j;
                }
                k = contrib[i].n++;
                contrib[i].p[k].pixel = n*3;
                contrib[i].p[k].weight = rweight;
            }
        }
    }

    /* apply filter to zoom vertically from tmp to dst */
    sp = malloc(tmp->height*3);

    for(k = 0; k < new_width; ++k) {
        CONTRIB *pp;

        p = dst->data + k*3;

        /* copy a column into a row */
        {
            int i;
            unsigned char *p, *d;

            d = sp;
            for(i = tmp->height, p = tmp->data + k*3; i-- > 0;
                p += tmp->width*3) {
                *d++ = *p;
                *d++ = *(p+1);
                *d++ = *(p+2);
            }
        }
        for(i = 0; i < new_height; ++i) {
            rweight = gweight = bweight = 0.0;

            pp = contrib[i].p;

            for(j = 0; j < contrib[i].n; ++j) {
                rweight += sp[pp[j].pixel] * pp[j].weight;
                gweight += sp[pp[j].pixel+1] * pp[j].weight;
                bweight += sp[pp[j].pixel+2] * pp[j].weight;
            }
            *p = CLAMP(rweight, 0, 255);
            *(p+1) = CLAMP(gweight, 0, 255);
            *(p+2) = CLAMP(bweight, 0, 255);
            p += new_width*3;
        }
    }
    free(sp);

    /* free the memory allocated for vertical filter weights */
    for(i = 0; i < dst->height; ++i) {
        free(contrib[i].p);
    }
    free(contrib);

    RReleaseImage(tmp);

    return dst;
}



syntax highlighted by Code2HTML, v. 0.9.1