From f526eb7cdc52754c187f6dd4442f69b60554c006 Mon Sep 17 00:00:00 2001
From: Daniel Ekman <knegge@gmail.com>
Date: Thu, 12 Jan 2023 13:46:45 +0100
Subject: [PATCH] Add docker compose and build

Signed-off-by: Daniel Ekman <knegge@gmail.com>
---
 Dockerfile         | 19 +++++++++++++++
 docker-compose.yml | 18 ++++++++++++++
 docker_rtlsdr.sh   | 61 ++++++++++++++++++++++++++++++++++++++++++++++
 user.env.example   | 26 ++++++++++++++++++++
 4 files changed, 124 insertions(+)
 create mode 100644 Dockerfile
 create mode 100644 docker-compose.yml
 create mode 100755 docker_rtlsdr.sh
 create mode 100644 user.env.example

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..3d105f1
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,19 @@
+FROM debian:bullseye
+MAINTAINER sa2kng <knegge@gmail.com>
+ARG HOMEDIR=/home/pi/horusdemodlib
+ENV PATH=${PATH}:${HOMEDIR}
+
+RUN apt-get -y update && apt -y upgrade && apt-get -y install --no-install-recommends cmake build-essential libusb-1.0-0-dev git python3-venv python3-crcmod python3-requests python3-pip sox bc rtl-sdr libatlas-base-dev rtl-sdr && rm -rf /var/lib/apt/lists/*
+
+COPY . ${HOMEDIR}
+WORKDIR ${HOMEDIR}
+
+RUN cmake -B build -DCMAKE_INSTALL_PREFIX=/usr -DCMAKE_BUILD_TYPE=Release &&\
+    cmake --build build --target install
+RUN echo '[global]\nextra-index-url=https://www.piwheels.org/simple' > /etc/pip.conf &&\
+    python3 -m venv venv &&\
+    . venv/bin/activate &&\
+    pip install --no-cache-dir --prefer-binary -r requirements.txt &&\
+    pip install --no-cache-dir --prefer-binary horusdemodlib
+
+CMD ["bash"]
diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..e704999
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,18 @@
+version: '3.7'
+services:
+  horusdemod:
+    build:
+      context: .
+    image: 'projecthorus/horusdemodlib:latest'
+    command: 'docker_rtlsdr.sh'
+      #read_only: true
+    device_cgroup_rules:
+      - 'c 189:* rwm'
+    env_file:
+      - ./user.env
+    volumes:
+      - type: 'tmpfs'
+        target: '/tmp'
+      - './user.cfg:/home/pi/horusdemodlib/user.cfg'
+      - '/dev/bus/usb:/dev/bus/usb'
+
diff --git a/docker_rtlsdr.sh b/docker_rtlsdr.sh
new file mode 100755
index 0000000..814bd25
--- /dev/null
+++ b/docker_rtlsdr.sh
@@ -0,0 +1,61 @@
+#!/usr/bin/env bash
+#
+#	Horus Binary RTLSDR Helper Script
+#
+#   Uses rtl_fm to receive a chunk of spectrum, and passes it into horus_demod.
+#
+
+# Check that the horus_demod decoder has been compiled.
+DECODER=./build/src/horus_demod
+if [ -f "$DECODER" ]; then
+    echo "Found horus_demod."
+else 
+    echo "ERROR - $DECODER does not exist - have you compiled it yet?"
+	exit 1
+fi
+
+# Check that bc is available on the system path.
+if echo "1+1" | bc > /dev/null; then
+    echo "Found bc."
+else 
+    echo "ERROR - Cannot find bc - Did you install it?"
+	exit 1
+fi
+
+# Use a local venv if it exists
+VENV_DIR=venv
+if [ -d "$VENV_DIR" ]; then
+    echo "Entering venv."
+    source $VENV_DIR/bin/activate
+fi
+
+# Calculate the SDR tuning frequency
+SDR_RX_FREQ=$(echo "$RXFREQ - $RXBANDWIDTH/2 - 1000" | bc)
+
+# Calculate the frequency estimator limits
+FSK_LOWER=1000
+FSK_UPPER=$(echo "$FSK_LOWER + $RXBANDWIDTH" | bc)
+
+echo "Using SDR Centre Frequency: $SDR_RX_FREQ Hz."
+echo "Using FSK estimation range: $FSK_LOWER - $FSK_UPPER Hz"
+
+BIAS_SETTING=""
+
+if [ "$BIAS" = "1" ]; then
+	echo "Enabling Bias Tee."
+	BIAS_SETTING=" -T"
+fi
+
+GAIN_SETTING=""
+if [ "$GAIN" = "0" ]; then
+	echo "Using AGC."
+	GAIN_SETTING=""
+else
+	echo "Using Manual Gain"
+	GAIN_SETTING=" -g $GAIN"
+fi
+
+# Start the receive chain.
+# Note that we now pass in the SDR centre frequency ($SDR_RX_FREQ) and 'target' signal frequency ($RXFREQ)
+# to enable providing additional metadata to Habitat / Sondehub.
+rtl_fm -M raw -F9 -s 48000 -p $PPM $GAIN_SETTING$BIAS_SETTING -f $SDR_RX_FREQ | $DECODER -q --stats=5 -g -m binary --fsk_lower=$FSK_LOWER --fsk_upper=$FSK_UPPER - - | python -m horusdemodlib.uploader --freq_hz $SDR_RX_FREQ --freq_target_hz $RXFREQ $@
diff --git a/user.env.example b/user.env.example
new file mode 100644
index 0000000..2f0f198
--- /dev/null
+++ b/user.env.example
@@ -0,0 +1,26 @@
+# Receive *centre* frequency, in Hz
+# Note: The SDR will be tuned to RXBANDWIDTH/2 below this frequency.
+RXFREQ=434200000
+
+# Receiver Gain. Set this to 0 to use automatic gain control, otherwise if running a
+# preamplifier, you may want to experiment with different gain settings to optimize
+# your receiver setup.
+# You can find what gain range is valid for your RTLSDR by running: rtl_test
+GAIN=0
+
+# Bias Tee Enable (1) or Disable (0)
+BIAS=0
+
+# Receiver PPM offset
+PPM=0
+
+# Frequency estimator bandwidth. The wider the bandwidth, the more drift and frequency error the modem can tolerate,
+# but the higher the chance that the modem will lock on to a strong spurious signal.
+# Note: The SDR will be tuned to RXFREQ-RXBANDWIDTH/2, and the estimator set to look at 0-RXBANDWIDTH Hz.
+RXBANDWIDTH=10000
+
+# Enable (1) or disable (0) modem statistics output.
+# If enabled, modem statistics are written to stats.txt, and can be observed
+# during decoding by running: tail -f stats.txt | python fskstats.py
+STATS_OUTPUT=0
+