blob: 5e821f5806dee5bc77fb6e91fef89562225ec62c [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
Brian Osmane9560492019-02-05 17:00:03 -05008#include "GrColorSpaceXform.h"
Brian Osman3b66ab62016-11-28 09:26:31 -05009#include "GrImageTextureMaker.h"
Brian Osman3b655982017-03-07 16:58:08 -050010#include "SkGr.h"
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040011#include "SkImage_GpuYUVA.h"
Brian Osmanbd659552018-09-11 10:03:19 -040012#include "SkImage_Lazy.h"
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040013#include "effects/GrYUVtoRGBEffect.h"
Brian Osman3b66ab62016-11-28 09:26:31 -050014
Brian Osmandf7e0752017-04-26 16:20:28 -040015GrImageTextureMaker::GrImageTextureMaker(GrContext* context, const SkImage* client,
16 SkImage::CachingHint chint)
17 : INHERITED(context, client->width(), client->height(), client->isAlphaOnly())
Brian Osmanbd659552018-09-11 10:03:19 -040018 , fImage(static_cast<const SkImage_Lazy*>(client))
Brian Osmandf7e0752017-04-26 16:20:28 -040019 , fCachingHint(chint) {
Brian Osmanbd659552018-09-11 10:03:19 -040020 SkASSERT(client->isLazyGenerated());
Brian Osmandf7e0752017-04-26 16:20:28 -040021 GrMakeKeyFromImageID(&fOriginalKey, client->uniqueID(),
22 SkIRect::MakeWH(this->width(), this->height()));
Brian Osman3b66ab62016-11-28 09:26:31 -050023}
24
Robert Phillips0c984a02017-03-16 07:51:56 -040025sk_sp<GrTextureProxy> GrImageTextureMaker::refOriginalTextureProxy(bool willBeMipped,
Stan Ilievba81af22017-06-08 15:16:53 -040026 AllowedTexGenType onlyIfFast) {
Brian Osmanbd659552018-09-11 10:03:19 -040027 return fImage->lockTextureProxy(this->context(), fOriginalKey, fCachingHint,
Brian Osmane7fd8c32018-10-19 13:30:39 -040028 willBeMipped, onlyIfFast);
Robert Phillips0c984a02017-03-16 07:51:56 -040029}
30
Brian Osmanb3f38302018-09-07 15:24:44 -040031void GrImageTextureMaker::makeCopyKey(const CopyParams& stretch, GrUniqueKey* paramsCopyKey) {
Brian Osman3b66ab62016-11-28 09:26:31 -050032 if (fOriginalKey.isValid() && SkImage::kAllow_CachingHint == fCachingHint) {
Brian Osman10494e32018-09-10 12:45:18 -040033 GrUniqueKey cacheKey;
Brian Osmanbd659552018-09-11 10:03:19 -040034 fImage->makeCacheKeyFromOrigKey(fOriginalKey, &cacheKey);
Brian Osman10494e32018-09-10 12:45:18 -040035 MakeCopyKeyFromOrigKey(cacheKey, stretch, paramsCopyKey);
Brian Osman3b66ab62016-11-28 09:26:31 -050036 }
37}
38
Brian Osman3b66ab62016-11-28 09:26:31 -050039SkAlphaType GrImageTextureMaker::alphaType() const {
Brian Osmanbd659552018-09-11 10:03:19 -040040 return fImage->alphaType();
Brian Osman3b66ab62016-11-28 09:26:31 -050041}
Brian Osman6064e1c2018-10-19 14:27:54 -040042SkColorSpace* GrImageTextureMaker::colorSpace() const {
43 return fImage->colorSpace();
Brian Osman3b66ab62016-11-28 09:26:31 -050044}
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040045
46/////////////////////////////////////////////////////////////////////////////////////////////////
47
48GrYUVAImageTextureMaker::GrYUVAImageTextureMaker(GrContext* context, const SkImage* client )
49 : INHERITED(context, client->width(), client->height(), client->isAlphaOnly())
50 , fImage(static_cast<const SkImage_GpuYUVA*>(client)) {
51 SkASSERT(as_IB(client)->isYUVA());
52 GrMakeKeyFromImageID(&fOriginalKey, client->uniqueID(),
53 SkIRect::MakeWH(this->width(), this->height()));
54}
55
56sk_sp<GrTextureProxy> GrYUVAImageTextureMaker::refOriginalTextureProxy(bool willBeMipped,
57 AllowedTexGenType onlyIfFast) {
Jim Van Verth7da46762018-11-05 14:24:23 -050058 if (AllowedTexGenType::kCheap == onlyIfFast) {
59 return nullptr;
60 }
61
Jim Van Verth803a5022018-11-05 15:55:53 -050062 if (willBeMipped) {
63 return fImage->asMippedTextureProxyRef();
64 } else {
65 return fImage->asTextureProxyRef();
Jim Van Verth7da46762018-11-05 14:24:23 -050066 }
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040067}
68
69void GrYUVAImageTextureMaker::makeCopyKey(const CopyParams& stretch, GrUniqueKey* paramsCopyKey) {
70 // TODO: Do we ever want to disable caching?
71 if (fOriginalKey.isValid()) {
72 GrUniqueKey cacheKey;
73 static const GrUniqueKey::Domain kDomain = GrUniqueKey::GenerateDomain();
74 GrUniqueKey::Builder builder(&cacheKey, fOriginalKey, kDomain, 0, "Image");
75 MakeCopyKeyFromOrigKey(cacheKey, stretch, paramsCopyKey);
76 }
77}
78
79SkAlphaType GrYUVAImageTextureMaker::alphaType() const {
80 return fImage->alphaType();
81}
82SkColorSpace* GrYUVAImageTextureMaker::colorSpace() const {
83 return fImage->colorSpace();
84}
85
86std::unique_ptr<GrFragmentProcessor> GrYUVAImageTextureMaker::createFragmentProcessor(
87 const SkMatrix& textureMatrix,
88 const SkRect& constraintRect,
89 FilterConstraint filterConstraint,
90 bool coordsLimitedToConstraintRect,
91 const GrSamplerState::Filter* filterOrNullForBicubic) {
92
93 // Check simple cases to see if we need to fall back to flattening the image
Jim Van Verth53275362018-11-09 15:42:35 -050094 // TODO: See if we can relax this -- for example, if filterConstraint
95 // is kYes_FilterConstraint we still may not need a TextureDomain
96 // in some cases.
Jim Van Verth30e0d7f2018-11-02 13:36:42 -040097 if (!textureMatrix.isIdentity() || kNo_FilterConstraint != filterConstraint ||
98 !coordsLimitedToConstraintRect || !filterOrNullForBicubic) {
99 return this->INHERITED::createFragmentProcessor(textureMatrix, constraintRect,
100 filterConstraint,
101 coordsLimitedToConstraintRect,
102 filterOrNullForBicubic);
103 }
104
Jim Van Verthf542cab2018-11-07 12:08:21 -0500105 // Check to see if the client has given us pre-mipped textures or we can generate them
Jim Van Verth30e0d7f2018-11-02 13:36:42 -0400106 // If not, fall back to bilerp
Jim Van Verth30e0d7f2018-11-02 13:36:42 -0400107 GrSamplerState::Filter filter = *filterOrNullForBicubic;
Jim Van Verthf542cab2018-11-07 12:08:21 -0500108 if (GrSamplerState::Filter::kMipMap == filter && !fImage->setupMipmapsForPlanes()) {
Jim Van Verth30e0d7f2018-11-02 13:36:42 -0400109 filter = GrSamplerState::Filter::kBilerp;
110 }
111
Brian Osmane9560492019-02-05 17:00:03 -0500112 auto fp = GrYUVtoRGBEffect::Make(fImage->fProxies, fImage->fYUVAIndices,
113 fImage->fYUVColorSpace, filter);
114 if (fImage->fTargetColorSpace) {
115 fp = GrColorSpaceXformEffect::Make(std::move(fp), fImage->fColorSpace.get(),
116 fImage->alphaType(), fImage->fTargetColorSpace.get());
117 }
118 return fp;
Jim Van Verth30e0d7f2018-11-02 13:36:42 -0400119}