Blame view

main.cu 22.7 KB
5ac53c9e   Jiaming Guo   add keyboardfunc
1
  #include <stdlib.h>
db598823   David Mayerich   fixed GCC errors ...
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
  #include <string>
  #include <fstream>
  #include <algorithm>
  
  //OpenGL includes
  #include <GL/glut.h>
  #include <GL/freeglut.h>
  
  //STIM includes
  #include <stim/visualization/gl_network.h>
  #include <stim/biomodels/network.h>
  #include <stim/visualization/gl_aaboundingbox.h>
  #include <stim/parser/arguments.h>
  #include <stim/visualization/camera.h>
  
f86a38d3   Jiaming Guo   add device choice...
17
18
19
20
21
  #ifdef __CUDACC__
  //CUDA includes
  #include <cuda.h>
  #endif
  
db598823   David Mayerich   fixed GCC errors ...
22
23
24
  //ANN includes
  //#include <ANN/ANN.h>
  
5ac53c9e   Jiaming Guo   add keyboardfunc
25
26
27
  //chrono includes
  //#include <chrono>
  
db598823   David Mayerich   fixed GCC errors ...
28
29
30
31
32
33
34
35
36
37
  //BOOST includes
  #include <boost/tuple/tuple.hpp>
  
  //visualization objects
  stim::gl_aaboundingbox<float> bb;			//axis-aligned bounding box object
  stim::camera cam;					//camera object
  
  unsigned num_nets = 0;
  stim::gl_network<float> GT;			//ground truth network
  stim::gl_network<float> T;			//test network
9627c6e6   Jiaming Guo   add splitting and...
38
39
40
41
  stim::gl_network<float> _GT;		//splitted GT
  stim::gl_network<float> _T;			//splitted T
  
  unsigned ind = 0;						//indicator of mapping
21f03d55   Jiaming Guo   add functions for...
42
  unsigned swc_ind = 0;					//indicator of rendering swc file as networks
9627c6e6   Jiaming Guo   add splitting and...
43
44
45
  
  std::vector<unsigned> _gt_t;								// store indices of nearest edge points in _T for _GT
  std::vector<unsigned> _t_gt;								// store indices of nearest edge points in _GT for _T
db598823   David Mayerich   fixed GCC errors ...
46
47
  
  //hard-coded parameters
f86a38d3   Jiaming Guo   add device choice...
48
49
50
  float resample_rate = 0.5f;			//sample rate for the network (fraction of sigma used as the maximum sample rate)
  float camera_factor = 1.2f;			//start point of the camera as a function of X and Y size
  float orbit_factor = 0.01f;			//degrees per pixel used to orbit the camera
db598823   David Mayerich   fixed GCC errors ...
51
  
9627c6e6   Jiaming Guo   add splitting and...
52
53
54
55
  //mouse click
  bool LButtonDown = false;			// true when left button down
  bool RButtonDown = false;
  
db598823   David Mayerich   fixed GCC errors ...
56
57
58
59
  //mouse position tracking
  int mouse_x;
  int mouse_y;
  
9627c6e6   Jiaming Guo   add splitting and...
60
61
62
  bool compareMode = true;			// default mode is compare mode
  bool mappingMode = false;
  
5ac53c9e   Jiaming Guo   add keyboardfunc
63
  // random color set
9627c6e6   Jiaming Guo   add splitting and...
64
65
  std::vector<float> colormap;
  
5ac53c9e   Jiaming Guo   add keyboardfunc
66
67
68
  // special key indicator
  int mods;
  
9627c6e6   Jiaming Guo   add splitting and...
69
70
71
72
  // create display lists
  GLuint dlist1;
  GLuint dlist2;
  
