d7cc093f1f
Constructors such as `float[2](0, 0)` add a type to the symbol table; this type needs to be copied into the new symbol table if the constructor is cloned by the inliner. Change-Id: Ifa8d2dec87103c6223ce493e2201a904c14c2137 Bug: oss-fuzz:28050 Reviewed-on: https://skia-review.googlesource.com/c/skia/+/339168 Auto-Submit: John Stiles <johnstiles@google.com> Commit-Queue: Brian Osman <brianosman@google.com> Reviewed-by: Brian Osman <brianosman@google.com>
103 lines
4.0 KiB
C++
103 lines
4.0 KiB
C++
/*
|
|
* Copyright 2020 Google LLC
|
|
*
|
|
* Use of this source code is governed by a BSD-style license that can be
|
|
* found in the LICENSE file.
|
|
*/
|
|
|
|
#ifndef SKSL_INLINER
|
|
#define SKSL_INLINER
|
|
|
|
#include <memory>
|
|
#include <unordered_map>
|
|
|
|
#include "src/sksl/ir/SkSLProgram.h"
|
|
#include "src/sksl/ir/SkSLVariableReference.h"
|
|
|
|
namespace SkSL {
|
|
|
|
class Block;
|
|
class Context;
|
|
class Expression;
|
|
class FunctionCall;
|
|
class FunctionDefinition;
|
|
struct InlineCandidate;
|
|
struct InlineCandidateList;
|
|
class ModifiersPool;
|
|
class Statement;
|
|
class SymbolTable;
|
|
class Variable;
|
|
|
|
/**
|
|
* Converts a FunctionCall in the IR to a set of statements to be injected ahead of the function
|
|
* call, and a replacement expression. Can also detect cases where inlining isn't cleanly possible
|
|
* (e.g. return statements nested inside of a loop construct). The inliner isn't able to guarantee
|
|
* identical-to-GLSL execution order if the inlined function has visible side effects.
|
|
*/
|
|
class Inliner {
|
|
public:
|
|
Inliner(const Context* context, const ShaderCapsClass* caps) : fContext(context), fCaps(caps) {}
|
|
|
|
void reset(ModifiersPool* modifiers, const Program::Settings*);
|
|
|
|
/** Inlines any eligible functions that are found. Returns true if any changes are made. */
|
|
bool analyze(const std::vector<std::unique_ptr<ProgramElement>>& elements,
|
|
SymbolTable* symbols,
|
|
ProgramUsage* usage);
|
|
|
|
private:
|
|
using VariableRewriteMap = std::unordered_map<const Variable*, std::unique_ptr<Expression>>;
|
|
|
|
String uniqueNameForInlineVar(const String& baseName, SymbolTable* symbolTable);
|
|
|
|
void buildCandidateList(const std::vector<std::unique_ptr<ProgramElement>>& elements,
|
|
SymbolTable* symbols, ProgramUsage* usage,
|
|
InlineCandidateList* candidateList);
|
|
|
|
std::unique_ptr<Expression> inlineExpression(int offset,
|
|
VariableRewriteMap* varMap,
|
|
SymbolTable* symbolTableForExpression,
|
|
const Expression& expression);
|
|
std::unique_ptr<Statement> inlineStatement(int offset,
|
|
VariableRewriteMap* varMap,
|
|
SymbolTable* symbolTableForStatement,
|
|
const Expression* resultExpr,
|
|
bool haveEarlyReturns,
|
|
const Statement& statement,
|
|
bool isBuiltinCode);
|
|
|
|
using InlinabilityCache = std::unordered_map<const FunctionDeclaration*, bool>;
|
|
bool candidateCanBeInlined(const InlineCandidate& candidate, InlinabilityCache* cache);
|
|
|
|
using FunctionSizeCache = std::unordered_map<const FunctionDeclaration*, int>;
|
|
int getFunctionSize(const FunctionDeclaration& fnDecl, FunctionSizeCache* cache);
|
|
|
|
/**
|
|
* Processes the passed-in FunctionCall expression. The FunctionCall expression should be
|
|
* replaced with `fReplacementExpr`. If non-null, `fInlinedBody` should be inserted immediately
|
|
* above the statement containing the inlined expression.
|
|
*/
|
|
struct InlinedCall {
|
|
std::unique_ptr<Block> fInlinedBody;
|
|
std::unique_ptr<Expression> fReplacementExpr;
|
|
};
|
|
InlinedCall inlineCall(FunctionCall*, SymbolTable*, const FunctionDeclaration* caller);
|
|
|
|
/** Adds a scope to inlined bodies returned by `inlineCall`, if one is required. */
|
|
void ensureScopedBlocks(Statement* inlinedBody, Statement* parentStmt);
|
|
|
|
/** Checks whether inlining is viable for a FunctionCall, modulo recursion and function size. */
|
|
bool isSafeToInline(const FunctionDefinition* functionDef);
|
|
|
|
const Context* fContext = nullptr;
|
|
ModifiersPool* fModifiers = nullptr;
|
|
const Program::Settings* fSettings = nullptr;
|
|
const ShaderCapsClass* fCaps = nullptr;
|
|
int fInlineVarCounter = 0;
|
|
int fInlinedStatementCounter = 0;
|
|
};
|
|
|
|
} // namespace SkSL
|
|
|
|
#endif // SKSL_INLINER
|