reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2015 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 | #ifndef GrYUVProvider_DEFINED |
| 9 | #define GrYUVProvider_DEFINED |
| 10 | |
| 11 | #include "GrTypes.h" |
| 12 | #include "SkImageInfo.h" |
msarett | 4984c3c | 2016-03-10 05:44:43 -0800 | [diff] [blame] | 13 | #include "SkYUVSizeInfo.h" |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 14 | |
| 15 | class GrContext; |
| 16 | class GrTexture; |
Brian Salomon | c65aec9 | 2017-03-09 09:03:58 -0500 | [diff] [blame] | 17 | class GrTextureProxy; |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 18 | |
| 19 | /** |
| 20 | * There are at least 2 different ways to extract/retrieve YUV planar data... |
| 21 | * - SkPixelRef |
Brian Osman | ef90ae4 | 2017-04-24 09:30:24 -0400 | [diff] [blame] | 22 | * - SkImageGenerator |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 23 | * |
| 24 | * To share common functionality around using the planar data, we use this abstract base-class |
| 25 | * to represent accessing that data. |
| 26 | */ |
| 27 | class GrYUVProvider { |
| 28 | public: |
| 29 | virtual ~GrYUVProvider() {} |
| 30 | |
| 31 | /** |
Robert Phillips | 538f1a3 | 2017-03-08 14:32:55 -0500 | [diff] [blame] | 32 | * On success, this returns a texture proxy that has converted the YUV data from the provider |
Greg Daniel | 1445da6 | 2018-01-04 10:27:29 -0500 | [diff] [blame] | 33 | * into a form that is supported by the GPU (typically transformed into RGB). The texture will |
| 34 | * automatically have a key added, so it can be retrieved from the cache (assuming it is |
| 35 | * requested by a provider w/ the same genID). If srcColorSpace and dstColorSpace are |
| 36 | * specified, then a color conversion from src to dst will be applied to the pixels. |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 37 | * |
| 38 | * On failure (e.g. the provider had no data), this returns NULL. |
| 39 | */ |
Greg Daniel | 1445da6 | 2018-01-04 10:27:29 -0500 | [diff] [blame] | 40 | sk_sp<GrTextureProxy> refAsTextureProxy(GrContext*, const GrSurfaceDesc&, |
Christopher Cameron | 77e9666 | 2017-07-08 01:47:47 -0700 | [diff] [blame] | 41 | const SkColorSpace* srcColorSpace, |
| 42 | const SkColorSpace* dstColorSpace); |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 43 | |
| 44 | virtual uint32_t onGetID() = 0; |
| 45 | |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 46 | // These are not meant to be called by a client, only by the implementation |
| 47 | |
| 48 | /** |
msarett | 4984c3c | 2016-03-10 05:44:43 -0800 | [diff] [blame] | 49 | * If decoding to YUV is supported, this returns true. Otherwise, this |
| 50 | * returns false and does not modify any of the parameters. |
| 51 | * |
| 52 | * @param sizeInfo Output parameter indicating the sizes and required |
| 53 | * allocation widths of the Y, U, and V planes. |
| 54 | * @param colorSpace Output parameter. |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 55 | */ |
msarett | 4984c3c | 2016-03-10 05:44:43 -0800 | [diff] [blame] | 56 | virtual bool onQueryYUV8(SkYUVSizeInfo* sizeInfo, SkYUVColorSpace* colorSpace) const = 0; |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 57 | |
| 58 | /** |
msarett | 4984c3c | 2016-03-10 05:44:43 -0800 | [diff] [blame] | 59 | * Returns true on success and false on failure. |
| 60 | * This always attempts to perform a full decode. If the client only |
| 61 | * wants size, it should call onQueryYUV8(). |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 62 | * |
msarett | 4984c3c | 2016-03-10 05:44:43 -0800 | [diff] [blame] | 63 | * @param sizeInfo Needs to exactly match the values returned by the |
| 64 | * query, except the WidthBytes may be larger than the |
| 65 | * recommendation (but not smaller). |
| 66 | * @param planes Memory for each of the Y, U, and V planes. |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 67 | */ |
msarett | 4984c3c | 2016-03-10 05:44:43 -0800 | [diff] [blame] | 68 | virtual bool onGetYUV8Planes(const SkYUVSizeInfo& sizeInfo, void* planes[3]) = 0; |
Greg Daniel | fb3abcd | 2018-02-02 15:48:33 -0500 | [diff] [blame] | 69 | |
| 70 | private: |
| 71 | // This is used as release callback for the YUV data that we capture in an SkImage when |
| 72 | // uploading to a gpu. When the upload is complete and we release the SkImage this callback will |
| 73 | // release the underlying data. |
| 74 | static void YUVGen_DataReleaseProc(const void*, void* data); |
reed | 43fe618 | 2015-09-08 08:37:36 -0700 | [diff] [blame] | 75 | }; |
| 76 | |
| 77 | #endif |