142 lines
3.3 KiB
C++
142 lines
3.3 KiB
C++
// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
|
|
|
|
#include "Luau/Quantify.h"
|
|
|
|
#include "Luau/VisitTypeVar.h"
|
|
|
|
LUAU_FASTFLAG(LuauTypecheckOptPass)
|
|
|
|
namespace Luau
|
|
{
|
|
|
|
struct Quantifier final : TypeVarOnceVisitor
|
|
{
|
|
TypeLevel level;
|
|
std::vector<TypeId> generics;
|
|
std::vector<TypePackId> genericPacks;
|
|
bool seenGenericType = false;
|
|
bool seenMutableType = false;
|
|
|
|
explicit Quantifier(TypeLevel level)
|
|
: level(level)
|
|
{
|
|
}
|
|
|
|
void cycle(TypeId) override {}
|
|
void cycle(TypePackId) override {}
|
|
|
|
bool operator()(TypeId ty, const FreeTypeVar& ftv)
|
|
{
|
|
return visit(ty, ftv);
|
|
}
|
|
|
|
template<typename T>
|
|
bool operator()(TypeId ty, const T& t)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
template<typename T>
|
|
bool operator()(TypePackId, const T&)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
bool operator()(TypeId ty, const ConstrainedTypeVar&)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
bool operator()(TypeId ty, const TableTypeVar& ttv)
|
|
{
|
|
return visit(ty, ttv);
|
|
}
|
|
|
|
bool operator()(TypePackId tp, const FreeTypePack& ftp)
|
|
{
|
|
return visit(tp, ftp);
|
|
}
|
|
|
|
bool visit(TypeId ty, const FreeTypeVar& ftv) override
|
|
{
|
|
if (FFlag::LuauTypecheckOptPass)
|
|
seenMutableType = true;
|
|
|
|
if (!level.subsumes(ftv.level))
|
|
return false;
|
|
|
|
*asMutable(ty) = GenericTypeVar{level};
|
|
generics.push_back(ty);
|
|
|
|
return false;
|
|
}
|
|
|
|
bool visit(TypeId ty, const TableTypeVar&) override
|
|
{
|
|
LUAU_ASSERT(getMutable<TableTypeVar>(ty));
|
|
TableTypeVar& ttv = *getMutable<TableTypeVar>(ty);
|
|
|
|
if (FFlag::LuauTypecheckOptPass)
|
|
{
|
|
if (ttv.state == TableState::Generic)
|
|
seenGenericType = true;
|
|
|
|
if (ttv.state == TableState::Free)
|
|
seenMutableType = true;
|
|
}
|
|
|
|
if (ttv.state == TableState::Sealed || ttv.state == TableState::Generic)
|
|
return false;
|
|
if (!level.subsumes(ttv.level))
|
|
{
|
|
if (FFlag::LuauTypecheckOptPass && ttv.state == TableState::Unsealed)
|
|
seenMutableType = true;
|
|
return false;
|
|
}
|
|
|
|
if (ttv.state == TableState::Free)
|
|
{
|
|
ttv.state = TableState::Generic;
|
|
|
|
if (FFlag::LuauTypecheckOptPass)
|
|
seenGenericType = true;
|
|
}
|
|
else if (ttv.state == TableState::Unsealed)
|
|
ttv.state = TableState::Sealed;
|
|
|
|
ttv.level = level;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool visit(TypePackId tp, const FreeTypePack& ftp) override
|
|
{
|
|
if (FFlag::LuauTypecheckOptPass)
|
|
seenMutableType = true;
|
|
|
|
if (!level.subsumes(ftp.level))
|
|
return false;
|
|
|
|
*asMutable(tp) = GenericTypePack{level};
|
|
genericPacks.push_back(tp);
|
|
return true;
|
|
}
|
|
};
|
|
|
|
void quantify(TypeId ty, TypeLevel level)
|
|
{
|
|
Quantifier q{level};
|
|
DenseHashSet<void*> seen{nullptr};
|
|
DEPRECATED_visitTypeVarOnce(ty, q, seen);
|
|
|
|
FunctionTypeVar* ftv = getMutable<FunctionTypeVar>(ty);
|
|
LUAU_ASSERT(ftv);
|
|
ftv->generics = q.generics;
|
|
ftv->genericPacks = q.genericPacks;
|
|
|
|
if (FFlag::LuauTypecheckOptPass && ftv->generics.empty() && ftv->genericPacks.empty() && !q.seenMutableType && !q.seenGenericType)
|
|
ftv->hasNoGenerics = true;
|
|
}
|
|
|
|
} // namespace Luau
|