db598823   David Mayerich   fixed GCC errors ...
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
  //OpenGL objects
  GLuint cmap_tex = 0;				//texture name for the color map
  
  //sets an OpenGL viewport taking up the entire window
  void glut_render_single_projection(){
  
  	glMatrixMode(GL_PROJECTION);					//load the projection matrix for editing
  	glLoadIdentity();								//start with the identity matrix
  	int X = glutGet(GLUT_WINDOW_WIDTH);				//use the whole screen for rendering
  	int Y = glutGet(GLUT_WINDOW_HEIGHT);
  	glViewport(0, 0, X, Y);							//specify a viewport for the entire window
  	float aspect = (float)X / (float)Y;				//calculate the aspect ratio
  	gluPerspective(60, aspect, 0.1, 1000000);		//set up a perspective projection
  }
  
  //sets an OpenGL viewport taking up the left half of the window
  void glut_render_left_projection(){
  
  	glMatrixMode(GL_PROJECTION);					//load the projection matrix for editing
  	glLoadIdentity();								//start with the identity matrix
  	int X = glutGet(GLUT_WINDOW_WIDTH) / 2;			//only use half of the screen for the viewport
  	int Y = glutGet(GLUT_WINDOW_HEIGHT);
  	glViewport(0, 0, X, Y);							//specify the viewport on the left
  	float aspect = (float)X / (float)Y;				//calculate the aspect ratio
  	gluPerspective(60, aspect, 0.1, 1000000);		//set up a perspective projection
  }
  
  //sets an OpenGL viewport taking up the right half of the window
  void glut_render_right_projection(){
  
  	glMatrixMode(GL_PROJECTION);					//load the projection matrix for editing
  	glLoadIdentity();								//start with the identity matrix
  	int X = glutGet(GLUT_WINDOW_WIDTH) / 2;			//only use half of the screen for the viewport
  	int Y = glutGet(GLUT_WINDOW_HEIGHT);
  	glViewport(X, 0, X, Y);							//specify the viewport on the right
  	float aspect = (float)X / (float)Y;				//calculate the aspect ratio
  	gluPerspective(60, aspect, 0.1, 1000000);		//set up a perspective projection
  }
  
  void glut_render_modelview(){
  
  	glMatrixMode(GL_MODELVIEW);						//load the modelview matrix for editing
  	glLoadIdentity();								//start with the identity matrix
  	stim::vec3<float> eye = cam.getPosition();		//get the camera position (eye point)
  	stim::vec3<float> focus = cam.getLookAt();		//get the camera focal point
  	stim::vec3<float> up = cam.getUp();				//get the camera "up" orientation
  
  	gluLookAt(eye[0], eye[1], eye[2], focus[0], focus[1], focus[2], up[0], up[1], up[2]);	//set up the OpenGL camera
  }
  
db598823   David Mayerich   fixed GCC errors ...
123
124
125
  //draws the network(s)
  void glut_render(void) {
  
21f03d55   Jiaming Guo   add functions for...
126
127
128
129
130
131
132
133
134
135
  	//no mapping, just comparing
  	if (ind == 0) {	
  		//working with obj files
  		if (swc_ind == 0) {
  			if (num_nets == 1) {										//if a single network is loaded
  				glut_render_single_projection();						//fill the entire viewport
  				glut_render_modelview();								//set up the modelview matrix with camera details
  				glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);		//clear the screen
  				GT.glCenterline0();										//render the GT network (the only one loaded)
  			}
f7d360df   Jiaming Guo   resoloved conflicts
136
  
21f03d55   Jiaming Guo   add functions for...
137
  			if (num_nets == 2) {										//if two networks are loaded	
f7d360df   Jiaming Guo   resoloved conflicts
138
  
9627c6e6   Jiaming Guo   add splitting and...
139
140
141
142
143
144
145
146
  				glut_render_left_projection();							//set up a projection for the left half of the window
  				glut_render_modelview();								//set up the modelview matrix using camera details
  				glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);		//clear the screen
  
  				glEnable(GL_TEXTURE_1D);										//enable texture mapping
  				glTexEnvf(GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_REPLACE);		//texture map will be used as the network color
  				glBindTexture(GL_TEXTURE_1D, cmap_tex);							//bind the Brewer texture map
  
21f03d55   Jiaming Guo   add functions for...
147
  				GT.glCenterline();							//render the GT network
9627c6e6   Jiaming Guo   add splitting and...
148
149
150
  
  				glut_render_right_projection();							//set up a projection for the right half of the window
  				glut_render_modelview();								//set up the modelview matrix using camera details
21f03d55   Jiaming Guo   add functions for...
151
152
153
154
155
156
157
158
159
160
  				T.glCenterline();							//render the T network
  			}
  		}
  		//working with swc files
  		else {
  			if (num_nets == 1) {										//if a single network is loaded
  				glut_render_single_projection();						//fill the entire viewport
  				glut_render_modelview();								//set up the modelview matrix with camera details
  				glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);		//clear the screen
  				GT.glCenterline0_swc();									//render the GT network (the only one loaded)
f7d360df   Jiaming Guo   resoloved conflicts
161
  			}
