diff --git a/src/main.go b/src/main.go index e9e0ad8..1868e3b 100644 --- a/src/main.go +++ b/src/main.go @@ -13,8 +13,8 @@ import ( rl "github.com/gen2brain/raylib-go/raylib" ) -const WIDTH = 200 -const HEIGHT = 200 +const WIDTH = 100 +const HEIGHT = 100 const TEXTURE_SCALE = 4 const MAX_DIST = 500.0 diff --git a/src/material.go b/src/material.go index ed6c735..d9075ee 100644 --- a/src/material.go +++ b/src/material.go @@ -1,8 +1,6 @@ package main -import ( - "math" -) +import "math" type Color interface { GetColor(Vector3) Vector3 @@ -70,14 +68,14 @@ func MixMat(mat1 MaterialBrut, mat2 MaterialBrut, t float64, p Vector3) Material } type MixedMaterial struct { - mat1 Material - mat2 Material + mat1 *Material + mat2 *Material t float64 p Vector3 } func (mixedMat MixedMaterial) GetMaterialBrut() MaterialBrut { - return MixMat((mixedMat.mat1).GetMaterialBrut(), (mixedMat.mat2).GetMaterialBrut(), mixedMat.t, mixedMat.p) + return MixMat((*mixedMat.mat1).GetMaterialBrut(), (*mixedMat.mat2).GetMaterialBrut(), mixedMat.t, mixedMat.p) } func DefaultMaterial(diffuseColor Color) MaterialBrut { diff --git a/src/primitives_sdf.go b/src/primitives_sdf.go index ba173ab..9999c1e 100644 --- a/src/primitives_sdf.go +++ b/src/primitives_sdf.go @@ -6,8 +6,8 @@ type Sphere struct { material Material } -func (s Sphere) Distance(p Vector3) (float64, Material) { - return p.Length() - s.radius, s.material +func (s Sphere) Distance(p Vector3) (float64, *Material) { + return p.Length() - s.radius, &s.material } // Box @@ -16,9 +16,9 @@ type Box struct { material Material } -func (s Box) Distance(p Vector3) (float64, Material) { +func (s Box) Distance(p Vector3) (float64, *Material) { q := p.Abs().Sub(s.dimensions) - return q.Max(0.0).Length() + min(max(q.X, max(q.Y, q.Z)), 0.0), s.material + return q.Max(0.0).Length() + min(max(q.X, max(q.Y, q.Z)), 0.0), &s.material } // Plane @@ -27,6 +27,6 @@ type Plane struct { material Material } -func (s Plane) Distance(p Vector3) (float64, Material) { - return p.Dot(s.normal), s.material +func (s Plane) Distance(p Vector3) (float64, *Material) { + return p.Dot(s.normal), &s.material } diff --git a/src/ray_marching.go b/src/ray_marching.go index 6af7946..d7b806f 100644 --- a/src/ray_marching.go +++ b/src/ray_marching.go @@ -39,7 +39,7 @@ func shadow(point Vector3) float64 { } res = min(res, SOFT_SHADOW_COEFF*dist/dist_total) dist_total += dist - p.Radd(direction.Scale(dist)) + p = p.Add(direction.Scale(dist)) } return res } @@ -50,13 +50,12 @@ func rayMarch(origin Vector3, direction Vector3) (bool, Vector3) { dist, mat := scene.Distance(p) if dist < EPS { normal := Gradient(scene, p, EPS).Normalized() - return true, phongShading(p, normal, mat).Scale(max(shadow(p), 0.3)) - // return true, phongShading(p, normal, *mat) + return true, phongShading(p, normal, *mat).Scale(max(shadow(p), 0.3)) } if dist > MAX_DIST { break } - p.Radd(direction.Scale(dist)) + p = p.Add(direction.Scale(dist)) } return false, GREY } diff --git a/src/sdf.go b/src/sdf.go index c57df23..64f52c0 100644 --- a/src/sdf.go +++ b/src/sdf.go @@ -5,7 +5,7 @@ import ( ) type SDF interface { - Distance(Vector3) (float64, Material) + Distance(Vector3) (float64, *Material) } func DistanceOnly(sdf SDF, p Vector3) float64 { @@ -27,7 +27,7 @@ type TranslatedSDF struct { translate Vector3 } -func (s TranslatedSDF) Distance(p Vector3) (float64, Material) { +func (s TranslatedSDF) Distance(p Vector3) (float64, *Material) { return s.primitive.Distance(p.Sub(s.translate)) } @@ -37,7 +37,7 @@ type RotatedSDF struct { angle float64 } -func (s RotatedSDF) Distance(p Vector3) (float64, Material) { +func (s RotatedSDF) Distance(p Vector3) (float64, *Material) { rotated_p := Rotate(p, s.rotVector, s.angle) return s.primitive.Distance(rotated_p) } @@ -47,7 +47,7 @@ type ScaledSDF struct { scale float64 } -func (s ScaledSDF) Distance(p Vector3) (float64, Material) { +func (s ScaledSDF) Distance(p Vector3) (float64, *Material) { dist, color := s.primitive.Distance(p.Scale(1 / s.scale)) return dist * s.scale, color } @@ -57,7 +57,7 @@ type RepeatSDF struct { cellSize Vector3 } -func (s RepeatSDF) Distance(p Vector3) (float64, Material) { +func (s RepeatSDF) Distance(p Vector3) (float64, *Material) { x, y, z := p.Unpack() sx, sy, sz := s.cellSize.Unpack() round := math.RoundToEven @@ -70,10 +70,10 @@ type UnionSDF struct { primitive2 SDF } -func (s UnionSDF) Distance(p Vector3) (float64, Material) { +func (s UnionSDF) Distance(p Vector3) (float64, *Material) { d1, color1 := s.primitive1.Distance(p) d2, color2 := s.primitive2.Distance(p) - d := min(d1, d2) + d := math.Min(d1, d2) color := color1 if d2 < d1 { @@ -87,11 +87,11 @@ type SubstractionSDF struct { primitive2 SDF } -func (s SubstractionSDF) Distance(p Vector3) (float64, Material) { +func (s SubstractionSDF) Distance(p Vector3) (float64, *Material) { d1, color1 := s.primitive1.Distance(p) d2, _ := s.primitive2.Distance(p) - d := max(d1, -d2) + d := math.Max(d1, -d2) return d, color1 } @@ -100,14 +100,14 @@ type IntersectionSDF struct { primitive2 SDF } -func (s IntersectionSDF) Distance(p Vector3) (float64, Material) { +func (s IntersectionSDF) Distance(p Vector3) (float64, *Material) { d1, mat1 := s.primitive1.Distance(p) d2, mat2 := s.primitive2.Distance(p) - d := max(d1, d2) + d := math.Max(d1, d2) var mat Material = MixedMaterial{mat1, mat2, 0.5, p} - return d, mat + return d, &mat } type SmoothUnionSDF struct { @@ -116,15 +116,15 @@ type SmoothUnionSDF struct { k float64 } -func (s SmoothUnionSDF) Distance(p Vector3) (float64, Material) { +func (s SmoothUnionSDF) Distance(p Vector3) (float64, *Material) { k := 4 * s.k d1, mat1 := s.primitive1.Distance(p) d2, mat2 := s.primitive2.Distance(p) - h := max(k-math.Abs(d1-d2), 0.0) - d := min(d1, d2) - h*h*0.25/k + h := math.Max(k-math.Abs(d1-d2), 0.0) + d := math.Min(d1, d2) - h*h*0.25/k t := SmoothStep(d2-d1, -k, k) var mat Material = MixedMaterial{mat2, mat1, t, p} - return d, mat + return d, &mat } type SmoothSubstractionSDF struct { @@ -133,15 +133,15 @@ type SmoothSubstractionSDF struct { k float64 } -func (s SmoothSubstractionSDF) Distance(p Vector3) (float64, Material) { +func (s SmoothSubstractionSDF) Distance(p Vector3) (float64, *Material) { k := 4 * s.k d1, mat1 := s.primitive1.Distance(p) d2, mat2 := s.primitive2.Distance(p) - h := max(k-math.Abs(-d1-d2), 0.0) - d := max(d1, -d2) + h*h*0.25/k + h := math.Max(k-math.Abs(-d1-d2), 0.0) + d := math.Max(d1, -d2) + h*h*0.25/k t := SmoothStep(d1-d2, -k, k) var mat Material = MixedMaterial{mat1, mat2, t, p} - return d, mat + return d, &mat } type SmoothIntersectionSDF struct { @@ -150,13 +150,13 @@ type SmoothIntersectionSDF struct { k float64 } -func (s SmoothIntersectionSDF) Distance(p Vector3) (float64, Material) { +func (s SmoothIntersectionSDF) Distance(p Vector3) (float64, *Material) { k := 4 * s.k d1, mat1 := s.primitive1.Distance(p) d2, mat2 := s.primitive2.Distance(p) - h := max(k-math.Abs(d1-d2), 0.0) - d := max(d1, d2) + h*h*0.25/k + h := math.Max(k-math.Abs(d1-d2), 0.0) + d := math.Max(d1, d2) + h*h*0.25/k t := SmoothStep(d2-d1, -k, k) var mat Material = MixedMaterial{mat1, mat2, t, p} - return d, mat + return d, &mat } diff --git a/src/vec3.go b/src/vec3.go index 3f02686..7aea4c9 100644 --- a/src/vec3.go +++ b/src/vec3.go @@ -16,18 +16,12 @@ func (u Vector3) Add(v Vector3) Vector3 { return Vector3{u.X + v.X, u.Y + v.Y, u.Z + v.Z} } -func (u *Vector3) Radd(v Vector3) { - u.X += v.X - u.Y += v.Y - u.Z += v.Z -} - func (u Vector3) Neg() Vector3 { return Vector3{-u.X, -u.Y, -u.Z} } func (u Vector3) Sub(v Vector3) Vector3 { - return Vector3{u.X - v.X, u.Y - v.Y, u.Z - v.Z} + return u.Add(v.Neg()) } func (u Vector3) Scale(a float64) Vector3 {