pyovr-examples/RiftDemo.py at master · OculusRiftInAction/pyovr-examples · 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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#! /usr/bin/env python
import pygame
import pygame.locals as pgl
import oculusvr as ovr

from RiftApp import RiftApp
from cgkit.cgtypes import mat4, vec3
from OpenGL.GL import *

def draw_color_cube(size=1.0):
  p = size / 2.0
  n = -p
  glBegin(GL_QUADS)

  # front
  glColor3f(1, 1, 0)
  glVertex3f(n, n, n)
  glVertex3f(p, n, n)
  glVertex3f(p, p, n)
  glVertex3f(n, p, n)
  # back
  glColor3f(0.2, 0.2, 1)
  glVertex3f(n, n, p)
  glVertex3f(p, n, p)
  glVertex3f(p, p, p)
  glVertex3f(n, p, p)
  # right
  glColor3f(1, 0, 0)
  glVertex3f(p, n, n)
  glVertex3f(p, n, p)
  glVertex3f(p, p, p)
  glVertex3f(p, p, n)
  # left
  glColor3f(0, 1, 1)
  glVertex3f(n, n, n)
  glVertex3f(n, n, p)
  glVertex3f(n, p, p)
  glVertex3f(n, p, n)
  # top
  glColor3f(0, 1, 0)
  glVertex3f(n, p, n)
  glVertex3f(p, p, n)
  glVertex3f(p, p, p)
  glVertex3f(n, p, p)
  # bottom
  glColor3f(1, 0, 1)
  glVertex3f(n, n, n)
  glVertex3f(p, n, n)
  glVertex3f(p, n, p)
  glVertex3f(n, n, p)
  glEnd()


class RiftDemo(RiftApp):
  def __init__(self):
    RiftApp.__init__(self)
    self.cube_size = self.hmd.get_float(
      ovr.OVR_KEY_IPD, ovr.OVR_DEFAULT_IPD)
    self.reset_camera()

  def reset_camera(self):
    self.camera = mat4(1.0)
    self.camera.translate(vec3(0, 0, 0.2))

  def recompose_camera(self):
    (tr, rot, sc) = self.camera.decompose()
    self.camera = mat4(1.0)
    self.camera.translate(tr)
    self.camera = self.camera * rot

  def init_gl(self):
    RiftApp.init_gl(self)
    glEnable(GL_DEPTH_TEST)
    glClearColor(0.1, 0.1, 0.1, 1)

  def update(self):
    RiftApp.update(self)
    pressed = pygame.key.get_pressed()

    if pressed[pgl.K_r]:
      self.reset_camera()

    rotation = 0.0

    if pressed[pgl.K_q]:
      rotation = +1.0
    if pressed[pgl.K_e]:
      rotation = -1.0
    if (rotation != 0.0):
      self.camera = self.camera * \
        mat4.rotation(rotation * 0.01, vec3(0, 1, 0))
      self.recompose_camera()

    # Modify direction vectors for key presses
    translation = vec3()
    if pressed[pgl.K_r]:
      self.hmd.recenter_pose()
    if pressed[pgl.K_w]:
      translation.z = -1.0
    elif pressed[pgl.K_s]:
      translation.z = +1.0
    if pressed[pgl.K_a]:
      translation.x = -1.0
    elif pressed[pgl.K_d]:
      translation.x = +1.0
    if (vec3.length(translation) > 0.1):
      translation = self.camera.getMat3() * (translation * 0.005)
      self.camera.translate(translation)
      self.recompose_camera()


  def render_scene(self):
    glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)

    # apply the camera position
    cameraview = self.eyeview * self.camera
    glMatrixMode(GL_MODELVIEW)
    glLoadMatrixf(cameraview.inverse().toList())

    glMultMatrixf(self.camera.inverse().toList())
    draw_color_cube(self.cube_size)


RiftDemo().run();