фикс RXLOSS для sbus

README.md

@@ -2,6 +2,7 @@
 
 Это проект с двумя частями программы:
 * программа `ground` - запускается на наземной станции
+* программа `air` - запускается на воздушной станции (SDRPi)
 
 
 # Зависимости для ground
@@ -22,3 +23,28 @@ brew install sdl2
 
 Под шиндовс, как всегда, надо страдать. Рекомендуется установить `sdl2` через `vcpkg`.
 
+# Запуск
+
+## на "воздушной" части:
+```shell
+sdrpi-fpv-control-air /dev/ttyPS0
+```
+
+> На данный момент эта команда добавлена в автозапуск, никаких дополнительных действий от пользователя не требуется.
+
+## На "земле"
+
+Если у вас windows, скачать релиз можно:
+* с шары (`\\SHARE\share\vlad405\sdripi-fpv-control`) [(ссылка для линуксоидов)](smb://share.local/share/vlad405/sdripi-fpv-control)
+* со [страницы релиза](https://git.wawaa.ru/VladislavOstapov/sdrpi-fpv-control/releases).
+
+Если вы ~~нормальный человек~~ разработчик, то проще собрать программу при помощи cmake.
+
+Запускать программу надо на ноуте, предварительно подключив пульт.
+```shell
+sdrpi-fpv-control-ground 192.168.1.5 20
+```
+У программы 2 необязательных аргумента. Первый - IP адрес "воздушной" SDRPi. Второй - количество пакетов управления в секунду.
+По умолчанию IP адрес - localhost, а количество пакетов в секунду - 5.
+Для мягкого управления рекомендуется ставить от 15 до 50 пакетов в секунду. При показе рекомендуется устанавливать значение 20.
+

air/main.cpp

@@ -1,4 +1,5 @@
 #include <sys/ioctl.h>
+#include <sys/poll.h>
 #include <asm/termbits.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
@@ -11,17 +12,26 @@
 #include <cstring>
 #include <atomic>
 #include <span>
+#include <sys/time.h>
 
+/**
+ * Вспомогательная функция для получения текущего времени в миллисекундах
+ */
+inline int64_t milliseconds() {
+    timeval tv{};
+    gettimeofday(&tv,nullptr);
+    return ((tv.tv_sec * 1000000l) + tv.tv_usec) / 1000;
+}
 
-
+constexpr int64_t SBUS_SEND_FRAME_INTERVAL = 15;
+constexpr int64_t SBUS_RXLOSS_INTERVAL = 500;
 
 class UDPServer {
-private:
+public:
     int sockfd;
     sockaddr_in server_addr, client_addr;
     socklen_t client_len;
 
-public:
     UDPServer(uint16_t port) : client_len(sizeof(client_addr)) {
         // Создание UDP сокета
         sockfd = socket(AF_INET, SOCK_DGRAM, 0);
@@ -84,7 +94,9 @@ struct SbusData {
     bool lost_frame = false;
     bool failsafe = false;
     bool ch17 = false, ch18 = false;
-    static constexpr size_t NUM_CH = 16;
+    static constexpr size_t NUM_CH = 18;
+    static constexpr int16_t SBUS_CH_MIN = 173 + 10;
+    static constexpr int16_t SBUS_CH_MAX = 1812 - 10;
     int16_t ch[NUM_CH];
 
     /* Message len */
@@ -98,52 +110,68 @@ struct SbusData {
     static constexpr uint8_t CH18_MASK_ = 0x02;
     static constexpr uint8_t LOST_FRAME_MASK_ = 0x04;
     static constexpr uint8_t FAILSAFE_MASK_ = 0x08;
-    /* Data */
-    uint8_t buf_[BUF_LEN_];
+    uint8_t binaryBuffer[25];
 
     void fillDataBuf() {
-        /* Assemble packet */
-        buf_[0] = HEADER_;
-        buf_[1] = static_cast<uint8_t>((ch[0] & 0x07FF));
-        buf_[2] = static_cast<uint8_t>((ch[0] & 0x07FF) >> 8 |
-                (ch[1] & 0x07FF) << 3);
-        buf_[3] = static_cast<uint8_t>((ch[1] & 0x07FF) >> 5 |
-                (ch[2] & 0x07FF) << 6);
-        buf_[4] = static_cast<uint8_t>((ch[2] & 0x07FF) >> 2);
-        buf_[5] = static_cast<uint8_t>((ch[2] & 0x07FF) >> 10 |
-                (ch[3] & 0x07FF) << 1);
-        buf_[6] = static_cast<uint8_t>((ch[3] & 0x07FF) >> 7 |
-                (ch[4] & 0x07FF) << 4);
-        buf_[7] = static_cast<uint8_t>((ch[4] & 0x07FF) >> 4 |
-                (ch[5] & 0x07FF) << 7);
-        buf_[8] = static_cast<uint8_t>((ch[5] & 0x07FF) >> 1);
-        buf_[9] = static_cast<uint8_t>((ch[5] & 0x07FF) >> 9  |
-                (ch[6] & 0x07FF) << 2);
-        buf_[10] = static_cast<uint8_t>((ch[6] & 0x07FF) >> 6  |
-                 (ch[7] & 0x07FF) << 5);
-        buf_[11] = static_cast<uint8_t>((ch[7] & 0x07FF) >> 3);
-        buf_[12] = static_cast<uint8_t>((ch[8] & 0x07FF));
-        buf_[13] = static_cast<uint8_t>((ch[8] & 0x07FF) >> 8 |
-                 (ch[9]  & 0x07FF) << 3);
-        buf_[14] = static_cast<uint8_t>((ch[9] & 0x07FF) >> 5 |
-                 (ch[10] & 0x07FF) << 6);
-        buf_[15] = static_cast<uint8_t>((ch[10] & 0x07FF) >> 2);
-        buf_[16] = static_cast<uint8_t>((ch[10] & 0x07FF) >> 10 |
-                 (ch[11] & 0x07FF) << 1);
-        buf_[17] = static_cast<uint8_t>((ch[11] & 0x07FF) >> 7  |
-                 (ch[12] & 0x07FF) << 4);
-        buf_[18] = static_cast<uint8_t>((ch[12] & 0x07FF) >> 4  |
-                 (ch[13] & 0x07FF) << 7);
-        buf_[19] = static_cast<uint8_t>((ch[13] & 0x07FF) >> 1);
-        buf_[20] = static_cast<uint8_t>((ch[13] & 0x07FF) >> 9  |
-                 (ch[14] & 0x07FF) << 2);
-        buf_[21] = static_cast<uint8_t>((ch[14] & 0x07FF) >> 6  |
-                 (ch[15] & 0x07FF) << 5);
-        buf_[22] = static_cast<uint8_t>((ch[15] & 0x07FF) >> 3);
-        buf_[23] = 0x00 | (ch17 * CH17_MASK_) | (ch18 * CH18_MASK_) |
-                 (failsafe * FAILSAFE_MASK_) |
-                 (lost_frame * LOST_FRAME_MASK_);
-        buf_[24] = FOOTER_;
+        typedef struct sbusChannels_s {
+            // 176 bits of data (11 bits per channel * 16 channels) = 22 bytes.
+            unsigned int chan0 : 11;
+            unsigned int chan1 : 11;
+            unsigned int chan2 : 11;
+            unsigned int chan3 : 11;
+            unsigned int chan4 : 11;
+            unsigned int chan5 : 11;
+            unsigned int chan6 : 11;
+            unsigned int chan7 : 11;
+            unsigned int chan8 : 11;
+            unsigned int chan9 : 11;
+            unsigned int chan10 : 11;
+            unsigned int chan11 : 11;
+            unsigned int chan12 : 11;
+            unsigned int chan13 : 11;
+            unsigned int chan14 : 11;
+            unsigned int chan15 : 11;
+        } __attribute__((__packed__)) sbusChannels_t;
+        static_assert(sizeof(sbusChannels_s) == 22);
+        struct sbusFrame_s {
+            uint8_t syncByte = HEADER_;
+            sbusChannels_t channels{};
+            uint8_t flags{};
+            /**
+             * The endByte is 0x00 on FrSky and some futaba RX's, on Some SBUS2 RX's the value indicates the telemetry byte that is sent after every 4th sbus frame.
+             *
+             * See https://github.com/cleanflight/cleanflight/issues/590#issuecomment-101027349
+             * and
+             * https://github.com/cleanflight/cleanflight/issues/590#issuecomment-101706023
+             */
+            uint8_t endByte = FOOTER_;
+        } __attribute__ ((__packed__));
+        static_assert(sizeof(sbusFrame_s) == sizeof(binaryBuffer));
+
+        auto* dest = reinterpret_cast<sbusFrame_s*>(binaryBuffer);
+        dest->syncByte = HEADER_;
+        dest->channels.chan0 = std::min(std::max(ch[0], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan1 = std::min(std::max(ch[1], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan2 = std::min(std::max(ch[2], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan3 = std::min(std::max(ch[3], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan4 = std::min(std::max(ch[4], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan5 = std::min(std::max(ch[5], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan6 = std::min(std::max(ch[6], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan7 = std::min(std::max(ch[7], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan8 = std::min(std::max(ch[8], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan9 = std::min(std::max(ch[9], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan10 = std::min(std::max(ch[10], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan11 = std::min(std::max(ch[11], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan12 = std::min(std::max(ch[12], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan13 = std::min(std::max(ch[13], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan14 = std::min(std::max(ch[14], SBUS_CH_MIN), SBUS_CH_MAX);
+        dest->channels.chan15 = std::min(std::max(ch[15], SBUS_CH_MIN), SBUS_CH_MAX);
+
+        dest->flags = 0;
+        if (ch[16] >= (SBUS_CH_MAX + SBUS_CH_MAX) / 2) { dest->flags |= CH17_MASK_; }
+        if (ch[16] >= (SBUS_CH_MAX + SBUS_CH_MAX) / 2) { dest->flags |= CH18_MASK_; }
+
+        dest->endByte = FOOTER_;
     }
 
 };
@@ -243,32 +271,57 @@ int main(int argc, char* argv[]) {
         std::cout << "Press Ctrl+C to exit" << std::endl;
 
         SbusData sb{};
+        int64_t lastSbusWrite = 0;
+        int64_t lastSbusRecv = 0;
 
-        int packet_count = 0;
+        int packetCount = 0;
+        int totalPacketCount = 0;
+        int64_t lastStatisticsShow = milliseconds();
         while (true) {
-            // Прием UDP пакета
-            std::vector<uint16_t> data = udp_server.receive();
+            pollfd udpFd{.fd = udp_server.sockfd, .events = POLLIN, .revents = 0};
+            poll(&udpFd, 1, SBUS_SEND_FRAME_INTERVAL / 4);
 
-            if (!data.empty()) {
-                packet_count++;
+            if (udpFd.revents & POLLIN) {
+                // Прием UDP пакета
+                std::vector<uint16_t> data = udp_server.receive();
+                lastSbusRecv = milliseconds();
 
-                for (int i = 0; i < data.size() && i < SbusData::NUM_CH; ++i) {
-                    auto item = static_cast<int16_t>(data[i]);
-                    sb.ch[i] = static_cast<int16_t>((item - 1000.0) * 2);
-                    if (sb.ch[i] < 50) {
-                        sb.ch[i] = 50; // минимальное число
-                    } else if (sb.ch[i] > 1900) {
-                        sb.ch[i] = 1900; // максимальное число
+                if (!data.empty()) {
+                    packetCount++;
+
+                    for (int i = 0; i < data.size() && i < SbusData::NUM_CH; ++i) {
+                        auto item = static_cast<double>(data[i]);
+                        item -= 1000.0;
+                        item = std::min(item, 1000.0);
+                        item = std::max(item, 0.0);
+                        item *= (SbusData::SBUS_CH_MAX - SbusData::SBUS_CH_MIN) / 1000.0;
+                        item += SbusData::SBUS_CH_MIN;
+
+                        sb.ch[i] = static_cast<int16_t>(item);
+                        if (sb.ch[i] < SbusData::SBUS_CH_MIN) {
+                            sb.ch[i] = SbusData::SBUS_CH_MIN; // минимальное число
+                        } else if (sb.ch[i] > SbusData::SBUS_CH_MAX) {
+                            sb.ch[i] = SbusData::SBUS_CH_MAX; // максимальное число
+                        }
                     }
+                    sb.fillDataBuf();
                 }
-                sb.fillDataBuf();
-                serial.write(sb.buf_);
+            }
 
-                // Выводим статистику каждые 100 пакетов
-                if (packet_count % 100 == 0) {
-                    std::cout << "Received " << packet_count << " packets total" << std::endl;
+            const auto now = milliseconds();
+            if (now - lastSbusWrite >= SBUS_SEND_FRAME_INTERVAL && now - lastSbusRecv <= SBUS_RXLOSS_INTERVAL) {
+                lastSbusWrite = now;
+                if (!serial.write(sb.binaryBuffer)) {
+                    break;
                 }
             }
+
+            if (now - lastStatisticsShow >= 1000) {
+                lastStatisticsShow = now;
+                totalPacketCount += packetCount;
+                std::cout << "Received " << totalPacketCount << " packets total, " << packetCount << " in last second" << std::endl;
+                packetCount = 0;
+            }
         }
 
     } catch (const std::exception& e) {

ground/joystick-reader.cpp

@@ -64,13 +64,12 @@ bool JoystickReader::readData(std::vector<uint16_t>& data) {
     // Читаем кнопки
     int buttons = SDL_JoystickNumButtons(joystick);
     for (int i = 0; i < buttons && i < data.size() - axes; ++i) {
-        auto buttonState = SDL_JoystickGetButton(joystick, i);
-        data[axes + i] = static_cast<uint16_t>(1000.0 + (buttonState * (1000.0 / 255.0)));
+        data[axes + i] = SDL_JoystickGetButton(joystick, i) ? 2000 : 1000;
     }
 
     for (auto& i: data) {
-        if (i < 950) i = 950;
-        if (i > 2050) i = 2050;
+        if (i < 1000) i = 1000;
+        if (i > 2000) i = 2000;
     }
     return true;
 }

ground/main.cpp

@@ -27,8 +27,8 @@ int main(int argc, char* argv[]) {
 
     JoystickReader reader;
     const int64_t timeInterval = 1000 / frequency;
-    while (!reader.initialize()) {
-        std::this_thread::sleep_for(std::chrono::milliseconds(1000));
+    if (!reader.initialize()) {
+        return 1;
     }
 
     UDPSocket udp(sendAddress, 1066);