film_grain_tmpl.c 21.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
/*
 * Copyright © 2018, Niklas Haas
 * Copyright © 2018, VideoLAN and dav1d authors
 * Copyright © 2018, Two Orioles, LLC
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this
 *    list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

29 30
#include "config.h"

31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
#include <assert.h>
#include <stdint.h>

#include "common.h"
#include "common/intops.h"
#include "common/bitdepth.h"
#include "tables.h"

#include "film_grain.h"

#if BITDEPTH == 8
typedef int8_t entry;
#else
typedef int16_t entry;
#endif

enum {
    GRAIN_WIDTH  = 82,
    GRAIN_HEIGHT = 73,
    SUB_GRAIN_WIDTH = 44,
    SUB_GRAIN_HEIGHT = 38,
    SUB_GRAIN_OFFSET = 6,
    BLOCK_SIZE = 32,
54 55 56 57 58
#if BITDEPTH == 8
    SCALING_SIZE = 256
#else
    SCALING_SIZE = 4096
#endif
59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
};

static inline int get_random_number(const int bits, unsigned *state) {
    const int r = *state;
    unsigned bit = ((r >> 0) ^ (r >> 1) ^ (r >> 3) ^ (r >> 12)) & 1;
    *state = (r >> 1) | (bit << 15);

    return (*state >> (16 - bits)) & ((1 << bits) - 1);
}

static inline int round2(const int x, const int shift) {
    return (x + ((1 << shift) >> 1)) >> shift;
}

static void generate_grain_y(const Dav1dPicture *const in,
                             entry buf[GRAIN_HEIGHT][GRAIN_WIDTH])
{
76
    const Dav1dFilmGrainData *data = &in->frame_hdr->film_grain.data;
77
    unsigned seed = data->seed;
78 79 80
    const int shift = 12 - in->p.bpc + data->grain_scale_shift;
    const int grain_ctr = 128 << (in->p.bpc - 8);
    const int grain_min = -grain_ctr, grain_max = grain_ctr - 1;
81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104

    for (int y = 0; y < GRAIN_HEIGHT; y++) {
        for (int x = 0; x < GRAIN_WIDTH; x++) {
            const int value = get_random_number(11, &seed);
            buf[y][x] = round2(dav1d_gaussian_sequence[ value ], shift);
        }
    }

    const int ar_pad = 3;
    const int ar_lag = data->ar_coeff_lag;

    for (int y = ar_pad; y < GRAIN_HEIGHT; y++) {
        for (int x = ar_pad; x < GRAIN_WIDTH - ar_pad; x++) {
            const int8_t *coeff = data->ar_coeffs_y;
            int sum = 0;
            for (int dy = -ar_lag; dy <= 0; dy++) {
                for (int dx = -ar_lag; dx <= ar_lag; dx++) {
                    if (!dx && !dy)
                        break;
                    sum += *(coeff++) * buf[y + dy][x + dx];
                }
            }

            int grain = buf[y][x] + round2(sum, data->ar_coeff_shift);
105
            buf[y][x] = iclip(grain, grain_min, grain_max);
106 107 108 109 110 111 112 113
        }
    }
}

static void generate_grain_uv(const Dav1dPicture *const in, int uv,
                              entry buf[GRAIN_HEIGHT][GRAIN_WIDTH],
                              entry buf_y[GRAIN_HEIGHT][GRAIN_WIDTH])
{
114
    const Dav1dFilmGrainData *data = &in->frame_hdr->film_grain.data;
115
    unsigned seed = data->seed ^ (uv ? 0x49d8 : 0xb524);
116 117 118
    const int shift = 12 - in->p.bpc + data->grain_scale_shift;
    const int grain_ctr = 128 << (in->p.bpc - 8);
    const int grain_min = -grain_ctr, grain_max = grain_ctr - 1;
119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164

    const int subx = in->p.layout != DAV1D_PIXEL_LAYOUT_I444;
    const int suby = in->p.layout == DAV1D_PIXEL_LAYOUT_I420;

    const int chromaW = subx ? SUB_GRAIN_WIDTH  : GRAIN_WIDTH;
    const int chromaH = suby ? SUB_GRAIN_HEIGHT : GRAIN_HEIGHT;

    for (int y = 0; y < chromaH; y++) {
        for (int x = 0; x < chromaW; x++) {
            const int value = get_random_number(11, &seed);
            buf[y][x] = round2(dav1d_gaussian_sequence[ value ], shift);
        }
    }

    const int ar_pad = 3;
    const int ar_lag = data->ar_coeff_lag;

    for (int y = ar_pad; y < chromaH; y++) {
        for (int x = ar_pad; x < chromaW - ar_pad; x++) {
            const int8_t *coeff = data->ar_coeffs_uv[uv];
            int sum = 0;
            for (int dy = -ar_lag; dy <= 0; dy++) {
                for (int dx = -ar_lag; dx <= ar_lag; dx++) {
                    // For the final (current) pixel, we need to add in the
                    // contribution from the luma grain texture
                    if (!dx && !dy) {
                        if (!data->num_y_points)
                            break;
                        int luma = 0;
                        const int lumaX = ((x - ar_pad) << subx) + ar_pad;
                        const int lumaY = ((y - ar_pad) << suby) + ar_pad;
                        for (int i = 0; i <= suby; i++) {
                            for (int j = 0; j <= subx; j++) {
                                luma += buf_y[lumaY + i][lumaX + j];
                            }
                        }
                        luma = round2(luma, subx + suby);
                        sum += luma * (*coeff);
                        break;
                    }

                    sum += *(coeff++) * buf[y + dy][x + dx];
                }
            }

            const int grain = buf[y][x] + round2(sum, data->ar_coeff_shift);
165
            buf[y][x] = iclip(grain, grain_min, grain_max);
166 167 168 169
        }
    }
}

170 171
static void generate_scaling(const int bitdepth,
                             const uint8_t points[][2], int num,
172 173
                             uint8_t scaling[SCALING_SIZE])
{
174 175
    const int shift_x = bitdepth - 8;
    const int scaling_size = 1 << bitdepth;
176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196

    // Fill up the preceding entries with the initial value
    for (int i = 0; i < points[0][0] << shift_x; i++)
        scaling[i] = points[0][1];

    // Linearly interpolate the values in the middle
    for (int i = 0; i < num - 1; i++) {
        const int bx = points[i][0] << shift_x;
        const int by = points[i][1];
        const int ex = points[i+1][0] << shift_x;
        const int ey = points[i+1][1];
        const int dx = ex - bx;
        const int dy = ey - by;
        const int delta = dy * ((0xFFFF + (dx >> 1))) / dx;
        for (int x = 0; x < dx; x++) {
            const int v = by + ((x * delta + 0x8000) >> 16);
            scaling[bx + x] = v;
        }
    }

    // Fill up the remaining entries with the final value
197
    for (int i = points[num - 1][0] << shift_x; i < scaling_size; i++)
198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217
        scaling[i] = points[num - 1][1];
}

// samples from the correct block of a grain LUT, while taking into account the
// offsets provided by the offsets cache
static inline entry sample_lut(entry grain_lut[GRAIN_HEIGHT][GRAIN_WIDTH],
                               int offsets[2][2], int subx, int suby,
                               int bx, int by, int x, int y)
{
    const int randval = offsets[bx][by];
    const int offx = 3 + (2 >> subx) * (3 + (randval >> 4));
    const int offy = 3 + (2 >> suby) * (3 + (randval & 0xF));
    return grain_lut[offy + y + (BLOCK_SIZE >> suby) * by]
                    [offx + x + (BLOCK_SIZE >> subx) * bx];
}

static void apply_to_row_y(Dav1dPicture *const out, const Dav1dPicture *const in,
                           entry grain_lut[GRAIN_HEIGHT][GRAIN_WIDTH],
                           uint8_t scaling[SCALING_SIZE], int row_num)
{
218
    const Dav1dFilmGrainData *const data = &out->frame_hdr->film_grain.data;
219
    const int rows = 1 + (data->overlap_flag && row_num > 0);
220 221 222
    const int bitdepth_min_8 = in->p.bpc - 8;
    const int grain_ctr = 128 << bitdepth_min_8;
    const int grain_min = -grain_ctr, grain_max = grain_ctr - 1;
223 224 225

    int min_value, max_value;
    if (data->clip_to_restricted_range) {
226 227
        min_value = 16 << bitdepth_min_8;
        max_value = 235 << bitdepth_min_8;
228 229
    } else {
        min_value = 0;
230
        max_value = (1U << in->p.bpc) - 1;
231 232 233 234 235 236 237 238 239 240 241 242 243
    }

    // seed[0] contains the current row, seed[1] contains the previous
    unsigned seed[2];
    for (int i = 0; i < rows; i++) {
        seed[i] = data->seed;
        seed[i] ^= (((row_num - i) * 37  + 178) & 0xFF) << 8;
        seed[i] ^= (((row_num - i) * 173 + 105) & 0xFF);
    }

    const ptrdiff_t stride = out->stride[0];
    assert(stride % (BLOCK_SIZE * sizeof(pixel)) == 0);
    assert(stride == in->stride[0]);
244 245
    pixel *const src_row = (pixel *)  in->data[0] + PXSTRIDE(stride) * row_num * BLOCK_SIZE;
    pixel *const dst_row = (pixel *) out->data[0] + PXSTRIDE(stride) * row_num * BLOCK_SIZE;
246 247 248 249

    int offsets[2 /* col offset */][2 /* row offset */];

    // process this row in BLOCK_SIZE^2 blocks
