WebGLGeometries.js 2.4 KB
Newer Older
M
Mr.doob 已提交
1
/**
M
Mr.doob 已提交
2 3 4 5
 * @author mrdoob / http://mrdoob.com/
 */

import { BufferGeometry } from '../../core/BufferGeometry';
M
Mr.doob 已提交
6

M
Mr.doob 已提交
7
function WebGLGeometries( gl, properties, info ) {
M
Mr.doob 已提交
8 9 10

	var geometries = {};

11 12 13 14 15
	function onGeometryDispose( event ) {

		var geometry = event.target;
		var buffergeometry = geometries[ geometry.id ];

16 17 18 19 20 21
		if ( buffergeometry.index !== null ) {

			deleteAttribute( buffergeometry.index );

		}

22 23
		deleteAttributes( buffergeometry.attributes );

24 25 26 27
		geometry.removeEventListener( 'dispose', onGeometryDispose );

		delete geometries[ geometry.id ];

M
Mr.doob 已提交
28 29
		// TODO

30
		var property = properties.get( geometry );
M
Mr.doob 已提交
31 32 33 34 35 36

		if ( property.wireframe ) {

			deleteAttribute( property.wireframe );

		}
37

38 39
		properties.delete( geometry );

40
		var bufferproperty = properties.get( buffergeometry );
M
Mr.doob 已提交
41 42 43 44 45 46

		if ( bufferproperty.wireframe ) {

			deleteAttribute( bufferproperty.wireframe );

		}
47 48 49

		properties.delete( buffergeometry );

M
Mr.doob 已提交
50 51
		//

M
Mr.doob 已提交
52 53 54 55
		info.memory.geometries --;

	}

56
	function getAttributeBuffer( attribute ) {
57

M
Mr.doob 已提交
58
		if ( attribute.isInterleavedBufferAttribute ) {
M
Mr.doob 已提交
59

60 61 62 63 64 65
			return properties.get( attribute.data ).__webglBuffer;

		}

		return properties.get( attribute ).__webglBuffer;

66 67
	}

68
	function deleteAttribute( attribute ) {
M
Mr.doob 已提交
69

70
		var buffer = getAttributeBuffer( attribute );
M
Mr.doob 已提交
71

72
		if ( buffer !== undefined ) {
M
Mr.doob 已提交
73

74 75
			gl.deleteBuffer( buffer );
			removeAttributeBuffer( attribute );
M
Mr.doob 已提交
76

77
		}
M
Mr.doob 已提交
78

79 80 81 82 83 84 85
	}

	function deleteAttributes( attributes ) {

		for ( var name in attributes ) {

			deleteAttribute( attributes[ name ] );
M
Mr.doob 已提交
86 87 88 89 90

		}

	}

91
	function removeAttributeBuffer( attribute ) {
92

M
Mr.doob 已提交
93
		if ( attribute.isInterleavedBufferAttribute ) {
94 95 96 97 98 99 100

			properties.delete( attribute.data );

		} else {

			properties.delete( attribute );

101 102 103 104
		}

	}

M
Mr.doob 已提交
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
	return {

		get: function ( object ) {

			var geometry = object.geometry;

			if ( geometries[ geometry.id ] !== undefined ) {

				return geometries[ geometry.id ];

			}

			geometry.addEventListener( 'dispose', onGeometryDispose );

			var buffergeometry;

			if ( geometry.isBufferGeometry ) {

				buffergeometry = geometry;

			} else if ( geometry.isGeometry ) {

				if ( geometry._bufferGeometry === undefined ) {

					geometry._bufferGeometry = new BufferGeometry().setFromObject( object );

				}

				buffergeometry = geometry._bufferGeometry;

			}

			geometries[ geometry.id ] = buffergeometry;

			info.memory.geometries ++;

			return buffergeometry;

		}

	};
146

M
Mr.doob 已提交
147
}
R
Rich Harris 已提交
148 149


150
export { WebGLGeometries };