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#include "config.h"
27#include "WHLSLSynthesizeArrayOperatorLength.h"
28
29#if ENABLE(WEBGPU)
30
31#include "WHLSLArrayType.h"
32#include "WHLSLProgram.h"
33#include "WHLSLTypeReference.h"
34#include "WHLSLVisitor.h"
35
36namespace WebCore {
37
38namespace WHLSL {
39
40class FindArrayTypes : public Visitor {
41public:
42 ~FindArrayTypes() = default;
43
44 void visit(AST::ArrayType& arrayType) override
45 {
46 m_arrayTypes.append(arrayType);
47 Visitor::visit(arrayType);
48 }
49
50 Vector<std::reference_wrapper<AST::ArrayType>>&& takeArrayTypes()
51 {
52 return WTFMove(m_arrayTypes);
53 }
54
55private:
56 Vector<std::reference_wrapper<AST::ArrayType>> m_arrayTypes;
57};
58
59bool synthesizeArrayOperatorLength(Program& program)
60{
61 FindArrayTypes findArrayTypes;
62 findArrayTypes.checkErrorAndVisit(program);
63 auto arrayTypes = findArrayTypes.takeArrayTypes();
64
65 bool isOperator = true;
66
67 for (auto& arrayType : arrayTypes) {
68 AST::VariableDeclaration variableDeclaration(Lexer::Token(arrayType.get().origin()), AST::Qualifiers(), { arrayType.get().clone() }, String(), WTF::nullopt, WTF::nullopt);
69 AST::VariableDeclarations parameters;
70 parameters.append(WTFMove(variableDeclaration));
71 AST::NativeFunctionDeclaration nativeFunctionDeclaration(AST::FunctionDeclaration(Lexer::Token(arrayType.get().origin()), AST::AttributeBlock(), WTF::nullopt, AST::TypeReference::wrap(Lexer::Token(arrayType.get().origin()), program.intrinsics().uintType()), "operator.length"_str, WTFMove(parameters), WTF::nullopt, isOperator));
72 if (!program.append(WTFMove(nativeFunctionDeclaration)))
73 return false;
74 }
75 return true;
76}
77
78} // namespace WHLSL
79
80} // namespace WebCore
81
82#endif // ENABLE(WEBGPU)
83