bstein/lesavka
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Video Fix & Script Updates

Browse Source
This commit is contained in:
Brad Stein 2025-06-27 22:51:50 -05:00
parent 43808f95d6
commit 8da4f36dc3
9 changed files with 81 additions and 74 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
client/src/app.rs
View File

@ -52,8 +52,8 @@ impl LesavkaClientApp {
.connect_lazy();
let vid_ep = Channel::from_shared(self.server_addr.clone())?
.initial_connection_window_size(2<<20)
.initial_stream_window_size(2<<20)
.initial_connection_window_size(4<<20)
.initial_stream_window_size(4<<20)
.tcp_nodelay(true)
.connect_lazy();

31
client/src/output/video.rs
View File

@ -25,14 +25,15 @@ impl MonitorWindow {
pub fn new(id: u32, el: &EventLoop<()>) -> anyhow::Result<Self> {
gst::init()?; // idempotent
/*────────────────────── window ──────────────────────*/
let window = el.create_window(
/* ---------- Wayland / X11 window ------------- */
let window = el
.create_window(
WindowAttributes::default()
.with_title(format!("Lesavka‑monitor‑{id}"))
.with_decorations(false),
.with_decorations(true),
)?;
/*────────────────────── pipeline ────────────────────*/
/* ---------- GStreamer pipeline --------------- */
let caps = gst::Caps::builder("video/x-h264")
.field("stream-format", &"byte-stream")
.field("alignment", &"au")
@ -44,27 +45,29 @@ impl MonitorWindow {
let src = pipeline
.by_name("src")
.expect("appsink")
.expect("appsrc element not found")
.downcast::<gst_app::AppSrc>()
.expect("appsink down‑cast");
.expect("appsrc down‑cast");
src.set_caps(Some(&caps));
src.set_format(gst::Format::Time); // running‑time PTS
src.set_property("blocksize", &0u32); // whole AU per buffer
src.set_latency(gst::ClockTime::NONE, gst::ClockTime::NONE);
src.set_format(gst::Format::Time); // downstream clock
src.set_property("blocksize", &0u32); // one AU per buffer
// NOTE: set_property() and friends return (), so no `?`
src.set_property("do-timestamp", &true);
src.set_latency(gst::ClockTime::NONE, gst::ClockTime::NONE);
pipeline.set_state(gst::State::Playing)?;
Ok(Self { id, _window: window, src })
}
/// Push one encoded access‑unit into the local pipeline.
/// Feed one H.264 access‑unit into the pipeline.
pub fn push_packet(&self, pkt: VideoPacket) {
let buf = gst::Buffer::from_slice(pkt.data); // no PTS manipulation
if let Some(mut b) = buf.get_mut() {
// Mutable so we can set the PTS:
let mut buf = gst::Buffer::from_slice(pkt.data);
if let Some(ref mut b) = buf.get_mut() {
b.set_pts(Some(gst::ClockTime::from_useconds(pkt.pts)));
}
let _ = self.src.push_buffer(buf);
let _ = self.src.push_buffer(buf); // ignore Eos / flushing
}
}

0
scripts/lesavka-core.sh → scripts/daemon/lesavka-core.sh
View File

2
scripts/install-client.sh → scripts/install/client.sh
View File

@ -1,5 +1,5 @@
#!/usr/bin/env bash
# install-client.sh - install and setup all client related apps and environments
# scripts/install/client.sh - install and setup all client related apps and environments
set -euo pipefail
ORIG_USER=${SUDO_USER:-$(id -un)}

8
scripts/install-server.sh → scripts/install/server.sh
View File

@ -1,5 +1,5 @@
#!/usr/bin/env bash
# install-server.sh - install and setup all server related apps and environments
# scripts/install/server.sh - install and setup all server related apps and environments
set -euo pipefail
ORIG_USER=${SUDO_USER:-$(id -un)}
@ -45,7 +45,7 @@ LEFT_TAG=${TAGS[0]}
RIGHT_TAG=${TAGS[1]}
sudo tee /etc/udev/rules.d/85-gc311.rules >/dev/null <<EOF
# auto-generated by lesavka/scripts/install-server.sh - DO NOT EDIT
# auto-generated by lesavka/scripts/daemon/install-server.sh - DO NOT EDIT
SUBSYSTEM=="video4linux", ENV{ID_PATH_TAG}=="$LEFT_TAG", SYMLINK+="lesavka_l_eye"
SUBSYSTEM=="video4linux", ENV{ID_PATH_TAG}=="$RIGHT_TAG", SYMLINK+="lesavka_r_eye"
EOF
@ -76,7 +76,7 @@ sudo -u "$ORIG_USER" bash -c "cd '$SRC_DIR/server' && cargo clean && cargo build
echo "==> 5. Install binaries"
sudo install -Dm755 "$SRC_DIR/server/target/release/lesavka-server" /usr/local/bin/lesavka-server
sudo install -Dm755 "$SRC_DIR/scripts/lesavka-core.sh" /usr/local/bin/lesavka-core.sh
sudo install -Dm755 "$SRC_DIR/scripts/daemon/lesavka-core.sh" /usr/local/bin/lesavka-core.sh
echo "==> 6a. Systemd units - lesavka-core"
cat <<'UNIT' | sudo tee /etc/systemd/system/lesavka-core.service >/dev/null
@ -105,7 +105,7 @@ After=network.target lesavka-core.service
[Service]
ExecStart=/usr/local/bin/lesavka-server
Restart=always
Environment=RUST_LOG=lesavka_server=debug,lesavka_server::usb_gadget=info
Environment=RUST_LOG=lesavka_server=debug,lesavka_server::video=trace,lesavka_server::usb_gadget=info
Environment=RUST_BACKTRACE=1
Restart=always
RestartSec=5

7
scripts/manual/usb-reset.sh Normal file
View File

@ -0,0 +1,7 @@
grpcurl \
-plaintext \
-import-path ./../../common/proto \
-proto lesavka.proto \
-d '{}' \
192.168.42.253:50051 \
lesavka.Relay/ResetUsb

8
scripts/manual/video-check.sh Normal file
View File

@ -0,0 +1,8 @@
grpcurl -plaintext \
-d '{"id":0,"max_bitrate":6000}' \
-import-path ./../../common/proto -proto lesavka.proto \
192.168.42.253:50051 \
lesavka.relay.Relay/CaptureVideo \
| jq -r '.data'
| base64 -d \
| gst-launch-1.0 fdsrc ! h264parse ! avdec_h264 ! autovideosink

4
server/src/main.rs
View File

@ -159,7 +159,7 @@ impl Relay for Handler {
_ => return Err(Status::invalid_argument("monitor id must be 0 or 1")),
};
info!("🎥 streaming {dev}");
let s = video::spawn_camera(dev, id, 6_000)
let s = video::eye_ball(dev, id, 6_000)
.await
.map_err(|e| Status::internal(format!("{e:#}")))?;
Ok(Response::new(Box::pin(s)))
@ -197,7 +197,7 @@ async fn main() -> anyhow::Result<()> {
info!("🌐 lesavka‑server listening on 0.0.0.0:50051");
Server::builder()
.tcp_nodelay(true)
.max_frame_size(Some(256*1024))
.max_frame_size(Some(2*1024*1024))
.add_service(RelayServer::new(handler))
.add_service(ReflBuilder::configure().build_v1().unwrap())
.serve(([0,0,0,0], 50051).into())

85
server/src/video.rs
View File

@ -8,18 +8,17 @@ use gst::log;
use lesavka_common::lesavka::VideoPacket;
use tokio_stream::wrappers::ReceiverStream;
use tonic::Status;
use tracing::{debug, enabled, Level};
use tracing::{debug, enabled, trace, Level};
static START: std::sync::OnceLock<gst::ClockTime> = std::sync::OnceLock::new();
pub async fn spawn_camera(
pub async fn eye_ball(
dev: &str,
id: u32,
_max_bitrate_kbit: u32,
) -> anyhow::Result<ReceiverStream<Result<VideoPacket, Status>>> {
gst::init().context("gst init")?;
// IMPORTANT: keep one AU per buffer, include regular SPS/PPS
let desc = format!(
"v4l2src device={dev} io-mode=mmap ! \
video/x-h264,stream-format=byte-stream,alignment=au ! \
@ -29,80 +28,70 @@ pub async fn spawn_camera(
let pipeline = gst::parse::launch(&desc)?
.downcast::<gst::Pipeline>()
.expect("pipeline down-cast");
.expect("pipeline down‑cast");
let sink = pipeline
.by_name("sink")
.expect("appsink")
.dynamic_cast::<gst_app::AppSink>()
.expect("appsink downcast");
.expect("appsink down‑cast");
let (tx, rx) = tokio::sync::mpsc::channel(256);
sink.set_callbacks(
gst_app::AppSinkCallbacks::builder()
// .new_sample(move |sink| {
// let sample = sink.pull_sample().map_err(|_| gst::FlowError::Eos)?;
// let buffer = sample.buffer().ok_or(gst::FlowError::Error)?;
// let map = buffer.map_readable().map_err(|_| gst::FlowError::Error)?;
// let pkt = VideoPacket {
// id,
// pts: buffer.pts().nseconds() / 1_000, // → µs
// data: map.as_slice().to_vec(),
// };
// // ignore back‑pressure: drop oldest if channel is full
// let _ = tx.try_send(Ok(pkt));
// Ok(gst::FlowSuccess::Ok)
// })
// .build(),
.new_sample(move |sink| {
/* -------- pull frame ---------- */
let sample = sink.pull_sample().map_err(|_| gst::FlowError::Eos)?;
let buffer = sample.buffer().ok_or(gst::FlowError::Error)?;
let origin = *START.get_or_init(|| buffer.pts().unwrap_or(gst::ClockTime::ZERO));
/* -------- basic counters ------ */
static FRAME: std::sync::atomic::AtomicU64 =
std::sync::atomic::AtomicU64::new(0);
let n = FRAME.fetch_add(1, std::sync::atomic::Ordering::Relaxed);
if n % 120 == 0 {
trace!("eye{id}: delivered {n} frames");
}
/* -------- map once, reuse ----- */
let map = buffer.map_readable().map_err(|_| gst::FlowError::Error)?;
// write first IDR to disk (quick sanity check)
if n == 0 {
std::fs::write(format!("/tmp/eye{id}-idr.h264"), map.as_slice()).ok();
}
/* -------- detect SPS / IDR ---- */
if enabled!(Level::DEBUG) {
if let Some(&nal) = map.as_slice().get(4) {
if (nal & 0x1F) == 0x05 /* IDR */ {
debug!("eye{id}: IDR");
}
}
}
/* -------- timestamps ---------- */
let origin = *START.get_or_init(|| buffer.pts().unwrap_or(gst::ClockTime::ZERO));
let pts_us = buffer
.pts()
.unwrap_or(gst::ClockTime::ZERO)
.saturating_sub(origin)
.nseconds() / 1_000;
let map = buffer.map_readable().map_err(|_| gst::FlowError::Error)?;
if enabled!(Level::DEBUG) {
if let Some(slice) = map.as_slice().get(0..5) {
match slice[4] & 0b1_1111 {
// 0x07 = SPS, 0x05 = IDR (Annex-B “byte-stream”)
0x07 | 0x05 => {
debug!(
"🎞️ monitor {id}: got NAL {:02X}, {} bytes",
slice[4],
map.as_slice().len()
);
}
_ => {}
}
}
}
.nseconds()
/ 1_000;
/* -------- ship over gRPC ----- */
let pkt = VideoPacket {
id,
pts: pts_us,
data: map.as_slice().to_vec(),
};
let _ = tx.try_send(Ok(pkt)); // drop on overflow
let _ = tx.try_send(Ok(pkt));
Ok(gst::FlowSuccess::Ok)
})
.build(),
);
// gst::debug_remove_default_log_function();
// gst::debug_add_default_log_function(|lvl, cat, msg| {
// println!("[GST] {lvl:?} {cat}: {msg}");
// });
// std::env::set_var("GST_DEBUG", "v4l2src:4,h264parse:3");
pipeline.set_state(gst::State::Playing)?;
Ok(ReceiverStream::new(rx))
}