263 lines
10 KiB
C++
263 lines
10 KiB
C++
#include "common/range_tools.hpp"
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
#include "customizer/cell_customizer.hpp"
|
|
#include "partition/multi_level_partition.hpp"
|
|
#include "util/static_graph.hpp"
|
|
|
|
using namespace osrm;
|
|
using namespace osrm::customizer;
|
|
using namespace osrm::partition;
|
|
using namespace osrm::util;
|
|
|
|
namespace
|
|
{
|
|
struct MockEdge
|
|
{
|
|
NodeID start;
|
|
NodeID target;
|
|
EdgeWeight weight;
|
|
};
|
|
|
|
auto makeGraph(const std::vector<MockEdge> &mock_edges)
|
|
{
|
|
struct EdgeData
|
|
{
|
|
EdgeWeight weight;
|
|
bool forward;
|
|
bool backward;
|
|
};
|
|
using Edge = static_graph_details::SortableEdgeWithData<EdgeData>;
|
|
std::vector<Edge> edges;
|
|
std::size_t max_id = 0;
|
|
for (const auto &m : mock_edges)
|
|
{
|
|
max_id = std::max<std::size_t>(max_id, std::max(m.start, m.target));
|
|
edges.push_back(Edge{m.start, m.target, m.weight, true, false});
|
|
edges.push_back(Edge{m.target, m.start, m.weight, false, true});
|
|
}
|
|
std::sort(edges.begin(), edges.end());
|
|
return util::StaticGraph<EdgeData>(max_id + 1, edges);
|
|
}
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE(cell_customization_tests)
|
|
|
|
BOOST_AUTO_TEST_CASE(two_level_test)
|
|
{
|
|
// node: 0 1 2 3
|
|
std::vector<CellID> l1{{0, 0, 1, 1}};
|
|
MultiLevelPartition mlp{{l1}, {2}};
|
|
|
|
BOOST_REQUIRE_EQUAL(mlp.GetNumberOfLevels(), 2);
|
|
|
|
std::vector<MockEdge> edges = {{0, 1, 1}, {0, 2, 1}, {2, 3, 1}, {3, 1, 1}, {3, 2, 1}};
|
|
|
|
auto graph = makeGraph(edges);
|
|
|
|
CellStorage storage(mlp, graph);
|
|
CellCustomizer customizer(mlp);
|
|
|
|
auto cell_1_0 = storage.GetCell(1, 0);
|
|
auto cell_1_1 = storage.GetCell(1, 1);
|
|
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetSourceNodes(), 1);
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetDestinationNodes(), 1);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetSourceNodes(), 2);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetDestinationNodes(), 2);
|
|
|
|
CHECK_EQUAL_RANGE(cell_1_0.GetSourceNodes(), 0);
|
|
CHECK_EQUAL_RANGE(cell_1_0.GetDestinationNodes(), 1);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetSourceNodes(), 2, 3);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetDestinationNodes(), 2, 3);
|
|
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetOutWeight(0), 1);
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetOutWeight(1), 0);
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetInWeight(1), 1);
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetInWeight(0), 0);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetOutWeight(2), 2);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetInWeight(3), 2);
|
|
|
|
customizer.Customize(graph, storage, 1, 0);
|
|
customizer.Customize(graph, storage, 1, 1);
|
|
|
|
// cell 0
|
|
// check row source -> destination
|
|
CHECK_EQUAL_RANGE(cell_1_0.GetOutWeight(0), 1);
|
|
// check column destination -> source
|
|
CHECK_EQUAL_RANGE(cell_1_0.GetInWeight(1), 1);
|
|
|
|
// cell 1
|
|
// check row source -> destination
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetOutWeight(2), 0, 1);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetOutWeight(3), 1, 0);
|
|
// check column destination -> source
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetInWeight(2), 0, 1);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetInWeight(3), 1, 0);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(four_levels_test)
|
|
{
|
|
// node: 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
|
|
std::vector<CellID> l1{{0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3}};
|
|
std::vector<CellID> l2{{0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1}};
|
|
std::vector<CellID> l3{{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}};
|
|
MultiLevelPartition mlp{{l1, l2, l3}, {4, 2, 1}};
|
|
|
|
BOOST_REQUIRE_EQUAL(mlp.GetNumberOfLevels(), 4);
|
|
|
|
std::vector<MockEdge> edges = {
|
|
{0, 1, 1}, // cell (0, 0, 0)
|
|
{0, 2, 1}, {3, 1, 1}, {3, 2, 1},
|
|
|
|
{4, 5, 1}, // cell (1, 0, 0)
|
|
{4, 6, 1}, {4, 7, 1}, {5, 4, 1}, {5, 6, 1}, {5, 7, 1}, {6, 4, 1},
|
|
{6, 5, 1}, {6, 7, 1}, {7, 4, 1}, {7, 5, 1}, {7, 6, 1},
|
|
|
|
{9, 11, 1}, // cell (2, 1, 0)
|
|
{10, 8, 1}, {11, 10, 1},
|
|
|
|
{13, 12, 10}, // cell (3, 1, 0)
|
|
{15, 14, 1},
|
|
|
|
{2, 4, 1}, // edge between cells (0, 0, 0) -> (1, 0, 0)
|
|
{5, 12, 1}, // edge between cells (1, 0, 0) -> (3, 1, 0)
|
|
{8, 3, 1}, // edge between cells (2, 1, 0) -> (0, 0, 0)
|
|
{9, 3, 1}, // edge between cells (2, 1, 0) -> (0, 0, 0)
|
|
{12, 5, 1}, // edge between cells (3, 1, 0) -> (1, 0, 0)
|
|
{13, 7, 1}, // edge between cells (3, 1, 0) -> (1, 0, 0)
|
|
{14, 9, 1}, // edge between cells (2, 1, 0) -> (0, 0, 0)
|
|
{14, 11, 1} // edge between cells (2, 1, 0) -> (0, 0, 0)
|
|
};
|
|
|
|
auto graph = makeGraph(edges);
|
|
|
|
CellStorage storage(mlp, graph);
|
|
|
|
auto cell_1_0 = storage.GetCell(1, 0);
|
|
auto cell_1_1 = storage.GetCell(1, 1);
|
|
auto cell_1_2 = storage.GetCell(1, 2);
|
|
auto cell_1_3 = storage.GetCell(1, 3);
|
|
auto cell_2_0 = storage.GetCell(2, 0);
|
|
auto cell_2_1 = storage.GetCell(2, 1);
|
|
auto cell_3_0 = storage.GetCell(3, 0);
|
|
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetSourceNodes(), 1);
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetDestinationNodes(), 1);
|
|
CHECK_EQUAL_RANGE(cell_1_0.GetSourceNodes(), 3);
|
|
CHECK_EQUAL_RANGE(cell_1_0.GetDestinationNodes(), 2);
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetOutWeight(3), 1);
|
|
REQUIRE_SIZE_RANGE(cell_1_0.GetInWeight(2), 1);
|
|
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetSourceNodes(), 3);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetDestinationNodes(), 3);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetSourceNodes(), 4, 5, 7);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetDestinationNodes(), 4, 5, 7);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetOutWeight(4), 3);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetOutWeight(5), 3);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetOutWeight(7), 3);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetInWeight(4), 3);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetInWeight(5), 3);
|
|
REQUIRE_SIZE_RANGE(cell_1_1.GetInWeight(7), 3);
|
|
|
|
REQUIRE_SIZE_RANGE(cell_1_2.GetSourceNodes(), 2);
|
|
REQUIRE_SIZE_RANGE(cell_1_2.GetDestinationNodes(), 2);
|
|
CHECK_EQUAL_RANGE(cell_1_2.GetSourceNodes(), 9, 11);
|
|
CHECK_EQUAL_RANGE(cell_1_2.GetDestinationNodes(), 8, 11);
|
|
REQUIRE_SIZE_RANGE(cell_1_2.GetOutWeight(9), 2);
|
|
REQUIRE_SIZE_RANGE(cell_1_2.GetOutWeight(11), 2);
|
|
REQUIRE_SIZE_RANGE(cell_1_2.GetInWeight(8), 2);
|
|
REQUIRE_SIZE_RANGE(cell_1_2.GetInWeight(11), 2);
|
|
|
|
REQUIRE_SIZE_RANGE(cell_1_3.GetSourceNodes(), 1);
|
|
REQUIRE_SIZE_RANGE(cell_1_3.GetDestinationNodes(), 2);
|
|
CHECK_EQUAL_RANGE(cell_1_3.GetSourceNodes(), 13);
|
|
CHECK_EQUAL_RANGE(cell_1_3.GetDestinationNodes(), 12, 14);
|
|
REQUIRE_SIZE_RANGE(cell_1_3.GetOutWeight(13), 2);
|
|
REQUIRE_SIZE_RANGE(cell_1_3.GetInWeight(12), 1);
|
|
REQUIRE_SIZE_RANGE(cell_1_3.GetInWeight(14), 1);
|
|
|
|
REQUIRE_SIZE_RANGE(cell_2_0.GetSourceNodes(), 3);
|
|
REQUIRE_SIZE_RANGE(cell_2_0.GetDestinationNodes(), 2);
|
|
CHECK_EQUAL_RANGE(cell_2_0.GetSourceNodes(), 3, 5, 7);
|
|
CHECK_EQUAL_RANGE(cell_2_0.GetDestinationNodes(), 5, 7);
|
|
REQUIRE_SIZE_RANGE(cell_2_0.GetOutWeight(3), 2);
|
|
REQUIRE_SIZE_RANGE(cell_2_0.GetOutWeight(5), 2);
|
|
REQUIRE_SIZE_RANGE(cell_2_0.GetOutWeight(7), 2);
|
|
REQUIRE_SIZE_RANGE(cell_2_0.GetInWeight(5), 3);
|
|
REQUIRE_SIZE_RANGE(cell_2_0.GetInWeight(7), 3);
|
|
|
|
REQUIRE_SIZE_RANGE(cell_2_1.GetSourceNodes(), 2);
|
|
REQUIRE_SIZE_RANGE(cell_2_1.GetDestinationNodes(), 3);
|
|
CHECK_EQUAL_RANGE(cell_2_1.GetSourceNodes(), 9, 13);
|
|
CHECK_EQUAL_RANGE(cell_2_1.GetDestinationNodes(), 8, 9, 12);
|
|
REQUIRE_SIZE_RANGE(cell_2_1.GetOutWeight(9), 3);
|
|
REQUIRE_SIZE_RANGE(cell_2_1.GetOutWeight(13), 3);
|
|
REQUIRE_SIZE_RANGE(cell_2_1.GetInWeight(8), 2);
|
|
REQUIRE_SIZE_RANGE(cell_2_1.GetInWeight(9), 2);
|
|
REQUIRE_SIZE_RANGE(cell_2_1.GetInWeight(12), 2);
|
|
|
|
REQUIRE_SIZE_RANGE(cell_3_0.GetSourceNodes(), 0);
|
|
REQUIRE_SIZE_RANGE(cell_3_0.GetDestinationNodes(), 0);
|
|
|
|
CellCustomizer customizer(mlp);
|
|
|
|
customizer.Customize(graph, storage, 1, 0);
|
|
customizer.Customize(graph, storage, 1, 1);
|
|
customizer.Customize(graph, storage, 1, 2);
|
|
customizer.Customize(graph, storage, 1, 3);
|
|
|
|
customizer.Customize(graph, storage, 2, 0);
|
|
customizer.Customize(graph, storage, 2, 1);
|
|
|
|
// level 1
|
|
// cell 0
|
|
CHECK_EQUAL_RANGE(cell_1_0.GetOutWeight(3), 1);
|
|
CHECK_EQUAL_RANGE(cell_1_0.GetInWeight(2), 1);
|
|
|
|
// cell 1
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetOutWeight(4), 0, 1, 1);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetOutWeight(5), 1, 0, 1);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetOutWeight(7), 1, 1, 0);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetInWeight(4), 0, 1, 1);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetInWeight(5), 1, 0, 1);
|
|
CHECK_EQUAL_RANGE(cell_1_1.GetInWeight(7), 1, 1, 0);
|
|
|
|
// cell 2
|
|
CHECK_EQUAL_RANGE(cell_1_2.GetOutWeight(9), 3, 1);
|
|
CHECK_EQUAL_RANGE(cell_1_2.GetOutWeight(11), 2, 0);
|
|
CHECK_EQUAL_RANGE(cell_1_2.GetInWeight(8), 3, 2);
|
|
CHECK_EQUAL_RANGE(cell_1_2.GetInWeight(11), 1, 0);
|
|
|
|
// cell 3
|
|
CHECK_EQUAL_RANGE(cell_1_3.GetOutWeight(13), 10, INVALID_EDGE_WEIGHT);
|
|
CHECK_EQUAL_RANGE(cell_1_3.GetInWeight(12), 10);
|
|
CHECK_EQUAL_RANGE(cell_1_3.GetInWeight(14), INVALID_EDGE_WEIGHT);
|
|
|
|
// level 2
|
|
// cell 0
|
|
CHECK_EQUAL_RANGE(cell_2_0.GetOutWeight(3), 3, 3);
|
|
CHECK_EQUAL_RANGE(cell_2_0.GetOutWeight(5), 0, 1);
|
|
CHECK_EQUAL_RANGE(cell_2_0.GetOutWeight(7), 1, 0);
|
|
CHECK_EQUAL_RANGE(cell_2_0.GetInWeight(5), 3, 0, 1);
|
|
CHECK_EQUAL_RANGE(cell_2_0.GetInWeight(7), 3, 1, 0);
|
|
|
|
// cell 1
|
|
CHECK_EQUAL_RANGE(cell_2_1.GetOutWeight(9), 3, 0, INVALID_EDGE_WEIGHT);
|
|
CHECK_EQUAL_RANGE(cell_2_1.GetOutWeight(13), INVALID_EDGE_WEIGHT, INVALID_EDGE_WEIGHT, 10);
|
|
CHECK_EQUAL_RANGE(cell_2_1.GetInWeight(8), 3, INVALID_EDGE_WEIGHT);
|
|
CHECK_EQUAL_RANGE(cell_2_1.GetInWeight(9), 0, INVALID_EDGE_WEIGHT);
|
|
CHECK_EQUAL_RANGE(cell_2_1.GetInWeight(12), INVALID_EDGE_WEIGHT, 10);
|
|
|
|
CellStorage storage_rec(mlp, graph);
|
|
customizer.Customize(graph, storage_rec);
|
|
|
|
CHECK_EQUAL_COLLECTIONS(cell_2_1.GetOutWeight(9), storage_rec.GetCell(2, 1).GetOutWeight(9));
|
|
CHECK_EQUAL_COLLECTIONS(cell_2_1.GetOutWeight(13), storage_rec.GetCell(2, 1).GetOutWeight(13));
|
|
CHECK_EQUAL_COLLECTIONS(cell_2_1.GetInWeight(8), storage_rec.GetCell(2, 1).GetInWeight(8));
|
|
CHECK_EQUAL_COLLECTIONS(cell_2_1.GetInWeight(9), storage_rec.GetCell(2, 1).GetInWeight(9));
|
|
CHECK_EQUAL_COLLECTIONS(cell_2_1.GetInWeight(12), storage_rec.GetCell(2, 1).GetInWeight(12));
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|