]> git.sesse.net Git - qscale/blobdiff - qscale.c
Added a load_from_stdio() parameter.
[qscale] / qscale.c
index 0f9598f32a29de1fed6b8ba694802a4dae1810a1..d89c8b9463d9a34cc367a22f72f4e4b9f75a7b7d 100644 (file)
--- a/qscale.c
+++ b/qscale.c
@@ -1,3 +1,21 @@
+/*
+ * qscale: Quick, high-quality JPEG-to-JPEG scaler.
+ * Copyright (C) 2008 Steinar H. Gunderson <sgunderson@bigfoot.com>
+ *
+ * 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, version 2 of the License.
+ *
+ * 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, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
+ */
+
 #include <stdio.h>
 #include <malloc.h>
 #include <math.h>
 #include <stdlib.h>
 #include "jpeglib.h"
 
+/* The number of pixels to process at a time when scaling vertically. */
 #define CACHE_LINE_FACTOR 16
 
+/* Whether to use SSE for horizontal scaling or not (requires SSE3). */
+#define USE_HORIZONTAL_SSE 1
+
+/* Whether to use SSE for vertical scaling or not (requires only SSE1). */
+#define USE_VERTICAL_SSE 1
+
+#if USE_VERTICAL_SSE
+#undef CACHE_LINE_FACTOR
+#define CACHE_LINE_FACTOR 16
+#endif
+
+#ifndef M_PI
+#define M_PI 3.14159265358979323846264
+#endif
+
 double sinc(double x)
 {
-       static const double cutoff = 1.220703668e-4;  // sqrt(sqrt(eps))
+       static const double cutoff = 1.220703668e-4;  /* sqrt(sqrt(eps)) */
 
        if (abs(x) < cutoff) {
-               // For small |x|, use Taylor series instead
+               /* For small |x|, use Taylor series instead */
                const double x2 = x * x;
                const double x4 = x2 * x2;
 
@@ -62,10 +96,11 @@ void hscale(float *pix, unsigned char *npix, unsigned w, unsigned h, unsigned nw
                        end = w - 1;
                }
 
-               // round up so we get a multiple of four for the SSE code
+#if USE_HORIZONTAL_SSE
+               /* round up so we get a multiple of four for the SSE code */
                int num = (end - start + 1);
                if (num % 4 != 0) {
-                       // prefer aligning it if possible
+                       /* prefer aligning it if possible */
                        if (start % 4 != 0 && start % 4 <= num % 4) {
                                num += start % 4;
                                start -= start % 4;
@@ -74,6 +109,7 @@ void hscale(float *pix, unsigned char *npix, unsigned w, unsigned h, unsigned nw
                                end += 4 - (num % 4);
                        }
                }
+#endif
 
                pd[x].start = start;
                pd[x].end = end;
@@ -101,21 +137,22 @@ void hscale(float *pix, unsigned char *npix, unsigned w, unsigned h, unsigned nw
                unsigned char *dptr = npix + y*dstride;
                unsigned char ch;
                for (x = 0; x < nw; ++x) {
+#if USE_HORIZONTAL_SSE
                        int result;
                        float acc;
                        long tmp;
                        static const float low = 0.0, high = 255.0;
-                       asm (
+                       __asm__ (
                                "pxor %1, %1               \n"
                                "xor %2, %2                \n"
-                               ".lbl2:                    \n"
+                               "0:                        \n"
                                "movups (%4,%2),%%xmm1     \n"
                                "movups (%3,%2),%%xmm2     \n"
                                "mulps %%xmm2,%%xmm1       \n"
                                "addps %%xmm1,%1           \n"
                                "add $16,%2                \n"
                                "dec %5                    \n"
-                               "jnz .lbl2                 \n"
+                               "jnz 0b                    \n"
                                "haddps %1,%1              \n"
                                "haddps %1,%1              \n"
                                "maxss %6,%1               \n"
@@ -132,7 +169,16 @@ void hscale(float *pix, unsigned char *npix, unsigned w, unsigned h, unsigned nw
                                : "memory", "xmm1", "xmm2"
                        );
 
-#if 0
+                       *dptr++ = (unsigned char)result;
+#else
+                       float acc = 0.0;
+                       float *cf = &coeffs[pd[x].startcoeff];
+                       unsigned sx;
+                       
+                       for (sx = pd[x].start; sx <= pd[x].end; ++sx) {
+                               acc += sptr[sx] * *cf++;
+                       }
+
                        if (acc < 0.0)
                                ch = 0;
                        else if (acc > 255.0)
@@ -141,7 +187,6 @@ void hscale(float *pix, unsigned char *npix, unsigned w, unsigned h, unsigned nw
                                ch = (unsigned char)acc;
                        *dptr++ = ch;
 #endif
-                       *dptr++ = (unsigned char)result;
                }
                ch = dptr[-1];
                for ( ; x < dstride; ++x) {
@@ -199,31 +244,13 @@ void vscale(unsigned char *pix, float *npix, unsigned w, unsigned h, unsigned nh
                unsigned char *sptr = pix + x;
                float *dptr = npix + x;
                for (y = 0; y < nh; ++y) {
-#if 0
-                       int i;
-                       float acc[CACHE_LINE_FACTOR];
-                       for (i = 0; i < CACHE_LINE_FACTOR; ++i)
-                               acc[i] = 0.0;
-                       float *cf = &coeffs[pd[y].startcoeff];
-                       unsigned sy;
-               
-                       for (sy = pd[y].start; sy <= pd[y].end; ++sy) {
-                               for (i = 0; i < CACHE_LINE_FACTOR; ++i) {
-                                       acc[i] += sptr[sy * w + i] * *cf;
-                               }
-                               ++cf;
-                       }
-
-                       for (i = 0; i < CACHE_LINE_FACTOR; ++i) {
-                               dptr[i] = acc[i];
-                       }
-#else
+#if USE_VERTICAL_SSE
                        /*
                         * xmm0 - xmm3: acc[0..15]
                         * xmm4: current filter coefficient
                         * xmm5, xmm6, xmm7: scratchpad
                         */
-                       asm (
+                       __asm__ (
                                /* clear */
                                "pxor %%xmm0, %%xmm0          \n"
                                "pxor %%xmm1, %%xmm1          \n"
@@ -231,7 +258,7 @@ void vscale(unsigned char *pix, float *npix, unsigned w, unsigned h, unsigned nh
                                "pxor %%xmm3, %%xmm3          \n"
 
                                /* main loop */
-                               ".lbl:                        \n"
+                               "0:                           \n"
                                
                                /* a zero is useful during unpacking */
                                "pxor %%xmm4, %%xmm4          \n"
@@ -290,7 +317,7 @@ void vscale(unsigned char *pix, float *npix, unsigned w, unsigned h, unsigned nh
                                "add $4, %2                   \n"
                                "add %3, %0                   \n"
                                "dec %1                       \n"
-                               "jnz .lbl                     \n"
+                               "jnz 0b                       \n"
 
                                /* store the values */
                                "movaps %%xmm0, (%4)          \n"
@@ -305,6 +332,24 @@ void vscale(unsigned char *pix, float *npix, unsigned w, unsigned h, unsigned nh
                                "r" (dptr)                           /* 4: dstptr base */
                                : "memory", "xmm0", "xmm1", "xmm2", "xmm3", "xmm4", "xmm5", "xmm6", "xmm7"
                        );
+#else
+                       int i;
+                       float acc[CACHE_LINE_FACTOR];
+                       for (i = 0; i < CACHE_LINE_FACTOR; ++i)
+                               acc[i] = 0.0;
+                       float *cf = &coeffs[pd[y].startcoeff];
+                       unsigned sy;
+               
+                       for (sy = pd[y].start; sy <= pd[y].end; ++sy) {
+                               for (i = 0; i < CACHE_LINE_FACTOR; ++i) {
+                                       acc[i] += sptr[sy * w + i] * *cf;
+                               }
+                               ++cf;
+                       }
+
+                       for (i = 0; i < CACHE_LINE_FACTOR; ++i) {
+                               dptr[i] = acc[i];
+                       }
 #endif
                        dptr += dstride;
                }
@@ -332,13 +377,27 @@ void vscale(unsigned char *pix, float *npix, unsigned w, unsigned h, unsigned nh
 
 int main(int argc, char **argv)
 {
+       /* user-settable parameters */
        unsigned nominal_w = atoi(argv[1]);
        unsigned nominal_h = atoi(argv[2]);
-
        unsigned samp_h0 = 2, samp_v0 = 2;
        unsigned samp_h1 = 1, samp_v1 = 1;
        unsigned samp_h2 = 1, samp_v2 = 1;
-       unsigned max_samp_h = 2, max_samp_v = 2;
+       unsigned jpeg_quality = 85;
+       /* end */
+
+       unsigned max_samp_h, max_samp_v;
+       max_samp_h = samp_h0;
+       if (samp_h1 > max_samp_h)
+               max_samp_h = samp_h1;
+       if (samp_h2 > max_samp_h)
+               max_samp_h = samp_h2;
+       
+       max_samp_v = samp_v0;
+       if (samp_v1 > max_samp_v)
+               max_samp_v = samp_v1;
+       if (samp_v2 > max_samp_v)
+               max_samp_v = samp_v2;
 
        unsigned nw0 = nominal_w * samp_h0 / max_samp_h, nh0 = nominal_h * samp_v0 / max_samp_v;
        unsigned nw1 = nominal_w * samp_h1 / max_samp_h, nh1 = nominal_h * samp_v1 / max_samp_v;
@@ -402,21 +461,21 @@ int main(int argc, char **argv)
        {
                float *npix = (float*)memalign(16, dinfo.comp_info[0].width_in_blocks * DCTSIZE * nh0 * sizeof(float)); 
                vscale(data_y, npix, dinfo.comp_info[0].width_in_blocks * DCTSIZE, h0, nh0, dinfo.comp_info[0].width_in_blocks * DCTSIZE);
-               data_ny = (unsigned char *)malloc(nw0 * stride0);
+               data_ny = (unsigned char *)malloc(nh0 * stride0);
                hscale(npix, data_ny, w0, nh0, nw0, dinfo.comp_info[0].width_in_blocks * DCTSIZE, stride0);
                free(npix);
        }
        {
                float *npix = (float*)memalign(16, dinfo.comp_info[1].width_in_blocks * DCTSIZE * nh1 * sizeof(float)); 
                vscale(data_cr, npix, dinfo.comp_info[1].width_in_blocks * DCTSIZE, h1, nh1, dinfo.comp_info[1].width_in_blocks * DCTSIZE);
-               data_ncr = (unsigned char *)malloc(nw1 * stride1);
+               data_ncr = (unsigned char *)malloc(nh1 * stride1);
                hscale(npix, data_ncr, w1, nh1, nw1, dinfo.comp_info[1].width_in_blocks * DCTSIZE, stride1);
                free(npix);
        }
        {
                float *npix = (float*)memalign(16, dinfo.comp_info[2].width_in_blocks * DCTSIZE * nh2 * sizeof(float)); 
                vscale(data_cb, npix, dinfo.comp_info[2].width_in_blocks * DCTSIZE, h2, nh2, dinfo.comp_info[2].width_in_blocks * DCTSIZE);
-               data_ncb = (unsigned char *)malloc(nw2 * stride2);
+               data_ncb = (unsigned char *)malloc(nh2 * stride2);
                hscale(npix, data_ncb, w2, nh2, nw2, dinfo.comp_info[2].width_in_blocks * DCTSIZE, stride2);
                free(npix);
        }
@@ -428,7 +487,7 @@ int main(int argc, char **argv)
        jpeg_stdio_dest(&cinfo, stdout);
        cinfo.input_components = 3;
        jpeg_set_defaults(&cinfo);
-       jpeg_set_quality(&cinfo, 85, FALSE);
+       jpeg_set_quality(&cinfo, jpeg_quality, FALSE);
        cinfo.image_width = nominal_w;
        cinfo.image_height = nominal_h;
        cinfo.raw_data_in = TRUE;
@@ -453,7 +512,7 @@ int main(int argc, char **argv)
                int i;
 
                for (i = 0; i < max_lines; ++i) {
-                       // simple edge extension
+                       /* simple edge extension */
                        int yline = i + blocks*DCTSIZE*cinfo.comp_info[0].v_samp_factor;
                        if (yline > nh0 - 1)
                                yline = nh0 - 1;