12 #include "fml/backtrace.h"
14 #include "inja/inja.hpp"
17 #include "impeller/runtime_stage/runtime_stage_flatbuffers.h"
18 #include "runtime_stage_types_flatbuffers.h"
29 FML_DCHECK(data_.find(
data->backend) == data_.end());
33 static std::optional<fb::Stage>
ToStage(spv::ExecutionModel stage) {
35 case spv::ExecutionModel::ExecutionModelVertex:
36 return fb::Stage::kVertex;
37 case spv::ExecutionModel::ExecutionModelFragment:
38 return fb::Stage::kFragment;
39 case spv::ExecutionModel::ExecutionModelGLCompute:
40 return fb::Stage::kCompute;
47 static std::optional<fb::Stage>
ToJsonStage(spv::ExecutionModel stage) {
49 case spv::ExecutionModel::ExecutionModelVertex:
50 return fb::Stage::kVertex;
51 case spv::ExecutionModel::ExecutionModelFragment:
52 return fb::Stage::kFragment;
53 case spv::ExecutionModel::ExecutionModelGLCompute:
54 return fb::Stage::kCompute;
62 spirv_cross::SPIRType::BaseType
type) {
64 case spirv_cross::SPIRType::Float:
66 case spirv_cross::SPIRType::SampledImage:
68 case spirv_cross::SPIRType::Struct:
70 case spirv_cross::SPIRType::Boolean:
71 case spirv_cross::SPIRType::SByte:
72 case spirv_cross::SPIRType::UByte:
73 case spirv_cross::SPIRType::Short:
74 case spirv_cross::SPIRType::UShort:
75 case spirv_cross::SPIRType::Int:
76 case spirv_cross::SPIRType::UInt:
77 case spirv_cross::SPIRType::Int64:
78 case spirv_cross::SPIRType::UInt64:
79 case spirv_cross::SPIRType::Half:
80 case spirv_cross::SPIRType::Double:
81 case spirv_cross::SPIRType::AccelerationStructure:
82 case spirv_cross::SPIRType::AtomicCounter:
83 case spirv_cross::SPIRType::Char:
84 case spirv_cross::SPIRType::ControlPointArray:
85 case spirv_cross::SPIRType::Image:
86 case spirv_cross::SPIRType::Interpolant:
87 case spirv_cross::SPIRType::RayQuery:
88 case spirv_cross::SPIRType::Sampler:
89 case spirv_cross::SPIRType::Unknown:
90 case spirv_cross::SPIRType::Void:
96 spirv_cross::SPIRType::BaseType
type) {
98 case spirv_cross::SPIRType::Boolean:
99 return fb::InputDataType::kBoolean;
100 case spirv_cross::SPIRType::SByte:
101 return fb::InputDataType::kSignedByte;
102 case spirv_cross::SPIRType::UByte:
103 return fb::InputDataType::kUnsignedByte;
104 case spirv_cross::SPIRType::Short:
105 return fb::InputDataType::kSignedShort;
106 case spirv_cross::SPIRType::UShort:
107 return fb::InputDataType::kUnsignedShort;
108 case spirv_cross::SPIRType::Int:
109 return fb::InputDataType::kSignedInt;
110 case spirv_cross::SPIRType::UInt:
111 return fb::InputDataType::kUnsignedInt;
112 case spirv_cross::SPIRType::Int64:
113 return fb::InputDataType::kSignedInt64;
114 case spirv_cross::SPIRType::UInt64:
115 return fb::InputDataType::kUnsignedInt64;
116 case spirv_cross::SPIRType::Float:
118 case spirv_cross::SPIRType::Double:
119 return fb::InputDataType::kDouble;
120 case spirv_cross::SPIRType::Unknown:
121 case spirv_cross::SPIRType::Void:
122 case spirv_cross::SPIRType::Half:
123 case spirv_cross::SPIRType::AtomicCounter:
124 case spirv_cross::SPIRType::Struct:
125 case spirv_cross::SPIRType::Image:
126 case spirv_cross::SPIRType::SampledImage:
127 case spirv_cross::SPIRType::Sampler:
128 case spirv_cross::SPIRType::AccelerationStructure:
129 case spirv_cross::SPIRType::RayQuery:
130 case spirv_cross::SPIRType::ControlPointArray:
131 case spirv_cross::SPIRType::Interpolant:
132 case spirv_cross::SPIRType::Char:
139 spirv_cross::SPIRType::BaseType
type) {
141 case spirv_cross::SPIRType::Boolean:
143 case spirv_cross::SPIRType::SByte:
145 case spirv_cross::SPIRType::UByte:
147 case spirv_cross::SPIRType::Short:
149 case spirv_cross::SPIRType::UShort:
151 case spirv_cross::SPIRType::Int:
153 case spirv_cross::SPIRType::UInt:
155 case spirv_cross::SPIRType::Int64:
157 case spirv_cross::SPIRType::UInt64:
159 case spirv_cross::SPIRType::Half:
161 case spirv_cross::SPIRType::Float:
163 case spirv_cross::SPIRType::Double:
165 case spirv_cross::SPIRType::SampledImage:
167 case spirv_cross::SPIRType::Struct:
169 case spirv_cross::SPIRType::AccelerationStructure:
170 case spirv_cross::SPIRType::AtomicCounter:
171 case spirv_cross::SPIRType::Char:
172 case spirv_cross::SPIRType::ControlPointArray:
173 case spirv_cross::SPIRType::Image:
174 case spirv_cross::SPIRType::Interpolant:
175 case spirv_cross::SPIRType::RayQuery:
176 case spirv_cross::SPIRType::Sampler:
177 case spirv_cross::SPIRType::Unknown:
178 case spirv_cross::SPIRType::Void:
235 for (
const auto& kvp : data_) {
236 nlohmann::json platform_object;
239 if (!stage.has_value()) {
243 platform_object[
kStageKey] =
static_cast<uint32_t
>(stage.value());
246 if (kvp.second->shader->GetSize() > 0u) {
248 reinterpret_cast<const char*
>(kvp.second->shader->GetMapping()),
249 kvp.second->shader->GetSize());
253 auto& uniforms = platform_object[
kUniformsKey] = nlohmann::json::array_t{};
254 for (
const auto& uniform : kvp.second->uniforms) {
255 nlohmann::json uniform_object;
262 if (!uniform_type.has_value()) {
270 uniform.array_elements.value_or(0);
272 uniforms.push_back(uniform_object);
278 auto json_string = std::make_shared<std::string>(root.dump(2u));
280 return std::make_shared<fml::NonOwnedMapping>(
281 reinterpret_cast<const uint8_t*
>(json_string->data()),
282 json_string->size(), [json_string](
auto,
auto) {});
287 auto kvp = data_.find(backend);
288 if (kvp == data_.end()) {
292 auto runtime_stage = std::make_unique<fb::RuntimeStageT>();
293 runtime_stage->entrypoint = kvp->second->entrypoint;
294 const auto stage =
ToStage(kvp->second->stage);
295 if (!stage.has_value()) {
299 runtime_stage->stage = stage.value();
300 if (!kvp->second->shader) {
304 if (kvp->second->shader->GetSize() > 0u) {
305 runtime_stage->shader = {
306 kvp->second->shader->GetMapping(),
307 kvp->second->shader->GetMapping() + kvp->second->shader->GetSize()};
309 for (
const auto& uniform : kvp->second->uniforms) {
310 auto desc = std::make_unique<fb::UniformDescriptionT>();
312 desc->name = uniform.name;
313 if (desc->name.empty()) {
317 desc->location = uniform.location;
318 desc->rows = uniform.rows;
319 desc->columns = uniform.columns;
321 if (!uniform_type.has_value()) {
325 desc->type = uniform_type.value();
326 desc->bit_width = uniform.bit_width;
327 if (uniform.array_elements.has_value()) {
328 desc->array_elements = uniform.array_elements.value();
331 for (
const auto& byte_type : uniform.struct_layout) {
332 desc->struct_layout.push_back(
static_cast<fb::StructByteType
>(byte_type));
334 desc->struct_float_count = uniform.struct_float_count;
336 runtime_stage->uniforms.emplace_back(std::move(desc));
339 for (
const auto& input : kvp->second->inputs) {
340 auto desc = std::make_unique<fb::StageInputT>();
342 desc->name = input.name;
344 if (desc->name.empty()) {
348 desc->location = input.location;
349 desc->set = input.set;
350 desc->binding = input.binding;
352 if (!input_type.has_value()) {
356 desc->type = input_type.value();
357 desc->bit_width = input.bit_width;
358 desc->vec_size = input.vec_size;
359 desc->columns = input.columns;
360 desc->offset = input.offset;
362 runtime_stage->inputs.emplace_back(std::move(desc));
365 return runtime_stage;
368 std::unique_ptr<fb::RuntimeStagesT>
372 auto runtime_stages = std::make_unique<fb::RuntimeStagesT>();
374 for (
const auto& kvp : data_) {
378 runtime_stages->sksl = std::move(runtime_stage);
381 runtime_stages->metal = std::move(runtime_stage);
384 runtime_stages->opengles = std::move(runtime_stage);
387 runtime_stages->vulkan = std::move(runtime_stage);
390 runtime_stages->opengles3 = std::move(runtime_stage);
394 return runtime_stages;
399 if (!runtime_stages) {
403 auto builder = std::make_shared<flatbuffers::FlatBufferBuilder>();
404 builder->Finish(fb::RuntimeStages::Pack(*builder.get(), runtime_stages.get()),
405 fb::RuntimeStagesIdentifier());
406 return std::make_shared<fml::NonOwnedMapping>(builder->GetBufferPointer(),
408 [builder](
auto,
auto) {});
std::unique_ptr< fb::RuntimeStagesT > CreateMultiStageFlatbuffer() const
std::shared_ptr< fml::Mapping > CreateMapping() const
std::unique_ptr< fb::RuntimeStageT > CreateStageFlatbuffer(impeller::RuntimeStageBackend backend) const
void AddShader(const std::shared_ptr< Shader > &data)
std::shared_ptr< fml::Mapping > CreateJsonMapping() const
static std::optional< fb::Stage > ToStage(spv::ExecutionModel stage)
static std::optional< fb::UniformDataType > ToUniformType(spirv_cross::SPIRType::BaseType type)
static const char * kUniformLocationKey
static const char * kUniformNameKey
static std::optional< fb::InputDataType > ToInputType(spirv_cross::SPIRType::BaseType type)
static std::optional< fb::Stage > ToJsonStage(spv::ExecutionModel stage)
static const char * kEntrypointKey
static const char * kUniformColumnsKey
static std::optional< uint32_t > ToJsonType(spirv_cross::SPIRType::BaseType type)
static std::string RuntimeStageBackendToString(RuntimeStageBackend backend)
static const char * kUniformTypeKey
static const char * kStageKey
static const char * kUniformsKey
static const char * kUniformRowsKey
static const char * kShaderKey
static const char * kUniformArrayElementsKey
static const char * kTargetPlatformKey
static const char * kUniformBitWidthKey
std::shared_ptr< const fml::Mapping > data