feat - Antialliasing and camera

1 files changed, 12 insertions, 3 deletions

diff --git a/main.go b/main.go
index daf2718..f85ad8a 100644
--- a/main.go
+++ b/main.go
@@ -1,15 +1,19 @@
 package main
 
+import "math/rand"
+
 const HEIGHT = 100
 const WIDTH = 200
 const T_MIN = 0.001
 const T_MAX = 1000
+const RAY_PER_PIXEL = 100
 
 func main() {
 	LOWER_LEFT_CORNER := Vec3D{-2.0, -1.0, -1.0}
 	ORIGIN := Vec3D{0.0, 0.0, 0.0}
 	HORIZONTAL := Vec3D{4.0, 0.0, 0.0}
 	VERTICAL := Vec3D{0.0, 2.0, 0.0}
+	CAMERA := Camera{LOWER_LEFT_CORNER, ORIGIN, HORIZONTAL, VERTICAL}
 
 	SPHERE := Sphere{Vec3D{0.0, 0.0, -1.0}, 0.5}
 	EARTH := Sphere{Vec3D{0.0, -100.5, -1.0}, 100.0}
@@ -18,9 +22,14 @@ func main() {
 	frame := Frame{}
 	for x := 0; x < HEIGHT; x++ {
 		for y := 0; y < WIDTH; y++ {
-			u, v := float64(y)/float64(WIDTH), float64(x)/float64(HEIGHT)
-			ray := Ray{ORIGIN, AddVec3D(AddVec3D(LOWER_LEFT_CORNER, ScalarProduct(u, HORIZONTAL)), ScalarProduct(v, VERTICAL))}
-			color := RayColor(ray, WORLD)
+			var color_list [RAY_PER_PIXEL]Color
+			for it := 0; it < RAY_PER_PIXEL; it++ {
+				u := (float64(y) + rand.Float64()) / float64(WIDTH)
+				v := (float64(x) + rand.Float64()) / float64(HEIGHT)
+				ray := CAMERA.GetRay(u, v)
+				color_list[it] = RayColor(ray, WORLD)
+			}
+			color := ColorMeanSquare(color_list)
 			frame[HEIGHT-1-x][y] = color.ToPixel()
 		}
 	}