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