mirror of
https://github.com/luau-lang/luau.git
synced 2024-12-13 13:30:40 +00:00
74c532053f
New Solver * New algorithm for inferring the types of locals that have no annotations. This algorithm is very conservative by default, but is augmented with some control flow awareness to handle most common scenarios. * Fix bugs in type inference of tables * Improve performance of by switching out standard C++ containers for `DenseHashMap` * Infrastructure to support clearer error messages in strict mode Native Code Generation * Fix a lowering issue with buffer.writeu8 and 0x80-0xff values: A constant argument wasn't truncated to the target type range and that causes an assertion failure in `build.mov`. * Store full lightuserdata value in loop iteration protocol lowering * Add analysis to compute function bytecode distribution * This includes a class to analyze the bytecode operator distribution per function and a CLI tool that produces a JSON report. See the new cmake target `Luau.Bytecode.CLI` --------- Co-authored-by: Aaron Weiss <aaronweiss@roblox.com> Co-authored-by: Alexander McCord <amccord@roblox.com> Co-authored-by: Andy Friesen <afriesen@roblox.com> Co-authored-by: Aviral Goel <agoel@roblox.com> Co-authored-by: Lily Brown <lbrown@roblox.com> Co-authored-by: Vyacheslav Egorov <vegorov@roblox.com>
103 lines
3.8 KiB
C++
103 lines
3.8 KiB
C++
// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
|
|
#pragma once
|
|
|
|
#include "Luau/Def.h"
|
|
#include "Luau/LValue.h"
|
|
#include "Luau/Location.h"
|
|
#include "Luau/NotNull.h"
|
|
#include "Luau/Type.h"
|
|
#include "Luau/DenseHash.h"
|
|
#include "Luau/Symbol.h"
|
|
#include "Luau/Unifiable.h"
|
|
|
|
#include <unordered_map>
|
|
#include <optional>
|
|
#include <memory>
|
|
|
|
namespace Luau
|
|
{
|
|
|
|
struct Scope;
|
|
|
|
using ScopePtr = std::shared_ptr<Scope>;
|
|
|
|
struct Binding
|
|
{
|
|
TypeId typeId;
|
|
Location location;
|
|
bool deprecated = false;
|
|
std::string deprecatedSuggestion;
|
|
std::optional<std::string> documentationSymbol;
|
|
};
|
|
|
|
struct Scope
|
|
{
|
|
explicit Scope(TypePackId returnType); // root scope
|
|
explicit Scope(const ScopePtr& parent, int subLevel = 0); // child scope. Parent must not be nullptr.
|
|
|
|
const ScopePtr parent; // null for the root
|
|
|
|
// All the children of this scope.
|
|
std::vector<NotNull<Scope>> children;
|
|
std::unordered_map<Symbol, Binding> bindings;
|
|
TypePackId returnType;
|
|
std::optional<TypePackId> varargPack;
|
|
|
|
TypeLevel level;
|
|
|
|
std::unordered_map<Name, TypeFun> exportedTypeBindings;
|
|
std::unordered_map<Name, TypeFun> privateTypeBindings;
|
|
std::unordered_map<Name, Location> typeAliasLocations;
|
|
std::unordered_map<Name, Location> typeAliasNameLocations;
|
|
std::unordered_map<Name, ModuleName> importedModules; // Mapping from the name in the require statement to the internal moduleName.
|
|
std::unordered_map<Name, std::unordered_map<Name, TypeFun>> importedTypeBindings;
|
|
|
|
DenseHashSet<Name> builtinTypeNames{""};
|
|
void addBuiltinTypeBinding(const Name& name, const TypeFun& tyFun);
|
|
|
|
std::optional<TypeId> lookup(Symbol sym) const;
|
|
std::optional<TypeId> lookupUnrefinedType(DefId def) const;
|
|
std::optional<TypeId> lookup(DefId def) const;
|
|
std::optional<std::pair<TypeId, Scope*>> lookupEx(DefId def);
|
|
std::optional<std::pair<Binding*, Scope*>> lookupEx(Symbol sym);
|
|
|
|
std::optional<TypeFun> lookupType(const Name& name) const;
|
|
std::optional<TypeFun> lookupImportedType(const Name& moduleAlias, const Name& name) const;
|
|
|
|
std::unordered_map<Name, TypePackId> privateTypePackBindings;
|
|
std::optional<TypePackId> lookupPack(const Name& name) const;
|
|
|
|
// WARNING: This function linearly scans for a string key of equal value! It is thus O(n**2)
|
|
std::optional<Binding> linearSearchForBinding(const std::string& name, bool traverseScopeChain = true) const;
|
|
|
|
RefinementMap refinements;
|
|
|
|
// This can be viewed as the "unrefined" type of each binding.
|
|
DenseHashMap<const Def*, TypeId> lvalueTypes{nullptr};
|
|
|
|
// Luau values are routinely refined more narrowly than their actual
|
|
// inferred type through control flow statements. We retain those refined
|
|
// types here.
|
|
DenseHashMap<const Def*, TypeId> rvalueRefinements{nullptr};
|
|
|
|
void inheritAssignments(const ScopePtr& childScope);
|
|
void inheritRefinements(const ScopePtr& childScope);
|
|
|
|
// For mutually recursive type aliases, it's important that
|
|
// they use the same types for the same names.
|
|
// For instance, in `type Tree<T> { data: T, children: Forest<T> } type Forest<T> = {Tree<T>}`
|
|
// we need that the generic type `T` in both cases is the same, so we use a cache.
|
|
std::unordered_map<Name, TypeId> typeAliasTypeParameters;
|
|
std::unordered_map<Name, TypePackId> typeAliasTypePackParameters;
|
|
};
|
|
|
|
// Returns true iff the left scope encloses the right scope. A Scope* equal to
|
|
// nullptr is considered to be the outermost-possible scope.
|
|
bool subsumesStrict(Scope* left, Scope* right);
|
|
|
|
// Returns true if the left scope encloses the right scope, or if they are the
|
|
// same scope. As in subsumesStrict(), nullptr is considered to be the
|
|
// outermost-possible scope.
|
|
bool subsumes(Scope* left, Scope* right);
|
|
|
|
} // namespace Luau
|