250
    const int bh = imin(out->p.h - row_num * BLOCK_SIZE, BLOCK_SIZE);
251
    for (int bx = 0; bx < out->p.w; bx += BLOCK_SIZE) {
252 253
        const int bw = imin(BLOCK_SIZE, out->p.w - bx);

254 255 256 257 258 259 260 261 262 263 264
        if (data->overlap_flag && bx) {
            // shift previous offsets left
            for (int i = 0; i < rows; i++)
                offsets[1][i] = offsets[0][i];
        }

        // update current offsets
        for (int i = 0; i < rows; i++)
            offsets[0][i] = get_random_number(8, &seed[i]);

        // x/y block offsets to compensate for overlapped regions
265 266
        const int ystart = data->overlap_flag && row_num ? imin(2, bh) : 0;
        const int xstart = data->overlap_flag && bx      ? imin(2, bw) : 0;
267 268 269 270

        static const int w[2][2] = { { 27, 17 }, { 17, 27 } };

#define add_noise_y(x, y, grain)                                                \
271 272
            pixel *src = src_row + (y) * PXSTRIDE(stride) + (bx + (x));         \
            pixel *dst = dst_row + (y) * PXSTRIDE(stride) + (bx + (x));         \
273 274 275
            int noise = round2(scaling[ *src ] * (grain), data->scaling_shift); \
            *dst = iclip(*src + noise, min_value, max_value);

276
        for (int y = ystart; y < bh; y++) {
277
            // Non-overlapped image region (straightforward)
278
            for (int x = xstart; x < bw; x++) {
279 280 281 282 283 284 285 286 287
                int grain = sample_lut(grain_lut, offsets, 0, 0, 0, 0, x, y);
                add_noise_y(x, y, grain);
            }

            // Special case for overlapped column
            for (int x = 0; x < xstart; x++) {
                int grain = sample_lut(grain_lut, offsets, 0, 0, 0, 0, x, y);
                int old   = sample_lut(grain_lut, offsets, 0, 0, 1, 0, x, y);
                grain = round2(old * w[x][0] + grain * w[x][1], 5);
288
                grain = iclip(grain, grain_min, grain_max);
289 290 291 292 293 294
                add_noise_y(x, y, grain);
            }
        }

        for (int y = 0; y < ystart; y++) {
            // Special case for overlapped row (sans corner)
295
            for (int x = xstart; x < bw; x++) {
296 297 298
                int grain = sample_lut(grain_lut, offsets, 0, 0, 0, 0, x, y);
                int old   = sample_lut(grain_lut, offsets, 0, 0, 0, 1, x, y);
                grain = round2(old * w[y][0] + grain * w[y][1], 5);
299
                grain = iclip(grain, grain_min, grain_max);
300 301 302 303 304 305 306 307 308
                add_noise_y(x, y, grain);
            }

            // Special case for doubly-overlapped corner
            for (int x = 0; x < xstart; x++) {
                // Blend the top pixel with the top left block
                int top = sample_lut(grain_lut, offsets, 0, 0, 0, 1, x, y);
                int old = sample_lut(grain_lut, offsets, 0, 0, 1, 1, x, y);
                top = round2(old * w[x][0] + top * w[x][1], 5);
309
                top = iclip(top, grain_min, grain_max);
310 311 312 313 314

                // Blend the current pixel with the left block
                int grain = sample_lut(grain_lut, offsets, 0, 0, 0, 0, x, y);
                old = sample_lut(grain_lut, offsets, 0, 0, 1, 0, x, y);
                grain = round2(old * w[x][0] + grain * w[x][1], 5);
315
                grain = iclip(grain, grain_min, grain_max);
316 317 318

                // Mix the row rows together and apply grain
                grain = round2(top * w[y][0] + grain * w[y][1], 5);
319
                grain = iclip(grain, grain_min, grain_max);
320 321 322 323 324 325 326 327 328 329
                add_noise_y(x, y, grain);
            }
        }
    }
}

