#!/bin/bash

INPUT_ARGS=$@

if [[ $@ =~ '--python2' ]]; then
    INPUT_ARGS=${INPUT_ARGS//--python2/}
    source /usr/share/trex-core/scripts/find_python.sh --python2
fi

if [[ $@ =~ '--python3' ]]; then
    INPUT_ARGS=${INPUT_ARGS//--python3/}
    source /usr/share/trex-core/scripts/find_python.sh --python3
fi

if [ -z "$PYTHON" ]; then
    source /usr/share/trex-core/scripts/find_python.sh
fi

export PYTHONPATH=/usr/share/trex-core/scripts/automation/trex_control_plane/interactive

if [[ $@ =~ '--stl' ]]; then
    INPUT_ARGS=${INPUT_ARGS//--stl/}
    $PYTHON -m trex.examples.stl.stl_ndr_bench_tool  $INPUT_ARGS
elif [[ $@ =~ '--astf' ]]; then
    INPUT_ARGS=${INPUT_ARGS//--astf/}
    $PYTHON -m trex.examples.astf.astf_ndr_bench_tool  $INPUT_ARGS
else
    echo "Please specify mode with --stl or --astf."
fi