build_grid_3d_neighbors()

std::unordered_map< size_t, std::vector< size_t > > MayaFlux::Nodes::NodeNetwork::build_grid_3d_neighbors ( size_t  width, size_t  height, size_t  depth  )

staticprotected

Build neighbor map for GRID_3D topology.

Parameters

width	Grid width
height	Grid height
depth	Grid depth

Returns

Map of node index to neighbor indices (6-connectivity)

Definition at line 709 of file ParticleNetwork.cpp.

{
    std::unordered_map<size_t, std::vector<size_t>> neighbors;
 
    for (size_t z = 0; z < depth; ++z) {
        for (size_t y = 0; y < height; ++y) {
            for (size_t x = 0; x < width; ++x) {
                size_t idx = z * (width * height) + y * width + x;
                std::vector<size_t> node_neighbors;
 
                if (x > 0) {
                    node_neighbors.push_back(idx - 1);
                }
                if (x < width - 1) {
                    node_neighbors.push_back(idx + 1);
                }
 
                if (y > 0) {
                    node_neighbors.push_back(idx - width);
                }
                if (y < height - 1) {
                    node_neighbors.push_back(idx + width);
                }
 
                if (z > 0) {
                    node_neighbors.push_back(idx - (width * height));
                }
                if (z < depth - 1) {
                    node_neighbors.push_back(idx + (width * height));
                }
 
                if (!node_neighbors.empty()) {
                    neighbors[idx] = std::move(node_neighbors);
                }
            }
        }
    }
 
    return neighbors;
}