Home | History | Annotate | Download | only in tests 
      1 /*
      2  * Copyright 2014 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 "Test.h"
      9 
     10 #include "SkPictureRecorder.h"
     11 #include "SkRecord.h"
     12 #include "SkRecorder.h"
     13 #include "SkRecords.h"
     14 #include "SkShader.h"
     15 #include "SkSurface.h"
     16 
     17 #define COUNT(T) + 1
     18 static const int kRecordTypes = SK_RECORD_TYPES(COUNT);
     19 #undef COUNT
     20 
     21 // Tallies the types of commands it sees into a histogram.
     22 class Tally {
     23 public:
     24     Tally() { sk_bzero(&fHistogram, sizeof(fHistogram)); }
     25 
     26     template <typename T>
     27     void operator()(const T&) { ++fHistogram[T::kType]; }
     28 
     29     template <typename T>
     30     int count() const { return fHistogram[T::kType]; }
     31 
     32     void apply(const SkRecord& record) {
     33         for (int i = 0; i < record.count(); i++) {
     34             record.visit<void>(i, *this);
     35         }
     36     }
     37 
     38 private:
     39     int fHistogram[kRecordTypes];
     40 };
     41 
     42 DEF_TEST(Recorder, r) {
     43     SkRecord record;
     44     SkRecorder recorder(&record, 1920, 1080);
     45 
     46     recorder.drawRect(SkRect::MakeWH(10, 10), SkPaint());
     47 
     48     Tally tally;
     49     tally.apply(record);
     50     REPORTER_ASSERT(r, 1 == tally.count<SkRecords::DrawRect>());
     51 }
     52 
     53 // Regression test for leaking refs held by optional arguments.
     54 DEF_TEST(Recorder_RefLeaking, r) {
     55     // We use SaveLayer to test:
     56     //   - its SkRect argument is optional and SkRect is POD.  Just testing that that works.
     57     //   - its SkPaint argument is optional and SkPaint is not POD.  The bug was here.
     58 
     59     SkRect bounds = SkRect::MakeWH(320, 240);
     60     SkPaint paint;
     61     paint.setShader(SkShader::CreateEmptyShader())->unref();
     62 
     63     REPORTER_ASSERT(r, paint.getShader()->unique());
     64     {
     65         SkRecord record;
     66         SkRecorder recorder(&record, 1920, 1080);
     67         recorder.saveLayer(&bounds, &paint);
     68         REPORTER_ASSERT(r, !paint.getShader()->unique());
     69     }
     70     REPORTER_ASSERT(r, paint.getShader()->unique());
     71 }
     72 
     73 DEF_TEST(Recorder_drawImage_takeReference, reporter) {
     74 
     75     SkAutoTUnref<SkImage> image;
     76     {
     77         SkAutoTUnref<SkSurface> surface(SkSurface::NewRasterN32Premul(100, 100));
     78         surface->getCanvas()->clear(SK_ColorGREEN);
     79         image.reset(surface->newImageSnapshot());
     80     }
     81     {
     82         SkRecord record;
     83         SkRecorder recorder(&record, 100, 100);
     84 
     85         // DrawImage is supposed to take a reference
     86         recorder.drawImage(image.get(), 0, 0);
     87         REPORTER_ASSERT(reporter, !image->unique());
     88 
     89         Tally tally;
     90         tally.apply(record);
     91 
     92         REPORTER_ASSERT(reporter, 1 == tally.count<SkRecords::DrawImage>());
     93     }
     94     REPORTER_ASSERT(reporter, image->unique());
     95 
     96     {
     97         SkRecord record;
     98         SkRecorder recorder(&record, 100, 100);
     99 
    100         // DrawImageRect is supposed to take a reference
    101         recorder.drawImageRect(image.get(), SkRect::MakeWH(100, 100), nullptr);
    102         REPORTER_ASSERT(reporter, !image->unique());
    103 
    104         Tally tally;
    105         tally.apply(record);
    106 
    107         REPORTER_ASSERT(reporter, 1 == tally.count<SkRecords::DrawImageRect>());
    108     }
    109     REPORTER_ASSERT(reporter, image->unique());
    110 }
    111