2a8ebc4906
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
86 lines
3.0 KiB
C++
86 lines
3.0 KiB
C++
#pragma once
|
|
|
|
#include <mln/style/expression/expression.hpp>
|
|
#include <mln/style/expression/parsing_context.hpp>
|
|
#include <mln/style/expression/get_covering_stops.hpp>
|
|
#include <mln/style/expression/interpolator.hpp>
|
|
#include <mln/style/conversion.hpp>
|
|
|
|
#include <memory>
|
|
#include <map>
|
|
#include <cmath>
|
|
#include <optional>
|
|
|
|
namespace mln {
|
|
namespace style {
|
|
namespace expression {
|
|
|
|
ParseResult parseInterpolate(const mln::style::conversion::Convertible& value, ParsingContext& ctx);
|
|
|
|
class Interpolate : public Expression {
|
|
public:
|
|
Interpolate(const type::Type& type_,
|
|
Interpolator interpolator_,
|
|
std::unique_ptr<Expression> input_,
|
|
std::map<double, std::unique_ptr<Expression>> stops_);
|
|
|
|
const std::unique_ptr<Expression>& getInput() const noexcept { return input; }
|
|
const Interpolator& getInterpolator() const noexcept { return interpolator; }
|
|
|
|
void eachChild(const std::function<void(const Expression&)>& visit) const override {
|
|
visit(*input);
|
|
for (const auto& stop : stops) {
|
|
visit(*stop.second);
|
|
}
|
|
}
|
|
|
|
std::size_t getStopCount() const { return stops.size(); }
|
|
|
|
void eachStop(const std::function<void(double, const Expression&)>& visit) const {
|
|
for (const auto& stop : stops) {
|
|
visit(stop.first, *stop.second);
|
|
}
|
|
}
|
|
|
|
// Return the smallest range of stops that covers the interval [lower, upper]
|
|
Range<float> getCoveringStops(const double lower, const double upper) const noexcept {
|
|
return ::mln::style::expression::getCoveringStops(stops, lower, upper);
|
|
}
|
|
|
|
double interpolationFactor(const Range<double>& inputLevels, const double inputValue) const {
|
|
return interpolator.match(
|
|
[&](const auto& interp) { return interp.interpolationFactor(inputLevels, inputValue); });
|
|
}
|
|
|
|
bool operator==(const Expression& e) const noexcept override {
|
|
if (e.getKind() == Kind::Interpolate) {
|
|
const auto* rhs = static_cast<const Interpolate*>(&e);
|
|
if (interpolator != rhs->interpolator || *input != *(rhs->input) || stops.size() != rhs->stops.size()) {
|
|
return false;
|
|
}
|
|
|
|
return Expression::childrenEqual(stops, rhs->stops);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
std::vector<std::optional<Value>> possibleOutputs() const override;
|
|
mln::Value serialize() const override;
|
|
std::string getOperator() const override { return "interpolate"; }
|
|
|
|
protected:
|
|
const Interpolator interpolator;
|
|
const std::unique_ptr<Expression> input;
|
|
const std::map<double, std::unique_ptr<Expression>> stops;
|
|
};
|
|
|
|
ParseResult createInterpolate(type::Type type,
|
|
Interpolator interpolator,
|
|
std::unique_ptr<Expression> input,
|
|
std::map<double, std::unique_ptr<Expression>> stops,
|
|
ParsingContext& ctx);
|
|
|
|
} // namespace expression
|
|
} // namespace style
|
|
} // namespace mln
|