kuhyx/WUT_Computer_Science
1
0
Fork 0
mirror of https://github.com/kuhyx/WUT_Computer_Science.git synced 2026-07-04 15:03:08 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: flake perfect

Browse Source
This commit is contained in:
Krzysztof Rudnicki 2024-12-27 12:33:06 +01:00
parent df9da83336
commit 0a497f35a2
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
code/main.py
View File

@ -56,7 +56,7 @@ def intersect_sphere(ray_origin, ray_direction, sphere_center, sphere_radius):
ray_origin (numpy.ndarray): A 3D point representing the origin of the ray. ray_origin (numpy.ndarray): A 3D point representing the origin of the ray.
ray_direction (numpy.ndarray): A normalized 3D vector representing the ray_direction (numpy.ndarray): A normalized 3D vector representing the
direction of the ray. direction of the ray.
sphere_center (numpy.ndarray): A 3D point representing sphere_center (numpy.ndarray): A 3D point representing
the center of the sphere. the center of the sphere.
sphere_radius (float): The radius of the sphere. sphere_radius (float): The radius of the sphere.
@ -77,7 +77,8 @@ def intersect_sphere(ray_origin, ray_direction, sphere_center, sphere_radius):
def calculate_sphere_intersection(a, b, c, disc): def calculate_sphere_intersection(a, b, c, disc):
""" """
Calculate the intersection distance of a ray with a sphere using the quadratic formula. Calculate the
intersection distance of a ray with a sphere using the quadratic formula.
Parameters: Parameters:
a (float): Coefficient of t^2 in the quadratic equation. a (float): Coefficient of t^2 in the quadratic equation.
@ -86,8 +87,10 @@ def calculate_sphere_intersection(a, b, c, disc):
disc (float): Discriminant of the quadratic equation. disc (float): Discriminant of the quadratic equation.
Returns: Returns:
float: The distance from the origin to the intersection point with the sphere. float:
Returns +inf if there is no intersection or if the intersection is behind the origin. The distance from the origin to the intersection point with the sphere.
Returns +inf if there is no intersection
or if the intersection is behind the origin.
""" """
if disc > 0: if disc > 0:
distance_squared = np.sqrt(disc) distance_squared = np.sqrt(disc)
@ -179,7 +182,7 @@ def trace_ray(ray_origin, ray_direction):
A normalized 3D vector representing the direction of the ray. A normalized 3D vector representing the direction of the ray.
Returns: Returns:
tuple: A tuple containing the object, tuple: A tuple containing the object,
intersection point, normal at the intersection, intersection point, normal at the intersection,
and the color at the intersection point. and the color at the intersection point.
Returns None if there is no intersection. Returns None if there is no intersection.
@ -358,7 +361,7 @@ SPECULAR_K = 50
DEPTH_MAX = 5 # Maximum number of light reflections. DEPTH_MAX = 5 # Maximum number of light reflections.
col = np.zeros(3) # Current color. col = np.zeros(3) # Current color.
O = np.array([0., 0.35, -1.]) # Camera. camera_origin = np.array([0., 0.35, -1.]) # Camera.
Q = np.array([0., 0., 0.]) # Camera pointing to. Q = np.array([0., 0., 0.]) # Camera pointing to.
img = np.zeros((IMAGE_HEIGHT, IMAGE_WIDTH, 3)) img = np.zeros((IMAGE_HEIGHT, IMAGE_WIDTH, 3))
@ -373,9 +376,9 @@ for i, x in enumerate(np.linspace(S[0], S[2], IMAGE_WIDTH)):
for j, y in enumerate(np.linspace(S[1], S[3], IMAGE_HEIGHT)): for j, y in enumerate(np.linspace(S[1], S[3], IMAGE_HEIGHT)):
col[:] = 0 col[:] = 0
Q[:2] = (x, y) Q[:2] = (x, y)
D = normalize(Q - O) D = normalize(Q - camera_origin)
DEPTH = 0 DEPTH = 0
rayO, rayD = O, D rayO, rayD = camera_origin, D
REFLECTION = 1. REFLECTION = 1.
# Loop through initial and secondary rays. # Loop through initial and secondary rays.
while DEPTH < DEPTH_MAX: while DEPTH < DEPTH_MAX: