[WHLSL] Implement parser AST nodes
[WebKit-https.git] / Source / WebCore / Modules / webgpu / WHLSL / AST / WHLSLReferenceType.h
1 /*
2  * Copyright (C) 2019 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #pragma once
27
28 #if ENABLE(WEBGPU)
29
30 #include "WHLSLLexer.h"
31 #include "WHLSLUnnamedType.h"
32 #include <wtf/UniqueRef.h>
33 #include <wtf/text/WTFString.h>
34
35 namespace WebCore {
36
37 namespace WHLSL {
38
39 namespace AST {
40
41 class ReferenceType : public UnnamedType {
42 public:
43     enum class AddressSpace : uint8_t {
44         Constant,
45         Device,
46         Threadgroup,
47         Thread
48     };
49
50     ReferenceType(Lexer::Token&& origin, AddressSpace addressSpace, UniqueRef<UnnamedType>&& elementType)
51         : UnnamedType(WTFMove(origin))
52         , m_addressSpace(addressSpace)
53         , m_elementType(WTFMove(elementType))
54     {
55     }
56
57     virtual ~ReferenceType() = default;
58
59     ReferenceType(const ReferenceType&) = delete;
60     ReferenceType(ReferenceType&&) = default;
61
62     bool isReferenceType() const override { return false; }
63
64     AddressSpace addressSpace() const { return m_addressSpace; }
65     const UnnamedType& elementType() const { return static_cast<const UnnamedType&>(m_elementType); }
66     UnnamedType& elementType() { return static_cast<UnnamedType&>(m_elementType); }
67
68 private:
69     AddressSpace m_addressSpace;
70     UniqueRef<UnnamedType> m_elementType;
71 };
72
73 } // namespace AST
74
75 }
76
77 }
78
79 SPECIALIZE_TYPE_TRAITS_WHLSL_UNNAMED_TYPE(ReferenceType, isReferenceType())
80
81 #endif