Brian Osman | 6b797fe | 2019-04-08 13:56:36 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2019 Google LLC. |
| 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 GrPersistentCacheEntry_DEFINED |
| 9 | #define GrPersistentCacheEntry_DEFINED |
| 10 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 11 | #include "include/core/SkData.h" |
| 12 | #include "include/private/GrTypesPriv.h" |
| 13 | #include "src/core/SkReader32.h" |
| 14 | #include "src/core/SkWriter32.h" |
| 15 | #include "src/sksl/SkSLString.h" |
| 16 | #include "src/sksl/ir/SkSLProgram.h" |
Brian Osman | 6b797fe | 2019-04-08 13:56:36 -0400 | [diff] [blame] | 17 | |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 18 | // The GrPersistentCache stores opaque blobs, as far as clients are concerned. It's helpful to |
| 19 | // inspect certain kinds of cached data within our tools, so for those cases (GLSL, SPIR-V), we |
| 20 | // put the serialization logic here, to be shared by the backend code and the tool code. |
Brian Osman | 6b797fe | 2019-04-08 13:56:36 -0400 | [diff] [blame] | 21 | namespace GrPersistentCacheUtils { |
| 22 | |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 23 | static inline sk_sp<SkData> PackCachedShaders(SkFourByteTag shaderType, |
| 24 | const SkSL::String shaders[], |
| 25 | const SkSL::Program::Inputs inputs[], |
Brian Osman | ed58e00 | 2019-09-06 14:42:43 -0400 | [diff] [blame] | 26 | int numInputs, |
| 27 | const SkSL::Program::Settings* settings) { |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 28 | // For consistency (so tools can blindly pack and unpack cached shaders), we always write |
| 29 | // kGrShaderTypeCount inputs. If the backend gives us fewer, we just replicate the last one. |
| 30 | SkASSERT(numInputs >= 1 && numInputs <= kGrShaderTypeCount); |
Brian Osman | 6b797fe | 2019-04-08 13:56:36 -0400 | [diff] [blame] | 31 | |
Brian Osman | e6ef03d | 2019-04-11 14:38:27 -0400 | [diff] [blame] | 32 | SkWriter32 writer; |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 33 | writer.write32(shaderType); |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 34 | for (int i = 0; i < kGrShaderTypeCount; ++i) { |
| 35 | writer.writeString(shaders[i].c_str(), shaders[i].size()); |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 36 | writer.writePad(&inputs[SkTMin(i, numInputs - 1)], sizeof(SkSL::Program::Inputs)); |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 37 | } |
Brian Osman | ed58e00 | 2019-09-06 14:42:43 -0400 | [diff] [blame] | 38 | writer.writeBool(SkToBool(settings)); |
| 39 | if (settings) { |
| 40 | writer.writeBool(settings->fFlipY); |
| 41 | writer.writeBool(settings->fFragColorIsInOut); |
| 42 | writer.writeBool(settings->fForceHighPrecision); |
| 43 | } |
Brian Osman | e6ef03d | 2019-04-11 14:38:27 -0400 | [diff] [blame] | 44 | return writer.snapshotAsData(); |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 45 | } |
| 46 | |
Brian Osman | a66081d | 2019-09-03 14:59:26 -0400 | [diff] [blame] | 47 | static inline void UnpackCachedShaders(SkReader32* reader, |
| 48 | SkSL::String shaders[], |
| 49 | SkSL::Program::Inputs inputs[], |
Brian Osman | ed58e00 | 2019-09-06 14:42:43 -0400 | [diff] [blame] | 50 | int numInputs, |
| 51 | SkSL::Program::Settings* settings = nullptr) { |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 52 | for (int i = 0; i < kGrShaderTypeCount; ++i) { |
| 53 | size_t stringLen = 0; |
Brian Osman | a66081d | 2019-09-03 14:59:26 -0400 | [diff] [blame] | 54 | const char* string = reader->readString(&stringLen); |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 55 | shaders[i] = SkSL::String(string, stringLen); |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 56 | |
| 57 | // GL, for example, only wants one set of Inputs |
| 58 | if (i < numInputs) { |
Brian Osman | a66081d | 2019-09-03 14:59:26 -0400 | [diff] [blame] | 59 | reader->read(&inputs[i], sizeof(inputs[i])); |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 60 | } else { |
Brian Osman | a66081d | 2019-09-03 14:59:26 -0400 | [diff] [blame] | 61 | reader->skip(sizeof(SkSL::Program::Inputs)); |
Brian Osman | a085a41 | 2019-04-25 09:44:43 -0400 | [diff] [blame] | 62 | } |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 63 | } |
Brian Osman | ed58e00 | 2019-09-06 14:42:43 -0400 | [diff] [blame] | 64 | if (reader->readBool() && settings) { |
| 65 | settings->fFlipY = reader->readBool(); |
| 66 | settings->fFragColorIsInOut = reader->readBool(); |
| 67 | settings->fForceHighPrecision = reader->readBool(); |
| 68 | } |
Brian Osman | a5a010b | 2019-04-08 15:01:32 -0400 | [diff] [blame] | 69 | } |
| 70 | |
Brian Osman | 6b797fe | 2019-04-08 13:56:36 -0400 | [diff] [blame] | 71 | } |
| 72 | |
| 73 | #endif |