#!/bin/sh
#===============================================================================
#
#  Copyright (C) 2024 by Digi International Inc.
#  All rights reserved.
#
#  This program is free software; you can redistribute it and/or modify it
#  under the terms of the GNU General Public License version 2 as published by
#  the Free Software Foundation.
#
#
#  !Description: Launch script for ST NPU demos
#
#===============================================================================

# Default demo to launch.
DEFAULT_DEMO="pose_estimation"

# Prepare MIPI camera.
media-ctl -d /dev/media0 --set-v4l2 "'ov5640 0-003c':0[fmt:SBGGR8_1X8/1280x720]"
media-ctl -d /dev/media0 --set-v4l2 "'48020000.csi':1[fmt:SBGGR8_1X8/1280x720]"
media-ctl -d /dev/media0 --set-v4l2 "'dcmipp_input':2[fmt:SBGGR8_1X8/1280x720]"
media-ctl -d /dev/media0 --set-v4l2 "'dcmipp_main_isp':1[fmt:RGB888_1X24/1280x720 field:none]"
media-ctl -d /dev/media0 --set-v4l2 "'dcmipp_main_postproc':0[compose:(0,0)/640x480]"
media-ctl -d /dev/media0 --set-v4l2 "'dcmipp_main_postproc':1[fmt:RGB565_2X8_LE/640x480]"

# Mirror the image.
v4l2-ctl -d /dev/v4l-subdev6 --set-ctrl "horizontal_flip=1"
#v4l2-ctl -d /dev/v4l-subdev6 --set-ctrl "vertical_flip=1"

# Configure Wayland/Weston settings.
export "DISPLAY=:0.0"
export "XDG_RUNTIME_DIR=/run/user/0"
export "WAYLAND_DISPLAY=wayland-1"

# Wait until Wayland framework is ready.
wait_for_wayland() {
    local count=20
    local wayland_socket="/run/user/0/${WAYLAND_DISPLAY}"

    while [ ! -S "${wayland_socket}" ]; do
        sleep 1
        count=$((count-1))
        if [ "${count}" = 0 ]; then
            return 1
        fi
    done
    return 0
}
[ -d "/usr/share/wayland" ] && wait_for_wayland

# Try to extract the demo to launch from the script name.
DEMO=${DEFAULT_DEMO}
DEMO_FROM_NAME="$(basename "${0}" | sed -n 's/^launch_npu_demo_\(.*\)\.sh$/\1/p')"

# Check if the demo to launch was passed as argument.
if [ -n "${1}" ]; then
    DEMO=${1}
elif [ -n "${DEMO_FROM_NAME}" ]; then
    DEMO=${DEMO_FROM_NAME}
fi

# Build the demo folder name.
DEMO_FOLDER_NAME="$(echo "${DEMO}" | sed 's/_/-/g')"

# Build the demo full path.
DEMO_DIR="/usr/local/x-linux-ai/${DEMO_FOLDER_NAME}"

# Verify that the demo directory exists.
[ -d "${DEMO_DIR}" ] || { echo "Error: Demo ${DEMO} does not exist"; exit 1; }

# Execute the demo.
"${DEMO_DIR}/launch_python_${DEMO}.sh"