Subversion Repositories gelsvn

Rev

Rev 411 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
386 jab 1
/*
2
 *  MeshEdit is a small application which allows you to load and edit a mesh.
3
 *  The mesh will be stored in GEL's half edge based Manifold data structure.
4
 *  A number of editing operations are supported. Most of these are accessible from the 
5
 *  console that pops up when you hit 'esc'.
6
 *
7
 *  Created by J. Andreas Bærentzen on 15/08/08.
8
 *  Copyright 2008 __MyCompanyName__. All rights reserved.
9
 *
10
 */
399 jab 11
 
12
#include <string>
386 jab 13
#include <iostream>
14
#include <CGLA/eigensolution.h>
15
#include <CGLA/Vec2d.h>
16
#include <CGLA/Vec3d.h>
17
#include <CGLA/Mat3x3d.h>
18
#include <CGLA/Mat2x2d.h>
19
#include <CGLA/Mat2x3d.h>
20
 
21
#include <LinAlg/Matrix.h>
22
#include <LinAlg/Vector.h>
23
#include <LinAlg/LapackFunc.h>
24
 
25
#include <Util/Timer.h>
26
#include <Util/ArgExtracter.h>
27
 
28
#include <GL/glew.h>
29
#include <GLGraphics/gel_glut.h>
30
 
31
#include <HMesh/Manifold.h>
32
#include <HMesh/VertexCirculator.h>
33
#include <HMesh/FaceCirculator.h>
34
#include <HMesh/build_manifold.h>
35
#include <HMesh/mesh_optimization.h>
36
#include <HMesh/triangulate.h>
37
#include <HMesh/load.h>
387 jab 38
#include <HMesh/quadric_simplify.h>
39
#include <HMesh/smooth.h>
386 jab 40
#include <HMesh/x3d_save.h>
388 jab 41
#include <HMesh/obj_save.h>
387 jab 42
#include <HMesh/mesh_optimization.h>
388 jab 43
#include <HMesh/triangulate.h>
44
#include <HMesh/close_holes.h>
45
#include <HMesh/caps_and_needles.h>
46
#include <HMesh/refine_edges.h>
47
#include <HMesh/subdivision.h>
386 jab 48
 
49
#include <GLConsole/GLConsole.h>
394 jab 50
#include <Util/Timer.h>
386 jab 51
#include "harmonics.h"
412 jab 52
#include "curvature.h"
399 jab 53
#include "Renderer.h"
54
#include "VisObj.h"
386 jab 55
 
391 jab 56
using namespace std;
57
using namespace HMesh;
58
using namespace Geometry;
59
using namespace GLGraphics;
60
using namespace CGLA;
61
using namespace Util;
62
using namespace LinAlg;
412 jab 63
using namespace CVarUtils;
386 jab 64
 
391 jab 65
inline VisObj& get_vis_obj(int i)
66
{
67
	static VisObj vo[9];
68
	return vo[i];
69
}
70
 
71
inline VisObj& avo()
72
{
412 jab 73
	static int& active = CreateCVar("active_mesh",0);
391 jab 74
	return get_vis_obj(active);
75
}
76
 
77
inline Manifold& active_mesh()
78
{
79
	return avo().mesh();
80
}
81
 
82
inline GLViewController& active_view_control()
83
{
84
	return avo().view_control();
85
}
86
 
386 jab 87
// Single global instance so glut can get access
412 jab 88
Trie CVarTrie;
386 jab 89
GLConsole theConsole;
90
 
91
////////////////////////////////////////////////////////////////////////////////
92
char* ConsoleHelp(std::vector<std::string> &args)
93
{
94
    theConsole.Printf("");
95
    theConsole.Printf("----------------- HELP -----------------");
391 jab 96
    theConsole.Printf("Press ESC key to open and close console");
386 jab 97
    theConsole.Printf("Press TAB to see the available commands and functions");
98
    theConsole.Printf("Functions are shown in green and variables in yellow");
99
    theConsole.Printf("Setting a value: [command] = value");
100
    theConsole.Printf("Getting a value: [command]");
101
    theConsole.Printf("Functions: [function] [arg1] [arg2] ...");
391 jab 102
    theConsole.Printf("Entering arg1=? or arg1=help will give a description.");
386 jab 103
    theConsole.Printf("History: Up and Down arrow keys move through history.");
104
    theConsole.Printf("Tab Completion: TAB does tab completion and makes suggestions.");
391 jab 105
    theConsole.Printf("");
106
    theConsole.Printf("Keyboard commands (when console is not active):");
399 jab 107
    theConsole.Printf("w   : switch to display.render_mode = wireframe");
108
    theConsole.Printf("i   : switch to display.render_mode = isophotes");
109
    theConsole.Printf("r   : switch to display.render_mode = reflection");
401 jab 110
    theConsole.Printf("m   : switch to display.render_mode = metallic");
111
    theConsole.Printf("g   : switch to display.render_mode = glazed");
112
    theConsole.Printf("n   : switch to display.render_mode = normal");
399 jab 113
    theConsole.Printf("h   : switch to display.render_mode = harmonics");
114
    theConsole.Printf("f   : toggle smooth/flat shading");
391 jab 115
    theConsole.Printf("1-9 : switch between active meshes.");
399 jab 116
    theConsole.Printf("d   : (display.render_mode = harmonics) diffuse light on and off");
117
    theConsole.Printf("h   : (display.render_mode = harmonics) highlight on and off ");
118
    theConsole.Printf("+/- : (display.render_mode = harmonics) which eigenvector to show");
391 jab 119
    theConsole.Printf("q   : quit program");
120
    theConsole.Printf("ESC : open console");
121
    theConsole.Printf("");
122
    theConsole.Printf("Mouse: Left button rotates, middle zooms, right pans");
386 jab 123
    theConsole.Printf("----------------- HELP -----------------");
124
    theConsole.Printf("");
125
    return "";
126
}
127
 
