mirror of
https://github.com/PhotonVision/photonvision
synced 2026-06-23 01:21:40 +00:00
Create timesync JNI for testing client (#1433)
This commit is contained in:
200
photon-targeting/src/main/native/cpp/net/TimeSyncClient.cpp
Normal file
200
photon-targeting/src/main/native/cpp/net/TimeSyncClient.cpp
Normal file
@@ -0,0 +1,200 @@
|
||||
/*
|
||||
* Copyright (C) Photon Vision.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include "net/TimeSyncClient.h"
|
||||
|
||||
#include <fmt/core.h>
|
||||
#include <wpinet/UDPClient.h>
|
||||
#include <wpinet/uv/util.h>
|
||||
|
||||
#include <atomic>
|
||||
#include <chrono>
|
||||
#include <cstdlib>
|
||||
#include <cstring>
|
||||
#include <ctime>
|
||||
#include <iostream>
|
||||
#include <mutex>
|
||||
#include <thread>
|
||||
|
||||
#include <Eigen/Core>
|
||||
#include <wpi/Logger.h>
|
||||
#include <wpi/struct/Struct.h>
|
||||
|
||||
#include "ntcore_cpp.h"
|
||||
|
||||
static void ClientLoggerFunc(unsigned int level, const char* file,
|
||||
unsigned int line, const char* msg) {
|
||||
if (level == 20) {
|
||||
fmt::print(stderr, "TimeSyncClient: {}\n", msg);
|
||||
return;
|
||||
}
|
||||
|
||||
std::string_view levelmsg;
|
||||
if (level >= 50) {
|
||||
levelmsg = "CRITICAL";
|
||||
} else if (level >= 40) {
|
||||
levelmsg = "ERROR";
|
||||
} else if (level >= 30) {
|
||||
levelmsg = "WARNING";
|
||||
} else {
|
||||
return;
|
||||
}
|
||||
fmt::print(stderr, "TimeSyncClient: {}: {} ({}:{})\n", levelmsg, msg, file,
|
||||
line);
|
||||
}
|
||||
|
||||
void wpi::tsp::TimeSyncClient::Tick() {
|
||||
// fmt::println("wpi::tsp::TimeSyncClient::Tick");
|
||||
// Regardless of if we've gotten a pong back yet, we'll ping again. this is
|
||||
// pretty naive but should be "fine" for now?
|
||||
|
||||
uint64_t ping_local_time{m_timeProvider()};
|
||||
|
||||
TspPing ping{.version = 1, .message_id = 1, .client_time = ping_local_time};
|
||||
|
||||
wpi::SmallVector<uint8_t, wpi::Struct<TspPing>::GetSize()> pingData(
|
||||
wpi::Struct<TspPing>::GetSize());
|
||||
wpi::PackStruct(pingData, ping);
|
||||
|
||||
// Wrap our buffer - pingData should free itself
|
||||
wpi::uv::Buffer pingBuf{pingData};
|
||||
int sent = m_udp->TrySend(wpi::SmallVector<wpi::uv::Buffer, 1>{pingBuf});
|
||||
|
||||
if (static_cast<size_t>(sent) != wpi::Struct<TspPing>::GetSize()) {
|
||||
WPI_ERROR(m_logger, "Didn't send the whole ping out? sent {} bytes", sent);
|
||||
return;
|
||||
}
|
||||
|
||||
{
|
||||
std::lock_guard lock{m_offsetMutex};
|
||||
m_metadata.pingsSent++;
|
||||
}
|
||||
|
||||
m_lastPing = ping;
|
||||
}
|
||||
|
||||
void wpi::tsp::TimeSyncClient::UdpCallback(uv::Buffer& buf, size_t nbytes,
|
||||
const sockaddr& sender,
|
||||
unsigned flags) {
|
||||
uint64_t pong_local_time = m_timeProvider();
|
||||
|
||||
if (static_cast<size_t>(nbytes) != wpi::Struct<TspPong>::GetSize()) {
|
||||
WPI_ERROR(m_logger, "Got {} bytes for pong?", nbytes);
|
||||
return;
|
||||
}
|
||||
|
||||
TspPong pong{
|
||||
wpi::UnpackStruct<TspPong>(buf.bytes()),
|
||||
};
|
||||
|
||||
// fmt::println("->[client] Got pong: {} {} {} {}", pong.version,
|
||||
// pong.message_id, pong.client_time, pong.server_time);
|
||||
|
||||
if (pong.version != 1) {
|
||||
fmt::println("Bad version from server?");
|
||||
return;
|
||||
}
|
||||
if (pong.message_id != 2) {
|
||||
fmt::println("Bad message id from server?");
|
||||
return;
|
||||
}
|
||||
|
||||
TspPing ping = m_lastPing;
|
||||
|
||||
if (pong.client_time != ping.client_time) {
|
||||
WPI_WARNING(m_logger,
|
||||
"Pong was not a reply to our ping? Got ping {} vs pong {}",
|
||||
ping.client_time, pong.client_time);
|
||||
return;
|
||||
}
|
||||
|
||||
// when time = send_time+rtt2/2, server time = server time
|
||||
// server time = local time + offset
|
||||
// offset = (server time - local time) = (server time) - (send_time +
|
||||
// rtt2/2)
|
||||
auto rtt2 = pong_local_time - ping.client_time;
|
||||
int64_t serverTimeOffsetUs = pong.server_time - rtt2 / 2 - ping.client_time;
|
||||
|
||||
auto filtered = m_lastOffsets.Calculate(serverTimeOffsetUs);
|
||||
|
||||
// fmt::println("Ping-ponged! RTT2 {} uS, offset {}/filtered offset {} uS",
|
||||
// rtt2,
|
||||
// serverTimeOffsetUs, filtered);
|
||||
|
||||
{
|
||||
std::lock_guard lock{m_offsetMutex};
|
||||
m_metadata.offset = filtered;
|
||||
m_metadata.rtt2 = rtt2;
|
||||
m_metadata.pongsReceived++;
|
||||
m_metadata.lastPongTime = pong_local_time;
|
||||
}
|
||||
|
||||
using std::cout;
|
||||
// fmt::println("Ping-ponged! RTT2 {} uS, offset {} uS", rtt2,
|
||||
// serverTimeOffsetUs);
|
||||
// fmt::println("Estimated server time {} s",
|
||||
// (m_timeProvider() + serverTimeOffsetUs) / 1000000.0);
|
||||
}
|
||||
|
||||
wpi::tsp::TimeSyncClient::TimeSyncClient(std::string_view server,
|
||||
int remote_port,
|
||||
std::chrono::milliseconds ping_delay,
|
||||
std::function<uint64_t()> timeProvider)
|
||||
: m_logger(::ClientLoggerFunc),
|
||||
m_timeProvider(timeProvider),
|
||||
m_udp{wpi::uv::Udp::Create(m_loopRunner.GetLoop(), AF_INET)},
|
||||
m_pingTimer{wpi::uv::Timer::Create(m_loopRunner.GetLoop())},
|
||||
m_serverIP{server},
|
||||
m_serverPort{remote_port},
|
||||
m_loopDelay(ping_delay) {
|
||||
struct sockaddr_in serverAddr;
|
||||
uv::NameToAddr(m_serverIP, m_serverPort, &serverAddr);
|
||||
|
||||
m_loopRunner.ExecSync(
|
||||
[this, serverAddr](uv::Loop&) { m_udp->Connect(serverAddr); });
|
||||
|
||||
// fmt::println("Starting client (with server address {}:{})", server,
|
||||
// remote_port);
|
||||
}
|
||||
|
||||
void wpi::tsp::TimeSyncClient::Start() {
|
||||
// fmt::println("Connecting received");
|
||||
|
||||
m_loopRunner.ExecSync([this](uv::Loop&) {
|
||||
m_udp->received.connect(&wpi::tsp::TimeSyncClient::UdpCallback, this);
|
||||
m_udp->StartRecv();
|
||||
});
|
||||
|
||||
// fmt::println("Starting pinger");
|
||||
using namespace std::chrono_literals;
|
||||
m_pingTimer->timeout.connect(&wpi::tsp::TimeSyncClient::Tick, this);
|
||||
|
||||
m_loopRunner.ExecSync(
|
||||
[this](uv::Loop&) { m_pingTimer->Start(m_loopDelay, m_loopDelay); });
|
||||
}
|
||||
|
||||
void wpi::tsp::TimeSyncClient::Stop() { m_loopRunner.Stop(); }
|
||||
|
||||
int64_t wpi::tsp::TimeSyncClient::GetOffset() {
|
||||
std::lock_guard lock{m_offsetMutex};
|
||||
return m_metadata.offset;
|
||||
}
|
||||
|
||||
wpi::tsp::TimeSyncClient::Metadata wpi::tsp::TimeSyncClient::GetMetadata() {
|
||||
std::lock_guard lock{m_offsetMutex};
|
||||
return m_metadata;
|
||||
}
|
||||
116
photon-targeting/src/main/native/cpp/net/TimeSyncServer.cpp
Normal file
116
photon-targeting/src/main/native/cpp/net/TimeSyncServer.cpp
Normal file
@@ -0,0 +1,116 @@
|
||||
/*
|
||||
* Copyright (C) Photon Vision.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include "net/TimeSyncServer.h"
|
||||
|
||||
#include <fmt/core.h>
|
||||
#include <wpinet/UDPClient.h>
|
||||
#include <wpinet/uv/util.h>
|
||||
|
||||
#include <atomic>
|
||||
#include <chrono>
|
||||
#include <cstdlib>
|
||||
#include <cstring>
|
||||
#include <ctime>
|
||||
#include <iostream>
|
||||
#include <mutex>
|
||||
#include <thread>
|
||||
|
||||
#include <wpi/Logger.h>
|
||||
#include <wpi/struct/Struct.h>
|
||||
|
||||
#include "ntcore_cpp.h"
|
||||
|
||||
static void ServerLoggerFunc(unsigned int level, const char* file,
|
||||
unsigned int line, const char* msg) {
|
||||
if (level == 20) {
|
||||
fmt::print(stderr, "TimeSyncServer: {}\n", msg);
|
||||
return;
|
||||
}
|
||||
|
||||
std::string_view levelmsg;
|
||||
if (level >= 50) {
|
||||
levelmsg = "CRITICAL";
|
||||
} else if (level >= 40) {
|
||||
levelmsg = "ERROR";
|
||||
} else if (level >= 30) {
|
||||
levelmsg = "WARNING";
|
||||
} else {
|
||||
return;
|
||||
}
|
||||
fmt::print(stderr, "TimeSyncServer: {}: {} ({}:{})\n", levelmsg, msg, file,
|
||||
line);
|
||||
}
|
||||
|
||||
void wpi::tsp::TimeSyncServer::UdpCallback(uv::Buffer& data, size_t n,
|
||||
const sockaddr& sender,
|
||||
unsigned flags) {
|
||||
// fmt::println("TimeSyncServer got ping!");
|
||||
|
||||
TspPing ping{wpi::UnpackStruct<TspPing>(data.bytes())};
|
||||
|
||||
if (ping.version != 1) {
|
||||
WPI_ERROR(m_logger, "Bad version from client?");
|
||||
return;
|
||||
}
|
||||
if (ping.message_id != 1) {
|
||||
WPI_ERROR(m_logger, "Bad message id from client?");
|
||||
return;
|
||||
}
|
||||
|
||||
uint64_t current_time = m_timeProvider();
|
||||
|
||||
TspPong pong{ping, current_time};
|
||||
pong.message_id = 2;
|
||||
|
||||
wpi::SmallVector<uint8_t, wpi::Struct<TspPong>::GetSize()> pongData(
|
||||
wpi::Struct<TspPong>::GetSize());
|
||||
wpi::PackStruct(pongData, pong);
|
||||
|
||||
// Wrap our buffer - pongData should free itself for free
|
||||
wpi::uv::Buffer pongBuf{pongData};
|
||||
int sent =
|
||||
m_udp->TrySend(sender, wpi::SmallVector<wpi::uv::Buffer, 1>{pongBuf});
|
||||
// fmt::println("Pong ret: {}", sent);
|
||||
if (static_cast<size_t>(sent) != wpi::Struct<TspPong>::GetSize()) {
|
||||
WPI_ERROR(m_logger, "Didn't send the whole pong back?");
|
||||
return;
|
||||
}
|
||||
|
||||
// WPI_INFO(m_logger, "Got ping: {} {} {}", ping.version, ping.message_id,
|
||||
// ping.client_time);
|
||||
// WPI_INFO(m_logger, "Sent pong: {} {} {} {}", pong.version, pong.message_id,
|
||||
// pong.client_time, pong.server_time);
|
||||
}
|
||||
|
||||
wpi::tsp::TimeSyncServer::TimeSyncServer(int port,
|
||||
std::function<uint64_t()> timeProvider)
|
||||
: m_logger{::ServerLoggerFunc},
|
||||
m_timeProvider{timeProvider},
|
||||
m_udp{wpi::uv::Udp::Create(m_loopRunner.GetLoop(), AF_INET)} {
|
||||
m_loopRunner.ExecSync(
|
||||
[this, port](uv::Loop&) { m_udp->Bind("0.0.0.0", port); });
|
||||
}
|
||||
|
||||
void wpi::tsp::TimeSyncServer::Start() {
|
||||
m_loopRunner.ExecSync([this](uv::Loop&) {
|
||||
m_udp->received.connect(&wpi::tsp::TimeSyncServer::UdpCallback, this);
|
||||
m_udp->StartRecv();
|
||||
});
|
||||
}
|
||||
|
||||
void wpi::tsp::TimeSyncServer::Stop() { m_loopRunner.Stop(); }
|
||||
103
photon-targeting/src/main/native/include/net/TimeSyncClient.h
Normal file
103
photon-targeting/src/main/native/include/net/TimeSyncClient.h
Normal file
@@ -0,0 +1,103 @@
|
||||
/*
|
||||
* Copyright (C) Photon Vision.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <fmt/core.h>
|
||||
#include <wpinet/EventLoopRunner.h>
|
||||
#include <wpinet/UDPClient.h>
|
||||
#include <wpinet/uv/Buffer.h>
|
||||
#include <wpinet/uv/Timer.h>
|
||||
#include <wpinet/uv/Udp.h>
|
||||
|
||||
#include <atomic>
|
||||
#include <chrono>
|
||||
#include <cstdlib>
|
||||
#include <cstring>
|
||||
#include <ctime>
|
||||
#include <functional>
|
||||
#include <iostream>
|
||||
#include <memory>
|
||||
#include <mutex>
|
||||
#include <string>
|
||||
#include <thread>
|
||||
|
||||
#include <frc/filter/MedianFilter.h>
|
||||
#include <wpi/Logger.h>
|
||||
#include <wpi/static_circular_buffer.h>
|
||||
#include <wpi/struct/Struct.h>
|
||||
|
||||
#include "TimeSyncStructs.h"
|
||||
#include "ntcore_cpp.h"
|
||||
|
||||
namespace wpi {
|
||||
namespace tsp {
|
||||
|
||||
class TimeSyncClient {
|
||||
public:
|
||||
struct Metadata {
|
||||
int64_t offset{0};
|
||||
int64_t rtt2{0};
|
||||
size_t pingsSent{0};
|
||||
size_t pongsReceived{0};
|
||||
uint64_t lastPongTime{0};
|
||||
};
|
||||
|
||||
private:
|
||||
using SharedUdpPtr = std::shared_ptr<uv::Udp>;
|
||||
using SharedTimerPtr = std::shared_ptr<uv::Timer>;
|
||||
|
||||
EventLoopRunner m_loopRunner{};
|
||||
|
||||
wpi::Logger m_logger;
|
||||
std::function<uint64_t()> m_timeProvider;
|
||||
|
||||
SharedUdpPtr m_udp;
|
||||
SharedTimerPtr m_pingTimer;
|
||||
|
||||
std::string m_serverIP;
|
||||
int m_serverPort;
|
||||
|
||||
std::chrono::milliseconds m_loopDelay;
|
||||
|
||||
std::mutex m_offsetMutex;
|
||||
Metadata m_metadata;
|
||||
|
||||
// We only allow the most recent ping to stay alive, so only keep track of it
|
||||
TspPing m_lastPing;
|
||||
|
||||
// 30s is a reasonable guess
|
||||
frc::MedianFilter<int64_t> m_lastOffsets{30};
|
||||
|
||||
void Tick();
|
||||
|
||||
void UdpCallback(uv::Buffer& buf, size_t nbytes, const sockaddr& sender,
|
||||
unsigned flags);
|
||||
|
||||
public:
|
||||
TimeSyncClient(std::string_view server, int remote_port,
|
||||
std::chrono::milliseconds ping_delay,
|
||||
std::function<uint64_t()> timeProvider = nt::Now);
|
||||
|
||||
void Start();
|
||||
void Stop();
|
||||
int64_t GetOffset();
|
||||
Metadata GetMetadata();
|
||||
};
|
||||
|
||||
} // namespace tsp
|
||||
} // namespace wpi
|
||||
@@ -0,0 +1,78 @@
|
||||
/*
|
||||
* Copyright (C) Photon Vision.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <fmt/core.h>
|
||||
#include <wpinet/EventLoopRunner.h>
|
||||
#include <wpinet/UDPClient.h>
|
||||
#include <wpinet/uv/Buffer.h>
|
||||
#include <wpinet/uv/Timer.h>
|
||||
#include <wpinet/uv/Udp.h>
|
||||
|
||||
#include <atomic>
|
||||
#include <chrono>
|
||||
#include <cstdlib>
|
||||
#include <cstring>
|
||||
#include <ctime>
|
||||
#include <functional>
|
||||
#include <iostream>
|
||||
#include <memory>
|
||||
#include <mutex>
|
||||
#include <string>
|
||||
#include <thread>
|
||||
|
||||
#include <wpi/Logger.h>
|
||||
#include <wpi/struct/Struct.h>
|
||||
|
||||
#include "TimeSyncStructs.h"
|
||||
#include "ntcore_cpp.h"
|
||||
|
||||
namespace wpi {
|
||||
namespace tsp {
|
||||
|
||||
class TimeSyncServer {
|
||||
using SharedUdpPtr = std::shared_ptr<uv::Udp>;
|
||||
|
||||
EventLoopRunner m_loopRunner{};
|
||||
|
||||
wpi::Logger m_logger;
|
||||
std::function<uint64_t()> m_timeProvider;
|
||||
SharedUdpPtr m_udp;
|
||||
|
||||
std::thread m_listener;
|
||||
|
||||
private:
|
||||
void UdpCallback(uv::Buffer& buf, size_t nbytes, const sockaddr& sender,
|
||||
unsigned flags);
|
||||
|
||||
public:
|
||||
explicit TimeSyncServer(int port = 5810,
|
||||
std::function<uint64_t()> timeProvider = nt::Now);
|
||||
|
||||
/**
|
||||
* Start listening for pings
|
||||
*/
|
||||
void Start();
|
||||
/**
|
||||
* Stop our loop runner. After stopping, we cannot restart.
|
||||
*/
|
||||
void Stop();
|
||||
};
|
||||
|
||||
} // namespace tsp
|
||||
} // namespace wpi
|
||||
@@ -0,0 +1,91 @@
|
||||
/*
|
||||
* Copyright (C) Photon Vision.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
#include <stdint.h>
|
||||
|
||||
#include <wpi/struct/Struct.h>
|
||||
|
||||
namespace wpi {
|
||||
namespace tsp {
|
||||
|
||||
struct TspPing {
|
||||
uint8_t version;
|
||||
uint8_t message_id;
|
||||
uint64_t client_time;
|
||||
};
|
||||
|
||||
struct TspPong : public TspPing {
|
||||
TspPong(TspPing ping, uint64_t servertime)
|
||||
: TspPing{ping}, server_time{servertime} {}
|
||||
uint64_t server_time;
|
||||
};
|
||||
|
||||
} // namespace tsp
|
||||
} // namespace wpi
|
||||
|
||||
template <>
|
||||
struct wpi::Struct<wpi::tsp::TspPing> {
|
||||
static constexpr std::string_view GetTypeName() { return "TspPing"; }
|
||||
static constexpr size_t GetSize() { return 10; }
|
||||
static constexpr std::string_view GetSchema() {
|
||||
return "uint8 version;uint8 message_id;uint64 client_time";
|
||||
}
|
||||
|
||||
static wpi::tsp::TspPing Unpack(std::span<const uint8_t> data) {
|
||||
return wpi::tsp::TspPing{
|
||||
wpi::UnpackStruct<uint8_t, 0>(data),
|
||||
wpi::UnpackStruct<uint8_t, 1>(data),
|
||||
wpi::UnpackStruct<uint64_t, 2>(data),
|
||||
};
|
||||
}
|
||||
static void Pack(std::span<uint8_t> data, const wpi::tsp::TspPing& value) {
|
||||
wpi::PackStruct<0>(data, value.version);
|
||||
wpi::PackStruct<1>(data, value.message_id);
|
||||
wpi::PackStruct<2>(data, value.client_time);
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct wpi::Struct<wpi::tsp::TspPong> {
|
||||
static constexpr std::string_view GetTypeName() { return "TspPong"; }
|
||||
static constexpr size_t GetSize() { return 18; }
|
||||
static constexpr std::string_view GetSchema() {
|
||||
return "uint8 version;uint8 message_id;uint64 client_time;uint64_t "
|
||||
"server_time";
|
||||
}
|
||||
|
||||
static wpi::tsp::TspPong Unpack(std::span<const uint8_t> data) {
|
||||
return wpi::tsp::TspPong{
|
||||
wpi::tsp::TspPing{
|
||||
wpi::UnpackStruct<uint8_t, 0>(data),
|
||||
wpi::UnpackStruct<uint8_t, 1>(data),
|
||||
wpi::UnpackStruct<uint64_t, 2>(data),
|
||||
},
|
||||
wpi::UnpackStruct<uint64_t, 10>(data),
|
||||
};
|
||||
}
|
||||
static void Pack(std::span<uint8_t> data, const wpi::tsp::TspPong& value) {
|
||||
wpi::PackStruct<0>(data, value.version);
|
||||
wpi::PackStruct<1>(data, value.message_id);
|
||||
wpi::PackStruct<2>(data, value.client_time);
|
||||
wpi::PackStruct<10>(data, value.server_time);
|
||||
}
|
||||
};
|
||||
|
||||
static_assert(wpi::StructSerializable<wpi::tsp::TspPong>);
|
||||
static_assert(wpi::StructSerializable<wpi::tsp::TspPing>);
|
||||
168
photon-targeting/src/main/native/jni/TimeSyncClientJNI.cpp
Normal file
168
photon-targeting/src/main/native/jni/TimeSyncClientJNI.cpp
Normal file
@@ -0,0 +1,168 @@
|
||||
/*
|
||||
* Copyright (C) Photon Vision.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <org_photonvision_jni_TimeSyncClient.h>
|
||||
|
||||
#include <cstdio>
|
||||
#include <string>
|
||||
|
||||
#include "net/TimeSyncClient.h"
|
||||
|
||||
using namespace wpi::tsp;
|
||||
|
||||
/**
|
||||
* Finds a class and keeps it as a global reference.
|
||||
*
|
||||
* Use with caution, as the destructor does NOT call DeleteGlobalRef due to
|
||||
* potential shutdown issues with doing so.
|
||||
*/
|
||||
class JClass {
|
||||
public:
|
||||
JClass() = default;
|
||||
|
||||
JClass(JNIEnv* env, const char* name) {
|
||||
jclass local = env->FindClass(name);
|
||||
if (!local) {
|
||||
return;
|
||||
}
|
||||
m_cls = static_cast<jclass>(env->NewGlobalRef(local));
|
||||
env->DeleteLocalRef(local);
|
||||
}
|
||||
|
||||
void free(JNIEnv* env) {
|
||||
if (m_cls) {
|
||||
env->DeleteGlobalRef(m_cls);
|
||||
}
|
||||
m_cls = nullptr;
|
||||
}
|
||||
|
||||
explicit operator bool() const { return m_cls; }
|
||||
|
||||
operator jclass() const { return m_cls; }
|
||||
|
||||
protected:
|
||||
jclass m_cls = nullptr;
|
||||
};
|
||||
|
||||
static JClass metadataClass;
|
||||
static jmethodID metadataCtor;
|
||||
|
||||
// TODO - only one onload allowed
|
||||
JNIEXPORT jint JNICALL JNI_OnLoad(JavaVM* vm, void* reserved) {
|
||||
JNIEnv* env;
|
||||
if (vm->GetEnv(reinterpret_cast<void**>(&env), JNI_VERSION_1_6) != JNI_OK) {
|
||||
return JNI_ERR;
|
||||
}
|
||||
|
||||
metadataClass =
|
||||
JClass(env, "org/photonvision/jni/TimeSyncClient$PingMetadata");
|
||||
|
||||
if (!metadataClass) {
|
||||
std::printf("Couldn't find class!");
|
||||
return JNI_ERR;
|
||||
}
|
||||
|
||||
metadataCtor = env->GetMethodID(metadataClass, "<init>", "(JJJJJ)V");
|
||||
if (!metadataCtor) {
|
||||
std::printf("Couldn't find constructor!");
|
||||
return JNI_ERR;
|
||||
}
|
||||
|
||||
return JNI_VERSION_1_6;
|
||||
}
|
||||
|
||||
extern "C" {
|
||||
|
||||
/*
|
||||
* Class: org_photonvision_jni_TimeSyncClient
|
||||
* Method: create
|
||||
* Signature: (Ljava/lang/String;ID)J
|
||||
*/
|
||||
JNIEXPORT jlong JNICALL
|
||||
Java_org_photonvision_jni_TimeSyncClient_create
|
||||
(JNIEnv* env, jclass, jstring name, jint port, jdouble interval)
|
||||
{
|
||||
using namespace std::chrono_literals;
|
||||
|
||||
const char* c_name{env->GetStringUTFChars(name, 0)};
|
||||
std::string cpp_name{c_name};
|
||||
jlong ret{reinterpret_cast<jlong>(
|
||||
new TimeSyncClient(cpp_name, static_cast<int>(port),
|
||||
std::chrono::duration_cast<std::chrono::milliseconds>(
|
||||
std::chrono::duration<double>(interval))))};
|
||||
env->ReleaseStringUTFChars(name, c_name);
|
||||
return ret;
|
||||
}
|
||||
|
||||
/*
|
||||
* Class: org_photonvision_jni_TimeSyncClient
|
||||
* Method: start
|
||||
* Signature: (J)V
|
||||
*/
|
||||
JNIEXPORT void JNICALL
|
||||
Java_org_photonvision_jni_TimeSyncClient_start
|
||||
(JNIEnv*, jclass, jlong ptr)
|
||||
{
|
||||
TimeSyncClient* client = reinterpret_cast<TimeSyncClient*>(ptr);
|
||||
client->Start();
|
||||
}
|
||||
|
||||
/*
|
||||
* Class: org_photonvision_jni_TimeSyncClient
|
||||
* Method: stop
|
||||
* Signature: (J)V
|
||||
*/
|
||||
JNIEXPORT void JNICALL
|
||||
Java_org_photonvision_jni_TimeSyncClient_stop
|
||||
(JNIEnv*, jclass, jlong ptr)
|
||||
{
|
||||
TimeSyncClient* client = reinterpret_cast<TimeSyncClient*>(ptr);
|
||||
client->Stop();
|
||||
delete client;
|
||||
}
|
||||
|
||||
/*
|
||||
* Class: org_photonvision_jni_TimeSyncClient
|
||||
* Method: getOffset
|
||||
* Signature: (J)J
|
||||
*/
|
||||
JNIEXPORT jlong JNICALL
|
||||
Java_org_photonvision_jni_TimeSyncClient_getOffset
|
||||
(JNIEnv*, jclass, jlong ptr)
|
||||
{
|
||||
TimeSyncClient* client = reinterpret_cast<TimeSyncClient*>(ptr);
|
||||
return client->GetOffset();
|
||||
}
|
||||
|
||||
/*
|
||||
* Class: org_photonvision_jni_TimeSyncClient
|
||||
* Method: getLatestMetadata
|
||||
* Signature: (J)Ljava/lang/Object;
|
||||
*/
|
||||
JNIEXPORT jobject JNICALL
|
||||
Java_org_photonvision_jni_TimeSyncClient_getLatestMetadata
|
||||
(JNIEnv* env, jclass, jlong ptr)
|
||||
{
|
||||
TimeSyncClient* client = reinterpret_cast<TimeSyncClient*>(ptr);
|
||||
auto m{client->GetMetadata()};
|
||||
auto ret = env->NewObject(metadataClass, metadataCtor, m.offset, m.pingsSent,
|
||||
m.pongsReceived, m.lastPongTime, m.rtt2);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
} // extern "C"
|
||||
68
photon-targeting/src/main/native/jni/TimeSyncServerJNI.cpp
Normal file
68
photon-targeting/src/main/native/jni/TimeSyncServerJNI.cpp
Normal file
@@ -0,0 +1,68 @@
|
||||
/*
|
||||
* Copyright (C) Photon Vision.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <org_photonvision_jni_TimeSyncClient.h>
|
||||
#include <org_photonvision_jni_TimeSyncServer.h>
|
||||
|
||||
#include <cstdio>
|
||||
|
||||
#include "net/TimeSyncServer.h"
|
||||
|
||||
using namespace wpi::tsp;
|
||||
|
||||
extern "C" {
|
||||
|
||||
/*
|
||||
* Class: org_photonvision_jni_TimeSyncServer
|
||||
* Method: create
|
||||
* Signature: (I)J
|
||||
*/
|
||||
JNIEXPORT jlong JNICALL
|
||||
Java_org_photonvision_jni_TimeSyncServer_create
|
||||
(JNIEnv*, jclass, jint port)
|
||||
{
|
||||
return reinterpret_cast<jlong>(new TimeSyncServer(port));
|
||||
}
|
||||
|
||||
/*
|
||||
* Class: org_photonvision_jni_TimeSyncServer
|
||||
* Method: start
|
||||
* Signature: (J)V
|
||||
*/
|
||||
JNIEXPORT void JNICALL
|
||||
Java_org_photonvision_jni_TimeSyncServer_start
|
||||
(JNIEnv*, jclass, jlong ptr)
|
||||
{
|
||||
TimeSyncServer* server = reinterpret_cast<TimeSyncServer*>(ptr);
|
||||
server->Start();
|
||||
}
|
||||
|
||||
/*
|
||||
* Class: org_photonvision_jni_TimeSyncServer
|
||||
* Method: stop
|
||||
* Signature: (J)V
|
||||
*/
|
||||
JNIEXPORT void JNICALL
|
||||
Java_org_photonvision_jni_TimeSyncServer_stop
|
||||
(JNIEnv*, jclass, jlong ptr)
|
||||
{
|
||||
TimeSyncServer* server = reinterpret_cast<TimeSyncServer*>(ptr);
|
||||
server->Stop();
|
||||
delete server;
|
||||
}
|
||||
|
||||
} // extern "C"
|
||||
Reference in New Issue
Block a user