image.cpp revision dfec28d4a9e05d1d525f377b380b4df3c5e07c7b 
1/*
2 * Copyright 2011 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 "gm.h"
9#include "SkSurface.h"
10#include "SkCanvas.h"
11#include "SkStream.h"
12#include "SkData.h"
13
14#if SK_SUPPORT_GPU
15#include "GrContext.h"
16
17namespace skiagm {
18extern GrContext* GetGr();
19};
20#endif
21
22static SkData* fileToData(const char path[]) {
23    SkFILEStream stream(path);
24    if (!stream.isValid()) {
25        return SkData::NewEmpty();
26    }
27    size_t size = stream.getLength();
28    void* mem = sk_malloc_throw(size);
29    stream.read(mem, size);
30    return SkData::NewFromMalloc(mem, size);
31}
32
33static void drawJpeg(SkCanvas* canvas, const SkISize& size) {
34    // TODO: Make this draw a file that is checked in, so it can
35    // be exercised on machines other than mike's. Will require a
36    // rebaseline.
37    SkAutoDataUnref data(fileToData("/Users/mike/Downloads/skia.google.jpeg"));
38    SkImage* image = SkImage::NewEncodedData(data);
39    if (image) {
40        SkAutoCanvasRestore acr(canvas, true);
41        canvas->scale(size.width() * 1.0f / image->width(),
42                      size.height() * 1.0f / image->height());
43        image->draw(canvas, 0, 0, NULL);
44        image->unref();
45    }
46}
47
48static void drawContents(SkSurface* surface, SkColor fillC) {
49    SkSize size = SkSize::Make(SkIntToScalar(surface->width()),
50                               SkIntToScalar(surface->height()));
51    SkCanvas* canvas = surface->getCanvas();
52
53    SkScalar stroke = size.fWidth / 10;
54    SkScalar radius = (size.fWidth - stroke) / 2;
55
56    SkPaint paint;
57
58    paint.setAntiAlias(true);
59    paint.setColor(fillC);
60    canvas->drawCircle(size.fWidth/2, size.fHeight/2, radius, paint);
61
62    paint.setStyle(SkPaint::kStroke_Style);
63    paint.setStrokeWidth(stroke);
64    paint.setColor(SK_ColorBLACK);
65    canvas->drawCircle(size.fWidth/2, size.fHeight/2, radius, paint);
66}
67
68static void test_surface(SkCanvas* canvas, SkSurface* surf, bool usePaint) {
69    drawContents(surf, SK_ColorRED);
70    SkImage* imgR = surf->newImageSnapshot();
71
72    if (true) {
73        SkImage* imgR2 = surf->newImageSnapshot();
74        SkASSERT(imgR == imgR2);
75        imgR2->unref();
76    }
77
78    drawContents(surf, SK_ColorGREEN);
79    SkImage* imgG = surf->newImageSnapshot();
80
81    // since we've drawn after we snapped imgR, imgG will be a different obj
82    SkASSERT(imgR != imgG);
83
84    drawContents(surf, SK_ColorBLUE);
85
86    SkPaint paint;
87//    paint.setFilterBitmap(true);
88//    paint.setAlpha(0x80);
89
90    imgR->draw(canvas, 0, 0, usePaint ? &paint : NULL);
91    imgG->draw(canvas, 0, 80, usePaint ? &paint : NULL);
92    surf->draw(canvas, 0, 160, usePaint ? &paint : NULL);
93
94    SkRect src1, src2, src3;
95    src1.iset(0, 0, surf->width(), surf->height());
96    src2.iset(-surf->width() / 2, -surf->height() / 2,
97             surf->width(), surf->height());
98    src3.iset(0, 0, surf->width() / 2, surf->height() / 2);
99
100    SkRect dst1, dst2, dst3, dst4;
101    dst1.set(0, 240, 65, 305);
102    dst2.set(0, 320, 65, 385);
103    dst3.set(0, 400, 65, 465);
104    dst4.set(0, 480, 65, 545);
105
106    imgR->draw(canvas, &src1, dst1, usePaint ? &paint : NULL);
107    imgG->draw(canvas, &src2, dst2, usePaint ? &paint : NULL);
108    imgR->draw(canvas, &src3, dst3, usePaint ? &paint : NULL);
109    imgG->draw(canvas, NULL, dst4, usePaint ? &paint : NULL);
110
111    imgG->unref();
112    imgR->unref();
113}
114
115class ImageGM : public skiagm::GM {
116    void*   fBuffer;
117    size_t  fBufferSize;
118    SkSize  fSize;
119    enum {
120        W = 64,
121        H = 64,
122        RB = W * 4 + 8,
123    };
124public:
125    ImageGM() {
126        fBufferSize = RB * H;
127        fBuffer = sk_malloc_throw(fBufferSize);
128        fSize.set(SkIntToScalar(W), SkIntToScalar(H));
129    }
130
131    virtual ~ImageGM() {
132        sk_free(fBuffer);
133    }
134
135
136protected:
137    virtual SkString onShortName() {
138        return SkString("image-surface");
139    }
140
141    virtual SkISize onISize() {
142        return SkISize::Make(960, 1200);
143    }
144
145    virtual void onDraw(SkCanvas* canvas) {
146        drawJpeg(canvas, this->getISize());
147
148        canvas->scale(2, 2);
149
150        static const char* kLabel1 = "Original Img";
151        static const char* kLabel2 = "Modified Img";
152        static const char* kLabel3 = "Cur Surface";
153        static const char* kLabel4 = "Full Crop";
154        static const char* kLabel5 = "Over-crop";
155        static const char* kLabel6 = "Upper-left";
156        static const char* kLabel7 = "No Crop";
157
158        static const char* kLabel8 = "Pre-Alloc Img";
159        static const char* kLabel9 = "New Alloc Img";
160        static const char* kLabel10 = "SkPicture";
161        static const char* kLabel11 = "Null Paint";
162        static const char* kLabel12 = "GPU";
163
164        SkPaint textPaint;
165
166        canvas->drawText(kLabel1, strlen(kLabel1), 10,  60, textPaint);
167        canvas->drawText(kLabel2, strlen(kLabel2), 10, 140, textPaint);
168        canvas->drawText(kLabel3, strlen(kLabel3), 10, 220, textPaint);
169        canvas->drawText(kLabel4, strlen(kLabel4), 10, 300, textPaint);
170        canvas->drawText(kLabel5, strlen(kLabel5), 10, 380, textPaint);
171        canvas->drawText(kLabel6, strlen(kLabel6), 10, 460, textPaint);
172        canvas->drawText(kLabel7, strlen(kLabel7), 10, 540, textPaint);
173
174        canvas->drawText(kLabel8, strlen(kLabel8),  80, 10, textPaint);
175        canvas->drawText(kLabel9, strlen(kLabel9), 160, 10, textPaint);
176        canvas->drawText(kLabel10, strlen(kLabel10), 250, 10, textPaint);
177        canvas->drawText(kLabel11, strlen(kLabel11), 320, 10, textPaint);
178        canvas->drawText(kLabel12, strlen(kLabel12), 410, 10, textPaint);
179
180        canvas->translate(80, 20);
181
182        // since we draw into this directly, we need to start fresh
183        sk_bzero(fBuffer, fBufferSize);
184
185        SkImage::Info info;
186
187        info.fWidth = W;
188        info.fHeight = H;
189        info.fColorType = SkImage::kPMColor_ColorType;
190        info.fAlphaType = SkImage::kPremul_AlphaType;
191        SkAutoTUnref<SkSurface> surf0(SkSurface::NewRasterDirect(info, fBuffer, RB));
192        SkAutoTUnref<SkSurface> surf1(SkSurface::NewRaster(info));
193        SkAutoTUnref<SkSurface> surf2(SkSurface::NewPicture(info.fWidth, info.fHeight));
194        SkAutoTUnref<SkSurface> surf3(SkSurface::NewPicture(info.fWidth, info.fHeight));
195#if SK_SUPPORT_GPU
196        GrContext* ctx = skiagm::GetGr();
197
198        SkAutoTUnref<SkSurface> surf4(SkSurface::NewRenderTarget(ctx, info, 0));
199#endif
200
201        test_surface(canvas, surf0, true);
202        canvas->translate(80, 0);
203        test_surface(canvas, surf1, true);
204        canvas->translate(80, 0);
205        test_surface(canvas, surf2, true);
206        canvas->translate(80, 0);
207        test_surface(canvas, surf3, false);
208#if SK_SUPPORT_GPU
209        if (NULL != ctx) {
210            canvas->translate(80, 0);
211            test_surface(canvas, surf4, true);
212        }
213#endif
214    }
215
216    virtual uint32_t onGetFlags() const SK_OVERRIDE {
217        return GM::kSkipPicture_Flag | GM::kSkipPipe_Flag;
218    }
219
220private:
221    typedef skiagm::GM INHERITED;
222};
223
224//////////////////////////////////////////////////////////////////////////////
225
226static skiagm::GM* MyFactory(void*) { return new ImageGM; }
227static skiagm::GMRegistry reg(MyFactory);
228