21f03d55   Jiaming Guo   add functions for...
162
163
164
  
  			if (num_nets == 2) {										//if two networks are loaded	
  
f7d360df   Jiaming Guo   resoloved conflicts
165
166
167
  				glut_render_left_projection();							//set up a projection for the left half of the window
  				glut_render_modelview();								//set up the modelview matrix using camera details
  				glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);		//clear the screen
9627c6e6   Jiaming Guo   add splitting and...
168
  
21f03d55   Jiaming Guo   add functions for...
169
170
171
172
173
  				glEnable(GL_TEXTURE_1D);										//enable texture mapping
  				glTexEnvf(GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_REPLACE);		//texture map will be used as the network color
  				glBindTexture(GL_TEXTURE_1D, cmap_tex);							//bind the Brewer texture map
  
  				GT.glCenterline();							//render the GT network
9627c6e6   Jiaming Guo   add splitting and...
174
175
176
  
  				glut_render_right_projection();							//set up a projection for the right half of the window
  				glut_render_modelview();								//set up the modelview matrix using camera details
21f03d55   Jiaming Guo   add functions for...
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
  				T.glCenterline();							//render the T network
  			}
  		}
  	}
  
  	//do mapping
  	else {	
  		//working with obj files
  		if (swc_ind == 0) {
  			if (num_nets == 1) {											//if a single network is loaded
  				std::cout << "You should have at least two networks to do mapping." << std::endl;	//exit program because there isn't enough network
  				exit(1);
  			}
  			if (num_nets == 2) {											//if two networks are loaded
  				if (compareMode) {
  					glut_render_left_projection();							//set up a projection for the left half of the window
  					glut_render_modelview();								//set up the modelview matrix using camera details
  					glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);		//clear the screen
  
  					glEnable(GL_TEXTURE_1D);										//enable texture mapping
  					glTexEnvf(GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_REPLACE);		//texture map will be used as the network color
  					glBindTexture(GL_TEXTURE_1D, cmap_tex);							//bind the Brewer texture map
  
  					_GT.glCenterline();							//render the GT network
  
  					glut_render_right_projection();							//set up a projection for the right half of the window
  					glut_render_modelview();								//set up the modelview matrix using camera details
  					_T.glCenterline();							//render the T network
  
  				}
  				else {
  					glEnable(GL_DEPTH_TEST);
  					glut_render_left_projection();							//set up a projection for the left half of the window
  					glut_render_modelview();								//set up the modelview matrix using camera details
  					glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);		//clear the screen
  
  																			//_GT.glRandColorCenterlineGT(dlist1, _gt_t, colormap);
  					_GT.glCylinderGT(dlist1, _gt_t, colormap);
  
  					glut_render_right_projection();							//set up a projection for the right half of the window
  					glut_render_modelview();								//set up the modelview matrix using camera details
  																			//_T.glRandColorCenterlineT(dlist2, _t_gt, colormap);
  					_T.glCylinderT(dlist2, _t_gt, colormap);
  				}
9627c6e6   Jiaming Guo   add splitting and...
221
222
  			}
  		}
db598823   David Mayerich   fixed GCC errors ...
223
224
  	}
  
