1/*
2 * Copyright 2013 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 "SkBitmap.h"
9#include "SkCanvas.h"
10#include "SkDocument.h"
11#include "SkImageInfo.h"
12#include "SkPixelRef.h"
13#include "SkRefCnt.h"
14#include "SkStream.h"
15
16#include "Test.h"
17
18namespace {
19
20// SkPixelRef which fails to lock, as a lazy pixel ref might if its pixels
21// cannot be generated.
22class InvalidPixelRef : public SkPixelRef {
23public:
24    InvalidPixelRef(const SkImageInfo& info) : SkPixelRef(info) {}
25private:
26    bool onNewLockPixels(LockRec*) override { return false; }
27    void onUnlockPixels() override {
28        SkDEBUGFAIL("InvalidPixelRef can't be locked");
29    }
30};
31
32SkBitmap make_invalid_bitmap(const SkImageInfo& imageInfo) {
33    SkBitmap bitmap;
34    bitmap.setInfo(imageInfo);
35    bitmap.setPixelRef(new InvalidPixelRef(imageInfo))->unref();
36    return bitmap;
37}
38
39SkBitmap make_invalid_bitmap(SkColorType colorType) {
40    return make_invalid_bitmap(
41        SkImageInfo::Make(100, 100, colorType, kPremul_SkAlphaType));
42}
43
44}  // namespace
45
46DEF_TEST(PDFInvalidBitmap, reporter) {
47    SkDynamicMemoryWStream stream;
48    SkAutoTUnref<SkDocument> document(SkDocument::CreatePDF(&stream));
49    SkCanvas* canvas = document->beginPage(100, 100);
50
51    canvas->drawBitmap(SkBitmap(), 0, 0);
52    canvas->drawBitmap(make_invalid_bitmap(SkImageInfo()), 0, 0);
53    canvas->drawBitmap(make_invalid_bitmap(kN32_SkColorType), 0, 0);
54    canvas->drawBitmap(make_invalid_bitmap(kIndex_8_SkColorType), 0, 0);
55    canvas->drawBitmap(make_invalid_bitmap(kARGB_4444_SkColorType), 0, 0);
56    canvas->drawBitmap(make_invalid_bitmap(kRGB_565_SkColorType), 0, 0);
57    canvas->drawBitmap(make_invalid_bitmap(kAlpha_8_SkColorType), 0, 0);
58
59    // This test passes if it does not crash.
60}
61