#!/bin/sh
#
# Build and/or update the sample of studying central X-ray brightness
# excess, by only copying/updating the required products from our complete
# mass sample.
#
# Aaron LI
# Created: 2016-03-02
# Updated: 2016-04-26
#
# Changelog:
# 2016-04-26:
#   * Add flag "UPDATE_MODE"
#   * Choose the correct 'img' directory for ZZH's data
#   * Simplify the argument processing
# 2016-04-15:
#   * Add support for zzh's sample
#   * Convert "_" to "-" in NAME
#

analyze_path() {
    # extract `obs_id' and `source name' from path
    echo "$@" | awk '
    # main part
    {
        if (NF==1) {
            ## oi & name
            input=($1 "/")
            if (input ~ /_oi/) {
                ## PATTERN: .../$name_oi$oi/...
                idx_oi = match(input, /oi[0-9]+/) + 2;    # "2" skip the "oi"
                len_oi = RLENGTH - 2;
                oi = substr(input, idx_oi, len_oi);
                idx_name = match(input, /\/[a-zA-Z0-9.+-]+_oi/) + 1;
                len_name = RLENGTH - 4;
                name = substr(input, idx_name, len_name);
                owner = "lwt";
            }
            else {
                ## PATTERN: .../$name/$oi/...
                idx_oi = match(input, /\/[0-9]+\//) + 1;
                len_oi = RLENGTH - 2;
                oi = substr(input, idx_oi, len_oi);
                idx_name1 = match(input, /\/[a-zA-Z0-9_.+-]+\/[0-9]+\//);
                len_name1 = RLENGTH;
                name1 = substr(input, idx_name1, len_name1);
                idx_name = match(name1, /\/[a-zA-Z0-9_.+-]+\//) + 1;
                len_name = RLENGTH - 2;
                name = substr(name1, idx_name, len_name);
                owner = "zzh";
            }
            ## output
            printf("input: %s\n", input)
            printf("oi: %s\nname: %s\nowner: %s\n", oi, name, owner)
        }
        else {
            printf("*** WARNING: invalid input: %s\n", $0)
        }
    }
    # END { }
    '
}


case "$1" in
    -[hH]*)
        echo "Usage:"
        echo "    `basename $0` <dest_root_dir> <repro_dir1> ..."
        exit 1
        ;;
esac

UPDATE_MODE="YES"

DEST_ROOT="$1"
if [ -n "${UPDATE_MODE}" -a ! -d "${DEST_ROOT}" ]; then
    echo "UPDATE MODE: '${DEST_ROOT}' dose not exist"
    exit 2
fi
[ ! -d  "${DEST_ROOT}" ] && mkdir -v "${DEST_ROOT}"

INIT_DIR=`pwd -P`
while [ ! -z "$2" ]; do
    repro_dir="$2"
    shift
    OI=`analyze_path "${repro_dir}" | grep '^oi:' | awk '{ print $2 }'`
    NAME=`analyze_path "${repro_dir}" | grep '^name:' | awk '{ print $2 }'`
    OWNER=`analyze_path "${repro_dir}" | grep '^owner:' | awk '{ print $2 }'`
    # avoid "_" in name
    NAME=`echo "${NAME}" | tr '_' '-'`
    echo "********* ${NAME}_oi${OI} *********"
    # create directories
    DEST="${DEST_ROOT}/${NAME}_oi${OI}"
    if [ -n "${UPDATE_MODE}" -a ! -d "${DEST}/repro" ]; then
        # skip if dest repro directory does not exist
        echo "Skipped!"
        continue
    fi
    [ ! -d "${DEST}/repro" ] && mkdir -pv ${DEST}/repro
    cd "${DEST}/repro"
    # simply copy the whole sub-directories
    cp -av ${repro_dir}/acisf*.fits .
    cp -av ${repro_dir}/acisf*.lis ${repro_dir}/pcadf*.fits .
    cp -av ${repro_dir}/*_INFO.json .
    cp -av ${repro_dir}/evt .
    cp -av ${repro_dir}/bkg .
    if [ "${OWNER}" = "lwt" ]; then
        cp -av ${repro_dir}/img .
    elif [ "${OWNER}" = "zzh" ]; then
        img_dir="${repro_dir}/../evt2/img"
        if [ -f "${img_dir}/sbprofile.txt" ]; then
            cp -av ${img_dir} .
            cp -av ${repro_dir}/../evt2/spc/profile/rspec.reg ./img/
        else
            echo "WARNING: '${img_dir}/sbprofile.txt' does not exists"
            cp -av ${repro_dir}/img .
        fi
    fi
    # apply clean up
    find . \( -iname '*_bak' -o -iname '_tmp*' \) -delete
    find . -type l -iname '*.dat' -delete
    cd "${INIT_DIR}"
done