f7d360df   Jiaming Guo   resoloved conflicts
225
  	if (num_nets == 2) {
5ac53c9e   Jiaming Guo   add keyboardfunc
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
  		std::ostringstream ss;
  		if (mappingMode)											// if it is in mapping mode
  			ss << "Mapping Mode";
  		else
  			ss << "Compare Mode";								// default mode is compare mode
  
  		glDisable(GL_TEXTURE_1D);
  		glMatrixMode(GL_PROJECTION);										//Set up the 2d viewport for mode text printing
  		glPushMatrix();
  		glLoadIdentity();
  		int X = glutGet(GLUT_WINDOW_WIDTH);
  		int Y = glutGet(GLUT_WINDOW_HEIGHT);
  		glViewport(0, 0, X / 2, Y);											// locate to left bottom corner
  		gluOrtho2D(0, X, 0, Y);												// define othogonal aspect
  		glColor3f(0.0, 1.0, 0.0);											// using green to show mode
  
  		glMatrixMode(GL_MODELVIEW);
  		glPushMatrix();
  		glLoadIdentity();
  
f7d360df   Jiaming Guo   resoloved conflicts
246
247
  		glRasterPos2f(0, 5);											//print text in the bottom left corner
  		glutBitmapString(GLUT_BITMAP_TIMES_ROMAN_24, (const unsigned char*)(ss.str().c_str()));
5ac53c9e   Jiaming Guo   add keyboardfunc
248
249
250
251
252
  
  		glPopMatrix();
  		glMatrixMode(GL_PROJECTION);
  		glPopMatrix();
  	}