391 jab 128
bool wantshelp(std::vector<std::string> &args)
129
{
130
	if(args.size()==0) return false;
131
	string str = args[0];
132
	if(str=="help" || str=="HELP" || str=="Help" || str=="?") return true;
133
	return false;
134
}
386 jab 135
 
391 jab 136
/// Function that aligns two meshes.
137
char* console_align(std::vector<std::string> &args)
138
{
139
	if(wantshelp(args)) 
399 jab 140
	{
141
		theConsole.Printf("usage: align <dest> <src>");
142
		theConsole.Printf("This function aligns dest mesh with src");
143
		theConsole.Printf("In practice the GLViewController of src is copied to dst.");
144
		theConsole.Printf("both arguments are mandatory and must be numbers between 1 and 9.");
145
		theConsole.Printf("Note that results might be unexpexted if the meshes are not on the same scale");
146
		return "";
147
	}
148
 
391 jab 149
	int dest = 0;
150
	if(args.size()>0)
151
	{
152
		istringstream a0(args[0]);
153
		a0 >> dest;
154
		--dest;
155
		if(dest <0 || dest>8) return "dest mesh out of range (1-9)";
156
	}
157
	else return "neither source nor destination mesh?!";
158
	int src = 0;
159
	if(args.size()>1)
160
	{
161
		istringstream a1(args[1]);
162
		a1 >> src;
163
		--src;
164
		if(src <0 || src>8) return "src mesh out of range (1-9)";
165
	}	
166
	else return "no src mesh?";
386 jab 167
 
391 jab 168
	get_vis_obj(dest).view_control() = get_vis_obj(src).view_control();
169
 
170
	return "";
386 jab 171
}
172
 
391 jab 173
 
388 jab 174
char* console_save(std::vector<std::string> &args)
175
{
391 jab 176
	if(wantshelp(args)) 
399 jab 177
	{
178
		theConsole.Printf("usage: save <name.x3d|name.obj> ");
179
		return "";
180
	}
388 jab 181
	string& file_name = args[0];
182
	if(args.size() == 1)
183
	{
184
		if(file_name.substr(file_name.length()-4,file_name.length())==".obj")
185
		{
391 jab 186
			obj_save(file_name, active_mesh());
388 jab 187
			return "";
188
		}
189
		else if(file_name.substr(file_name.length()-4,file_name.length())==".x3d")
190
		{
391 jab 191
			x3d_save(file_name, active_mesh());
388 jab 192
			return "";
193
		}
194
		return "unknown format";
195
	}
196
	return "usage: save <name.x3d|name.obj> ";
197
}
386 jab 198
 
412 jab 199
char* console_save_history(std::vector<std::string> &args)
200
{
201
	if(wantshelp(args)) 
202
	{
203
		theConsole.Printf("usage: save_history <file> ... writes history to file");
204
		return "";
205
	}
206
	if(args.size()>0)
207
	{
208
		string& file_name = args[0];
209
		theConsole.SaveHistoryScript(file_name.c_str());
210
		return "";
211
	}
212
	return "must supply file name\n";
213
}
214
 
215
char* console_exec_script(std::vector<std::string> &args)
216
{
217
	if(wantshelp(args)) 
218
	{
219
		theConsole.Printf("usage: source <file> ... runs script");
220
		return "";
221
	}
222
	if(args.size()>0)
223
	{
224
		string& file_name = args[0];
225
		theConsole.ExecuteScript(file_name.c_str());
226
		return "";
227
	}
228
	return "must supply file name\n";
229
}
230
 
231
 
388 jab 232
char* console_refine_edges(std::vector<std::string> &args)
233
{
391 jab 234
	if(wantshelp(args)) 
399 jab 235
	{
236
		theConsole.Printf("usage: refine.split_edges <length>");
237
		theConsole.Printf("splits edges longer than <length>; default is 0.5 times average length");
238
		return "";
239
	}
240
 
388 jab 241
	float thresh = 0.5;
242
	if(args.size()>0)
243
	{
244
		istringstream a0(args[0]);
245
		a0 >> thresh;
246
	}
391 jab 247
	float avg_length = average_edge_length(active_mesh());
248
	refine_edges(active_mesh(), thresh * avg_length);
388 jab 249
	return "";
391 jab 250
 
388 jab 251
}
252
 
253
char* console_refine_faces(std::vector<std::string> &args)
254
{
391 jab 255
	if(wantshelp(args)) 
399 jab 256
	{
257
		theConsole.Printf("usage: refine.split_faces ");
258
		theConsole.Printf("usage:  Takes no arguments. Inserts a vertex at the centre of each face.");
259
		return "";
260
	}
261
 
391 jab 262
	safe_triangulate(active_mesh());
388 jab 263
	return "";
391 jab 264
 
388 jab 265
}
266
 
