rev |
line source |
nuclear@1
|
1 #include <stdio.h>
|
nuclear@12
|
2 #include <GL/glew.h>
|
nuclear@1
|
3 #include "vr.h"
|
nuclear@1
|
4 #include "vr_impl.h"
|
nuclear@15
|
5 #include "vr_sdr.h"
|
nuclear@12
|
6
|
nuclear@18
|
7 static void init_ctx();
|
nuclear@1
|
8 static bool init_ovr();
|
nuclear@12
|
9 static bool init_sdr();
|
nuclear@1
|
10
|
nuclear@3
|
11 VRContext vr_ctx;
|
nuclear@12
|
12 static unsigned int sdrprog;
|
nuclear@1
|
13
|
nuclear@1
|
14 extern "C" int vr_init(enum vr_init_mode mode)
|
nuclear@1
|
15 {
|
nuclear@12
|
16 glewInit();
|
nuclear@12
|
17
|
nuclear@18
|
18 init_ctx();
|
nuclear@18
|
19
|
nuclear@1
|
20 if(!init_ovr()) {
|
nuclear@1
|
21 return -1;
|
nuclear@1
|
22 }
|
nuclear@12
|
23
|
nuclear@12
|
24 if(!init_sdr()) {
|
nuclear@12
|
25 return -1;
|
nuclear@12
|
26 }
|
nuclear@12
|
27
|
nuclear@1
|
28 return 0;
|
nuclear@1
|
29 }
|
nuclear@1
|
30
|
nuclear@1
|
31 extern "C" void vr_shutdown(void)
|
nuclear@1
|
32 {
|
nuclear@15
|
33 delete [] vr_ctx.info.display;
|
nuclear@3
|
34 //System::Destroy();
|
nuclear@1
|
35 }
|
nuclear@1
|
36
|
nuclear@18
|
37 static void init_ctx()
|
nuclear@18
|
38 {
|
nuclear@18
|
39 vr_ctx.info.width = 1280;
|
nuclear@18
|
40 vr_ctx.info.height = 800;
|
nuclear@18
|
41 vr_ctx.info.fov = M_PI / 2.0;
|
nuclear@18
|
42 vr_ctx.info.aspect = (float)vr_ctx.info.width / (float)vr_ctx.info.height;
|
nuclear@18
|
43 vr_ctx.info.ipd = 0.035;
|
nuclear@18
|
44 vr_ctx.info.scale = 1.0;
|
nuclear@18
|
45 }
|
nuclear@18
|
46
|
nuclear@1
|
47 static bool init_ovr()
|
nuclear@1
|
48 {
|
nuclear@8
|
49 LogMaskConstants log_level = LogMask_All;
|
nuclear@1
|
50 // initialize Oculus SDK
|
nuclear@5
|
51 const char *logenv = getenv("VR_LOGLEVEL");
|
nuclear@5
|
52 if(logenv) {
|
nuclear@5
|
53 switch(atoi(logenv)) {
|
nuclear@5
|
54 case 0:
|
nuclear@5
|
55 log_level = LogMask_None;
|
nuclear@5
|
56 break;
|
nuclear@5
|
57 case 1:
|
nuclear@5
|
58 log_level = LogMask_Regular;
|
nuclear@5
|
59 break;
|
nuclear@5
|
60 case 2:
|
nuclear@5
|
61 default:
|
nuclear@5
|
62 log_level = LogMask_All;
|
nuclear@5
|
63 break;
|
nuclear@5
|
64 }
|
nuclear@5
|
65 }
|
nuclear@5
|
66
|
nuclear@5
|
67 System::Init(Log::ConfigureDefaultLog(log_level));
|
nuclear@3
|
68 if(!(vr_ctx.ovr_devman = DeviceManager::Create())) {
|
nuclear@1
|
69 fprintf(stderr, "failed to create OVR device manager\n");
|
nuclear@1
|
70 return false;
|
nuclear@1
|
71 }
|
nuclear@1
|
72
|
nuclear@1
|
73 // create the display device
|
nuclear@4
|
74 HMDInfo info;
|
nuclear@3
|
75 if(!(vr_ctx.ovr_hmd_dev = vr_ctx.ovr_devman->EnumerateDevices<HMDDevice>().CreateDevice())) {
|
nuclear@1
|
76 fprintf(stderr, "no oculus rift devices found\n");
|
nuclear@4
|
77 } else {
|
nuclear@4
|
78 if(vr_ctx.ovr_hmd_dev->GetDeviceInfo(&info)) {
|
nuclear@4
|
79 printf("oculus device info:\n");
|
nuclear@4
|
80 printf(" name: %s\n", info.DisplayDeviceName);
|
nuclear@4
|
81 printf(" ipd: %f\n", info.InterpupillaryDistance);
|
nuclear@4
|
82 printf(" distortion: %f %f %f %f\n", info.DistortionK[0],
|
nuclear@4
|
83 info.DistortionK[1], info.DistortionK[2], info.DistortionK[3]);
|
nuclear@4
|
84 }
|
nuclear@1
|
85
|
nuclear@4
|
86 // calculate and store viewing parameters
|
nuclear@4
|
87 vr_ctx.info.width = info.HResolution;
|
nuclear@4
|
88 vr_ctx.info.height = info.VResolution;
|
nuclear@4
|
89 vr_ctx.info.aspect = (float)vr_ctx.info.width / (float)vr_ctx.info.height;
|
nuclear@4
|
90
|
nuclear@4
|
91 vr_ctx.info.ipd = info.InterpupillaryDistance;
|
nuclear@4
|
92 for(int i=0; i<4; i++) {
|
nuclear@4
|
93 vr_ctx.info.distort[i] = info.DistortionK[i];
|
nuclear@4
|
94 }
|
nuclear@12
|
95
|
nuclear@14
|
96 Util::Render::StereoConfig stereohelp;
|
nuclear@14
|
97 stereohelp.SetFullViewport(Util::Render::Viewport(0, 0, vr_ctx.info.width, vr_ctx.info.height));
|
nuclear@14
|
98 stereohelp.SetStereoMode(Util::Render::Stereo_LeftRight_Multipass);
|
nuclear@14
|
99 stereohelp.SetHMDInfo(info);
|
nuclear@14
|
100 stereohelp.SetDistortionFitPointVP(-1.0, 0.0);
|
nuclear@14
|
101
|
nuclear@14
|
102 vr_ctx.info.scale = stereohelp.GetDistortionScale();
|
nuclear@13
|
103
|
nuclear@13
|
104 float vhalfsz = vr_ctx.info.scale * info.VScreenSize * 0.5;
|
nuclear@13
|
105 vr_ctx.info.fov = 2.0 * atan(vhalfsz / info.EyeToScreenDistance);
|
nuclear@13
|
106
|
nuclear@13
|
107 vr_ctx.info.lens_center_offset = 0.5 - info.LensSeparationDistance / info.HScreenSize;
|
nuclear@13
|
108
|
nuclear@13
|
109 // calculate center of projection shift to match the lens positions
|
nuclear@13
|
110 float center_dist_meters = info.HScreenSize * 0.25;
|
nuclear@13
|
111 float proj_shift = center_dist_meters - info.LensSeparationDistance * 0.5;
|
nuclear@13
|
112 vr_ctx.info.proj_center_offset = 4.0 * proj_shift / info.HScreenSize;
|
nuclear@15
|
113
|
nuclear@16
|
114 // grab the display info
|
nuclear@15
|
115 vr_ctx.info.display = new char[strlen(info.DisplayDeviceName) + 1];
|
nuclear@15
|
116 strcpy(vr_ctx.info.display, info.DisplayDeviceName);
|
nuclear@16
|
117
|
nuclear@16
|
118 vr_ctx.info.display_xoffs = info.DesktopX;
|
nuclear@16
|
119 vr_ctx.info.display_yoffs = info.DesktopY;
|
nuclear@16
|
120
|
nuclear@16
|
121 printf("display: \"%s\" offset: %+d %+d\n", vr_ctx.info.display,
|
nuclear@16
|
122 vr_ctx.info.display_xoffs, vr_ctx.info.display_yoffs);
|
nuclear@1
|
123 }
|
nuclear@1
|
124
|
nuclear@1
|
125 // get the sensor device
|
nuclear@4
|
126 if(vr_ctx.ovr_hmd_dev) {
|
nuclear@4
|
127 if(!(vr_ctx.ovr_sensor_dev = vr_ctx.ovr_hmd_dev->GetSensor())) {
|
nuclear@4
|
128 fprintf(stderr, "failed to get oculus sensor device\n");
|
nuclear@4
|
129 }
|
nuclear@4
|
130 } else {
|
nuclear@4
|
131 if(!(vr_ctx.ovr_sensor_dev = vr_ctx.ovr_devman->EnumerateDevices<SensorDevice>().CreateDevice())) {
|
nuclear@4
|
132 fprintf(stderr, "failed to get oculus sensor device\n");
|
nuclear@4
|
133 }
|
nuclear@1
|
134 }
|
nuclear@1
|
135
|
nuclear@4
|
136 if(vr_ctx.ovr_sensor_dev) {
|
nuclear@4
|
137 SensorInfo sinfo;
|
nuclear@4
|
138 if(vr_ctx.ovr_sensor_dev->GetDeviceInfo(&sinfo)) {
|
nuclear@4
|
139 printf("oculus sensor device info:\n");
|
nuclear@4
|
140 printf(" name: %s\n", sinfo.ProductName);
|
nuclear@4
|
141 }
|
nuclear@4
|
142
|
nuclear@4
|
143 vr_ctx.ovr_sfusion.AttachToSensor(vr_ctx.ovr_sensor_dev);
|
nuclear@1
|
144 }
|
nuclear@1
|
145 return true;
|
nuclear@1
|
146 }
|
nuclear@3
|
147
|
nuclear@13
|
148 #define EXTERNAL_SHADER
|
nuclear@13
|
149
|
nuclear@12
|
150 static bool init_sdr()
|
nuclear@12
|
151 {
|
nuclear@12
|
152 int status;
|
nuclear@12
|
153
|
nuclear@13
|
154 #ifdef EXTERNAL_SHADER
|
nuclear@17
|
155 FILE *fp = fopen("sdr/sdr.glsl", "rb");
|
nuclear@13
|
156 if(!fp) {
|
nuclear@13
|
157 perror("failed to load sdr.glsl");
|
nuclear@13
|
158 return false;
|
nuclear@13
|
159 }
|
nuclear@13
|
160 fseek(fp, 0, SEEK_END);
|
nuclear@13
|
161 long sz = ftell(fp);
|
nuclear@13
|
162 rewind(fp);
|
nuclear@13
|
163
|
nuclear@13
|
164 char *buf = (char*)alloca(sz + 1);
|
nuclear@13
|
165 fread(buf, 1, sz, fp);
|
nuclear@13
|
166 buf[sz] = 0;
|
nuclear@13
|
167 sdr_src = buf;
|
nuclear@13
|
168
|
nuclear@13
|
169 fclose(fp);
|
nuclear@13
|
170 #endif
|
nuclear@13
|
171
|
nuclear@12
|
172 unsigned int sdr = glCreateShader(GL_FRAGMENT_SHADER);
|
nuclear@12
|
173 glShaderSource(sdr, 1, &sdr_src, 0);
|
nuclear@12
|
174 glCompileShader(sdr);
|
nuclear@12
|
175 glGetShaderiv(sdr, GL_COMPILE_STATUS, &status);
|
nuclear@12
|
176 if(!status) {
|
nuclear@12
|
177 fprintf(stderr, "failed to compile distortion shader\n");
|
nuclear@14
|
178
|
nuclear@14
|
179 int loglen;
|
nuclear@14
|
180 glGetShaderiv(sdr, GL_INFO_LOG_LENGTH, &loglen);
|
nuclear@14
|
181
|
nuclear@14
|
182 if(loglen > 0) {
|
nuclear@14
|
183 char *log = (char*)alloca(loglen);
|
nuclear@14
|
184 glGetShaderInfoLog(sdr, loglen, &loglen, log);
|
nuclear@14
|
185 fprintf(stderr, "%s\n", log);
|
nuclear@14
|
186 }
|
nuclear@14
|
187
|
nuclear@12
|
188 return false;
|
nuclear@12
|
189 }
|
nuclear@12
|
190
|
nuclear@12
|
191 sdrprog = glCreateProgram();
|
nuclear@12
|
192 glAttachShader(sdrprog, sdr);
|
nuclear@12
|
193 glLinkProgram(sdrprog);
|
nuclear@12
|
194 if(!status) {
|
nuclear@12
|
195 fprintf(stderr, "failed to link distortion shader program\n");
|
nuclear@12
|
196 glDeleteShader(sdr);
|
nuclear@12
|
197 return false;
|
nuclear@12
|
198 }
|
nuclear@12
|
199
|
nuclear@12
|
200 int loc;
|
nuclear@12
|
201
|
nuclear@12
|
202 glUseProgram(sdrprog);
|
nuclear@12
|
203
|
nuclear@12
|
204 if((loc = glGetUniformLocation(sdrprog, "tex")) != -1) {
|
nuclear@12
|
205 glUniform1i(loc, 0);
|
nuclear@12
|
206 }
|
nuclear@13
|
207 if((loc = glGetUniformLocation(sdrprog, "lens_center_offset")) != -1) {
|
nuclear@13
|
208 glUniform1f(loc, 0);
|
nuclear@12
|
209 }
|
nuclear@12
|
210 if((loc = glGetUniformLocation(sdrprog, "scr_center")) != -1) {
|
nuclear@12
|
211 glUniform2f(loc, 0, 0);
|
nuclear@12
|
212 }
|
nuclear@12
|
213 if((loc = glGetUniformLocation(sdrprog, "scale")) != -1) {
|
nuclear@13
|
214 glUniform1f(loc, vr_ctx.info.scale);
|
nuclear@13
|
215 }
|
nuclear@13
|
216 if((loc = glGetUniformLocation(sdrprog, "aspect")) != -1) {
|
nuclear@13
|
217 printf("setting aspect: %f\n", vr_ctx.info.aspect / 2.0);
|
nuclear@13
|
218 glUniform1f(loc, vr_ctx.info.aspect / 2.0);
|
nuclear@12
|
219 }
|
nuclear@12
|
220 if((loc = glGetUniformLocation(sdrprog, "scale_in")) != -1) {
|
nuclear@12
|
221 glUniform2f(loc, 1, 1);
|
nuclear@12
|
222 }
|
nuclear@13
|
223 if((loc = glGetUniformLocation(sdrprog, "dist_factors")) != -1) {
|
nuclear@12
|
224 glUniform4f(loc, vr_ctx.info.distort[0], vr_ctx.info.distort[1],
|
nuclear@12
|
225 vr_ctx.info.distort[2], vr_ctx.info.distort[3]);
|
nuclear@12
|
226 }
|
nuclear@12
|
227
|
nuclear@12
|
228 return true;
|
nuclear@12
|
229 }
|
nuclear@12
|
230
|
nuclear@16
|
231 extern "C" const char *vr_get_display_name(void)
|
nuclear@16
|
232 {
|
nuclear@16
|
233 return vr_ctx.info.display;
|
nuclear@16
|
234 }
|
nuclear@16
|
235
|
nuclear@16
|
236 extern "C" void vr_get_display_pos(int *xptr, int *yptr)
|
nuclear@16
|
237 {
|
nuclear@16
|
238 *xptr = vr_ctx.info.display_xoffs;
|
nuclear@16
|
239 *yptr = vr_ctx.info.display_yoffs;
|
nuclear@16
|
240 }
|
nuclear@16
|
241
|
nuclear@3
|
242 extern "C" int vr_get_width(void)
|
nuclear@3
|
243 {
|
nuclear@3
|
244 return vr_ctx.info.width;
|
nuclear@3
|
245 }
|
nuclear@3
|
246
|
nuclear@3
|
247 extern "C" int vr_get_height(void)
|
nuclear@3
|
248 {
|
nuclear@3
|
249 return vr_ctx.info.height;
|
nuclear@3
|
250 }
|
nuclear@3
|
251
|
nuclear@3
|
252 extern "C" float vr_get_fov(void)
|
nuclear@3
|
253 {
|
nuclear@3
|
254 return vr_ctx.info.fov;
|
nuclear@3
|
255 }
|
nuclear@3
|
256
|
nuclear@3
|
257 extern "C" float vr_get_aspect(void)
|
nuclear@3
|
258 {
|
nuclear@3
|
259 return vr_ctx.info.aspect;
|
nuclear@3
|
260 }
|
nuclear@3
|
261
|
nuclear@3
|
262 extern "C" void vr_set_eyedist(float ipd)
|
nuclear@3
|
263 {
|
nuclear@3
|
264 vr_ctx.info.ipd = ipd;
|
nuclear@3
|
265 }
|
nuclear@3
|
266
|
nuclear@3
|
267 extern "C" float vr_get_eyedist(void)
|
nuclear@3
|
268 {
|
nuclear@3
|
269 return vr_ctx.info.ipd;
|
nuclear@3
|
270 }
|
nuclear@3
|
271
|
nuclear@3
|
272 extern "C" void vr_set_distort(const float *coef)
|
nuclear@3
|
273 {
|
nuclear@3
|
274 memcpy(vr_ctx.info.distort, coef, sizeof vr_ctx.info.distort);
|
nuclear@3
|
275 }
|
nuclear@3
|
276
|
nuclear@3
|
277 extern "C" void vr_get_distort(float *coef)
|
nuclear@3
|
278 {
|
nuclear@3
|
279 memcpy(coef, vr_ctx.info.distort, sizeof vr_ctx.info.distort);
|
nuclear@3
|
280 }
|
nuclear@3
|
281
|
nuclear@3
|
282 extern "C" void vr_set_prediction_sec(float dt)
|
nuclear@3
|
283 {
|
nuclear@3
|
284 vr_ctx.ovr_sfusion.SetPrediction(dt);
|
nuclear@3
|
285 }
|
nuclear@3
|
286
|
nuclear@3
|
287 extern "C" float vr_get_prediction_sec(void)
|
nuclear@3
|
288 {
|
nuclear@3
|
289 return vr_ctx.ovr_sfusion.GetPredictionDelta();
|
nuclear@3
|
290 }
|
nuclear@3
|
291
|
nuclear@13
|
292 extern "C" void vr_get_view_matrix(float *res, int eye)
|
nuclear@13
|
293 {
|
nuclear@13
|
294 // TODO
|
nuclear@13
|
295 }
|
nuclear@13
|
296
|
nuclear@13
|
297 extern "C" void vr_get_proj_matrix(float *res, int eye)
|
nuclear@13
|
298 {
|
nuclear@13
|
299 static float eye_scale[] = {0.0, 1.0, -1.0};
|
nuclear@13
|
300
|
nuclear@13
|
301 Matrix4f proj = Matrix4f::PerspectiveRH(vr_ctx.info.fov, vr_ctx.info.aspect / 2.0, 0.3, 1000.0);
|
nuclear@13
|
302 proj = Matrix4f::Translation(vr_ctx.info.proj_center_offset * eye_scale[eye], 0, 0) * proj;
|
nuclear@13
|
303
|
nuclear@13
|
304 memcpy(res, proj.M[0], 16 * sizeof(float));
|
nuclear@13
|
305 }
|
nuclear@13
|
306
|
nuclear@3
|
307 extern "C" void vr_get_translation(float *offs)
|
nuclear@3
|
308 {
|
nuclear@3
|
309 // current oculus devkit doesn't do translation
|
nuclear@3
|
310 offs[0] = offs[1] = offs[2] = 0.0f;
|
nuclear@3
|
311 }
|
nuclear@3
|
312
|
nuclear@3
|
313 extern "C" void vr_get_rotation(float *quat)
|
nuclear@3
|
314 {
|
nuclear@3
|
315 Quatf oq = vr_ctx.ovr_sfusion.GetPredictedOrientation();
|
nuclear@3
|
316 quat[0] = oq.x;
|
nuclear@3
|
317 quat[1] = oq.y;
|
nuclear@3
|
318 quat[2] = oq.z;
|
nuclear@3
|
319 quat[3] = oq.w;
|
nuclear@3
|
320 }
|
nuclear@3
|
321
|
nuclear@3
|
322 extern "C" void vr_get_rotation_euler(float *euler)
|
nuclear@3
|
323 {
|
nuclear@3
|
324 Quatf oq = vr_ctx.ovr_sfusion.GetPredictedOrientation();
|
nuclear@3
|
325 oq.GetEulerAngles<Axis_Y, Axis_X, Axis_Z>(euler + 1, euler, euler + 2);
|
nuclear@3
|
326 }
|
nuclear@12
|
327
|
nuclear@18
|
328 extern "C" void vr_draw_eye(int eye, unsigned int tex, float tex_scale_x, float tex_scale_y)
|
nuclear@12
|
329 {
|
nuclear@12
|
330 static const float rects[3][4] = {
|
nuclear@12
|
331 {-1, -1, 1, 1},
|
nuclear@12
|
332 {-1, -1, 0, 1},
|
nuclear@12
|
333 {0, -1, 1, 1}
|
nuclear@12
|
334 };
|
nuclear@12
|
335 static const float offs_scale[3] = {0.0, -1.0, 1.0};
|
nuclear@12
|
336
|
nuclear@12
|
337 glPushAttrib(GL_ENABLE_BIT);
|
nuclear@12
|
338 glDisable(GL_DEPTH_TEST);
|
nuclear@12
|
339 glDisable(GL_LIGHTING);
|
nuclear@12
|
340 glEnable(GL_TEXTURE_2D);
|
nuclear@12
|
341
|
nuclear@12
|
342 glMatrixMode(GL_MODELVIEW);
|
nuclear@12
|
343 glPushMatrix();
|
nuclear@12
|
344 glLoadIdentity();
|
nuclear@12
|
345
|
nuclear@12
|
346 glMatrixMode(GL_PROJECTION);
|
nuclear@12
|
347 glPushMatrix();
|
nuclear@12
|
348 glLoadIdentity();
|
nuclear@12
|
349
|
nuclear@12
|
350 glUseProgram(sdrprog);
|
nuclear@12
|
351
|
nuclear@13
|
352 if(sdrprog) {
|
nuclear@13
|
353 int loc;
|
nuclear@13
|
354 if((loc = glGetUniformLocation(sdrprog, "lens_center_offset")) != -1) {
|
nuclear@13
|
355 float offset = vr_ctx.info.lens_center_offset * offs_scale[eye];
|
nuclear@13
|
356 glUniform1f(loc, offset);
|
nuclear@13
|
357 }
|
nuclear@18
|
358
|
nuclear@18
|
359 if((loc = glGetUniformLocation(sdrprog, "tex_scale")) != -1) {
|
nuclear@18
|
360 glUniform2f(loc, tex_scale_x, tex_scale_y);
|
nuclear@18
|
361 }
|
nuclear@12
|
362 }
|
nuclear@12
|
363
|
nuclear@12
|
364 glBindTexture(GL_TEXTURE_2D, tex);
|
nuclear@12
|
365 glBegin(GL_QUADS);
|
nuclear@12
|
366 glColor4f(1, 1, 1, 1);
|
nuclear@12
|
367 glTexCoord2f(0, 0); glVertex2f(rects[eye][0], rects[eye][1]);
|
nuclear@12
|
368 glTexCoord2f(1, 0); glVertex2f(rects[eye][2], rects[eye][1]);
|
nuclear@12
|
369 glTexCoord2f(1, 1); glVertex2f(rects[eye][2], rects[eye][3]);
|
nuclear@12
|
370 glTexCoord2f(0, 1); glVertex2f(rects[eye][0], rects[eye][3]);
|
nuclear@12
|
371 glEnd();
|
nuclear@12
|
372
|
nuclear@12
|
373 glUseProgram(0);
|
nuclear@12
|
374
|
nuclear@12
|
375 glPopMatrix();
|
nuclear@12
|
376 glMatrixMode(GL_MODELVIEW);
|
nuclear@12
|
377 glPopMatrix();
|
nuclear@12
|
378
|
nuclear@12
|
379 glPopAttrib();
|
nuclear@12
|
380 }
|