KMZLoader.js 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132
  1. import {
  2. FileLoader,
  3. Group,
  4. Loader,
  5. LoadingManager
  6. } from '../../../build/three.module.js';
  7. import { ColladaLoader } from '../loaders/ColladaLoader.js';
  8. import * as fflate from '../libs/fflate.module.min.js';
  9. var KMZLoader = function ( manager ) {
  10. Loader.call( this, manager );
  11. };
  12. KMZLoader.prototype = Object.assign( Object.create( Loader.prototype ), {
  13. constructor: KMZLoader,
  14. load: function ( url, onLoad, onProgress, onError ) {
  15. var scope = this;
  16. var loader = new FileLoader( scope.manager );
  17. loader.setPath( scope.path );
  18. loader.setResponseType( 'arraybuffer' );
  19. loader.setRequestHeader( scope.requestHeader );
  20. loader.setWithCredentials( scope.withCredentials );
  21. loader.load( url, function ( text ) {
  22. try {
  23. onLoad( scope.parse( text ) );
  24. } catch ( e ) {
  25. if ( onError ) {
  26. onError( e );
  27. } else {
  28. console.error( e );
  29. }
  30. scope.manager.itemError( url );
  31. }
  32. }, onProgress, onError );
  33. },
  34. parse: function ( data ) {
  35. function findFile( url ) {
  36. for ( var path in zip ) {
  37. if ( path.substr( - url.length ) === url ) {
  38. return zip[ path ];
  39. }
  40. }
  41. }
  42. var manager = new LoadingManager();
  43. manager.setURLModifier( function ( url ) {
  44. var image = findFile( url );
  45. if ( image ) {
  46. console.log( 'Loading', url );
  47. var blob = new Blob( [ image.buffer ], { type: 'application/octet-stream' } );
  48. return URL.createObjectURL( blob );
  49. }
  50. return url;
  51. } );
  52. //
  53. var zip = fflate.unzipSync( new Uint8Array( data ) ); // eslint-disable-line no-undef
  54. if ( zip[ 'doc.kml' ] ) {
  55. var xml = new DOMParser().parseFromString( fflate.strFromU8( zip[ 'doc.kml' ] ), 'application/xml' ); // eslint-disable-line no-undef
  56. var model = xml.querySelector( 'Placemark Model Link href' );
  57. if ( model ) {
  58. var loader = new ColladaLoader( manager );
  59. return loader.parse( fflate.strFromU8( zip[ model.textContent ] ) ); // eslint-disable-line no-undef
  60. }
  61. } else {
  62. console.warn( 'KMZLoader: Missing doc.kml file.' );
  63. for ( var path in zip ) {
  64. var extension = path.split( '.' ).pop().toLowerCase();
  65. if ( extension === 'dae' ) {
  66. var loader = new ColladaLoader( manager );
  67. return loader.parse( fflate.strFromU8( zip[ path ] ) ); // eslint-disable-line no-undef
  68. }
  69. }
  70. }
  71. console.error( 'KMZLoader: Couldn\'t find .dae file.' );
  72. return { scene: new Group() };
  73. }
  74. } );
  75. export { KMZLoader };