267
char* console_cc_subdivide(std::vector<std::string> &args)
268
{
391 jab 269
	if(wantshelp(args)) 
399 jab 270
	{
271
		theConsole.Printf("usage: refine.catmull_clark ");
272
		theConsole.Printf("Splits each polygon into four (Catmull Clark style)");
273
		return "";
274
	}
391 jab 275
	cc_split(active_mesh(),active_mesh());
388 jab 276
	return "";
277
}
278
 
394 jab 279
char* console_dual(std::vector<std::string> &args)
280
{
281
	if(wantshelp(args)) 
282
	{
283
		theConsole.Printf("usage: dual ");
284
		theConsole.Printf("Produces the dual by converting each face to a vertex placed at the barycenter.");
285
		return "";
286
	}
287
 
288
	Manifold& m = active_mesh();
289
 
290
	// make sure every face knows its number
291
	m.enumerate_faces();
292
 
293
	vector<Vec3f> vertices(m.no_faces());
399 jab 294
	vector<int> faces;
394 jab 295
	vector<int> indices;
296
 
297
	// Create new vertices. Each face becomes a vertex whose position
298
	// is the centre of the face
299
	int i=0;
300
	for(FaceIter f=m.faces_begin(); f!=m.faces_end(); ++f,++i)
301
		vertices[i] = centre(f);
302
 
303
	// Create new faces. Each vertex is a new face with N=valency of vertex
304
	// edges.
305
	i=0;
306
	for(VertexIter v=m.vertices_begin(); v!= m.vertices_end(); ++v,++i)
399 jab 307
		if(!is_boundary(v))
308
		{
309
			VertexCirculator vc(v);
310
			vector<int> index_tmp;
311
			for(; !vc.end(); ++vc)
312
				index_tmp.push_back(vc.get_face()->touched);
313
 
314
			// Push vertex indices for this face onto indices vector.
315
			// The circulator moves around the face in a clockwise fashion
316
			// so we just reverse the ordering.
317
			indices.insert(indices.end(), index_tmp.rbegin(), index_tmp.rend());
318
 
319
			// Insert face valency in the face vector.
320
			faces.push_back(vc.no_steps());
321
		}
394 jab 322
 
323
	// Clear the manifold before new geometry is inserted.
324
	m.clear();
325
 
326
	// And build
327
	build_manifold(m, vertices.size(), &vertices[0], faces.size(),
328
				   &faces[0],&indices[0]);
329
 
330
	return "";
331
}
388 jab 332
 
333
 
387 jab 334
char* console_minimize_curvature(std::vector<std::string> &args)
335
{
391 jab 336
	if(wantshelp(args)) 
399 jab 337
	{
338
		theConsole.Printf("usage: optimize.minimize_curvature <anneal>");
339
		theConsole.Printf("Flip edges to minimize mean curvature.");
340
		theConsole.Printf("If anneal is true, simulated annealing (slow) is used rather than a greedy scheme");
341
		return "";
342
	}
387 jab 343
	bool anneal=false;
344
	if(args.size()>0)
345
	{
346
		istringstream a0(args[0]);
347
		a0 >> anneal;
348
	}
391 jab 349
 
350
	minimize_curvature(active_mesh(), anneal);
351
	avo().post_create_display_list();
387 jab 352
	return "";
353
}
354
 
355
char* console_minimize_dihedral(std::vector<std::string> &args)
356
{
391 jab 357
	if(wantshelp(args)) 
399 jab 358
	{
359
		theConsole.Printf("usage: optimize.minimize_dihedral <iter> <anneal> <use_alpha> <gamma> ");
360
		theConsole.Printf("Flip edges to minimize dihedral angles.");
361
		theConsole.Printf("Iter is the max number of iterations. anneal tells us whether to use ");
362
		theConsole.Printf("simulated annealing and not greedy optimization. use_alpha (default=true) ");
363
		theConsole.Printf("means to use angle and not cosine of anglegamma (default=4) is the power ");
364
		theConsole.Printf("to which we raise the dihedral angle");
365
		return "";
366
	}
387 jab 367
	int iter = 1000;
368
	if(args.size()>0)
369
	{
370
		istringstream a0(args[0]);
371
		a0 >> iter;
372
	}
391 jab 373
 
387 jab 374
	bool anneal = false;
375
	if(args.size()>1)
376
	{
377
		istringstream a0(args[0]);
378
		a0 >> anneal;
379
	}
391 jab 380
 
387 jab 381
	bool use_alpha = true;
388 jab 382
	if(args.size()>2)
387 jab 383
	{
384
		istringstream a0(args[0]);
385
		a0 >> use_alpha;
386
	}
391 jab 387
 
387 jab 388
	float gamma = 4.0;
389
	if(args.size()>3)
390
	{
391
		istringstream a0(args[0]);
392
		a0 >> gamma;
393
	}
394
 
395
 
391 jab 396
	minimize_dihedral_angle(active_mesh(), iter, anneal, use_alpha, gamma);
387 jab 397
	return "";
398
}
399
 
