| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 1 | /* |
| Ethan Nicholas | 130fb3f | 2018-02-01 12:14:34 -0500 | [diff] [blame] | 2 | * Copyright 2018 Google Inc. |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 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 | |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 8 | #ifndef GrYUVtoRGBEffect_DEFINED |
| 9 | #define GrYUVtoRGBEffect_DEFINED |
| Robert Phillips | 94ade75 | 2018-10-09 12:32:31 -0400 | [diff] [blame] | 10 | |
| Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 11 | #include "include/core/SkTypes.h" |
| Robert Phillips | 94ade75 | 2018-10-09 12:32:31 -0400 | [diff] [blame] | 12 | |
| Brian Salomon | ca6b2f4 | 2020-01-24 11:31:21 -0500 | [diff] [blame] | 13 | #include "include/core/SkYUVAIndex.h" |
| Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 14 | #include "src/gpu/GrCoordTransform.h" |
| 15 | #include "src/gpu/GrFragmentProcessor.h" |
| Robert Phillips | 94ade75 | 2018-10-09 12:32:31 -0400 | [diff] [blame] | 16 | |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 17 | class GrYUVtoRGBEffect : public GrFragmentProcessor { |
| 18 | public: |
| Greg Daniel | c767209 | 2020-02-06 14:32:54 -0500 | [diff] [blame] | 19 | static std::unique_ptr<GrFragmentProcessor> Make(GrSurfaceProxyView views[], |
| Robert Phillips | 94ade75 | 2018-10-09 12:32:31 -0400 | [diff] [blame] | 20 | const SkYUVAIndex indices[4], |
| Jim Van Verth | 30e0d7f | 2018-11-02 13:36:42 -0400 | [diff] [blame] | 21 | SkYUVColorSpace yuvColorSpace, |
| Brian Salomon | d71548a | 2020-02-29 19:43:30 -0500 | [diff] [blame] | 22 | GrSamplerState samplerState, |
| Brian Salomon | ca6b2f4 | 2020-01-24 11:31:21 -0500 | [diff] [blame] | 23 | const GrCaps&, |
| Michael Ludwig | a6a8400 | 2019-04-12 15:03:02 -0400 | [diff] [blame] | 24 | const SkMatrix& localMatrix = SkMatrix::I(), |
| Brian Salomon | d71548a | 2020-02-29 19:43:30 -0500 | [diff] [blame] | 25 | const SkRect* subset = nullptr); |
| Brian Osman | 9a390ac | 2018-11-12 09:47:48 -0500 | [diff] [blame] | 26 | #ifdef SK_DEBUG |
| Robert Phillips | ba5c439 | 2018-07-25 12:37:14 -0400 | [diff] [blame] | 27 | SkString dumpInfo() const override; |
| Brian Osman | 9a390ac | 2018-11-12 09:47:48 -0500 | [diff] [blame] | 28 | #endif |
| Robert Phillips | 94ade75 | 2018-10-09 12:32:31 -0400 | [diff] [blame] | 29 | |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 30 | std::unique_ptr<GrFragmentProcessor> clone() const override; |
| Brian Salomon | fa9232c | 2019-12-04 16:09:04 -0500 | [diff] [blame] | 31 | |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 32 | const char* name() const override { return "YUVtoRGBEffect"; } |
| 33 | |
| 34 | private: |
| Brian Salomon | fa9232c | 2019-12-04 16:09:04 -0500 | [diff] [blame] | 35 | GrYUVtoRGBEffect(std::unique_ptr<GrFragmentProcessor> planeFPs[4], int numPlanes, |
| 36 | const SkYUVAIndex yuvaIndices[4], SkYUVColorSpace yuvColorSpace); |
| Michael Ludwig | a6a8400 | 2019-04-12 15:03:02 -0400 | [diff] [blame] | 37 | |
| Brian Salomon | fa9232c | 2019-12-04 16:09:04 -0500 | [diff] [blame] | 38 | GrYUVtoRGBEffect(const GrYUVtoRGBEffect& src); |
| Michael Ludwig | a6a8400 | 2019-04-12 15:03:02 -0400 | [diff] [blame] | 39 | |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 40 | GrGLSLFragmentProcessor* onCreateGLSLInstance() const override; |
| Brian Salomon | fa9232c | 2019-12-04 16:09:04 -0500 | [diff] [blame] | 41 | |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 42 | void onGetGLSLProcessorKey(const GrShaderCaps&, GrProcessorKeyBuilder*) const override; |
| Brian Salomon | fa9232c | 2019-12-04 16:09:04 -0500 | [diff] [blame] | 43 | |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 44 | bool onIsEqual(const GrFragmentProcessor&) const override; |
| Brian Salomon | fa9232c | 2019-12-04 16:09:04 -0500 | [diff] [blame] | 45 | |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 46 | GR_DECLARE_FRAGMENT_PROCESSOR_TEST |
| Robert Phillips | 94ade75 | 2018-10-09 12:32:31 -0400 | [diff] [blame] | 47 | |
| Robert Phillips | 94ade75 | 2018-10-09 12:32:31 -0400 | [diff] [blame] | 48 | SkYUVAIndex fYUVAIndices[4]; |
| Robert Phillips | b651aac | 2019-03-15 12:18:49 -0400 | [diff] [blame] | 49 | SkYUVColorSpace fYUVColorSpace; |
| Ethan Nicholas | 7461a4a | 2017-12-21 14:18:01 -0500 | [diff] [blame] | 50 | }; |
| 51 | #endif |