camera.py

from operation_vector import Vector

class Camera:
    def __init__(self, image_nrows, image_ncols, distance_focale):
        self.image_nrows = image_nrows
        self.image_ncols = image_ncols
        self.focal_length = distance_focale

    def ray_at(self, row, col):
        n = self.image_nrows//2
        x = (n-row)/n
        n = self.image_ncols//2
        y = (n-col)/n
        return Vector((x, y, self.focal_length))