388 jab 400
char* console_maximize_min_angle(std::vector<std::string> &args)
401
{
391 jab 402
	if(wantshelp(args)) 
399 jab 403
	{
404
		theConsole.Printf("usage: optimize.maximize_min_angle <thresh> <anneal>");
405
		theConsole.Printf("Flip edges to maximize min angle - to make mesh more Delaunay.");
406
		theConsole.Printf("If the dot product of the normals between adjacent faces < thresh");
407
		theConsole.Printf("no flip will be made. anneal selects simulated annealing rather ");
408
		theConsole.Printf("nthan greedy optimization.");
409
		return "";
410
	}
388 jab 411
	float thresh=0.0;
412
	if(args.size()>0)
413
	{
414
		istringstream a0(args[0]);
415
		a0 >> thresh;
416
	}
417
	bool anneal=false;
418
	if(args.size()>1)
419
	{
420
		istringstream a0(args[0]);
421
		a0 >> anneal;
422
	}
391 jab 423
	maximize_min_angle(active_mesh(),thresh,anneal);
388 jab 424
	return "";
425
}
426
 
427
 
387 jab 428
char* console_optimize_valency(std::vector<std::string> &args)
429
{
391 jab 430
	if(wantshelp(args)) 
399 jab 431
	{
432
		theConsole.Printf("usage: optimize.valency <anneal> ");
433
		theConsole.Printf("Optimizes valency for triangle meshes. Anneal selects simulated annealing rather than greedy optim.");
434
		return "";
435
	}
387 jab 436
	bool anneal=false;
437
	if(args.size()>0)
438
	{
439
		istringstream a0(args[0]);
440
		a0 >> anneal;
441
	}
391 jab 442
	optimize_valency(active_mesh(), anneal);
387 jab 443
	return "";
444
}
445
 
388 jab 446
char* console_analyze(std::vector<std::string> &args)
447
{
391 jab 448
	if(wantshelp(args)) 
399 jab 449
	{
450
		theConsole.Printf("usage:  harmonics.analyze");
451
		theConsole.Printf("Creates the Laplace Beltrami operator for the mesh and finds all eigensolutions.");
452
		theConsole.Printf("It also projects the vertices onto the eigenvectors - thus transforming the mesh");
453
		theConsole.Printf("to this basis.");
454
		theConsole.Printf("Note that this will stall the computer for a large mesh - as long as we use Lapack.");
455
		return "";
456
	}
391 jab 457
	avo().harmonics_analyze_mesh();
388 jab 458
	return "";
459
}
460
 
461
 
386 jab 462
char* console_partial_reconstruct(std::vector<std::string> &args)
463
{
411 jrf 464
  if(args.size() != 3)
465
		theConsole.Printf("usage: haramonics.partial_reconstruct <e0> <e1> <s>");
466
 
467
  if(wantshelp(args)) 
399 jab 468
	{
469
		theConsole.Printf("Reconstruct from projections onto eigenvectors. The two first arguments indicate");
470
		theConsole.Printf("the eigenvector interval that we reconstruct from. The last argument is the ");
471
		theConsole.Printf("scaling factor. Thus, for a vertex, v, the formula for computing the position, p, is:");
472
		theConsole.Printf("for (i=e0; i<=e1;++i) p += proj[i] * Q[i][v] * s;");
473
		theConsole.Printf("where proj[i] is the 3D vector containing the x, y, and z projections of the mesh onto");
474
		theConsole.Printf("eigenvector i. Q[i][v] is the v'th coordinate of the i'th eigenvector.");
475
		theConsole.Printf("Note that if vertex coordinates are not first reset, the result is probably unexpected.");
411 jrf 476
	}
477
 
478
  if(args.size() != 3)
399 jab 479
		return "";
411 jrf 480
 
481
  int E0,E1;
386 jab 482
	float scale;
483
	istringstream a0(args[0]);
484
	a0 >> E0;
485
	istringstream a1(args[1]);
486
	a1 >> E1;
487
	istringstream a2(args[2]);
488
	a2 >> scale;
391 jab 489
	avo().harmonics_partial_reconstruct(E0,E1,scale);
386 jab 490
	return "";
491
}
492
 
493
char* console_reset_shape(std::vector<std::string> &args)
494
{
391 jab 495
	if(wantshelp(args)) 
399 jab 496
	{
497
		theConsole.Printf("usage: harmonics.reset_shape ");
498
		theConsole.Printf("Simply sets all vertices to 0,0,0. Call this before doing partial_reconstruct");
499
		theConsole.Printf("unless you know what you are doing.");
500
		return "";
501
	}
391 jab 502
	avo().harmonics_reset_shape();
386 jab 503
	return "";
504
}
505
 
506
 
388 jab 507
char* console_close_holes(std::vector<std::string> &args)
508
{
391 jab 509
	if(wantshelp(args)) 
399 jab 510
	{
511
		theConsole.Printf("usage: cleanup.close_holes");
512
		theConsole.Printf("This function closes holes. It simply follows the loop of halfvectors which");
513
		theConsole.Printf("enclose the hole and add a face to which they all point.");
514
		return "";
515
	}
391 jab 516
	close_holes(active_mesh());
388 jab 517
	return "";
518
}
386 jab 519
 
388 jab 520
char* console_reload(std::vector<std::string> &args)
521
{
391 jab 522
	if(wantshelp(args)) 
399 jab 523
	{
412 jab 524
		theConsole.Printf("usage:  load <file>");
525
		theConsole.Printf("(Re)loads the current file if no argument is given, but");
399 jab 526
		theConsole.Printf("if an argument is given, then that becomes the current file");
527
		return "";
528
	}
391 jab 529
	if(!avo().reload(args.size()>0 ? args[0]:""))
530
		return "failed to load";
388 jab 531
	return "";
532
}
533
 
