123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163 |
- import {
- FileLoader,
- Group,
- Loader,
- LoadingManager
- } from 'three';
- import { ColladaLoader } from '../loaders/ColladaLoader.js';
- import * as fflate from '../libs/fflate.module.js';
- /**
- * A loader for the KMZ format.
- *
- * ```js
- * const loader = new KMZLoader();
- * const kmz = await loader.loadAsync( './models/kmz/Box.kmz' );
- *
- * scene.add( kmz.scene );
- * ```
- *
- * @augments Loader
- * @three_import import { KMZLoader } from 'three/addons/loaders/KMZLoader.js';
- */
- class KMZLoader extends Loader {
- /**
- * Constructs a new KMZ loader.
- *
- * @param {LoadingManager} [manager] - The loading manager.
- */
- constructor( manager ) {
- super( manager );
- }
- /**
- * Starts loading from the given URL and passes the loaded KMZ asset
- * to the `onLoad()` callback.
- *
- * @param {string} url - The path/URL of the file to be loaded. This can also be a data URI.
- * @param {function({scene:Group})} onLoad - Executed when the loading process has been finished.
- * @param {onProgressCallback} onProgress - Executed while the loading is in progress.
- * @param {onErrorCallback} onError - Executed when errors occur.
- */
- 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 );
- }
- /**
- * Parses the given KMZ data and returns an object holding the scene.
- *
- * @param {ArrayBuffer} data - The raw KMZ data as an array buffer.
- * @return {{scene:Group}} The parsed KMZ asset.
- */
- 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 };
|