#!/bin/bash
DATASET_DIR=${1:-"./datasets/maestro"}  # The first argument is dataset directory.
WORKSPACE=${2:-"./workspaces/bytesep"}  # The second argument is workspace directory.

echo "DATASET_DIR=${DATASET_DIR}"
echo "WORKSPACE=${WORKSPACE}"

# Users can change the following settings.
SAMPLE_RATE=44100
CHANNELS=2

# Paths
HDF5S_DIR="${WORKSPACE}/hdf5s/maestro/sr=${SAMPLE_RATE}_chn=${CHANNELS}/train"

python3 bytesep/dataset_creation/pack_audios_to_hdf5s/maestro.py \
    --dataset_dir=$DATASET_DIR \
    --split="train" \
    --hdf5s_dir=$HDF5S_DIR \
    --sample_rate=$SAMPLE_RATE \
    --channels=$CHANNELS