Martino Ferrari
286 lines
11 KiB
Bash
Executable File
286 lines
11 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# run_streamhub.sh — Launch a MARTe2 app with UDPStreamer + StreamHub
|
|
#
|
|
# Usage:
|
|
# ./run_streamhub.sh [OPTIONS]
|
|
#
|
|
# Options:
|
|
# -m <MARTe2_DIR> Override MARTe2 installation dir (default: $MARTe2_DIR)
|
|
# -c <MARTe2_Components_DIR> Override MARTe2-components dir (default: $MARTe2_Components_DIR)
|
|
# -b <BUILD_TARGET> Build target (default: x86-linux)
|
|
# -p <WS_PORT> StreamHub WebSocket port (default: 8090)
|
|
# -n <MAX_POINTS> StreamHub ring-buffer size per signal (default: 10000)
|
|
# -s Skip building — run with whatever is already built
|
|
# -g Launch the ImGui desktop client after start
|
|
# -w Build and launch the web UI server (Client/webui)
|
|
# -h Show this help
|
|
#
|
|
# Ports used:
|
|
# 44500/udp UDPStreamer scalar signals (unicast control)
|
|
# 44503/udp UDPStreamer scalar signals (multicast data, group 239.0.0.1)
|
|
# 44501/udp UDPStreamer array signals (FirstSample / LastSample)
|
|
# 44502/udp UDPStreamer array signals (FullArray)
|
|
# 8080/tcp DebugService control
|
|
# 8081/udp DebugService stream
|
|
# 9090/tcp TCPLogger
|
|
# 8090/tcp StreamHub WebSocket (default, override with -p)
|
|
#
|
|
# Environment:
|
|
# MARTe2_DIR must be set (or passed via -m)
|
|
# MARTe2_Components_DIR must be set (or passed via -c)
|
|
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
MARTe_CFG="${SCRIPT_DIR}/Test/Configurations/streamhub_demo.cfg"
|
|
BUILD_TARGET="${TARGET:-x86-linux}"
|
|
WS_PORT=8090
|
|
MAX_POINTS=1000000
|
|
SKIP_BUILD=0
|
|
START_GUI=0
|
|
START_WEBUI=0
|
|
WEBUI_PORT=8080
|
|
|
|
# ── Parse arguments ───────────────────────────────────────────────────────────
|
|
while getopts "m:c:b:p:n:sgwh" opt; do
|
|
case "$opt" in
|
|
m) MARTe2_DIR="$OPTARG" ;;
|
|
c) MARTe2_Components_DIR="$OPTARG" ;;
|
|
b) BUILD_TARGET="$OPTARG" ;;
|
|
p) WS_PORT="$OPTARG" ;;
|
|
n) MAX_POINTS="$OPTARG" ;;
|
|
s) SKIP_BUILD=1 ;;
|
|
g) START_GUI=1 ;;
|
|
w) START_WEBUI=1 ;;
|
|
h)
|
|
sed -n '2,30p' "$0" | grep '^#' | sed 's/^# \?//'
|
|
exit 0
|
|
;;
|
|
*) echo "Unknown option: -$OPTARG" >&2; exit 1 ;;
|
|
esac
|
|
done
|
|
|
|
# ── Validate environment ──────────────────────────────────────────────────────
|
|
if [[ -z "${MARTe2_DIR:-}" ]]; then
|
|
echo "ERROR: MARTe2_DIR is not set. Source env.sh first or pass -m <dir>."
|
|
echo " source ${SCRIPT_DIR}/env.sh"
|
|
exit 1
|
|
fi
|
|
if [[ -z "${MARTe2_Components_DIR:-}" ]]; then
|
|
echo "ERROR: MARTe2_Components_DIR is not set. Source env.sh first or pass -c <dir>."
|
|
exit 1
|
|
fi
|
|
|
|
BUILD_DIR="${SCRIPT_DIR}/Build/${BUILD_TARGET}"
|
|
STREAMHUB_EX="${BUILD_DIR}/StreamHub/StreamHub.ex"
|
|
IMGUI_CLIENT="${SCRIPT_DIR}/Client/streamhub/build/StreamHubClient"
|
|
WEBUI_DIR="${SCRIPT_DIR}/Client/webui"
|
|
WEBUI_BIN="${WEBUI_DIR}/streamhub-webui"
|
|
|
|
MARTE2_BIN="${MARTe2_DIR}/Build/${BUILD_TARGET}/App/MARTeApp.ex"
|
|
if [[ ! -x "$MARTE2_BIN" ]]; then
|
|
MARTE2_BIN="${MARTe2_DIR}/Build/${BUILD_TARGET}/App/MARTe2.sh"
|
|
fi
|
|
if [[ ! -x "$MARTE2_BIN" ]]; then
|
|
echo "ERROR: MARTe2 executable not found at ${MARTe2_DIR}/Build/${BUILD_TARGET}/App/"
|
|
exit 1
|
|
fi
|
|
|
|
# ── Build ─────────────────────────────────────────────────────────────────────
|
|
if [[ "$SKIP_BUILD" -eq 0 ]]; then
|
|
echo "==> Building MARTe2 components (TARGET=${BUILD_TARGET})..."
|
|
make -C "${SCRIPT_DIR}" -f Makefile.gcc TARGET="${BUILD_TARGET}" 2>&1 | tail -10
|
|
|
|
echo "==> Building StreamHub (TARGET=${BUILD_TARGET})..."
|
|
make -C "${SCRIPT_DIR}/Source/Applications/StreamHub" \
|
|
-f Makefile.gcc TARGET="${BUILD_TARGET}" \
|
|
MARTe2_DIR="${MARTe2_DIR}" 2>&1 | tail -10
|
|
|
|
if [[ "$START_GUI" -eq 1 ]]; then
|
|
if [[ -d "${SCRIPT_DIR}/Client/streamhub/build" ]]; then
|
|
echo "==> Building ImGui client (a full rebuild can take ~2 min;"
|
|
echo " ImPlot's implot_items.cpp is one slow -O3 translation unit)..."
|
|
cmake --build "${SCRIPT_DIR}/Client/streamhub/build" -j"$(nproc)"
|
|
fi
|
|
fi
|
|
|
|
if [[ "$START_WEBUI" -eq 1 ]]; then
|
|
echo "==> Building web UI server..."
|
|
(cd "${WEBUI_DIR}" && go build -o streamhub-webui .)
|
|
fi
|
|
|
|
echo "==> Build done."
|
|
fi
|
|
|
|
# ── Sanity-check binaries ─────────────────────────────────────────────────────
|
|
if [[ ! -x "$STREAMHUB_EX" ]]; then
|
|
echo "ERROR: StreamHub binary not found: ${STREAMHUB_EX}"
|
|
echo " Build it with: make -C Source/Applications/StreamHub -f Makefile.gcc"
|
|
exit 1
|
|
fi
|
|
|
|
# ── Write StreamHub config ────────────────────────────────────────────────────
|
|
HUB_CFG="$(mktemp /tmp/streamhub_XXXXXX.cfg)"
|
|
|
|
cat > "$HUB_CFG" <<EOF
|
|
/**
|
|
* StreamHub configuration — auto-generated by run_streamhub.sh
|
|
*
|
|
* Three sources matching the streamhub_demo MARTe2 configuration:
|
|
* scalar : 1 kHz scalar sines @ 1 ksps (Sine1, Sine2)
|
|
* med : 1 kHz arrays @ 1 Msps (Ch1, Ch2)
|
|
* fast : 5 kHz arrays @ 5 Msps (Ch3, Ch4)
|
|
*/
|
|
Hub = {
|
|
WSPort = ${WS_PORT}
|
|
MaxPoints = ${MAX_POINTS}
|
|
PushRate = 30
|
|
MaxPushPoints = 2000
|
|
RingTemporal = 1000000
|
|
RingScalar = 100000
|
|
+History = {
|
|
Directory = "/tmp/streamhub_history"
|
|
DurationHours = 1
|
|
Decimation = 10
|
|
FlushIntervalSec = 5
|
|
MinDiskFreeMB = 200
|
|
}
|
|
+Recorder = {
|
|
Enabled = 0
|
|
AutoStart = 1
|
|
Directory = "/tmp/streamhub_rec"
|
|
MaxFileMB = 256
|
|
KeepFiles = 8
|
|
StagingMB = 8
|
|
FlushIntervalSec = 5
|
|
MinDiskFreeMB = 500
|
|
Signals = "all"
|
|
}
|
|
Sources = {
|
|
scalar = {
|
|
Label = "Scalar Sines (1 ksps)"
|
|
Addr = "127.0.0.1"
|
|
Port = 44500
|
|
MulticastGroup = "239.0.0.1"
|
|
DataPort = 44503
|
|
}
|
|
med = {
|
|
Label = "1 Msps Sines (Ch1 1kHz, Ch2 5kHz)"
|
|
Addr = "127.0.0.1"
|
|
Port = 44501
|
|
}
|
|
fast = {
|
|
Label = "5 Msps Sines (Ch3 10kHz, Ch4 50kHz)"
|
|
Addr = "127.0.0.1"
|
|
Port = 44502
|
|
}
|
|
}
|
|
}
|
|
EOF
|
|
|
|
# ── Library path — covers both MARTe2 app and StreamHub ──────────────────────
|
|
export LD_LIBRARY_PATH="\
|
|
${MARTe2_DIR}/Build/${BUILD_TARGET}/Core:\
|
|
${MARTe2_Components_DIR}/Build/${BUILD_TARGET}/Components/DataSources/LinuxTimer:\
|
|
${MARTe2_Components_DIR}/Build/${BUILD_TARGET}/Components/GAMs/IOGAM:\
|
|
${BUILD_DIR}/Components/DataSources/UDPStreamer:\
|
|
${BUILD_DIR}/Components/GAMs/SineArrayGAM:\
|
|
${BUILD_DIR}/Components/GAMs/TimeArrayGAM:\
|
|
${BUILD_DIR}/Components/Interfaces/UDPStream:\
|
|
${LD_LIBRARY_PATH:-}"
|
|
|
|
# ── Cleanup handler ───────────────────────────────────────────────────────────
|
|
MARTE_PID=""
|
|
HUB_PID=""
|
|
GUI_PID=""
|
|
WEBUI_PID=""
|
|
|
|
cleanup() {
|
|
echo ""
|
|
echo "==> Shutting down..."
|
|
[[ -n "$WEBUI_PID" ]] && kill "$WEBUI_PID" 2>/dev/null || true
|
|
[[ -n "$GUI_PID" ]] && kill "$GUI_PID" 2>/dev/null || true
|
|
[[ -n "$HUB_PID" ]] && kill "$HUB_PID" 2>/dev/null || true
|
|
[[ -n "$MARTE_PID" ]] && kill "$MARTE_PID" 2>/dev/null || true
|
|
wait "$WEBUI_PID" 2>/dev/null || true
|
|
wait "$GUI_PID" 2>/dev/null || true
|
|
wait "$HUB_PID" 2>/dev/null || true
|
|
wait "$MARTE_PID" 2>/dev/null || true
|
|
rm -f "$HUB_CFG"
|
|
echo "==> Done."
|
|
}
|
|
trap cleanup EXIT INT TERM
|
|
|
|
# ── Launch MARTe2 ─────────────────────────────────────────────────────────────
|
|
echo ""
|
|
echo "==> Launching MARTe2..."
|
|
echo " Binary : ${MARTE2_BIN}"
|
|
echo " Config : ${MARTe_CFG}"
|
|
echo " Signals: Sine1 (1 Hz), Sine2 (0.3 Hz), Ch1-Ch4 (arrays)"
|
|
echo ""
|
|
|
|
"${MARTE2_BIN}" \
|
|
-l RealTimeLoader \
|
|
-f "${MARTe_CFG}" \
|
|
-s Running \
|
|
-m StateMachine:START &
|
|
MARTE_PID="$!"
|
|
|
|
# Give MARTe2 a moment to bind its UDP ports before StreamHub connects
|
|
sleep 1
|
|
|
|
# ── Launch StreamHub ──────────────────────────────────────────────────────────
|
|
echo "==> Launching StreamHub..."
|
|
echo " Binary : ${STREAMHUB_EX}"
|
|
echo " Config : ${HUB_CFG}"
|
|
echo " WS port : ${WS_PORT}"
|
|
echo " MaxPoints: ${MAX_POINTS}"
|
|
echo ""
|
|
|
|
"${STREAMHUB_EX}" -cfg "${HUB_CFG}" &
|
|
HUB_PID="$!"
|
|
|
|
# ── Optionally launch the ImGui client ───────────────────────────────────────
|
|
if [[ "$START_GUI" -eq 1 ]]; then
|
|
if [[ ! -x "$IMGUI_CLIENT" ]]; then
|
|
echo "WARNING: ImGui client not found at ${IMGUI_CLIENT}"
|
|
echo " Build it with: cd Client/streamhub && cmake -B build && cmake --build build"
|
|
else
|
|
sleep 0.5
|
|
echo "==> Launching ImGui client (127.0.0.1:${WS_PORT})..."
|
|
"${IMGUI_CLIENT}" -host 127.0.0.1 -port "${WS_PORT}" &
|
|
GUI_PID="$!"
|
|
fi
|
|
fi
|
|
|
|
# ── Optionally launch the web UI server ──────────────────────────────────────
|
|
if [[ "$START_WEBUI" -eq 1 ]]; then
|
|
if [[ ! -x "$WEBUI_BIN" ]]; then
|
|
echo "WARNING: webui binary not found at ${WEBUI_BIN}"
|
|
echo " Build it with: cd Client/webui && go build -o streamhub-webui ."
|
|
else
|
|
echo "==> Launching web UI server (:${WEBUI_PORT})..."
|
|
"${WEBUI_BIN}" -addr ":${WEBUI_PORT}" \
|
|
-hub "localhost:${WS_PORT}" \
|
|
-static "${SCRIPT_DIR}/Client/udpstreamer/static" &
|
|
WEBUI_PID="$!"
|
|
fi
|
|
fi
|
|
|
|
# ── Status ────────────────────────────────────────────────────────────────────
|
|
echo " MARTe2 PID : ${MARTE_PID}"
|
|
echo " StreamHub PID: ${HUB_PID}"
|
|
[[ -n "$GUI_PID" ]] && echo " ImGui PID : ${GUI_PID}"
|
|
[[ -n "$WEBUI_PID" ]] && echo " WebUI PID : ${WEBUI_PID}"
|
|
echo ""
|
|
echo " StreamHub WebSocket: ws://127.0.0.1:${WS_PORT}"
|
|
[[ -n "$WEBUI_PID" ]] && echo " Browser client : http://localhost:${WEBUI_PORT}/"
|
|
[[ -x "$IMGUI_CLIENT" ]] && echo " ImGui client : ${IMGUI_CLIENT} -host 127.0.0.1 -port ${WS_PORT}"
|
|
echo ""
|
|
echo " Press Ctrl-C to stop all processes."
|
|
echo ""
|
|
|
|
# ── Wait until any child exits ────────────────────────────────────────────────
|
|
wait -n "${MARTE_PID}" "${HUB_PID}" ${GUI_PID:-} ${WEBUI_PID:-} 2>/dev/null || true
|
|
echo "==> A process exited — stopping remaining processes."
|