Home My Page Projects Code Snippets Project Openings diderot
Summary Activity Tracker Tasks SCM

SCM Repository

[diderot] Annotation of /examples/vr-demo/main.c
ViewVC logotype

Annotation of /examples/vr-demo/main.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2061 - (view) (download) (as text)

1 : jhr 2058 /*! \file main.c
2 :     *
3 :     * \author John Reppy
4 :     *
5 :     * Main program for Diderot VR viewer.
6 :     */
7 :    
8 :     /*
9 :     * COPYRIGHT (c) 2012 The Diderot Project (http://diderot-language.cs.uchicago.edu)
10 :     * All rights reserved.
11 :     */
12 :    
13 :     #include "GL/glfw.h"
14 :     #include "util.h"
15 :     #include "vr-lite.h"
16 :     #include <unistd.h>
17 :    
18 : jhr 2061 /* minimum and maximum values from */
19 :     #define MIN_VALUE 0
20 :     #define MAX_VALUE 2632
21 :    
22 :     /* half the width of the opacity range */
23 :     #define OPAC_HALF_WID 150
24 :    
25 :     #define MIN_OPAC_MID ((float)(MIN_VALUE+OPAC_HALF_WID))
26 :     #define MAX_OPAC_MID ((float)(MAX_VALUE-OPAC_HALF_WID))
27 :    
28 : jhr 2058 /* pixel dimensions of output */
29 : jhr 2061 #define WIDTH 480
30 :     #define HEIGHT 345
31 : jhr 2058
32 :     /* world bounds */
33 :     #define WRLD_MIN_X 0.0
34 :     #define WRLD_MIN_Y 0.0
35 :     #define WRLD_MAX_X 1.0
36 :     #define WRLD_MAX_Y 1.0
37 :     #define WRLD_CENTER_X 0.5
38 :     #define WRLD_CENTER_Y 0.5
39 :    
40 :     #define STEP_SIZE 1
41 :     #define SLEEP_MS 250
42 :    
43 :     static inline void CheckError ()
44 :     {
45 :     GLenum errCode;
46 :     if ((errCode = glGetError()) != GL_NO_ERROR) {
47 :     fprintf(stderr, "OpenGL error: %s\n", gluErrorString(errCode));
48 :     exit (1);
49 :     }
50 :     }
51 :    
52 :     /***** Globals for viewing, etc. *****/
53 :    
54 :     Nrrd *Diderot; // input image nrrd
55 :     unsigned int Width; // view window width
56 :     unsigned int Height; // view window height
57 :     GLuint ImageTexture; // Texture ID of background image
58 :    
59 :    
60 : jhr 2061
61 : jhr 2058 // Callback function called by GLUT to render screen
62 :     void Draw (Nrrd *img)
63 :     {
64 :     // Clear frame buffer
65 :     glClearColor (0, 0, 0, 1);
66 :     glClear (GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
67 :    
68 :     glDisable(GL_DEPTH_TEST);
69 :     glDisable(GL_CULL_FACE);
70 :    
71 :     // Draw image
72 :     glEnable (GL_TEXTURE_2D);
73 :     glActiveTexture (GL_TEXTURE0);
74 :     glBindTexture (GL_TEXTURE_2D, ImageTexture);
75 :     glTexEnvf (GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_REPLACE);
76 :     glBegin (GL_QUADS);
77 :     glColor3f (0.1, 0.1, 0.1);
78 :     glTexCoord2f (0.0, 0.0);
79 :     glVertex2f (WrldOrigin[0], WrldOrigin[1]);
80 :     glTexCoord2f (1.0, 0.0);
81 :     glVertex2f (WrldOrigin[0]+WrldDim[0], WrldOrigin[1]);
82 :     glTexCoord2f (1.0, 1.0);
83 :     glVertex2f (WrldOrigin[0]+WrldDim[0], WrldOrigin[1]+WrldDim[1]);
84 :     glTexCoord2f (0.0, 1.0);
85 :     glVertex2f (WrldOrigin[0], WrldOrigin[1]+WrldDim[1]);
86 :     glEnd();
87 :     glDisable (GL_TEXTURE_2D);
88 :     CheckError ();
89 :    
90 :     // Draw current particle state
91 :     glEnable (GL_POINT_SMOOTH);
92 :     // glPointSize (2.0);
93 :     // it would be more efficient to do this as a single buffer, but it is fast enough
94 :     float *p = (float *)(positions->data);
95 :     unsigned int nPoints = positions->axis[1].size;
96 :     glBegin (GL_POINTS);
97 :     // first draw some points at known locations for testing purposes
98 :     // glColor3f (1.0, 0.0, 0.0);
99 :     // glVertex2f (WRLD_MIN_X, WRLD_MIN_Y);
100 :     // glVertex2f (WRLD_MAX_X, WRLD_MIN_Y);
101 :     // glVertex2f (WRLD_MAX_X, WRLD_MAX_Y);
102 :     // glVertex2f (WRLD_MIN_X, WRLD_MAX_Y);
103 :     glColor3f (0.2, 1.0, 0.2);
104 :     for (int i = 0; i < nPoints; i++) {
105 :     glVertex2fv(p);
106 :     p += 2;
107 :     }
108 :     glEnd ();
109 :    
110 :     // Draw tweak bars
111 :     TwDraw();
112 :    
113 :     // Present frame buffer
114 :     glfwSwapBuffers();
115 :     }
116 :    
117 :     // Callback function called by GLFW when window size changes
118 :     void GLFWCALL WindowSizeCB (int width, int height)
119 :     {
120 :     printf("window size = %d x %d\n", width, height);
121 :     // Set OpenGL viewport and camera
122 :     glViewport(0, 0, width, height);
123 :    
124 :     glMatrixMode (GL_PROJECTION);
125 :     glLoadIdentity ();
126 :     double pad = 0.05*(WRLD_MAX_X - WRLD_MIN_X); // add 10% to world dimensions
127 :     gluOrtho2D (
128 :     WRLD_MIN_X - pad, WRLD_MAX_X + pad,
129 :     WRLD_MAX_Y + pad, WRLD_MIN_Y - pad);
130 :    
131 :     glMatrixMode (GL_MODELVIEW);
132 :     glLoadIdentity ();
133 :     gluLookAt (
134 :     // WRLD_CENTER_X, WRLD_CENTER_Y, 1.0,
135 :     0.0, 0.0, 1.0,
136 :     // WRLD_CENTER_X, WRLD_CENTER_Y, 0.0,
137 :     0.0, 0.0, 0.0,
138 :     0.0, 1.0, 0.0);
139 :    
140 :     // Send the new window size to AntTweakBar
141 :     TwWindowSize(width, height);
142 :    
143 :     // remember width and height
144 :     Width = width;
145 :     Height = height;
146 :    
147 :     }
148 :    
149 :     // initialize a texture from a grayscale Nrrd file
150 :     void InitTexture ()
151 :     {
152 :     Image2D_t *img = LoadImage ("data/ddro-tx.png", true, GRAY_IMAGE);
153 :     if (img == 0) {
154 :     fprintf (stderr, "unable to load texture file\n");
155 :     exit (1);
156 :     }
157 :    
158 :     // generate the TeXID
159 :     GLuint texId;
160 :     glGenTextures (1, &texId);
161 :    
162 :     // load the texture data
163 :     glBindTexture (GL_TEXTURE_2D, texId);
164 :     int sts = TexImage (img);
165 :     if (sts != GL_NO_ERROR) {
166 :     fprintf(stderr, "error loading texture\n");
167 :     exit (1);
168 :     }
169 :     glTexParameteri (GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
170 :     glTexParameteri (GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
171 :    
172 :     FreeImage (img);
173 :    
174 :     ImageTexture = texId;
175 :     }
176 :    
177 :     void TW_CALL StartButtonCB (void *data)
178 :     {
179 :     State_t *st = (State_t *)data;
180 :    
181 :     if (st->running == 0) {
182 :     st->running = 1;
183 :     TwSetParam (st->ctls, "StartButton", "label", TW_PARAM_CSTRING, 1, "Stop");
184 :     }
185 :     else {
186 :     st->running = 0;
187 :     TwSetParam (st->ctls, "StartButton", "label", TW_PARAM_CSTRING, 1, "Start");
188 :     }
189 :     }
190 :    
191 : jhr 2061 void GetData (VR_World_t *wrld, Nrrd *nData)
192 : jhr 2058 {
193 :     // get snapshot of state
194 : jhr 2061 if (VR_Snapshot_pos (wrld, nData)) {
195 : jhr 2058 // error
196 :     fprintf(stderr, "Error getting nrrd data\n");
197 :     exit(1);
198 :     }
199 :     }
200 :    
201 :     int main (int argc, const char **argv)
202 :     {
203 :     // Initialize GLFW
204 :     if (0 == glfwInit()) {
205 :     // An error occured
206 :     fprintf(stderr, "GLFW initialization failed\n");
207 :     return 1;
208 :     }
209 :    
210 :     GLFWvidmode mode;
211 :     glfwGetDesktopMode (&mode);
212 :     if (0 == glfwOpenWindow(WIDTH, HEIGHT, mode.RedBits, mode.GreenBits, mode.BlueBits, 0, 0, 0, GLFW_WINDOW) ) {
213 :     // A fatal error occured
214 :     fprintf(stderr, "Cannot open GLFW window\n");
215 :     glfwTerminate();
216 :     return 1;
217 :     }
218 :     glfwEnable(GLFW_KEY_REPEAT);
219 : jhr 2061 glfwSetWindowTitle("Diderot VR Demo");
220 : jhr 2058
221 :     // Set GLFW event callbacks
222 :     glfwSetWindowSizeCallback (WindowSizeCB);
223 : jhr 2061 glfwSetKeyCallback ((GLFWkeyfun)TwEventKeyGLFW);
224 : jhr 2058
225 :     // load Diderot image for background texture
226 :     InitTexture ();
227 :    
228 :     // setup the world coords.
229 :     // FIXME: we should get these from the nrrd file, but for now I'm hard coding them
230 :     WrldOrigin[0] = -0.0184375;
231 :     WrldOrigin[1] = -0.0184375;
232 :     WrldDim[0] = 0.013125 * 80.0;
233 :     WrldDim[1] = 0.013125 * 80.0;
234 :    
235 :     // initialize the Diderot program
236 : jhr 2061 VR_World_t *wrld = VR_Init ();
237 : jhr 2058
238 :     // nrrd for getting computational state
239 : jhr 2061 Nrrd *nRGB = nrrdNew();
240 : jhr 2058
241 :     // get and render initial state
242 :     GetData (wrld, nData);
243 :     Draw (nData);
244 :    
245 :     // Main loop (repeated while window is not closed and [ESC] is not pressed)
246 :     while (glfwGetWindowParam(GLFW_OPENED) && !glfwGetKey(GLFW_KEY_ESC)) {
247 : jhr 2061 if (NeedsRecompute) {
248 :     // recompute
249 :     VR_Initially (wrld);
250 :     VR_Run (wrld, 0);
251 :     // get output image
252 :     if (VR_OutputGet_outRGBA (wrld, nRGB)) {
253 :     // error
254 :     fprintf(stderr, "Error getting nrrd data\n");
255 :     exit(1);
256 :     }
257 :     }
258 :     if (NeedsRedraw) {
259 :     }
260 :     glfwWaitEvents();
261 : jhr 2058 }
262 :    
263 :     // shutdown the world
264 : jhr 2061 VR_Shutdown (wrld);
265 : jhr 2058
266 :     // Terminate AntTweakBar and GLFW
267 :     TwTerminate();
268 :     glfwTerminate();
269 :    
270 :     return 0;
271 :     }

root@smlnj-gforge.cs.uchicago.edu
ViewVC Help
Powered by ViewVC 1.0.0