stage_1.sh

#!/bin/bash

#stage1 (mriqc, fmriprep_fit, freesurfer, qsiprep, smriprep, magetbrain_init):

# Function to calculate and submit array jobs
submit_array_job() {
    local script=$1
    local sub_size=$2
    local n_subjects=$(( $( wc -l ./data/local/bids/participants.tsv | cut -f1 -d' ' ) - 1 ))
    local array_job_length=$(( n_subjects / sub_size ))
    echo "Submitting job for $script with array size: ${array_job_length}"
    sbatch --array=0-${array_job_length} $script
}

# Function to prompt user and run selected pipeline
run_pipeline() {
    local pipeline_name=$1
    local script_path=$2
    local sub_size=$3
    read -p "Do you want to run the $pipeline_name pipeline? (yes/no): " run_pipeline
    if [[ "$run_pipeline" =~ ^(yes|y)$ ]]; then
        echo "Running $pipeline_name..."
        submit_array_job $script_path $sub_size
    else
        echo "Skipping $pipeline_name."
    fi
}

# Prompt user for each pipeline
run_pipeline "mriqc" "./code/01_mriqc_scinet.sh" 1
run_pipeline "fmriprep_fit" "code/01_fmriprep_fit_scinet.sh" 1
run_pipeline "freesurfer" "code/01_freesurfer_long_scinet.sh" 1
run_pipeline "qsiprep" "./code/01_qsiprep_scinet.sh" 1
run_pipeline "smriprep" "./code/01_smriprep_scinet.sh" 1


# Prompt for magetbrain_init
read -p "Do you want to run the magetbrain_init pipeline? (yes/no): " run_magetbrain
if [[ "$run_magetbrain" =~ ^(yes|y)$ ]]; then
    echo "Running magetbrain_init..."
    sbatch ./code/01_magetbrain_init_scinet.sh
else
    echo "Skipping magetbrain_init."
fi