summaryrefslogtreecommitdiff
path: root/compiler/tf2circle-dredd-pbtxt-test/runner.sh
blob: 7c7b123c4a25302a45603e4ad2d5b393ea566bbf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
#!/bin/bash

# This script checks circle file generated by tf2circle

# exit if unknown var is used
set -u

WORKDIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)

# Need at least toolchain.config
if [[ $# -lt 1 ]]; then
  echo "USAGE: $0 ..."
  echo
  echo "ARGUMENTS:"
  echo "  [toolchain.config path]"
  echo "  [Prefix1]"
  echo "  [Prefix2]"
  echo "  ..."
  exit 255
fi

CONFIG_PATH="$1"; shift

source "${CONFIG_PATH}"

echo "-- Found circle-inspect: ${CIRCLE_INSPECT_PATH}"
echo "-- Found circle-verify: ${CIRCLE_VERIFY_PATH}"
echo "-- Found tf2circle: ${TF2CIRCLE_PATH}"
echo "-- Found workdir: ${WORKDIR}"

TESTED=()
PASSED=()
FAILED=()

pushd "${WORKDIR}"

# running each rule file

while [[ $# -ne 0 ]]; do
  PREFIX="$1"; shift

  echo "[ RUN      ] ${PREFIX}"

  TESTED+=("${PREFIX}")

  PASSED_TAG="${PREFIX}.passed"

  rm -f "${PASSED_TAG}"

  cat > "${PREFIX}.log" <(
    exec 2>&1

    source "${PREFIX}.test"

    echo "-- Use '${MODEL_PB_PATH}', '${MODEL_INFO_PATH}', and '${MODEL_RULE_PATH}'"

    # Exit immediately if any command fails
    set -e
    # Show commands
    set -x

    # Generate circle
    "${TF2CIRCLE_PATH}" \
      "${MODEL_INFO_PATH}" \
      "${MODEL_PB_PATH}" \
      "${WORKDIR}/${PREFIX}.circle"

    #
    # Run rule prepared to check circle file
    #

    # set vars needed by rule file
    CIRCLE_PATH="${WORKDIR}/${PREFIX}.circle"

    # Note: turn off 'command printing'. Otherwise printing will be so messy
    set +x

    # set vars required by rule-lib.sh and rule file
    COMPILED_FILE=${CIRCLE_PATH}
    INSPECT_PROG_PATH=${CIRCLE_INSPECT_PATH}
    VERIFY_PROG_PATH=${CIRCLE_VERIFY_PATH}
    ERROR_LOG="${PREFIX}.error"

    rm -f "${ERROR_LOG}"

    # in case error while running ${MODEL_RULE_PATH}, prints error msg
    trap 'echo "** ERROR **" ; cat "${ERROR_LOG}"' ERR

    source rule-lib.sh
    source "${MODEL_RULE_PATH}"

    # unset
    trap - ERR
    set -x

    # At this point, the exit code of all commands is 0
    # If not 0, execution of this script ends because of "set -e"
    touch "${PASSED_TAG}"
  )

  if [[ -f "${PASSED_TAG}" ]]; then
    echo "[       OK ] ${PREFIX}"
    PASSED+=("$PREFIX")
  else
    echo "[      FAIL] ${PREFIX}"
    FAILED+=("$PREFIX")
  fi
done
popd

if [[ ${#TESTED[@]} -ne ${#PASSED[@]} ]]; then
  echo "FAILED"
  for TEST in "${FAILED[@]}"
  do
    echo "- ${TEST}"
  done
  exit 255
fi

echo "PASSED"
exit 0