added color-mapped networks, parallel viewing of two networks

David Mayerich
1 parent bbb519e6
Showing 1 changed file with 171 additions and 119 deletions Show diff stats
main.cpp
@@ -8,7 +8,7 @@
 //STIM includes
 #include <stim/visualization/gl_network.h>
-#include <stim/visualization/gl_aabb.h>
+#include <stim/visualization/gl_aaboundingbox.h>
 #include <stim/parser/arguments.h>
 #include <stim/visualization/camera.h>
@@ -18,62 +18,117 @@
 //BOOST includes
 #include <boost/tuple/tuple.hpp>
-stim::gl_aabb<float> bb;
-stim::camera cam;
+//visualization objects
+stim::gl_aaboundingbox<float> bb;			//axis-aligned bounding box object
+stim::camera cam;					//camera object
-stim::gl_network<float> GT;
-stim::gl_network<float> T;
+unsigned num_nets = 0;
+stim::gl_network<float> GT;			//ground truth network
+stim::gl_network<float> T;			//test network
 //hard-coded parameters
-float resample_rate = 0.5;
+float resample_rate = 0.5;			//sample rate for the network (fraction of sigma used as the maximum sample rate)
+float camera_factor = 1.2;			//start point of the camera as a function of X and Y size
+float orbit_factor = 0.01;			//degrees per pixel used to orbit the camera
 //mouse position tracking
 int mouse_x;
 int mouse_y;
+//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::vec<float> eye = cam.getPosition();		//get the camera position (eye point)
+	stim::vec<float> focus = cam.getLookAt();		//get the camera focal point
+	stim::vec<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
+}
+
+//draws the network(s)
 void glut_render(void) {
-	glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
+	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.glCenterline(GT.nmags() - 1);						//render the GT network (the only one loaded)
+	}
-	glMatrixMode(GL_MODELVIEW);
-	glLoadIdentity();
-	stim::vec<float> eye = cam.getPosition();
-	stim::vec<float> focus = cam.getLookAt();
-	stim::vec<float> up = cam.getUp();
+	if(num_nets == 2){											//if two networks are loaded	
-	gluLookAt(eye[0], eye[1], eye[2], focus[0], focus[1], focus[2], up[0], up[1], up[2]);
+		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
-	// render the bounding box
-	glColor3f(0.0, 0.0, 1.0);
-	bb.glWire();
-	
+		GT.glCenterline(GT.nmags() - 1);						//render the GT network
-	//render the GT network (red)
-	glColor3f(1.0, 0.0, 0.0);
-	GT.glCenterline();
+		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(T.nmags() - 1);							//render the T network
-	//render the T network (green)
-	glColor3f(0.0, 1.0, 0.0);
-	T.glCenterline();
+	}
 	glutSwapBuffers();
 }
 // defines camera motion based on mouse dragging
 void glut_motion(int x, int y){
-	float factor = 0.01;
+	
-	float theta = factor * (mouse_x - x);
-	float phi = factor * (y - mouse_y);
+	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);
+	cam.OrbitFocus(theta, phi);						//rotate the camera around the focal point
-	mouse_x = x;
+	mouse_x = x;									//update the mouse position
 	mouse_y = y;
-
-	glutPostRedisplay();
+		
+	glutPostRedisplay();							//re-draw the visualization
 }
 // sets the mouse position when clicked
@@ -82,19 +137,78 @@ void glut_mouse(int button, int state, int x, int y){
 	mouse_y = y;
 }
-// re-calculates the projection matrix if the window is reshaped
-void glut_reshape(int x, int y){
+#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
+	glutInitWindowSize(320,320);								//set the initial window size
+	glutCreateWindow("NetMets - STIM Lab, UH");					//set the dialog box title
-	glMatrixMode(GL_PROJECTION);
-	glLoadIdentity();
-	glViewport(0, 0, x, y);
-	float aspect = (float)x / (float)y;
-	gluPerspective(60, aspect, 0.1, 1000000);
+	
+	// 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
+
+	texture_initialize();					//set up texture mapping (create texture maps, enable features)
+
+	stim::vec<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
 }
