parent
d515c33afc
commit
e05335b153
@ -0,0 +1,92 @@ |
||||
#include "pathing.hpp" |
||||
#include <vector> |
||||
|
||||
using std::vector; |
||||
|
||||
inline void add_neighbors(PointList &neighbors, Matrix &closed, size_t y, size_t x) { |
||||
size_t h = closed.size(); |
||||
size_t w = closed[0].size(); |
||||
vector<size_t> rows{y - 1, y, y + 1}; |
||||
vector<size_t> cols{x - 1, x, x + 1}; |
||||
|
||||
for(size_t row : rows) { |
||||
for(size_t col : cols) { |
||||
if((0 <= row && row < h) && |
||||
(0 <= col && col < w) && |
||||
closed[row][col] == 0) |
||||
{ |
||||
closed[row][col] = 1; |
||||
neighbors.push_back({.x=col, .y=row}); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
inline void matrix_assign(Matrix &out, int new_value) { |
||||
for(auto &row : out) { |
||||
row.assign(row.size(), new_value); |
||||
} |
||||
} |
||||
|
||||
void Pathing::compute_paths(Matrix &walls) { |
||||
INVARIANT(); |
||||
// Initialize the new array with every pixel at limit distance
|
||||
// NOTE: this is normally ones() * limit
|
||||
int limit = $limit == 0 ? $height * $width : $limit; |
||||
matrix_assign($paths, limit); |
||||
|
||||
Matrix closed = walls; |
||||
PointList starting_pixels; |
||||
PointList open_pixels; |
||||
|
||||
// First pass: Add starting pixels and put them in closed
|
||||
for(size_t counter = 0; counter < $height * $width; counter++) { |
||||
size_t x = counter % $width; // BUG: is this right?
|
||||
size_t y = counter / $width; |
||||
if($input[y][x] == 0) { |
||||
$paths[y][x] = 0; |
||||
closed[y][x] = 1; |
||||
starting_pixels.push_back({.x=x,.y=y}); |
||||
} |
||||
} |
||||
|
||||
// Second pass: Add border to open
|
||||
for(auto sp : starting_pixels) { |
||||
add_neighbors(open_pixels, closed, sp.y, sp.x); |
||||
} |
||||
|
||||
// Third pass: Iterate filling in the open list
|
||||
int counter = 1; // leave this here so it's available below
|
||||
for(; counter < limit && !open_pixels.empty(); ++counter) { |
||||
PointList next_open; |
||||
for(auto sp : open_pixels) { |
||||
$paths[sp.y][sp.x] = counter; |
||||
add_neighbors(next_open, closed, sp.y, sp.x); |
||||
} |
||||
open_pixels = next_open; |
||||
} |
||||
|
||||
// Last pass: flood last pixels
|
||||
for(auto sp : open_pixels) { |
||||
$paths[sp.y][sp.x] = counter; |
||||
} |
||||
} |
||||
|
||||
void Pathing::set_target(const Point &at, int value) { |
||||
$input[at.y][at.x] = 0; |
||||
} |
||||
|
||||
void Pathing::clear_target(const Point &at) { |
||||
$input[at.y][at.x] = 1; |
||||
} |
||||
|
||||
bool Pathing::INVARIANT() { |
||||
using dbc::check; |
||||
|
||||
check($paths.size() == $height, "paths wrong height"); |
||||
check($paths[0].size() == $width, "paths wrong width"); |
||||
check($input.size() == $height, "input wrong height"); |
||||
check($input[0].size() == $width, "input wrong width"); |
||||
|
||||
return true; |
||||
} |
@ -0,0 +1,25 @@ |
||||
#pragma once |
||||
#include "map.hpp" |
||||
#include "point.hpp" |
||||
|
||||
class Pathing { |
||||
public: |
||||
int $limit; |
||||
size_t $width; |
||||
size_t $height; |
||||
Matrix $paths; |
||||
Matrix $input; |
||||
|
||||
Pathing(size_t width, size_t height, int limit) : |
||||
$limit(limit), |
||||
$width(width), |
||||
$height(height), |
||||
$paths(height, MatrixRow(width, 1)), |
||||
$input(height, MatrixRow(width, 1)) |
||||
{} |
||||
|
||||
void compute_paths(Matrix &walls); |
||||
void set_target(const Point &at, int value=0); |
||||
void clear_target(const Point &at); |
||||
bool INVARIANT(); |
||||
}; |
@ -0,0 +1,34 @@ |
||||
#include <catch2/catch_test_macros.hpp> |
||||
#include <fmt/core.h> |
||||
#include <nlohmann/json.hpp> |
||||
#include <fstream> |
||||
#include "pathing.hpp" |
||||
|
||||
using namespace fmt; |
||||
using namespace nlohmann; |
||||
using std::string; |
||||
|
||||
json load_test_pathing(const string &fname) { |
||||
std::ifstream infile(fname); |
||||
return json::parse(infile); |
||||
} |
||||
|
||||
TEST_CASE("dijkstra algo test", "[pathing]") { |
||||
json data = load_test_pathing("./tests/dijkstra.json"); |
||||
|
||||
for(auto &test : data) { |
||||
Matrix expected = test["expected"]; |
||||
Matrix walls = test["walls"]; |
||||
int limit = test["limit"]; |
||||
|
||||
Pathing pathing(walls[0].size(), walls.size(), limit); |
||||
|
||||
pathing.$input = test["input"]; |
||||
|
||||
REQUIRE(pathing.INVARIANT()); |
||||
pathing.compute_paths(walls); |
||||
|
||||
REQUIRE(pathing.INVARIANT()); |
||||
REQUIRE(pathing.$paths == expected); |
||||
} |
||||
} |
Loading…
Reference in new issue