534
 
387 jab 535
char* console_simplify(std::vector<std::string> &args)
536
{
391 jab 537
	if(wantshelp(args)) 
399 jab 538
	{
539
		theConsole.Printf("usage: simplify <fraction> ");
540
		theConsole.Printf("Performs Garland Heckbert (quadric based) mesh simplification.");
541
		theConsole.Printf("The only argument is the fraction of vertices to keep.");
542
		return "";
543
	}
387 jab 544
	float keep_fraction;
545
	if(args.size()==0) return "you must specify fraction of vertices to keep";
546
	istringstream a0(args[0]);
547
	a0 >> keep_fraction;
391 jab 548
 
387 jab 549
	Vec3f p0, p7;
391 jab 550
	active_mesh().get_bbox(p0, p7);
387 jab 551
	Vec3f d = p7-p0;
552
	float s = 1.0/d.max_coord();
553
	Vec3f pcentre = (p7+p0)/2.0;
391 jab 554
	for(VertexIter vi = active_mesh().vertices_begin(); vi != active_mesh().vertices_end(); ++vi)
387 jab 555
		vi->pos = (vi->pos - pcentre) * s;
391 jab 556
	quadric_simplify(active_mesh(),keep_fraction,0.0001f,true);
557
	for(VertexIter vi = active_mesh().vertices_begin(); vi != active_mesh().vertices_end(); ++vi)
387 jab 558
		vi->pos = vi->pos*d.max_coord() + pcentre;
559
	return "";
560
}
561
 
391 jab 562
char* console_vertex_noise(std::vector<std::string> &args)
563
{
564
	if(wantshelp(args)) 
399 jab 565
	{
566
		theConsole.Printf("usage: noise.perturb_vertices <amplitude>");
403 jab 567
		theConsole.Printf("adds a random vector to each vertex. A random vector in the unit cube is generated and");
568
		theConsole.Printf("to ensure an isotropic distribution, vectors outside the unit ball are discarded.");
569
		theConsole.Printf("The vector is multiplied by the average edge length and then by the amplitude specified.");
570
		theConsole.Printf("If no amplitude is specified, the default (0.5) is used.");
399 jab 571
		return "";
572
	}
391 jab 573
	float avg_length = average_edge_length(active_mesh());
574
 
575
	float noise_amplitude = 0.5;
576
	if(args.size()>0) 
577
	{
578
		istringstream a0(args[0]);
579
		a0 >> noise_amplitude;
580
	}
581
 
582
	srand(0);
583
	for(VertexIter vi = active_mesh().vertices_begin(); vi != active_mesh().vertices_end(); ++vi)
584
	{
585
		Vec3f v;
586
		do {
587
			v = Vec3f(rand(),rand(),rand());
588
			v /= RAND_MAX;
589
		} while(sqr_length(v) > 1.0);
590
		v -= Vec3f(0.5);
591
		v *= 2.0;
592
		v *= noise_amplitude;
593
		v *= avg_length;
594
		vi->pos += v;
595
	}		
596
	return "";
597
}
598
 
412 jab 599
char* console_perpendicular_vertex_noise(std::vector<std::string> &args)
600
{
601
	if(wantshelp(args)) 
602
	{
603
		theConsole.Printf("usage: noise.perturb_vertices_perpendicular <amplitude>");
604
		theConsole.Printf("adds the normal times a random scalar times amplitude times");
605
		theConsole.Printf("times average edge length to the vertex. (default amplitude=0.5)");
606
		return "";
607
	}
608
	float avg_length = average_edge_length(active_mesh());
609
 
610
	float noise_amplitude = 0.5;
611
	if(args.size()>0) 
612
	{
613
		istringstream a0(args[0]);
614
		a0 >> noise_amplitude;
615
	}
616
 
617
	vector<Vec3f> normals(active_mesh().no_vertices());
618
	int i=0;
619
	for(VertexIter vi = active_mesh().vertices_begin(); vi != active_mesh().vertices_end(); ++vi,++i)
620
		normals[i] = normal(vi);
621
	i=0;
622
	srandom(0);
623
	for(VertexIter vi = active_mesh().vertices_begin(); vi != active_mesh().vertices_end(); ++vi,++i)
624
	{
625
		float rval = 0.5-random()/float(RAND_MAX);
626
		vi->pos += normals[i]*rval*noise_amplitude*avg_length*2.0;
627
	}
628
	return "";
629
}
630
 
391 jab 631
char* console_noisy_flips(std::vector<std::string> &args)
632
{
633
	if(wantshelp(args)) 
399 jab 634
	{
635
		theConsole.Printf("usage:  noise.perturb_topology <iter>");
636
		theConsole.Printf("Perform random flips. iter (default=1) is the number of iterations.");
637
		theConsole.Printf("mostly for making nasty synthetic test cases.");
638
		return "";
639
	}
391 jab 640
	int iter=1;
641
	if(args.size()>0)
642
	{
643
		istringstream a0(args[0]);
644
		a0 >> iter;
645
	}
646
 
647
	randomize_mesh(active_mesh(),  iter);
648
	return "";
649
}
650
 
