lock stdout

Author: Jeremy5909

Diff for: src/camera.rs

@@ -1,4 +1,4 @@
-use std::{f64::INFINITY, io::stdout};
+use std::{f64::INFINITY, io::{self, stdout, BufWriter}};
 
 use rand::{random, thread_rng, Rng};
 
@@ -48,6 +48,7 @@ impl Camera {
 	}
 	pub fn render(&mut self, world: &dyn Hittable) {
 		self.initialize();
+        let mut writer = BufWriter::new(io::stdout());
 
 		println!("P3\n{} {}\n255", self.image_width, self.image_height);
     	for j in 0..=self.image_height {
@@ -58,7 +59,7 @@ impl Camera {
 					let r = self.get_ray(i, j);
 					pixel_color += Self::ray_color(&r, self.max_depth, world);
 				}
-            	write_color(&mut stdout(), self.pixel_samples_scale * pixel_color);
+            	write_color(&mut writer, self.pixel_samples_scale * pixel_color);
         }
     }
     eprintln!("\rDone.                 \n");

Diff for: src/color.rs

@@ -1,27 +1,29 @@
-use std::io::{Stdout, Write};
+use std::io::Write;
 
 use crate::{interval::Interval, vec3::Vec3};
 
 pub type Color = Vec3;
 
-pub fn write_color(out: &mut Stdout, pixel_color: Color) {
-	let r = pixel_color.x();
-	let g = pixel_color.y();
-	let b = pixel_color.z();
+pub fn write_color<W: Write>(mut out: W, pixel_color: Color) {
+    let r = pixel_color.x();
+    let g = pixel_color.y();
+    let b = pixel_color.z();
 
-	let r = linear_to_gamma(r);
-	let g = linear_to_gamma(g);
-	let b = linear_to_gamma(b);
+    let r = linear_to_gamma(r);
+    let g = linear_to_gamma(g);
+    let b = linear_to_gamma(b);
 
-	let intensity = Interval::new(0.0, 0.999);
-	let rbyte = (256.0 * intensity.clamp(r)) as i32;
-	let gbyte = (256.0 * intensity.clamp(g)) as i32;
-	let bbyte = (256.0 * intensity.clamp(b)) as i32;
-	out.write_all(format!("{rbyte} {gbyte} {bbyte}\n").as_bytes()).unwrap();
+    let intensity = Interval::new(0.0, 0.999);
+    let rbyte = (256.0 * intensity.clamp(r)) as i32;
+    let gbyte = (256.0 * intensity.clamp(g)) as i32;
+    let bbyte = (256.0 * intensity.clamp(b)) as i32;
+    out.write_all(format!("{rbyte} {gbyte} {bbyte}\n").as_bytes())
+        .unwrap();
 }
 pub fn linear_to_gamma(linear_component: f64) -> f64 {
-	if linear_component > 0.0 {
-		return linear_component.sqrt()
-	}
-	0.0
-}
+    if linear_component > 0.0 {
+        return linear_component.sqrt();
+    }
+    0.0
+}
+