131 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
		
		
			
		
	
	
			131 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
|  | import { | ||
|  | 	FileLoader, | ||
|  | 	Group, | ||
|  | 	Loader, | ||
|  | 	LoadingManager | ||
|  | } from 'three'; | ||
|  | import { ColladaLoader } from '../loaders/ColladaLoader.js'; | ||
|  | import * as fflate from '../libs/fflate.module.js'; | ||
|  | 
 | ||
|  | class KMZLoader extends Loader { | ||
|  | 
 | ||
|  | 	constructor( manager ) { | ||
|  | 
 | ||
|  | 		super( manager ); | ||
|  | 
 | ||
|  | 	} | ||
|  | 
 | ||
|  | 	load( url, onLoad, onProgress, onError ) { | ||
|  | 
 | ||
|  | 		const scope = this; | ||
|  | 
 | ||
|  | 		const loader = new FileLoader( scope.manager ); | ||
|  | 		loader.setPath( scope.path ); | ||
|  | 		loader.setResponseType( 'arraybuffer' ); | ||
|  | 		loader.setRequestHeader( scope.requestHeader ); | ||
|  | 		loader.setWithCredentials( scope.withCredentials ); | ||
|  | 		loader.load( url, function ( text ) { | ||
|  | 
 | ||
|  | 			try { | ||
|  | 
 | ||
|  | 				onLoad( scope.parse( text ) ); | ||
|  | 
 | ||
|  | 			} catch ( e ) { | ||
|  | 
 | ||
|  | 				if ( onError ) { | ||
|  | 
 | ||
|  | 					onError( e ); | ||
|  | 
 | ||
|  | 				} else { | ||
|  | 
 | ||
|  | 					console.error( e ); | ||
|  | 
 | ||
|  | 				} | ||
|  | 
 | ||
|  | 				scope.manager.itemError( url ); | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 		}, onProgress, onError ); | ||
|  | 
 | ||
|  | 	} | ||
|  | 
 | ||
|  | 	parse( data ) { | ||
|  | 
 | ||
|  | 		function findFile( url ) { | ||
|  | 
 | ||
|  | 			for ( const path in zip ) { | ||
|  | 
 | ||
|  | 				if ( path.slice( - url.length ) === url ) { | ||
|  | 
 | ||
|  | 					return zip[ path ]; | ||
|  | 
 | ||
|  | 				} | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 		} | ||
|  | 
 | ||
|  | 		const manager = new LoadingManager(); | ||
|  | 		manager.setURLModifier( function ( url ) { | ||
|  | 
 | ||
|  | 			const image = findFile( url ); | ||
|  | 
 | ||
|  | 			if ( image ) { | ||
|  | 
 | ||
|  | 				console.log( 'Loading', url ); | ||
|  | 
 | ||
|  | 				const blob = new Blob( [ image.buffer ], { type: 'application/octet-stream' } ); | ||
|  | 				return URL.createObjectURL( blob ); | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 			return url; | ||
|  | 
 | ||
|  | 		} ); | ||
|  | 
 | ||
|  | 		//
 | ||
|  | 
 | ||
|  | 		const zip = fflate.unzipSync( new Uint8Array( data ) ); | ||
|  | 
 | ||
|  | 		if ( zip[ 'doc.kml' ] ) { | ||
|  | 
 | ||
|  | 			const xml = new DOMParser().parseFromString( fflate.strFromU8( zip[ 'doc.kml' ] ), 'application/xml' ); | ||
|  | 
 | ||
|  | 			const model = xml.querySelector( 'Placemark Model Link href' ); | ||
|  | 
 | ||
|  | 			if ( model ) { | ||
|  | 
 | ||
|  | 				const loader = new ColladaLoader( manager ); | ||
|  | 				return loader.parse( fflate.strFromU8( zip[ model.textContent ] ) ); | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 		} else { | ||
|  | 
 | ||
|  | 			console.warn( 'KMZLoader: Missing doc.kml file.' ); | ||
|  | 
 | ||
|  | 			for ( const path in zip ) { | ||
|  | 
 | ||
|  | 				const extension = path.split( '.' ).pop().toLowerCase(); | ||
|  | 
 | ||
|  | 				if ( extension === 'dae' ) { | ||
|  | 
 | ||
|  | 					const loader = new ColladaLoader( manager ); | ||
|  | 					return loader.parse( fflate.strFromU8( zip[ path ] ) ); | ||
|  | 
 | ||
|  | 				} | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 		} | ||
|  | 
 | ||
|  | 		console.error( 'KMZLoader: Couldn\'t find .dae file.' ); | ||
|  | 		return { scene: new Group() }; | ||
|  | 
 | ||
|  | 	} | ||
|  | 
 | ||
|  | } | ||
|  | 
 | ||
|  | export { KMZLoader }; |