387 jab 651
char* console_laplacian_smooth(std::vector<std::string> &args)
652
{
391 jab 653
	if(wantshelp(args)) 
399 jab 654
	{
412 jab 655
		theConsole.Printf("usage:  smooth.laplacian <weight> <iter>");
399 jab 656
		theConsole.Printf("Perform Laplacian smoothing. weight is the scaling factor for the Laplacian.");
412 jab 657
		theConsole.Printf("default weight = 1.0. Default number of iterations = 1");
399 jab 658
		return "";
659
	}
387 jab 660
	float t=1.0;
661
	if(args.size()>0)
662
	{
663
		istringstream a0(args[0]);
664
		a0 >> t;
665
	}
412 jab 666
	int iter=1;
667
	if(args.size()>1)
668
	{
669
		istringstream a0(args[1]);
670
		a0 >> iter;
671
	}
672
 
387 jab 673
	/// Simple laplacian smoothing with an optional weight.
412 jab 674
	for(int i=0;i<iter;++i) laplacian_smooth(active_mesh(), t);
387 jab 675
	return "";
676
}
677
 
412 jab 678
char* console_mean_curvature_smooth(std::vector<std::string> &args)
679
{
680
	if(wantshelp(args)) 
681
	{
682
		theConsole.Printf("usage:  smooth.mean_curvature <weight> <iter>");
683
		theConsole.Printf("Perform mean curvature smoothing. weight is the scaling factor for the");
684
		theConsole.Printf("mean curvature vector which has been normalized by dividing by edge lengths");
685
		theConsole.Printf("this allows for larger steps as suggested by Desbrun et al.");
686
		theConsole.Printf("default weight = 1.0. Default number of iterations = 1");
687
		return "";
688
	}
689
	float t=1.0;
690
	if(args.size()>0)
691
	{
692
		istringstream a0(args[0]);
693
		a0 >> t;
694
	}
695
	int iter=1;
696
	if(args.size()>1)
697
	{
698
		istringstream a0(args[1]);
699
		a0 >> iter;
700
	}	
701
	vector<Vec3d> new_pos(active_mesh().no_vertices());
702
	for(int j=0;j<iter;++j)
703
	{
704
		int i=0;
705
		for(VertexIter v = active_mesh().vertices_begin(); v != active_mesh().vertices_end(); ++v,++i) {
706
			Vec3d m;
707
			double w_sum;
708
			unnormalized_mean_curvature_normal(v, m, w_sum);
709
			new_pos[i] = Vec3d(v->pos)  - (t * m/w_sum);
710
		}
711
		i=0;
712
		for(VertexIter v = active_mesh().vertices_begin(); v != active_mesh().vertices_end(); ++v,++i)
713
			v->pos = Vec3f(new_pos[i]);
714
	}
715
	return "";
716
}
717
 
718
 
387 jab 719
char* console_taubin_smooth(std::vector<std::string> &args)
720
{
391 jab 721
	if(wantshelp(args)) 
399 jab 722
	{
723
		theConsole.Printf("usage:  smooth.taubin <iter>");
724
		theConsole.Printf("Perform Taubin smoothing. iter (default=1) is the number of iterations.");
725
		return "";
726
	}
387 jab 727
	int iter=1;
728
	if(args.size()>0)
729
	{
730
		istringstream a0(args[0]);
731
		a0 >> iter;
732
	}
733
 
734
	/// Taubin smoothing is similar to laplacian smoothing but reduces shrinkage
391 jab 735
	taubin_smooth(active_mesh(),  iter);
387 jab 736
	return "";
737
}
738
 
739
char* console_fvm_smooth(std::vector<std::string> &args)
740
{	
391 jab 741
	if(wantshelp(args)) 
399 jab 742
	{
743
		theConsole.Printf("usage: smooth.fuzzy_vector_median <iter>");
744
		theConsole.Printf("Smooth normals using fuzzy vector median smoothing. iter (default=1) is the number of iterations");
745
		theConsole.Printf("This function does a very good job of preserving sharp edges.");
746
		return "";
747
	}
387 jab 748
	int iter=1;
749
	if(args.size()>0)
750
	{
751
		istringstream a0(args[0]);
752
		a0 >> iter;
753
	}
754
	/** Fuzzy vector median smoothing is effective when it comes to
755
	 preserving sharp edges. */
391 jab 756
	fvm_smooth(active_mesh(),  iter);
387 jab 757
	return "";
391 jab 758
 
387 jab 759
}
760
 
388 jab 761
char* console_triangulate(std::vector<std::string> &args)
762
{	
391 jab 763
	if(wantshelp(args)) 
399 jab 764
	{
765
		theConsole.Printf("usage:  triangulate");
766
		theConsole.Printf("This function triangulates all non triangular faces of the mesh.");
767
		theConsole.Printf("you may want to call it after hole closing. For a polygon it simply connects");
768
		theConsole.Printf("the two closest vertices in a recursive manner until only triangles remain");
769
		return "";
770
	}
391 jab 771
	shortest_edge_triangulate(active_mesh());
388 jab 772
	return "";
773
}
774
 
775
 
776
char* console_remove_caps(std::vector<std::string> &args)
777
{	
391 jab 778
	if(wantshelp(args)) 
399 jab 779
	{
780
		theConsole.Printf("usage:  cleanup.remove_caps thresh");
781
		theConsole.Printf("Remove caps (triangles with one very big angle). The thresh argument is the fraction of PI to");
782
		theConsole.Printf("use as threshold for big angle. Default is 0.85. Caps are removed by flipping.");
783
		return "";
784
	}
391 jab 785
	float t=0.85;
786
	if(args.size()>0)
787
	{
788
		istringstream a0(args[0]);
789
		a0 >> t;
790
	}
399 jab 791
 
391 jab 792
	remove_caps_from_trimesh(active_mesh(), static_cast<float>(M_PI) *t);
388 jab 793
	return "";
794
}
795
 
