1// Copyright 2011 Google Inc. All Rights Reserved.
2//
3// This code is licensed under the same terms as WebM:
4//  Software License Agreement:  http://www.webmproject.org/license/software/
5//  Additional IP Rights Grant:  http://www.webmproject.org/license/additional/
6// -----------------------------------------------------------------------------
7//
8// functions for sample output.
9//
10// Author: Skal (pascal.massimino@gmail.com)
11
12#include <assert.h>
13#include <stdlib.h>
14#include "../dec/vp8i.h"
15#include "./webpi.h"
16#include "../dsp/dsp.h"
17#include "../dsp/yuv.h"
18
19#if defined(__cplusplus) || defined(c_plusplus)
20extern "C" {
21#endif
22
23//------------------------------------------------------------------------------
24// Main YUV<->RGB conversion functions
25
26static int EmitYUV(const VP8Io* const io, WebPDecParams* const p) {
27  WebPDecBuffer* output = p->output;
28  const WebPYUVABuffer* const buf = &output->u.YUVA;
29  uint8_t* const y_dst = buf->y + io->mb_y * buf->y_stride;
30  uint8_t* const u_dst = buf->u + (io->mb_y >> 1) * buf->u_stride;
31  uint8_t* const v_dst = buf->v + (io->mb_y >> 1) * buf->v_stride;
32  const int mb_w = io->mb_w;
33  const int mb_h = io->mb_h;
34  const int uv_w = (mb_w + 1) / 2;
35  const int uv_h = (mb_h + 1) / 2;
36  int j;
37  for (j = 0; j < mb_h; ++j) {
38    memcpy(y_dst + j * buf->y_stride, io->y + j * io->y_stride, mb_w);
39  }
40  for (j = 0; j < uv_h; ++j) {
41    memcpy(u_dst + j * buf->u_stride, io->u + j * io->uv_stride, uv_w);
42    memcpy(v_dst + j * buf->v_stride, io->v + j * io->uv_stride, uv_w);
43  }
44  return io->mb_h;
45}
46
47// Point-sampling U/V sampler.
48static int EmitSampledRGB(const VP8Io* const io, WebPDecParams* const p) {
49  WebPDecBuffer* output = p->output;
50  const WebPRGBABuffer* const buf = &output->u.RGBA;
51  uint8_t* dst = buf->rgba + io->mb_y * buf->stride;
52  const uint8_t* y_src = io->y;
53  const uint8_t* u_src = io->u;
54  const uint8_t* v_src = io->v;
55  const WebPSampleLinePairFunc sample = WebPSamplers[output->colorspace];
56  const int mb_w = io->mb_w;
57  const int last = io->mb_h - 1;
58  int j;
59  for (j = 0; j < last; j += 2) {
60    sample(y_src, y_src + io->y_stride, u_src, v_src,
61           dst, dst + buf->stride, mb_w);
62    y_src += 2 * io->y_stride;
63    u_src += io->uv_stride;
64    v_src += io->uv_stride;
65    dst += 2 * buf->stride;
66  }
67  if (j == last) {  // Just do the last line twice
68    sample(y_src, y_src, u_src, v_src, dst, dst, mb_w);
69  }
70  return io->mb_h;
71}
72
73//------------------------------------------------------------------------------
74// YUV444 -> RGB conversion
75
76#if 0   // TODO(skal): this is for future rescaling.
77static int EmitRGB(const VP8Io* const io, WebPDecParams* const p) {
78  WebPDecBuffer* output = p->output;
79  const WebPRGBABuffer* const buf = &output->u.RGBA;
80  uint8_t* dst = buf->rgba + io->mb_y * buf->stride;
81  const uint8_t* y_src = io->y;
82  const uint8_t* u_src = io->u;
83  const uint8_t* v_src = io->v;
84  const WebPYUV444Converter convert = WebPYUV444Converters[output->colorspace];
85  const int mb_w = io->mb_w;
86  const int last = io->mb_h;
87  int j;
88  for (j = 0; j < last; ++j) {
89    convert(y_src, u_src, v_src, dst, mb_w);
90    y_src += io->y_stride;
91    u_src += io->uv_stride;
92    v_src += io->uv_stride;
93    dst += buf->stride;
94  }
95  return io->mb_h;
96}
97#endif
98
99//------------------------------------------------------------------------------
100// Fancy upsampling
101
102#ifdef FANCY_UPSAMPLING
103static int EmitFancyRGB(const VP8Io* const io, WebPDecParams* const p) {
104  int num_lines_out = io->mb_h;   // a priori guess
105  const WebPRGBABuffer* const buf = &p->output->u.RGBA;
106  uint8_t* dst = buf->rgba + io->mb_y * buf->stride;
107  WebPUpsampleLinePairFunc upsample = WebPUpsamplers[p->output->colorspace];
108  const uint8_t* cur_y = io->y;
109  const uint8_t* cur_u = io->u;
110  const uint8_t* cur_v = io->v;
111  const uint8_t* top_u = p->tmp_u;
112  const uint8_t* top_v = p->tmp_v;
113  int y = io->mb_y;
114  int y_end = io->mb_y + io->mb_h;
115  const int mb_w = io->mb_w;
116  const int uv_w = (mb_w + 1) / 2;
117
118  if (y == 0) {
119    // First line is special cased. We mirror the u/v samples at boundary.
120    upsample(NULL, cur_y, cur_u, cur_v, cur_u, cur_v, NULL, dst, mb_w);
121  } else {
122    // We can finish the left-over line from previous call.
123    upsample(p->tmp_y, cur_y, top_u, top_v, cur_u, cur_v,
124             dst - buf->stride, dst, mb_w);
125    ++num_lines_out;
126  }
127  // Loop over each output pairs of row.
128  for (; y + 2 < y_end; y += 2) {
129    top_u = cur_u;
130    top_v = cur_v;
131    cur_u += io->uv_stride;
132    cur_v += io->uv_stride;
133    dst += 2 * buf->stride;
134    cur_y += 2 * io->y_stride;
135    upsample(cur_y - io->y_stride, cur_y,
136             top_u, top_v, cur_u, cur_v,
137             dst - buf->stride, dst, mb_w);
138  }
139  // move to last row
140  cur_y += io->y_stride;
141  if (io->crop_top + y_end < io->crop_bottom) {
142    // Save the unfinished samples for next call (as we're not done yet).
143    memcpy(p->tmp_y, cur_y, mb_w * sizeof(*p->tmp_y));
144    memcpy(p->tmp_u, cur_u, uv_w * sizeof(*p->tmp_u));
145    memcpy(p->tmp_v, cur_v, uv_w * sizeof(*p->tmp_v));
146    // The fancy upsampler leaves a row unfinished behind
147    // (except for the very last row)
148    num_lines_out--;
149  } else {
150    // Process the very last row of even-sized picture
151    if (!(y_end & 1)) {
152      upsample(cur_y, NULL, cur_u, cur_v, cur_u, cur_v,
153              dst + buf->stride, NULL, mb_w);
154    }
155  }
156  return num_lines_out;
157}
158
159#endif    /* FANCY_UPSAMPLING */
160
161//------------------------------------------------------------------------------
162
163static int EmitAlphaYUV(const VP8Io* const io, WebPDecParams* const p) {
164  const uint8_t* alpha = io->a;
165  const WebPYUVABuffer* const buf = &p->output->u.YUVA;
166  const int mb_w = io->mb_w;
167  const int mb_h = io->mb_h;
168  uint8_t* dst = buf->a + io->mb_y * buf->a_stride;
169  int j;
170
171  if (alpha != NULL) {
172    for (j = 0; j < mb_h; ++j) {
173      memcpy(dst, alpha, mb_w * sizeof(*dst));
174      alpha += io->width;
175      dst += buf->a_stride;
176    }
177  } else if (buf->a != NULL) {
178    // the user requested alpha, but there is none, set it to opaque.
179    for (j = 0; j < mb_h; ++j) {
180      memset(dst, 0xff, mb_w * sizeof(*dst));
181      dst += buf->a_stride;
182    }
183  }
184  return 0;
185}
186
187static int GetAlphaSourceRow(const VP8Io* const io,
188                             const uint8_t** alpha, int* const num_rows) {
189  int start_y = io->mb_y;
190  *num_rows = io->mb_h;
191
192  // Compensate for the 1-line delay of the fancy upscaler.
193  // This is similar to EmitFancyRGB().
194  if (io->fancy_upsampling) {
195    if (start_y == 0) {
196      // We don't process the last row yet. It'll be done during the next call.
197      --*num_rows;
198    } else {
199      --start_y;
200      // Fortunately, *alpha data is persistent, so we can go back
201      // one row and finish alpha blending, now that the fancy upscaler
202      // completed the YUV->RGB interpolation.
203      *alpha -= io->width;
204    }
205    if (io->crop_top + io->mb_y + io->mb_h == io->crop_bottom) {
206      // If it's the very last call, we process all the remaing rows!
207      *num_rows = io->crop_bottom - io->crop_top - start_y;
208    }
209  }
210  return start_y;
211}
212
213static int EmitAlphaRGB(const VP8Io* const io, WebPDecParams* const p) {
214  const uint8_t* alpha = io->a;
215  if (alpha != NULL) {
216    const int mb_w = io->mb_w;
217    int i, j;
218    const WEBP_CSP_MODE colorspace = p->output->colorspace;
219    const int alpha_first =
220        (colorspace == MODE_ARGB || colorspace == MODE_Argb);
221    const WebPRGBABuffer* const buf = &p->output->u.RGBA;
222    int num_rows;
223    const int start_y = GetAlphaSourceRow(io, &alpha, &num_rows);
224    uint32_t alpha_mask = 0xff;
225
226    {
227      uint8_t* const base_rgba = buf->rgba + start_y * buf->stride;
228      uint8_t* dst = base_rgba + (alpha_first ? 0 : 3);
229      for (j = 0; j < num_rows; ++j) {
230        for (i = 0; i < mb_w; ++i) {
231          const uint32_t alpha_value = alpha[i];
232          dst[4 * i] = alpha_value;
233          alpha_mask &= alpha_value;
234        }
235        alpha += io->width;
236        dst += buf->stride;
237      }
238      // alpha_mask is < 0xff if there's non-trivial alpha to premultiply with.
239      if (alpha_mask != 0xff && WebPIsPremultipliedMode(colorspace)) {
240        WebPApplyAlphaMultiply(base_rgba, alpha_first,
241                               mb_w, num_rows, buf->stride);
242      }
243    }
244  }
245  return 0;
246}
247
248static int EmitAlphaRGBA4444(const VP8Io* const io, WebPDecParams* const p) {
249  const uint8_t* alpha = io->a;
250  if (alpha != NULL) {
251    const int mb_w = io->mb_w;
252    int i, j;
253    const WebPRGBABuffer* const buf = &p->output->u.RGBA;
254    int num_rows;
255    const int start_y = GetAlphaSourceRow(io, &alpha, &num_rows);
256    uint32_t alpha_mask = 0x0f;
257
258    {
259      uint8_t* const base_rgba = buf->rgba + start_y * buf->stride;
260      uint8_t* alpha_dst = base_rgba + 1;
261      for (j = 0; j < num_rows; ++j) {
262        for (i = 0; i < mb_w; ++i) {
263          // Fill in the alpha value (converted to 4 bits).
264          const uint32_t alpha_value = alpha[i] >> 4;
265          alpha_dst[2 * i] = (alpha_dst[2 * i] & 0xf0) | alpha_value;
266          alpha_mask &= alpha_value;
267        }
268        alpha += io->width;
269        alpha_dst += buf->stride;
270      }
271      if (alpha_mask != 0x0f && p->output->colorspace == MODE_rgbA_4444) {
272        WebPApplyAlphaMultiply4444(base_rgba, mb_w, num_rows, buf->stride);
273      }
274    }
275  }
276  return 0;
277}
278
279//------------------------------------------------------------------------------
280// YUV rescaling (no final RGB conversion needed)
281
282static int Rescale(const uint8_t* src, int src_stride,
283                   int new_lines, WebPRescaler* const wrk) {
284  int num_lines_out = 0;
285  while (new_lines > 0) {    // import new contributions of source rows.
286    const int lines_in = WebPRescalerImport(wrk, new_lines, src, src_stride);
287    src += lines_in * src_stride;
288    new_lines -= lines_in;
289    num_lines_out += WebPRescalerExport(wrk);    // emit output row(s)
290  }
291  return num_lines_out;
292}
293
294static int EmitRescaledYUV(const VP8Io* const io, WebPDecParams* const p) {
295  const int mb_h = io->mb_h;
296  const int uv_mb_h = (mb_h + 1) >> 1;
297  const int num_lines_out = Rescale(io->y, io->y_stride, mb_h, &p->scaler_y);
298  Rescale(io->u, io->uv_stride, uv_mb_h, &p->scaler_u);
299  Rescale(io->v, io->uv_stride, uv_mb_h, &p->scaler_v);
300  return num_lines_out;
301}
302
303static int EmitRescaledAlphaYUV(const VP8Io* const io, WebPDecParams* const p) {
304  if (io->a != NULL) {
305    Rescale(io->a, io->width, io->mb_h, &p->scaler_a);
306  }
307  return 0;
308}
309
310static int InitYUVRescaler(const VP8Io* const io, WebPDecParams* const p) {
311  const int has_alpha = WebPIsAlphaMode(p->output->colorspace);
312  const WebPYUVABuffer* const buf = &p->output->u.YUVA;
313  const int out_width  = io->scaled_width;
314  const int out_height = io->scaled_height;
315  const int uv_out_width  = (out_width + 1) >> 1;
316  const int uv_out_height = (out_height + 1) >> 1;
317  const int uv_in_width  = (io->mb_w + 1) >> 1;
318  const int uv_in_height = (io->mb_h + 1) >> 1;
319  const size_t work_size = 2 * out_width;   // scratch memory for luma rescaler
320  const size_t uv_work_size = 2 * uv_out_width;  // and for each u/v ones
321  size_t tmp_size;
322  int32_t* work;
323
324  tmp_size = work_size + 2 * uv_work_size;
325  if (has_alpha) {
326    tmp_size += work_size;
327  }
328  p->memory = calloc(1, tmp_size * sizeof(*work));
329  if (p->memory == NULL) {
330    return 0;   // memory error
331  }
332  work = (int32_t*)p->memory;
333  WebPRescalerInit(&p->scaler_y, io->mb_w, io->mb_h,
334                   buf->y, out_width, out_height, buf->y_stride, 1,
335                   io->mb_w, out_width, io->mb_h, out_height,
336                   work);
337  WebPRescalerInit(&p->scaler_u, uv_in_width, uv_in_height,
338                   buf->u, uv_out_width, uv_out_height, buf->u_stride, 1,
339                   uv_in_width, uv_out_width,
340                   uv_in_height, uv_out_height,
341                   work + work_size);
342  WebPRescalerInit(&p->scaler_v, uv_in_width, uv_in_height,
343                   buf->v, uv_out_width, uv_out_height, buf->v_stride, 1,
344                   uv_in_width, uv_out_width,
345                   uv_in_height, uv_out_height,
346                   work + work_size + uv_work_size);
347  p->emit = EmitRescaledYUV;
348
349  if (has_alpha) {
350    WebPRescalerInit(&p->scaler_a, io->mb_w, io->mb_h,
351                     buf->a, out_width, out_height, buf->a_stride, 1,
352                     io->mb_w, out_width, io->mb_h, out_height,
353                     work + work_size + 2 * uv_work_size);
354    p->emit_alpha = EmitRescaledAlphaYUV;
355  }
356  return 1;
357}
358
359//------------------------------------------------------------------------------
360// RGBA rescaling
361
362static int ExportRGB(WebPDecParams* const p, int y_pos) {
363  const WebPYUV444Converter convert =
364      WebPYUV444Converters[p->output->colorspace];
365  const WebPRGBABuffer* const buf = &p->output->u.RGBA;
366  uint8_t* dst = buf->rgba + (p->last_y + y_pos) * buf->stride;
367  int num_lines_out = 0;
368  // For RGB rescaling, because of the YUV420, current scan position
369  // U/V can be +1/-1 line from the Y one.  Hence the double test.
370  while (WebPRescalerHasPendingOutput(&p->scaler_y) &&
371         WebPRescalerHasPendingOutput(&p->scaler_u)) {
372    assert(p->last_y + y_pos + num_lines_out < p->output->height);
373    assert(p->scaler_u.y_accum == p->scaler_v.y_accum);
374    WebPRescalerExportRow(&p->scaler_y);
375    WebPRescalerExportRow(&p->scaler_u);
376    WebPRescalerExportRow(&p->scaler_v);
377    convert(p->scaler_y.dst, p->scaler_u.dst, p->scaler_v.dst,
378            dst, p->scaler_y.dst_width);
379    dst += buf->stride;
380    ++num_lines_out;
381  }
382  return num_lines_out;
383}
384
385static int EmitRescaledRGB(const VP8Io* const io, WebPDecParams* const p) {
386  const int mb_h = io->mb_h;
387  const int uv_mb_h = (mb_h + 1) >> 1;
388  int j = 0, uv_j = 0;
389  int num_lines_out = 0;
390  while (j < mb_h) {
391    const int y_lines_in =
392        WebPRescalerImport(&p->scaler_y, mb_h - j,
393                           io->y + j * io->y_stride, io->y_stride);
394    const int u_lines_in =
395        WebPRescalerImport(&p->scaler_u, uv_mb_h - uv_j,
396                           io->u + uv_j * io->uv_stride, io->uv_stride);
397    const int v_lines_in =
398        WebPRescalerImport(&p->scaler_v, uv_mb_h - uv_j,
399                           io->v + uv_j * io->uv_stride, io->uv_stride);
400    (void)v_lines_in;   // remove a gcc warning
401    assert(u_lines_in == v_lines_in);
402    j += y_lines_in;
403    uv_j += u_lines_in;
404    num_lines_out += ExportRGB(p, num_lines_out);
405  }
406  return num_lines_out;
407}
408
409static int ExportAlpha(WebPDecParams* const p, int y_pos) {
410  const WebPRGBABuffer* const buf = &p->output->u.RGBA;
411  uint8_t* const base_rgba = buf->rgba + (p->last_y + y_pos) * buf->stride;
412  const WEBP_CSP_MODE colorspace = p->output->colorspace;
413  const int alpha_first =
414      (colorspace == MODE_ARGB || colorspace == MODE_Argb);
415  uint8_t* dst = base_rgba + (alpha_first ? 0 : 3);
416  int num_lines_out = 0;
417  const int is_premult_alpha = WebPIsPremultipliedMode(colorspace);
418  uint32_t alpha_mask = 0xff;
419  const int width = p->scaler_a.dst_width;
420
421  while (WebPRescalerHasPendingOutput(&p->scaler_a)) {
422    int i;
423    assert(p->last_y + y_pos + num_lines_out < p->output->height);
424    WebPRescalerExportRow(&p->scaler_a);
425    for (i = 0; i < width; ++i) {
426      const uint32_t alpha_value = p->scaler_a.dst[i];
427      dst[4 * i] = alpha_value;
428      alpha_mask &= alpha_value;
429    }
430    dst += buf->stride;
431    ++num_lines_out;
432  }
433  if (is_premult_alpha && alpha_mask != 0xff) {
434    WebPApplyAlphaMultiply(base_rgba, alpha_first,
435                           width, num_lines_out, buf->stride);
436  }
437  return num_lines_out;
438}
439
440static int ExportAlphaRGBA4444(WebPDecParams* const p, int y_pos) {
441  const WebPRGBABuffer* const buf = &p->output->u.RGBA;
442  uint8_t* const base_rgba = buf->rgba + (p->last_y + y_pos) * buf->stride;
443  uint8_t* alpha_dst = base_rgba + 1;
444  int num_lines_out = 0;
445  const WEBP_CSP_MODE colorspace = p->output->colorspace;
446  const int width = p->scaler_a.dst_width;
447  const int is_premult_alpha = WebPIsPremultipliedMode(colorspace);
448  uint32_t alpha_mask = 0x0f;
449
450  while (WebPRescalerHasPendingOutput(&p->scaler_a)) {
451    int i;
452    assert(p->last_y + y_pos + num_lines_out < p->output->height);
453    WebPRescalerExportRow(&p->scaler_a);
454    for (i = 0; i < width; ++i) {
455      // Fill in the alpha value (converted to 4 bits).
456      const uint32_t alpha_value = p->scaler_a.dst[i] >> 4;
457      alpha_dst[2 * i] = (alpha_dst[2 * i] & 0xf0) | alpha_value;
458      alpha_mask &= alpha_value;
459    }
460    alpha_dst += buf->stride;
461    ++num_lines_out;
462  }
463  if (is_premult_alpha && alpha_mask != 0x0f) {
464    WebPApplyAlphaMultiply4444(base_rgba, width, num_lines_out, buf->stride);
465  }
466  return num_lines_out;
467}
468
469static int EmitRescaledAlphaRGB(const VP8Io* const io, WebPDecParams* const p) {
470  if (io->a != NULL) {
471    WebPRescaler* const scaler = &p->scaler_a;
472    int j = 0;
473    int pos = 0;
474    while (j < io->mb_h) {
475      j += WebPRescalerImport(scaler, io->mb_h - j,
476                              io->a + j * io->width, io->width);
477      pos += p->emit_alpha_row(p, pos);
478    }
479  }
480  return 0;
481}
482
483static int InitRGBRescaler(const VP8Io* const io, WebPDecParams* const p) {
484  const int has_alpha = WebPIsAlphaMode(p->output->colorspace);
485  const int out_width  = io->scaled_width;
486  const int out_height = io->scaled_height;
487  const int uv_in_width  = (io->mb_w + 1) >> 1;
488  const int uv_in_height = (io->mb_h + 1) >> 1;
489  const size_t work_size = 2 * out_width;   // scratch memory for one rescaler
490  int32_t* work;  // rescalers work area
491  uint8_t* tmp;   // tmp storage for scaled YUV444 samples before RGB conversion
492  size_t tmp_size1, tmp_size2;
493
494  tmp_size1 = 3 * work_size;
495  tmp_size2 = 3 * out_width;
496  if (has_alpha) {
497    tmp_size1 += work_size;
498    tmp_size2 += out_width;
499  }
500  p->memory =
501      calloc(1, tmp_size1 * sizeof(*work) + tmp_size2 * sizeof(*tmp));
502  if (p->memory == NULL) {
503    return 0;   // memory error
504  }
505  work = (int32_t*)p->memory;
506  tmp = (uint8_t*)(work + tmp_size1);
507  WebPRescalerInit(&p->scaler_y, io->mb_w, io->mb_h,
508                   tmp + 0 * out_width, out_width, out_height, 0, 1,
509                   io->mb_w, out_width, io->mb_h, out_height,
510                   work + 0 * work_size);
511  WebPRescalerInit(&p->scaler_u, uv_in_width, uv_in_height,
512                   tmp + 1 * out_width, out_width, out_height, 0, 1,
513                   io->mb_w, 2 * out_width, io->mb_h, 2 * out_height,
514                   work + 1 * work_size);
515  WebPRescalerInit(&p->scaler_v, uv_in_width, uv_in_height,
516                   tmp + 2 * out_width, out_width, out_height, 0, 1,
517                   io->mb_w, 2 * out_width, io->mb_h, 2 * out_height,
518                   work + 2 * work_size);
519  p->emit = EmitRescaledRGB;
520
521  if (has_alpha) {
522    WebPRescalerInit(&p->scaler_a, io->mb_w, io->mb_h,
523                     tmp + 3 * out_width, out_width, out_height, 0, 1,
524                     io->mb_w, out_width, io->mb_h, out_height,
525                     work + 3 * work_size);
526    p->emit_alpha = EmitRescaledAlphaRGB;
527    if (p->output->colorspace == MODE_RGBA_4444 ||
528        p->output->colorspace == MODE_rgbA_4444) {
529      p->emit_alpha_row = ExportAlphaRGBA4444;
530    } else {
531      p->emit_alpha_row = ExportAlpha;
532    }
533  }
534  return 1;
535}
536
537//------------------------------------------------------------------------------
538// Default custom functions
539
540static int CustomSetup(VP8Io* io) {
541  WebPDecParams* const p = (WebPDecParams*)io->opaque;
542  const WEBP_CSP_MODE colorspace = p->output->colorspace;
543  const int is_rgb = WebPIsRGBMode(colorspace);
544  const int is_alpha = WebPIsAlphaMode(colorspace);
545
546  p->memory = NULL;
547  p->emit = NULL;
548  p->emit_alpha = NULL;
549  p->emit_alpha_row = NULL;
550  if (!WebPIoInitFromOptions(p->options, io, is_alpha ? MODE_YUV : MODE_YUVA)) {
551    return 0;
552  }
553
554  if (io->use_scaling) {
555    const int ok = is_rgb ? InitRGBRescaler(io, p) : InitYUVRescaler(io, p);
556    if (!ok) {
557      return 0;    // memory error
558    }
559  } else {
560    if (is_rgb) {
561      p->emit = EmitSampledRGB;   // default
562#ifdef FANCY_UPSAMPLING
563      if (io->fancy_upsampling) {
564        const int uv_width = (io->mb_w + 1) >> 1;
565        p->memory = malloc(io->mb_w + 2 * uv_width);
566        if (p->memory == NULL) {
567          return 0;   // memory error.
568        }
569        p->tmp_y = (uint8_t*)p->memory;
570        p->tmp_u = p->tmp_y + io->mb_w;
571        p->tmp_v = p->tmp_u + uv_width;
572        p->emit = EmitFancyRGB;
573        WebPInitUpsamplers();
574      }
575#endif
576    } else {
577      p->emit = EmitYUV;
578    }
579    if (is_alpha) {  // need transparency output
580      if (WebPIsPremultipliedMode(colorspace)) WebPInitPremultiply();
581      p->emit_alpha =
582          (colorspace == MODE_RGBA_4444 || colorspace == MODE_rgbA_4444) ?
583              EmitAlphaRGBA4444
584          : is_rgb ? EmitAlphaRGB
585          : EmitAlphaYUV;
586    }
587  }
588
589  if (is_rgb) {
590    VP8YUVInit();
591  }
592  return 1;
593}
594
595//------------------------------------------------------------------------------
596
597static int CustomPut(const VP8Io* io) {
598  WebPDecParams* p = (WebPDecParams*)io->opaque;
599  const int mb_w = io->mb_w;
600  const int mb_h = io->mb_h;
601  int num_lines_out;
602  assert(!(io->mb_y & 1));
603
604  if (mb_w <= 0 || mb_h <= 0) {
605    return 0;
606  }
607  num_lines_out = p->emit(io, p);
608  if (p->emit_alpha) {
609    p->emit_alpha(io, p);
610  }
611  p->last_y += num_lines_out;
612  return 1;
613}
614
615//------------------------------------------------------------------------------
616
617static void CustomTeardown(const VP8Io* io) {
618  WebPDecParams* const p = (WebPDecParams*)io->opaque;
619  free(p->memory);
620  p->memory = NULL;
621}
622
623//------------------------------------------------------------------------------
624// Main entry point
625
626void WebPInitCustomIo(WebPDecParams* const params, VP8Io* const io) {
627  io->put      = CustomPut;
628  io->setup    = CustomSetup;
629  io->teardown = CustomTeardown;
630  io->opaque   = params;
631}
632
633//------------------------------------------------------------------------------
634
635#if defined(__cplusplus) || defined(c_plusplus)
636}    // extern "C"
637#endif
638