Fixed capture

src/programs/viewer.rs

@@ -85,6 +85,7 @@ fn main() {
 
     let mut events_loop = EventsLoop::new();
     let window = WindowBuilder::new().with_visibility(false);
+
     let context = glutin::ContextBuilder::new().with_depth_buffer(24);
     let display = Display::new(window, context, &events_loop).unwrap();
     let mut renderer = Renderer::new(display);
@@ -142,6 +143,8 @@ fn main() {
 
     while !closed {
 
+        let mut should_screenshot = false;
+
         events_loop.poll_events(|ev| {
 
             controls.manage_event(&ev, &mut camera, &renderer);
@@ -191,11 +194,9 @@ fn main() {
 
                 } => {
 
-                    // Make a screenshot
-                    let size = renderer.gl_window().get_inner_size().unwrap();
-                    let cap = renderer.capture(size.into());
-                    cap.save(format!("capture-{}.png", capture_count)).unwrap();
-                    capture_count += 1;
+                    should_screenshot = true;
+
+
 
                 },
 
@@ -205,6 +206,14 @@ fn main() {
 
         controls.update(&mut camera, &renderer);
         renderer.render(&scene, &camera);
+
+        if should_screenshot {
+            // Make a screenshot
+            let cap = renderer.capture();
+            cap.save(format!("capture-{}.png", capture_count)).unwrap();
+            capture_count += 1;
+        }
+
         let elapsed = as_millis(Instant::now().duration_since(before));
 
         if elapsed < 20 {