WHLSLSynthesizeConstructors.cpp [plain text]
#include "config.h"
#include "WHLSLSynthesizeConstructors.h"
#if ENABLE(WEBGPU)
#include "WHLSLArrayType.h"
#include "WHLSLArrayReferenceType.h"
#include "WHLSLEnumerationDefinition.h"
#include "WHLSLInferTypes.h"
#include "WHLSLNativeFunctionDeclaration.h"
#include "WHLSLNativeTypeDeclaration.h"
#include "WHLSLPointerType.h"
#include "WHLSLProgram.h"
#include "WHLSLStructureDefinition.h"
#include "WHLSLTypeReference.h"
#include "WHLSLUnnamedTypeHash.h"
#include "WHLSLVariableDeclaration.h"
#include "WHLSLVisitor.h"
namespace WebCore {
namespace WHLSL {
class FindAllTypes : public Visitor {
public:
virtual ~FindAllTypes() = default;
void visit(AST::PointerType& pointerType) override
{
m_unnamedTypes.add(UnnamedTypeKey { pointerType });
Visitor::visit(pointerType);
}
void visit(AST::ArrayReferenceType& arrayReferenceType) override
{
m_unnamedTypes.add(UnnamedTypeKey { arrayReferenceType });
Visitor::visit(arrayReferenceType);
}
void visit(AST::ArrayType& arrayType) override
{
m_unnamedTypes.add(UnnamedTypeKey { arrayType });
Visitor::visit(arrayType);
}
void visit(AST::EnumerationDefinition& enumerationDefinition) override
{
appendNamedType(enumerationDefinition);
Visitor::visit(enumerationDefinition);
}
void visit(AST::StructureDefinition& structureDefinition) override
{
appendNamedType(structureDefinition);
Visitor::visit(structureDefinition);
}
void visit(AST::NativeTypeDeclaration& nativeTypeDeclaration) override
{
appendNamedType(nativeTypeDeclaration);
Visitor::visit(nativeTypeDeclaration);
}
HashSet<UnnamedTypeKey> takeUnnamedTypes()
{
return WTFMove(m_unnamedTypes);
}
Vector<std::reference_wrapper<AST::NamedType>> takeNamedTypes()
{
return WTFMove(m_namedTypes);
}
private:
void appendNamedType(AST::NamedType& type)
{
#if ASSERT_ENABLED
for (auto& entry : m_namedTypes)
ASSERT(&entry.get().unifyNode() != &type.unifyNode());
#endif
m_namedTypes.append(type);
}
HashSet<UnnamedTypeKey> m_unnamedTypes;
Vector<std::reference_wrapper<AST::NamedType>> m_namedTypes;
};
Expected<void, Error> synthesizeConstructors(Program& program)
{
FindAllTypes findAllTypes;
findAllTypes.checkErrorAndVisit(program);
auto unnamedTypes = findAllTypes.takeUnnamedTypes();
auto namedTypes = findAllTypes.takeNamedTypes();
bool isOperator = true;
for (auto& unnamedTypeKey : unnamedTypes) {
auto& unnamedType = unnamedTypeKey.unnamedType();
auto location = unnamedType.codeLocation();
{
auto variableDeclaration = makeUniqueRef<AST::VariableDeclaration>(location, AST::Qualifiers(), &unnamedType, String(), nullptr, nullptr);
AST::VariableDeclarations parameters;
parameters.append(WTFMove(variableDeclaration));
AST::NativeFunctionDeclaration copyConstructor(AST::FunctionDeclaration(location, AST::AttributeBlock(), WTF::nullopt, unnamedType, "operator cast"_str, WTFMove(parameters), nullptr, isOperator, ParsingMode::StandardLibrary));
program.append(WTFMove(copyConstructor));
}
AST::NativeFunctionDeclaration defaultConstructor(AST::FunctionDeclaration(location, AST::AttributeBlock(), WTF::nullopt, unnamedType, "operator cast"_str, AST::VariableDeclarations(), nullptr, isOperator, ParsingMode::StandardLibrary));
if (!program.append(WTFMove(defaultConstructor)))
return makeUnexpected(Error("Could not synthesize default constructor"));
}
for (auto& namedType : namedTypes) {
if (matches(namedType, program.intrinsics().voidType()))
continue;
if (is<AST::NativeTypeDeclaration>(static_cast<AST::NamedType&>(namedType)) && downcast<AST::NativeTypeDeclaration>(static_cast<AST::NamedType&>(namedType)).isAtomic())
continue;
auto location = namedType.get().codeLocation();
auto variableDeclaration = makeUniqueRef<AST::VariableDeclaration>(location, AST::Qualifiers(), AST::TypeReference::wrap(location, namedType.get()), String(), nullptr, nullptr);
AST::VariableDeclarations parameters;
parameters.append(WTFMove(variableDeclaration));
AST::NativeFunctionDeclaration copyConstructor(AST::FunctionDeclaration(location, AST::AttributeBlock(), WTF::nullopt, AST::TypeReference::wrap(location, namedType.get()), "operator cast"_str, WTFMove(parameters), nullptr, isOperator, ParsingMode::StandardLibrary));
program.append(WTFMove(copyConstructor));
if (is<AST::NativeTypeDeclaration>(static_cast<AST::NamedType&>(namedType))) {
auto& nativeTypeDeclaration = downcast<AST::NativeTypeDeclaration>(static_cast<AST::NamedType&>(namedType));
if (nativeTypeDeclaration.isOpaqueType())
continue;
}
AST::NativeFunctionDeclaration defaultConstructor(AST::FunctionDeclaration(location, AST::AttributeBlock(), WTF::nullopt, AST::TypeReference::wrap(location, namedType.get()), "operator cast"_str, AST::VariableDeclarations(), nullptr, isOperator, ParsingMode::StandardLibrary));
if (!program.append(WTFMove(defaultConstructor)))
return makeUnexpected(Error("Could not synthesize default constructor"));
}
return { };
}
}
}
#endif // ENABLE(WEBGPU)