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 "SkImageDecoder.h"
9#include "SkImage_Base.h"
10#include "SkBitmap.h"
11#include "SkCanvas.h"
12#include "SkData.h"
13
14class SkImage_Codec : public SkImage_Base {
15public:
16    static SkImage* NewEmpty();
17
18    SkImage_Codec(SkData* encodedData, int width, int height);
19    virtual ~SkImage_Codec();
20
21    virtual void onDraw(SkCanvas*, SkScalar, SkScalar, const SkPaint*) SK_OVERRIDE;
22    virtual void onDrawRectToRect(SkCanvas*, const SkRect*, const SkRect&, const SkPaint*) SK_OVERRIDE;
23
24private:
25    SkData*     fEncodedData;
26    SkBitmap    fBitmap;
27
28    typedef SkImage_Base INHERITED;
29};
30
31///////////////////////////////////////////////////////////////////////////////
32
33SkImage_Codec::SkImage_Codec(SkData* data, int width, int height) : INHERITED(width, height) {
34    fEncodedData = data;
35    fEncodedData->ref();
36}
37
38SkImage_Codec::~SkImage_Codec() {
39    fEncodedData->unref();
40}
41
42void SkImage_Codec::onDraw(SkCanvas* canvas, SkScalar x, SkScalar y, const SkPaint* paint) {
43    if (!fBitmap.pixelRef()) {
44        if (!SkImageDecoder::DecodeMemory(fEncodedData->bytes(), fEncodedData->size(), &fBitmap)) {
45            return;
46        }
47    }
48    canvas->drawBitmap(fBitmap, x, y, paint);
49}
50
51void SkImage_Codec::onDrawRectToRect(SkCanvas* canvas, const SkRect* src,
52                                     const SkRect& dst, const SkPaint* paint) {
53    if (!fBitmap.pixelRef()) {
54        if (!SkImageDecoder::DecodeMemory(fEncodedData->bytes(), fEncodedData->size(), &fBitmap)) {
55            return;
56        }
57    }
58    canvas->drawBitmapRectToRect(fBitmap, src, dst, paint);
59}
60
61///////////////////////////////////////////////////////////////////////////////
62
63SkImage* SkImage::NewEncodedData(SkData* data) {
64    if (NULL == data) {
65        return NULL;
66    }
67
68    SkBitmap bitmap;
69    if (!SkImageDecoder::DecodeMemory(data->bytes(), data->size(), &bitmap, kUnknown_SkColorType,
70                                      SkImageDecoder::kDecodeBounds_Mode)) {
71        return NULL;
72    }
73
74    return SkNEW_ARGS(SkImage_Codec, (data, bitmap.width(), bitmap.height()));
75}
76