2a8ebc4906
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
85 lines
2.6 KiB
C++
85 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <mln/style/expression/expression.hpp>
|
|
#include <mln/style/expression/parsing_context.hpp>
|
|
#include <mln/style/conversion.hpp>
|
|
|
|
#include <memory>
|
|
#include <map>
|
|
#include <optional>
|
|
|
|
namespace mln {
|
|
namespace style {
|
|
namespace expression {
|
|
|
|
class Let : public Expression {
|
|
public:
|
|
using Bindings = std::map<std::string, std::shared_ptr<Expression>>;
|
|
|
|
Let(Bindings bindings_, std::unique_ptr<Expression> result_)
|
|
: Expression(
|
|
Kind::Let, result_->getType(), collectDependencies(bindings_) | depsOf(result_) | Dependency::Bind),
|
|
bindings(std::move(bindings_)),
|
|
result(std::move(result_)) {}
|
|
|
|
static ParseResult parse(const mln::style::conversion::Convertible&, ParsingContext&);
|
|
|
|
EvaluationResult evaluate(const EvaluationContext& params) const override;
|
|
void eachChild(const std::function<void(const Expression&)>&) const override;
|
|
|
|
bool operator==(const Expression& e) const noexcept override {
|
|
if (e.getKind() == Kind::Let) {
|
|
const auto* rhs = static_cast<const Let*>(&e);
|
|
return *result == *(rhs->result);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
std::vector<std::optional<Value>> possibleOutputs() const override;
|
|
|
|
Expression* getResult() const noexcept { return result.get(); }
|
|
|
|
mln::Value serialize() const override;
|
|
std::string getOperator() const override { return "let"; }
|
|
|
|
private:
|
|
Bindings bindings;
|
|
std::unique_ptr<Expression> result;
|
|
};
|
|
|
|
class Var : public Expression {
|
|
public:
|
|
Var(std::string name_, std::shared_ptr<Expression> value_)
|
|
: Expression(Kind::Var, value_->getType(), depsOf(value_) | Dependency::Var),
|
|
name(std::move(name_)),
|
|
value(std::move(value_)) {}
|
|
|
|
static ParseResult parse(const mln::style::conversion::Convertible&, ParsingContext&);
|
|
|
|
EvaluationResult evaluate(const EvaluationContext& params) const override;
|
|
void eachChild(const std::function<void(const Expression&)>&) const override;
|
|
|
|
bool operator==(const Expression& e) const noexcept override {
|
|
if (e.getKind() == Kind::Var) {
|
|
auto rhs = static_cast<const Var*>(&e);
|
|
return *value == *(rhs->value);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
std::vector<std::optional<Value>> possibleOutputs() const override;
|
|
|
|
mln::Value serialize() const override;
|
|
std::string getOperator() const override { return "var"; }
|
|
|
|
const std::shared_ptr<Expression>& getBoundExpression() const noexcept { return value; }
|
|
|
|
private:
|
|
std::string name;
|
|
std::shared_ptr<Expression> value;
|
|
};
|
|
|
|
} // namespace expression
|
|
} // namespace style
|
|
} // namespace mln
|