diff options
Diffstat (limited to 'include/mbgl/style/expression/parse/match.hpp')
-rw-r--r-- | include/mbgl/style/expression/parse/match.hpp | 203 |
1 files changed, 203 insertions, 0 deletions
diff --git a/include/mbgl/style/expression/parse/match.hpp b/include/mbgl/style/expression/parse/match.hpp new file mode 100644 index 0000000000..4406b74002 --- /dev/null +++ b/include/mbgl/style/expression/parse/match.hpp @@ -0,0 +1,203 @@ +#pragma once + +#include <memory> +#include <mbgl/style/expression/check_subtype.hpp> +#include <mbgl/style/expression/match.hpp> +#include <mbgl/style/expression/type.hpp> +#include <mbgl/style/expression/parsing_context.hpp> +#include <mbgl/style/conversion.hpp> + +namespace mbgl { +namespace style { +namespace expression { + +struct ParseMatch { + template <class V> + static ParseResult parse(const V& value, ParsingContext ctx) { + using namespace mbgl::style::conversion; + + assert(isArray(value)); + auto length = arrayLength(value); + if (length < 5) { + ctx.error( + "Expected at least 4 arguments, but found only " + std::to_string(length - 1) + "." + ); + return ParseResult(); + } + + // Expect odd-length array: ["match", input, 2 * (n pairs)..., otherwise] + if (length % 2 != 1) { + ctx.error("Expected an even number of arguments."); + return ParseResult(); + } + + optional<type::Type> inputType; + optional<type::Type> outputType = ctx.expected; + std::vector<std::pair<std::vector<InputType>, + std::unique_ptr<Expression>>> cases; + + for (size_t i = 2; i + 1 < length; i += 2) { + const auto& label = arrayMember(value, i); + + ParsingContext labelContext(ctx, i); + std::vector<InputType> labels; + // Match pair inputs are provided as either a literal value or a + // raw JSON array of string / number / boolean values. + if (isArray(label)) { + auto groupLength = arrayLength(label); + if (groupLength == 0) { + labelContext.error("Expected at least one branch label."); + return ParseResult(); + } + + for (size_t j = 0; j < groupLength; j++) { + const optional<InputType>& inputValue = parseInputValue(arrayMember(label, j), ParsingContext(ctx, i), inputType); + if (!inputValue) { + return ParseResult(); + } + labels.push_back(*inputValue); + } + } else { + const optional<InputType>& inputValue = parseInputValue(label, ParsingContext(ctx, i), inputType); + if (!inputValue) { + return ParseResult(); + } + labels.push_back(*inputValue); + } + + ParseResult output = parseExpression(arrayMember(value, i + 1), ParsingContext(ctx, i + 1, outputType)); + if (!output) { + return ParseResult(); + } + + if (!outputType) { + outputType = (*output)->getType(); + } + + cases.push_back(std::make_pair(std::move(labels), std::move(*output))); + } + + auto input = parseExpression(arrayMember(value, 1), ParsingContext(ctx, 1, inputType)); + if (!input) { + return ParseResult(); + } + + auto otherwise = parseExpression(arrayMember(value, length - 1), ParsingContext(ctx, length - 1, outputType)); + if (!otherwise) { + return ParseResult(); + } + + assert(inputType && outputType); + + return inputType->match( + [&](const type::NumberType&) { + return create<int64_t>(*outputType, std::move(*input), std::move(cases), std::move(*otherwise), ctx); + }, + [&](const type::StringType&) { + return create<std::string>(*outputType, std::move(*input), std::move(cases), std::move(*otherwise), ctx); + }, + [&](const auto&) { + assert(false); + return ParseResult(); + } + ); + } + +private: + template <typename V> + static optional<InputType> parseInputValue(const V& input, ParsingContext ctx, optional<type::Type>& inputType) { + using namespace mbgl::style::conversion; + optional<InputType> result; + optional<type::Type> type; + + auto value = toValue(input); + + if (value) { + value->match( + [&] (uint64_t n) { + if (!Value::isSafeNumericValue(n)) { + ctx.error("Numeric values must be no larger than " + std::to_string(Value::max()) + "."); + } else { + type = {type::Number}; + result = {static_cast<int64_t>(n)}; + } + }, + [&] (int64_t n) { + if (!Value::isSafeNumericValue(n)) { + ctx.error("Numeric values must be no larger than " + std::to_string(Value::max()) + "."); + } else { + type = {type::Number}; + result = {n}; + } + }, + [&] (double n) { + if (!Value::isSafeNumericValue(n)) { + ctx.error("Numeric values must be no larger than " + std::to_string(Value::max()) + "."); + } else if (n != ceilf(n)) { + ctx.error("Numeric branch labels must be integer values."); + } else { + type = {type::Number}; + result = {static_cast<int64_t>(n)}; + } + }, + [&] (const std::string& s) { + type = {type::String}; + result = {s}; + }, + [&] (const auto&) { + ctx.error("Branch labels must be numbers or strings."); + } + ); + } else { + ctx.error("Branch labels must be numbers or strings."); + } + + if (!type) { + return result; + } + + if (!inputType) { + inputType = type; + } else if (checkSubtype(*inputType, *type, ctx)) { + return optional<InputType>(); + } + + return result; + } + + template <typename T> + static ParseResult create(type::Type outputType, + std::unique_ptr<Expression>input, + std::vector<std::pair<std::vector<InputType>, + std::unique_ptr<Expression>>> cases, + std::unique_ptr<Expression> otherwise, + ParsingContext ctx) { + typename Match<T>::Cases typedCases; + + std::size_t index = 2; + for (std::pair<std::vector<InputType>, + std::unique_ptr<Expression>>& pair : cases) { + std::shared_ptr<Expression> result = std::move(pair.second); + for (const InputType& label : pair.first) { + const auto& typedLabel = label.template get<T>(); + if (typedCases.find(typedLabel) != typedCases.end()) { + ctx.error("Branch labels must be unique.", index); + return ParseResult(); + } + typedCases.emplace(typedLabel, result); + } + + index += 2; + } + return ParseResult(std::make_unique<Match<T>>( + outputType, + std::move(input), + std::move(typedCases), + std::move(otherwise) + )); + } +}; + +} // namespace expression +} // namespace style +} // namespace mbgl |