static void apply_to_row_uv(Dav1dPicture *const out, const Dav1dPicture *const in,
                            entry grain_lut[GRAIN_HEIGHT][GRAIN_WIDTH],
                            uint8_t scaling[SCALING_SIZE], int uv, int row_num)
{
330
    const Dav1dFilmGrainData *const data = &out->frame_hdr->film_grain.data;
331
    const int rows = 1 + (data->overlap_flag && row_num > 0);
332 333 334 335
    const int bitdepth_max = (1 << in->p.bpc) - 1;
    const int bitdepth_min_8 = in->p.bpc - 8;
    const int grain_ctr = 128 << bitdepth_min_8;
    const int grain_min = -grain_ctr, grain_max = grain_ctr - 1;
336 337 338

    int min_value, max_value;
    if (data->clip_to_restricted_range) {
339
        min_value = 16 << bitdepth_min_8;
340
        if (out->seq_hdr->mtrx == DAV1D_MC_IDENTITY) {
341
            max_value = 235 << bitdepth_min_8;
342
        } else {
343
            max_value = 240 << bitdepth_min_8;
344 345 346
        }
    } else {
        min_value = 0;
347
        max_value = bitdepth_max;
348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365
    }

    const int sx = in->p.layout != DAV1D_PIXEL_LAYOUT_I444;
    const int sy = in->p.layout == DAV1D_PIXEL_LAYOUT_I420;

    // seed[0] contains the current row, seed[1] contains the previous
    unsigned seed[2];
    for (int i = 0; i < rows; i++) {
        seed[i] = data->seed;
        seed[i] ^= (((row_num - i) * 37  + 178) & 0xFF) << 8;
        seed[i] ^= (((row_num - i) * 173 + 105) & 0xFF);
    }

    const ptrdiff_t stride = out->stride[1];
    assert(stride % (BLOCK_SIZE * sizeof(pixel)) == 0);
    assert(stride == in->stride[1]);

    const int by = row_num * (BLOCK_SIZE >> sy);
366 367 368
    pixel *const dst_row = (pixel *) out->data[1 + uv] + PXSTRIDE(stride) * by;
    pixel *const src_row = (pixel *)  in->data[1 + uv] + PXSTRIDE(stride) * by;
    pixel *const luma_row = (pixel *) out->data[0] + PXSTRIDE(out->stride[0]) * row_num * BLOCK_SIZE;
369 370 371 372

    int offsets[2 /* col offset */][2 /* row offset */];

    // process this row in BLOCK_SIZE^2 blocks (subsampled)
373
    const int bh = (imin(out->p.h - row_num * BLOCK_SIZE, BLOCK_SIZE) + sy) >> sy;
374
    for (int bx = 0; bx < (out->p.w + sx) >> sx; bx += BLOCK_SIZE >> sx) {
375
        const int bw = (imin(BLOCK_SIZE, out->p.w - (bx << sx)) + sx) >> sx;
376 377 378 379 380 381 382 383 384 385 386
        if (data->overlap_flag && bx) {
            // shift previous offsets left
            for (int i = 0; i < rows; i++)
                offsets[1][i] = offsets[0][i];
        }

        // update current offsets
        for (int i = 0; i < rows; i++)
            offsets[0][i] = get_random_number(8, &seed[i]);

        // x/y block offsets to compensate for overlapped regions
387 388
        const int ystart = data->overlap_flag && row_num ? imin(2 >> sy, bh) : 0;
        const int xstart = data->overlap_flag && bx      ? imin(2 >> sx, bw) : 0;
389 390 391 392 393 394 395 396 397

        static const int w[2 /* sub */][2 /* off */][2] = {
            { { 27, 17 }, { 17, 27 } },
            { { 23, 22 } },
        };

#define add_noise_uv(x, y, grain)                                               \
            const int lx = (bx + x) << sx;                                      \
            const int ly = y << sy;                                             \
398
            pixel *luma = luma_row + ly * PXSTRIDE(out->stride[0]) + lx;        \
399 400 401 402
            pixel avg = luma[0];                                                \
            if (sx && lx + 1 < out->p.w)                                        \
                avg = (avg + luma[1] + 1) >> 1;                                 \
                                                                                \
403 404
            pixel *src = src_row + (y) * PXSTRIDE(stride) + (bx + (x));         \
            pixel *dst = dst_row + (y) * PXSTRIDE(stride) + (bx + (x));         \
405 406 407 408 409
            int val = avg;                                                      \
            if (!data->chroma_scaling_from_luma) {                              \
                int combined = avg * data->uv_luma_mult[uv] +                   \
                               *src * data->uv_mult[uv];                        \
                val = iclip_pixel( (combined >> 6) +                            \
410
                                   (data->uv_offset[uv] * (1 << bitdepth_min_8)) );   \
411 412 413 414 415
            }                                                                   \
                                                                                \
            int noise = round2(scaling[ val ] * (grain), data->scaling_shift);  \
            *dst = iclip(*src + noise, min_value, max_value);

416
        for (int y = ystart; y < bh; y++) {
417
            // Non-overlapped image region (straightforward)
418
            for (int x = xstart; x < bw; x++) {
419 420 421 422 423 424 425 426 427
                int grain = sample_lut(grain_lut, offsets, sx, sy, 0, 0, x, y);
                add_noise_uv(x, y, grain);
            }

            // Special case for overlapped column
            for (int x = 0; x < xstart; x++) {
                int grain = sample_lut(grain_lut, offsets, sx, sy, 0, 0, x, y);
                int old   = sample_lut(grain_lut, offsets, sx, sy, 1, 0, x, y);
                grain = (old * w[sx][x][0] + grain * w[sx][x][1] + 16) >> 5;
428
                grain = iclip(grain, grain_min, grain_max);
429 430 431 432 433 434
                add_noise_uv(x, y, grain);
            }
        }

        for (int y = 0; y < ystart; y++) {
            // Special case for overlapped row (sans corner)
435
            for (int x = xstart; x < bw; x++) {
436 437 438
                int grain = sample_lut(grain_lut, offsets, sx, sy, 0, 0, x, y);
                int old   = sample_lut(grain_lut, offsets, sx, sy, 0, 1, x, y);
                grain = (old * w[sy][y][0] + grain * w[sy][y][1] + 16) >> 5;
439
                grain = iclip(grain, grain_min, grain_max);
440 441 442 443 444 445 446 447 448
                add_noise_uv(x, y, grain);
            }

            // Special case for doubly-overlapped corner
            for (int x = 0; x < xstart; x++) {
                // Blend the top pixel with the top left block
                int top = sample_lut(grain_lut, offsets, sx, sy, 0, 1, x, y);
                int old = sample_lut(grain_lut, offsets, sx, sy, 1, 1, x, y);
                top = (old * w[sx][x][0] + top * w[sx][x][1] + 16) >> 5;
449
                top = iclip(top, grain_min, grain_max);
450 451 452 453 454

                // Blend the current pixel with the left block
                int grain = sample_lut(grain_lut, offsets, sx, sy, 0, 0, x, y);
                old = sample_lut(grain_lut, offsets, sx, sy, 1, 0, x, y);
                grain = (old * w[sx][x][0] + grain * w[sx][x][1] + 16) >> 5;
455
                grain = iclip(grain, grain_min, grain_max);
456 457 458

                // Mix the row rows together and apply to image
                grain = (top * w[sy][y][0] + grain * w[sy][y][1] + 16) >> 5;
459
                grain = iclip(grain, grain_min, grain_max);
460 461 462 463 464 465 466 467 468
                add_noise_uv(x, y, grain);
            }
        }
    }
}

