Home | History | Annotate | Download | only in graphics
      1 /*
      2 **
      3 ** Copyright 2006, The Android Open Source Project
      4 **
      5 ** Licensed under the Apache License, Version 2.0 (the "License");
      6 ** you may not use this file except in compliance with the License.
      7 ** You may obtain a copy of the License at
      8 **
      9 **     http://www.apache.org/licenses/LICENSE-2.0
     10 **
     11 ** Unless required by applicable law or agreed to in writing, software
     12 ** distributed under the License is distributed on an "AS IS" BASIS,
     13 ** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     14 ** See the License for the specific language governing permissions and
     15 ** limitations under the License.
     16 */
     17 
     18 #define LOG_TAG "NinePatch"
     19 #define LOG_NDEBUG 1
     20 
     21 #include <utils/ResourceTypes.h>
     22 #include <utils/Log.h>
     23 
     24 #include "SkBitmap.h"
     25 #include "SkCanvas.h"
     26 #include "SkNinePatch.h"
     27 #include "SkPaint.h"
     28 #include "SkUnPreMultiply.h"
     29 
     30 #define USE_TRACE
     31 
     32 #ifdef USE_TRACE
     33     static bool gTrace;
     34 #endif
     35 
     36 #include "SkColorPriv.h"
     37 
     38 #include <utils/Log.h>
     39 
     40 static bool getColor(const SkBitmap& bitmap, int x, int y, SkColor* c) {
     41     switch (bitmap.getConfig()) {
     42         case SkBitmap::kARGB_8888_Config:
     43             *c = SkUnPreMultiply::PMColorToColor(*bitmap.getAddr32(x, y));
     44             break;
     45         case SkBitmap::kRGB_565_Config:
     46             *c = SkPixel16ToPixel32(*bitmap.getAddr16(x, y));
     47             break;
     48         case SkBitmap::kARGB_4444_Config:
     49             *c = SkUnPreMultiply::PMColorToColor(
     50                                 SkPixel4444ToPixel32(*bitmap.getAddr16(x, y)));
     51             break;
     52         case SkBitmap::kIndex8_Config: {
     53             SkColorTable* ctable = bitmap.getColorTable();
     54             *c = SkUnPreMultiply::PMColorToColor(
     55                                             (*ctable)[*bitmap.getAddr8(x, y)]);
     56             break;
     57         }
     58         default:
     59             return false;
     60     }
     61     return true;
     62 }
     63 
     64 static SkColor modAlpha(SkColor c, int alpha) {
     65     int scale = alpha + (alpha >> 7);
     66     int a = SkColorGetA(c) * scale >> 8;
     67     return SkColorSetA(c, a);
     68 }
     69 
     70 static void drawStretchyPatch(SkCanvas* canvas, SkIRect& src, const SkRect& dst,
     71                               const SkBitmap& bitmap, const SkPaint& paint,
     72                               SkColor initColor, uint32_t colorHint,
     73                               bool hasXfer) {
     74     if (colorHint !=  android::Res_png_9patch::NO_COLOR) {
     75         ((SkPaint*)&paint)->setColor(modAlpha(colorHint, paint.getAlpha()));
     76         canvas->drawRect(dst, paint);
     77         ((SkPaint*)&paint)->setColor(initColor);
     78     } else if (src.width() == 1 && src.height() == 1) {
     79         SkColor c;
     80         if (!getColor(bitmap, src.fLeft, src.fTop, &c)) {
     81             goto SLOW_CASE;
     82         }
     83         if (0 != c || hasXfer) {
     84             SkColor prev = paint.getColor();
     85             ((SkPaint*)&paint)->setColor(c);
     86             canvas->drawRect(dst, paint);
     87             ((SkPaint*)&paint)->setColor(prev);
     88         }
     89     } else {
     90     SLOW_CASE:
     91         canvas->drawBitmapRect(bitmap, &src, dst, &paint);
     92     }
     93 }
     94 
     95 SkScalar calculateStretch(SkScalar boundsLimit, SkScalar startingPoint,
     96                           int srcSpace, int numStrechyPixelsRemaining,
     97                           int numFixedPixelsRemaining) {
     98     SkScalar spaceRemaining = boundsLimit - startingPoint;
     99     SkScalar stretchySpaceRemaining =
    100                 spaceRemaining - SkIntToScalar(numFixedPixelsRemaining);
    101     return SkScalarMulDiv(srcSpace, stretchySpaceRemaining,
    102                           numStrechyPixelsRemaining);
    103 }
    104 
    105 void NinePatch_Draw(SkCanvas* canvas, const SkRect& bounds,
    106                        const SkBitmap& bitmap, const android::Res_png_9patch& chunk,
    107                        const SkPaint* paint, SkRegion** outRegion) {
    108     if (canvas && canvas->quickReject(bounds, SkCanvas::kBW_EdgeType)) {
    109         return;
    110     }
    111 
    112     SkPaint defaultPaint;
    113     if (NULL == paint) {
    114         // matches default dither in NinePatchDrawable.java.
    115         defaultPaint.setDither(true);
    116         paint = &defaultPaint;
    117     }
    118 
    119     // if our canvas is GL, draw this as a mesh, which will be faster than
    120     // in parts (which is faster for raster)
    121     if (canvas && canvas->getViewport(NULL)) {
    122         SkNinePatch::DrawMesh(canvas, bounds, bitmap,
    123                               chunk.xDivs, chunk.numXDivs,
    124                               chunk.yDivs, chunk.numYDivs,
    125                               paint);
    126         return;
    127     }
    128 
    129 #ifdef USE_TRACE
    130     gTrace = true;
    131 #endif
    132 
    133     SkASSERT(canvas || outRegion);
    134 
    135 #ifdef USE_TRACE
    136     if (canvas) {
    137         const SkMatrix& m = canvas->getTotalMatrix();
    138         LOGV("ninepatch [%g %g %g] [%g %g %g]\n",
    139                  SkScalarToFloat(m[0]), SkScalarToFloat(m[1]), SkScalarToFloat(m[2]),
    140                  SkScalarToFloat(m[3]), SkScalarToFloat(m[4]), SkScalarToFloat(m[5]));
    141     }
    142 #endif
    143 
    144 #ifdef USE_TRACE
    145     if (gTrace) {
    146         LOGV("======== ninepatch bounds [%g %g]\n", SkScalarToFloat(bounds.width()), SkScalarToFloat(bounds.height()));
    147         LOGV("======== ninepatch paint bm [%d,%d]\n", bitmap.width(), bitmap.height());
    148         LOGV("======== ninepatch xDivs [%d,%d]\n", chunk.xDivs[0], chunk.xDivs[1]);
    149         LOGV("======== ninepatch yDivs [%d,%d]\n", chunk.yDivs[0], chunk.yDivs[1]);
    150     }
    151 #endif
    152 
    153     if (bounds.isEmpty() ||
    154         bitmap.width() == 0 || bitmap.height() == 0 ||
    155         (paint && paint->getXfermode() == NULL && paint->getAlpha() == 0))
    156     {
    157 #ifdef USE_TRACE
    158         if (gTrace) LOGV("======== abort ninepatch draw\n");
    159 #endif
    160         return;
    161     }
    162 
    163     // should try a quick-reject test before calling lockPixels
    164 
    165     SkAutoLockPixels alp(bitmap);
    166     // after the lock, it is valid to check getPixels()
    167     if (bitmap.getPixels() == NULL)
    168         return;
    169 
    170     const bool hasXfer = paint->getXfermode() != NULL;
    171     SkRect      dst;
    172     SkIRect     src;
    173 
    174     const int32_t x0 = chunk.xDivs[0];
    175     const int32_t y0 = chunk.yDivs[0];
    176     const SkColor initColor = ((SkPaint*)paint)->getColor();
    177     const uint8_t numXDivs = chunk.numXDivs;
    178     const uint8_t numYDivs = chunk.numYDivs;
    179     int i;
    180     int j;
    181     int colorIndex = 0;
    182     uint32_t color;
    183     bool xIsStretchable;
    184     const bool initialXIsStretchable =  (x0 == 0);
    185     bool yIsStretchable = (y0 == 0);
    186     const int bitmapWidth = bitmap.width();
    187     const int bitmapHeight = bitmap.height();
    188 
    189     SkScalar* dstRights = (SkScalar*) alloca((numXDivs + 1) * sizeof(SkScalar));
    190     bool dstRightsHaveBeenCached = false;
    191 
    192     int numStretchyXPixelsRemaining = 0;
    193     for (i = 0; i < numXDivs; i += 2) {
    194         numStretchyXPixelsRemaining += chunk.xDivs[i + 1] - chunk.xDivs[i];
    195     }
    196     int numFixedXPixelsRemaining = bitmapWidth - numStretchyXPixelsRemaining;
    197     int numStretchyYPixelsRemaining = 0;
    198     for (i = 0; i < numYDivs; i += 2) {
    199         numStretchyYPixelsRemaining += chunk.yDivs[i + 1] - chunk.yDivs[i];
    200     }
    201     int numFixedYPixelsRemaining = bitmapHeight - numStretchyYPixelsRemaining;
    202 
    203 #ifdef USE_TRACE
    204     LOGV("NinePatch [%d %d] bounds [%g %g %g %g] divs [%d %d]\n",
    205              bitmap.width(), bitmap.height(),
    206              SkScalarToFloat(bounds.fLeft), SkScalarToFloat(bounds.fTop),
    207              SkScalarToFloat(bounds.width()), SkScalarToFloat(bounds.height()),
    208              numXDivs, numYDivs);
    209 #endif
    210 
    211     src.fTop = 0;
    212     dst.fTop = bounds.fTop;
    213     // The first row always starts with the top being at y=0 and the bottom
    214     // being either yDivs[1] (if yDivs[0]=0) of yDivs[0].  In the former case
    215     // the first row is stretchable along the Y axis, otherwise it is fixed.
    216     // The last row always ends with the bottom being bitmap.height and the top
    217     // being either yDivs[numYDivs-2] (if yDivs[numYDivs-1]=bitmap.height) or
    218     // yDivs[numYDivs-1]. In the former case the last row is stretchable along
    219     // the Y axis, otherwise it is fixed.
    220     //
    221     // The first and last columns are similarly treated with respect to the X
    222     // axis.
    223     //
    224     // The above is to help explain some of the special casing that goes on the
    225     // code below.
    226 
    227     // The initial yDiv and whether the first row is considered stretchable or
    228     // not depends on whether yDiv[0] was zero or not.
    229     for (j = yIsStretchable ? 1 : 0;
    230           j <= numYDivs && src.fTop < bitmapHeight;
    231           j++, yIsStretchable = !yIsStretchable) {
    232         src.fLeft = 0;
    233         dst.fLeft = bounds.fLeft;
    234         if (j == numYDivs) {
    235             src.fBottom = bitmapHeight;
    236             dst.fBottom = bounds.fBottom;
    237         } else {
    238             src.fBottom = chunk.yDivs[j];
    239             const int srcYSize = src.fBottom - src.fTop;
    240             if (yIsStretchable) {
    241                 dst.fBottom = dst.fTop + calculateStretch(bounds.fBottom, dst.fTop,
    242                                                           srcYSize,
    243                                                           numStretchyYPixelsRemaining,
    244                                                           numFixedYPixelsRemaining);
    245                 numStretchyYPixelsRemaining -= srcYSize;
    246             } else {
    247                 dst.fBottom = dst.fTop + SkIntToScalar(srcYSize);
    248                 numFixedYPixelsRemaining -= srcYSize;
    249             }
    250         }
    251 
    252         xIsStretchable = initialXIsStretchable;
    253         // The initial xDiv and whether the first column is considered
    254         // stretchable or not depends on whether xDiv[0] was zero or not.
    255         for (i = xIsStretchable ? 1 : 0;
    256               i <= numXDivs && src.fLeft < bitmapWidth;
    257               i++, xIsStretchable = !xIsStretchable) {
    258             color = chunk.colors[colorIndex++];
    259             if (i == numXDivs) {
    260                 src.fRight = bitmapWidth;
    261                 dst.fRight = bounds.fRight;
    262             } else {
    263                 src.fRight = chunk.xDivs[i];
    264                 if (dstRightsHaveBeenCached) {
    265                     dst.fRight = dstRights[i];
    266                 } else {
    267                     const int srcXSize = src.fRight - src.fLeft;
    268                     if (xIsStretchable) {
    269                         dst.fRight = dst.fLeft + calculateStretch(bounds.fRight, dst.fLeft,
    270                                                                   srcXSize,
    271                                                                   numStretchyXPixelsRemaining,
    272                                                                   numFixedXPixelsRemaining);
    273                         numStretchyXPixelsRemaining -= srcXSize;
    274                     } else {
    275                         dst.fRight = dst.fLeft + SkIntToScalar(srcXSize);
    276                         numFixedXPixelsRemaining -= srcXSize;
    277                     }
    278                     dstRights[i] = dst.fRight;
    279                 }
    280             }
    281             // If this horizontal patch is too small to be displayed, leave
    282             // the destination left edge where it is and go on to the next patch
    283             // in the source.
    284             if (src.fLeft >= src.fRight) {
    285                 src.fLeft = src.fRight;
    286                 continue;
    287             }
    288             // Make sure that we actually have room to draw any bits
    289             if (dst.fRight <= dst.fLeft || dst.fBottom <= dst.fTop) {
    290                 goto nextDiv;
    291             }
    292             // If this patch is transparent, skip and don't draw.
    293             if (color == android::Res_png_9patch::TRANSPARENT_COLOR && !hasXfer) {
    294                 if (outRegion) {
    295                     if (*outRegion == NULL) {
    296                         *outRegion = new SkRegion();
    297                     }
    298                     SkIRect idst;
    299                     dst.round(&idst);
    300                     //LOGI("Adding trans rect: (%d,%d)-(%d,%d)\n",
    301                     //     idst.fLeft, idst.fTop, idst.fRight, idst.fBottom);
    302                     (*outRegion)->op(idst, SkRegion::kUnion_Op);
    303                 }
    304                 goto nextDiv;
    305             }
    306             if (canvas) {
    307 #ifdef USE_TRACE
    308                 LOGV("-- src [%d %d %d %d] dst [%g %g %g %g]\n",
    309                          src.fLeft, src.fTop, src.width(), src.height(),
    310                          SkScalarToFloat(dst.fLeft), SkScalarToFloat(dst.fTop),
    311                          SkScalarToFloat(dst.width()), SkScalarToFloat(dst.height()));
    312                 if (2 == src.width() && SkIntToScalar(5) == dst.width()) {
    313                     LOGV("--- skip patch\n");
    314                 }
    315 #endif
    316                 drawStretchyPatch(canvas, src, dst, bitmap, *paint, initColor,
    317                                   color, hasXfer);
    318             }
    319 
    320 nextDiv:
    321             src.fLeft = src.fRight;
    322             dst.fLeft = dst.fRight;
    323         }
    324         src.fTop = src.fBottom;
    325         dst.fTop = dst.fBottom;
    326         dstRightsHaveBeenCached = true;
    327     }
    328 }
    329