#!/usr/bin/env bash
# ------------------------------------------------------------
#  generate_lrmc_seeds.sh
#
#  Run generate_lrmc_seeds.py for num_nodes = 1, 101, 201, …,
#  2601, 2701 and then once more for 2708.
#
#  Usage: ./generate_lrmc_seeds.sh
# ------------------------------------------------------------

set -euo pipefail          # Exit on error, undefined var, or pipe failure

# ------------------------------------------------------------------
# Configuration – adjust these if your file layout changes
# ------------------------------------------------------------------
INPUT_EDGELIST="cora_seeds/edgelist.txt"
OUT_DIR="cora_seeds"
LEVELS=1
BASELINE="random"

# ------------------------------------------------------------------
# Main loop – 1 … 2708 stepping by 100
# ------------------------------------------------------------------
echo "Starting loop over num_nodes = 1, 101, 201, …, 2601, 2701 …"

for NUM_NODES in $(seq 1 100 2708); do
    echo ">>> num_nodes=$NUM_NODES"
    python3 generate_lrmc_seeds.py \
        --input_edgelist "$INPUT_EDGELIST" \
        --out_dir "$OUT_DIR" \
        --levels "$LEVELS" \
        --baseline "$BASELINE" \
        --num_nodes "$NUM_NODES"
done

# ------------------------------------------------------------------
# Explicitly run for 2708 (not hit by the 100‑step sequence)
# ------------------------------------------------------------------
echo ">>> num_nodes=2708 (explicitly added)"
python3 generate_lrmc_seeds.py \
    --input_edgelist "$INPUT_EDGELIST" \
    --out_dir "$OUT_DIR" \
    --levels "$LEVELS" \
    --baseline "$BASELINE" \
    --num_nodes 2708

echo "All done!"