1/*
2 * Copyright 2015 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 "SkImageGeneratorUtils.h"
9#include "SkBitmap.h"
10
11class GeneratorFromEmpty : public SkImageGenerator {
12public:
13    GeneratorFromEmpty(const SkImageInfo& info) : SkImageGenerator(info) {}
14};
15
16SkImageGenerator* SkImageGeneratorUtils::NewEmpty(const SkImageInfo& info) {
17    return new GeneratorFromEmpty(info);
18}
19
20///////////////////////////////////////////////////////////////////////////////////////////////////
21
22class GeneratorFromBitmap : public SkImageGenerator {
23public:
24    GeneratorFromBitmap(const SkBitmap& bm) : SkImageGenerator(bm.info()), fBM(bm) {
25        if (!bm.isImmutable()) {
26            bm.copyTo(&fBM);
27        }
28    }
29
30protected:
31    bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes,
32                     SkPMColor*, int*) override {
33        return fBM.readPixels(info, pixels, rowBytes, 0, 0);
34    }
35
36private:
37    SkBitmap fBM;
38};
39
40SkImageGenerator* SkImageGeneratorUtils::NewFromBitmap(const SkBitmap& bm) {
41    return new GeneratorFromBitmap(bm);
42}
43
44///////////////////////////////////////////////////////////////////////////////////////////////////
45
46#if SK_SUPPORT_GPU
47
48#include "GrContext.h"
49#include "GrTexture.h"
50#include "SkGrPriv.h"
51
52class GeneratorFromTexture : public SkImageGenerator {
53public:
54    GeneratorFromTexture(GrContext* ctx, GrTexture* tex, const SkImageInfo& info)
55        : SkImageGenerator(info), fCtx(ctx), fTexture(tex)
56    {}
57
58protected:
59    GrTexture* onGenerateTexture(GrContext* ctx, const SkIRect* subset) override {
60        if (ctx) {
61            SkASSERT(ctx == fCtx.get());
62        }
63
64        if (!subset) {
65            return SkRef(fTexture.get());
66        }
67        // need to copy the subset into a new texture
68        GrSurfaceDesc desc = fTexture->desc();
69        desc.fWidth = subset->width();
70        desc.fHeight = subset->height();
71
72        GrTexture* dst = fCtx->textureProvider()->createTexture(desc, SkBudgeted::kNo);
73        fCtx->copySurface(dst, fTexture, *subset, SkIPoint::Make(0, 0));
74        return dst;
75    }
76private:
77    SkAutoTUnref<GrContext> fCtx;
78    SkAutoTUnref<GrTexture> fTexture;
79};
80#endif
81
82SkImageGenerator* SkImageGeneratorUtils::NewFromTexture(GrContext* ctx, GrTexture* tex) {
83#if SK_SUPPORT_GPU
84    if (ctx && tex) {
85        const GrSurfaceDesc desc = tex->desc();
86
87        SkColorType ct;
88        SkColorProfileType cpt;
89        if (!GrPixelConfig2ColorAndProfileType(desc.fConfig, &ct, &cpt)) {
90            return nullptr;
91        }
92        const SkAlphaType at = kPremul_SkAlphaType; // take isOpaque from caller?
93        SkImageInfo info = SkImageInfo::Make(desc.fWidth, desc.fHeight, ct, at, cpt);
94        return new GeneratorFromTexture(ctx, tex, info);
95    }
96#endif
97    return nullptr;
98}
99
100///////////////////////////////////////////////////////////////////////////////////////////////////
101
102#include "SkImage.h"
103
104class GeneratorFromImage : public SkImageGenerator {
105public:
106    GeneratorFromImage(const SkImage* image, const SkImageInfo& info)
107        : SkImageGenerator(info), fImage(image) {}
108
109protected:
110    bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes,
111                     SkPMColor*, int*) override {
112        return fImage->readPixels(info, pixels, rowBytes, 0, 0);
113    }
114
115    GrTexture* onGenerateTexture(GrContext* ctx, const SkIRect* subset) override {
116        // waiting on https://code.google.com/p/skia/issues/detail?id=4233
117        return nullptr;
118    }
119
120private:
121    SkAutoTUnref<const SkImage> fImage;
122};
123
124SkImageGenerator* SkImageGeneratorUtils::NewFromImage(const SkImage* image) {
125    if (image) {
126        const SkColorType ct = kN32_SkColorType;
127        const SkAlphaType at = image->isOpaque() ? kOpaque_SkAlphaType : kPremul_SkAlphaType;
128        const SkImageInfo info = SkImageInfo::Make(image->width(), image->height(), ct, at);
129        return new GeneratorFromImage(image, info);
130    }
131    return nullptr;
132}
133
134
135