1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
#include <string.h>
#include <mbgl/style/conversion_impl.hpp>
#include <mbgl/style/expression/in.hpp>
#include <mbgl/style/expression/type.hpp>
#include <mbgl/util/string.hpp>
namespace mbgl {
namespace style {
namespace expression {
EvaluationResult In::evaluate(const EvaluationContext& params) const {
const EvaluationResult evaluatedNeedle = needle->evaluate(params);
const EvaluationResult evaluatedHeystack = haystack->evaluate(params);
if (!evaluatedNeedle) {
return evaluatedNeedle.error();
}
if (!evaluatedHeystack) {
return evaluatedHeystack.error();
}
type::Type evaluatedNeedleType = typeOf(*evaluatedNeedle);
if (!isComparableRuntimeValue(evaluatedNeedleType)) {
return EvaluationError{"Expected first argument to be of type boolean, string or number, but found " +
toString(evaluatedNeedleType) + " instead."};
}
type::Type evaluatedHeystackType = typeOf(*evaluatedHeystack);
if (!isSearchableRuntimeValue(evaluatedHeystackType)) {
return EvaluationError{"Expected second argument to be of type array or string, but found " +
toString(evaluatedHeystackType) + " instead."};
}
if (evaluatedNeedleType == type::Null || evaluatedHeystackType == type::Null) {
return EvaluationResult(false);
}
if (evaluatedHeystackType == type::String) {
const auto heystackString = evaluatedHeystack->get<std::string>();
std::string needleValue = "";
if (evaluatedNeedleType == type::Boolean) {
needleValue = evaluatedNeedle->get<bool>() ? "true" : " false";
} else if (evaluatedNeedleType == type::String) {
needleValue = evaluatedNeedle->get<std::string>();
} else if (evaluatedNeedleType == type::Number) {
needleValue = std::to_string(evaluatedNeedle->get<double>());
needleValue.erase(needleValue.find_last_not_of('0') + 1, std::string::npos);
needleValue.erase(needleValue.find_last_not_of('.') + 1, std::string::npos);
}
return EvaluationResult(heystackString.find(needleValue) != std::string::npos);
} else {
const auto heystackArray = evaluatedHeystack->get<std::vector<Value>>();
bool result = false;
if (evaluatedNeedleType == type::Boolean) {
auto needleValue = evaluatedNeedle->get<bool>();
result = find(heystackArray.begin(), heystackArray.end(), needleValue) != heystackArray.end();
} else if (evaluatedNeedleType == type::String) {
auto needleValue = evaluatedNeedle->get<std::string>();
result = find(heystackArray.begin(), heystackArray.end(), needleValue) != heystackArray.end();
} else if (evaluatedNeedleType == type::Number) {
auto needleValue = evaluatedNeedle->get<double>();
result = find(heystackArray.begin(), heystackArray.end(), needleValue) != heystackArray.end();
}
return EvaluationResult(result);
}
}
void In::eachChild(const std::function<void(const Expression&)>& visit) const {
visit(*needle);
visit(*haystack);
}
using namespace mbgl::style::conversion;
ParseResult In::parse(const Convertible& value, ParsingContext& ctx) {
assert(isArray(value));
std::size_t length = arrayLength(value);
if (length != 3) {
ctx.error("Expected 2 arguments, but found " + util::toString(length - 1) + " instead.");
return ParseResult();
}
ParseResult needle = ctx.parse(arrayMember(value, 1), 1, {type::Value});
ParseResult haystack = ctx.parse(arrayMember(value, 2), 2, {type::Value});
if (!needle || !haystack) return ParseResult();
type::Type needleType = (*needle)->getType();
type::Type haystackType = (*haystack)->getType();
if (!isComparableType(needleType)) {
ctx.error("Expected first argument to be of type boolean, string or number, but found " + toString(needleType) +
" instead.");
return ParseResult();
}
return ParseResult(std::make_unique<In>(std::move(*needle), std::move(*haystack)));
}
bool In::isComparableType(type::Type type) {
return type == type::Boolean || type == type::String || type == type::Number || type == type::Null ||
type == type::Value;
}
bool In::isComparableRuntimeValue(type::Type type) {
return type == type::Boolean || type == type::String || type == type::Number || type == type::Null;
}
bool In::isSearchableRuntimeValue(type::Type type) {
return type == type::String || type.is<type::Array>() || type == type::Null;
}
} // namespace expression
} // namespace style
} // namespace mbgl
|