KMZLoader.js 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. import {
  2. FileLoader,
  3. Group,
  4. Loader,
  5. LoadingManager
  6. } from 'three';
  7. import { ColladaLoader } from '../loaders/ColladaLoader.js';
  8. import * as fflate from '../libs/fflate.module.js';
  9. /**
  10. * A loader for the KMZ format.
  11. *
  12. * ```js
  13. * const loader = new KMZLoader();
  14. * const kmz = await loader.loadAsync( './models/kmz/Box.kmz' );
  15. *
  16. * scene.add( kmz.scene );
  17. * ```
  18. *
  19. * @augments Loader
  20. * @three_import import { KMZLoader } from 'three/addons/loaders/KMZLoader.js';
  21. */
  22. class KMZLoader extends Loader {
  23. /**
  24. * Constructs a new KMZ loader.
  25. *
  26. * @param {LoadingManager} [manager] - The loading manager.
  27. */
  28. constructor( manager ) {
  29. super( manager );
  30. }
  31. /**
  32. * Starts loading from the given URL and passes the loaded KMZ asset
  33. * to the `onLoad()` callback.
  34. *
  35. * @param {string} url - The path/URL of the file to be loaded. This can also be a data URI.
  36. * @param {function({scene:Group})} onLoad - Executed when the loading process has been finished.
  37. * @param {onProgressCallback} onProgress - Executed while the loading is in progress.
  38. * @param {onErrorCallback} onError - Executed when errors occur.
  39. */
  40. load( url, onLoad, onProgress, onError ) {
  41. const scope = this;
  42. const loader = new FileLoader( scope.manager );
  43. loader.setPath( scope.path );
  44. loader.setResponseType( 'arraybuffer' );
  45. loader.setRequestHeader( scope.requestHeader );
  46. loader.setWithCredentials( scope.withCredentials );
  47. loader.load( url, function ( text ) {
  48. try {
  49. onLoad( scope.parse( text ) );
  50. } catch ( e ) {
  51. if ( onError ) {
  52. onError( e );
  53. } else {
  54. console.error( e );
  55. }
  56. scope.manager.itemError( url );
  57. }
  58. }, onProgress, onError );
  59. }
  60. /**
  61. * Parses the given KMZ data and returns an object holding the scene.
  62. *
  63. * @param {ArrayBuffer} data - The raw KMZ data as an array buffer.
  64. * @return {{scene:Group}} The parsed KMZ asset.
  65. */
  66. parse( data ) {
  67. function findFile( url ) {
  68. for ( const path in zip ) {
  69. if ( path.slice( - url.length ) === url ) {
  70. return zip[ path ];
  71. }
  72. }
  73. }
  74. const manager = new LoadingManager();
  75. manager.setURLModifier( function ( url ) {
  76. const image = findFile( url );
  77. if ( image ) {
  78. console.log( 'Loading', url );
  79. const blob = new Blob( [ image.buffer ], { type: 'application/octet-stream' } );
  80. return URL.createObjectURL( blob );
  81. }
  82. return url;
  83. } );
  84. //
  85. const zip = fflate.unzipSync( new Uint8Array( data ) );
  86. if ( zip[ 'doc.kml' ] ) {
  87. const xml = new DOMParser().parseFromString( fflate.strFromU8( zip[ 'doc.kml' ] ), 'application/xml' );
  88. const model = xml.querySelector( 'Placemark Model Link href' );
  89. if ( model ) {
  90. const loader = new ColladaLoader( manager );
  91. return loader.parse( fflate.strFromU8( zip[ model.textContent ] ) );
  92. }
  93. } else {
  94. console.warn( 'KMZLoader: Missing doc.kml file.' );
  95. for ( const path in zip ) {
  96. const extension = path.split( '.' ).pop().toLowerCase();
  97. if ( extension === 'dae' ) {
  98. const loader = new ColladaLoader( manager );
  99. return loader.parse( fflate.strFromU8( zip[ path ] ) );
  100. }
  101. }
  102. }
  103. console.error( 'KMZLoader: Couldn\'t find .dae file.' );
  104. return { scene: new Group() };
  105. }
  106. }
  107. export { KMZLoader };