BinaryLoader.js 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. import * as THREE from "../../libs/three.js/build/three.module.js";
  2. import {Version} from "../Version.js";
  3. import {XHRFactory} from "../XHRFactory.js";
  4. //加载 解析点云
  5. export class BinaryLoader{
  6. constructor(version, boundingBox, scale){
  7. if (typeof (version) === 'string') {
  8. this.version = new Version(version);
  9. } else {
  10. this.version = version;
  11. }
  12. this.boundingBox = boundingBox;
  13. this.scale = scale;
  14. }
  15. load(node){
  16. if (node.loaded) {
  17. return;
  18. }
  19. let url = node.getURL();
  20. if (this.version.equalOrHigher('1.4')) {
  21. url += '.bin';
  22. }
  23. url += '?m='+node.pcoGeometry.timeStamp //add
  24. let xhr = XHRFactory.createXMLHttpRequest();
  25. xhr.open('GET', url, true);
  26. xhr.responseType = 'arraybuffer';
  27. xhr.overrideMimeType('text/plain; charset=x-user-defined');
  28. xhr.onreadystatechange = () => {
  29. if (xhr.readyState === 4) {
  30. if((xhr.status === 200 || xhr.status === 0) && xhr.response !== null){
  31. let buffer = xhr.response;
  32. this.parse(node, buffer);
  33. } else {
  34. //console.error(`Failed to load file! HTTP status: ${xhr.status}, file: ${url}`);
  35. throw new Error(`Failed to load file! HTTP status: ${xhr.status}, file: ${url}`);
  36. }
  37. }
  38. };
  39. try {
  40. xhr.send(null);
  41. } catch (e) {
  42. console.log('fehler beim laden der punktwolke: ' + e);
  43. }
  44. };
  45. parse(node, buffer){ //解析点云
  46. let pointAttributes = node.pcoGeometry.pointAttributes;
  47. let numPoints = buffer.byteLength / node.pcoGeometry.pointAttributes.byteSize;
  48. if (this.version.upTo('1.5')) {
  49. node.numPoints = numPoints;
  50. }
  51. let workerPath = Potree.scriptPath + '/workers/BinaryDecoderWorker.js';
  52. let worker = Potree.workerPool.getWorker(workerPath);
  53. worker.onmessage = function (e) {
  54. let data = e.data;
  55. let buffers = data.attributeBuffers;
  56. let tightBoundingBox = new THREE.Box3(
  57. new THREE.Vector3().fromArray(data.tightBoundingBox.min),
  58. new THREE.Vector3().fromArray(data.tightBoundingBox.max)
  59. );
  60. Potree.workerPool.returnWorker(workerPath, worker);
  61. let geometry = new THREE.BufferGeometry();
  62. for(let property in buffers){
  63. let buffer = buffers[property].buffer;
  64. let batchAttribute = buffers[property].attribute;
  65. if (property === "POSITION_CARTESIAN") {
  66. geometry.setAttribute('position', new THREE.BufferAttribute(new Float32Array(buffer), 3));
  67. } else if (property === "rgba") {
  68. geometry.setAttribute("rgba", new THREE.BufferAttribute(new Uint8Array(buffer), 4, true));
  69. } else if (property === "NORMAL_SPHEREMAPPED") {
  70. geometry.setAttribute('normal', new THREE.BufferAttribute(new Float32Array(buffer), 3));
  71. } else if (property === "NORMAL_OCT16") {
  72. geometry.setAttribute('normal', new THREE.BufferAttribute(new Float32Array(buffer), 3));
  73. } else if (property === "NORMAL") {
  74. geometry.setAttribute('normal', new THREE.BufferAttribute(new Float32Array(buffer), 3));
  75. } else if (property === "INDICES") {
  76. let bufferAttribute = new THREE.BufferAttribute(new Uint8Array(buffer), 4);
  77. bufferAttribute.normalized = true;
  78. geometry.setAttribute('indices', bufferAttribute);
  79. } else if (property === "SPACING") {
  80. let bufferAttribute = new THREE.BufferAttribute(new Float32Array(buffer), 1);
  81. geometry.setAttribute('spacing', bufferAttribute);
  82. } else {
  83. const bufferAttribute = new THREE.BufferAttribute(new Float32Array(buffer), 1);
  84. bufferAttribute.potree = {
  85. offset: buffers[property].offset,
  86. scale: buffers[property].scale,
  87. preciseBuffer: buffers[property].preciseBuffer,
  88. range: batchAttribute.range,
  89. };
  90. geometry.setAttribute(property, bufferAttribute);
  91. const attribute = pointAttributes.attributes.find(a => a.name === batchAttribute.name);
  92. attribute.range[0] = Math.min(attribute.range[0], batchAttribute.range[0]);
  93. attribute.range[1] = Math.max(attribute.range[1], batchAttribute.range[1]);
  94. if(node.getLevel() === 0){
  95. attribute.initialRange = batchAttribute.range;
  96. }
  97. }
  98. }
  99. tightBoundingBox.max.sub(tightBoundingBox.min);
  100. tightBoundingBox.min.set(0, 0, 0);
  101. let numPoints = e.data.buffer.byteLength / pointAttributes.byteSize;
  102. node.numPoints = numPoints;
  103. node.geometry = geometry;
  104. node.mean = new THREE.Vector3(...data.mean);
  105. node.tightBoundingBox = tightBoundingBox;
  106. node.loaded = true;
  107. node.loading = false;
  108. node.estimatedSpacing = data.estimatedSpacing;
  109. Potree.numNodesLoading--;
  110. };
  111. let message = {
  112. buffer: buffer,
  113. pointAttributes: pointAttributes,
  114. version: this.version.version,
  115. min: [ node.boundingBox.min.x, node.boundingBox.min.y, node.boundingBox.min.z ],
  116. offset: [node.pcoGeometry.offset.x, node.pcoGeometry.offset.y, node.pcoGeometry.offset.z],
  117. scale: this.scale,
  118. spacing: node.spacing,
  119. hasChildren: node.hasChildren,
  120. name: node.name
  121. };
  122. worker.postMessage(message, [message.buffer]);
  123. };
  124. }