166 lines
6.6 KiB
C++
166 lines
6.6 KiB
C++
#include "catch.hpp"
|
|
|
|
#include <mapnik/expression.hpp>
|
|
#include <mapnik/expression_evaluator.hpp>
|
|
#include <mapnik/expression_string.hpp>
|
|
#include <mapnik/wkt/wkt_factory.hpp>
|
|
#include <mapnik/feature.hpp>
|
|
#include <mapnik/feature_factory.hpp>
|
|
#include <mapnik/unicode.hpp>
|
|
#include <vector>
|
|
|
|
namespace {
|
|
|
|
template <typename Properties>
|
|
mapnik::feature_ptr make_test_feature(mapnik::value_integer id, std::string const& wkt, Properties const& prop)
|
|
{
|
|
auto ctx = std::make_shared<mapnik::context_type>();
|
|
mapnik::feature_ptr feature(mapnik::feature_factory::create(ctx, id));
|
|
mapnik::geometry::geometry<double> geom;
|
|
if (mapnik::from_wkt(wkt, geom))
|
|
{
|
|
feature->set_geometry(std::move(geom));
|
|
}
|
|
|
|
for (auto const& kv : prop)
|
|
{
|
|
feature->put_new(kv.first, kv.second);
|
|
}
|
|
return feature;
|
|
}
|
|
|
|
template <typename Feature, typename Expression>
|
|
mapnik::value_type evaluate(Feature const& feature, Expression const& expr)
|
|
{
|
|
auto value = mapnik::util::apply_visitor(
|
|
mapnik::evaluate<Feature, mapnik::value_type, mapnik::attributes>(
|
|
feature, mapnik::attributes()), expr);
|
|
return value;
|
|
}
|
|
|
|
}
|
|
TEST_CASE("expressions")
|
|
{
|
|
using properties_type = std::vector<std::pair<std::string, mapnik::value> > ;
|
|
mapnik::transcoder tr("utf8");
|
|
|
|
properties_type prop = {{ "foo" , tr.transcode("bar") },
|
|
{ "name" , tr.transcode("Québec")},
|
|
{ "double", mapnik::value_double(1.23456)},
|
|
{ "int" , mapnik::value_integer(123)},
|
|
{ "bool" , mapnik::value_bool(true)},
|
|
{ "null" , mapnik::value_null()}};
|
|
|
|
auto feature = make_test_feature(1, "POINT(100 200)", prop);
|
|
|
|
auto expr = mapnik::parse_expression("[foo]='bar'");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
|
|
// primary expressions
|
|
// null
|
|
expr = mapnik::parse_expression("null");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "null");
|
|
// boolean
|
|
expr = mapnik::parse_expression("true");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "true");
|
|
expr = mapnik::parse_expression("false");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "false");
|
|
// floating point
|
|
expr = mapnik::parse_expression("3.14159");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "3.14159");
|
|
// integer
|
|
expr = mapnik::parse_expression("123");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "123");
|
|
// unicode
|
|
expr = mapnik::parse_expression("'single_quoted_string'");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "'single_quoted_string'");
|
|
expr = mapnik::parse_expression("\"double_quoted_string\"");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "'double_quoted_string'");
|
|
|
|
// floating point constants
|
|
expr = mapnik::parse_expression("pi");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "3.14159");
|
|
expr = mapnik::parse_expression("deg_to_rad");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "0.0174533");
|
|
expr = mapnik::parse_expression("rad_to_deg");
|
|
REQUIRE(mapnik::to_expression_string(*expr) == "57.2958");
|
|
|
|
// unary functions
|
|
// sin / cos
|
|
expr = mapnik::parse_expression("sin(0.25 * pi)/cos(0.25 * pi)");
|
|
double value = evaluate(*feature, *expr).to_double();
|
|
REQUIRE(std::fabs(value - 1.0) < 1e-6);
|
|
// tan
|
|
auto expr2 = mapnik::parse_expression("tan(0.25 * pi)");
|
|
double value2 = evaluate(*feature, *expr).to_double();
|
|
REQUIRE(value == value2);
|
|
// atan
|
|
expr = mapnik::parse_expression("rad_to_deg * atan(1.0)");
|
|
REQUIRE(std::fabs(evaluate(*feature, *expr).to_double() - 45.0) < 1e-6);
|
|
// exp
|
|
expr = mapnik::parse_expression("exp(0.0)");
|
|
REQUIRE(evaluate(*feature, *expr).to_double() == 1.0);
|
|
// abs
|
|
expr = mapnik::parse_expression("abs(cos(-pi))");
|
|
REQUIRE(evaluate(*feature, *expr).to_double() == 1.0);
|
|
// length (string)
|
|
expr = mapnik::parse_expression("length('1234567890')");
|
|
REQUIRE(evaluate(*feature, *expr).to_int() == 10);
|
|
|
|
// binary functions
|
|
// min
|
|
expr = mapnik::parse_expression("min(-0.01, 0.001)");
|
|
REQUIRE(evaluate(*feature, *expr).to_double() == -0.01);
|
|
// max
|
|
expr = mapnik::parse_expression("max(0.01, -0.1)");
|
|
REQUIRE(evaluate(*feature, *expr).to_double() == 0.01);
|
|
// pow
|
|
expr = mapnik::parse_expression("pow(2, 32)");
|
|
REQUIRE(evaluate(*feature, *expr).to_double() == 4294967296.0);
|
|
|
|
// geometry types
|
|
expr = mapnik::parse_expression("[mapnik::geometry_type] = point");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
expr = mapnik::parse_expression("[mapnik::geometry_type] <> linestring");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
expr = mapnik::parse_expression("[mapnik::geometry_type] != polygon");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
expr = mapnik::parse_expression("[mapnik::geometry_type] neq collection");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
expr = mapnik::parse_expression("[mapnik::geometry_type] eq collection");
|
|
REQUIRE(evaluate(*feature, *expr) == false);
|
|
|
|
//unary expression
|
|
expr = mapnik::parse_expression("-123.456");
|
|
REQUIRE(evaluate(*feature, *expr).to_double() == -123.456);
|
|
expr = mapnik::parse_expression("+123.456");
|
|
REQUIRE(evaluate(*feature, *expr).to_double() == 123.456);
|
|
|
|
// multiplicative/additive
|
|
expr = mapnik::parse_expression("(2.0 * 2.0 + 3.0 * 3.0)/(2.0 * 2.0 - 3.0 * 3.0)");
|
|
REQUIRE(evaluate(*feature, *expr).to_double() == -2.6);
|
|
expr2 = mapnik::parse_expression("(2.0 * 2.0 + 3.0 * 3.0)/((2.0 - 3.0) * (2.0 + 3.0))");
|
|
REQUIRE(evaluate(*feature, *expr).to_double() == evaluate(*feature, *expr2).to_double());
|
|
|
|
// logical
|
|
expr = mapnik::parse_expression("[int] = 123 and [double] = 1.23456 && [bool] = true and [null] = null && [foo] = 'bar'");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
expr = mapnik::parse_expression("[int] = 456 or [foo].match('foo') || length([foo]) = 3");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
|
|
// relational
|
|
expr = mapnik::parse_expression("[int] > 100 and [int] gt 100.0 and [double] < 2 and [double] lt 2.0");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
expr = mapnik::parse_expression("[int] >= 123 and [int] ge 123.0 and [double] <= 1.23456 and [double] le 1.23456");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
|
|
// regex
|
|
// replace
|
|
expr = mapnik::parse_expression("[foo].replace('(\\B)|( )','$1 ')");
|
|
REQUIRE(evaluate(*feature, *expr) == tr.transcode("b a r"));
|
|
|
|
// match
|
|
expr = mapnik::parse_expression("[name].match('Québec')");
|
|
REQUIRE(evaluate(*feature, *expr) == true);
|
|
|
|
}
|