796
char* console_remove_needles(std::vector<std::string> &args)
797
{	
391 jab 798
	if(wantshelp(args)) 
399 jab 799
	{
800
		theConsole.Printf("usage: cleanup.remove_needles <thresh>");
801
		theConsole.Printf("Removes very short edges by collapse. thresh is multiplied by the average edge length");
802
		theConsole.Printf("to get the length shorter than which we collapse. Default = 0.1");
803
		return "";
804
	}
388 jab 805
	float thresh = 0.1;
806
	if(args.size()>0)
807
	{
808
		istringstream a0(args[0]);
809
		a0 >> thresh;
810
	}
391 jab 811
	float avg_length = average_edge_length(active_mesh());
812
	remove_needles_from_trimesh(active_mesh(), thresh * avg_length);
388 jab 813
	return "";
814
}
815
 
391 jab 816
void reshape(int W, int H)
817
{
818
	active_view_control().reshape(W,H);
819
}
820
 
386 jab 821
void display() 
822
{
823
	glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
824
 
412 jab 825
	static string& display_render_mode = CreateCVar<string>("display.render_mode","");
826
	static int& display_smooth= CreateCVar<int>("display.smooth_shading",1);
386 jab 827
 
828
	glPushMatrix();
829
 
399 jab 830
	avo().display(display_render_mode, display_smooth);
386 jab 831
 
832
	glPopMatrix();
833
 
834
	glUseProgram(0);
835
	theConsole.RenderConsole();
836
 
837
	glutSwapBuffers();
838
}
839
 
840
void animate() 
841
{	
394 jab 842
	//usleep( (int)1e4 );
391 jab 843
	active_view_control().try_spin();
386 jab 844
	glutPostRedisplay();
845
}
846
 
847
 
848
void mouse(int button, int state, int x, int y) 
849
{
850
	Vec2i pos(x,y);
851
	if (state==GLUT_DOWN) 
852
	{
853
		if (button==GLUT_LEFT_BUTTON) 
391 jab 854
			active_view_control().grab_ball(ROTATE_ACTION,pos);
386 jab 855
		else if (button==GLUT_MIDDLE_BUTTON) 
391 jab 856
			active_view_control().grab_ball(ZOOM_ACTION,pos);
386 jab 857
		else if (button==GLUT_RIGHT_BUTTON) 
391 jab 858
			active_view_control().grab_ball(PAN_ACTION,pos);
386 jab 859
	}
860
	else if (state==GLUT_UP)
391 jab 861
		active_view_control().release_ball();
386 jab 862
}
863
 
864
void motion(int x, int y) {
865
	Vec2i pos(x,y);
391 jab 866
	active_view_control().roll_ball(Vec2i(x,y));
386 jab 867
}
868
 
869
void keyboard_spec(int key, int x, int y)
870
{
391 jab 871
	int mod = glutGetModifiers();
412 jab 872
	if( theConsole.IsOpen() ) {
391 jab 873
		// If shift held, scroll the console
874
		if( mod == GLUT_ACTIVE_SHIFT ) {
875
			switch (key){
876
				case GLUT_KEY_UP:
877
					theConsole.ScrollDownLine();
878
					break;
879
				case GLUT_KEY_DOWN: 
880
					theConsole.ScrollUpLine();
881
					break;
882
			}
883
		} else {
884
			theConsole.StandardKeyBindings( key );
885
		}
886
	}
386 jab 887
}
888
 
889
 
890
void keyboard(unsigned char key, int x, int y) 
891
{	
412 jab 892
	if(theConsole.IsOpen())
391 jab 893
	{
386 jab 894
		switch(key) {
895
			case '\033': 
896
				theConsole.ToggleConsole();
412 jab 897
			default:
898
				theConsole.EnterCommandCharacter(key);
899
			break;
388 jab 900
		}
391 jab 901
		if(key == 13)	avo().post_create_display_list();
902
 
903
	}	
386 jab 904
	else {
412 jab 905
		string& display_render_mode = GetCVarRef<string>("display.render_mode");
906
		int& display_smooth = GetCVarRef<int>("display.smooth_shading");
907
		int& active  = GetCVarRef<int>("active_mesh");
386 jab 908
 
391 jab 909
 
386 jab 910
		switch(key) {
911
			case 'q': exit(0);
912
			case '\033':
913
				theConsole.ToggleConsole();
914
				break;
391 jab 915
			case '1':
916
			case '2':
917
			case '3':
918
			case '4':
919
			case '5':
920
			case '6':
921
			case '7':
922
			case '8':
923
			case '9':
924
				active = key - '1'; break;
399 jab 925
			case 'f': display_smooth = !display_smooth; break;
386 jab 926
			case 'w':
399 jab 927
				display_render_mode = "wire"; break;
928
			case 'n':
929
				display_render_mode = "normal"; break;
930
			case 'i':
931
				display_render_mode = "isophotes"; break;
932
			case 'r':
933
				display_render_mode = "reflection"; break;
934
			case 'h':
935
				display_render_mode = "harmonics"; break;
403 jab 936
			case 't':
937
				display_render_mode = "toon"; break;
401 jab 938
			case 'g':
939
				display_render_mode = "glazed"; break;
406 jab 940
			case 'a':
941
				display_render_mode = "ambient_occlusion"; break;
942
			case 'c':
943
				display_render_mode = "copper"; break;
403 jab 944
			case 'C':
945
				display_render_mode = "curvature_lines"; break;
946
			case 'M':
947
				display_render_mode = "mean_curvature"; break;
948
			case 'G':
949
				display_render_mode = "gaussian_curvature"; break;
386 jab 950
		}
391 jab 951
 
399 jab 952
		if(display_render_mode.substr(0,4) == "harm")
391 jab 953
			avo().harmonics_parse_key(key);
954
 
403 jab 955
		if(key != '\033') avo().post_create_display_list();		
386 jab 956
	}
957
}
958
 
