95 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			95 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			C++
		
	
	
	
| /*
 | |
|  * Copyright 2016 Google Inc.
 | |
|  *
 | |
|  * Use of this source code is governed by a BSD-style license that can be
 | |
|  * found in the LICENSE file.
 | |
|  */
 | |
| 
 | |
| #include "gm/gm.h"
 | |
| #include "include/core/SkCanvas.h"
 | |
| #include "include/core/SkColorSpace.h"
 | |
| #include "include/core/SkImage.h"
 | |
| #include "include/core/SkImageGenerator.h"
 | |
| #include "include/core/SkImageInfo.h"
 | |
| #include "include/core/SkPaint.h"
 | |
| #include "include/core/SkRect.h"
 | |
| #include "include/core/SkRefCnt.h"
 | |
| #include "include/core/SkSize.h"
 | |
| #include "include/core/SkSurface.h"
 | |
| #include "include/core/SkTypes.h"
 | |
| #include "include/gpu/GrDirectContext.h"
 | |
| #include "tools/ToolUtils.h"
 | |
| 
 | |
| namespace {
 | |
| 
 | |
| const SkISize   kSize = SkISize::Make(100, 100);
 | |
| const SkIRect kSubset = SkIRect::MakeLTRB(25, 25, 75, 75);
 | |
| const SkRect    kDest = SkRect::MakeXYWH(10, 10, 100, 100);
 | |
| 
 | |
| sk_sp<SkImage> make_mask(const sk_sp<SkSurface>& surface) {
 | |
|     ToolUtils::draw_checkerboard(surface->getCanvas(), 0x80808080, 0x00000000, 5);
 | |
|     return surface->makeImageSnapshot();
 | |
| }
 | |
| 
 | |
| class MaskGenerator final : public SkImageGenerator {
 | |
| public:
 | |
|     MaskGenerator(const SkImageInfo& info) : INHERITED(info) {}
 | |
| 
 | |
|     bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes, const Options&)
 | |
|     override {
 | |
|         SkImageInfo surfaceInfo = info;
 | |
|         if (kAlpha_8_SkColorType == info.colorType()) {
 | |
|             surfaceInfo = surfaceInfo.makeColorSpace(nullptr);
 | |
|         }
 | |
| 
 | |
|         make_mask(SkSurface::MakeRasterDirect(surfaceInfo, pixels, rowBytes));
 | |
|         return true;
 | |
|     }
 | |
| 
 | |
| private:
 | |
|     using INHERITED = SkImageGenerator;
 | |
| };
 | |
| 
 | |
| using MakerT = sk_sp<SkImage>(*)(SkCanvas*, const SkImageInfo&);
 | |
| const MakerT makers[] = {
 | |
|     // SkImage_Raster
 | |
|     [](SkCanvas*, const SkImageInfo& info) -> sk_sp<SkImage> {
 | |
|         return make_mask(SkSurface::MakeRaster(info));
 | |
|     },
 | |
| 
 | |
|     // SkImage_Gpu
 | |
|     [](SkCanvas* c, const SkImageInfo& info) -> sk_sp<SkImage> {
 | |
|         sk_sp<SkSurface> surface;
 | |
|         surface = SkSurface::MakeRenderTarget(c->recordingContext(), SkBudgeted::kNo, info);
 | |
|         return make_mask(surface ? surface : SkSurface::MakeRaster(info));
 | |
|     },
 | |
| 
 | |
|     // SkImage_Lazy
 | |
|     [](SkCanvas*, const SkImageInfo& info) -> sk_sp<SkImage> {
 | |
|         return SkImage::MakeFromGenerator(std::make_unique<MaskGenerator>(info));
 | |
|     },
 | |
| };
 | |
| 
 | |
| }  // namespace
 | |
| 
 | |
| // Checks whether subset SkImages preserve the original color type (A8 in this case).
 | |
| DEF_SIMPLE_GM(imagemasksubset, canvas, 480, 480) {
 | |
|     SkPaint paint;
 | |
|     paint.setColor(0xff00ff00);
 | |
| 
 | |
|     const SkImageInfo info = SkImageInfo::MakeA8(kSize.width(), kSize.height());
 | |
| 
 | |
|     for (size_t i = 0; i < SK_ARRAY_COUNT(makers); ++i) {
 | |
|         sk_sp<SkImage> image = makers[i](canvas, info);
 | |
|         if (image) {
 | |
|             canvas->drawImageRect(image, SkRect::Make(kSubset), kDest, SkSamplingOptions(),
 | |
|                                   &paint, SkCanvas::kStrict_SrcRectConstraint);
 | |
|             auto direct = GrAsDirectContext(canvas->recordingContext());
 | |
|             sk_sp<SkImage> subset = image->makeSubset(kSubset, direct);
 | |
|             canvas->drawImageRect(subset, kDest.makeOffset(kSize.width() * 1.5f, 0),
 | |
|                                   SkSamplingOptions(), &paint);
 | |
|         }
 | |
|         canvas->translate(0, kSize.height() * 1.5f);
 | |
|     }
 | |
| }
 |