#!/usr/bin/env bash
set -euo pipefail

# Copy reference arrays used for validation into solution/arrays/
# Source path can be overridden via REF_SOLN_DIR environment variable

SRC_DIR=${REF_SOLN_DIR:-/global/cfs/projectdirs/atlas/dwkim/llm4hep/solution/arrays}
DST_DIR=$(dirname "$0")/../solution/arrays

echo "Source: $SRC_DIR"
echo "Dest:   $DST_DIR"

if [[ ! -d "$SRC_DIR" ]]; then
  echo "ERROR: Source directory not found: $SRC_DIR" >&2
  exit 1
fi

mkdir -p "$DST_DIR"

shopt -s nullglob
files=(
  bkgd.npy bkgd_scores.npy boundaries.npy
  data_raw.npy data_A_raw.npy file_list.txt root_summary.txt
  signal.npy signal_scores.npy signal_raw.npy signal_WH_raw.npy
  significances.npy
)

copied=0
for f in "${files[@]}"; do
  if [[ -e "$SRC_DIR/$f" ]]; then
    echo "Copying $f"
    cp -f "$SRC_DIR/$f" "$DST_DIR/"
    copied=$((copied+1))
  else
    echo "Missing in source: $f (skipping)"
  fi
done

echo "Copied $copied files to $DST_DIR"
echo "Done."