PictureTest.cpp revision 5fb2ce38b3dcb8e60e9e112df23c9d42456d7069 
1/*
2 * Copyright 2012 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8#include "SkBitmapDevice.h"
9#include "SkCanvas.h"
10#include "SkColorPriv.h"
11#include "SkData.h"
12#include "SkDecodingImageGenerator.h"
13#include "SkError.h"
14#include "SkImageEncoder.h"
15#include "SkImageGenerator.h"
16#include "SkPaint.h"
17#include "SkPicture.h"
18#include "SkPictureUtils.h"
19#include "SkQuadTreePicture.h"
20#include "SkRRect.h"
21#include "SkRandom.h"
22#include "SkRTreePicture.h"
23#include "SkShader.h"
24#include "SkStream.h"
25#include "SkTileGrid.h"
26#include "Test.h"
27
28static const int gColorScale = 30;
29static const int gColorOffset = 60;
30
31static void make_bm(SkBitmap* bm, int w, int h, SkColor color, bool immutable) {
32    bm->allocN32Pixels(w, h);
33    bm->eraseColor(color);
34    if (immutable) {
35        bm->setImmutable();
36    }
37}
38
39static void make_checkerboard(SkBitmap* bm, int w, int h, bool immutable) {
40    SkASSERT(w % 2 == 0);
41    SkASSERT(h % 2 == 0);
42    bm->allocPixels(SkImageInfo::Make(w, h, kAlpha_8_SkColorType,
43                                      kPremul_SkAlphaType));
44    SkAutoLockPixels lock(*bm);
45    for (int y = 0; y < h; y += 2) {
46        uint8_t* s = bm->getAddr8(0, y);
47        for (int x = 0; x < w; x += 2) {
48            *s++ = 0xFF;
49            *s++ = 0x00;
50        }
51        s = bm->getAddr8(0, y + 1);
52        for (int x = 0; x < w; x += 2) {
53            *s++ = 0x00;
54            *s++ = 0xFF;
55        }
56    }
57    if (immutable) {
58        bm->setImmutable();
59    }
60}
61
62static void init_paint(SkPaint* paint, const SkBitmap &bm) {
63    SkShader* shader = SkShader::CreateBitmapShader(bm,
64                                                    SkShader::kClamp_TileMode,
65                                                    SkShader::kClamp_TileMode);
66    paint->setShader(shader)->unref();
67}
68
69typedef void (*DrawBitmapProc)(SkCanvas*, const SkBitmap&,
70                               const SkBitmap&, const SkPoint&,
71                               SkTDArray<SkPixelRef*>* usedPixRefs);
72
73static void drawpaint_proc(SkCanvas* canvas, const SkBitmap& bm,
74                           const SkBitmap& altBM, const SkPoint& pos,
75                           SkTDArray<SkPixelRef*>* usedPixRefs) {
76    SkPaint paint;
77    init_paint(&paint, bm);
78
79    canvas->drawPaint(paint);
80    *usedPixRefs->append() = bm.pixelRef();
81}
82
83static void drawpoints_proc(SkCanvas* canvas, const SkBitmap& bm,
84                            const SkBitmap& altBM, const SkPoint& pos,
85                            SkTDArray<SkPixelRef*>* usedPixRefs) {
86    SkPaint paint;
87    init_paint(&paint, bm);
88
89    // draw a rect
90    SkPoint points[5] = {
91        { pos.fX, pos.fY },
92        { pos.fX + bm.width() - 1, pos.fY },
93        { pos.fX + bm.width() - 1, pos.fY + bm.height() - 1 },
94        { pos.fX, pos.fY + bm.height() - 1 },
95        { pos.fX, pos.fY },
96    };
97
98    canvas->drawPoints(SkCanvas::kPolygon_PointMode, 5, points, paint);
99    *usedPixRefs->append() = bm.pixelRef();
100}
101
102static void drawrect_proc(SkCanvas* canvas, const SkBitmap& bm,
103                          const SkBitmap& altBM, const SkPoint& pos,
104                          SkTDArray<SkPixelRef*>* usedPixRefs) {
105    SkPaint paint;
106    init_paint(&paint, bm);
107
108    SkRect r = { 0, 0, SkIntToScalar(bm.width()), SkIntToScalar(bm.height()) };
109    r.offset(pos.fX, pos.fY);
110
111    canvas->drawRect(r, paint);
112    *usedPixRefs->append() = bm.pixelRef();
113}
114
115static void drawoval_proc(SkCanvas* canvas, const SkBitmap& bm,
116                          const SkBitmap& altBM, const SkPoint& pos,
117                          SkTDArray<SkPixelRef*>* usedPixRefs) {
118    SkPaint paint;
119    init_paint(&paint, bm);
120
121    SkRect r = { 0, 0, SkIntToScalar(bm.width()), SkIntToScalar(bm.height()) };
122    r.offset(pos.fX, pos.fY);
123
124    canvas->drawOval(r, paint);
125    *usedPixRefs->append() = bm.pixelRef();
126}
127
128static void drawrrect_proc(SkCanvas* canvas, const SkBitmap& bm,
129                           const SkBitmap& altBM, const SkPoint& pos,
130                           SkTDArray<SkPixelRef*>* usedPixRefs) {
131    SkPaint paint;
132    init_paint(&paint, bm);
133
134    SkRect r = { 0, 0, SkIntToScalar(bm.width()), SkIntToScalar(bm.height()) };
135    r.offset(pos.fX, pos.fY);
136
137    SkRRect rr;
138    rr.setRectXY(r, SkIntToScalar(bm.width())/4, SkIntToScalar(bm.height())/4);
139    canvas->drawRRect(rr, paint);
140    *usedPixRefs->append() = bm.pixelRef();
141}
142
143static void drawpath_proc(SkCanvas* canvas, const SkBitmap& bm,
144                          const SkBitmap& altBM, const SkPoint& pos,
145                          SkTDArray<SkPixelRef*>* usedPixRefs) {
146    SkPaint paint;
147    init_paint(&paint, bm);
148
149    SkPath path;
150    path.lineTo(bm.width()/2.0f, SkIntToScalar(bm.height()));
151    path.lineTo(SkIntToScalar(bm.width()), 0);
152    path.close();
153    path.offset(pos.fX, pos.fY);
154
155    canvas->drawPath(path, paint);
156    *usedPixRefs->append() = bm.pixelRef();
157}
158
159static void drawbitmap_proc(SkCanvas* canvas, const SkBitmap& bm,
160                            const SkBitmap& altBM, const SkPoint& pos,
161                            SkTDArray<SkPixelRef*>* usedPixRefs) {
162    canvas->drawBitmap(bm, pos.fX, pos.fY, NULL);
163    *usedPixRefs->append() = bm.pixelRef();
164}
165
166static void drawbitmap_withshader_proc(SkCanvas* canvas, const SkBitmap& bm,
167                                       const SkBitmap& altBM, const SkPoint& pos,
168                                       SkTDArray<SkPixelRef*>* usedPixRefs) {
169    SkPaint paint;
170    init_paint(&paint, bm);
171
172    // The bitmap in the paint is ignored unless we're drawing an A8 bitmap
173    canvas->drawBitmap(altBM, pos.fX, pos.fY, &paint);
174    *usedPixRefs->append() = bm.pixelRef();
175    *usedPixRefs->append() = altBM.pixelRef();
176}
177
178static void drawsprite_proc(SkCanvas* canvas, const SkBitmap& bm,
179                            const SkBitmap& altBM, const SkPoint& pos,
180                            SkTDArray<SkPixelRef*>* usedPixRefs) {
181    const SkMatrix& ctm = canvas->getTotalMatrix();
182
183    SkPoint p(pos);
184    ctm.mapPoints(&p, 1);
185
186    canvas->drawSprite(bm, (int)p.fX, (int)p.fY, NULL);
187    *usedPixRefs->append() = bm.pixelRef();
188}
189
190#if 0
191// Although specifiable, this case doesn't seem to make sense (i.e., the
192// bitmap in the shader is never used).
193static void drawsprite_withshader_proc(SkCanvas* canvas, const SkBitmap& bm,
194                                       const SkBitmap& altBM, const SkPoint& pos,
195                                       SkTDArray<SkPixelRef*>* usedPixRefs) {
196    SkPaint paint;
197    init_paint(&paint, bm);
198
199    const SkMatrix& ctm = canvas->getTotalMatrix();
200
201    SkPoint p(pos);
202    ctm.mapPoints(&p, 1);
203
204    canvas->drawSprite(altBM, (int)p.fX, (int)p.fY, &paint);
205    *usedPixRefs->append() = bm.pixelRef();
206    *usedPixRefs->append() = altBM.pixelRef();
207}
208#endif
209
210static void drawbitmaprect_proc(SkCanvas* canvas, const SkBitmap& bm,
211                                const SkBitmap& altBM, const SkPoint& pos,
212                                SkTDArray<SkPixelRef*>* usedPixRefs) {
213    SkRect r = { 0, 0, SkIntToScalar(bm.width()), SkIntToScalar(bm.height()) };
214
215    r.offset(pos.fX, pos.fY);
216    canvas->drawBitmapRectToRect(bm, NULL, r, NULL);
217    *usedPixRefs->append() = bm.pixelRef();
218}
219
220static void drawbitmaprect_withshader_proc(SkCanvas* canvas,
221                                           const SkBitmap& bm,
222                                           const SkBitmap& altBM,
223                                           const SkPoint& pos,
224                                           SkTDArray<SkPixelRef*>* usedPixRefs) {
225    SkPaint paint;
226    init_paint(&paint, bm);
227
228    SkRect r = { 0, 0, SkIntToScalar(bm.width()), SkIntToScalar(bm.height()) };
229    r.offset(pos.fX, pos.fY);
230
231    // The bitmap in the paint is ignored unless we're drawing an A8 bitmap
232    canvas->drawBitmapRectToRect(altBM, NULL, r, &paint);
233    *usedPixRefs->append() = bm.pixelRef();
234    *usedPixRefs->append() = altBM.pixelRef();
235}
236
237static void drawtext_proc(SkCanvas* canvas, const SkBitmap& bm,
238                          const SkBitmap& altBM, const SkPoint& pos,
239                          SkTDArray<SkPixelRef*>* usedPixRefs) {
240    SkPaint paint;
241    init_paint(&paint, bm);
242    paint.setTextSize(SkIntToScalar(1.5*bm.width()));
243
244    canvas->drawText("0", 1, pos.fX, pos.fY+bm.width(), paint);
245    *usedPixRefs->append() = bm.pixelRef();
246}
247
248static void drawpostext_proc(SkCanvas* canvas, const SkBitmap& bm,
249                             const SkBitmap& altBM, const SkPoint& pos,
250                             SkTDArray<SkPixelRef*>* usedPixRefs) {
251    SkPaint paint;
252    init_paint(&paint, bm);
253    paint.setTextSize(SkIntToScalar(1.5*bm.width()));
254
255    SkPoint point = { pos.fX, pos.fY + bm.height() };
256    canvas->drawPosText("O", 1, &point, paint);
257    *usedPixRefs->append() = bm.pixelRef();
258}
259
260static void drawtextonpath_proc(SkCanvas* canvas, const SkBitmap& bm,
261                                const SkBitmap& altBM, const SkPoint& pos,
262                                SkTDArray<SkPixelRef*>* usedPixRefs) {
263    SkPaint paint;
264
265    init_paint(&paint, bm);
266    paint.setTextSize(SkIntToScalar(1.5*bm.width()));
267
268    SkPath path;
269    path.lineTo(SkIntToScalar(bm.width()), 0);
270    path.offset(pos.fX, pos.fY+bm.height());
271
272    canvas->drawTextOnPath("O", 1, path, NULL, paint);
273    *usedPixRefs->append() = bm.pixelRef();
274}
275
276static void drawverts_proc(SkCanvas* canvas, const SkBitmap& bm,
277                           const SkBitmap& altBM, const SkPoint& pos,
278                           SkTDArray<SkPixelRef*>* usedPixRefs) {
279    SkPaint paint;
280    init_paint(&paint, bm);
281
282    SkPoint verts[4] = {
283        { pos.fX, pos.fY },
284        { pos.fX + bm.width(), pos.fY },
285        { pos.fX + bm.width(), pos.fY + bm.height() },
286        { pos.fX, pos.fY + bm.height() }
287    };
288    SkPoint texs[4] = { { 0, 0 },
289                        { SkIntToScalar(bm.width()), 0 },
290                        { SkIntToScalar(bm.width()), SkIntToScalar(bm.height()) },
291                        { 0, SkIntToScalar(bm.height()) } };
292    uint16_t indices[6] = { 0, 1, 2, 0, 2, 3 };
293
294    canvas->drawVertices(SkCanvas::kTriangles_VertexMode, 4, verts, texs, NULL, NULL,
295                         indices, 6, paint);
296    *usedPixRefs->append() = bm.pixelRef();
297}
298
299// Return a picture with the bitmaps drawn at the specified positions.
300static SkPicture* record_bitmaps(const SkBitmap bm[],
301                                 const SkPoint pos[],
302                                 SkTDArray<SkPixelRef*> analytic[],
303                                 int count,
304                                 DrawBitmapProc proc) {
305    SkPictureRecorder recorder;
306    SkCanvas* canvas = recorder.beginRecording(1000, 1000, NULL, 0);
307    for (int i = 0; i < count; ++i) {
308        analytic[i].rewind();
309        canvas->save();
310        SkRect clipRect = SkRect::MakeXYWH(pos[i].fX, pos[i].fY,
311                                           SkIntToScalar(bm[i].width()),
312                                           SkIntToScalar(bm[i].height()));
313        canvas->clipRect(clipRect, SkRegion::kIntersect_Op);
314        proc(canvas, bm[i], bm[count+i], pos[i], &analytic[i]);
315        canvas->restore();
316    }
317    return recorder.endRecording();
318}
319
320static void rand_rect(SkRect* rect, SkRandom& rand, SkScalar W, SkScalar H) {
321    rect->fLeft   = rand.nextRangeScalar(-W, 2*W);
322    rect->fTop    = rand.nextRangeScalar(-H, 2*H);
323    rect->fRight  = rect->fLeft + rand.nextRangeScalar(0, W);
324    rect->fBottom = rect->fTop + rand.nextRangeScalar(0, H);
325
326    // we integralize rect to make our tests more predictable, since Gather is
327    // a little sloppy.
328    SkIRect ir;
329    rect->round(&ir);
330    rect->set(ir);
331}
332
333static void draw(SkPicture* pic, int width, int height, SkBitmap* result) {
334    make_bm(result, width, height, SK_ColorBLACK, false);
335
336    SkCanvas canvas(*result);
337    canvas.drawPicture(*pic);
338}
339
340template <typename T> int find_index(const T* array, T elem, int count) {
341    for (int i = 0; i < count; ++i) {
342        if (array[i] == elem) {
343            return i;
344        }
345    }
346    return -1;
347}
348
349// Return true if 'ref' is found in array[]
350static bool find(SkPixelRef const * const * array, SkPixelRef const * ref, int count) {
351    return find_index<const SkPixelRef*>(array, ref, count) >= 0;
352}
353
354// Look at each pixel that is inside 'subset', and if its color appears in
355// colors[], find the corresponding value in refs[] and append that ref into
356// array, skipping duplicates of the same value.
357// Note that gathering pixelRefs from rendered colors suffers from the problem
358// that multiple simultaneous textures (e.g., A8 for alpha and 8888 for color)
359// isn't easy to reconstruct.
360static void gather_from_image(const SkBitmap& bm, SkPixelRef* const refs[],
361                              int count, SkTDArray<SkPixelRef*>* array,
362                              const SkRect& subset) {
363    SkIRect ir;
364    subset.roundOut(&ir);
365
366    if (!ir.intersect(0, 0, bm.width()-1, bm.height()-1)) {
367        return;
368    }
369
370    // Since we only want to return unique values in array, when we scan we just
371    // set a bit for each index'd color found. In practice we only have a few
372    // distinct colors, so we just use an int's bits as our array. Hence the
373    // assert that count <= number-of-bits-in-our-int.
374    SkASSERT((unsigned)count <= 32);
375    uint32_t bitarray = 0;
376
377    SkAutoLockPixels alp(bm);
378
379    for (int y = ir.fTop; y < ir.fBottom; ++y) {
380        for (int x = ir.fLeft; x < ir.fRight; ++x) {
381            SkPMColor pmc = *bm.getAddr32(x, y);
382            // the only good case where the color is not found would be if
383            // the color is transparent, meaning no bitmap was drawn in that
384            // pixel.
385            if (pmc) {
386                uint32_t index = SkGetPackedR32(pmc);
387                SkASSERT(SkGetPackedG32(pmc) == index);
388                SkASSERT(SkGetPackedB32(pmc) == index);
389                if (0 == index) {
390                    continue;           // background color
391                }
392                SkASSERT(0 == (index - gColorOffset) % gColorScale);
393                index = (index - gColorOffset) / gColorScale;
394                SkASSERT(static_cast<int>(index) < count);
395                bitarray |= 1 << index;
396            }
397        }
398    }
399
400    for (int i = 0; i < count; ++i) {
401        if (bitarray & (1 << i)) {
402            *array->append() = refs[i];
403        }
404    }
405}
406
407static void gather_from_analytic(const SkPoint pos[], SkScalar w, SkScalar h,
408                                 const SkTDArray<SkPixelRef*> analytic[],
409                                 int count,
410                                 SkTDArray<SkPixelRef*>* result,
411                                 const SkRect& subset) {
412    for (int i = 0; i < count; ++i) {
413        SkRect rect = SkRect::MakeXYWH(pos[i].fX, pos[i].fY, w, h);
414
415        if (SkRect::Intersects(subset, rect)) {
416            result->append(analytic[i].count(), analytic[i].begin());
417        }
418    }
419}
420
421
422static const struct {
423    const DrawBitmapProc proc;
424    const char* const desc;
425} gProcs[] = {
426    {drawpaint_proc, "drawpaint"},
427    {drawpoints_proc, "drawpoints"},
428    {drawrect_proc, "drawrect"},
429    {drawoval_proc, "drawoval"},
430    {drawrrect_proc, "drawrrect"},
431    {drawpath_proc, "drawpath"},
432    {drawbitmap_proc, "drawbitmap"},
433    {drawbitmap_withshader_proc, "drawbitmap_withshader"},
434    {drawsprite_proc, "drawsprite"},
435#if 0
436    {drawsprite_withshader_proc, "drawsprite_withshader"},
437#endif
438    {drawbitmaprect_proc, "drawbitmaprect"},
439    {drawbitmaprect_withshader_proc, "drawbitmaprect_withshader"},
440    {drawtext_proc, "drawtext"},
441    {drawpostext_proc, "drawpostext"},
442    {drawtextonpath_proc, "drawtextonpath"},
443    {drawverts_proc, "drawverts"},
444};
445
446static void create_textures(SkBitmap* bm, SkPixelRef** refs, int num, int w, int h) {
447    // Our convention is that the color components contain an encoding of
448    // the index of their corresponding bitmap/pixelref. (0,0,0,0) is
449    // reserved for the background
450    for (int i = 0; i < num; ++i) {
451        make_bm(&bm[i], w, h,
452                SkColorSetARGB(0xFF,
453                               gColorScale*i+gColorOffset,
454                               gColorScale*i+gColorOffset,
455                               gColorScale*i+gColorOffset),
456                true);
457        refs[i] = bm[i].pixelRef();
458    }
459
460    // The A8 alternate bitmaps are all BW checkerboards
461    for (int i = 0; i < num; ++i) {
462        make_checkerboard(&bm[num+i], w, h, true);
463        refs[num+i] = bm[num+i].pixelRef();
464    }
465}
466
467static void test_gatherpixelrefs(skiatest::Reporter* reporter) {
468    const int IW = 32;
469    const int IH = IW;
470    const SkScalar W = SkIntToScalar(IW);
471    const SkScalar H = W;
472
473    static const int N = 4;
474    SkBitmap bm[2*N];
475    SkPixelRef* refs[2*N];
476    SkTDArray<SkPixelRef*> analytic[N];
477
478    const SkPoint pos[N] = {
479        { 0, 0 }, { W, 0 }, { 0, H }, { W, H }
480    };
481
482    create_textures(bm, refs, N, IW, IH);
483
484    SkRandom rand;
485    for (size_t k = 0; k < SK_ARRAY_COUNT(gProcs); ++k) {
486        SkAutoTUnref<SkPicture> pic(
487            record_bitmaps(bm, pos, analytic, N, gProcs[k].proc));
488
489        REPORTER_ASSERT(reporter, pic->willPlayBackBitmaps() || N == 0);
490        // quick check for a small piece of each quadrant, which should just
491        // contain 1 or 2 bitmaps.
492        for (size_t  i = 0; i < SK_ARRAY_COUNT(pos); ++i) {
493            SkRect r;
494            r.set(2, 2, W - 2, H - 2);
495            r.offset(pos[i].fX, pos[i].fY);
496            SkAutoDataUnref data(SkPictureUtils::GatherPixelRefs(pic, r));
497            if (!data) {
498                ERRORF(reporter, "SkPictureUtils::GatherPixelRefs returned "
499                       "NULL for %s.", gProcs[k].desc);
500                continue;
501            }
502            SkPixelRef** gatheredRefs = (SkPixelRef**)data->data();
503            int count = static_cast<int>(data->size() / sizeof(SkPixelRef*));
504            REPORTER_ASSERT(reporter, 1 == count || 2 == count);
505            if (1 == count) {
506                REPORTER_ASSERT(reporter, gatheredRefs[0] == refs[i]);
507            } else if (2 == count) {
508                REPORTER_ASSERT(reporter,
509                    (gatheredRefs[0] == refs[i] && gatheredRefs[1] == refs[i+N]) ||
510                    (gatheredRefs[1] == refs[i] && gatheredRefs[0] == refs[i+N]));
511            }
512        }
513
514        SkBitmap image;
515        draw(pic, 2*IW, 2*IH, &image);
516
517        // Test a bunch of random (mostly) rects, and compare the gather results
518        // with a deduced list of refs by looking at the colors drawn.
519        for (int j = 0; j < 100; ++j) {
520            SkRect r;
521            rand_rect(&r, rand, 2*W, 2*H);
522
523            SkTDArray<SkPixelRef*> fromImage;
524            gather_from_image(image, refs, N, &fromImage, r);
525
526            SkTDArray<SkPixelRef*> fromAnalytic;
527            gather_from_analytic(pos, W, H, analytic, N, &fromAnalytic, r);
528
529            SkData* data = SkPictureUtils::GatherPixelRefs(pic, r);
530            size_t dataSize = data ? data->size() : 0;
531            int gatherCount = static_cast<int>(dataSize / sizeof(SkPixelRef*));
532            SkASSERT(gatherCount * sizeof(SkPixelRef*) == dataSize);
533            SkPixelRef** gatherRefs = data ? (SkPixelRef**)(data->data()) : NULL;
534            SkAutoDataUnref adu(data);
535
536            // Everything that we saw drawn should appear in the analytic list
537            // but the analytic list may contain some pixelRefs that were not
538            // seen in the image (e.g., A8 textures used as masks)
539            for (int i = 0; i < fromImage.count(); ++i) {
540                if (-1 == fromAnalytic.find(fromImage[i])) {
541                    ERRORF(reporter, "PixelRef missing %d %s",
542                           i, gProcs[k].desc);
543                }
544            }
545
546            /*
547             *  GatherPixelRefs is conservative, so it can return more bitmaps
548             *  than are strictly required. Thus our check here is only that
549             *  Gather didn't miss any that we actually needed. Even that isn't
550             *  a strict requirement on Gather, which is meant to be quick and
551             *  only mostly-correct, but at the moment this test should work.
552             */
553            for (int i = 0; i < fromAnalytic.count(); ++i) {
554                bool found = find(gatherRefs, fromAnalytic[i], gatherCount);
555                if (!found) {
556                    ERRORF(reporter, "PixelRef missing %d %s",
557                           i, gProcs[k].desc);
558                }
559#if 0
560                // enable this block of code to debug failures, as it will rerun
561                // the case that failed.
562                if (!found) {
563                    SkData* data = SkPictureUtils::GatherPixelRefs(pic, r);
564                    size_t dataSize = data ? data->size() : 0;
565                }
566#endif
567            }
568        }
569    }
570}
571
572static void test_gatherpixelrefsandrects(skiatest::Reporter* reporter) {
573    const int IW = 32;
574    const int IH = IW;
575    const SkScalar W = SkIntToScalar(IW);
576    const SkScalar H = W;
577
578    static const int N = 4;
579    SkBitmap bm[2*N];
580    SkPixelRef* refs[2*N];
581    SkTDArray<SkPixelRef*> analytic[N];
582
583    const SkPoint pos[N] = {
584        { 0, 0 }, { W, 0 }, { 0, H }, { W, H }
585    };
586
587    create_textures(bm, refs, N, IW, IH);
588
589    SkRandom rand;
590    for (size_t k = 0; k < SK_ARRAY_COUNT(gProcs); ++k) {
591        SkAutoTUnref<SkPicture> pic(
592            record_bitmaps(bm, pos, analytic, N, gProcs[k].proc));
593
594        REPORTER_ASSERT(reporter, pic->willPlayBackBitmaps() || N == 0);
595
596        SkAutoTUnref<SkPictureUtils::SkPixelRefContainer> prCont(
597                                new SkPictureUtils::SkPixelRefsAndRectsList);
598
599        SkPictureUtils::GatherPixelRefsAndRects(pic, prCont);
600
601        // quick check for a small piece of each quadrant, which should just
602        // contain 1 or 2 bitmaps.
603        for (size_t  i = 0; i < SK_ARRAY_COUNT(pos); ++i) {
604            SkRect r;
605            r.set(2, 2, W - 2, H - 2);
606            r.offset(pos[i].fX, pos[i].fY);
607
608            SkTDArray<SkPixelRef*> gatheredRefs;
609            prCont->query(r, &gatheredRefs);
610
611            int count = gatheredRefs.count();
612            REPORTER_ASSERT(reporter, 1 == count || 2 == count);
613            if (1 == count) {
614                REPORTER_ASSERT(reporter, gatheredRefs[0] == refs[i]);
615            } else if (2 == count) {
616                REPORTER_ASSERT(reporter,
617                    (gatheredRefs[0] == refs[i] && gatheredRefs[1] == refs[i+N]) ||
618                    (gatheredRefs[1] == refs[i] && gatheredRefs[0] == refs[i+N]));
619            }
620        }
621
622        SkBitmap image;
623        draw(pic, 2*IW, 2*IH, &image);
624
625        // Test a bunch of random (mostly) rects, and compare the gather results
626        // with the analytic results and the pixel refs seen in a rendering.
627        for (int j = 0; j < 100; ++j) {
628            SkRect r;
629            rand_rect(&r, rand, 2*W, 2*H);
630
631            SkTDArray<SkPixelRef*> fromImage;
632            gather_from_image(image, refs, N, &fromImage, r);
633
634            SkTDArray<SkPixelRef*> fromAnalytic;
635            gather_from_analytic(pos, W, H, analytic, N, &fromAnalytic, r);
636
637            SkTDArray<SkPixelRef*> gatheredRefs;
638            prCont->query(r, &gatheredRefs);
639
640            // Everything that we saw drawn should appear in the analytic list
641            // but the analytic list may contain some pixelRefs that were not
642            // seen in the image (e.g., A8 textures used as masks)
643            for (int i = 0; i < fromImage.count(); ++i) {
644                REPORTER_ASSERT(reporter, -1 != fromAnalytic.find(fromImage[i]));
645            }
646
647            // Everything in the analytic list should appear in the gathered
648            // list.
649            for (int i = 0; i < fromAnalytic.count(); ++i) {
650                REPORTER_ASSERT(reporter, -1 != gatheredRefs.find(fromAnalytic[i]));
651            }
652        }
653    }
654}
655
656#ifdef SK_DEBUG
657// Ensure that deleting SkPicturePlayback does not assert. Asserts only fire in debug mode, so only
658// run in debug mode.
659static void test_deleting_empty_playback() {
660    SkPictureRecorder recorder;
661    // Creates an SkPictureRecord
662    recorder.beginRecording(0, 0, NULL, 0);
663    // Turns that into an SkPicturePlayback
664    SkAutoTUnref<SkPicture> picture(recorder.endRecording());
665    // Deletes the old SkPicturePlayback, and creates a new SkPictureRecord
666    recorder.beginRecording(0, 0, NULL, 0);
667}
668
669// Ensure that serializing an empty picture does not assert. Likewise only runs in debug mode.
670static void test_serializing_empty_picture() {
671    SkPictureRecorder recorder;
672    recorder.beginRecording(0, 0, NULL, 0);
673    SkAutoTUnref<SkPicture> picture(recorder.endRecording());
674    SkDynamicMemoryWStream stream;
675    picture->serialize(&stream);
676}
677#endif
678
679static void rand_op(SkCanvas* canvas, SkRandom& rand) {
680    SkPaint paint;
681    SkRect rect = SkRect::MakeWH(50, 50);
682
683    SkScalar unit = rand.nextUScalar1();
684    if (unit <= 0.3) {
685//        SkDebugf("save\n");
686        canvas->save();
687    } else if (unit <= 0.6) {
688//        SkDebugf("restore\n");
689        canvas->restore();
690    } else if (unit <= 0.9) {
691//        SkDebugf("clip\n");
692        canvas->clipRect(rect);
693    } else {
694//        SkDebugf("draw\n");
695        canvas->drawPaint(paint);
696    }
697}
698
699static void set_canvas_to_save_count_4(SkCanvas* canvas) {
700    canvas->restoreToCount(1);
701    canvas->save();
702    canvas->save();
703    canvas->save();
704}
705
706static void test_unbalanced_save_restores(skiatest::Reporter* reporter) {
707    SkCanvas testCanvas(100, 100);
708    set_canvas_to_save_count_4(&testCanvas);
709
710    REPORTER_ASSERT(reporter, 4 == testCanvas.getSaveCount());
711
712    SkPaint paint;
713    SkRect rect = SkRect::MakeLTRB(-10000000, -10000000, 10000000, 10000000);
714
715    SkPictureRecorder recorder;
716
717    {
718        // Create picture with 2 unbalanced saves
719        SkCanvas* canvas = recorder.beginRecording(100, 100, NULL, 0);
720        canvas->save();
721        canvas->translate(10, 10);
722        canvas->drawRect(rect, paint);
723        canvas->save();
724        canvas->translate(10, 10);
725        canvas->drawRect(rect, paint);
726        SkAutoTUnref<SkPicture> extraSavePicture(recorder.endRecording());
727
728        testCanvas.drawPicture(*extraSavePicture);
729        REPORTER_ASSERT(reporter, 4 == testCanvas.getSaveCount());
730    }
731
732    set_canvas_to_save_count_4(&testCanvas);
733
734    {
735        // Create picture with 2 unbalanced restores
736        SkCanvas* canvas = recorder.beginRecording(100, 100, NULL, 0);
737        canvas->save();
738        canvas->translate(10, 10);
739        canvas->drawRect(rect, paint);
740        canvas->save();
741        canvas->translate(10, 10);
742        canvas->drawRect(rect, paint);
743        canvas->restore();
744        canvas->restore();
745        canvas->restore();
746        canvas->restore();
747        SkAutoTUnref<SkPicture> extraRestorePicture(recorder.endRecording());
748
749        testCanvas.drawPicture(*extraRestorePicture);
750        REPORTER_ASSERT(reporter, 4 == testCanvas.getSaveCount());
751    }
752
753    set_canvas_to_save_count_4(&testCanvas);
754
755    {
756        SkCanvas* canvas = recorder.beginRecording(100, 100, NULL, 0);
757        canvas->translate(10, 10);
758        canvas->drawRect(rect, paint);
759        SkAutoTUnref<SkPicture> noSavePicture(recorder.endRecording());
760
761        testCanvas.drawPicture(*noSavePicture);
762        REPORTER_ASSERT(reporter, 4 == testCanvas.getSaveCount());
763        REPORTER_ASSERT(reporter, testCanvas.getTotalMatrix().isIdentity());
764    }
765}
766
767static void test_peephole() {
768    SkRandom rand;
769
770    SkPictureRecorder recorder;
771
772    for (int j = 0; j < 100; j++) {
773        SkRandom rand2(rand); // remember the seed
774
775        SkCanvas* canvas = recorder.beginRecording(100, 100, NULL, 0);
776
777        for (int i = 0; i < 1000; ++i) {
778            rand_op(canvas, rand);
779        }
780        SkAutoTUnref<SkPicture> picture(recorder.endRecording());
781
782        rand = rand2;
783    }
784
785    {
786        SkCanvas* canvas = recorder.beginRecording(100, 100, NULL, 0);
787        SkRect rect = SkRect::MakeWH(50, 50);
788
789        for (int i = 0; i < 100; ++i) {
790            canvas->save();
791        }
792        while (canvas->getSaveCount() > 1) {
793            canvas->clipRect(rect);
794            canvas->restore();
795        }
796        SkAutoTUnref<SkPicture> picture(recorder.endRecording());
797    }
798}
799
800#ifndef SK_DEBUG
801// Only test this is in release mode. We deliberately crash in debug mode, since a valid caller
802// should never do this.
803static void test_bad_bitmap() {
804    // This bitmap has a width and height but no pixels. As a result, attempting to record it will
805    // fail.
806    SkBitmap bm;
807    bm.setConfig(SkImageInfo::MakeN32Premul(100, 100));
808    SkPictureRecorder recorder;
809    SkCanvas* recordingCanvas = recorder.beginRecording(100, 100, NULL, 0);
810    recordingCanvas->drawBitmap(bm, 0, 0);
811    SkAutoTUnref<SkPicture> picture(recorder.endRecording());
812
813    SkCanvas canvas;
814    canvas.drawPicture(*picture);
815}
816#endif
817
818static SkData* encode_bitmap_to_data(size_t*, const SkBitmap& bm) {
819    return SkImageEncoder::EncodeData(bm, SkImageEncoder::kPNG_Type, 100);
820}
821
822static SkData* serialized_picture_from_bitmap(const SkBitmap& bitmap) {
823    SkPictureRecorder recorder;
824    SkCanvas* canvas = recorder.beginRecording(bitmap.width(), bitmap.height(), NULL, 0);
825    canvas->drawBitmap(bitmap, 0, 0);
826    SkAutoTUnref<SkPicture> picture(recorder.endRecording());
827
828    SkDynamicMemoryWStream wStream;
829    picture->serialize(&wStream, &encode_bitmap_to_data);
830    return wStream.copyToData();
831}
832
833struct ErrorContext {
834    int fErrors;
835    skiatest::Reporter* fReporter;
836};
837
838static void assert_one_parse_error_cb(SkError error, void* context) {
839    ErrorContext* errorContext = static_cast<ErrorContext*>(context);
840    errorContext->fErrors++;
841    // This test only expects one error, and that is a kParseError. If there are others,
842    // there is some unknown problem.
843    REPORTER_ASSERT_MESSAGE(errorContext->fReporter, 1 == errorContext->fErrors,
844                            "This threw more errors than expected.");
845    REPORTER_ASSERT_MESSAGE(errorContext->fReporter, kParseError_SkError == error,
846                            SkGetLastErrorString());
847}
848
849static void test_bitmap_with_encoded_data(skiatest::Reporter* reporter) {
850    // Create a bitmap that will be encoded.
851    SkBitmap original;
852    make_bm(&original, 100, 100, SK_ColorBLUE, true);
853    SkDynamicMemoryWStream wStream;
854    if (!SkImageEncoder::EncodeStream(&wStream, original, SkImageEncoder::kPNG_Type, 100)) {
855        return;
856    }
857    SkAutoDataUnref data(wStream.copyToData());
858
859    SkBitmap bm;
860    bool installSuccess = SkInstallDiscardablePixelRef(
861         SkDecodingImageGenerator::Create(data, SkDecodingImageGenerator::Options()), &bm, NULL);
862    REPORTER_ASSERT(reporter, installSuccess);
863
864    // Write both bitmaps to pictures, and ensure that the resulting data streams are the same.
865    // Flattening original will follow the old path of performing an encode, while flattening bm
866    // will use the already encoded data.
867    SkAutoDataUnref picture1(serialized_picture_from_bitmap(original));
868    SkAutoDataUnref picture2(serialized_picture_from_bitmap(bm));
869    REPORTER_ASSERT(reporter, picture1->equals(picture2));
870    // Now test that a parse error was generated when trying to create a new SkPicture without
871    // providing a function to decode the bitmap.
872    ErrorContext context;
873    context.fErrors = 0;
874    context.fReporter = reporter;
875    SkSetErrorCallback(assert_one_parse_error_cb, &context);
876    SkMemoryStream pictureStream(picture1);
877    SkClearLastError();
878    SkAutoUnref pictureFromStream(SkPicture::CreateFromStream(&pictureStream, NULL));
879    REPORTER_ASSERT(reporter, pictureFromStream.get() != NULL);
880    SkClearLastError();
881    SkSetErrorCallback(NULL, NULL);
882}
883
884static void test_clone_empty(skiatest::Reporter* reporter) {
885    // This is a regression test for crbug.com/172062
886    // Before the fix, we used to crash accessing a null pointer when we
887    // had a picture with no paints. This test passes by not crashing.
888    {
889        SkPictureRecorder recorder;
890        recorder.beginRecording(1, 1, NULL, 0);
891        SkAutoTUnref<SkPicture> picture(recorder.endRecording());
892        SkAutoTUnref<SkPicture> destPicture(picture->clone());
893        REPORTER_ASSERT(reporter, NULL != destPicture);
894    }
895}
896
897static void test_draw_empty(skiatest::Reporter* reporter) {
898    SkBitmap result;
899    make_bm(&result, 2, 2, SK_ColorBLACK, false);
900
901    SkCanvas canvas(result);
902
903    {
904        // stock SkPicture
905        SkPictureRecorder recorder;
906        recorder.beginRecording(1, 1, NULL, 0);
907        SkAutoTUnref<SkPicture> picture(recorder.endRecording());
908
909        canvas.drawPicture(*picture);
910    }
911
912    {
913        // tile grid
914        SkTileGridFactory::TileGridInfo gridInfo;
915        gridInfo.fMargin.setEmpty();
916        gridInfo.fOffset.setZero();
917        gridInfo.fTileInterval.set(1, 1);
918
919        SkTileGridFactory factory(gridInfo);
920        SkPictureRecorder recorder;
921        recorder.beginRecording(1, 1, &factory, 0);
922        SkAutoTUnref<SkPicture> picture(recorder.endRecording());
923
924        canvas.drawPicture(*picture);
925    }
926
927    {
928        // RTree
929        SkRTreeFactory factory;
930        SkPictureRecorder recorder;
931        recorder.beginRecording(1, 1, &factory, 0);
932        SkAutoTUnref<SkPicture> picture(recorder.endRecording());
933
934        canvas.drawPicture(*picture);
935    }
936
937    {
938        // quad tree
939        SkQuadTreeFactory factory;
940        SkPictureRecorder recorder;
941        recorder.beginRecording(1, 1, &factory, 0);
942        SkAutoTUnref<SkPicture> picture(recorder.endRecording());
943
944        canvas.drawPicture(*picture);
945    }
946}
947
948static void test_clip_bound_opt(skiatest::Reporter* reporter) {
949    // Test for crbug.com/229011
950    SkRect rect1 = SkRect::MakeXYWH(SkIntToScalar(4), SkIntToScalar(4),
951                                    SkIntToScalar(2), SkIntToScalar(2));
952    SkRect rect2 = SkRect::MakeXYWH(SkIntToScalar(7), SkIntToScalar(7),
953                                    SkIntToScalar(1), SkIntToScalar(1));
954    SkRect rect3 = SkRect::MakeXYWH(SkIntToScalar(6), SkIntToScalar(6),
955                                    SkIntToScalar(1), SkIntToScalar(1));
956
957    SkPath invPath;
958    invPath.addOval(rect1);
959    invPath.setFillType(SkPath::kInverseEvenOdd_FillType);
960    SkPath path;
961    path.addOval(rect2);
962    SkPath path2;
963    path2.addOval(rect3);
964    SkIRect clipBounds;
965    SkPictureRecorder recorder;
966    // Minimalist test set for 100% code coverage of
967    // SkPictureRecord::updateClipConservativelyUsingBounds
968    {
969        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL,
970            SkPicture::kUsePathBoundsForClip_RecordingFlag);
971        canvas->clipPath(invPath, SkRegion::kIntersect_Op);
972        bool nonEmpty = canvas->getClipDeviceBounds(&clipBounds);
973        REPORTER_ASSERT(reporter, true == nonEmpty);
974        REPORTER_ASSERT(reporter, 0 == clipBounds.fLeft);
975        REPORTER_ASSERT(reporter, 0 == clipBounds.fTop);
976        REPORTER_ASSERT(reporter, 10 == clipBounds.fBottom);
977        REPORTER_ASSERT(reporter, 10 == clipBounds.fRight);
978    }
979    {
980        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL,
981            SkPicture::kUsePathBoundsForClip_RecordingFlag);
982        canvas->clipPath(path, SkRegion::kIntersect_Op);
983        canvas->clipPath(invPath, SkRegion::kIntersect_Op);
984        bool nonEmpty = canvas->getClipDeviceBounds(&clipBounds);
985        REPORTER_ASSERT(reporter, true == nonEmpty);
986        REPORTER_ASSERT(reporter, 7 == clipBounds.fLeft);
987        REPORTER_ASSERT(reporter, 7 == clipBounds.fTop);
988        REPORTER_ASSERT(reporter, 8 == clipBounds.fBottom);
989        REPORTER_ASSERT(reporter, 8 == clipBounds.fRight);
990    }
991    {
992        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL,
993            SkPicture::kUsePathBoundsForClip_RecordingFlag);
994        canvas->clipPath(path, SkRegion::kIntersect_Op);
995        canvas->clipPath(invPath, SkRegion::kUnion_Op);
996        bool nonEmpty = canvas->getClipDeviceBounds(&clipBounds);
997        REPORTER_ASSERT(reporter, true == nonEmpty);
998        REPORTER_ASSERT(reporter, 0 == clipBounds.fLeft);
999        REPORTER_ASSERT(reporter, 0 == clipBounds.fTop);
1000        REPORTER_ASSERT(reporter, 10 == clipBounds.fBottom);
1001        REPORTER_ASSERT(reporter, 10 == clipBounds.fRight);
1002    }
1003    {
1004        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL,
1005            SkPicture::kUsePathBoundsForClip_RecordingFlag);
1006        canvas->clipPath(path, SkRegion::kDifference_Op);
1007        bool nonEmpty = canvas->getClipDeviceBounds(&clipBounds);
1008        REPORTER_ASSERT(reporter, true == nonEmpty);
1009        REPORTER_ASSERT(reporter, 0 == clipBounds.fLeft);
1010        REPORTER_ASSERT(reporter, 0 == clipBounds.fTop);
1011        REPORTER_ASSERT(reporter, 10 == clipBounds.fBottom);
1012        REPORTER_ASSERT(reporter, 10 == clipBounds.fRight);
1013    }
1014    {
1015        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL,
1016            SkPicture::kUsePathBoundsForClip_RecordingFlag);
1017        canvas->clipPath(path, SkRegion::kReverseDifference_Op);
1018        bool nonEmpty = canvas->getClipDeviceBounds(&clipBounds);
1019        // True clip is actually empty in this case, but the best
1020        // determination we can make using only bounds as input is that the
1021        // clip is included in the bounds of 'path'.
1022        REPORTER_ASSERT(reporter, true == nonEmpty);
1023        REPORTER_ASSERT(reporter, 7 == clipBounds.fLeft);
1024        REPORTER_ASSERT(reporter, 7 == clipBounds.fTop);
1025        REPORTER_ASSERT(reporter, 8 == clipBounds.fBottom);
1026        REPORTER_ASSERT(reporter, 8 == clipBounds.fRight);
1027    }
1028    {
1029        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL,
1030            SkPicture::kUsePathBoundsForClip_RecordingFlag);
1031        canvas->clipPath(path, SkRegion::kIntersect_Op);
1032        canvas->clipPath(path2, SkRegion::kXOR_Op);
1033        bool nonEmpty = canvas->getClipDeviceBounds(&clipBounds);
1034        REPORTER_ASSERT(reporter, true == nonEmpty);
1035        REPORTER_ASSERT(reporter, 6 == clipBounds.fLeft);
1036        REPORTER_ASSERT(reporter, 6 == clipBounds.fTop);
1037        REPORTER_ASSERT(reporter, 8 == clipBounds.fBottom);
1038        REPORTER_ASSERT(reporter, 8 == clipBounds.fRight);
1039    }
1040}
1041
1042/**
1043 * A canvas that records the number of clip commands.
1044 */
1045class ClipCountingCanvas : public SkCanvas {
1046public:
1047    explicit ClipCountingCanvas(int width, int height)
1048        : INHERITED(width, height)
1049        , fClipCount(0){
1050    }
1051
1052    virtual void onClipRect(const SkRect& r,
1053                            SkRegion::Op op,
1054                            ClipEdgeStyle edgeStyle) SK_OVERRIDE {
1055        fClipCount += 1;
1056        this->INHERITED::onClipRect(r, op, edgeStyle);
1057    }
1058
1059    virtual void onClipRRect(const SkRRect& rrect,
1060                             SkRegion::Op op,
1061                             ClipEdgeStyle edgeStyle)SK_OVERRIDE {
1062        fClipCount += 1;
1063        this->INHERITED::onClipRRect(rrect, op, edgeStyle);
1064    }
1065
1066    virtual void onClipPath(const SkPath& path,
1067                            SkRegion::Op op,
1068                            ClipEdgeStyle edgeStyle) SK_OVERRIDE {
1069        fClipCount += 1;
1070        this->INHERITED::onClipPath(path, op, edgeStyle);
1071    }
1072
1073    virtual void onClipRegion(const SkRegion& deviceRgn, SkRegion::Op op) SK_OVERRIDE {
1074        fClipCount += 1;
1075        this->INHERITED::onClipRegion(deviceRgn, op);
1076    }
1077
1078    unsigned getClipCount() const { return fClipCount; }
1079
1080private:
1081    unsigned fClipCount;
1082
1083    typedef SkCanvas INHERITED;
1084};
1085
1086static void test_clip_expansion(skiatest::Reporter* reporter) {
1087    SkPictureRecorder recorder;
1088    SkCanvas* canvas = recorder.beginRecording(10, 10, NULL, 0);
1089
1090    canvas->clipRect(SkRect::MakeEmpty(), SkRegion::kReplace_Op);
1091    // The following expanding clip should not be skipped.
1092    canvas->clipRect(SkRect::MakeXYWH(4, 4, 3, 3), SkRegion::kUnion_Op);
1093    // Draw something so the optimizer doesn't just fold the world.
1094    SkPaint p;
1095    p.setColor(SK_ColorBLUE);
1096    canvas->drawPaint(p);
1097    SkAutoTUnref<SkPicture> picture(recorder.endRecording());
1098
1099    ClipCountingCanvas testCanvas(10, 10);
1100    picture->draw(&testCanvas);
1101
1102    // Both clips should be present on playback.
1103    REPORTER_ASSERT(reporter, testCanvas.getClipCount() == 2);
1104}
1105
1106static void test_hierarchical(skiatest::Reporter* reporter) {
1107    SkBitmap bm;
1108    make_bm(&bm, 10, 10, SK_ColorRED, true);
1109
1110    SkPictureRecorder recorder;
1111
1112    recorder.beginRecording(10, 10, NULL, 0);
1113    SkAutoTUnref<SkPicture> childPlain(recorder.endRecording());
1114    REPORTER_ASSERT(reporter, !childPlain->willPlayBackBitmaps()); // 0
1115
1116    recorder.beginRecording(10, 10, NULL, 0)->drawBitmap(bm, 0, 0);
1117    SkAutoTUnref<SkPicture> childWithBitmap(recorder.endRecording());
1118    REPORTER_ASSERT(reporter, childWithBitmap->willPlayBackBitmaps()); // 1
1119
1120    {
1121        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL, 0);
1122        canvas->drawPicture(*childPlain);
1123        SkAutoTUnref<SkPicture> parentPP(recorder.endRecording());
1124        REPORTER_ASSERT(reporter, !parentPP->willPlayBackBitmaps()); // 0
1125    }
1126    {
1127        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL, 0);
1128        canvas->drawPicture(*childWithBitmap);
1129        SkAutoTUnref<SkPicture> parentPWB(recorder.endRecording());
1130        REPORTER_ASSERT(reporter, parentPWB->willPlayBackBitmaps()); // 1
1131    }
1132    {
1133        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL, 0);
1134        canvas->drawBitmap(bm, 0, 0);
1135        canvas->drawPicture(*childPlain);
1136        SkAutoTUnref<SkPicture> parentWBP(recorder.endRecording());
1137        REPORTER_ASSERT(reporter, parentWBP->willPlayBackBitmaps()); // 1
1138    }
1139    {
1140        SkCanvas* canvas = recorder.beginRecording(10, 10, NULL, 0);
1141        canvas->drawBitmap(bm, 0, 0);
1142        canvas->drawPicture(*childWithBitmap);
1143        SkAutoTUnref<SkPicture> parentWBWB(recorder.endRecording());
1144        REPORTER_ASSERT(reporter, parentWBWB->willPlayBackBitmaps()); // 2
1145    }
1146}
1147
1148static void test_gen_id(skiatest::Reporter* reporter) {
1149
1150    SkPicture empty;
1151
1152    // Empty pictures should still have a valid ID
1153    REPORTER_ASSERT(reporter, empty.uniqueID() != SK_InvalidGenID);
1154
1155    SkPictureRecorder recorder;
1156
1157    SkCanvas* canvas = recorder.beginRecording(1, 1, NULL, 0);
1158    canvas->drawARGB(255, 255, 255, 255);
1159    SkAutoTUnref<SkPicture> hasData(recorder.endRecording());
1160    // picture should have a non-zero id after recording
1161    REPORTER_ASSERT(reporter, hasData->uniqueID() != SK_InvalidGenID);
1162
1163    // both pictures should have different ids
1164    REPORTER_ASSERT(reporter, hasData->uniqueID() != empty.uniqueID());
1165
1166    // test out copy constructor
1167    SkPicture copyWithData(*hasData);
1168    REPORTER_ASSERT(reporter, hasData->uniqueID() == copyWithData.uniqueID());
1169
1170    SkPicture emptyCopy(empty);
1171    REPORTER_ASSERT(reporter, empty.uniqueID() != emptyCopy.uniqueID());
1172
1173    // test out swap
1174    {
1175        SkPicture swapWithData;
1176        uint32_t beforeID1 = swapWithData.uniqueID();
1177        uint32_t beforeID2 = copyWithData.uniqueID();
1178        swapWithData.swap(copyWithData);
1179        REPORTER_ASSERT(reporter, copyWithData.uniqueID() == beforeID1);
1180        REPORTER_ASSERT(reporter, swapWithData.uniqueID() == beforeID2);
1181    }
1182
1183    // test out clone
1184    {
1185        SkAutoTUnref<SkPicture> cloneWithData(hasData->clone());
1186        REPORTER_ASSERT(reporter, hasData->uniqueID() == cloneWithData->uniqueID());
1187
1188        SkAutoTUnref<SkPicture> emptyClone(empty.clone());
1189        REPORTER_ASSERT(reporter, empty.uniqueID() != emptyClone->uniqueID());
1190    }
1191}
1192
1193DEF_TEST(Picture, reporter) {
1194#ifdef SK_DEBUG
1195    test_deleting_empty_playback();
1196    test_serializing_empty_picture();
1197#else
1198    test_bad_bitmap();
1199#endif
1200    test_unbalanced_save_restores(reporter);
1201    test_peephole();
1202    test_gatherpixelrefs(reporter);
1203    test_gatherpixelrefsandrects(reporter);
1204    test_bitmap_with_encoded_data(reporter);
1205    test_clone_empty(reporter);
1206    test_draw_empty(reporter);
1207    test_clip_bound_opt(reporter);
1208    test_clip_expansion(reporter);
1209    test_hierarchical(reporter);
1210    test_gen_id(reporter);
1211}
1212
1213static void draw_bitmaps(const SkBitmap bitmap, SkCanvas* canvas) {
1214    const SkPaint paint;
1215    const SkRect rect = { 5.0f, 5.0f, 8.0f, 8.0f };
1216    const SkIRect irect =  { 2, 2, 3, 3 };
1217
1218    // Don't care what these record, as long as they're legal.
1219    canvas->drawBitmap(bitmap, 0.0f, 0.0f, &paint);
1220    canvas->drawBitmapRectToRect(bitmap, &rect, rect, &paint, SkCanvas::kNone_DrawBitmapRectFlag);
1221    canvas->drawBitmapMatrix(bitmap, SkMatrix::I(), &paint);
1222    canvas->drawBitmapNine(bitmap, irect, rect, &paint);
1223    canvas->drawSprite(bitmap, 1, 1);
1224}
1225
1226static void test_draw_bitmaps(SkCanvas* canvas) {
1227    SkBitmap empty;
1228    draw_bitmaps(empty, canvas);
1229    empty.setConfig(SkImageInfo::MakeN32Premul(10, 10));
1230    draw_bitmaps(empty, canvas);
1231}
1232
1233DEF_TEST(Picture_EmptyBitmap, r) {
1234    SkPictureRecorder recorder;
1235    test_draw_bitmaps(recorder.beginRecording(10, 10, NULL, 0));
1236    SkAutoTUnref<SkPicture> picture(recorder.endRecording());
1237}
1238
1239DEF_TEST(Canvas_EmptyBitmap, r) {
1240    SkBitmap dst;
1241    dst.allocN32Pixels(10, 10);
1242    SkCanvas canvas(dst);
1243
1244    test_draw_bitmaps(&canvas);
1245}
1246