blob: 92e0d9a0b61382210ac24f2267a09af81865e6fc [file] [log] [blame]
/*
* Copyright 2016 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef SKSL_FUNCTIONDEFINITION
#define SKSL_FUNCTIONDEFINITION
#include "include/private/SkSLProgramElement.h"
#include "src/sksl/ir/SkSLBlock.h"
#include "src/sksl/ir/SkSLFunctionDeclaration.h"
namespace SkSL {
/**
* A function definition (a declaration plus an associated block of code).
*/
class FunctionDefinition final : public ProgramElement {
public:
inline static constexpr Kind kProgramElementKind = Kind::kFunction;
using FunctionSet = std::unordered_set<const FunctionDeclaration*>;
FunctionDefinition(int line, const FunctionDeclaration* declaration, bool builtin,
std::unique_ptr<Statement> body, FunctionSet referencedBuiltinFunctions)
: INHERITED(line, kProgramElementKind)
, fDeclaration(declaration)
, fBuiltin(builtin)
, fBody(std::move(body))
, fReferencedBuiltinFunctions(std::move(referencedBuiltinFunctions)) {}
/**
* Coerces `return` statements to the return type of the function, and reports errors in the
* function that can't be detected at the individual statement level:
* - `break` and `continue` statements must be in reasonable places.
* - non-void functions are required to return a value on all paths.
* - vertex main() functions don't allow early returns.
*
* This will return a FunctionDefinition even if an error is detected; this leads to better
* diagnostics overall. (Returning null here leads to spurious "function 'f()' was not defined"
* errors when trying to call a function with an error in it.)
*/
static std::unique_ptr<FunctionDefinition> Convert(const Context& context,
int line,
const FunctionDeclaration& function,
std::unique_ptr<Statement> body,
bool builtin);
const FunctionDeclaration& declaration() const {
return *fDeclaration;
}
bool isBuiltin() const {
return fBuiltin;
}
std::unique_ptr<Statement>& body() {
return fBody;
}
const std::unique_ptr<Statement>& body() const {
return fBody;
}
const FunctionSet& referencedBuiltinFunctions() const {
return fReferencedBuiltinFunctions;
}
std::unique_ptr<ProgramElement> clone() const override {
return std::make_unique<FunctionDefinition>(fLine, &this->declaration(),
/*builtin=*/false, this->body()->clone(),
this->referencedBuiltinFunctions());
}
std::string description() const override {
return this->declaration().description() + " " + this->body()->description();
}
private:
const FunctionDeclaration* fDeclaration;
bool fBuiltin;
std::unique_ptr<Statement> fBody;
// We track the builtin functions we reference so that we can ensure that all of them end up
// copied into the final output.
FunctionSet fReferencedBuiltinFunctions;
using INHERITED = ProgramElement;
};
} // namespace SkSL
#endif