| // Copyright 2020 The Tint Authors. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #include "src/tint/lang/msl/writer/writer.h" |
| |
| #include <memory> |
| #include <utility> |
| |
| #include "src/tint/lang/msl/writer/ast_printer/ast_printer.h" |
| #include "src/tint/lang/msl/writer/printer/printer.h" |
| #include "src/tint/lang/msl/writer/raise/raise.h" |
| #include "src/tint/lang/wgsl/reader/program_to_ir/program_to_ir.h" |
| |
| namespace tint::msl::writer { |
| |
| Result<Output, std::string> Generate(const Program& program, const Options& options) { |
| if (!program.IsValid()) { |
| return std::string("input program is not valid"); |
| } |
| |
| Output output; |
| |
| if (options.use_tint_ir) { |
| // Convert the AST program to an IR module. |
| auto converted = wgsl::reader::ProgramToIR(program); |
| if (!converted) { |
| return std::string("IR converter: " + converted.Failure()); |
| } |
| |
| auto ir = converted.Move(); |
| |
| // Raise the IR to the MSL dialect. |
| auto raised = raise::Raise(&ir); |
| if (!raised) { |
| return std::move(raised.Failure()); |
| } |
| |
| // Generate the MSL code. |
| auto impl = std::make_unique<Printer>(&ir); |
| auto result = impl->Generate(); |
| if (!result) { |
| return result.Failure(); |
| } |
| output.msl = impl->Result(); |
| } else { |
| // Sanitize the program. |
| auto sanitized_result = Sanitize(program, options); |
| if (!sanitized_result.program.IsValid()) { |
| return sanitized_result.program.Diagnostics().str(); |
| } |
| output.needs_storage_buffer_sizes = sanitized_result.needs_storage_buffer_sizes; |
| output.used_array_length_from_uniform_indices = |
| std::move(sanitized_result.used_array_length_from_uniform_indices); |
| |
| // Generate the MSL code. |
| auto impl = std::make_unique<ASTPrinter>(sanitized_result.program); |
| if (!impl->Generate()) { |
| return impl->Diagnostics().str(); |
| } |
| output.msl = impl->Result(); |
| output.has_invariant_attribute = impl->HasInvariant(); |
| output.workgroup_allocations = impl->DynamicWorkgroupAllocations(); |
| } |
| |
| return output; |
| } |
| |
| } // namespace tint::msl::writer |