gen_test_data.sh

#!/usr/bin/bash

if [[ $# -eq 2 ]]; then # Generate custom-length command list to the specified output directory
    num_cmds=$1
    tests_dir=$2
    # NOTE: hacky and will break when other tests are created.
    mkdir -p \
        ${tests_dir}/binheap/round_robin \
        ${tests_dir}/binheap/strict \
        ${tests_dir}/round_robin \
        ${tests_dir}/strict 
    echo "Number of commands: ${num_cmds}; output directory: ${tests_dir}"
else
    num_cmds=20000
    tests_dir="$(dirname "$0")/../tests"
fi

queue_size=16

data_gen_dir="$(dirname "$0")"


# For SDN, we use piezo mode when making the data file and use pifotree_oracle to 
# generate the expected output.

python3 ${data_gen_dir}/gen_oracle_data.py $num_cmds --no-err $queue_size \
    > ${tests_dir}/sdn_test.data
[[ $? -eq 0 ]] && echo "Generated sdn_test.data"

cat ${tests_dir}/sdn_test.data \
    | python3 ${data_gen_dir}/pifo_tree_oracle.py $num_cmds $queue_size --keepgoing \
    > ${tests_dir}/sdn_test.expect
[[ $? -eq 0 ]] && echo "Generated sdn_test.expect"


# For the others, we drop piezo mode for data gen, and we use the appropriate
# oracle, which is one of the following:
# - fifo_oracle.py
# - pifo_oracle.py
# - pifo_tree_oracle.py

for queue_kind in fifo pifo_tree complex_tree; do
    python3 ${data_gen_dir}/gen_oracle_data.py $num_cmds \
        > ${tests_dir}/${queue_kind}_test.data
    [[ $? -eq 0 ]] && echo "Generated ${queue_kind}_test.data"

    cat ${tests_dir}/${queue_kind}_test.data \
        | python3 ${data_gen_dir}/${queue_kind}_oracle.py $num_cmds $queue_size --keepgoing \
        > ${tests_dir}/${queue_kind}_test.expect
    [[ $? -eq 0 ]] && echo "Generated ${queue_kind}_test.expect"
done


# For the PIEO and PCQ, we drop piezo mode and enable ranks and readiness times for 
# data gen and use nwc_oracle to generate the expected output.

python3 ${data_gen_dir}/gen_oracle_data.py $num_cmds --nwc-en \
    > ${tests_dir}/pieo_test.data
[[ $? -eq 0 ]] && echo "Generated pieo_test.data"

cat ${tests_dir}/pieo_test.data \
    | python3 ${data_gen_dir}/nwc_oracle.py $num_cmds $queue_size --keepgoing \
    > ${tests_dir}/pieo_test.expect
[[ $? -eq 0 ]] && echo "Generated pieo_test.expect"

cp ${tests_dir}/pieo_test.data ${tests_dir}/pcq_test.data
[[ $? -eq 0 ]] && echo "Generated pcq_test.data"

cp ${tests_dir}/pieo_test.expect ${tests_dir}/pcq_test.expect
[[ $? -eq 0 ]] && echo "Generated pcq_test.expect"


# For the Binary Heap, we drop piezo mode and enable ranks for data gen and
# use binheap_oracle to generate the expected output.

python3 ${data_gen_dir}/gen_oracle_data.py $num_cmds --use-rank \
    > ${tests_dir}/binheap/stable_binheap_test.data
[[ $? -eq 0 ]] && echo "Generated binheap/stable_binheap_test.data"

cat ${tests_dir}/binheap/stable_binheap_test.data \
    | python3 ${data_gen_dir}/binheap_oracle.py $num_cmds $queue_size --keepgoing \
    > ${tests_dir}/binheap/stable_binheap_test.expect
[[ $? -eq 0 ]] && echo "Generated binheap/stable_binheap_test.expect"


# For the Round Robin queues, we drop piezo mode as well and use rr_oracle to
# generate the expected output for queues with 2..7 flows. This generates 6 data 
# expect file pairs.

for n in {2..7}; do
    python3 ${data_gen_dir}/gen_oracle_data.py $num_cmds \
        > ${tests_dir}/round_robin/rr_${n}flow_test.data
    [[ $? -eq 0 ]] && echo "Generated round_robin/rr_${n}flow_test.data"

    cat ${tests_dir}/round_robin/rr_${n}flow_test.data \
        | python3 ${data_gen_dir}/rr_oracle.py $num_cmds $queue_size $n --keepgoing \
        > ${tests_dir}/round_robin/rr_${n}flow_test.expect
    [[ $? -eq 0 ]] && echo "Generated round_robin/rr_${n}flow_test.expect"
done


# For Strict queues, we use strict_oracle.py to generate the expected output
# for queues with 2..6 flows, each with a different strict ordering. This generates 5
# expect file pairs.

for n in {2..7}; do
    python3 ${data_gen_dir}/gen_oracle_data.py $num_cmds \
        > ${tests_dir}/strict/strict_${n}flow_test.data
    [[ $? -eq 0 ]] && echo "Generated strict/strict_${n}flow_test.data"

    cat ${tests_dir}/strict/strict_${n}flow_test.data \
        | python3 ${data_gen_dir}/strict_oracle.py $num_cmds $queue_size $n --keepgoing \
        > ${tests_dir}/strict/strict_${n}flow_test.expect
    [[ $? -eq 0 ]] && echo "Generated strict/strict_${n}flow_test.expect"
done


# Copying into binheap/

cp ${tests_dir}/fifo_test.data ${tests_dir}/binheap/
[[ $? -eq 0 ]] && echo "Generated binheap/fifo_test.data"

cp ${tests_dir}/fifo_test.expect ${tests_dir}/binheap/
[[ $? -eq 0 ]] && echo "Generated binheap/fifo_test.expect"

for sched_algo in round_robin strict; do 
    for i in ${tests_dir}/${sched_algo}/*.data; do
        name="$(basename $i .data)"

        cp ${tests_dir}/${sched_algo}/${name}.data \
            ${tests_dir}/binheap/${sched_algo}/${name}.data
        [[ $? -eq 0 ]] && echo "Generated binheap/${sched_algo}/${name}.data"

        cp ${tests_dir}/${sched_algo}/${name}.expect \
            ${tests_dir}/binheap/${sched_algo}/${name}.expect
        [[ $? -eq 0 ]] && echo "Generated binheap/${sched_algo}/${name}.expect"
    done
done