KMZLoader.js 2.3 KB

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