#!/bin/bash

# Usage: set_topology <TOPO>
# Fills the NODES array based on TOPO value

set_topology() {
  local topo="$1"

    # Define node lists based on topology
    case "$TOPO" in
    2x2)
        NODES=( 0 1 
                5 6 )
        ;;
    3x3)
        NODES=(	0  1  2 
                5  6  7
                10 11 12)
        ;;
    4x4)
        NODES=(	0  1  2  3 
                5  6  7  8 
                10 11 12 13 
                15 16 17 18)
        ;;
    5x5)
        NODES=( 0  1  2  3  4 
                5  6  7  8  9 
                10 11 12 13 14 
                15 16 17 18 19 
                20 21 22 23 24)
        ;;
    8hop)
        NODES=( 0 1 2 3 4 
                        9
                        14
                        19
                        24)
        ;;
    ring)
        NODES=( 0  1  2  3  4 
                5 		    9
                10		    14
                15		    19
                20 21 22 23 24)
        ;;
    *)
        echo "Unknown topology: $TOPO"
        exit 1
        ;;
    esac

  return 0
}