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 17 namespace skiagm { 18 extern GrContext* GetGr(); 19 }; 20 #endif 21 22 static 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 33 static 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 48 static 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 68 static 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 115 class 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 }; 124 public: 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 136 protected: 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 220 private: 221 typedef skiagm::GM INHERITED; 222 }; 223 224 ////////////////////////////////////////////////////////////////////////////// 225 226 static skiagm::GM* MyFactory(void*) { return new ImageGM; } 227 static skiagm::GMRegistry reg(MyFactory); 228
