/*****************************************************************************/ // Copyright 2008 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. /*****************************************************************************/ /* $Id: //mondo/dng_sdk_1_4/dng_sdk/source/dng_bad_pixels.cpp#1 $ */ /* $DateTime: 2012/05/30 13:28:51 $ */ /* $Change: 832332 $ */ /* $Author: tknoll $ */ /*****************************************************************************/ #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 /*****************************************************************************/ 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 &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 (" Pixel [%u]: v=%d, h=%d\n", (unsigned) index, (int) fList->Point (index).v, (int) fList->Point (index).h); } if (pCount > gDumpLineLimit) { printf (" ... %u bad pixels skipped\n", (unsigned) (pCount - gDumpLineLimit)); } printf ("Bad Rects: %u\n", (unsigned) rCount); for (index = 0; index < rCount && index < gDumpLineLimit; index++) { printf (" Rect [%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 (" ... %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; } 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]; sprintf (s, "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; } 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"); } #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); } /*****************************************************************************/