chr-wei
diff --git a/‎geom.py‎
Lines changed: 18 additions & 5 deletions b/‎geom.py‎
Lines changed: 18 additions & 5 deletions
diff --git a/‎tiny_shaders.py‎
Lines changed: 19 additions & 2 deletions b/‎tiny_shaders.py‎
Lines changed: 19 additions & 2 deletions
@@ -6,9 +6,11 @@
 from collections.abc import Iterable
 
 from itertools import chain
+from functools import reduce
 import operator
 
-import math
+from math import sqrt
+
 import numpy as np
 
 class Vector4DType(Enum):
@@ -185,6 +187,10 @@ def tr(self): # pylint: disable=invalid-name
         elems = self._asdict().values()
         return type(self)(elems, shape = (cols, rows))
 
+    def abs(self):
+        """Returns length of vector."""
+        return vect_norm(self.get_field_values())
+
 class Point2D(MixinVector, metaclass=NamedTupleMetaEx):
     """Two-dimensional point with x and y ordinate."""
     _shape = (2,1)
@@ -203,6 +209,12 @@ class Barycentric(MixinVector, metaclass=NamedTupleMetaEx):
     one_u_v: float
     u: float
     v: float
+
+class Vector2D(MixinVector, metaclass=NamedTupleMetaEx):
+    """Two-dimensional point with x and y ordinate."""
+    _shape = (2,1)
+    x: float
+    y: float
 class Vector3D(MixinVector, metaclass=NamedTupleMetaEx):
     """Three-dimensional vector with x, y, z component."""
     _shape = (3,1)
@@ -227,10 +239,6 @@ def expand_4D(self, vtype): # pylint: disable=invalid-name
 
         return ValueError
 
-    def abs(self):
-        """Returns length of vector."""
-        return math.sqrt(self.x**2 + self.y**2 + self.z**2)
-
     def normalize(self):
         """Normalizes vector to length = 1.0."""
         abl = self.abs()
@@ -464,6 +472,11 @@ def compfloor(mat_0: list, shape_0: tuple, divisor: float):
         # Return coefficients and shape tuple
         return [int(e // divisor) for e in mat_0], shape_0
 
+def vect_norm(all_elems: list):
+    """Return norm of n-dim vector."""
+    squared = [elem**2 for elem in all_elems]
+    return sqrt(reduce(operator.add, squared))
+
 def matadd(mat_0: list, shape_0: tuple, mat_1: list, shape_1: tuple):
     """Performing componentwise addition."""
     (rows_0, cols_0) = shape_0
 
@@ -4,7 +4,7 @@
 import math
 import our_gl as gl
 from geom import Matrix4D, Matrix3D, MatrixUV, \
-                 Vector4DType, Vector3D, Barycentric, PointUV, \
+                 Vector4DType, Vector3D, Vector2D, Point2D, Barycentric, PointUV, \
                  transform_3D4D3D, transform_vertex_to_screen, \
                  cross_product, comp_min
 
@@ -470,4 +470,21 @@ def fragment(self, bary: Barycentric):
         color = comp_min(Vector3D(255, 255, 255), color) // 1
 
         # Do not discard pixel and return color
-        return (False, color)
+        return (False, color)
+
+def get_max_elevation_angle(pt_amb: Point2D, z_buffer: list, sweep_dir: Vector2D):
+    """Returns max elevation angle ray-casted from starting point pt_amb."""
+    (im_w, im_h) = len(z_buffer)
+    fact = 1 / max(pt_amb)
+    vect_amb = Vector3D(pt_amb.x, pt_amb.x, z_buffer[pt_amb.x][pt_amb.y])
+    sweep_proj = vect_amb
+    max_angle = 0
+
+    while 0 <= sweep_proj.x < im_w and 0 <= sweep_proj.y < im_h:
+        sweep_proj += fact * Vector3D(sweep_dir.x, sweep_dir.y, 0)
+        z_height = z_buffer[int(sweep_proj.x // 1)][int(sweep_proj.y // 1)]
+        sweep = Vector3D(sweep_proj.x, sweep_proj.y, z_height)
+        alpha = math.acos((sweep - vect_amb).normalize() * sweep_proj.normalize())
+        max_angle = max(alpha, max_angle)
+
+    return max_angle