1 /* 2 * Copyright 2013 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 "SkDisplacementMapEffect.h" 10 #include "SkBitmapSource.h" 11 12 namespace skiagm { 13 14 class DisplacementMapGM : public GM { 15 public: 16 DisplacementMapGM() : fInitialized(false) { 17 this->setBGColor(0xFF000000); 18 } 19 20 protected: 21 virtual SkString onShortName() { 22 return SkString("displacement"); 23 } 24 25 void make_bitmaps() { 26 make_bitmap(); 27 make_checkerboard(&fCheckerboard, 80, 80); 28 make_checkerboard(&fSmall, 64, 64); 29 make_checkerboard(&fLarge, 96, 96); 30 make_checkerboard(&fLargeW, 96, 64); 31 make_checkerboard(&fLargeH, 64, 96); 32 } 33 34 void make_bitmap() { 35 fBitmap.allocN32Pixels(80, 80); 36 SkCanvas canvas(fBitmap); 37 canvas.clear(0x00000000); 38 SkPaint paint; 39 paint.setAntiAlias(true); 40 paint.setColor(0xFF884422); 41 paint.setTextSize(SkIntToScalar(96)); 42 const char* str = "g"; 43 canvas.drawText(str, strlen(str), SkIntToScalar(15), SkIntToScalar(55), paint); 44 } 45 46 void make_checkerboard(SkBitmap* bitmap, int w, int h) { 47 bitmap->allocN32Pixels(w, h); 48 SkCanvas canvas(*bitmap); 49 canvas.clear(0x00000000); 50 SkPaint darkPaint; 51 darkPaint.setColor(0xFF804020); 52 SkPaint lightPaint; 53 lightPaint.setColor(0xFF244484); 54 for (int y = 0; y < h; y += 16) { 55 for (int x = 0; x < w; x += 16) { 56 canvas.save(); 57 canvas.translate(SkIntToScalar(x), SkIntToScalar(y)); 58 canvas.drawRect(SkRect::MakeXYWH(0, 0, 8, 8), darkPaint); 59 canvas.drawRect(SkRect::MakeXYWH(8, 0, 8, 8), lightPaint); 60 canvas.drawRect(SkRect::MakeXYWH(0, 8, 8, 8), lightPaint); 61 canvas.drawRect(SkRect::MakeXYWH(8, 8, 8, 8), darkPaint); 62 canvas.restore(); 63 } 64 } 65 } 66 67 virtual SkISize onISize() { 68 return SkISize::Make(500, 500); 69 } 70 71 void drawClippedBitmap(SkCanvas* canvas, int x, int y, const SkPaint& paint) { 72 canvas->save(); 73 canvas->translate(SkIntToScalar(x), SkIntToScalar(y)); 74 canvas->clipRect(SkRect::MakeWH(SkIntToScalar(fBitmap.width()), SkIntToScalar(fBitmap.height()))); 75 canvas->drawBitmap(fBitmap, 0, 0, &paint); 76 canvas->restore(); 77 } 78 79 virtual void onDraw(SkCanvas* canvas) { 80 if (!fInitialized) { 81 make_bitmaps(); 82 fInitialized = true; 83 } 84 canvas->clear(0x00000000); 85 SkPaint paint; 86 SkAutoTUnref<SkImageFilter> displ(SkBitmapSource::Create(fCheckerboard)); 87 paint.setImageFilter(SkDisplacementMapEffect::Create( 88 SkDisplacementMapEffect::kR_ChannelSelectorType, 89 SkDisplacementMapEffect::kG_ChannelSelectorType, 0.0f, displ))->unref(); 90 drawClippedBitmap(canvas, 0, 0, paint); 91 paint.setImageFilter(SkDisplacementMapEffect::Create( 92 SkDisplacementMapEffect::kB_ChannelSelectorType, 93 SkDisplacementMapEffect::kA_ChannelSelectorType, 16.0f, displ))->unref(); 94 drawClippedBitmap(canvas, 100, 0, paint); 95 paint.setImageFilter(SkDisplacementMapEffect::Create( 96 SkDisplacementMapEffect::kR_ChannelSelectorType, 97 SkDisplacementMapEffect::kB_ChannelSelectorType, 32.0f, displ))->unref(); 98 drawClippedBitmap(canvas, 200, 0, paint); 99 paint.setImageFilter(SkDisplacementMapEffect::Create( 100 SkDisplacementMapEffect::kG_ChannelSelectorType, 101 SkDisplacementMapEffect::kA_ChannelSelectorType, 48.0f, displ))->unref(); 102 drawClippedBitmap(canvas, 300, 0, paint); 103 paint.setImageFilter(SkDisplacementMapEffect::Create( 104 SkDisplacementMapEffect::kR_ChannelSelectorType, 105 SkDisplacementMapEffect::kA_ChannelSelectorType, 64.0f, displ))->unref(); 106 drawClippedBitmap(canvas, 400, 0, paint); 107 108 paint.setImageFilter(SkDisplacementMapEffect::Create( 109 SkDisplacementMapEffect::kR_ChannelSelectorType, 110 SkDisplacementMapEffect::kG_ChannelSelectorType, 40.0f, displ))->unref(); 111 drawClippedBitmap(canvas, 0, 100, paint); 112 paint.setImageFilter(SkDisplacementMapEffect::Create( 113 SkDisplacementMapEffect::kB_ChannelSelectorType, 114 SkDisplacementMapEffect::kA_ChannelSelectorType, 40.0f, displ))->unref(); 115 drawClippedBitmap(canvas, 100, 100, paint); 116 paint.setImageFilter(SkDisplacementMapEffect::Create( 117 SkDisplacementMapEffect::kR_ChannelSelectorType, 118 SkDisplacementMapEffect::kB_ChannelSelectorType, 40.0f, displ))->unref(); 119 drawClippedBitmap(canvas, 200, 100, paint); 120 paint.setImageFilter(SkDisplacementMapEffect::Create( 121 SkDisplacementMapEffect::kG_ChannelSelectorType, 122 SkDisplacementMapEffect::kA_ChannelSelectorType, 40.0f, displ))->unref(); 123 drawClippedBitmap(canvas, 300, 100, paint); 124 paint.setImageFilter(SkDisplacementMapEffect::Create( 125 SkDisplacementMapEffect::kR_ChannelSelectorType, 126 SkDisplacementMapEffect::kA_ChannelSelectorType, 40.0f, displ))->unref(); 127 drawClippedBitmap(canvas, 400, 100, paint); 128 129 SkImageFilter::CropRect cropRect(SkRect::MakeXYWH(30, 30, 40, 40)); 130 paint.setImageFilter(SkDisplacementMapEffect::Create( 131 SkDisplacementMapEffect::kR_ChannelSelectorType, 132 SkDisplacementMapEffect::kG_ChannelSelectorType, 133 0.0f, displ, NULL, &cropRect))->unref(); 134 drawClippedBitmap(canvas, 0, 200, paint); 135 paint.setImageFilter(SkDisplacementMapEffect::Create( 136 SkDisplacementMapEffect::kB_ChannelSelectorType, 137 SkDisplacementMapEffect::kA_ChannelSelectorType, 138 16.0f, displ, NULL, &cropRect))->unref(); 139 drawClippedBitmap(canvas, 100, 200, paint); 140 paint.setImageFilter(SkDisplacementMapEffect::Create( 141 SkDisplacementMapEffect::kR_ChannelSelectorType, 142 SkDisplacementMapEffect::kB_ChannelSelectorType, 143 32.0f, displ, NULL, &cropRect))->unref(); 144 drawClippedBitmap(canvas, 200, 200, paint); 145 paint.setImageFilter(SkDisplacementMapEffect::Create( 146 SkDisplacementMapEffect::kG_ChannelSelectorType, 147 SkDisplacementMapEffect::kA_ChannelSelectorType, 148 48.0f, displ, NULL, &cropRect))->unref(); 149 drawClippedBitmap(canvas, 300, 200, paint); 150 paint.setImageFilter(SkDisplacementMapEffect::Create( 151 SkDisplacementMapEffect::kR_ChannelSelectorType, 152 SkDisplacementMapEffect::kA_ChannelSelectorType, 153 64.0f, displ, NULL, &cropRect))->unref(); 154 drawClippedBitmap(canvas, 400, 200, paint); 155 156 paint.setImageFilter(SkDisplacementMapEffect::Create( 157 SkDisplacementMapEffect::kR_ChannelSelectorType, 158 SkDisplacementMapEffect::kG_ChannelSelectorType, 159 40.0f, displ, NULL, &cropRect))->unref(); 160 drawClippedBitmap(canvas, 0, 300, paint); 161 paint.setImageFilter(SkDisplacementMapEffect::Create( 162 SkDisplacementMapEffect::kB_ChannelSelectorType, 163 SkDisplacementMapEffect::kA_ChannelSelectorType, 164 40.0f, displ, NULL, &cropRect))->unref(); 165 drawClippedBitmap(canvas, 100, 300, paint); 166 paint.setImageFilter(SkDisplacementMapEffect::Create( 167 SkDisplacementMapEffect::kR_ChannelSelectorType, 168 SkDisplacementMapEffect::kB_ChannelSelectorType, 169 40.0f, displ, NULL, &cropRect))->unref(); 170 drawClippedBitmap(canvas, 200, 300, paint); 171 paint.setImageFilter(SkDisplacementMapEffect::Create( 172 SkDisplacementMapEffect::kG_ChannelSelectorType, 173 SkDisplacementMapEffect::kA_ChannelSelectorType, 174 40.0f, displ, NULL, &cropRect))->unref(); 175 drawClippedBitmap(canvas, 300, 300, paint); 176 paint.setImageFilter(SkDisplacementMapEffect::Create( 177 SkDisplacementMapEffect::kR_ChannelSelectorType, 178 SkDisplacementMapEffect::kA_ChannelSelectorType, 179 40.0f, displ, NULL, &cropRect))->unref(); 180 drawClippedBitmap(canvas, 400, 300, paint); 181 182 // Tests for images of different sizes 183 displ.reset(SkBitmapSource::Create(fSmall)); 184 paint.setImageFilter(SkDisplacementMapEffect::Create( 185 SkDisplacementMapEffect::kR_ChannelSelectorType, 186 SkDisplacementMapEffect::kG_ChannelSelectorType, 187 40.0f, displ))->unref(); 188 drawClippedBitmap(canvas, 0, 400, paint); 189 displ.reset(SkBitmapSource::Create(fLarge)); 190 paint.setImageFilter(SkDisplacementMapEffect::Create( 191 SkDisplacementMapEffect::kB_ChannelSelectorType, 192 SkDisplacementMapEffect::kA_ChannelSelectorType, 193 40.0f, displ))->unref(); 194 drawClippedBitmap(canvas, 100, 400, paint); 195 displ.reset(SkBitmapSource::Create(fLargeW)); 196 paint.setImageFilter(SkDisplacementMapEffect::Create( 197 SkDisplacementMapEffect::kR_ChannelSelectorType, 198 SkDisplacementMapEffect::kB_ChannelSelectorType, 199 40.0f, displ))->unref(); 200 drawClippedBitmap(canvas, 200, 400, paint); 201 displ.reset(SkBitmapSource::Create(fLargeH)); 202 paint.setImageFilter(SkDisplacementMapEffect::Create( 203 SkDisplacementMapEffect::kG_ChannelSelectorType, 204 SkDisplacementMapEffect::kA_ChannelSelectorType, 205 40.0f, displ))->unref(); 206 drawClippedBitmap(canvas, 300, 400, paint); 207 208 // Test for no given displacement input. In this case, both displacement 209 // and color should use the same bitmap, given to SkCanvas::drawBitmap() 210 // as an input argument. 211 paint.setImageFilter(SkDisplacementMapEffect::Create( 212 SkDisplacementMapEffect::kG_ChannelSelectorType, 213 SkDisplacementMapEffect::kA_ChannelSelectorType, 214 40.0f, NULL))->unref(); 215 drawClippedBitmap(canvas, 400, 400, paint); 216 } 217 218 private: 219 typedef GM INHERITED; 220 SkBitmap fBitmap, fCheckerboard, fSmall, fLarge, fLargeW, fLargeH; 221 bool fInitialized; 222 }; 223 224 ////////////////////////////////////////////////////////////////////////////// 225 226 static GM* MyFactory(void*) { return new DisplacementMapGM; } 227 static GMRegistry reg(MyFactory); 228 229 } 230
