Grid3D<uint32_t> computeDistanceMap26_WithQueue(const GridType& grid, Predicate isInObject, bool outsideIsObject) { auto w = grid.width(), h = grid.height(), d = grid.depth(); uint32_t maxDist = std::numeric_limits<uint32_t>::max(); Grid3D<uint32_t> distanceGrid(w, h, d, maxDist); std::queue<Vec4i> queue; for(auto z = 0u; z < d; ++z) { for(auto y = 0u; y < h; ++y) { for(auto x = 0u; x < w; ++x) { auto voxel = Vec3i(x, y, z); if(!isInObject(x, y, z, grid)) { queue.push(Vec4i(voxel, 0u)); } else if(!outsideIsObject && (x == 0 || y == 0 || z == 0 || x == w - 1 || y == h - 1 || z == d - 1)) { queue.push(Vec4i(voxel, 1u)); } } } } while(!queue.empty()) { auto voxel = Vec3i(queue.front()); auto dist = queue.front().w; queue.pop(); if(dist < distanceGrid(voxel)) { distanceGrid(voxel) = dist; foreach26Neighbour(grid.resolution(), voxel, [&](const Vec3i& neighbour) { auto newDist = dist + 1; if(newDist < distanceGrid(neighbour)) { queue.push(Vec4i(neighbour, newDist)); } }); } } return distanceGrid; }
Grid3D<uint32_t> computeDistanceGrid6Connexity_outsideIsInObject(const GridType& grid, Predicate isInObject) { auto w = grid.width(), h = grid.height(), d = grid.depth(); auto maxDist = w + h + d; Grid3D<uint32_t> distanceGrid(w, h, d, maxDist); using DistType = decltype(maxDist); auto processVoxel = [&distanceGrid, &grid, &isInObject, maxDist](DistType& dist, int x, int y, int z) { if(!isInObject(x, y, z, grid)) { dist = 0u; } else if(dist != maxDist) { ++dist; } if(dist < distanceGrid(x, y, z)) { distanceGrid(x, y, z) = dist; } else { dist = distanceGrid(x, y, z); } }; for(auto y = 0u; y < h; ++y) { for(auto x = 0u; x < w; ++x) { auto dist = maxDist; for(auto z = 0; z < d; ++z) { processVoxel(dist, x, y, z); } dist = maxDist; for(int z = d - 1; z >= 0; --z) { processVoxel(dist, x, y, z); } } } for(auto z = 0u; z < d; ++z) { for(auto x = 0u; x < w; ++x) { auto dist = maxDist; for(auto y = 0; y < h; ++y) { processVoxel(dist, x, y, z); } dist = maxDist; for(int y = h - 1; y >= 0; --y) { processVoxel(dist, x, y, z); } } } for(auto z = 0u; z < d; ++z) { for(auto y = 0u; y < h; ++y) { auto dist = maxDist; for(auto x = 0; x < w; ++x) { processVoxel(dist, x, y, z); } dist = maxDist; for(int x = w - 1; x >= 0; --x) { processVoxel(dist, x, y, z); } } } return distanceGrid; }
Grid3D<uint32_t> computeDistanceMap26(const GridType& grid, Predicate isInObject, bool outsideIsObject) { int w = grid.width(), h = grid.height(), d = grid.depth(); uint32_t maxDist = std::numeric_limits<uint32_t>::max(); Grid3D<uint32_t> distanceGrid(w, h, d, maxDist); static const auto MASK_SIZE = 13; auto processVoxel = [&distanceGrid, &grid, &isInObject, maxDist, outsideIsObject, w, h, d](const Vec3i* mask, int x, int y, int z) { auto voxel = Vec3i(x, y, z); if(!isInObject(x, y, z, grid)) { distanceGrid(voxel) = 0u; } else if(!outsideIsObject && (x == 0 || y == 0 || z == 0 || x == w - 1 || y == h - 1 || z == d - 1)) { distanceGrid(voxel) = 1u; } auto currentDist = distanceGrid(voxel); if(currentDist != maxDist) { auto dist = currentDist + 1; for(auto i = 0u; i < MASK_SIZE; ++i) { auto offset = mask[i]; auto neighbour = voxel + offset; if(grid.contains(neighbour) && isInObject(neighbour.x, neighbour.y, neighbour.z, grid) && dist < distanceGrid(neighbour)) { distanceGrid(neighbour) = dist; } } } }; static const Vec3i masks[][MASK_SIZE] = { { Vec3i(1, 0, 0), Vec3i(1, 1, 0), Vec3i(0, 1, 0), Vec3i(-1, 1, 0), Vec3i(0, 0, 1), Vec3i(1, 0, 1), Vec3i(1, 1, 1), Vec3i(0, 1, 1), Vec3i(-1, 1, 1), Vec3i(-1, 0, 1), Vec3i(-1, -1, 1), Vec3i(0, -1, 1), Vec3i(1, -1, 1), }, { -Vec3i(1, 0, 0), -Vec3i(1, 1, 0), -Vec3i(0, 1, 0), -Vec3i(-1, 1, 0), -Vec3i(0, 0, 1), -Vec3i(1, 0, 1), -Vec3i(1, 1, 1), -Vec3i(0, 1, 1), -Vec3i(-1, 1, 1), -Vec3i(-1, 0, 1), -Vec3i(-1, -1, 1), -Vec3i(0, -1, 1), -Vec3i(1, -1, 1), } }; for(auto z = 0; z < d; ++z) { for(auto y = 0; y < h; ++y) { for(auto x = 0; x < w; ++x) { processVoxel(masks[0], x, y, z); } } } for(auto z = d - 1; z >= 0; --z) { for(auto y = h - 1; y >= 0; --y) { for(auto x = w - 1; x >= 0; --x) { processVoxel(masks[1], x, y, z); } } } return distanceGrid; }