Austin Eng | cc2516a | 2023-10-17 20:57:54 +0000 | [diff] [blame] | 1 | // Copyright 2021 The Dawn & Tint Authors |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 2 | // |
Austin Eng | cc2516a | 2023-10-17 20:57:54 +0000 | [diff] [blame] | 3 | // Redistribution and use in source and binary forms, with or without |
| 4 | // modification, are permitted provided that the following conditions are met: |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 5 | // |
Austin Eng | cc2516a | 2023-10-17 20:57:54 +0000 | [diff] [blame] | 6 | // 1. Redistributions of source code must retain the above copyright notice, this |
| 7 | // list of conditions and the following disclaimer. |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 8 | // |
Austin Eng | cc2516a | 2023-10-17 20:57:54 +0000 | [diff] [blame] | 9 | // 2. Redistributions in binary form must reproduce the above copyright notice, |
| 10 | // this list of conditions and the following disclaimer in the documentation |
| 11 | // and/or other materials provided with the distribution. |
| 12 | // |
| 13 | // 3. Neither the name of the copyright holder nor the names of its |
| 14 | // contributors may be used to endorse or promote products derived from |
| 15 | // this software without specific prior written permission. |
| 16 | // |
| 17 | // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
| 18 | // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| 19 | // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| 20 | // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE |
| 21 | // FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| 22 | // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR |
| 23 | // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER |
| 24 | // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
| 25 | // OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 26 | // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 27 | |
dan sinclair | 99181d8 | 2023-07-20 01:14:15 +0000 | [diff] [blame] | 28 | #ifndef SRC_TINT_LANG_WGSL_AST_TRANSFORM_BINDING_REMAPPER_H_ |
| 29 | #define SRC_TINT_LANG_WGSL_AST_TRANSFORM_BINDING_REMAPPER_H_ |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 30 | |
| 31 | #include <unordered_map> |
| 32 | |
Ben Clayton | c1942cf | 2023-08-16 19:15:21 +0000 | [diff] [blame] | 33 | #include "src/tint/api/common/binding_point.h" |
Ben Clayton | cd52f38 | 2023-08-07 13:11:08 +0000 | [diff] [blame] | 34 | #include "src/tint/lang/core/access.h" |
dan sinclair | 99181d8 | 2023-07-20 01:14:15 +0000 | [diff] [blame] | 35 | #include "src/tint/lang/wgsl/ast/transform/transform.h" |
Ben Clayton | b3484ec | 2024-04-25 12:05:22 +0000 | [diff] [blame] | 36 | #include "src/tint/utils/reflection/reflection.h" |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 37 | |
James Price | b4acbb8 | 2023-05-11 21:27:16 +0000 | [diff] [blame] | 38 | namespace tint::ast::transform { |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 39 | |
Ben Clayton | 1025258 | 2023-07-25 20:53:25 +0000 | [diff] [blame] | 40 | /// BindingPoint is an alias to BindingPoint |
| 41 | using BindingPoint = BindingPoint; |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 42 | |
| 43 | /// BindingRemapper is a transform used to remap resource binding points and |
| 44 | /// access controls. |
dan sinclair | bae54e7 | 2023-07-28 15:01:54 +0000 | [diff] [blame] | 45 | class BindingRemapper final : public Castable<BindingRemapper, Transform> { |
dan sinclair | 41e4d9a | 2022-05-01 14:40:55 +0000 | [diff] [blame] | 46 | public: |
| 47 | /// BindingPoints is a map of old binding point to new binding point |
| 48 | using BindingPoints = std::unordered_map<BindingPoint, BindingPoint>; |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 49 | |
dan sinclair | 41e4d9a | 2022-05-01 14:40:55 +0000 | [diff] [blame] | 50 | /// AccessControls is a map of old binding point to new access control |
Ben Clayton | cd52f38 | 2023-08-07 13:11:08 +0000 | [diff] [blame] | 51 | using AccessControls = std::unordered_map<BindingPoint, core::Access>; |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 52 | |
dan sinclair | 41e4d9a | 2022-05-01 14:40:55 +0000 | [diff] [blame] | 53 | /// Remappings is consumed by the BindingRemapper transform. |
| 54 | /// Data holds information about shader usage and constant buffer offsets. |
dan sinclair | bae54e7 | 2023-07-28 15:01:54 +0000 | [diff] [blame] | 55 | struct Remappings final : public Castable<Remappings, Data> { |
dan sinclair | 41e4d9a | 2022-05-01 14:40:55 +0000 | [diff] [blame] | 56 | /// Constructor |
Ben Clayton | b3484ec | 2024-04-25 12:05:22 +0000 | [diff] [blame] | 57 | Remappings(); |
| 58 | |
| 59 | /// Constructor |
dan sinclair | 41e4d9a | 2022-05-01 14:40:55 +0000 | [diff] [blame] | 60 | /// @param bp a map of new binding points |
| 61 | /// @param ac a map of new access controls |
| 62 | /// @param may_collide If true, then validation will be disabled for |
| 63 | /// binding point collisions generated by this transform |
| 64 | Remappings(BindingPoints bp, AccessControls ac, bool may_collide = true); |
| 65 | |
| 66 | /// Copy constructor |
| 67 | Remappings(const Remappings&); |
| 68 | |
| 69 | /// Destructor |
| 70 | ~Remappings() override; |
| 71 | |
| 72 | /// A map of old binding point to new binding point |
Ben Clayton | b3484ec | 2024-04-25 12:05:22 +0000 | [diff] [blame] | 73 | BindingPoints binding_points; |
dan sinclair | 41e4d9a | 2022-05-01 14:40:55 +0000 | [diff] [blame] | 74 | |
| 75 | /// A map of old binding point to new access controls |
Ben Clayton | b3484ec | 2024-04-25 12:05:22 +0000 | [diff] [blame] | 76 | AccessControls access_controls; |
dan sinclair | 41e4d9a | 2022-05-01 14:40:55 +0000 | [diff] [blame] | 77 | |
| 78 | /// If true, then validation will be disabled for binding point collisions |
| 79 | /// generated by this transform |
Ben Clayton | b3484ec | 2024-04-25 12:05:22 +0000 | [diff] [blame] | 80 | bool allow_collisions = false; |
| 81 | |
| 82 | /// Reflection for this class |
| 83 | TINT_REFLECT(Remappings, binding_points, access_controls, allow_collisions); |
dan sinclair | 41e4d9a | 2022-05-01 14:40:55 +0000 | [diff] [blame] | 84 | }; |
| 85 | |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 86 | /// Constructor |
dan sinclair | 41e4d9a | 2022-05-01 14:40:55 +0000 | [diff] [blame] | 87 | BindingRemapper(); |
| 88 | ~BindingRemapper() override; |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 89 | |
Ben Clayton | c6b3814 | 2022-11-03 08:41:19 +0000 | [diff] [blame] | 90 | /// @copydoc Transform::Apply |
Ben Clayton | 5ed5cc4 | 2023-09-22 10:31:04 +0000 | [diff] [blame] | 91 | ApplyResult Apply(const Program& program, |
Ben Clayton | c6b3814 | 2022-11-03 08:41:19 +0000 | [diff] [blame] | 92 | const DataMap& inputs, |
| 93 | DataMap& outputs) const override; |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 94 | }; |
| 95 | |
James Price | b4acbb8 | 2023-05-11 21:27:16 +0000 | [diff] [blame] | 96 | } // namespace tint::ast::transform |
Ryan Harrison | dbc13af | 2022-02-21 15:19:07 +0000 | [diff] [blame] | 97 | |
dan sinclair | 99181d8 | 2023-07-20 01:14:15 +0000 | [diff] [blame] | 98 | #endif // SRC_TINT_LANG_WGSL_AST_TRANSFORM_BINDING_REMAPPER_H_ |