TheLanc3/rustdesk
1
0
Fork 0
mirror of https://github.com/rustdesk/rustdesk.git synced 2026-04-01 20:51:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

mobile & web rgba stream

Browse Source
This commit is contained in:
csf
2022-05-19 23:45:44 +08:00
parent 9ecacadd4a
commit 7c5a136b6b
6 changed files with 76 additions and 130 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/mobile.rs
View File

@@ -1,5 +1,5 @@
use crate::client::*;
use flutter_rust_bridge::StreamSink;
use flutter_rust_bridge::{StreamSink, ZeroCopyBuffer};
use hbb_common::{
allow_err,
compress::decompress,
@@ -24,7 +24,8 @@ use std::{
lazy_static::lazy_static! {
static ref SESSION: Arc<RwLock<Option<Session>>> = Default::default();
pub static ref EVENT_STREAM: RwLock<Option<StreamSink<String>>> = Default::default(); // rust to dart channel
pub static ref EVENT_STREAM: RwLock<Option<StreamSink<String>>> = Default::default(); // rust to dart event channel
pub static ref RGBA_STREAM: RwLock<Option<StreamSink<ZeroCopyBuffer<Vec<u8>>>>> = Default::default(); // rust to dart rgba (big u8 list) channel
}
#[derive(Clone, Default)]
@@ -33,7 +34,6 @@ pub struct Session {
sender: Arc<RwLock<Option<mpsc::UnboundedSender<Data>>>>,
lc: Arc<RwLock<LoginConfigHandler>>,
events2ui: Arc<RwLock<VecDeque<String>>>,
rgba: Arc<RwLock<Option<Vec<u8>>>>,
}
impl Session {
@@ -89,14 +89,6 @@ impl Session {
}
}
pub fn rgba() -> Option<Vec<u8>> {
if let Some(session) = SESSION.read().unwrap().as_ref() {
session.rgba.write().unwrap().take()
} else {
None
}
}
pub fn pop_event() -> Option<String> {
if let Some(session) = SESSION.read().unwrap().as_ref() {
session.events2ui.write().unwrap().pop_front()
@@ -607,8 +599,11 @@ impl Connection {
if !self.first_frame {
self.first_frame = true;
}
if let Ok(true) = self.video_handler.handle_frame(vf) {
*self.session.rgba.write().unwrap() = Some(self.video_handler.rgb.clone());
if let (Ok(true), Some(s)) = (
self.video_handler.handle_frame(vf),
RGBA_STREAM.read().unwrap().as_ref(),
) {
s.add(ZeroCopyBuffer(self.video_handler.rgb.clone()));
}
}
Some(message::Union::hash(hash)) => {