/*****************************************************************************/
// Copyright 2008-2019 Adobe Systems Incorporated
// All Rights Reserved.
//
// NOTICE:  Adobe permits you to use, modify, and distribute this file in
// accordance with the terms of the Adobe license agreement accompanying it.
/*****************************************************************************/

#include "dng_bad_pixels.h"

#include "dng_filter_task.h"
#include "dng_globals.h"
#include "dng_host.h"
#include "dng_image.h"
#include "dng_negative.h"
#include "dng_safe_arithmetic.h"

#include <algorithm>

/*****************************************************************************/

dng_opcode_FixBadPixelsConstant::dng_opcode_FixBadPixelsConstant
                                 (uint32 constant,
                                  uint32 bayerPhase)

    :   dng_filter_opcode (dngOpcode_FixBadPixelsConstant,
                           dngVersion_1_3_0_0,
                           0)

    ,   fConstant   (constant)
    ,   fBayerPhase (bayerPhase)

    {

    }

/*****************************************************************************/

dng_opcode_FixBadPixelsConstant::dng_opcode_FixBadPixelsConstant
                                 (dng_stream &stream)

    :   dng_filter_opcode (dngOpcode_FixBadPixelsConstant,
                           stream,
                           "FixBadPixelsConstant")

    ,   fConstant   (0)
    ,   fBayerPhase (0)

    {

    if (stream.Get_uint32 () != 8)
        {
        ThrowBadFormat ();
        }

    fConstant   = stream.Get_uint32 ();
    fBayerPhase = stream.Get_uint32 ();

    #if qDNGValidate

    if (gVerbose)
        {

        printf ("Constant: %u\n", (unsigned) fConstant);

        printf ("Bayer Phase: %u\n", (unsigned) fBayerPhase);

        }

    #endif

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsConstant::PutData (dng_stream &stream) const
    {

    stream.Put_uint32 (8);

    stream.Put_uint32 (fConstant  );
    stream.Put_uint32 (fBayerPhase);

    }

/*****************************************************************************/

dng_point dng_opcode_FixBadPixelsConstant::SrcRepeat ()
    {

    return dng_point (2, 2);

    }

/*****************************************************************************/

dng_rect dng_opcode_FixBadPixelsConstant::SrcArea (const dng_rect &dstArea,
                                                   const dng_rect & /* imageBounds */)
    {

    dng_rect srcArea = dstArea;

    srcArea.t -= 2;
    srcArea.l -= 2;

    srcArea.b += 2;
    srcArea.r += 2;

    return srcArea;

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsConstant::Prepare (dng_negative & /* negative */,
                                               uint32 /* threadCount */,
                                               const dng_point & /* tileSize */,
                                               const dng_rect & /* imageBounds */,
                                               uint32 imagePlanes,
                                               uint32 bufferPixelType,
                                               dng_memory_allocator & /* allocator */)
    {

    // This opcode is restricted to single channel images.

    if (imagePlanes != 1)
        {

        ThrowBadFormat ();

        }

    // This opcode is restricted to 16-bit images.

    if (bufferPixelType != ttShort)
        {

        ThrowBadFormat ();

        }

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsConstant::ProcessArea (dng_negative & /* negative */,
                                                   uint32 /* threadIndex */,
                                                   dng_pixel_buffer &srcBuffer,
                                                   dng_pixel_buffer &dstBuffer,
                                                   const dng_rect &dstArea,
                                                   const dng_rect & /* imageBounds */)
    {

    dstBuffer.CopyArea (srcBuffer,
                        dstArea,
                        0,
                        dstBuffer.fPlanes);

    uint16 badPixel = (uint16) fConstant;

    for (int32 dstRow = dstArea.t; dstRow < dstArea.b; dstRow++)
        {

        const uint16 *sPtr = srcBuffer.ConstPixel_uint16 (dstRow, dstArea.l, 0);
              uint16 *dPtr = dstBuffer.DirtyPixel_uint16 (dstRow, dstArea.l, 0);

        for (int32 dstCol = dstArea.l; dstCol < dstArea.r; dstCol++)
            {

            if (*sPtr == badPixel)
                {

                uint32 count = 0;
                uint32 total = 0;

                uint16 value;

                if (IsGreen (dstRow, dstCol))   // Green pixel
                    {

                    value = sPtr [-srcBuffer.fRowStep - 1];

                    if (value != badPixel)
                        {
                        count += 1;
                        total += value;
                        }

                    value = sPtr [-srcBuffer.fRowStep + 1];

                    if (value != badPixel)
                        {
                        count += 1;
                        total += value;
                        }

                    value = sPtr [srcBuffer.fRowStep - 1];

                    if (value != badPixel)
                        {
                        count += 1;
                        total += value;
                        }

                    value = sPtr [srcBuffer.fRowStep + 1];

                    if (value != badPixel)
                        {
                        count += 1;
                        total += value;
                        }

                    }

                else    // Red/blue pixel.
                    {

                    value = sPtr [-srcBuffer.fRowStep * 2];

                    if (value != badPixel)
                        {
                        count += 1;
                        total += value;
                        }

                    value = sPtr [srcBuffer.fRowStep * 2];

                    if (value != badPixel)
                        {
                        count += 1;
                        total += value;
                        }

                    value = sPtr [-2];

                    if (value != badPixel)
                        {
                        count += 1;
                        total += value;
                        }

                    value = sPtr [2];

                    if (value != badPixel)
                        {
                        count += 1;
                        total += value;
                        }

                    }

                if (count == 4)     // Most common case.
                    {

                    *dPtr = (uint16) ((total + 2) >> 2);

                    }

                else if (count > 0)
                    {

                    *dPtr = (uint16) ((total + (count >> 1)) / count);

                    }

                }

            sPtr++;
            dPtr++;

            }

        }

    }

/*****************************************************************************/

dng_bad_pixel_list::dng_bad_pixel_list ()

    :   fBadPoints ()
    ,   fBadRects  ()

    {

    }

/*****************************************************************************/

void dng_bad_pixel_list::AddPoint (const dng_point &pt)
    {

    fBadPoints.push_back (pt);

    }

/*****************************************************************************/

void dng_bad_pixel_list::AddRect (const dng_rect &r)
    {

    fBadRects.push_back (r);

    }

/*****************************************************************************/

static bool SortBadPoints (const dng_point &a,
                           const dng_point &b)
    {

    if (a.v < b.v)
        return true;

    if (a.v > b.v)
        return false;

    return a.h < b.h;

    }

/*****************************************************************************/

static bool SortBadRects (const dng_rect &a,
                          const dng_rect &b)
    {

    if (a.t < b.t)
        return true;

    if (a.t > b.t)
        return false;

    if (a.l < b.l)
        return true;

    if (a.l > b.l)
        return false;

    if (a.b < b.b)
        return true;

    if (a.b > b.b)
        return false;

    return a.r < b.r;

    }

/*****************************************************************************/

void dng_bad_pixel_list::Sort ()
    {

    if (PointCount () > 1)
        {

        std::sort (fBadPoints.begin (),
                   fBadPoints.end   (),
                   SortBadPoints);

        }

    if (RectCount () > 1)
        {

        std::sort (fBadRects.begin (),
                   fBadRects.end   (),
                   SortBadRects);

        }

    }

/*****************************************************************************/

bool dng_bad_pixel_list::IsPointIsolated (uint32 index,
                                          uint32 radius) const
    {

    dng_point pt = Point (index);

    // Search backward through bad point list.

    for (int32 j = index - 1; j >= 0; j--)
        {

        const dng_point &pt2 = Point (j);

        if (pt2.v < pt.v - (int32) radius)
            {
            break;
            }

        if (Abs_int32 (pt2.h - pt.h) <= radius)
            {
            return false;
            }

        }

    // Search forward through bad point list.

    for (uint32 k = index + 1; k < PointCount (); k++)
        {

        const dng_point &pt2 = Point (k);

        if (pt2.v > pt.v + (int32) radius)
            {
            break;
            }

        if (Abs_int32 (pt2.h - pt.h) <= radius)
            {
            return false;
            }

        }

    // Search through bad rectangle list.

    dng_rect testRect (pt.v - radius,
                       pt.h - radius,
                       pt.v + radius + 1,
                       pt.h + radius + 1);

    for (uint32 n = 0; n < RectCount (); n++)
        {

        if ((testRect & Rect (n)).NotEmpty ())
            {
            return false;
            }

        }

    // Did not find point anywhere, so bad pixel is isolated.

    return true;

    }

/*****************************************************************************/

bool dng_bad_pixel_list::IsRectIsolated (uint32 index,
                                         uint32 radius) const
    {

    dng_rect testRect = Rect (index);

    testRect.t -= radius;
    testRect.l -= radius;
    testRect.b += radius;
    testRect.r += radius;

    for (uint32 n = 0; n < RectCount (); n++)
        {

        if (n != index)
            {

            if ((testRect & Rect (n)).NotEmpty ())
                {
                return false;
                }

            }

        }

    return true;

    }

/*****************************************************************************/

bool dng_bad_pixel_list::IsPointValid (const dng_point &pt,
                                       const dng_rect &imageBounds,
                                       uint32 index) const
    {

    // The point must be in the image bounds to be valid.

    if (pt.v <  imageBounds.t ||
        pt.h <  imageBounds.l ||
        pt.v >= imageBounds.b ||
        pt.h >= imageBounds.r)
        {
        return false;
        }

    // Only search the bad point list if we have a starting search index.

    if (index != kNoIndex)
        {

        // Search backward through bad point list.

        for (int32 j = index - 1; j >= 0; j--)
            {

            const dng_point &pt2 = Point (j);

            if (pt2.v < pt.v)
                {
                break;
                }

            if (pt2 == pt)
                {
                return false;
                }

            }

        // Search forward through bad point list.

        for (uint32 k = index + 1; k < PointCount (); k++)
            {

            const dng_point &pt2 = Point (k);

            if (pt2.v > pt.v)
                {
                break;
                }

            if (pt2 == pt)
                {
                return false;
                }

            }

        }

    // Search through bad rectangle list.

    for (uint32 n = 0; n < RectCount (); n++)
        {

        const dng_rect &r = Rect (n);

        if (pt.v >= r.t &&
            pt.h >= r.l &&
            pt.v <  r.b &&
            pt.h <  r.r)
            {
            return false;
            }

        }

    // Did not find point anywhere, so pixel is valid.

    return true;

    }

/*****************************************************************************/

dng_opcode_FixBadPixelsList::dng_opcode_FixBadPixelsList
                             (AutoPtr<dng_bad_pixel_list> &list,
                              uint32 bayerPhase)


    :   dng_filter_opcode (dngOpcode_FixBadPixelsList,
                           dngVersion_1_3_0_0,
                           0)

    ,   fList ()

    ,   fBayerPhase (bayerPhase)

    {

    fList.Reset (list.Release ());

    fList->Sort ();

    }

/*****************************************************************************/

dng_opcode_FixBadPixelsList::dng_opcode_FixBadPixelsList (dng_stream &stream)

    :   dng_filter_opcode (dngOpcode_FixBadPixelsList,
                           stream,
                           "FixBadPixelsList")

    ,   fList ()

    ,   fBayerPhase (0)

    {

    uint32 size = stream.Get_uint32 ();

    fBayerPhase = stream.Get_uint32 ();

    uint32 pCount = stream.Get_uint32 ();
    uint32 rCount = stream.Get_uint32 ();

    uint32 expectedSize =
        SafeUint32Add (12,
                       SafeUint32Add (SafeUint32Mult (pCount, 8),
                                      SafeUint32Mult (rCount, 16)));

    if (size != expectedSize)
        {
        ThrowBadFormat ();
        }

    fList.Reset (new dng_bad_pixel_list);

    uint32 index;

    for (index = 0; index < pCount; index++)
        {

        dng_point pt;

        pt.v = stream.Get_int32 ();
        pt.h = stream.Get_int32 ();

        fList->AddPoint (pt);

        }

    for (index = 0; index < rCount; index++)
        {

        dng_rect r;

        r.t = stream.Get_int32 ();
        r.l = stream.Get_int32 ();
        r.b = stream.Get_int32 ();
        r.r = stream.Get_int32 ();

        fList->AddRect (r);

        }

    fList->Sort ();

    #if qDNGValidate

    if (gVerbose)
        {

        printf ("Bayer Phase: %u\n", (unsigned) fBayerPhase);

        printf ("Bad Pixels: %u\n", (unsigned) pCount);

        for (index = 0; index < pCount && index < gDumpLineLimit; index++)
            {
            printf ("\tPixel [%u]: v=%d, h=%d\n",
                    (unsigned) index,
                    (int) fList->Point (index).v,
                    (int) fList->Point (index).h);
            }

        if (pCount > gDumpLineLimit)
            {
            printf ("\t ... %u bad pixels skipped\n", (unsigned) (pCount - gDumpLineLimit));
            }

        printf ("Bad Rects: %u\n", (unsigned) rCount);

        for (index = 0; index < rCount && index < gDumpLineLimit; index++)
            {
            printf ("\tRect [%u]: t=%d, l=%d, b=%d, r=%d\n",
                    (unsigned) index,
                    (int) fList->Rect (index).t,
                    (int) fList->Rect (index).l,
                    (int) fList->Rect (index).b,
                    (int) fList->Rect (index).r);
            }

        if (rCount > gDumpLineLimit)
            {
            printf ("\t ... %u bad rects skipped\n", (unsigned) (rCount - gDumpLineLimit));
            }

        }

    #endif

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsList::PutData (dng_stream &stream) const
    {

    uint32 pCount = fList->PointCount ();
    uint32 rCount = fList->RectCount  ();

    stream.Put_uint32 (12 + pCount * 8 + rCount * 16);

    stream.Put_uint32 (fBayerPhase);

    stream.Put_uint32 (pCount);
    stream.Put_uint32 (rCount);

    uint32 index;

    for (index = 0; index < pCount; index++)
        {

        const dng_point &pt (fList->Point (index));

        stream.Put_int32 (pt.v);
        stream.Put_int32 (pt.h);

        }

    for (index = 0; index < rCount; index++)
        {

        const dng_rect &r (fList->Rect (index));

        stream.Put_int32 (r.t);
        stream.Put_int32 (r.l);
        stream.Put_int32 (r.b);
        stream.Put_int32 (r.r);

        }

    }

/*****************************************************************************/

dng_rect dng_opcode_FixBadPixelsList::SrcArea (const dng_rect &dstArea,
                                               const dng_rect & /* imageBounds */)
    {

    int32 padding = 0;

    if (fList->PointCount ())
        {
        padding += kBadPointPadding;
        }

    if (fList->RectCount ())
        {
        padding += kBadRectPadding;
        }

    dng_rect srcArea = dstArea;

    srcArea.t -= padding;
    srcArea.l -= padding;

    srcArea.b += padding;
    srcArea.r += padding;

    return srcArea;

    }

/*****************************************************************************/

dng_point dng_opcode_FixBadPixelsList::SrcRepeat ()
    {

    return dng_point (2, 2);

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsList::Prepare (dng_negative & /* negative */,
                                           uint32 /* threadCount */,
                                           const dng_point & /* tileSize */,
                                           const dng_rect & /* imageBounds */,
                                           uint32 imagePlanes,
                                           uint32 bufferPixelType,
                                           dng_memory_allocator & /* allocator */)
    {

    // This opcode is restricted to single channel images.

    if (imagePlanes != 1)
        {

        ThrowBadFormat ();

        }

    // This opcode is restricted to 16-bit images.

    if (bufferPixelType != ttShort)
        {

        ThrowBadFormat ();

        }

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsList::FixIsolatedPixel (dng_pixel_buffer &buffer,
                                                    dng_point &badPoint)
    {

    uint16 *p0 = buffer.DirtyPixel_uint16 (badPoint.v - 2, badPoint.h - 2, 0);
    uint16 *p1 = buffer.DirtyPixel_uint16 (badPoint.v - 1, badPoint.h - 2, 0);
    uint16 *p2 = buffer.DirtyPixel_uint16 (badPoint.v    , badPoint.h - 2, 0);
    uint16 *p3 = buffer.DirtyPixel_uint16 (badPoint.v + 1, badPoint.h - 2, 0);
    uint16 *p4 = buffer.DirtyPixel_uint16 (badPoint.v + 2, badPoint.h - 2, 0);

    uint32 est0;
    uint32 est1;
    uint32 est2;
    uint32 est3;

    uint32 grad0;
    uint32 grad1;
    uint32 grad2;
    uint32 grad3;

    if (IsGreen (badPoint.v, badPoint.h))       // Green pixel
        {

        // g00 b01 g02 b03 g04
        // r10 g11 r12 g13 r14
        // g20 b21 g22 b23 g24
        // r30 g31 r32 g33 r34
        // g40 b41 g42 b43 g44

        int32 b01 = p0 [1];
        int32 g02 = p0 [2];
        int32 b03 = p0 [3];

        int32 r10 = p1 [0];
        int32 g11 = p1 [1];
        int32 r12 = p1 [2];
        int32 g13 = p1 [3];
        int32 r14 = p1 [4];

        int32 g20 = p2 [0];
        int32 b21 = p2 [1];
        int32 b23 = p2 [3];
        int32 g24 = p2 [4];

        int32 r30 = p3 [0];
        int32 g31 = p3 [1];
        int32 r32 = p3 [2];
        int32 g33 = p3 [3];
        int32 r34 = p3 [4];

        int32 b41 = p4 [1];
        int32 g42 = p4 [2];
        int32 b43 = p4 [3];

        est0 = g02 + g42;

        grad0 = Abs_int32 (g02 - g42) +
                Abs_int32 (g11 - g31) +
                Abs_int32 (g13 - g33) +
                Abs_int32 (b01 - b21) +
                Abs_int32 (b03 - b23) +
                Abs_int32 (b21 - b41) +
                Abs_int32 (b23 - b43);

        est1 = g11 + g33;

        grad1 = Abs_int32 (g11 - g33) +
                Abs_int32 (g02 - g24) +
                Abs_int32 (g20 - g42) +
                Abs_int32 (b01 - b23) +
                Abs_int32 (r10 - r32) +
                Abs_int32 (r12 - r34) +
                Abs_int32 (b21 - b43);

        est2 = g20 + g24;

        grad2 = Abs_int32 (g20 - g24) +
                Abs_int32 (g11 - g13) +
                Abs_int32 (g31 - g33) +
                Abs_int32 (r10 - r12) +
                Abs_int32 (r30 - r32) +
                Abs_int32 (r12 - r14) +
                Abs_int32 (r32 - r34);

        est3 = g13 + g31;

        grad3 = Abs_int32 (g13 - g31) +
                Abs_int32 (g02 - g20) +
                Abs_int32 (g24 - g42) +
                Abs_int32 (b03 - b21) +
                Abs_int32 (r14 - r32) +
                Abs_int32 (r12 - r30) +
                Abs_int32 (b23 - b41);

        }

    else        // Red/blue pixel
        {

        // b00 g01 b02 g03 b04
        // g10 r11 g12 r13 g14
        // b20 g21 b22 g23 b24
        // g30 r31 g32 r33 g34
        // b40 g41 b42 g43 b44

        int32 b00 = p0 [0];
        int32 g01 = p0 [1];
        int32 b02 = p0 [2];
        int32 g03 = p0 [3];
        int32 b04 = p0 [4];

        int32 g10 = p1 [0];
        int32 r11 = p1 [1];
        int32 g12 = p1 [2];
        int32 r13 = p1 [3];
        int32 g14 = p1 [4];

        int32 b20 = p2 [0];
        int32 g21 = p2 [1];
        int32 g23 = p2 [3];
        int32 b24 = p2 [4];

        int32 g30 = p3 [0];
        int32 r31 = p3 [1];
        int32 g32 = p3 [2];
        int32 r33 = p3 [3];
        int32 g34 = p3 [4];

        int32 b40 = p4 [0];
        int32 g41 = p4 [1];
        int32 b42 = p4 [2];
        int32 g43 = p4 [3];
        int32 b44 = p4 [4];

        est0 = b02 + b42;

        grad0 = Abs_int32 (b02 - b42) +
                Abs_int32 (g12 - g32) +
                Abs_int32 (g01 - g21) +
                Abs_int32 (g21 - g41) +
                Abs_int32 (g03 - g23) +
                Abs_int32 (g23 - g43) +
                Abs_int32 (r11 - r31) +
                Abs_int32 (r13 - r33);

        est1 = b00 + b44;

        grad1 = Abs_int32 (b00 - b44) +
                Abs_int32 (r11 - r33) +
                Abs_int32 (g01 - g23) +
                Abs_int32 (g10 - g32) +
                Abs_int32 (g12 - g34) +
                Abs_int32 (g21 - g43) +
                Abs_int32 (b02 - b24) +
                Abs_int32 (b20 - b42);

        est2 = b20 + b24;

        grad2 = Abs_int32 (b20 - b24) +
                Abs_int32 (g21 - g23) +
                Abs_int32 (g10 - g12) +
                Abs_int32 (g12 - g14) +
                Abs_int32 (g30 - g32) +
                Abs_int32 (g32 - g34) +
                Abs_int32 (r11 - r13) +
                Abs_int32 (r31 - r33);

        est3 = b04 + b40;

        grad3 = Abs_int32 (b04 - b40) +
                Abs_int32 (r13 - r31) +
                Abs_int32 (g03 - g21) +
                Abs_int32 (g14 - g32) +
                Abs_int32 (g12 - g30) +
                Abs_int32 (g23 - g41) +
                Abs_int32 (b02 - b20) +
                Abs_int32 (b24 - b42);

        }

    uint32 minGrad = Min_uint32 (grad0, grad1);

    minGrad = Min_uint32 (minGrad, grad2);
    minGrad = Min_uint32 (minGrad, grad3);

    uint32 limit = (minGrad * 3) >> 1;

    uint32 total = 0;
    uint32 count = 0;

    if (grad0 <= limit)
        {
        total += est0;
        count += 2;
        }

    if (grad1 <= limit)
        {
        total += est1;
        count += 2;
        }

    if (grad2 <= limit)
        {
        total += est2;
        count += 2;
        }

    if (grad3 <= limit)
        {
        total += est3;
        count += 2;
        }

    count = Max_uint32 (count, 1);           // Suppress div-by-zero warning.

    uint32 estimate = (total + (count >> 1)) / count;

    p2 [2] = (uint16) estimate;

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsList::FixClusteredPixel (dng_pixel_buffer &buffer,
                                                     uint32 pointIndex,
                                                     const dng_rect &imageBounds)
    {

    const uint32 kNumSets = 3;
    const uint32 kSetSize = 4;

    static const int32 kOffset [kNumSets] [kSetSize] [2] =
        {
            {
                { -1,  1 },
                { -1, -1 },
                {  1, -1 },
                {  1,  1 }
            },
            {
                { -2,  0 },
                {  2,  0 },
                {  0, -2 },
                {  0,  2 }
            },
            {
                { -2, -2 },
                { -2,  2 },
                {  2, -2 },
                {  2,  2 }
            }
        };

    dng_point badPoint = fList->Point (pointIndex);

    bool isGreen = IsGreen (badPoint.v, badPoint.h);

    uint16 *p = buffer.DirtyPixel_uint16 (badPoint.v, badPoint.h, 0);

    for (uint32 set = 0; set < kNumSets; set++)
        {

        if (!isGreen && (kOffset [set] [0] [0] & 1) == 1)
            {
            continue;
            }

        uint32 total = 0;
        uint32 count = 0;

        for (uint32 entry = 0; entry < kSetSize; entry++)
            {

            dng_point offset (kOffset [set] [entry] [0],
                              kOffset [set] [entry] [1]);

            if (fList->IsPointValid (badPoint + offset,
                                     imageBounds,
                                     pointIndex))
                {

                total += p [offset.v * buffer.fRowStep +
                            offset.h * buffer.fColStep];

                count++;

                }

            }

        if (count)
            {

            uint32 estimate = (total + (count >> 1)) / count;

            p [0] = (uint16) estimate;

            return;

            }

        }

    // Unable to patch bad pixel.  Leave pixel as is.

    #if qDNGValidate

    char s [256];

    snprintf (s,
              256,
              "Unable to repair bad pixel, row %d, column %d",
              (int) badPoint.v,
              (int) badPoint.h);

    ReportWarning (s);

    #endif

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsList::FixSingleColumn (dng_pixel_buffer &buffer,
                                                   const dng_rect &badRect)
    {

    int32 cs = buffer.fColStep;

    for (int32 row = badRect.t; row < badRect.b; row++)
        {

        uint16 *p0 = buffer.DirtyPixel_uint16 (row - 4, badRect.l - 4, 0);
        uint16 *p1 = buffer.DirtyPixel_uint16 (row - 3, badRect.l - 4, 0);
        uint16 *p2 = buffer.DirtyPixel_uint16 (row - 2, badRect.l - 4, 0);
        uint16 *p3 = buffer.DirtyPixel_uint16 (row - 1, badRect.l - 4, 0);
        uint16 *p4 = buffer.DirtyPixel_uint16 (row    , badRect.l - 4, 0);
        uint16 *p5 = buffer.DirtyPixel_uint16 (row + 1, badRect.l - 4, 0);
        uint16 *p6 = buffer.DirtyPixel_uint16 (row + 2, badRect.l - 4, 0);
        uint16 *p7 = buffer.DirtyPixel_uint16 (row + 3, badRect.l - 4, 0);
        uint16 *p8 = buffer.DirtyPixel_uint16 (row + 4, badRect.l - 4, 0);

        uint32 est0;
        uint32 est1;
        uint32 est2;
        uint32 est3;
        uint32 est4;
        uint32 est5;
        uint32 est6;

        uint32 grad0;
        uint32 grad1;
        uint32 grad2;
        uint32 grad3;
        uint32 grad4;
        uint32 grad5;
        uint32 grad6;

        uint32 lower = 0;
        uint32 upper = 0x0FFFF;

        if (IsGreen (row, badRect.l))       // Green pixel
            {

            // g00 b01 g02 b03 g04 b05 g06 b07 g08
            // r10 g11 r12 g13 r14 g15 r16 g17 r18
            // g20 b21 g22 b23 g24 b25 g26 b27 g28
            // r30 g31 r32 g33 r34 g35 r36 g37 r38
            // g40 b41 g42 b43 g44 b45 g46 b47 g48
            // r50 g51 r52 g53 r54 g55 r56 g57 r58
            // g60 b61 g62 b63 g64 b65 g66 b67 g68
            // r70 g71 r72 g73 r74 g75 r76 g77 r78
            // g80 b81 g82 b83 g84 b85 g86 b87 g88

            int32 b03 = p0 [3 * cs];
            int32 b05 = p0 [5 * cs];

            int32 g11 = p1 [1 * cs];
            int32 g13 = p1 [3 * cs];
            int32 g15 = p1 [5 * cs];
            int32 g17 = p1 [7 * cs];

            int32 g22 = p2 [2 * cs];
            int32 b23 = p2 [3 * cs];
            int32 b25 = p2 [5 * cs];
            int32 g26 = p2 [6 * cs];

            int32 r30 = p3 [0 * cs];
            int32 g31 = p3 [1 * cs];
            int32 r32 = p3 [2 * cs];
            int32 g33 = p3 [3 * cs];
            int32 g35 = p3 [5 * cs];
            int32 r36 = p3 [6 * cs];
            int32 g37 = p3 [7 * cs];
            int32 r38 = p3 [8 * cs];

            int32 g40 = p4 [0 * cs];
            int32 g42 = p4 [2 * cs];
            int32 b43 = p4 [3 * cs];
            int32 b45 = p4 [5 * cs];
            int32 g46 = p4 [6 * cs];
            int32 g48 = p4 [8 * cs];

            int32 r50 = p5 [0 * cs];
            int32 g51 = p5 [1 * cs];
            int32 r52 = p5 [2 * cs];
            int32 g53 = p5 [3 * cs];
            int32 g55 = p5 [5 * cs];
            int32 r56 = p5 [6 * cs];
            int32 g57 = p5 [7 * cs];
            int32 r58 = p5 [8 * cs];

            int32 g62 = p6 [2 * cs];
            int32 b63 = p6 [3 * cs];
            int32 b65 = p6 [5 * cs];
            int32 g66 = p6 [6 * cs];

            int32 g71 = p7 [1 * cs];
            int32 g73 = p7 [3 * cs];
            int32 g75 = p7 [5 * cs];
            int32 g77 = p7 [7 * cs];

            int32 b83 = p8 [3 * cs];
            int32 b85 = p8 [5 * cs];

            // In case there is some green split, make an estimate of
            // of the local difference between the greens, and adjust
            // the estimated green values for the difference
            // between the two types of green pixels when estimating
            // across green types.

            int32 split = ((g22 + g62 + g26 + g66) * 4 +
                           (g42 + g46            ) * 8 -
                           (g11 + g13 + g15 + g17)     -
                           (g31 + g33 + g35 + g37) * 3 -
                           (g51 + g53 + g55 + g57) * 3 -
                           (g71 + g73 + g75 + g77) + 16) >> 5;

            est0 = g13 + g75 + split * 2;

            grad0 = Abs_int32 (g13 - g75) +
                    Abs_int32 (g15 - g46) +
                    Abs_int32 (g22 - g53) +
                    Abs_int32 (g35 - g66) +
                    Abs_int32 (g42 - g73) +
                    Abs_int32 (b03 - b65) +
                    Abs_int32 (b23 - b85);

            est1 = g33 + g55 + split * 2;

            grad1 = Abs_int32 (g33 - g55) +
                    Abs_int32 (g22 - g55) +
                    Abs_int32 (g33 - g66) +
                    Abs_int32 (g13 - g35) +
                    Abs_int32 (g53 - g75) +
                    Abs_int32 (b23 - b45) +
                    Abs_int32 (b43 - b65);

            est2 = g31 + g57 + split * 2;

            grad2 = Abs_int32 (g31 - g57) +
                    Abs_int32 (g33 - g46) +
                    Abs_int32 (g35 - g48) +
                    Abs_int32 (g40 - g53) +
                    Abs_int32 (g42 - g55) +
                    Abs_int32 (r30 - r56) +
                    Abs_int32 (r32 - r58);

            est3 = g42 + g46;

            grad3 = Abs_int32 (g42 - g46) * 2 +
                    Abs_int32 (g33 - g35) +
                    Abs_int32 (g53 - g55) +
                    Abs_int32 (b23 - b25) +
                    Abs_int32 (b43 - b45) +
                    Abs_int32 (b63 - b65);

            est4 = g37 + g51 + split * 2;

            grad4 = Abs_int32 (g37 - g51) +
                    Abs_int32 (g35 - g42) +
                    Abs_int32 (g33 - g40) +
                    Abs_int32 (g48 - g55) +
                    Abs_int32 (g46 - g53) +
                    Abs_int32 (r38 - r52) +
                    Abs_int32 (r36 - r50);

            est5 = g35 + g53 + split * 2;

            grad5 = Abs_int32 (g35 - g53) +
                    Abs_int32 (g26 - g53) +
                    Abs_int32 (g35 - g62) +
                    Abs_int32 (g15 - g33) +
                    Abs_int32 (g55 - g73) +
                    Abs_int32 (b25 - b43) +
                    Abs_int32 (b45 - b63);

            est6 = g15 + g73 + split * 2;

            grad6 = Abs_int32 (g15 - g73) +
                    Abs_int32 (g13 - g42) +
                    Abs_int32 (g26 - g55) +
                    Abs_int32 (g33 - g62) +
                    Abs_int32 (g46 - g75) +
                    Abs_int32 (b05 - b63) +
                    Abs_int32 (b25 - b83);

            lower = Min_uint32 (Min_uint32 (g33, g35),
                                Min_uint32 (g53, g55));

            upper = Max_uint32 (Max_uint32 (g33, g35),
                                Max_uint32 (g53, g55));

            lower = Pin_int32 (0, lower + split, 65535);
            upper = Pin_int32 (0, upper + split, 65535);

            }

        else        // Red/blue pixel
            {

            // b00 g01 b02 g03 b04 g05 b06 g07 b08
            // g10 r11 g12 r13 g14 r15 g16 r17 g18
            // b20 g21 b22 g23 b24 g25 b26 g27 b28
            // g30 r31 g32 r33 g34 r35 g36 r37 g38
            // b40 g41 b42 g43 b44 g45 b46 g47 b48
            // g50 r51 g52 r53 g54 r55 g56 r57 g58
            // b60 g61 b62 g63 b64 g65 b66 g67 b68
            // g70 r71 g72 r73 g74 r75 g76 r77 g78
            // b80 g81 b82 g83 b84 g85 b86 g87 b88

            int32 b02 = p0 [2 * cs];
            int32 g03 = p0 [3 * cs];
            int32 g05 = p0 [5 * cs];
            int32 b06 = p0 [6 * cs];

            int32 r13 = p1 [3 * cs];
            int32 r15 = p1 [5 * cs];

            int32 b20 = p2 [0 * cs];
            int32 b22 = p2 [2 * cs];
            int32 g23 = p2 [3 * cs];
            int32 g25 = p2 [5 * cs];
            int32 b26 = p2 [6 * cs];
            int32 b28 = p2 [8 * cs];

            int32 r31 = p3 [1 * cs];
            int32 g32 = p3 [2 * cs];
            int32 r33 = p3 [3 * cs];
            int32 r35 = p3 [5 * cs];
            int32 g36 = p3 [6 * cs];
            int32 r37 = p3 [7 * cs];

            int32 g41 = p4 [1 * cs];
            int32 b42 = p4 [2 * cs];
            int32 g43 = p4 [3 * cs];
            int32 g45 = p4 [5 * cs];
            int32 b46 = p4 [6 * cs];
            int32 g47 = p4 [7 * cs];

            int32 r51 = p5 [1 * cs];
            int32 g52 = p5 [2 * cs];
            int32 r53 = p5 [3 * cs];
            int32 r55 = p5 [5 * cs];
            int32 g56 = p5 [6 * cs];
            int32 r57 = p5 [7 * cs];

            int32 b60 = p6 [0 * cs];
            int32 b62 = p6 [2 * cs];
            int32 g63 = p6 [3 * cs];
            int32 g65 = p6 [5 * cs];
            int32 b66 = p6 [6 * cs];
            int32 b68 = p6 [8 * cs];

            int32 r73 = p7 [3 * cs];
            int32 r75 = p7 [5 * cs];

            int32 b82 = p8 [2 * cs];
            int32 g83 = p8 [3 * cs];
            int32 g85 = p8 [5 * cs];
            int32 b86 = p8 [6 * cs];

            est0 = b02 + b86;

            grad0 = Abs_int32 (b02 - b86) +
                    Abs_int32 (r13 - r55) +
                    Abs_int32 (r33 - r75) +
                    Abs_int32 (g03 - g45) +
                    Abs_int32 (g23 - g65) +
                    Abs_int32 (g43 - g85);

            est1 = b22 + b66;

            grad1 = Abs_int32 (b22 - b66) +
                    Abs_int32 (r13 - r35) +
                    Abs_int32 (r33 - r55) +
                    Abs_int32 (r53 - r75) +
                    Abs_int32 (g23 - g45) +
                    Abs_int32 (g43 - g65);

            est2 = b20 + b68;

            grad2 = Abs_int32 (b20 - b68) +
                    Abs_int32 (r31 - r55) +
                    Abs_int32 (r33 - r57) +
                    Abs_int32 (g23 - g47) +
                    Abs_int32 (g32 - g56) +
                    Abs_int32 (g41 - g65);

            est3 = b42 + b46;

            grad3 = Abs_int32 (b42 - b46) +
                    Abs_int32 (r33 - r35) +
                    Abs_int32 (r53 - r55) +
                    Abs_int32 (g32 - g36) +
                    Abs_int32 (g43 - g43) +
                    Abs_int32 (g52 - g56);

            est4 = b28 + b60;

            grad4 = Abs_int32 (b28 - b60) +
                    Abs_int32 (r37 - r53) +
                    Abs_int32 (r35 - r51) +
                    Abs_int32 (g25 - g41) +
                    Abs_int32 (g36 - g52) +
                    Abs_int32 (g47 - g63);

            est5 = b26 + b62;

            grad5 = Abs_int32 (b26 - b62) +
                    Abs_int32 (r15 - r33) +
                    Abs_int32 (r35 - r53) +
                    Abs_int32 (r55 - r73) +
                    Abs_int32 (g25 - g43) +
                    Abs_int32 (g45 - g63);

            est6 = b06 + b82;

            grad6 = Abs_int32 (b06 - b82) +
                    Abs_int32 (r15 - r53) +
                    Abs_int32 (r35 - r73) +
                    Abs_int32 (g05 - g43) +
                    Abs_int32 (g25 - g63) +
                    Abs_int32 (g45 - g83);

            lower = Min_uint32 (b42, b46);
            upper = Max_uint32 (b42, b46);

            }

        uint32 minGrad = Min_uint32 (grad0, grad1);

        minGrad = Min_uint32 (minGrad, grad2);
        minGrad = Min_uint32 (minGrad, grad3);
        minGrad = Min_uint32 (minGrad, grad4);
        minGrad = Min_uint32 (minGrad, grad5);
        minGrad = Min_uint32 (minGrad, grad6);

        uint32 limit = (minGrad * 3) >> 1;

        uint32 total = 0;
        uint32 count = 0;

        if (grad0 <= limit)
            {
            total += est0;
            count += 2;
            }

        if (grad1 <= limit)
            {
            total += est1;
            count += 2;
            }

        if (grad2 <= limit)
            {
            total += est2;
            count += 2;
            }

        if (grad3 <= limit)
            {
            total += est3;
            count += 2;
            }

        if (grad4 <= limit)
            {
            total += est4;
            count += 2;
            }

        if (grad5 <= limit)
            {
            total += est5;
            count += 2;
            }

        if (grad6 <= limit)
            {
            total += est6;
            count += 2;
            }

        count = Max_uint32 (count, 1);       // Suppress div-by-zero warning.

        uint32 estimate = (total + (count >> 1)) / count;

        p4 [4] = (uint16) Pin_uint32 (lower, estimate, upper);

        }

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsList::FixSingleRow (dng_pixel_buffer &buffer,
                                                const dng_rect &badRect)
    {

    dng_pixel_buffer tBuffer = buffer;

    tBuffer.fArea = Transpose (buffer.fArea);

    tBuffer.fRowStep = buffer.fColStep;
    tBuffer.fColStep = buffer.fRowStep;

    dng_rect tBadRect = Transpose (badRect);

    FixSingleColumn (tBuffer, tBadRect);

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsList::FixClusteredRect (dng_pixel_buffer &buffer,
                                                    const dng_rect &badRect,
                                                    const dng_rect &imageBounds)
    {

    const uint32 kNumSets = 8;
    const uint32 kSetSize = 8;

    static const int32 kOffset [kNumSets] [kSetSize] [2] =
        {
            {
                { -1,  1 },
                { -1, -1 },
                {  1, -1 },
                {  1,  1 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 }
            },
            {
                { -2,  0 },
                {  2,  0 },
                {  0, -2 },
                {  0,  2 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 }
            },
            {
                { -2, -2 },
                { -2,  2 },
                {  2, -2 },
                {  2,  2 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 }
            },
            {
                { -1, -3 },
                { -3, -1 },
                {  1, -3 },
                {  3, -1 },
                { -1,  3 },
                { -3,  1 },
                {  1,  3 },
                {  3,  1 }
            },
            {
                { -4,  0 },
                {  4,  0 },
                {  0, -4 },
                {  0,  4 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 }
            },
            {
                { -3, -3 },
                { -3,  3 },
                {  3, -3 },
                {  3,  3 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 }
            },
            {
                { -2, -4 },
                { -4, -2 },
                {  2, -4 },
                {  4, -2 },
                { -2,  4 },
                { -4,  2 },
                {  2,  4 },
                {  4,  2 }
            },
            {
                { -4, -4 },
                { -4,  4 },
                {  4, -4 },
                {  4,  4 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 },
                {  0,  0 }
            }
        };

    bool didFail = false;

    for (int32 row = badRect.t; row < badRect.b; row++)
        {

        for (int32 col = badRect.l; col < badRect.r; col++)
            {

            uint16 *p = buffer.DirtyPixel_uint16 (row, col, 0);

            bool isGreen = IsGreen (row, col);

            bool didFixPixel = false;

            for (uint32 set = 0; set < kNumSets && !didFixPixel; set++)
                {

                if (!isGreen && (kOffset [set] [0] [0] & 1) == 1)
                    {
                    continue;
                    }

                uint32 total = 0;
                uint32 count = 0;

                for (uint32 entry = 0; entry < kSetSize; entry++)
                    {

                    dng_point offset (kOffset [set] [entry] [0],
                                      kOffset [set] [entry] [1]);

                    if (offset.v == 0 &&
                        offset.h == 0)
                        {
                        break;
                        }

                    if (fList->IsPointValid (dng_point (row, col) + offset,
                                             imageBounds))
                        {

                        total += p [offset.v * buffer.fRowStep +
                                    offset.h * buffer.fColStep];

                        count++;

                        }

                    }

                if (count)
                    {

                    uint32 estimate = (total + (count >> 1)) / count;

                    p [0] = (uint16) estimate;

                    didFixPixel = true;

                    }

                }

            if (!didFixPixel)
                {

                didFail = true;

                }

            }

        }

    #if qDNGValidate

    if (didFail)
        {

        ReportWarning ("Unable to repair bad rectangle");

        }

    #else

    (void) didFail;

    #endif

    }

/*****************************************************************************/

void dng_opcode_FixBadPixelsList::ProcessArea (dng_negative & /* negative */,
                                               uint32 /* threadIndex */,
                                               dng_pixel_buffer &srcBuffer,
                                               dng_pixel_buffer &dstBuffer,
                                               const dng_rect &dstArea,
                                               const dng_rect &imageBounds)
    {

    uint32 pointCount = fList->PointCount ();
    uint32 rectCount  = fList->RectCount  ();

    dng_rect fixArea = dstArea;

    if (rectCount)
        {
        fixArea.t -= kBadRectPadding;
        fixArea.l -= kBadRectPadding;
        fixArea.b += kBadRectPadding;
        fixArea.r += kBadRectPadding;
        }

    bool didFixPoint = false;

    if (pointCount)
        {

        for (uint32 pointIndex = 0; pointIndex < pointCount; pointIndex++)
            {

            dng_point badPoint = fList->Point (pointIndex);

            if (badPoint.v >= fixArea.t &&
                badPoint.h >= fixArea.l &&
                badPoint.v <  fixArea.b &&
                badPoint.h <  fixArea.r)
                {

                bool isIsolated = fList->IsPointIsolated (pointIndex,
                                                          kBadPointPadding);

                if (isIsolated &&
                    badPoint.v >= imageBounds.t + kBadPointPadding &&
                    badPoint.h >= imageBounds.l + kBadPointPadding &&
                    badPoint.v <  imageBounds.b - kBadPointPadding &&
                    badPoint.h <  imageBounds.r - kBadPointPadding)
                    {

                    FixIsolatedPixel (srcBuffer,
                                      badPoint);

                    }

                else
                    {

                    FixClusteredPixel (srcBuffer,
                                       pointIndex,
                                       imageBounds);

                    }

                didFixPoint = true;

                }

            }

        }

    if (rectCount)
        {

        if (didFixPoint)
            {

            srcBuffer.RepeatSubArea (imageBounds,
                                     SrcRepeat ().v,
                                     SrcRepeat ().h);

            }

        for (uint32 rectIndex = 0; rectIndex < rectCount; rectIndex++)
            {

            dng_rect badRect = fList->Rect (rectIndex);

            dng_rect overlap = dstArea & badRect;

            if (overlap.NotEmpty ())
                {

                bool isIsolated = fList->IsRectIsolated (rectIndex,
                                                         kBadRectPadding);

                if (isIsolated &&
                    badRect.r == badRect.l + 1 &&
                    badRect.l >= imageBounds.l + SrcRepeat ().h &&
                    badRect.r <= imageBounds.r - SrcRepeat ().v)
                    {

                    FixSingleColumn (srcBuffer,
                                     overlap);

                    }

                else if (isIsolated &&
                         badRect.b == badRect.t + 1 &&
                         badRect.t >= imageBounds.t + SrcRepeat ().h &&
                         badRect.b <= imageBounds.b - SrcRepeat ().v)
                    {

                    FixSingleRow (srcBuffer,
                                  overlap);

                    }

                else
                    {

                    FixClusteredRect (srcBuffer,
                                      overlap,
                                      imageBounds);

                    }

                }

            }

        }

    dstBuffer.CopyArea (srcBuffer,
                        dstArea,
                        0,
                        dstBuffer.fPlanes);

    }

/*****************************************************************************/