+//compare both networks and fill the networks with error information
+void compare(float sigma){
+
+	GT = GT.compare(T, sigma);						//compare the ground truth to the test case - store errors in GT
+    T = T.compare(GT, sigma);						//compare the test case to the ground truth - store errors in T
+
+	//calculate the metrics
+	float FPR = GT.average(1);						//calculate the metrics
+	float FNR = T.average(1);
+	
+	std::cout << "FNR: " << FPR << std::endl;		//print false alarms and misses
+	std::cout << "FPR: " << FNR << std::endl;
+}
+
+// Output an advertisement for the lab, authors, and usage information
 void advertise(){
-	//output advertisement
 	std::cout<<std::endl<<std::endl;
 	std::cout<<"========================================================================="<<std::endl;
 	std::cout<<"Thank you for using the NetMets network comparison tool!"<<std::endl;
@@ -107,94 +221,32 @@ void advertise(){
 	std::cout<<"            compare two files with a tolerance of 10 (units defined by the network)"<<std::endl;
 	std::cout<<"       netmets file1 --gui"<<std::endl<<std::endl;
 	std::cout<<"            load a file and display it using OpenGL"<<std::endl;
-
-}
-
-void compare(float sigma){
-
-	// compare ground truth with truth and viceversa
-	stim::network<float> cGT = GT.compare(T, sigma);
-    stim::network<float> cT = T.compare(GT, sigma);
-
-	//calculate the metrics
-	float FPR = cGT.average(1);
-	float FNR = cT.average(1);
-	// print false alarms and misses
-	std::cout << "FNR: " << FPR << std::endl;
-	std::cout << "FPR: " << FNR << std::endl;
-}
-
-void glut_initialize(){
-	float factor = 1.2;
-
-	// init GLUT and create Window
-	int myargc = 1;
-	char* myargv[1];
-	myargv [0]=strdup ("netmets");
-
-	glutInit(&myargc, myargv);
-	glutInitDisplayMode(GLUT_DEPTH | GLUT_DOUBLE | GLUT_RGBA);
-	glutInitWindowPosition(100,100);
-	glutInitWindowSize(320,320);
-	glutCreateWindow("Lighthouse3D- GLUT Tutorial");
-
-	
-	// register callbacks
-	glutDisplayFunc(glut_render);
-	glutMouseFunc(glut_mouse);
-	glutMotionFunc(glut_motion);
-	glutReshapeFunc(glut_reshape);
-
-	//set up the camera
-	stim::vec<float> c = bb.center();
-	//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, factor * std::max(bb.size()[0], bb.size()[1])));
-	cam.LookAt(c[0], c[1], c[2]);						//look at the center
-
-}
-
-void display(){
-
-
-	//generate a bounding volume
-	bb = GT.boundingbox();
-
-	//create the GLUT window and set callback functions
-	glut_initialize();
-
-	glut_reshape(glutGet(GLUT_WINDOW_WIDTH), glutGet(GLUT_WINDOW_HEIGHT));
-
-	// enter GLUT event processing cycle
-	glutMainLoop();
 }
 int main(int argc, char* argv[])
 {
-	//create an instance of arglist
-	stim::arglist args;
+	stim::arglist args;						//create an instance of arglist
 	//add arguments
 	args.add("help", "prints this help");
 	args.add("sigma", "force a sigma value to specify the tolerance of the network comparison", "10");
 	args.add("gui", "display the network or network comparison using OpenGL");
-	//parse the user arguments
-	args.parse(argc, argv);
+	args.parse(argc, argv);					//parse the user arguments
-	//test for help
-	if(args["help"].is_set() || args.nargs() == 0){
-		advertise();
-		std::cout<<args.str();
-		exit(1);
+	if(args["help"].is_set() || args.nargs() == 0){			//test for help
+		advertise();										//output the advertisement
+		std::cout<<args.str();								//output arguments
+		exit(1);											//exit
 	}
-
-	//if one file is specified, load it as the ground truth
-	if(args.nargs() >= 1){
-		GT.load_obj(args.arg(0));
+	
+	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
 	}
-
-	//if two files are provided, compare them
-	if(args.nargs() == 2){
+	
+	if(args.nargs() == 2){			//if two files are specified, they will be displayed in neighboring viewports and compared
+		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
@@ -205,9 +257,9 @@ int main(int argc, char* argv[])
 	}
 	//if a GUI is requested, display the network using OpenGL
-	if(args["gui"].is_set())
-		display();
-	
-
-	
+	if(args["gui"].is_set()){		
+		bb = GT.boundingbox();					//generate a bounding volume		
+		glut_initialize();						//create the GLUT window and set callback functions		
+		glutMainLoop();							// enter GLUT event processing cycle
+	}	
 }
 \ No newline at end of file