Add option to skip rects to drawImageLattice()

src/core/SkLatticeIter.cpp

 /*
  * Copyright 2015 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "SkLatticeIter.h"
 #include "SkRect.h"
 
 /**
  *  Divs must be in increasing order with no duplicates.
  */
 static bool valid_divs(const int* divs, int count, int len) {
-    if (count <= 0) {
-        return false;
-    }
-
     int prev = -1;
     for (int i = 0; i < count; i++) {
-        if (prev >= divs[i] || divs[i] > len) {
+        if (prev >= divs[i] || divs[i] >= len) {
             return false;
         }
     }
 
     return true;
 }
 
 bool SkLatticeIter::Valid(int width, int height, const SkCanvas::Lattice& lattice) {
+    bool zeroXDivs = lattice.fXCount <= 0 || (1 == lattice.fXCount && 0 == lattice.fXDivs[0]);
+    bool zeroYDivs = lattice.fYCount <= 0 || (1 == lattice.fYCount && 0 == lattice.fYDivs[0]);
+    if (zeroXDivs && zeroYDivs) {
+        return false;
+    }
+
     return valid_divs(lattice.fXDivs, lattice.fXCount, width) &&
            valid_divs(lattice.fYDivs, lattice.fYCount, height);
 }
 
 /**
  *  Count the number of pixels that are in "scalable" patches.
  */
 static int count_scalable_pixels(const int32_t* divs, int numDivs, bool firstIsScalable,
                                  int length) {
     if (0 == numDivs) {
@@ skipping 57 matching lines @@
     }
 
     src[divCount + 1] = (float) srcLen;
     dst[divCount + 1] = dstStop;
 }
 
 SkLatticeIter::SkLatticeIter(int srcWidth, int srcHeight, const SkCanvas::Lattice& lattice,
                              const SkRect& dst)
 {
     const int* xDivs = lattice.fXDivs;
-    int xCount = lattice.fXCount;
+    const int origXCount = lattice.fXCount;
     const int* yDivs = lattice.fYDivs;
-    int yCount = lattice.fYCount;
+    const int origYCount = lattice.fYCount;
 
     // In the x-dimension, the first rectangle always starts at x = 0 and is "scalable".
     // If xDiv[0] is 0, it indicates that the first rectangle is degenerate, so the
     // first real rectangle "scalable" in the x-direction.
     //
     // The same interpretation applies to the y-dimension.
     //
     // As we move left to right across the image, alternating patches will be "fixed" or
     // "scalable" in the x-direction.  Similarly, as move top to bottom, alternating
     // patches will be "fixed" or "scalable" in the y-direction.
-    SkASSERT(xCount > 0 && yCount > 0);
-    bool xIsScalable = (0 == xDivs[0]);
+    int xCount = origXCount;
+    int yCount = origYCount;
+    bool xIsScalable = (xCount > 0 && 0 == xDivs[0]);
     if (xIsScalable) {
         // Once we've decided that the first patch is "scalable", we don't need the
         // xDiv.  It is always implied that we start at zero.
         xDivs++;
         xCount--;
     }
-    bool yIsScalable = (0 == yDivs[0]);
+    bool yIsScalable = (yCount > 0 && 0 == yDivs[0]);
     if (yIsScalable) {
         // Once we've decided that the first patch is "scalable", we don't need the
         // yDiv.  It is always implied that we start at zero.
         yDivs++;
         yCount--;
     }
 
-    // We never need the final xDiv/yDiv if it is equal to the width/height.  This is implied.
-    if (xCount > 0 && srcWidth == xDivs[xCount - 1]) {
-        xCount--;
-    }
-    if (yCount > 0 && srcHeight == yDivs[yCount - 1]) {
-        yCount--;
-    }
-
     // Count "scalable" and "fixed" pixels in each dimension.
     int xCountScalable = count_scalable_pixels(xDivs, xCount, xIsScalable, srcWidth);
     int xCountFixed = srcWidth - xCountScalable;
     int yCountScalable = count_scalable_pixels(yDivs, yCount, yIsScalable, srcHeight);
     int yCountFixed = srcHeight - yCountScalable;
 
     fSrcX.reset(xCount + 2);
     fDstX.reset(xCount + 2);
     set_points(fDstX.begin(), fSrcX.begin(), xDivs, xCount, xCountFixed, xCountScalable,
                dst.fLeft, dst.fRight, xIsScalable);
 
     fSrcY.reset(yCount + 2);
     fDstY.reset(yCount + 2);
     set_points(fDstY.begin(), fSrcY.begin(), yDivs, yCount, yCountFixed, yCountScalable,
                dst.fTop, dst.fBottom, yIsScalable);
 
     fCurrX = fCurrY = 0;
-    fDone = false;
-    fNumRects = (xCount + 1) * (yCount + 1);
+    fNumRectsInLattice = (xCount + 1) * (yCount + 1);
+    fNumRectsToDraw = fNumRectsInLattice;
+
+    if (lattice.fFlags) {
+        fFlags.push_back_n(fNumRectsInLattice);
+
+        const SkCanvas::Lattice::Flags* flags = lattice.fFlags;
+
+        bool hasPadRow = (yCount != origYCount);
+        bool hasPadCol = (xCount != origXCount);
+        if (hasPadRow) {
+            // The first row of rects are all empty, skip the first row of flags.
+            flags += origXCount + 1;
+        }
+
+        int i = 0;
+        for (int y = 0; y < yCount + 1; y++) {
+            for (int x = 0; x < origXCount + 1; x++) {
+                if (0 == x && hasPadCol) {
+                    // The first column of rects are all empty.  Skip a rect.
+                    flags++;
+                    continue;
+                }
+
+                fFlags[i] = *flags;
+                flags++;
+                i++;
+            }
+        }
+
+        for (int j = 0; j < fFlags.count(); j++) {
+            if (SkCanvas::Lattice::kTransparent_Flags == fFlags[j]) {
+                fNumRectsToDraw--;
+            }
+        }
+    }
 }
 
 bool SkLatticeIter::Valid(int width, int height, const SkIRect& center) {
     return !center.isEmpty() && SkIRect::MakeWH(width, height).contains(center);
 }
 
 SkLatticeIter::SkLatticeIter(int w, int h, const SkIRect& c, const SkRect& dst) {
     SkASSERT(SkIRect::MakeWH(w, h).contains(c));
 
     fSrcX.reset(4);
@@ skipping 25 matching lines @@
         fDstX[1] = fDstX[0] + (fDstX[3] - fDstX[0]) * c.fLeft / (w - c.width());
         fDstX[2] = fDstX[1];
     }
 
     if (fDstY[1] > fDstY[2]) {
         fDstY[1] = fDstY[0] + (fDstY[3] - fDstY[0]) * c.fTop / (h - c.height());
         fDstY[2] = fDstY[1];
     }
 
     fCurrX = fCurrY = 0;
-    fDone = false;
-    fNumRects = 9;
+    fNumRectsInLattice = 9;
+    fNumRectsToDraw = 9;
 }
 
 bool SkLatticeIter::next(SkRect* src, SkRect* dst) {
-    if (fDone) {
+    int currRect = fCurrX + fCurrY * (fSrcX.count() - 1);
+    if (currRect == fNumRectsInLattice) {
         return false;
     }
 
     const int x = fCurrX;
     const int y = fCurrY;
     SkASSERT(x >= 0 && x < fSrcX.count() - 1);
     SkASSERT(y >= 0 && y < fSrcY.count() - 1);
 
-    src->set(fSrcX[x], fSrcY[y], fSrcX[x + 1], fSrcY[y + 1]);
-    dst->set(fDstX[x], fDstY[y], fDstX[x + 1], fDstY[y + 1]);
     if (fSrcX.count() - 1 == ++fCurrX) {
         fCurrX = 0;
         fCurrY += 1;
-        if (fCurrY >= fSrcY.count() - 1) {
-            fDone = true;
-        }
     }
+
+    if (fFlags.count() > 0 && SkToBool(SkCanvas::Lattice::kTransparent_Flags & fFlags[currRect])) {
+        return this->next(src, dst);
+    }
+
+    src->set(fSrcX[x], fSrcY[y], fSrcX[x + 1], fSrcY[y + 1]);
+    dst->set(fDstX[x], fDstY[y], fDstX[x + 1], fDstY[y + 1]);
     return true;
 }
 
 void SkLatticeIter::mapDstScaleTranslate(const SkMatrix& matrix) {
     SkASSERT(matrix.isScaleTranslate());
     SkScalar tx = matrix.getTranslateX();
     SkScalar sx = matrix.getScaleX();
     for (int i = 0; i < fDstX.count(); i++) {
         fDstX[i] = fDstX[i] * sx + tx;
     }
 
     SkScalar ty = matrix.getTranslateY();
     SkScalar sy = matrix.getScaleY();
     for (int i = 0; i < fDstY.count(); i++) {
         fDstY[i] = fDstY[i] * sy + ty;
     }
 }