KMZLoader.js 2.1 KB

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