добавил все для наземной станции
This commit is contained in:
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
@@ -0,0 +1,3 @@
|
||||
cmake-build-*
|
||||
.idea
|
||||
|
||||
38
CMakeLists.txt
Normal file
38
CMakeLists.txt
Normal file
@@ -0,0 +1,38 @@
|
||||
cmake_minimum_required(VERSION 3.28)
|
||||
project(sdrpi-fpv-control)
|
||||
|
||||
set(CMAKE_CXX_STANDARD 20)
|
||||
|
||||
option(ENABLE_GROUND_BUILD "Enable build ground station" OFF)
|
||||
option(ENABLE_AIR_BUILD "Enable build air" ON)
|
||||
|
||||
if (ENABLE_GROUND_BUILD)
|
||||
message("Enabled ground build!")
|
||||
# Настройки для Windows
|
||||
if(WIN32)
|
||||
add_definitions(-DWIN32_LEAN_AND_MEAN)
|
||||
set(EXTRA_LIBS ws2_32)
|
||||
endif()
|
||||
|
||||
find_package(SDL2 REQUIRED)
|
||||
include_directories(${SDL2_INCLUDE_DIRS})
|
||||
|
||||
add_executable(${PROJECT_NAME}-ground
|
||||
ground/joystick-reader.cpp
|
||||
ground/joystick-reader.h
|
||||
ground/main.cpp
|
||||
ground/udp-driver.cpp
|
||||
ground/udp-driver.h
|
||||
)
|
||||
|
||||
target_link_libraries(${PROJECT_NAME}-ground ${SDL2_LIBRARIES} ${EXTRA_LIBS})
|
||||
endif()
|
||||
|
||||
if (ENABLE_AIR_BUILD)
|
||||
message("Enabled air build!")
|
||||
|
||||
add_executable(${PROJECT_NAME}-air
|
||||
air/main.cpp
|
||||
)
|
||||
endif()
|
||||
|
||||
25
README.md
Normal file
25
README.md
Normal file
@@ -0,0 +1,25 @@
|
||||
# SDRPi fpv control
|
||||
|
||||
Это проект с двумя частями программы:
|
||||
* программа `ground` - запускается на наземной станции
|
||||
|
||||
|
||||
# Зависимости для ground
|
||||
|
||||
## Linux
|
||||
|
||||
```shell
|
||||
sudo apt install libhidapi-dev
|
||||
```
|
||||
|
||||
## MacOS
|
||||
|
||||
```shell
|
||||
brew install hidapi
|
||||
```
|
||||
|
||||
## Шиндовс
|
||||
|
||||
Под шиндовс, как всегда, надо страдать. Рекомендуется установить `libusb` и `hidapi` через `vcpkg`.
|
||||
|
||||
#
|
||||
81
ground/joystick-reader.cpp
Normal file
81
ground/joystick-reader.cpp
Normal file
@@ -0,0 +1,81 @@
|
||||
#include "joystick-reader.h"
|
||||
#include <iostream>
|
||||
#include <thread>
|
||||
#include <chrono>
|
||||
|
||||
JoystickReader::JoystickReader(int frequency)
|
||||
: joystick(nullptr), updateIntervalMs(1000 / frequency) {}
|
||||
|
||||
JoystickReader::~JoystickReader() {
|
||||
if (joystick) {
|
||||
SDL_JoystickClose(joystick);
|
||||
}
|
||||
SDL_Quit();
|
||||
}
|
||||
|
||||
bool JoystickReader::initialize() {
|
||||
if (SDL_Init(SDL_INIT_JOYSTICK) < 0) {
|
||||
std::cerr << "SDL initialization failed: " << SDL_GetError() << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
int numJoysticks = SDL_NumJoysticks();
|
||||
if (numJoysticks < 1) {
|
||||
std::cerr << "No joysticks found" << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
std::cout << "Found " << numJoysticks << " joystick(s)" << std::endl;
|
||||
|
||||
// Открываем первый джойстик
|
||||
joystick = SDL_JoystickOpen(0);
|
||||
if (!joystick) {
|
||||
std::cerr << "Failed to open joystick: " << SDL_GetError() << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
joystickName = SDL_JoystickName(joystick);
|
||||
std::cout << "Opened joystick: " << joystickName << std::endl;
|
||||
std::cout << "Axes: " << SDL_JoystickNumAxes(joystick)
|
||||
<< ", Buttons: " << SDL_JoystickNumButtons(joystick)
|
||||
<< ", Balls: " << SDL_JoystickNumBalls(joystick)
|
||||
<< ", Hats: " << SDL_JoystickNumHats(joystick) << std::endl;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool JoystickReader::readData(std::vector<uint16_t>& data) {
|
||||
if (!joystick) return false;
|
||||
if (SDL_JoystickGetAttached(joystick) != SDL_TRUE) { return false; }
|
||||
|
||||
SDL_JoystickUpdate();
|
||||
|
||||
data.resize(16, 1500); // Заполняем нейтральным значением
|
||||
|
||||
// Читаем оси (обычно до 6 осей)
|
||||
int axes = std::min(SDL_JoystickNumAxes(joystick), 6);
|
||||
for (int i = 0; i < axes; ++i) {
|
||||
Sint16 axisValue = SDL_JoystickGetAxis(joystick, i);
|
||||
// Преобразуем из [-32768, 32767] в [1000, 2000]
|
||||
data[i] = static_cast<uint16_t>((axisValue + 32768) * 1000 / 65536 + 1000);
|
||||
}
|
||||
|
||||
// Читаем кнопки
|
||||
int buttons = SDL_JoystickNumButtons(joystick);
|
||||
for (int i = 0; i < buttons && i < data.size() - axes; ++i) {
|
||||
Uint8 buttonState = SDL_JoystickGetButton(joystick, i);
|
||||
data[axes + i] = buttonState ? 2000 : 1000;
|
||||
}
|
||||
|
||||
for (auto& i: data) {
|
||||
if (i < 900) i = 900;
|
||||
if (i > 1100) i = 1100;
|
||||
}
|
||||
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(updateIntervalMs));
|
||||
return true;
|
||||
}
|
||||
|
||||
std::string JoystickReader::getJoystickName() const {
|
||||
return joystickName;
|
||||
}
|
||||
29
ground/joystick-reader.h
Normal file
29
ground/joystick-reader.h
Normal file
@@ -0,0 +1,29 @@
|
||||
#ifndef SDRPI_FPV_CONTROL_GROUND_JOYSTICK_READER_H
|
||||
#define SDRPI_FPV_CONTROL_GROUND_JOYSTICK_READER_H
|
||||
|
||||
#include <cstdint>
|
||||
#include <vector>
|
||||
#include <string>
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <SDL.h>
|
||||
#else
|
||||
#include <SDL2/SDL.h>
|
||||
#endif
|
||||
|
||||
class JoystickReader {
|
||||
public:
|
||||
JoystickReader(int frequency = 50);
|
||||
~JoystickReader();
|
||||
|
||||
bool initialize();
|
||||
bool readData(std::vector<uint16_t>& data);
|
||||
std::string getJoystickName() const;
|
||||
|
||||
private:
|
||||
SDL_Joystick* joystick;
|
||||
int updateIntervalMs;
|
||||
std::string joystickName;
|
||||
};
|
||||
|
||||
#endif //SDRPI_FPV_CONTROL_GROUND_JOYSTICK_READER_H
|
||||
52
ground/main.cpp
Normal file
52
ground/main.cpp
Normal file
@@ -0,0 +1,52 @@
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <thread>
|
||||
#include <csignal>
|
||||
|
||||
#include "joystick-reader.h"
|
||||
#include "udp-driver.h"
|
||||
|
||||
int main(int argc, char* argv[]) {
|
||||
|
||||
// Парсим аргументы командной строки для частоты
|
||||
int frequency = 50;
|
||||
const char* sendAddress;
|
||||
if (argc == 1) {
|
||||
sendAddress = "127.0.0.1";
|
||||
} else if (argc == 2) {
|
||||
sendAddress = argv[1];
|
||||
} else if (argc == 3) {
|
||||
sendAddress = argv[1];
|
||||
frequency = std::atoi(argv[2]);
|
||||
} else {
|
||||
std::cerr << "Usage: " << argv[0] << " send.ip.addr.v4 [Frequency]" << std::endl;
|
||||
return 1;
|
||||
}
|
||||
|
||||
std::cout << "Starting joystick reader with frequency: " << frequency << " Hz" << std::endl;
|
||||
|
||||
JoystickReader reader(frequency);
|
||||
while (!reader.initialize()) {
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(1000));
|
||||
}
|
||||
|
||||
UDPSocket udp(sendAddress, 1066);
|
||||
std::vector<uint16_t> data;
|
||||
|
||||
std::cout << "Sending data to " << sendAddress << ":1066" << std::endl;
|
||||
|
||||
int counter = 0;
|
||||
while (reader.readData(data)) {
|
||||
if (!udp.sendFrame(data)) {
|
||||
std::cerr << "Failed to send UDP packet" << std::endl;
|
||||
}
|
||||
|
||||
// Выводим прогресс каждые 100 пакетов
|
||||
if (++counter % 100 == 0) {
|
||||
std::cout << "Sent " << counter << " packets" << std::endl;
|
||||
}
|
||||
}
|
||||
|
||||
std::cout << "Exiting..." << std::endl;
|
||||
return 0;
|
||||
}
|
||||
111
ground/udp-driver.cpp
Normal file
111
ground/udp-driver.cpp
Normal file
@@ -0,0 +1,111 @@
|
||||
#include "udp-driver.h"
|
||||
#include <iostream>
|
||||
#include <cstring>
|
||||
|
||||
UDPSocket::UDPSocket(const std::string& ip, uint16_t port) {
|
||||
#ifdef _WIN32
|
||||
sockfd = INVALID_SOCKET;
|
||||
#else
|
||||
sockfd = -1;
|
||||
#endif
|
||||
|
||||
if (!initialize()) {
|
||||
return;
|
||||
}
|
||||
|
||||
// Создание сокета
|
||||
#ifdef _WIN32
|
||||
sockfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
||||
if (sockfd == INVALID_SOCKET) {
|
||||
std::cerr << "Socket creation failed: " << WSAGetLastError() << std::endl;
|
||||
return;
|
||||
}
|
||||
#else
|
||||
sockfd = socket(AF_INET, SOCK_DGRAM, 0);
|
||||
if (sockfd < 0) {
|
||||
std::cerr << "Socket creation failed" << std::endl;
|
||||
return;
|
||||
}
|
||||
#endif
|
||||
|
||||
// Настройка адреса назначения
|
||||
memset(&dest_addr, 0, sizeof(dest_addr));
|
||||
dest_addr.sin_family = AF_INET;
|
||||
dest_addr.sin_port = htons(port);
|
||||
|
||||
#ifdef _WIN32
|
||||
dest_addr.sin_addr.s_addr = inet_addr(ip.c_str());
|
||||
if (dest_addr.sin_addr.s_addr == INADDR_NONE) {
|
||||
std::cerr << "Invalid address: " << ip << std::endl;
|
||||
closesocket(sockfd);
|
||||
sockfd = INVALID_SOCKET;
|
||||
}
|
||||
#else
|
||||
if (inet_pton(AF_INET, ip.c_str(), &dest_addr.sin_addr) <= 0) {
|
||||
std::cerr << "Invalid address: " << ip << std::endl;
|
||||
close(sockfd);
|
||||
sockfd = -1;
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
bool UDPSocket::initialize() {
|
||||
#ifdef _WIN32
|
||||
// Инициализация Winsock
|
||||
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
|
||||
std::cerr << "WSAStartup failed: " << WSAGetLastError() << std::endl;
|
||||
return false;
|
||||
}
|
||||
#endif
|
||||
return true;
|
||||
}
|
||||
|
||||
UDPSocket::~UDPSocket() {
|
||||
#ifdef _WIN32
|
||||
if (sockfd != INVALID_SOCKET) {
|
||||
closesocket(sockfd);
|
||||
}
|
||||
WSACleanup();
|
||||
#else
|
||||
if (sockfd >= 0) {
|
||||
close(sockfd);
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
bool UDPSocket::sendFrame(const std::vector<uint16_t>& data) {
|
||||
#ifdef _WIN32
|
||||
if (sockfd == INVALID_SOCKET) return false;
|
||||
#else
|
||||
if (sockfd < 0) return false;
|
||||
#endif
|
||||
|
||||
if (data.empty()) return false;
|
||||
|
||||
// Отправка данных
|
||||
#ifdef _WIN32
|
||||
int result = sendto(sockfd,
|
||||
reinterpret_cast<const char*>(data.data()),
|
||||
data.size() * sizeof(uint16_t),
|
||||
0,
|
||||
(sockaddr*)&dest_addr,
|
||||
sizeof(dest_addr));
|
||||
if (result == SOCKET_ERROR) {
|
||||
std::cerr << "Send failed: " << WSAGetLastError() << std::endl;
|
||||
return false;
|
||||
}
|
||||
#else
|
||||
ssize_t result = sendto(sockfd,
|
||||
data.data(),
|
||||
data.size() * sizeof(uint16_t),
|
||||
0,
|
||||
(sockaddr*)&dest_addr,
|
||||
sizeof(dest_addr));
|
||||
if (result < 0) {
|
||||
std::cerr << "Send failed" << std::endl;
|
||||
return false;
|
||||
}
|
||||
#endif
|
||||
|
||||
return true;
|
||||
}
|
||||
37
ground/udp-driver.h
Normal file
37
ground/udp-driver.h
Normal file
@@ -0,0 +1,37 @@
|
||||
#ifndef SDRPI_FPV_CONTROL_GROUND_UDP_DRIVER_H
|
||||
#define SDRPI_FPV_CONTROL_GROUND_UDP_DRIVER_H
|
||||
|
||||
#include <string>
|
||||
#include <cstdint>
|
||||
#include <vector>
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <winsock2.h>
|
||||
#include <ws2tcpip.h>
|
||||
#pragma comment(lib, "ws2_32.lib")
|
||||
#else
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <unistd.h>
|
||||
#endif
|
||||
|
||||
class UDPSocket {
|
||||
public:
|
||||
UDPSocket(const std::string& ip, uint16_t port);
|
||||
~UDPSocket();
|
||||
bool sendFrame(const std::vector<uint16_t>& data);
|
||||
|
||||
private:
|
||||
#ifdef _WIN32
|
||||
SOCKET sockfd;
|
||||
WSADATA wsaData;
|
||||
#else
|
||||
int sockfd;
|
||||
#endif
|
||||
sockaddr_in dest_addr;
|
||||
|
||||
bool initialize();
|
||||
};
|
||||
|
||||
#endif //SDRPI_FPV_CONTROL_GROUND_UDP_DRIVER_H
|
||||
Reference in New Issue
Block a user