def create_pyramid():
primitives = []
# create the sphere primitive
material = None
object_to_world = Transform()
world_to_object = Transform()
sphere = Sphere(object_to_world, world_to_object, False, 1.0, -1.0, 1.0,
360)
primitive = GeometricPrimitive(sphere, material, None)
# now create the instances
# create the objects, with this layout (shown in 2D)
# O O O O level 0, width 4, pos_x -6
# O O O level 1, width 3, pos_x -4
# O O level 2, width 2, pos_x -2
# O level 3, width 1, pos_x 0
for level in range(4):
width_array = 4 - level
start_pos = Point(-2.0 * (3 - level), -2.0 * (3 - level), level * 2.0)
for i in range(width_array):
start_pos_i = start_pos + i * Vector(2.0, 0.0, 0.0)
for j in range(width_array):
pos = start_pos_i + j * Vector(0.0, 2.0, 0.0)
transform = translate(pos).inverse()
world_to_primitive = AnimatedTransform(transform, 0.0,
transform, 1.0)
instance = TransformedPrimitive(primitive, world_to_primitive)
primitives.append(instance)
return primitives
def setUp(self):
# create a transform
o2w = translate(Vector(10, 0, 0)) * scale(1.3, 1.8, 2.0)
w2o = o2w.inverse()
# create the sphere
self.sphere = Sphere(o2w, w2o, False, 1.0, -1.0, 1.0, 360)
def create_simple_sphere():
primitives = []
material = None
object_to_world = translate(Point(0, 1, 0))
world_to_object = object_to_world.inverse()
sphere = Sphere(object_to_world, world_to_object, False, 1.0, -1.0, 1.0,
360)
primitive = GeometricPrimitive(sphere, material, None)
primitives.append(primitive)
return primitives
def setUp(self):
# create a transform
o2w1 = translate(Vector(10, 0, 0)) * scale(1.3, 1.8, 2.0)
w2o1 = o2w1.inverse()
# create a sphere
self.sphere1 = Sphere(o2w1, w2o1, False, 1.0, -1.0, 1.0, 360)
self.primitive_sphere1 = GeometricPrimitive(self.sphere1, None, None)
# create a transform
o2w2 = translate(Vector(-10, 0, 0)) * scale(1.3, 1.8, 2.0)
w2o2 = o2w2.inverse()
# create a sphere
self.sphere2 = Sphere(o2w2, w2o2, False, 1.0, -1.0, 1.0, 360)
self.primitive_sphere2 = GeometricPrimitive(self.sphere2, None, None)
primitives = [self.primitive_sphere1, self.primitive_sphere2]
self.grid_accel = GridAccel(primitives, True)
def time_sphere_intersection():
# create a transform
o2w = translate(Vector(10,0,0)) * scale(1.3, 1.8, 2.0)
w2o = o2w.inverse()
# create the sphere
sphere = Sphere(o2w, w2o, False, 1.0, -1.0, 1.0, 360)
# create a large amount of rays,
# choose so that half of them will intersect the ray
positions = [Point(random.randint(0,100),
random.randint(0,100),
random.randint(0,100)
) for i in range(size)]
ray = Ray(Point(0,0,0), Vector(1.0, 1.0, 1.0))
vectors = []
for i in xrange(size):
position = positions[i]
if i%2 == 0:
# make sure this ray hit the sphere
vector = sphere.object_to_world(Point(0, 0, 0)) - position
vector /= float(random.randint(1,10))
else:
# construct a random vector
vector = Vector((random.random()-0.5)*random.randint(1,5),
(random.random()-0.5)*random.randint(1,5),
(random.random()-0.5*random.randint(1,5)))
vectors.append(vector)
intersections = 0
t1 = time.time()
for i in xrange(nb_calls):
ray.o = positions[i%size]
ray.d = vectors[i%size]
if sphere.intersect_p(ray):
intersections += 1
t2 = time.time()
for i in xrange(nb_calls):
ray.o = positions[i%size]
ray.d = vectors[i%size]
sphere.intersect(ray)
t3 = time.time()
print "%d calls, %d intersections" % (nb_calls, intersections)
print "Sphere.intersect_p() %.2fms" % ((t2-t1)/float(nb_calls)*1000.0)
print "Sphere.intersect() %.2fms" % ((t3-t2)/float(nb_calls)*1000.0)
