medium_ray_tracing_code_12.py

# global variable along with width, height, etc.
max_depth = 3

# everything that follows is inside the double for loop
color = np.zeros((3))
reflection = 1

for k in range(max_depth):
    nearest_object, min_distance = # ...
    # ...
    illumination += # ...

    # reflection
    color += reflection * illumination
    reflection *= nearest_object['reflection']

    # new ray origin and direction
    origin = shifted_point
    direction = reflected(direction, normal_to_surface)

image[i, j] = np.clip(color, 0, 1)