python-ray-tracing/main.py at main · lucashnss/python-ray-tracing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
from obj_reader import ObjReader
from camera import Camera
from renderer import Renderer
from point import Point
from vector import Vector
from sphere import Sphere
from plane import Plane

# Divirtam-se :)


def main():
    # Configurações de câmera
    camera = Camera(
        camera_point=Point(0, 0, 5),
        target_point=Point(0, 0, 0),
        vector_up=Vector(0, 1, 0),
        target_distance=1,
        hres=500,
        vres=500
    )
    print(camera)

    # Criar objetos

    # Novas esferas

    sphere4 = Sphere(center=Point(2, -4.5, -2), radius=0.5, color=(0, 0, 255))
    sphere5 = Sphere(center=Point(0, -4, -2), radius=1, color=(0,255,0))
    sphere6 = Sphere(center=Point(-3, -3.5, -2), radius=1.5, color=(255,0,0))


    # Novos planos
    plane3 = Plane(point=Point(5, 0, 0), normal=Vector(-1, 0, 0), color=(0,255,0))
    plane4 = Plane(point=Point(-5, 0, 0), normal=Vector(1, 0, 0), color=(0, 0, 255))
    plane5 = Plane(point=Point(0, -5, 0), normal=Vector(0, 1, 0), color=(255,255,255))
    plane6 = Plane(point=Point(0, 5, 0), normal=Vector(0, -1, 0), color=(255,255,255))
    plane7 = Plane(point=Point(0, 0, -5), normal=Vector(0, 0, 1), color=(255,255,255))
    plane8 = Plane(point=Point(0, 0, 6), normal=Vector(0, 0, -1), color=(255,255,255))


    # Renderizar a cena
    objects = [sphere4, sphere5, sphere6, plane3, plane4, plane5, plane6, plane7, plane8]
    renderer = Renderer(camera, objects, hres=500, vres=500)
    renderer.render()


if __name__ == "__main__":
    main()