motive.load_project("vr_demo.ttp")
motive.update()
# Create Arena and cube
reader = WavefrontReader(ratcave.graphics.resources.obj_arena)
arena = reader.get_mesh('Arena', lighting=True, centered=False)
arena_rb = motive.get_rigid_bodies()['Arena']
player = motive.get_rigid_bodies()['CalibWand']
wand = motive.get_rigid_bodies()['CalibWand']
# Calculate Arena's orientation
for attempt in range(3):
arena_rb.reset_orientation()
motive.update()
arena_markers = np.array(arena_rb.point_cloud_markers)
additional_rotation = rotate_to_var(arena_markers)
# Create Virtual Objects
reader = WavefrontReader('vr_demo.obj')
meshes = {name:reader.get_mesh(name, lighting=True, centered=False) for name in reader.mesh_names}
meshes['Arena'].visible = False
meshes['StarGrid'].drawstyle = 'point'
meshes['Monkey'].load_texture(resources.img_colorgrid)
meshes['Monkey'].material.spec_weight = 10.
meshes['Monkey'].material.diffuse.rgb = (1.,)*3
meshes['SkyBox'].material.diffuse.rgb = 1., 0., 0.
#meshes['SkyBox'].data.normals *= -1.
meshes['SkyBox'].local.scale = 1.
meshes['SkyBox'].lighting = False
def display(optitrack_ip="127.0.0.1", calib_object_name=''):
# Connect to Optitrack
# tracker = ratcave.devices.Optitrack(client_ip=optitrack_ip)
# Create Arena and cube
reader = WavefrontReader(ratcave.graphics.resources.obj_arena)
arena = reader.get_mesh('Arena', lighting=True, centered=False)
arena.load_texture(ratcave.graphics.resources.img_colorgrid)
meshes = [arena]
if calib_object_name:
reader = WavefrontReader(ratcave.graphics.resources.obj_primitives)
cube = reader.get_mesh('Sphere', lighting=True, scale=.02, centered=True)
meshes.append(cube)
# Create Scene and Window
scene = Scene(meshes)
scene.camera = projector
scene.camera.fov_y = 27.8
scene.light.position = scene.camera.position
window = Window(scene, screen=1, fullscr=True)
# Update Everything's Position
# arena.world.position = tracker.rigid_bodies['Arena'].position
# arena.world.rotation = tracker.rigid_bodies['Arena'].rotation_pca_y
#
print(motive.get_rigid_bodies())
arena_rb = motive.get_rigid_bodies()['Arena']
arena.world.position = arena_rb.location
arena.world.rotation = arena_rb.rotation
motive.update()
for attempt in range(3):
arena_rb.reset_orientation()
motive.update()
markers = np.array(arena_rb.point_cloud_markers)
additional_rotation = rotate_to_var(markers)
# Print the Following every time a key is detected:
print "Camera settings:\n -shift: {0}, {1}\n -position: {2}\n -fov_y(xz): {3}\n -rotation: {4}\n\n".format(
scene.camera.x_shift, scene.camera.y_shift, scene.camera.position, scene.camera.fov_y, scene.camera.rotation)
print "Arena settings:\n -local\n\tPosition: {}\n\tRotation: {}\n -world\n\tPosition: {}\n\tRotation: {}".format(
arena.local.position, arena.local.rotation, arena.world.position, arena.world.rotation)
aa = 0
while True:
# Update Everything's Position
motive.update()
arena.world.position = arena_rb.location
arena.world.rotation = arena_rb.rotation_global
arena.world.rotation[1] += additional_rotation
arena.world.rotation[1] += aa
# If there's another object to track, then track it.
if calib_object_name:
cube.local.position = motive.get_rigid_bodies()[calib_object_name].location
# Re-Draw Everything
window.draw()
window.flip()
keys = event.getKeys()
if 'escape' in keys:
window.close()
break
elif 'up' in keys:
scene.camera.fov_y += .1
print('fov_y: {}'.format(scene.camera.fov_y))
elif 'down' in keys:
scene.camera.fov_y -= .1
print('fov_y: {}'.format(scene.camera.fov_y))
elif 'space' in keys:
aa += 180