chris/lumina-qml
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

making obs scene switching more coherent, but still not functional

Browse source
This commit is contained in:
Chris Cochrun 2024-04-07 06:56:44 -05:00
parent c96f188c70
commit 00fcf565ef
5 changed files with 133 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

59
src/rust/obs.rs
View file

@ -75,31 +75,33 @@ impl Obs {
}
pub fn set_scene(
self,
&self,
scene: String,
) -> Result<(), Box<dyn Error>> {
debug!("Starting function");
if let Some(client) = self.client {
debug!("Starting to set");
let runtime =
tokio::runtime::Builder::new_current_thread()
.thread_keep_alive(Duration::from_secs(1))
.enable_all()
.build()
.unwrap();
// debug!("Starting function");
if let Some(client) = &self.client {
debug!(scene, "setting scene in obs");
let client = make_client();
let handle = runtime.spawn(async move {
debug!(scene, "working in thread");
client.scenes()
.set_current_program_scene(&scene)
.await
});
loop {
sleep(Duration::from_millis(100));
if handle.is_finished() {
break;
let runtime = tokio::runtime::Runtime::new()?;
let handle = runtime.spawn(async move {
debug!("in spawn: before setting");
let res = client.scenes().set_current_program_scene(&scene).await;
match res {
Ok(o) => debug!("in spawn: after setting: success"),
Err(e) => error!(?e, "in spawn: after setting")
}
}
});
debug!("after spawn");
// loop {
// sleep(Duration::from_millis(100));
// if handle.is_finished() {
// break;
// }
// }
Ok(())
} else {
Err("There is no client".to_owned())?
@ -180,13 +182,13 @@ impl obs_model::ObsModel {
tokio::runtime::Runtime::new().unwrap().block_on(async {
match Obs::new().await {
Ok(o) => {
if let Some(scene) = &o.current_program_scene {
let scene = QString::from(scene);
self.as_mut().set_current_program_scene(scene);
}
self.as_mut().set_connected(true);
self.as_mut().rust_mut().obs = Some(o);
self.as_mut().update_scenes();
if let Some(scene) = o.current_program_scene {
let scene = QString::from(&scene);
self.as_mut().set_current_program_scene(scene);
}
}
Err(e) => {
error!(e);
@ -220,6 +222,11 @@ mod test {
#[test]
pub fn test_obs_setting_scene() {
assert_eq!(true, true)
let runtime = tokio::runtime::Runtime::new().unwrap();
let future = Client::connect("localhost", 4455, Some(""));
let client = runtime.block_on(future).unwrap();
let scene = String::from("me");
let res = runtime.block_on(client.scenes().set_current_program_scene(&scene));
debug_assert!(res.is_ok());
}
}

76
src/rust/slide_model.rs
View file

@ -47,6 +47,7 @@ mod slide_model {
VideoStartTime,
VideoEndTime,
Html,
ObsScene,
}
unsafe extern "RustQt" {
@ -123,6 +124,13 @@ mod slide_model {
#[qinvokable]
fn activate(self: Pin<&mut SlideModel>, index: i32) -> bool;
#[qinvokable]
fn update_obs_scene(
self: Pin<&mut SlideModel>,
index: i32,
obs_scene: QString,
);
}
impl cxx_qt::Threading for SlideModel {}
@ -271,7 +279,7 @@ impl Default for Slide {
}
}
#[derive(Default, Debug)]
#[derive(Debug)]
pub struct SlideModelRust {
id: i32,
slides: Vec<Slide>,
@ -279,6 +287,28 @@ pub struct SlideModelRust {
count: i32,
}
impl Default for SlideModelRust {
fn default() -> Self {
let obs =
tokio::runtime::Runtime::new().unwrap().block_on(async {
match Obs::new().await {
Ok(o) => Some(o),
Err(e) => {
error!(e);
None
}
}
});
Self {
id: 0,
slides: Vec::new(),
obs,
count: 0,
}
}
}
impl slide_model::SlideModel {
pub fn add_video_thumbnail(
mut self: Pin<&mut Self>,
@ -1154,25 +1184,26 @@ impl slide_model::SlideModel {
slide.active = false;
}
let obs = self.as_mut().obs.clone();
if let Some(slide) =
self.as_mut().rust_mut().slides.get_mut(index as usize)
{
debug!(
slide = index,
service_item = slide.service_item_id,
obs = slide.obs_scene.to_string(),
"This slide is activating"
);
let obs_scene = slide.obs_scene.to_string();
slide.active = true;
self.as_mut().data_changed(tl, br, &vector_roles);
if let Some(obs) = obs {
if let Some(obs) = self.as_ref().obs.clone() {
match obs.set_scene(obs_scene) {
Ok(()) => debug!("Successfully set scene"),
Err(e) => error!(e),
}
} else {
debug!("Obs isn't connected");
}
// We use this signal generated by our signals enum to tell QML that
@ -1184,6 +1215,24 @@ impl slide_model::SlideModel {
}
}
pub fn update_obs_scene(
mut self: Pin<&mut Self>,
index: i32,
obs_scene: QString,
) {
let rc = self.as_ref().count() - 1;
let tl = &self.as_ref().index(0, 0, &QModelIndex::default());
let br = &self.as_ref().index(rc, 0, &QModelIndex::default());
let mut vector_roles = QVector_i32::default();
vector_roles.append(self.get_role(SlideRoles::ObsScene));
if let Some(slide) =
self.as_mut().rust_mut().slides.get_mut(index as usize)
{
slide.obs_scene = obs_scene;
self.as_mut().data_changed(tl, br, &vector_roles);
}
}
fn get_role(&self, role: SlideRoles) -> i32 {
match role {
SlideRoles::Text => 1,
@ -1193,6 +1242,8 @@ impl slide_model::SlideModel {
SlideRoles::VideoThumbnail => 15,
SlideRoles::VideoStartTime => 16,
SlideRoles::VideoEndTime => 17,
SlideRoles::Html => 18,
SlideRoles::ObsScene => 19,
_ => 0,
}
}
@ -1247,6 +1298,9 @@ impl slide_model::SlideModel {
QVariant::from(&slide.video_end_time)
}
SlideRoles::Html => QVariant::from(&slide.html),
SlideRoles::ObsScene => {
QVariant::from(&slide.obs_scene)
}
_ => QVariant::default(),
};
}
@ -1325,6 +1379,10 @@ impl slide_model::SlideModel {
QByteArray::from("videoEndTime"),
);
roles.insert(SlideRoles::Html.repr, QByteArray::from("html"));
roles.insert(
SlideRoles::ObsScene.repr,
QByteArray::from("obsScene"),
);
roles
}
@ -1334,3 +1392,13 @@ impl slide_model::SlideModel {
cnt
}
}
#[cfg(test)]
mod test {
use super::*;
#[test]
pub fn test_obs_setting_scene() {
assert_eq!(true, true)
}
}