db598823   David Mayerich   fixed GCC errors ...
253
254
255
256
257
258
  	glutSwapBuffers();
  }
  
  // defines camera motion based on mouse dragging
  void glut_motion(int x, int y){
  	
5ac53c9e   Jiaming Guo   add keyboardfunc
259
  	if(LButtonDown == true && RButtonDown == false && mods != GLUT_ACTIVE_CTRL){
db598823   David Mayerich   fixed GCC errors ...
260
261
262
263
264
265
266
267
268
269
  
  	float theta = orbit_factor * (mouse_x - x);		//determine the number of degrees along the x-axis to rotate
  	float phi = orbit_factor * (y - mouse_y);		//number of degrees along the y-axis to rotate
  
  	cam.OrbitFocus(theta, phi);						//rotate the camera around the focal point
  
  	mouse_x = x;									//update the mouse position
  	mouse_y = y;
  		
  	glutPostRedisplay();							//re-draw the visualization
9627c6e6   Jiaming Guo   add splitting and...
270
  	}
db598823   David Mayerich   fixed GCC errors ...
271
272
273
274
  }
  
  // sets the mouse position when clicked
  void glut_mouse(int button, int state, int x, int y){
9627c6e6   Jiaming Guo   add splitting and...
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
  	
  	if(button == GLUT_LEFT_BUTTON && state == GLUT_DOWN){
  		mouse_x = x;
  		mouse_y = y;
  		LButtonDown = true;
  	}
  	else if(button == GLUT_RIGHT_BUTTON && state == GLUT_DOWN){
  		mouse_x = x;
  		mouse_y = y;
  		RButtonDown = true;
  	}
  	else if(button == GLUT_LEFT_BUTTON && state == GLUT_UP){
  		mouse_x = x;
  		mouse_y = y;
  		LButtonDown = false;
  	}
  	else if(button == GLUT_RIGHT_BUTTON && state == GLUT_UP){
  		mouse_x = x;
  		mouse_y = y;
  		RButtonDown = false;
  	}
5ac53c9e   Jiaming Guo   add keyboardfunc
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
  
  	/// implementation of mouse click mapping feedback
  	mods = glutGetModifiers();											// get modifier keys
  	if (mods == GLUT_ACTIVE_CTRL) 										// if the CTRL key is pressed
  		if (button == GLUT_LEFT_BUTTON && state == GLUT_DOWN) {
  			std::cout << "( " << x << ", " << y << " )" << std::endl;	// if the CTRL key is pressed and LEFT BUTTON is DOWN, print the window coordinates
  
  			GLint    viewport[4];
  			GLdouble modelview[16];
  			GLdouble projection[16];
  			GLdouble winX, winY, winZ;
  			GLdouble posX, posY, posZ;
  
  			glGetIntegerv(GL_VIEWPORT, viewport);
  			glGetDoublev(GL_MODELVIEW_MATRIX, modelview);
  			glGetDoublev(GL_PROJECTION_MATRIX, projection);
  
  			winX = (GLdouble)x;
  			winY = viewport[3] - (GLdouble)y;
  			glReadPixels((GLint)winX, (GLint)winY, (GLsizei)1, (GLsizei)1, GL_DEPTH_COMPONENT, GL_FLOAT, &winZ);		// need frame buffer FBO
  			gluUnProject(winX, winY, winZ, modelview, projection, viewport, &posX, &posY, &posZ);						// not sure why it should add 1 to the winZ
  			
  			std::cout << "( " << posX << ", " << posY << ", "<< posZ <<" )" << std::endl;
  		}
9627c6e6   Jiaming Guo   add splitting and...
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
  }
  
  void glut_keyboard(unsigned char key, int x, int y){
  	if(key == 'm')											// press m to change mode
  	{
  		if(compareMode && !mappingMode){					// if it is in compare mode
  			compareMode = false;
  			mappingMode = true;
  		}
  		else{												// if it is in mapping mode
  			compareMode = true;
  			mappingMode = false;
  		}
  	}
  	glutPostRedisplay();
db598823   David Mayerich   fixed GCC errors ...
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
  }
  
  #define BREWER_CTRL_PTS 11							//number of control points in the Brewer map
  void texture_initialize(){
  
  	//define the colormap
  	static float  brewer_map[BREWER_CTRL_PTS][3] = {			//generate a Brewer color map (blue to red)
  		{0.192157f, 0.211765f, 0.584314f},
  		{0.270588f, 0.458824f, 0.705882f},
  		{0.454902f, 0.678431f, 0.819608f},
  		{0.670588f, 0.85098f, 0.913725f},
  		{0.878431f, 0.952941f, 0.972549f},
  		{1.0f, 1.0f, 0.74902f},
  		{0.996078f, 0.878431f, 0.564706f},
  		{0.992157f, 0.682353f, 0.380392f},
  		{0.956863f, 0.427451f, 0.262745f},
  		{0.843137f, 0.188235f, 0.152941f},
  		{0.647059f, 0.0f, 0.14902f}
  	};
  
  	glGenTextures(1, &cmap_tex);								//generate a texture map name
  	glBindTexture(GL_TEXTURE_1D, cmap_tex);						//bind the texture map
  
  	glTexParameteri(GL_TEXTURE_1D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);		//enable linear interpolation
  	glTexParameteri(GL_TEXTURE_1D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
  	glTexParameteri(GL_TEXTURE_1D, GL_TEXTURE_WRAP_S, GL_CLAMP);			//clamp the values at the minimum and maximum
  	glTexImage1D(GL_TEXTURE_1D, 0, 3, BREWER_CTRL_PTS, 0, GL_RGB, GL_FLOAT,	//upload the texture map to the GPU
  					brewer_map);
  }
  
  //Initialize the OpenGL (GLUT) window, including starting resolution, callbacks, texture maps, and camera
  void glut_initialize(){
  	
  	int myargc = 1;					//GLUT requires arguments, so create some bogus ones
  	char* myargv[1];
  	myargv [0]=strdup ("netmets");
  
  	glutInit(&myargc, myargv);									//pass bogus arguments to glutInit()
  	glutInitDisplayMode(GLUT_DEPTH | GLUT_DOUBLE | GLUT_RGBA);	//generate a color buffer, depth buffer, and enable double buffering
  	glutInitWindowPosition(100,100);							//set the initial window position
5ac53c9e   Jiaming Guo   add keyboardfunc
375
  	glutInitWindowSize(320, 320);								//set the initial window size
db598823   David Mayerich   fixed GCC errors ...
376
377
378
379
380
381
382
  	glutCreateWindow("NetMets - STIM Lab, UH");					//set the dialog box title
  
  	
  	// register callback functions
  	glutDisplayFunc(glut_render);			//function executed for rendering - renders networks
  	glutMouseFunc(glut_mouse);				//executed on a mouse click - sets starting mouse positions for rotations
  	glutMotionFunc(glut_motion);			//executed when the mouse is moved while a button is pressed
9627c6e6   Jiaming Guo   add splitting and...
383
384
  	if(ind == 1)							//only in mapping mode, keyboard will be used
  		glutKeyboardFunc(glut_keyboard);
db598823   David Mayerich   fixed GCC errors ...
385
  
9627c6e6   Jiaming Guo   add splitting and...
386
  	texture_initialize();									//set up texture mapping (create texture maps, enable features)
db598823   David Mayerich   fixed GCC errors ...
387
388
389
390
391
392
  
  	stim::vec3<float> c = bb.center();		//get the center of the network bounding box
  
  	//place the camera along the z-axis at a distance determined by the network size along x and y
  	cam.setPosition(c + stim::vec<float>(0, 0, camera_factor * std::max(bb.size()[0], bb.size()[1])));
  	cam.LookAt(c[0], c[1], c[2]);						//look at the center of the network
db598823   David Mayerich   fixed GCC errors ...
393
394
  }
  
f86a38d3   Jiaming Guo   add device choice...
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
  #ifdef __CUDACC__
  void setdevice(int &device){
  	int count;
  	cudaGetDeviceCount(&count);				// numbers of device that are available
  	if(count < device + 1){
  	std::cout<<"No such device available, please set another device"<<std::endl;
  	exit(1);
  	}
  }
  #else
  void setdevice(int &device){
  	device = -1;
  }
  #endif
  
db598823   David Mayerich   fixed GCC errors ...
410
  //compare both networks and fill the networks with error information
f86a38d3   Jiaming Guo   add device choice...
411
  void compare(float sigma, int device){
db598823   David Mayerich   fixed GCC errors ...
412
  
f86a38d3   Jiaming Guo   add device choice...
413
414
  	GT = GT.compare(T, sigma, device);						//compare the ground truth to the test case - store errors in GT
      T = T.compare(GT, sigma, device);						//compare the test case to the ground truth - store errors in T
db598823   David Mayerich   fixed GCC errors ...
415
416
  
  	//calculate the metrics
691fa079   David Mayerich   initial pass adap...
417
418
  	float FPR = GT.average();						//calculate the metrics
  	float FNR = T.average();
db598823   David Mayerich   fixed GCC errors ...
419
420
421
422
423
  	
  	std::cout << "FNR: " << FPR << std::endl;		//print false alarms and misses
  	std::cout << "FPR: " << FNR << std::endl;
  }
  
21f03d55   Jiaming Guo   add functions for...
424
  void map(float sigma, int device, float threshold){
9627c6e6   Jiaming Guo   add splitting and...
425
  
21f03d55   Jiaming Guo   add functions for...
426
427
  	_GT.split(GT, T, sigma, device, threshold);
  	_T.split(T, GT, sigma, device, threshold);
9627c6e6   Jiaming Guo   add splitting and...
428
  
21f03d55   Jiaming Guo   add functions for...
429
430
  	_GT.mapping(_T, _gt_t, device, threshold);
  	_T.mapping(_GT, _t_gt, device, threshold);
9627c6e6   Jiaming Guo   add splitting and...
431
432
  
  	size_t num = _gt_t.size();							// also create random color for unmapping edge, but won't be used though
5ac53c9e   Jiaming Guo   add keyboardfunc
433
434
  	colormap.resize(3 * num);							// 3 portions compound RGB
  	for(int i = 0; i < 3 * num; i++)
9627c6e6   Jiaming Guo   add splitting and...
435
436
437
438
439
440
441
442
443
444
  		colormap[i] = rand()/(float)RAND_MAX;
  	
  	//calculate the metrics
  	float FPR = _GT.average(0);						//calculate the metrics
  	float FNR = _T.average(0);
  	
  	std::cout << "FNR: " << FPR << std::endl;		//print false alarms and misses
  	std::cout << "FPR: " << FNR << std::endl;
  }
  
db598823   David Mayerich   fixed GCC errors ...
445
446
447
  // writes features of the networks i.e average segment length, tortuosity, branching index, contraction, fractal dimension, number of end and branch points to a csv file
  // Pranathi wrote this - saves network features to a CSV file
  void features(std::string filename){
f86a38d3   Jiaming Guo   add device choice...
448
  		double avgL_t, avgL_gt, avgT_t, avgT_gt, avgB_t, avgB_gt, avgC_t, avgC_gt, avgFD_t, avgFD_gt;
db598823   David Mayerich   fixed GCC errors ...
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
  		unsigned int e_t, e_gt, b_gt, b_t;
  		avgL_gt = GT.Lengths();
  		avgT_gt = GT.Tortuosities();
  		avgL_t = T.Lengths();
  		avgT_t = T.Tortuosities();
  		avgB_gt = GT.BranchingIndex();
  		avgB_t = T.BranchingIndex();
  		avgC_gt = GT.Contractions();
  		avgFD_gt = GT.FractalDimensions();
  		avgC_t = T.Contractions();
  		avgFD_t = T.FractalDimensions();
  		e_gt = GT.EndP();
  		e_t = T.EndP();
  		b_gt = GT.BranchP();
  		b_t = T.BranchP();
  		std::ofstream myfile;
  		myfile.open (filename.c_str());
  		myfile << "Length, Tortuosity, Contraction, Fractal Dimension, Branch Points, End points, Branching Index, \n";
  		myfile << avgL_gt << "," << avgT_gt << "," << avgC_gt << "," << avgFD_gt << "," << b_gt << "," << e_gt << "," << avgB_gt <<std::endl;
  		myfile << avgL_t << "," << avgT_t << "," << avgC_t << "," << avgFD_t << "," << b_t << "," << e_t << "," << avgB_t <<std::endl;
  		myfile.close();
  }
  
  // Output an advertisement for the lab, authors, and usage information
  void advertise(){
  	std::cout<<std::endl<<std::endl;
  	std::cout<<"========================================================================="<<std::endl;
  	std::cout<<"Thank you for using the NetMets network comparison tool!"<<std::endl;
  	std::cout<<"Scalable Tissue Imaging and Modeling (STIM) Lab, University of Houston"<<std::endl;
21f03d55   Jiaming Guo   add functions for...
478
  	std::cout<<"Developers: Pranathi Vemuri, David Mayerich, Jiaming Guo"<<std::endl;
9627c6e6   Jiaming Guo   add splitting and...
479
  	std::cout<<"Source: https://git.stim.ee.uh.edu/segmentation/netmets" <<std::endl;
db598823   David Mayerich   fixed GCC errors ...
480
481
  	std::cout<<"========================================================================="<<std::endl<<std::endl;
  
f86a38d3   Jiaming Guo   add device choice...
482
483
484
485
486
487
  	std::cout<<"usage: netmets file1 file2 --sigma 3"<<std::endl;
  	std::cout<<"            compare two files with a tolerance of 3 (units defined by the network)"<<std::endl<<std::endl;
  	std::cout<<"       netmets file1 --gui"<<std::endl;
  	std::cout<<"            load a file and display it using OpenGL"<<std::endl<<std::endl;
  	std::cout<<"       netmets file1 file2 --device 0"<<std::endl;
  	std::cout<<"            compare two files using device 0 (if there isn't a gpu, use cpu)"<<std::endl<<std::endl;
21f03d55   Jiaming Guo   add functions for...
488
489
  	std::cout<<"       netmets file1 file2 --mapping value"<<std::endl;
  	std::cout<<"            mapping two files in random colors with a threshold of value"<<std::endl<<std::endl;
db598823   David Mayerich   fixed GCC errors ...
490
491
492
493
494
495
496
497
  }
  
  int main(int argc, char* argv[])
  {
  	stim::arglist args;						//create an instance of arglist
  
  	//add arguments
  	args.add("help", "prints this help");
f86a38d3   Jiaming Guo   add device choice...
498
  	args.add("sigma", "force a sigma value to specify the tolerance of the network comparison", "3");
db598823   David Mayerich   fixed GCC errors ...
499
  	args.add("gui", "display the network or network comparison using OpenGL");
f86a38d3   Jiaming Guo   add device choice...
500
  	args.add("device", "choose specific device to run", "0");
db598823   David Mayerich   fixed GCC errors ...
501
  	args.add("features", "save features to a CSV file, specify file name");
9627c6e6   Jiaming Guo   add splitting and...
502
  	args.add("mapping", "mapping input according to similarity");
21f03d55   Jiaming Guo   add functions for...
503
  	args.add("swc", "load swc file instead of obj file");
db598823   David Mayerich   fixed GCC errors ...
504
505
506
507
508
509
510
511
512
  
  	args.parse(argc, argv);					//parse the user arguments
  
  	if(args["help"].is_set() || args.nargs() == 0){			//test for help
  		advertise();										//output the advertisement
  		std::cout<<args.str();								//output arguments
  		exit(1);											//exit
  	}
  	
21f03d55   Jiaming Guo   add functions for...
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
  	if (args["swc"].is_set()) {				// if it is to load a swc file, right now we only load two files(.swc and .obj)
  		if (args.nargs() >= 1) {			// if at least one network file is specified
  			num_nets = 1;					// set the number of networks to one
  			GT.load_swc(args.arg(0));		// load the specified file as the ground truth
  		}
  		if (args.nargs() == 2) {			//if two files are specified, they will be displayed in neighboring viewports and compared
  			int device = args["device"].as_int();				//get the device value from the user
  			num_nets = 2;										//set the number of networks to two
  			//does it need to be resampled??
  			float sigma = args["sigma"].as_float();				//get the sigma value from the user
  			T.load_swc(args.arg(1));                            //load the second (test) network
  			if (args["features"].is_set())						//if the user wants to save features
  				features(args["features"].as_string());
  			GT = GT.resample(resample_rate * sigma);			//resample both networks based on the sigma value
  			T = T.resample(resample_rate * sigma);
  			if (args["mapping"].is_set()) {
  				std::cout << "right now networks that are loaded from swc files do not need to be mapped with each other!" << std::endl;
  				exit(1);
  			}
  			else
  				compare(sigma, device);							//run the comparison algorithm
  		}
db598823   David Mayerich   fixed GCC errors ...
535
536
  	}
  	
21f03d55   Jiaming Guo   add functions for...
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
  	else {									//if it is to load a obj file	
  		if (args.nargs() >= 1) {			//if at least one network file is specified
  			num_nets = 1;					//set the number of networks to one
  			GT.load_obj(args.arg(0));		//load the specified file as the ground truth
  			/*GT.to_txt("Graph.txt");*/
  		}
  		if (args.nargs() == 2) {								//if two files are specified, they will be displayed in neighboring viewports and compared
  			int device = args["device"].as_int();				//get the device value from the user
  			num_nets = 2;										//set the number of networks to two
  			float sigma = args["sigma"].as_float();				//get the sigma value from the user
  			T.load_obj(args.arg(1));                            //load the second (test) network
  			if (args["features"].is_set())						//if the user wants to save features
  				features(args["features"].as_string());
  			GT = GT.resample(resample_rate * sigma);			//resample both networks based on the sigma value
  			T = T.resample(resample_rate * sigma);
  			if (args["mapping"].is_set()) {
  				float threshold = args["mapping"].as_float();
  				map(sigma, device, threshold);
  			}
  			else
  				compare(sigma, device);										//run the comparison algorithm
9627c6e6   Jiaming Guo   add splitting and...
558
  		}
db598823   David Mayerich   fixed GCC errors ...
559
560
561
  	}
  
  	//if a GUI is requested, display the network using OpenGL
9627c6e6   Jiaming Guo   add splitting and...
562
  	if(args["gui"].is_set()){
21f03d55   Jiaming Guo   add functions for...
563
564
565
566
567
  		if (args["swc"].is_set()) {
  			swc_ind = 1;
  		}
  		if (args["mapping"].is_set()) {
  			ind = 1;								//set indicator of mapping to 1(true)
9627c6e6   Jiaming Guo   add splitting and...
568
569
  			bb = _GT.boundingbox();					//generate a bounding volume		
  			glut_initialize();						//create the GLUT window and set callback functions		
21f03d55   Jiaming Guo   add functions for...
570
  			glutMainLoop();							//enter GLUT event processing cycle
9627c6e6   Jiaming Guo   add splitting and...
571
  		}
21f03d55   Jiaming Guo   add functions for...
572
  		else {
9627c6e6   Jiaming Guo   add splitting and...
573
574
  			bb = GT.boundingbox();					//generate a bounding volume		
  			glut_initialize();						//create the GLUT window and set callback functions		
21f03d55   Jiaming Guo   add functions for...
575
  			glutMainLoop();							//enter GLUT event processing cycle
9627c6e6   Jiaming Guo   add splitting and...
576
577
  		}
  	}
db598823   David Mayerich   fixed GCC errors ...
578
  }