1 files changed, 212 insertions, 0 deletions
diff --git a/gfx/angle/checkout/src/compiler/translator/tree_util/AsNode.h b/gfx/angle/checkout/src/compiler/translator/tree_util/AsNode.h
new file mode 100644
index 0000000000..1c6de182d9
--- /dev/null
+++ b/gfx/angle/checkout/src/compiler/translator/tree_util/AsNode.h
@@ -0,0 +1,212 @@
+//
+// Copyright 2020 The ANGLE Project Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+//
+
+#ifndef COMPILER_TRANSLATOR_TREEUTIL_INTERMASNODE_H_
+#define COMPILER_TRANSLATOR_TREEUTIL_INTERMASNODE_H_
+
+#include "common/angleutils.h"
+#include "compiler/translator/IntermNode.h"
+
+#include <utility>
+
+namespace sh
+{
+
+namespace priv
+{
+
+template <typename T>
+struct AsNode
+{};
+
+template <>
+struct AsNode<TIntermNode>
+{
+    static ANGLE_INLINE TIntermNode *exec(TIntermNode *node) { return node; }
+};
+
+template <>
+struct AsNode<TIntermTyped>
+{
+    static ANGLE_INLINE TIntermTyped *exec(TIntermNode *node)
+    {
+        return node ? node->getAsTyped() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermSymbol>
+{
+    static ANGLE_INLINE TIntermSymbol *exec(TIntermNode *node)
+    {
+        return node ? node->getAsSymbolNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermConstantUnion>
+{
+    static ANGLE_INLINE TIntermConstantUnion *exec(TIntermNode *node)
+    {
+        return node ? node->getAsConstantUnion() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermFunctionPrototype>
+{
+    static ANGLE_INLINE TIntermFunctionPrototype *exec(TIntermNode *node)
+    {
+        return node ? node->getAsFunctionPrototypeNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermPreprocessorDirective>
+{
+    static ANGLE_INLINE TIntermPreprocessorDirective *exec(TIntermNode *node)
+    {
+        return node ? node->getAsPreprocessorDirective() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermSwizzle>
+{
+    static ANGLE_INLINE TIntermSwizzle *exec(TIntermNode *node)
+    {
+        return node ? node->getAsSwizzleNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermBinary>
+{
+    static ANGLE_INLINE TIntermBinary *exec(TIntermNode *node)
+    {
+        return node ? node->getAsBinaryNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermUnary>
+{
+    static ANGLE_INLINE TIntermUnary *exec(TIntermNode *node)
+    {
+        return node ? node->getAsUnaryNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermTernary>
+{
+    static ANGLE_INLINE TIntermTernary *exec(TIntermNode *node)
+    {
+        return node ? node->getAsTernaryNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermIfElse>
+{
+    static ANGLE_INLINE TIntermIfElse *exec(TIntermNode *node)
+    {
+        return node ? node->getAsIfElseNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermSwitch>
+{
+    static ANGLE_INLINE TIntermSwitch *exec(TIntermNode *node)
+    {
+        return node ? node->getAsSwitchNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermCase>
+{
+    static ANGLE_INLINE TIntermCase *exec(TIntermNode *node)
+    {
+        return node ? node->getAsCaseNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermFunctionDefinition>
+{
+    static ANGLE_INLINE TIntermFunctionDefinition *exec(TIntermNode *node)
+    {
+        return node ? node->getAsFunctionDefinition() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermAggregate>
+{
+    static ANGLE_INLINE TIntermAggregate *exec(TIntermNode *node)
+    {
+        return node ? node->getAsAggregate() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermBlock>
+{
+    static ANGLE_INLINE TIntermBlock *exec(TIntermNode *node)
+    {
+        return node ? node->getAsBlock() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermGlobalQualifierDeclaration>
+{
+    static ANGLE_INLINE TIntermGlobalQualifierDeclaration *exec(TIntermNode *node)
+    {
+        return node ? node->getAsGlobalQualifierDeclarationNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermDeclaration>
+{
+    static ANGLE_INLINE TIntermDeclaration *exec(TIntermNode *node)
+    {
+        return node ? node->getAsDeclarationNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermLoop>
+{
+    static ANGLE_INLINE TIntermLoop *exec(TIntermNode *node)
+    {
+        return node ? node->getAsLoopNode() : nullptr;
+    }
+};
+
+template <>
+struct AsNode<TIntermBranch>
+{
+    static ANGLE_INLINE TIntermBranch *exec(TIntermNode *node)
+    {
+        return node ? node->getAsBranchNode() : nullptr;
+    }
+};
+
+}  // namespace priv
+
+template <typename T>
+ANGLE_INLINE T *asNode(TIntermNode *node)
+{
+    return priv::AsNode<T>::exec(node);
+}
+
+}  // namespace sh
+
+#endif  // COMPILER_TRANSLATOR_TREEUTIL_INTERMASNODE_H_