blob: 0752e487bcd2ed783a4dbe1b9fc68fe898447185 [file] [log] [blame]
Brian Osman3b66ab62016-11-28 09:26:31 -05001/*
2 * Copyright 2016 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
Mike Kleinc0bd9f92019-04-23 12:05:21 -05008#include "src/gpu/GrImageTextureMaker.h"
Brian Salomone7499c72019-06-24 12:12:36 -04009
10#include "src/gpu/GrColorSpaceXform.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050011#include "src/gpu/SkGr.h"
12#include "src/gpu/effects/GrYUVtoRGBEffect.h"
13#include "src/image/SkImage_GpuYUVA.h"
14#include "src/image/SkImage_Lazy.h"
Brian Osman3b66ab62016-11-28 09:26:31 -050015
Brian Salomone7499c72019-06-24 12:12:36 -040016static GrColorSpaceInfo make_info(const SkImage*& image) {
Brian Salomond6287472019-06-24 15:50:07 -040017 return GrColorSpaceInfo(SkColorTypeToGrColorType(image->colorType()),
18 image->alphaType(),
19 image->refColorSpace(),
Brian Salomone7499c72019-06-24 12:12:36 -040020 SkImageInfo2GrPixelConfig(image->imageInfo()));
21}
22
Robert Phillips9338c602019-02-19 12:52:29 -050023GrImageTextureMaker::GrImageTextureMaker(GrRecordingContext* context, const SkImage* client,
Michael Ludwigddeed372019-02-20 16:50:10 -050024 SkImage::CachingHint chint, bool useDecal)
Brian Salomone7499c72019-06-24 12:12:36 -040025 : INHERITED(context, client->width(), client->height(), make_info(client), useDecal)
Brian Osmanbd659552018-09-11 10:03:19 -040026 , fImage(static_cast<const SkImage_Lazy*>(client))
Brian Osmandf7e0752017-04-26 16:20:28 -040027 , fCachingHint(chint) {
Brian Osmanbd659552018-09-11 10:03:19 -040028 SkASSERT(client->isLazyGenerated());
Brian Osmandf7e0752017-04-26 16:20:28 -040029 GrMakeKeyFromImageID(&fOriginalKey, client->uniqueID(),
30 SkIRect::MakeWH(this->width(), this->height()));
Brian Osman3b66ab62016-11-28 09:26:31 -050031}
32
Robert Phillips0c984a02017-03-16 07:51:56 -040033sk_sp<GrTextureProxy> GrImageTextureMaker::refOriginalTextureProxy(bool willBeMipped,
Stan Ilievba81af22017-06-08 15:16:53 -040034 AllowedTexGenType onlyIfFast) {
Brian Osmanbd659552018-09-11 10:03:19 -040035 return fImage->lockTextureProxy(this->context(), fOriginalKey, fCachingHint,
Brian Osmane7fd8c32018-10-19 13:30:39 -040036 willBeMipped, onlyIfFast);
Robert Phillips0c984a02017-03-16 07:51:56 -040037}
38
Brian Osmanb3f38302018-09-07 15:24:44 -040039void GrImageTextureMaker::makeCopyKey(const CopyParams& stretch, GrUniqueKey* paramsCopyKey) {
Brian Osman3b66ab62016-11-28 09:26:31 -050040 if (fOriginalKey.isValid() && SkImage::kAllow_CachingHint == fCachingHint) {
Brian Osman10494e32018-09-10 12:45:18 -040041 GrUniqueKey cacheKey;
Brian Osmanbd659552018-09-11 10:03:19 -040042 fImage->makeCacheKeyFromOrigKey(fOriginalKey, &cacheKey);
Brian Osman10494e32018-09-10 12:45:18 -040043 MakeCopyKeyFromOrigKey(cacheKey, stretch, paramsCopyKey);
Brian Osman3b66ab62016-11-28 09:26:31 -050044 }
45}
46
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040047
48/////////////////////////////////////////////////////////////////////////////////////////////////
49
Michael Ludwigddeed372019-02-20 16:50:10 -050050GrYUVAImageTextureMaker::GrYUVAImageTextureMaker(GrContext* context, const SkImage* client,
51 bool useDecal)
Brian Salomone7499c72019-06-24 12:12:36 -040052 : INHERITED(context, client->width(), client->height(), make_info(client), useDecal)
53 , fImage(static_cast<const SkImage_GpuYUVA*>(client)) {
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040054 SkASSERT(as_IB(client)->isYUVA());
55 GrMakeKeyFromImageID(&fOriginalKey, client->uniqueID(),
56 SkIRect::MakeWH(this->width(), this->height()));
57}
58
59sk_sp<GrTextureProxy> GrYUVAImageTextureMaker::refOriginalTextureProxy(bool willBeMipped,
60 AllowedTexGenType onlyIfFast) {
Jim Van Verth7da46762018-11-05 14:24:23 -050061 if (AllowedTexGenType::kCheap == onlyIfFast) {
62 return nullptr;
63 }
64
Jim Van Verth803a5022018-11-05 15:55:53 -050065 if (willBeMipped) {
Robert Phillips6603a172019-03-05 12:35:44 -050066 return fImage->asMippedTextureProxyRef(this->context());
Jim Van Verth803a5022018-11-05 15:55:53 -050067 } else {
Robert Phillips6603a172019-03-05 12:35:44 -050068 return fImage->asTextureProxyRef(this->context());
Jim Van Verth7da46762018-11-05 14:24:23 -050069 }
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040070}
71
72void GrYUVAImageTextureMaker::makeCopyKey(const CopyParams& stretch, GrUniqueKey* paramsCopyKey) {
73 // TODO: Do we ever want to disable caching?
74 if (fOriginalKey.isValid()) {
75 GrUniqueKey cacheKey;
76 static const GrUniqueKey::Domain kDomain = GrUniqueKey::GenerateDomain();
77 GrUniqueKey::Builder builder(&cacheKey, fOriginalKey, kDomain, 0, "Image");
78 MakeCopyKeyFromOrigKey(cacheKey, stretch, paramsCopyKey);
79 }
80}
81
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040082std::unique_ptr<GrFragmentProcessor> GrYUVAImageTextureMaker::createFragmentProcessor(
83 const SkMatrix& textureMatrix,
84 const SkRect& constraintRect,
85 FilterConstraint filterConstraint,
86 bool coordsLimitedToConstraintRect,
87 const GrSamplerState::Filter* filterOrNullForBicubic) {
88
Brian Salomonad8efda2019-05-10 09:22:49 -040089 // Check simple cases to see if we need to fall back to flattening the image (or whether it's
90 // already been flattened.)
91 if (!filterOrNullForBicubic || this->domainNeedsDecal() || fImage->fRGBProxy) {
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040092 return this->INHERITED::createFragmentProcessor(textureMatrix, constraintRect,
93 filterConstraint,
94 coordsLimitedToConstraintRect,
95 filterOrNullForBicubic);
96 }
97
Jim Van Verthf542cab2018-11-07 12:08:21 -050098 // Check to see if the client has given us pre-mipped textures or we can generate them
Michael Ludwiga6a84002019-04-12 15:03:02 -040099 // If not, fall back to bilerp. Also fall back to bilerp when a domain is requested
Jim Van Verth30e0d7f2018-11-02 13:36:42 -0400100 GrSamplerState::Filter filter = *filterOrNullForBicubic;
Robert Phillips8defcc12019-03-05 15:58:59 -0500101 if (GrSamplerState::Filter::kMipMap == filter &&
Michael Ludwiga6a84002019-04-12 15:03:02 -0400102 (filterConstraint == GrTextureProducer::kYes_FilterConstraint ||
103 !fImage->setupMipmapsForPlanes(this->context()))) {
Jim Van Verth30e0d7f2018-11-02 13:36:42 -0400104 filter = GrSamplerState::Filter::kBilerp;
105 }
106
Michael Ludwiga6a84002019-04-12 15:03:02 -0400107 // Cannot rely on GrTextureProducer's domain infrastructure since we need to calculate domain's
108 // per plane, which may be different, so respect the filterConstraint without any additional
109 // analysis.
110 const SkRect* domain = nullptr;
111 if (filterConstraint == GrTextureProducer::kYes_FilterConstraint) {
112 domain = &constraintRect;
113 }
114
Brian Osmane9560492019-02-05 17:00:03 -0500115 auto fp = GrYUVtoRGBEffect::Make(fImage->fProxies, fImage->fYUVAIndices,
Michael Ludwiga6a84002019-04-12 15:03:02 -0400116 fImage->fYUVColorSpace, filter, textureMatrix, domain);
Brian Salomon5ad6fd32019-03-21 15:30:08 -0400117 if (fImage->fFromColorSpace) {
118 fp = GrColorSpaceXformEffect::Make(std::move(fp), fImage->fFromColorSpace.get(),
119 fImage->alphaType(), fImage->colorSpace());
Brian Osmane9560492019-02-05 17:00:03 -0500120 }
121 return fp;
Jim Van Verth30e0d7f2018-11-02 13:36:42 -0400122}