tint: Add enable directive for extensions
In this CL the enable directive is implemented.
1. Add AST node for enable directive, assign a ExtensionKind (enum) for
each supported extension.
2. Use an unorder_set in ast::Module to record all required extensions'
kind.
3. Provide inspector methods for getting names of used extension, and
getting all used enable directives' extension names and location.
4. For different writer, the extension nodes are handled in different
ways. MSL and HLSL writers will just ignore the extension nodes, while
SPIRV and GLSL writers will emit corresponding code.
5. Implement unittests and end2end test for enable directive and
inspector, using a reserved extension name `InternalExtensionForTesting`.
Bug: tint:1472
Change-Id: I40cb4061554deb477bc2005d7e38c9718385f825
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/86623
Kokoro: Kokoro <noreply+kokoro@google.com>
Reviewed-by: Ben Clayton <bclayton@google.com>
Commit-Queue: Zhaoming Jiang <zhaoming.jiang@intel.com>
diff --git a/src/tint/inspector/inspector.cc b/src/tint/inspector/inspector.cc
index bfebda5..40c5f5a 100644
--- a/src/tint/inspector/inspector.cc
+++ b/src/tint/inspector/inspector.cc
@@ -568,6 +568,33 @@
return total_size;
}
+std::vector<std::string> Inspector::GetUsedExtensionNames() {
+ std::vector<std::string> result;
+
+ ast::ExtensionSet set = program_->AST().Extensions();
+ result.reserve(set.size());
+ for (auto kind : set) {
+ std::string name = ast::Enable::KindToName(kind);
+ result.push_back(name);
+ }
+
+ return result;
+}
+
+std::vector<std::pair<std::string, Source>> Inspector::GetEnableDirectives() {
+ std::vector<std::pair<std::string, Source>> result;
+
+ // Ast nodes for enable directive are stored within global declarations list
+ auto global_decls = program_->AST().GlobalDeclarations();
+ for (auto node : global_decls) {
+ if (auto ext = node->As<ast::Enable>()) {
+ result.push_back({ext->name, ext->source});
+ }
+ }
+
+ return result;
+}
+
const ast::Function* Inspector::FindEntryPointByName(const std::string& name) {
auto* func = program_->AST().Functions().Find(program_->Symbols().Get(name));
if (!func) {
