WHLSLRecursiveTypeChecker.cpp [plain text]
#include "config.h"
#include "WHLSLRecursiveTypeChecker.h"
#if ENABLE(WEBGPU)
#include "WHLSLStructureDefinition.h"
#include "WHLSLTypeDefinition.h"
#include "WHLSLProgram.h"
#include "WHLSLTypeReference.h"
#include "WHLSLVisitor.h"
#include <wtf/HashSet.h>
namespace WebCore {
namespace WHLSL {
class RecursiveTypeChecker : public Visitor {
public:
void visit(AST::TypeDefinition&) override;
void visit(AST::StructureDefinition&) override;
void visit(AST::TypeReference&) override;
void visit(AST::ReferenceType&) override;
private:
HashSet<AST::Type*> m_startedVisiting;
HashSet<AST::Type*> m_finishedVisiting;
};
#define START_VISITING(t) \
do { \
if (m_finishedVisiting.contains(t)) \
return; \
auto resultStartedVisiting = m_startedVisiting.add(t); \
if (!resultStartedVisiting.isNewEntry) { \
setError(Error("Cannot declare recursive types.", (t)->codeLocation())); \
return; \
} \
} while (false);
#define END_VISITING(t) \
do { \
auto resultFinishedVisiting = m_finishedVisiting.add(t); \
ASSERT_UNUSED(resultFinishedVisiting, resultFinishedVisiting.isNewEntry); \
} while (false);
void RecursiveTypeChecker::visit(AST::TypeDefinition& typeDefinition)
{
START_VISITING(&typeDefinition);
Visitor::visit(typeDefinition);
END_VISITING(&typeDefinition);
}
void RecursiveTypeChecker::visit(AST::StructureDefinition& structureDefinition)
{
START_VISITING(&structureDefinition);
Visitor::visit(structureDefinition);
END_VISITING(&structureDefinition);
}
void RecursiveTypeChecker::visit(AST::TypeReference& typeReference)
{
START_VISITING(&typeReference);
for (auto& typeArgument : typeReference.typeArguments())
checkErrorAndVisit(typeArgument);
if (typeReference.maybeResolvedType()) checkErrorAndVisit(typeReference.resolvedType());
END_VISITING(&typeReference);
}
void RecursiveTypeChecker::visit(AST::ReferenceType&)
{
}
Expected<void, Error> checkRecursiveTypes(Program& program)
{
RecursiveTypeChecker recursiveTypeChecker;
for (auto& typeDefinition : program.typeDefinitions())
recursiveTypeChecker.checkErrorAndVisit(typeDefinition);
for (auto& structureDefinition : program.structureDefinitions())
recursiveTypeChecker.checkErrorAndVisit(structureDefinition);
return recursiveTypeChecker.result();
}
}
}
#endif // ENABLE(WEBGPU)