959
void init_glut(int argc, char** argv)
960
{
397 jab 961
	glutInitDisplayMode(GLUT_RGBA|GLUT_DOUBLE|GLUT_DEPTH|GLUT_ALPHA);
386 jab 962
	glutInitWindowSize(WINX, WINY);
963
	glutInit(&argc, argv);
964
	glutCreateWindow("Shape Harmonics");
965
	glutDisplayFunc(display);
966
	glutKeyboardFunc(keyboard);
967
	glutSpecialFunc(keyboard_spec);
968
	glutReshapeFunc(reshape);
969
	glutMouseFunc(mouse);
970
	glutMotionFunc(motion);
971
	glutIdleFunc(animate);
972
}
973
 
974
void init_gl()
975
{
976
	glewInit();
977
	glEnable(GL_LIGHTING);
978
	glEnable(GL_LIGHT0);
388 jab 979
	glLightModeli(GL_LIGHT_MODEL_TWO_SIDE, 1);
399 jab 980
 
386 jab 981
	// Set the value of a uniform
982
	//glUniform2f(glGetUniformLocation(prog_P0,"WIN_SCALE"), win_size_x/2.0, win_size_y/2.0);
983
 
984
	glMatrixMode(GL_MODELVIEW);
985
	glLoadIdentity();
986
	glClearColor(0.50f, 0.50f, 0.50f, 0.f);
987
	glColor4f(1.0f, 1.0f, 1.0f, 0.f);
406 jab 988
	float material[4] = {1,1,1,1};
989
	glMaterialfv(GL_FRONT_AND_BACK, GL_DIFFUSE, material);
386 jab 990
	glEnable(GL_DEPTH_TEST);
991
 
412 jab 992
	CreateCVar( "help", ConsoleHelp );
993
	CreateCVar("harmonics.reset_shape", console_reset_shape);
994
	CreateCVar("harmonics.analyze", console_analyze);
995
	CreateCVar("harmonics.partial_reconstruct", console_partial_reconstruct);
996
	CreateCVar("simplify", console_simplify);
997
	CreateCVar("smooth.mean_curvature", console_mean_curvature_smooth);
998
	CreateCVar("smooth.laplacian", console_laplacian_smooth);
999
	CreateCVar("smooth.taubin", console_taubin_smooth);
1000
	CreateCVar("smooth.fuzzy_vector_median", console_fvm_smooth);
391 jab 1001
 
412 jab 1002
	CreateCVar("optimize.valency", console_optimize_valency);
1003
	CreateCVar("optimize.minimize_dihedral_angles", console_minimize_dihedral);
1004
	CreateCVar("optimize.minimize_curvature", console_minimize_curvature);
1005
	CreateCVar("optimize.maximize_min_angle", console_maximize_min_angle);
1006
	CreateCVar("cleanup.close_holes", console_close_holes);
1007
	CreateCVar("load", console_reload);
391 jab 1008
 
412 jab 1009
	CreateCVar("cleanup.remove_caps", console_remove_caps);
1010
	CreateCVar("cleanup.remove_needles", console_remove_needles);
1011
	CreateCVar("triangulate", console_triangulate);
1012
	CreateCVar("refine.split_edges", console_refine_edges);
1013
	CreateCVar("refine.split_faces", console_refine_faces);
1014
	CreateCVar("refine.catmull_clark", console_cc_subdivide);
1015
	CreateCVar("save", console_save);
1016
	CreateCVar("noise.perturb_vertices", console_vertex_noise);
1017
	CreateCVar("noise.perturb_vertices_perpendicular", console_perpendicular_vertex_noise);
1018
	CreateCVar("noise.perturb_topology", console_noisy_flips);
399 jab 1019
 
412 jab 1020
	CreateCVar("dual", console_dual);
399 jab 1021
 
412 jab 1022
	CreateCVar("align", console_align);
1023
	CreateCVar("save_history", console_save_history);
1024
	CreateCVar("source", console_exec_script);
387 jab 1025
 
1026
 
386 jab 1027
}
1028
 
1029
int main(int argc, char** argv)
1030
{
1031
	ArgExtracter ae(argc, argv);
391 jab 1032
 
1033
	init_glut(argc,argv);
1034
	init_gl();
1035
 
1036
	Harmonics::init();
1037
 
399 jab 1038
	if(argc>1)
386 jab 1039
	{		
391 jab 1040
		string file = ae.get_last_arg();
1041
		avo().reload(file);
386 jab 1042
	}
1043
 
399 jab 1044
 
386 jab 1045
	glutMainLoop();
1046
}
1047
 
1048