From 5cb23f06076cd281549465cac8ac383351a3c718 Mon Sep 17 00:00:00 2001
From: Brooks Palin <brookspalin710@gmail.com>
Date: Thu, 29 Feb 2024 22:20:32 -0700
Subject: [PATCH] Headless mode Orbit (#66)

---
 studio/src/main-headless.rs | 8 ++++----
 studio/src/screenshot.rs    | 4 ++--
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/studio/src/main-headless.rs b/studio/src/main-headless.rs
index c193025..18bdd76 100644
--- a/studio/src/main-headless.rs
+++ b/studio/src/main-headless.rs
@@ -97,7 +97,7 @@ fn main() {
             );
 
             if orbit {
-                app.insert_resource(OrbitSettings { radius: 10.0 })
+                app.insert_resource(OrbitSettings { radius: 5.0 })
                     .add_systems(Update, orbit_scene);
             }
 
@@ -164,7 +164,7 @@ struct OrbitSettings {
 }
 
 fn orbit_scene(
-    mut camera: Query<&mut Transform, With<PrimaryWindow>>,
+    mut camera: Query<&mut Transform, With<Camera>>,
     orbit_settings: Res<OrbitSettings>,
     time: Res<Time>,
 ) {
@@ -173,9 +173,9 @@ fn orbit_scene(
     };
 
     main_cam_transform.translation = Vec3::new(
-        orbit_settings.radius * time.elapsed_seconds().cos() * 10.0,
-        orbit_settings.radius * time.elapsed_seconds().sin(),
+        orbit_settings.radius * time.elapsed_seconds().cos(),
         6.0,
+        orbit_settings.radius * time.elapsed_seconds().sin(),
     );
     main_cam_transform.look_at(Vec3::ZERO, Vec3::Y);
 }
diff --git a/studio/src/screenshot.rs b/studio/src/screenshot.rs
index c8d0b1c..849db65 100644
--- a/studio/src/screenshot.rs
+++ b/studio/src/screenshot.rs
@@ -147,7 +147,7 @@ fn extract_screen_capture_components(
 ) {
     for (entity, screen_capture) in screen_capure_comps.iter() {
         if screen_capture.should_capture() {
-            println!(
+            info!(
                 "Capturing Frame: {} / {}",
                 screen_capture.frames, screen_capture.max_frames
             );
@@ -275,7 +275,7 @@ fn save_buffer_to_file(
             .expect("Output path could not be created");
 
         let path = format!(
-            "{}img_{:0.5}.png",
+            "{}img_{:0>4}.png",
             screen_capture.to_directory, screen_capture.frames
         );