6f20b8d5f7
Change-Id: I895c3d94c9a49cc0350b287d3aa722050b91295e Reviewed-on: https://skia-review.googlesource.com/c/skia/+/443036 Commit-Queue: Ethan Nicholas <ethannicholas@google.com> Reviewed-by: John Stiles <johnstiles@google.com>
115 lines
3.7 KiB
C++
115 lines
3.7 KiB
C++
/*
|
|
* Copyright 2021 Google LLC.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license that can be
|
|
* found in the LICENSE file.
|
|
*/
|
|
|
|
#ifndef SKSL_DSL_FUNCTION
|
|
#define SKSL_DSL_FUNCTION
|
|
|
|
#include "include/sksl/DSLBlock.h"
|
|
#include "include/sksl/DSLExpression.h"
|
|
#include "include/sksl/DSLType.h"
|
|
#include "include/sksl/DSLVar.h"
|
|
#include "include/sksl/DSLWrapper.h"
|
|
|
|
namespace SkSL {
|
|
|
|
class Block;
|
|
class FunctionDeclaration;
|
|
class Variable;
|
|
|
|
namespace dsl {
|
|
|
|
class DSLType;
|
|
|
|
class DSLFunction {
|
|
public:
|
|
template<class... Parameters>
|
|
DSLFunction(const DSLType& returnType, skstd::string_view name, Parameters&... parameters)
|
|
: DSLFunction(DSLModifiers(), returnType, name, parameters...) {}
|
|
|
|
template<class... Parameters>
|
|
DSLFunction(const DSLModifiers& modifiers, const DSLType& returnType, skstd::string_view name,
|
|
Parameters&... parameters) {
|
|
SkTArray<DSLParameter*> parameterArray;
|
|
parameterArray.reserve_back(sizeof...(parameters));
|
|
|
|
// in C++17, we could just do:
|
|
// (parameterArray.push_back(¶meters), ...);
|
|
int unused[] = {0, (static_cast<void>(parameterArray.push_back(¶meters)), 0)...};
|
|
static_cast<void>(unused);
|
|
// We can't have a default parameter and a template parameter pack at the same time, so
|
|
// unfortunately we can't capture position info from this overload.
|
|
this->init(modifiers, returnType, name, std::move(parameterArray), PositionInfo());
|
|
}
|
|
|
|
DSLFunction(const DSLType& returnType, skstd::string_view name,
|
|
SkTArray<DSLParameter*> parameters, PositionInfo pos = PositionInfo::Capture()) {
|
|
this->init(DSLModifiers(), returnType, name, std::move(parameters), pos);
|
|
}
|
|
|
|
DSLFunction(const DSLModifiers& modifiers, const DSLType& returnType, skstd::string_view name,
|
|
SkTArray<DSLParameter*> parameters, PositionInfo pos = PositionInfo::Capture()) {
|
|
this->init(modifiers, returnType, name, std::move(parameters), pos);
|
|
}
|
|
|
|
DSLFunction(const SkSL::FunctionDeclaration* decl)
|
|
: fDecl(decl) {}
|
|
|
|
virtual ~DSLFunction() = default;
|
|
|
|
template<class... Stmt>
|
|
void define(Stmt... stmts) {
|
|
DSLBlock block = DSLBlock(DSLStatement(std::move(stmts))...);
|
|
this->define(std::move(block));
|
|
}
|
|
|
|
void define(DSLBlock block);
|
|
|
|
/**
|
|
* Invokes the function with the given arguments.
|
|
*/
|
|
template<class... Args>
|
|
DSLExpression operator()(Args&&... args) {
|
|
SkTArray<DSLWrapper<DSLExpression>> argArray;
|
|
argArray.reserve_back(sizeof...(args));
|
|
this->collectArgs(argArray, std::forward<Args>(args)...);
|
|
return this->call(std::move(argArray));
|
|
}
|
|
|
|
/**
|
|
* Invokes the function with the given arguments.
|
|
*/
|
|
DSLExpression call(SkTArray<DSLWrapper<DSLExpression>> args);
|
|
|
|
private:
|
|
void collectArgs(SkTArray<DSLWrapper<DSLExpression>>& args) {}
|
|
|
|
template<class... RemainingArgs>
|
|
void collectArgs(SkTArray<DSLWrapper<DSLExpression>>& args, DSLVar& var,
|
|
RemainingArgs&&... remaining) {
|
|
args.push_back(DSLWrapper<DSLExpression>(var));
|
|
collectArgs(args, std::forward<RemainingArgs>(remaining)...);
|
|
}
|
|
|
|
template<class... RemainingArgs>
|
|
void collectArgs(SkTArray<DSLWrapper<DSLExpression>>& args, DSLExpression expr,
|
|
RemainingArgs&&... remaining) {
|
|
args.push_back(DSLWrapper<DSLExpression>(std::move(expr)));
|
|
collectArgs(args, std::forward<RemainingArgs>(remaining)...);
|
|
}
|
|
|
|
void init(DSLModifiers modifiers, const DSLType& returnType, skstd::string_view name,
|
|
SkTArray<DSLParameter*> params, PositionInfo pos);
|
|
|
|
const SkSL::FunctionDeclaration* fDecl = nullptr;
|
|
};
|
|
|
|
} // namespace dsl
|
|
|
|
} // namespace SkSL
|
|
|
|
#endif
|