Projector.js 14.1 KB
Newer Older
1 2 3 4 5 6 7 8
/**
 * @author mr.doob / http://mrdoob.com/
 * @author supereggbert / http://www.paulbrunt.co.uk/
 * @author julianwa / https://github.com/julianwa
 */

THREE.Projector = function() {

M
Mr.doob 已提交
9 10
	var _object, _objectCount, _objectPool = [],
	_vertex, _vertexCount, _vertexPool = [],
11 12 13 14
	_face, _face3Count, _face3Pool = [], _face4Count, _face4Pool = [],
	_line, _lineCount, _linePool = [],
	_particle, _particleCount, _particlePool = [],

M
Mr.doob 已提交
15
	_renderData = { objects: [], sprites: [], lights: [], elements: [] },
16

17
	_vector3 = new THREE.Vector3(),
18
	_vector4 = new THREE.Vector4(),
19

20
	_projScreenMatrix = new THREE.Matrix4(),
M
Mr.doob 已提交
21
	_projScreenobjectMatrixWorld = new THREE.Matrix4(),
22

A
alteredq 已提交
23
	_frustum = new THREE.Frustum(),
24 25 26 27 28 29 30

	_clippedVertex1PositionScreen = new THREE.Vector4(),
	_clippedVertex2PositionScreen = new THREE.Vector4(),

	_face3VertexNormals;

	this.projectVector = function ( vector, camera ) {
31

32
		camera.matrixWorldInverse.getInverse( camera.matrixWorld );
33

J
Justin Sermeno 已提交
34
		_projScreenMatrix.multiply( camera.projectionMatrix, camera.matrixWorldInverse );
35 36 37
		_projScreenMatrix.multiplyVector3( vector );

		return vector;
38

39 40
	};

J
Justin Sermeno 已提交
41
	this.unprojectVector = function ( vector, camera ) {
42

43 44 45
		camera.projectionMatrixInverse.getInverse( camera.projectionMatrix );

		_projScreenMatrix.multiply( camera.matrixWorld, camera.projectionMatrixInverse );
46 47
		_projScreenMatrix.multiplyVector3( vector );

J
Justin Sermeno 已提交
48
		return vector;
49

J
Justin Sermeno 已提交
50
	};
J
Justin Sermeno 已提交
51

J
Justin Sermeno 已提交
52
	this.pickingRay = function ( vector, camera ) {
J
Justin Sermeno 已提交
53

54 55 56
		var end, ray, t;

		// set two vectors with opposing z values
J
Justin Sermeno 已提交
57 58
		vector.z = -1.0;
		end = new THREE.Vector3( vector.x, vector.y, 1.0 );
J
Justin Sermeno 已提交
59

J
Justin Sermeno 已提交
60 61 62
		this.unprojectVector( vector, camera );
		this.unprojectVector( end, camera );

63 64
		// find direction from vector to end
		end.subSelf( vector ).normalize();
65

66
		return new THREE.Ray( vector, end );
67

68 69
	};

M
Mr.doob 已提交
70 71 72
	this.projectGraph = function ( root, sort ) {

		_objectCount = 0;
73

74
		_renderData.objects.length = 0;
M
Mr.doob 已提交
75
		_renderData.sprites.length = 0;
76
		_renderData.lights.length = 0;
77

M
Mr.doob 已提交
78
		var projectObject = function ( object ) {
79

M
Mr.doob 已提交
80
			if ( object.visible === false ) return;
81

M
Mr.doob 已提交
82
			if ( ( object instanceof THREE.Mesh || object instanceof THREE.Line ) &&
M
Mr.doob 已提交
83
			( object.frustumCulled === false || _frustum.contains( object ) === true ) ) {
84

85 86
				_vector3.copy( object.matrixWorld.getPosition() );
				_projScreenMatrix.multiplyVector3( _vector3 );
87

M
Mr.doob 已提交
88 89 90
				_object = getNextObjectInPool();
				_object.object = object;
				_object.z = _vector3.z;
M
Mr.doob 已提交
91

M
Mr.doob 已提交
92
				_renderData.objects.push( _object );
M
Mr.doob 已提交
93 94 95

			} else if ( object instanceof THREE.Sprite || object instanceof THREE.Particle ) {

96 97
				_vector3.copy( object.matrixWorld.getPosition() );
				_projScreenMatrix.multiplyVector3( _vector3 );
M
Mr.doob 已提交
98 99 100 101 102 103

				_object = getNextObjectInPool();
				_object.object = object;
				_object.z = _vector3.z;

				_renderData.sprites.push( _object );
M
Mr.doob 已提交
104

105
			} else if ( object instanceof THREE.Light ) {
106

107
				_renderData.lights.push( object );
108

M
Mr.doob 已提交
109
			}
110

M
Mr.doob 已提交
111 112 113 114 115 116 117 118
			for ( var c = 0, cl = object.children.length; c < cl; c ++ ) {

				projectObject( object.children[ c ] );

			}

		};

M
Mr.doob 已提交
119 120
		projectObject( root );

M
Mr.doob 已提交
121
		if ( sort === true ) _renderData.objects.sort( painterSort );
122

123
		return _renderData;
124 125 126 127 128

	};

	this.projectScene = function ( scene, camera, sort ) {

129
		var near = camera.near, far = camera.far, visible = false,
130
		o, ol, v, vl, f, fl, n, nl, c, cl, u, ul, object,
131
		objectMatrixWorld, objectMatrixWorldRotation,
132
		geometry, geometryMaterials, vertices, vertex, vertexPositionScreen,
133 134 135 136 137 138 139 140
		faces, face, faceVertexNormals, normal, faceVertexUvs, uvs,
		v1, v2, v3, v4;

		_face3Count = 0;
		_face4Count = 0;
		_lineCount = 0;
		_particleCount = 0;

141 142
		_renderData.elements.length = 0;

M
Mr.doob 已提交
143 144 145 146 147 148
		if ( camera.parent === undefined ) {

			console.warn( 'DEPRECATED: Camera hasn\'t been added to a Scene. Adding it...' );
			scene.add( camera );

		}
M
Mr.doob 已提交
149 150

		scene.updateMatrixWorld();
151

152
		camera.matrixWorldInverse.getInverse( camera.matrixWorld );
153

154
		_projScreenMatrix.multiply( camera.projectionMatrix, camera.matrixWorldInverse );
155

A
alteredq 已提交
156
		_frustum.setFromMatrix( _projScreenMatrix );
157

M
Mr.doob 已提交
158
		_renderData = this.projectGraph( scene, false );
159

160
		for ( o = 0, ol = _renderData.objects.length; o < ol; o++ ) {
161

M
Mr.doob 已提交
162
			object = _renderData.objects[ o ].object;
163

M
Mr.doob 已提交
164
			objectMatrixWorld = object.matrixWorld;
165 166 167 168 169 170

			_vertexCount = 0;

			if ( object instanceof THREE.Mesh ) {

				geometry = object.geometry;
171
				geometryMaterials = object.geometry.materials;
172 173 174 175
				vertices = geometry.vertices;
				faces = geometry.faces;
				faceVertexUvs = geometry.faceVertexUvs;

M
Mr.doob 已提交
176
				objectMatrixWorldRotation = object.matrixRotationWorld.extractRotation( objectMatrixWorld );
M
Mr.doob 已提交
177

178 179 180
				for ( v = 0, vl = vertices.length; v < vl; v ++ ) {

					_vertex = getNextVertexInPool();
181
					_vertex.positionWorld.copy( vertices[ v ] );
182

M
Mr.doob 已提交
183
					objectMatrixWorld.multiplyVector3( _vertex.positionWorld );
184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204

					_vertex.positionScreen.copy( _vertex.positionWorld );
					_projScreenMatrix.multiplyVector4( _vertex.positionScreen );

					_vertex.positionScreen.x /= _vertex.positionScreen.w;
					_vertex.positionScreen.y /= _vertex.positionScreen.w;

					_vertex.visible = _vertex.positionScreen.z > near && _vertex.positionScreen.z < far;

				}

				for ( f = 0, fl = faces.length; f < fl; f ++ ) {

					face = faces[ f ];

					if ( face instanceof THREE.Face3 ) {

						v1 = _vertexPool[ face.a ];
						v2 = _vertexPool[ face.b ];
						v3 = _vertexPool[ face.c ];

M
Mr.doob 已提交
205
						if ( v1.visible === true && v2.visible === true && v3.visible === true ) {
206

207 208
							visible = ( ( v3.positionScreen.x - v1.positionScreen.x ) * ( v2.positionScreen.y - v1.positionScreen.y ) -
								( v3.positionScreen.y - v1.positionScreen.y ) * ( v2.positionScreen.x - v1.positionScreen.x ) ) < 0;
209

M
Mr.doob 已提交
210
							if ( object.doubleSided === true || visible !== object.flipSided ) {
211 212 213 214 215 216 217 218 219 220 221 222

								_face = getNextFace3InPool();

								_face.v1.copy( v1 );
								_face.v2.copy( v2 );
								_face.v3.copy( v3 );

							} else {

								continue;

							}
223 224 225 226 227 228 229 230 231 232 233 234 235 236

						} else {

							continue;

						}

					} else if ( face instanceof THREE.Face4 ) {

						v1 = _vertexPool[ face.a ];
						v2 = _vertexPool[ face.b ];
						v3 = _vertexPool[ face.c ];
						v4 = _vertexPool[ face.d ];

M
Mr.doob 已提交
237
						if ( v1.visible === true && v2.visible === true && v3.visible === true && v4.visible === true ) {
238 239 240 241 242 243 244

							visible = ( v4.positionScreen.x - v1.positionScreen.x ) * ( v2.positionScreen.y - v1.positionScreen.y ) -
								( v4.positionScreen.y - v1.positionScreen.y ) * ( v2.positionScreen.x - v1.positionScreen.x ) < 0 ||
								( v2.positionScreen.x - v3.positionScreen.x ) * ( v4.positionScreen.y - v3.positionScreen.y ) -
								( v2.positionScreen.y - v3.positionScreen.y ) * ( v4.positionScreen.x - v3.positionScreen.x ) < 0;


M
Mr.doob 已提交
245
							if ( object.doubleSided === true || visible !== object.flipSided ) {
246 247 248 249 250 251 252 253 254

								_face = getNextFace4InPool();

								_face.v1.copy( v1 );
								_face.v2.copy( v2 );
								_face.v3.copy( v3 );
								_face.v4.copy( v4 );

							} else {
255

256
								continue;
257

258
							}
259 260 261 262 263 264 265 266 267 268

						} else {

							continue;

						}

					}

					_face.normalWorld.copy( face.normal );
M
Mr.doob 已提交
269
					if ( visible === false && ( object.flipSided === true || object.doubleSided === true ) ) _face.normalWorld.negate();
M
Mr.doob 已提交
270
					objectMatrixWorldRotation.multiplyVector3( _face.normalWorld );
271 272

					_face.centroidWorld.copy( face.centroid );
M
Mr.doob 已提交
273
					objectMatrixWorld.multiplyVector3( _face.centroidWorld );
274 275 276 277 278 279 280 281 282 283

					_face.centroidScreen.copy( _face.centroidWorld );
					_projScreenMatrix.multiplyVector3( _face.centroidScreen );

					faceVertexNormals = face.vertexNormals;

					for ( n = 0, nl = faceVertexNormals.length; n < nl; n ++ ) {

						normal = _face.vertexNormalsWorld[ n ];
						normal.copy( faceVertexNormals[ n ] );
M
Mr.doob 已提交
284 285 286

						if ( visible === false && ( object.flipSided === true || object.doubleSided === true ) ) normal.negate();

M
Mr.doob 已提交
287
						objectMatrixWorldRotation.multiplyVector3( normal );
288 289 290 291 292 293 294

					}

					for ( c = 0, cl = faceVertexUvs.length; c < cl; c ++ ) {

						uvs = faceVertexUvs[ c ][ f ];

M
Mr.doob 已提交
295
						if ( uvs === undefined ) continue;
296 297 298 299 300 301 302 303 304

						for ( u = 0, ul = uvs.length; u < ul; u ++ ) {

							_face.uvs[ c ][ u ] = uvs[ u ];

						}

					}

305
					_face.material = object.material;
306
					_face.faceMaterial = face.materialIndex !== null ? geometryMaterials[ face.materialIndex ] : null;
307 308 309

					_face.z = _face.centroidScreen.z;

310
					_renderData.elements.push( _face );
311 312 313

				}

M
Mr.doob 已提交
314
			} else if ( object instanceof THREE.Line ) {
M
Mr.doob 已提交
315 316

				_projScreenobjectMatrixWorld.multiply( _projScreenMatrix, objectMatrixWorld );
317 318

				vertices = object.geometry.vertices;
M
Mr.doob 已提交
319
				
320
				v1 = getNextVertexInPool();
321
				v1.positionScreen.copy( vertices[ 0 ] );
M
Mr.doob 已提交
322
				_projScreenobjectMatrixWorld.multiplyVector4( v1.positionScreen );
323

M
Mr.doob 已提交
324 325 326 327
				// Handle LineStrip and LinePieces
				var step = object.type === THREE.LinePieces ? 2 : 1;

				for ( v = 1, vl = vertices.length; v < vl; v ++ ) {
328 329

					v1 = getNextVertexInPool();
330
					v1.positionScreen.copy( vertices[ v ] );
M
Mr.doob 已提交
331
					_projScreenobjectMatrixWorld.multiplyVector4( v1.positionScreen );
332

M
Mr.doob 已提交
333
					if ( ( v + 1 ) % step > 0 ) continue;
M
Mr.doob 已提交
334

335 336 337 338 339
					v2 = _vertexPool[ _vertexCount - 2 ];

					_clippedVertex1PositionScreen.copy( v1.positionScreen );
					_clippedVertex2PositionScreen.copy( v2.positionScreen );

M
Mr.doob 已提交
340
					if ( clipLine( _clippedVertex1PositionScreen, _clippedVertex2PositionScreen ) === true ) {
341 342 343 344 345 346 347 348 349 350 351

						// Perform the perspective divide
						_clippedVertex1PositionScreen.multiplyScalar( 1 / _clippedVertex1PositionScreen.w );
						_clippedVertex2PositionScreen.multiplyScalar( 1 / _clippedVertex2PositionScreen.w );

						_line = getNextLineInPool();
						_line.v1.positionScreen.copy( _clippedVertex1PositionScreen );
						_line.v2.positionScreen.copy( _clippedVertex2PositionScreen );

						_line.z = Math.max( _clippedVertex1PositionScreen.z, _clippedVertex2PositionScreen.z );

352
						_line.material = object.material;
353

354
						_renderData.elements.push( _line );
355 356

					}
M
Mr.doob 已提交
357

358 359
				}

M
Mr.doob 已提交
360
			}
M
Mr.doob 已提交
361

M
Mr.doob 已提交
362 363 364 365
		}

		for ( o = 0, ol = _renderData.sprites.length; o < ol; o++ ) {

M
Mr.doob 已提交
366
			object = _renderData.sprites[ o ].object;
M
Mr.doob 已提交
367 368 369 370

			objectMatrixWorld = object.matrixWorld;

			if ( object instanceof THREE.Particle ) {
371

372
				_vector4.set( objectMatrixWorld.elements[12], objectMatrixWorld.elements[13], objectMatrixWorld.elements[14], 1 );
373 374 375 376 377 378 379 380 381 382 383 384 385
				_projScreenMatrix.multiplyVector4( _vector4 );

				_vector4.z /= _vector4.w;

				if ( _vector4.z > 0 && _vector4.z < 1 ) {

					_particle = getNextParticleInPool();
					_particle.x = _vector4.x / _vector4.w;
					_particle.y = _vector4.y / _vector4.w;
					_particle.z = _vector4.z;

					_particle.rotation = object.rotation.z;

386 387
					_particle.scale.x = object.scale.x * Math.abs( _particle.x - ( _vector4.x + camera.projectionMatrix.elements[0] ) / ( _vector4.w + camera.projectionMatrix.elements[12] ) );
					_particle.scale.y = object.scale.y * Math.abs( _particle.y - ( _vector4.y + camera.projectionMatrix.elements[5] ) / ( _vector4.w + camera.projectionMatrix.elements[13] ) );
388

389
					_particle.material = object.material;
390

391
					_renderData.elements.push( _particle );
392 393 394 395 396 397 398

				}

			}

		}

399
		sort && _renderData.elements.sort( painterSort );
400

401
		return _renderData;
402 403 404 405 406

	};

	// Pools

M
Mr.doob 已提交
407 408
	function getNextObjectInPool() {

M
Mr.doob 已提交
409 410 411 412 413 414 415 416 417 418 419 420
		var object;

		if ( _objectCount === _objectPool.length ) {

			object = new THREE.RenderableObject();
			_objectPool.push( object );

		} else {

			object =  _objectPool[ _objectCount ];

		}
M
Mr.doob 已提交
421 422 423 424 425 426 427

		_objectCount ++;

		return object;

	}

428 429
	function getNextVertexInPool() {

M
Mr.doob 已提交
430 431 432 433 434 435 436 437 438 439 440 441
		var vertex;

		if ( _vertexCount === _vertexPool.length ) {

			vertex = new THREE.RenderableVertex();
			_vertexPool.push( vertex );

		} else {

			vertex =  _vertexPool[ _vertexCount ];

		}
442 443 444 445 446 447 448 449 450

		_vertexCount ++;

		return vertex;

	}

	function getNextFace3InPool() {

M
Mr.doob 已提交
451 452 453 454 455 456 457 458 459 460 461 462
		var face;

		if ( _face3Count === _face3Pool.length ) {

			face = new THREE.RenderableFace3();
			_face3Pool.push( face );

		} else {

			face =  _face3Pool[ _face3Count ];

		}
463 464 465 466 467

		_face3Count ++;

		return face;

M
Mr.doob 已提交
468

469 470 471 472
	}

	function getNextFace4InPool() {

M
Mr.doob 已提交
473 474 475 476 477 478 479 480 481 482 483 484
		var face;

		if ( _face4Count === _face4Pool.length ) {

			face = new THREE.RenderableFace4();
			_face4Pool.push( face );

		} else {

			face =  _face4Pool[ _face4Count ];

		}
485 486 487 488 489 490 491 492 493

		_face4Count ++;

		return face;

	}

	function getNextLineInPool() {

M
Mr.doob 已提交
494 495 496 497 498 499 500 501 502 503 504 505
		var line;

		if ( _lineCount === _linePool.length ) {

			line = new THREE.RenderableLine();
			_linePool.push( line );

		} else {

			line =  _linePool[ _lineCount ];

		}
506 507 508 509 510 511 512 513 514

		_lineCount ++;

		return line;

	}

	function getNextParticleInPool() {

M
Mr.doob 已提交
515 516 517 518 519 520 521 522 523 524 525 526 527
		var particle;

		if ( _particleCount === _particlePool.length ) {

			particle = new THREE.RenderableParticle();
			_particlePool.push( particle );

		} else {

			particle =  _particlePool[ _particleCount ];

		}

528
		_particleCount ++;
M
Mr.doob 已提交
529

530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612
		return particle;

	}

	//

	function painterSort( a, b ) {

		return b.z - a.z;

	}

	function clipLine( s1, s2 ) {

		var alpha1 = 0, alpha2 = 1,

		// Calculate the boundary coordinate of each vertex for the near and far clip planes,
		// Z = -1 and Z = +1, respectively.
		bc1near =  s1.z + s1.w,
		bc2near =  s2.z + s2.w,
		bc1far =  - s1.z + s1.w,
		bc2far =  - s2.z + s2.w;

		if ( bc1near >= 0 && bc2near >= 0 && bc1far >= 0 && bc2far >= 0 ) {

			// Both vertices lie entirely within all clip planes.
			return true;

		} else if ( ( bc1near < 0 && bc2near < 0) || (bc1far < 0 && bc2far < 0 ) ) {

			// Both vertices lie entirely outside one of the clip planes.
			return false;

		} else {

			// The line segment spans at least one clip plane.

			if ( bc1near < 0 ) {

				// v1 lies outside the near plane, v2 inside
				alpha1 = Math.max( alpha1, bc1near / ( bc1near - bc2near ) );

			} else if ( bc2near < 0 ) {

				// v2 lies outside the near plane, v1 inside
				alpha2 = Math.min( alpha2, bc1near / ( bc1near - bc2near ) );

			}

			if ( bc1far < 0 ) {

				// v1 lies outside the far plane, v2 inside
				alpha1 = Math.max( alpha1, bc1far / ( bc1far - bc2far ) );

			} else if ( bc2far < 0 ) {

				// v2 lies outside the far plane, v2 inside
				alpha2 = Math.min( alpha2, bc1far / ( bc1far - bc2far ) );

			}

			if ( alpha2 < alpha1 ) {

				// The line segment spans two boundaries, but is outside both of them.
				// (This can't happen when we're only clipping against just near/far but good
				//  to leave the check here for future usage if other clip planes are added.)
				return false;

			} else {

				// Update the s1 and s2 vertices to match the clipped line segment.
				s1.lerpSelf( s2, alpha1 );
				s2.lerpSelf( s1, 1 - alpha2 );

				return true;

			}

		}

	}

};