1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285 |
- #define JPEG_INTERNALS
- #include "jinclude.h"
- #include "jpeglib.h"
- #ifdef QUANT_2PASS_SUPPORTED
- #define R_SCALE 2
- #define G_SCALE 3
- #define B_SCALE 1
- static const int c_scales[3] = { R_SCALE, G_SCALE, B_SCALE };
- #define C0_SCALE c_scales[rgb_red[cinfo->out_color_space]]
- #define C1_SCALE c_scales[rgb_green[cinfo->out_color_space]]
- #define C2_SCALE c_scales[rgb_blue[cinfo->out_color_space]]
- #define MAXNUMCOLORS (MAXJSAMPLE + 1)
- #define HIST_C0_BITS 5
- #define HIST_C1_BITS 6
- #define HIST_C2_BITS 5
- #define HIST_C0_ELEMS (1 << HIST_C0_BITS)
- #define HIST_C1_ELEMS (1 << HIST_C1_BITS)
- #define HIST_C2_ELEMS (1 << HIST_C2_BITS)
- #define C0_SHIFT (BITS_IN_JSAMPLE - HIST_C0_BITS)
- #define C1_SHIFT (BITS_IN_JSAMPLE - HIST_C1_BITS)
- #define C2_SHIFT (BITS_IN_JSAMPLE - HIST_C2_BITS)
- typedef UINT16 histcell;
- typedef histcell *histptr;
- typedef histcell hist1d[HIST_C2_ELEMS];
- typedef hist1d *hist2d;
- typedef hist2d *hist3d;
- #if BITS_IN_JSAMPLE == 8
- typedef INT16 FSERROR;
- typedef int LOCFSERROR;
- #else
- typedef JLONG FSERROR;
- typedef JLONG LOCFSERROR;
- #endif
- typedef FSERROR *FSERRPTR;
- typedef struct {
- struct jpeg_color_quantizer pub;
-
- JSAMPARRAY sv_colormap;
- int desired;
-
- hist3d histogram;
- boolean needs_zeroed;
-
- FSERRPTR fserrors;
- boolean on_odd_row;
- int *error_limiter;
- } my_cquantizer;
- typedef my_cquantizer *my_cquantize_ptr;
- METHODDEF(void)
- prescan_quantize(j_decompress_ptr cinfo, JSAMPARRAY input_buf,
- JSAMPARRAY output_buf, int num_rows)
- {
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
- register JSAMPROW ptr;
- register histptr histp;
- register hist3d histogram = cquantize->histogram;
- int row;
- JDIMENSION col;
- JDIMENSION width = cinfo->output_width;
- for (row = 0; row < num_rows; row++) {
- ptr = input_buf[row];
- for (col = width; col > 0; col--) {
-
- histp = &histogram[GETJSAMPLE(ptr[0]) >> C0_SHIFT]
- [GETJSAMPLE(ptr[1]) >> C1_SHIFT]
- [GETJSAMPLE(ptr[2]) >> C2_SHIFT];
-
- if (++(*histp) <= 0)
- (*histp)--;
- ptr += 3;
- }
- }
- }
- typedef struct {
-
- int c0min, c0max;
- int c1min, c1max;
- int c2min, c2max;
-
- JLONG volume;
-
- long colorcount;
- } box;
- typedef box *boxptr;
- LOCAL(boxptr)
- find_biggest_color_pop(boxptr boxlist, int numboxes)
- {
- register boxptr boxp;
- register int i;
- register long maxc = 0;
- boxptr which = NULL;
- for (i = 0, boxp = boxlist; i < numboxes; i++, boxp++) {
- if (boxp->colorcount > maxc && boxp->volume > 0) {
- which = boxp;
- maxc = boxp->colorcount;
- }
- }
- return which;
- }
- LOCAL(boxptr)
- find_biggest_volume(boxptr boxlist, int numboxes)
- {
- register boxptr boxp;
- register int i;
- register JLONG maxv = 0;
- boxptr which = NULL;
- for (i = 0, boxp = boxlist; i < numboxes; i++, boxp++) {
- if (boxp->volume > maxv) {
- which = boxp;
- maxv = boxp->volume;
- }
- }
- return which;
- }
- LOCAL(void)
- update_box(j_decompress_ptr cinfo, boxptr boxp)
- {
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
- hist3d histogram = cquantize->histogram;
- histptr histp;
- int c0, c1, c2;
- int c0min, c0max, c1min, c1max, c2min, c2max;
- JLONG dist0, dist1, dist2;
- long ccount;
- c0min = boxp->c0min; c0max = boxp->c0max;
- c1min = boxp->c1min; c1max = boxp->c1max;
- c2min = boxp->c2min; c2max = boxp->c2max;
- if (c0max > c0min)
- for (c0 = c0min; c0 <= c0max; c0++)
- for (c1 = c1min; c1 <= c1max; c1++) {
- histp = &histogram[c0][c1][c2min];
- for (c2 = c2min; c2 <= c2max; c2++)
- if (*histp++ != 0) {
- boxp->c0min = c0min = c0;
- goto have_c0min;
- }
- }
- have_c0min:
- if (c0max > c0min)
- for (c0 = c0max; c0 >= c0min; c0--)
- for (c1 = c1min; c1 <= c1max; c1++) {
- histp = &histogram[c0][c1][c2min];
- for (c2 = c2min; c2 <= c2max; c2++)
- if (*histp++ != 0) {
- boxp->c0max = c0max = c0;
- goto have_c0max;
- }
- }
- have_c0max:
- if (c1max > c1min)
- for (c1 = c1min; c1 <= c1max; c1++)
- for (c0 = c0min; c0 <= c0max; c0++) {
- histp = &histogram[c0][c1][c2min];
- for (c2 = c2min; c2 <= c2max; c2++)
- if (*histp++ != 0) {
- boxp->c1min = c1min = c1;
- goto have_c1min;
- }
- }
- have_c1min:
- if (c1max > c1min)
- for (c1 = c1max; c1 >= c1min; c1--)
- for (c0 = c0min; c0 <= c0max; c0++) {
- histp = &histogram[c0][c1][c2min];
- for (c2 = c2min; c2 <= c2max; c2++)
- if (*histp++ != 0) {
- boxp->c1max = c1max = c1;
- goto have_c1max;
- }
- }
- have_c1max:
- if (c2max > c2min)
- for (c2 = c2min; c2 <= c2max; c2++)
- for (c0 = c0min; c0 <= c0max; c0++) {
- histp = &histogram[c0][c1min][c2];
- for (c1 = c1min; c1 <= c1max; c1++, histp += HIST_C2_ELEMS)
- if (*histp != 0) {
- boxp->c2min = c2min = c2;
- goto have_c2min;
- }
- }
- have_c2min:
- if (c2max > c2min)
- for (c2 = c2max; c2 >= c2min; c2--)
- for (c0 = c0min; c0 <= c0max; c0++) {
- histp = &histogram[c0][c1min][c2];
- for (c1 = c1min; c1 <= c1max; c1++, histp += HIST_C2_ELEMS)
- if (*histp != 0) {
- boxp->c2max = c2max = c2;
- goto have_c2max;
- }
- }
- have_c2max:
-
- dist0 = ((c0max - c0min) << C0_SHIFT) * C0_SCALE;
- dist1 = ((c1max - c1min) << C1_SHIFT) * C1_SCALE;
- dist2 = ((c2max - c2min) << C2_SHIFT) * C2_SCALE;
- boxp->volume = dist0 * dist0 + dist1 * dist1 + dist2 * dist2;
-
- ccount = 0;
- for (c0 = c0min; c0 <= c0max; c0++)
- for (c1 = c1min; c1 <= c1max; c1++) {
- histp = &histogram[c0][c1][c2min];
- for (c2 = c2min; c2 <= c2max; c2++, histp++)
- if (*histp != 0) {
- ccount++;
- }
- }
- boxp->colorcount = ccount;
- }
- LOCAL(int)
- median_cut(j_decompress_ptr cinfo, boxptr boxlist, int numboxes,
- int desired_colors)
- {
- int n, lb;
- int c0, c1, c2, cmax;
- register boxptr b1, b2;
- while (numboxes < desired_colors) {
-
- if (numboxes * 2 <= desired_colors) {
- b1 = find_biggest_color_pop(boxlist, numboxes);
- } else {
- b1 = find_biggest_volume(boxlist, numboxes);
- }
- if (b1 == NULL)
- break;
- b2 = &boxlist[numboxes];
-
- b2->c0max = b1->c0max; b2->c1max = b1->c1max; b2->c2max = b1->c2max;
- b2->c0min = b1->c0min; b2->c1min = b1->c1min; b2->c2min = b1->c2min;
-
- c0 = ((b1->c0max - b1->c0min) << C0_SHIFT) * C0_SCALE;
- c1 = ((b1->c1max - b1->c1min) << C1_SHIFT) * C1_SCALE;
- c2 = ((b1->c2max - b1->c2min) << C2_SHIFT) * C2_SCALE;
-
- if (rgb_red[cinfo->out_color_space] == 0) {
- cmax = c1; n = 1;
- if (c0 > cmax) { cmax = c0; n = 0; }
- if (c2 > cmax) { n = 2; }
- } else {
- cmax = c1; n = 1;
- if (c2 > cmax) { cmax = c2; n = 2; }
- if (c0 > cmax) { n = 0; }
- }
-
- switch (n) {
- case 0:
- lb = (b1->c0max + b1->c0min) / 2;
- b1->c0max = lb;
- b2->c0min = lb + 1;
- break;
- case 1:
- lb = (b1->c1max + b1->c1min) / 2;
- b1->c1max = lb;
- b2->c1min = lb + 1;
- break;
- case 2:
- lb = (b1->c2max + b1->c2min) / 2;
- b1->c2max = lb;
- b2->c2min = lb + 1;
- break;
- }
-
- update_box(cinfo, b1);
- update_box(cinfo, b2);
- numboxes++;
- }
- return numboxes;
- }
- LOCAL(void)
- compute_color(j_decompress_ptr cinfo, boxptr boxp, int icolor)
- {
-
-
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
- hist3d histogram = cquantize->histogram;
- histptr histp;
- int c0, c1, c2;
- int c0min, c0max, c1min, c1max, c2min, c2max;
- long count;
- long total = 0;
- long c0total = 0;
- long c1total = 0;
- long c2total = 0;
- c0min = boxp->c0min; c0max = boxp->c0max;
- c1min = boxp->c1min; c1max = boxp->c1max;
- c2min = boxp->c2min; c2max = boxp->c2max;
- for (c0 = c0min; c0 <= c0max; c0++)
- for (c1 = c1min; c1 <= c1max; c1++) {
- histp = &histogram[c0][c1][c2min];
- for (c2 = c2min; c2 <= c2max; c2++) {
- if ((count = *histp++) != 0) {
- total += count;
- c0total += ((c0 << C0_SHIFT) + ((1 << C0_SHIFT) >> 1)) * count;
- c1total += ((c1 << C1_SHIFT) + ((1 << C1_SHIFT) >> 1)) * count;
- c2total += ((c2 << C2_SHIFT) + ((1 << C2_SHIFT) >> 1)) * count;
- }
- }
- }
- cinfo->colormap[0][icolor] = (JSAMPLE)((c0total + (total >> 1)) / total);
- cinfo->colormap[1][icolor] = (JSAMPLE)((c1total + (total >> 1)) / total);
- cinfo->colormap[2][icolor] = (JSAMPLE)((c2total + (total >> 1)) / total);
- }
- LOCAL(void)
- select_colors(j_decompress_ptr cinfo, int desired_colors)
- {
- boxptr boxlist;
- int numboxes;
- int i;
-
- boxlist = (boxptr)(*cinfo->mem->alloc_small)
- ((j_common_ptr)cinfo, JPOOL_IMAGE, desired_colors * sizeof(box));
-
- numboxes = 1;
- boxlist[0].c0min = 0;
- boxlist[0].c0max = MAXJSAMPLE >> C0_SHIFT;
- boxlist[0].c1min = 0;
- boxlist[0].c1max = MAXJSAMPLE >> C1_SHIFT;
- boxlist[0].c2min = 0;
- boxlist[0].c2max = MAXJSAMPLE >> C2_SHIFT;
-
- update_box(cinfo, &boxlist[0]);
-
- numboxes = median_cut(cinfo, boxlist, numboxes, desired_colors);
-
- for (i = 0; i < numboxes; i++)
- compute_color(cinfo, &boxlist[i], i);
- cinfo->actual_number_of_colors = numboxes;
- TRACEMS1(cinfo, 1, JTRC_QUANT_SELECTED, numboxes);
- }
- #define BOX_C0_LOG (HIST_C0_BITS - 3)
- #define BOX_C1_LOG (HIST_C1_BITS - 3)
- #define BOX_C2_LOG (HIST_C2_BITS - 3)
- #define BOX_C0_ELEMS (1 << BOX_C0_LOG)
- #define BOX_C1_ELEMS (1 << BOX_C1_LOG)
- #define BOX_C2_ELEMS (1 << BOX_C2_LOG)
- #define BOX_C0_SHIFT (C0_SHIFT + BOX_C0_LOG)
- #define BOX_C1_SHIFT (C1_SHIFT + BOX_C1_LOG)
- #define BOX_C2_SHIFT (C2_SHIFT + BOX_C2_LOG)
- LOCAL(int)
- find_nearby_colors(j_decompress_ptr cinfo, int minc0, int minc1, int minc2,
- JSAMPLE colorlist[])
- {
- int numcolors = cinfo->actual_number_of_colors;
- int maxc0, maxc1, maxc2;
- int centerc0, centerc1, centerc2;
- int i, x, ncolors;
- JLONG minmaxdist, min_dist, max_dist, tdist;
- JLONG mindist[MAXNUMCOLORS];
-
- maxc0 = minc0 + ((1 << BOX_C0_SHIFT) - (1 << C0_SHIFT));
- centerc0 = (minc0 + maxc0) >> 1;
- maxc1 = minc1 + ((1 << BOX_C1_SHIFT) - (1 << C1_SHIFT));
- centerc1 = (minc1 + maxc1) >> 1;
- maxc2 = minc2 + ((1 << BOX_C2_SHIFT) - (1 << C2_SHIFT));
- centerc2 = (minc2 + maxc2) >> 1;
-
- minmaxdist = 0x7FFFFFFFL;
- for (i = 0; i < numcolors; i++) {
-
- x = GETJSAMPLE(cinfo->colormap[0][i]);
- if (x < minc0) {
- tdist = (x - minc0) * C0_SCALE;
- min_dist = tdist * tdist;
- tdist = (x - maxc0) * C0_SCALE;
- max_dist = tdist * tdist;
- } else if (x > maxc0) {
- tdist = (x - maxc0) * C0_SCALE;
- min_dist = tdist * tdist;
- tdist = (x - minc0) * C0_SCALE;
- max_dist = tdist * tdist;
- } else {
-
- min_dist = 0;
- if (x <= centerc0) {
- tdist = (x - maxc0) * C0_SCALE;
- max_dist = tdist * tdist;
- } else {
- tdist = (x - minc0) * C0_SCALE;
- max_dist = tdist * tdist;
- }
- }
- x = GETJSAMPLE(cinfo->colormap[1][i]);
- if (x < minc1) {
- tdist = (x - minc1) * C1_SCALE;
- min_dist += tdist * tdist;
- tdist = (x - maxc1) * C1_SCALE;
- max_dist += tdist * tdist;
- } else if (x > maxc1) {
- tdist = (x - maxc1) * C1_SCALE;
- min_dist += tdist * tdist;
- tdist = (x - minc1) * C1_SCALE;
- max_dist += tdist * tdist;
- } else {
-
- if (x <= centerc1) {
- tdist = (x - maxc1) * C1_SCALE;
- max_dist += tdist * tdist;
- } else {
- tdist = (x - minc1) * C1_SCALE;
- max_dist += tdist * tdist;
- }
- }
- x = GETJSAMPLE(cinfo->colormap[2][i]);
- if (x < minc2) {
- tdist = (x - minc2) * C2_SCALE;
- min_dist += tdist * tdist;
- tdist = (x - maxc2) * C2_SCALE;
- max_dist += tdist * tdist;
- } else if (x > maxc2) {
- tdist = (x - maxc2) * C2_SCALE;
- min_dist += tdist * tdist;
- tdist = (x - minc2) * C2_SCALE;
- max_dist += tdist * tdist;
- } else {
-
- if (x <= centerc2) {
- tdist = (x - maxc2) * C2_SCALE;
- max_dist += tdist * tdist;
- } else {
- tdist = (x - minc2) * C2_SCALE;
- max_dist += tdist * tdist;
- }
- }
- mindist[i] = min_dist;
- if (max_dist < minmaxdist)
- minmaxdist = max_dist;
- }
-
- ncolors = 0;
- for (i = 0; i < numcolors; i++) {
- if (mindist[i] <= minmaxdist)
- colorlist[ncolors++] = (JSAMPLE)i;
- }
- return ncolors;
- }
- LOCAL(void)
- find_best_colors(j_decompress_ptr cinfo, int minc0, int minc1, int minc2,
- int numcolors, JSAMPLE colorlist[], JSAMPLE bestcolor[])
- {
- int ic0, ic1, ic2;
- int i, icolor;
- register JLONG *bptr;
- JSAMPLE *cptr;
- JLONG dist0, dist1;
- register JLONG dist2;
- JLONG xx0, xx1;
- register JLONG xx2;
- JLONG inc0, inc1, inc2;
-
- JLONG bestdist[BOX_C0_ELEMS * BOX_C1_ELEMS * BOX_C2_ELEMS];
-
- bptr = bestdist;
- for (i = BOX_C0_ELEMS * BOX_C1_ELEMS * BOX_C2_ELEMS - 1; i >= 0; i--)
- *bptr++ = 0x7FFFFFFFL;
-
-
- #define STEP_C0 ((1 << C0_SHIFT) * C0_SCALE)
- #define STEP_C1 ((1 << C1_SHIFT) * C1_SCALE)
- #define STEP_C2 ((1 << C2_SHIFT) * C2_SCALE)
- for (i = 0; i < numcolors; i++) {
- icolor = GETJSAMPLE(colorlist[i]);
-
- inc0 = (minc0 - GETJSAMPLE(cinfo->colormap[0][icolor])) * C0_SCALE;
- dist0 = inc0 * inc0;
- inc1 = (minc1 - GETJSAMPLE(cinfo->colormap[1][icolor])) * C1_SCALE;
- dist0 += inc1 * inc1;
- inc2 = (minc2 - GETJSAMPLE(cinfo->colormap[2][icolor])) * C2_SCALE;
- dist0 += inc2 * inc2;
-
- inc0 = inc0 * (2 * STEP_C0) + STEP_C0 * STEP_C0;
- inc1 = inc1 * (2 * STEP_C1) + STEP_C1 * STEP_C1;
- inc2 = inc2 * (2 * STEP_C2) + STEP_C2 * STEP_C2;
-
- bptr = bestdist;
- cptr = bestcolor;
- xx0 = inc0;
- for (ic0 = BOX_C0_ELEMS - 1; ic0 >= 0; ic0--) {
- dist1 = dist0;
- xx1 = inc1;
- for (ic1 = BOX_C1_ELEMS - 1; ic1 >= 0; ic1--) {
- dist2 = dist1;
- xx2 = inc2;
- for (ic2 = BOX_C2_ELEMS - 1; ic2 >= 0; ic2--) {
- if (dist2 < *bptr) {
- *bptr = dist2;
- *cptr = (JSAMPLE)icolor;
- }
- dist2 += xx2;
- xx2 += 2 * STEP_C2 * STEP_C2;
- bptr++;
- cptr++;
- }
- dist1 += xx1;
- xx1 += 2 * STEP_C1 * STEP_C1;
- }
- dist0 += xx0;
- xx0 += 2 * STEP_C0 * STEP_C0;
- }
- }
- }
- LOCAL(void)
- fill_inverse_cmap(j_decompress_ptr cinfo, int c0, int c1, int c2)
- {
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
- hist3d histogram = cquantize->histogram;
- int minc0, minc1, minc2;
- int ic0, ic1, ic2;
- register JSAMPLE *cptr;
- register histptr cachep;
-
- JSAMPLE colorlist[MAXNUMCOLORS];
- int numcolors;
-
- JSAMPLE bestcolor[BOX_C0_ELEMS * BOX_C1_ELEMS * BOX_C2_ELEMS];
-
- c0 >>= BOX_C0_LOG;
- c1 >>= BOX_C1_LOG;
- c2 >>= BOX_C2_LOG;
-
- minc0 = (c0 << BOX_C0_SHIFT) + ((1 << C0_SHIFT) >> 1);
- minc1 = (c1 << BOX_C1_SHIFT) + ((1 << C1_SHIFT) >> 1);
- minc2 = (c2 << BOX_C2_SHIFT) + ((1 << C2_SHIFT) >> 1);
-
- numcolors = find_nearby_colors(cinfo, minc0, minc1, minc2, colorlist);
-
- find_best_colors(cinfo, minc0, minc1, minc2, numcolors, colorlist,
- bestcolor);
-
- c0 <<= BOX_C0_LOG;
- c1 <<= BOX_C1_LOG;
- c2 <<= BOX_C2_LOG;
- cptr = bestcolor;
- for (ic0 = 0; ic0 < BOX_C0_ELEMS; ic0++) {
- for (ic1 = 0; ic1 < BOX_C1_ELEMS; ic1++) {
- cachep = &histogram[c0 + ic0][c1 + ic1][c2];
- for (ic2 = 0; ic2 < BOX_C2_ELEMS; ic2++) {
- *cachep++ = (histcell)(GETJSAMPLE(*cptr++) + 1);
- }
- }
- }
- }
- METHODDEF(void)
- pass2_no_dither(j_decompress_ptr cinfo, JSAMPARRAY input_buf,
- JSAMPARRAY output_buf, int num_rows)
- {
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
- hist3d histogram = cquantize->histogram;
- register JSAMPROW inptr, outptr;
- register histptr cachep;
- register int c0, c1, c2;
- int row;
- JDIMENSION col;
- JDIMENSION width = cinfo->output_width;
- for (row = 0; row < num_rows; row++) {
- inptr = input_buf[row];
- outptr = output_buf[row];
- for (col = width; col > 0; col--) {
-
- c0 = GETJSAMPLE(*inptr++) >> C0_SHIFT;
- c1 = GETJSAMPLE(*inptr++) >> C1_SHIFT;
- c2 = GETJSAMPLE(*inptr++) >> C2_SHIFT;
- cachep = &histogram[c0][c1][c2];
-
-
- if (*cachep == 0)
- fill_inverse_cmap(cinfo, c0, c1, c2);
-
- *outptr++ = (JSAMPLE)(*cachep - 1);
- }
- }
- }
- METHODDEF(void)
- pass2_fs_dither(j_decompress_ptr cinfo, JSAMPARRAY input_buf,
- JSAMPARRAY output_buf, int num_rows)
- {
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
- hist3d histogram = cquantize->histogram;
- register LOCFSERROR cur0, cur1, cur2;
- LOCFSERROR belowerr0, belowerr1, belowerr2;
- LOCFSERROR bpreverr0, bpreverr1, bpreverr2;
- register FSERRPTR errorptr;
- JSAMPROW inptr;
- JSAMPROW outptr;
- histptr cachep;
- int dir;
- int dir3;
- int row;
- JDIMENSION col;
- JDIMENSION width = cinfo->output_width;
- JSAMPLE *range_limit = cinfo->sample_range_limit;
- int *error_limit = cquantize->error_limiter;
- JSAMPROW colormap0 = cinfo->colormap[0];
- JSAMPROW colormap1 = cinfo->colormap[1];
- JSAMPROW colormap2 = cinfo->colormap[2];
- SHIFT_TEMPS
- for (row = 0; row < num_rows; row++) {
- inptr = input_buf[row];
- outptr = output_buf[row];
- if (cquantize->on_odd_row) {
-
- inptr += (width - 1) * 3;
- outptr += width - 1;
- dir = -1;
- dir3 = -3;
- errorptr = cquantize->fserrors + (width + 1) * 3;
- cquantize->on_odd_row = FALSE;
- } else {
-
- dir = 1;
- dir3 = 3;
- errorptr = cquantize->fserrors;
- cquantize->on_odd_row = TRUE;
- }
-
- cur0 = cur1 = cur2 = 0;
-
- belowerr0 = belowerr1 = belowerr2 = 0;
- bpreverr0 = bpreverr1 = bpreverr2 = 0;
- for (col = width; col > 0; col--) {
-
- cur0 = RIGHT_SHIFT(cur0 + errorptr[dir3 + 0] + 8, 4);
- cur1 = RIGHT_SHIFT(cur1 + errorptr[dir3 + 1] + 8, 4);
- cur2 = RIGHT_SHIFT(cur2 + errorptr[dir3 + 2] + 8, 4);
-
- cur0 = error_limit[cur0];
- cur1 = error_limit[cur1];
- cur2 = error_limit[cur2];
-
- cur0 += GETJSAMPLE(inptr[0]);
- cur1 += GETJSAMPLE(inptr[1]);
- cur2 += GETJSAMPLE(inptr[2]);
- cur0 = GETJSAMPLE(range_limit[cur0]);
- cur1 = GETJSAMPLE(range_limit[cur1]);
- cur2 = GETJSAMPLE(range_limit[cur2]);
-
- cachep =
- &histogram[cur0 >> C0_SHIFT][cur1 >> C1_SHIFT][cur2 >> C2_SHIFT];
-
-
- if (*cachep == 0)
- fill_inverse_cmap(cinfo, cur0 >> C0_SHIFT, cur1 >> C1_SHIFT,
- cur2 >> C2_SHIFT);
-
- {
- register int pixcode = *cachep - 1;
- *outptr = (JSAMPLE)pixcode;
-
- cur0 -= GETJSAMPLE(colormap0[pixcode]);
- cur1 -= GETJSAMPLE(colormap1[pixcode]);
- cur2 -= GETJSAMPLE(colormap2[pixcode]);
- }
-
- {
- register LOCFSERROR bnexterr;
- bnexterr = cur0;
- errorptr[0] = (FSERROR)(bpreverr0 + cur0 * 3);
- bpreverr0 = belowerr0 + cur0 * 5;
- belowerr0 = bnexterr;
- cur0 *= 7;
- bnexterr = cur1;
- errorptr[1] = (FSERROR)(bpreverr1 + cur1 * 3);
- bpreverr1 = belowerr1 + cur1 * 5;
- belowerr1 = bnexterr;
- cur1 *= 7;
- bnexterr = cur2;
- errorptr[2] = (FSERROR)(bpreverr2 + cur2 * 3);
- bpreverr2 = belowerr2 + cur2 * 5;
- belowerr2 = bnexterr;
- cur2 *= 7;
- }
-
- inptr += dir3;
- outptr += dir;
- errorptr += dir3;
- }
-
- errorptr[0] = (FSERROR)bpreverr0;
- errorptr[1] = (FSERROR)bpreverr1;
- errorptr[2] = (FSERROR)bpreverr2;
- }
- }
- LOCAL(void)
- init_error_limit(j_decompress_ptr cinfo)
- {
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
- int *table;
- int in, out;
- table = (int *)(*cinfo->mem->alloc_small)
- ((j_common_ptr)cinfo, JPOOL_IMAGE, (MAXJSAMPLE * 2 + 1) * sizeof(int));
- table += MAXJSAMPLE;
- cquantize->error_limiter = table;
- #define STEPSIZE ((MAXJSAMPLE + 1) / 16)
-
- out = 0;
- for (in = 0; in < STEPSIZE; in++, out++) {
- table[in] = out; table[-in] = -out;
- }
-
- for (; in < STEPSIZE * 3; in++, out += (in & 1) ? 0 : 1) {
- table[in] = out; table[-in] = -out;
- }
-
- for (; in <= MAXJSAMPLE; in++) {
- table[in] = out; table[-in] = -out;
- }
- #undef STEPSIZE
- }
- METHODDEF(void)
- finish_pass1(j_decompress_ptr cinfo)
- {
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
-
- cinfo->colormap = cquantize->sv_colormap;
- select_colors(cinfo, cquantize->desired);
-
- cquantize->needs_zeroed = TRUE;
- }
- METHODDEF(void)
- finish_pass2(j_decompress_ptr cinfo)
- {
-
- }
- METHODDEF(void)
- start_pass_2_quant(j_decompress_ptr cinfo, boolean is_pre_scan)
- {
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
- hist3d histogram = cquantize->histogram;
- int i;
-
-
- if (cinfo->dither_mode != JDITHER_NONE)
- cinfo->dither_mode = JDITHER_FS;
- if (is_pre_scan) {
-
- cquantize->pub.color_quantize = prescan_quantize;
- cquantize->pub.finish_pass = finish_pass1;
- cquantize->needs_zeroed = TRUE;
- } else {
-
- if (cinfo->dither_mode == JDITHER_FS)
- cquantize->pub.color_quantize = pass2_fs_dither;
- else
- cquantize->pub.color_quantize = pass2_no_dither;
- cquantize->pub.finish_pass = finish_pass2;
-
- i = cinfo->actual_number_of_colors;
- if (i < 1)
- ERREXIT1(cinfo, JERR_QUANT_FEW_COLORS, 1);
- if (i > MAXNUMCOLORS)
- ERREXIT1(cinfo, JERR_QUANT_MANY_COLORS, MAXNUMCOLORS);
- if (cinfo->dither_mode == JDITHER_FS) {
- size_t arraysize =
- (size_t)((cinfo->output_width + 2) * (3 * sizeof(FSERROR)));
-
- if (cquantize->fserrors == NULL)
- cquantize->fserrors = (FSERRPTR)(*cinfo->mem->alloc_large)
- ((j_common_ptr)cinfo, JPOOL_IMAGE, arraysize);
-
- jzero_far((void *)cquantize->fserrors, arraysize);
-
- if (cquantize->error_limiter == NULL)
- init_error_limit(cinfo);
- cquantize->on_odd_row = FALSE;
- }
- }
-
- if (cquantize->needs_zeroed) {
- for (i = 0; i < HIST_C0_ELEMS; i++) {
- jzero_far((void *)histogram[i],
- HIST_C1_ELEMS * HIST_C2_ELEMS * sizeof(histcell));
- }
- cquantize->needs_zeroed = FALSE;
- }
- }
- METHODDEF(void)
- new_color_map_2_quant(j_decompress_ptr cinfo)
- {
- my_cquantize_ptr cquantize = (my_cquantize_ptr)cinfo->cquantize;
-
- cquantize->needs_zeroed = TRUE;
- }
- GLOBAL(void)
- jinit_2pass_quantizer(j_decompress_ptr cinfo)
- {
- my_cquantize_ptr cquantize;
- int i;
- cquantize = (my_cquantize_ptr)
- (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE,
- sizeof(my_cquantizer));
- cinfo->cquantize = (struct jpeg_color_quantizer *)cquantize;
- cquantize->pub.start_pass = start_pass_2_quant;
- cquantize->pub.new_color_map = new_color_map_2_quant;
- cquantize->fserrors = NULL;
- cquantize->error_limiter = NULL;
-
- if (cinfo->out_color_components != 3)
- ERREXIT(cinfo, JERR_NOTIMPL);
-
- cquantize->histogram = (hist3d)(*cinfo->mem->alloc_small)
- ((j_common_ptr)cinfo, JPOOL_IMAGE, HIST_C0_ELEMS * sizeof(hist2d));
- for (i = 0; i < HIST_C0_ELEMS; i++) {
- cquantize->histogram[i] = (hist2d)(*cinfo->mem->alloc_large)
- ((j_common_ptr)cinfo, JPOOL_IMAGE,
- HIST_C1_ELEMS * HIST_C2_ELEMS * sizeof(histcell));
- }
- cquantize->needs_zeroed = TRUE;
-
- if (cinfo->enable_2pass_quant) {
-
- int desired = cinfo->desired_number_of_colors;
-
- if (desired < 8)
- ERREXIT1(cinfo, JERR_QUANT_FEW_COLORS, 8);
-
- if (desired > MAXNUMCOLORS)
- ERREXIT1(cinfo, JERR_QUANT_MANY_COLORS, MAXNUMCOLORS);
- cquantize->sv_colormap = (*cinfo->mem->alloc_sarray)
- ((j_common_ptr)cinfo, JPOOL_IMAGE, (JDIMENSION)desired, (JDIMENSION)3);
- cquantize->desired = desired;
- } else
- cquantize->sv_colormap = NULL;
-
-
- if (cinfo->dither_mode != JDITHER_NONE)
- cinfo->dither_mode = JDITHER_FS;
-
- if (cinfo->dither_mode == JDITHER_FS) {
- cquantize->fserrors = (FSERRPTR)(*cinfo->mem->alloc_large)
- ((j_common_ptr)cinfo, JPOOL_IMAGE,
- (size_t)((cinfo->output_width + 2) * (3 * sizeof(FSERROR))));
-
- init_error_limit(cinfo);
- }
- }
- #endif
|