Headless Mode fixes for the Backend (#65)

This PR brings some fixes that are needed for the backend to render
scenes.

studio/src/headless.rs

@@ -9,6 +9,7 @@ use wgpu::Extent3d;
 
 use crate::{
     bvh::{BvhAnimCompleteEvent, TestBvh},
+    scene::SceneState,
     screenshot::{ScreenCapture, ScreenShotPlugin},
     BvhPlugin, CommonPlugins, LoadLibraryObjectEvent, LoadSceneEvent, MainCamera,
     MixamoImportEvent, MixamoPlugin, SkyboxId, StudioMode, WorldGridPlugin,
@@ -64,7 +65,10 @@ pub fn start_headless(
                 },
             );
 
-            app.add_systems(PostUpdate, check_for_bvh_capture_complete);
+            app.add_systems(
+                PostUpdate,
+                check_for_bvh_capture_complete.run_if(in_state(SceneState::Active)),
+            );
         }
         HeadlessMode::Scene(scene, skybox) => {
             app.add_systems(
@@ -87,7 +91,10 @@ pub fn start_headless(
                     commands.insert_resource(SkyboxId(skybox.clone()));
                 },
             );
-            app.add_systems(PostUpdate, check_for_mixamo_capture_complete);
+            app.add_systems(
+                PostUpdate,
+                check_for_mixamo_capture_complete.run_if(in_state(SceneState::Active)),
+            );
 
             app.run();
         }
@@ -135,7 +142,6 @@ fn check_for_bvh_capture_complete(
         return;
     };
 
-    println!("Frames captured {}", screen_capture.frames_captured());
     if !screen_capture.should_capture() || !bvh_event_reader.is_empty() {
         screen_capture.stop_capturing();
         close_app.send(AppExit);

studio/src/lib.rs

@@ -7,6 +7,7 @@ use scene::ImportSceneEvent;
 
 #[cfg(feature = "wasm")]
 use serde_repr::{Deserialize_repr, Serialize_repr};
+use space_editor::space_prefab::plugins::PrefabPlugin;
 #[cfg(feature = "wasm")]
 use wasm_bindgen::prelude::*;
 

studio/src/main-headless.rs

@@ -1,4 +1,7 @@
-use bevy::render::render_resource::Extent3d;
+use bevy::{
+    prelude::*, render::render_resource::Extent3d, transform::components::Transform,
+    window::PrimaryWindow,
+};
 use clap::{Args, Parser, Subcommand};
 use studio::headless::{start_headless, HeadlessMode};
 
@@ -19,6 +22,8 @@ struct StudioArgs {
 enum Subcommands {
     Scene {
         scene_dir: String,
+        #[arg(long)]
+        orbit: bool,
         #[arg(long, short)]
         output_dir: Option<String>,
         #[arg(long)]
@@ -67,6 +72,7 @@ fn main() {
     let mut app = match args.subcommand {
         Subcommands::Scene {
             scene_dir,
+            orbit,
             skybox,
             output_dir,
             output_resolution,
@@ -82,13 +88,20 @@ fn main() {
             let fps = fps.unwrap_or(24);
             let frames = frames.unwrap_or(30);
             let skybox = skybox.unwrap_or("gum_trees_4k".to_string());
-            start_headless(
+            let mut app = start_headless(
                 HeadlessMode::Scene(scene_dir, skybox),
                 fps,
                 &output_dir,
                 frames,
                 output_res,
-            )
+            );
+
+            if orbit {
+                app.insert_resource(OrbitSettings { radius: 10.0 })
+                    .add_systems(Update, orbit_scene);
+            }
+
+            app
         }
         Subcommands::Bvh {
             bvh_dir,
@@ -145,6 +158,28 @@ fn main() {
     app.run();
 }
 
+#[derive(Resource)]
+struct OrbitSettings {
+    radius: f32,
+}
+
+fn orbit_scene(
+    mut camera: Query<&mut Transform, With<PrimaryWindow>>,
+    orbit_settings: Res<OrbitSettings>,
+    time: Res<Time>,
+) {
+    let Ok(mut main_cam_transform) = camera.get_single_mut() else {
+        return;
+    };
+
+    main_cam_transform.translation = Vec3::new(
+        orbit_settings.radius * time.elapsed_seconds().cos() * 10.0,
+        orbit_settings.radius * time.elapsed_seconds().sin(),
+        6.0,
+    );
+    main_cam_transform.look_at(Vec3::ZERO, Vec3::Y);
+}
+
 //==============================================================================
 //                        Parsers
 //==============================================================================

studio/src/scene.rs

@@ -26,7 +26,7 @@ use crate::{
     camera_controller::{CameraController, CameraControllerBundle},
     change_tracking::{ChangeTracker, ChangeTrackingPlugin},
     debug::DebugPoint,
-    headless::{is_headless, is_not_headless, HeadlessSettings},
+    headless::{self, is_headless, is_not_headless, HeadlessSettings},
     math::{Combinable, Extrema},
     screenshot::ScreenCapture,
     MainCamera,
@@ -326,7 +326,8 @@ fn initialiaze_headless(
             ..default()
         },
         CameraControllerBundle::new(focus, camera_xform.translation),
-        ScreenCapture::from_out(&headless_settings.output_dir),
+        ScreenCapture::from_out(&headless_settings.output_dir)
+            .with_max_frames(headless_settings.max_frames),
     ));
 }
 

studio/src/screenshot.rs

@@ -1,4 +1,5 @@
 use bevy::{
+    app::AppExit,
     core_pipeline::{core_3d, prepass::ViewPrepassTextures},
     prelude::*,
     render::{
@@ -123,9 +124,16 @@ impl ScreenCapture {
     }
 }
 
-fn update_screen_capture_on_extract_step(mut screen_captures: Query<&mut ScreenCapture>) {
+fn update_screen_capture_on_extract_step(
+    mut screen_captures: Query<&mut ScreenCapture>,
+    mut exit_event: EventWriter<AppExit>,
+) {
     for mut screen_capture in screen_captures.iter_mut() {
         screen_capture.frames += 1;
+
+        if !screen_capture.should_capture() {
+            exit_event.send(AppExit);
+        }
         // if screen_capture.number_of_frames == 0 {
         //     commands.entity(entity).remove::<ScreenCapture>();
         // }
@@ -139,6 +147,10 @@ fn extract_screen_capture_components(
 ) {
     for (entity, screen_capture) in screen_capure_comps.iter() {
         if screen_capture.should_capture() {
+            println!(
+                "Capturing Frame: {} / {}",
+                screen_capture.frames, screen_capture.max_frames
+            );
             commands.get_or_spawn(entity).insert(screen_capture.clone());
         }
     }