void bitfn(dav1d_apply_grain)(Dav1dPicture *const out,
                              const Dav1dPicture *const in)
{
469
    const Dav1dFilmGrainData *const data = &out->frame_hdr->film_grain.data;
470 471 472 473 474 475 476 477 478 479 480 481 482

    entry grain_lut[3][GRAIN_HEIGHT][GRAIN_WIDTH];
    uint8_t scaling[3][SCALING_SIZE];

    // Generate grain LUTs as needed
    generate_grain_y(out, grain_lut[0]); // always needed
    if (data->num_uv_points[0] || data->chroma_scaling_from_luma)
        generate_grain_uv(out, 0, grain_lut[1], grain_lut[0]);
    if (data->num_uv_points[1] || data->chroma_scaling_from_luma)
        generate_grain_uv(out, 1, grain_lut[2], grain_lut[0]);

    // Generate scaling LUTs as needed
    if (data->num_y_points)
483
        generate_scaling(in->p.bpc, data->y_points, data->num_y_points, scaling[0]);
484
    if (data->num_uv_points[0])
485
        generate_scaling(in->p.bpc, data->uv_points[0], data->num_uv_points[0], scaling[1]);
486
    if (data->num_uv_points[1])
487
        generate_scaling(in->p.bpc, data->uv_points[1], data->num_uv_points[1], scaling[2]);
488 489 490 491 492 493 494 495

    // Copy over the non-modified planes
    // TODO: eliminate in favor of per-plane refs
    if (!data->num_y_points) {
        assert(out->stride[0] == in->stride[0]);
        memcpy(out->data[0], in->data[0], out->p.h * out->stride[0]);
    }

496 497 498 499 500 501 502 503
    if (in->p.layout != DAV1D_PIXEL_LAYOUT_I400) {
        for (int i = 0; i < 2; i++) {
            if (!data->num_uv_points[i] && !data->chroma_scaling_from_luma) {
                const int suby = in->p.layout == DAV1D_PIXEL_LAYOUT_I420;
                assert(out->stride[1] == in->stride[1]);
                memcpy(out->data[1+i], in->data[1+i],
                       (out->p.h >> suby) * out->stride[1]);
            }
504 505
        }
    }
506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522

    // Synthesize grain for the affected planes
    int rows = (out->p.h + 31) >> 5;
    for (int row = 0; row < rows; row++) {
        if (data->num_y_points)
            apply_to_row_y(out, in, grain_lut[0], scaling[0], row);

        if (data->chroma_scaling_from_luma) {
            apply_to_row_uv(out, in, grain_lut[1], scaling[0], 0, row);
            apply_to_row_uv(out, in, grain_lut[2], scaling[0], 1, row);
        } else {
            if (data->num_uv_points[0])
                apply_to_row_uv(out, in, grain_lut[1], scaling[1], 0, row);
            if (data->num_uv_points[1])
                apply_to_row_uv(out, in, grain_lut[2], scaling[2], 1, row);
        }
    }
523
}