@dvt3d/splat-mesh 1.0.0 → 1.1.0

package/dist/index.js

@@ -1 +1,3 @@
-import{BufferAttribute as I,BufferGeometry as P,CustomBlending as W,DataTexture as A,DynamicDrawUsage as V,FloatType as U,InstancedBufferAttribute as z,InstancedBufferGeometry as B,Mesh as R,OneFactor as j,RGBAFormat as G,RGBAIntegerFormat as H,ShaderMaterial as E,UnsignedIntType as q,Vector4 as $,Frustum as J,Matrix4 as L}from"three";var g="precision highp sampler2D;precision highp usampler2D;out vec4 vColor;out vec2 vPosition;uniform vec2 viewport;uniform mat4 gsModelViewMatrix;attribute uint splatIndex;uniform sampler2D centerAndScaleTexture;uniform usampler2D covAndColorTexture;vec2 unpackInt16(in uint value){int v=int(value);int v0=v>>16;int v1=(v&0xFFFF);if((v&0x8000)!=0)v1|=0xFFFF0000;return vec2(float(v1),float(v0));}vec4 calcCovVectors(vec3 viewPos,mat3 Vrk){float focal=(viewport.y/2.0)*abs(projectionMatrix[1][1]);mat3 J=mat3(focal/viewPos.z,0.,-(focal*viewPos.x)/(viewPos.z*viewPos.z),0.,focal/viewPos.z,-(focal*viewPos.y)/(viewPos.z*viewPos.z),0.,0.,0.);mat3 W=transpose(mat3(gsModelViewMatrix));mat3 T=W*J;mat3 cov=transpose(T)*Vrk*T;float diagonal1=cov[0][0]+0.3;float offDiagonal=cov[0][1];float diagonal2=cov[1][1]+0.3;float mid=0.5*(diagonal1+diagonal2);float radius=length(vec2((diagonal1-diagonal2)/2.0,offDiagonal));float lambda1=mid+radius;float lambda2=max(mid-radius,0.1);vec2 diagonalVector=normalize(vec2(offDiagonal,lambda1-diagonal1));vec2 v1=min(sqrt(2.0*lambda1),1024.0)*diagonalVector;vec2 v2=min(sqrt(2.0*lambda2),1024.0)*vec2(diagonalVector.y,-diagonalVector.x);return vec4(v1,v2);}void main(){ivec2 texSize=textureSize(centerAndScaleTexture,0);ivec2 texPos=ivec2(splatIndex % uint(texSize.x),splatIndex/uint(texSize.x));vec4 centerAndScaleData=texelFetch(centerAndScaleTexture,texPos,0);vec4 viewPos=gsModelViewMatrix*vec4(centerAndScaleData.xyz,1);vec4 pos2d=projectionMatrix*viewPos;float clip=1.2*pos2d.w;if(pos2d.z<-pos2d.w||pos2d.x<-clip||pos2d.x>clip||pos2d.y<-clip||pos2d.y>clip){gl_Position=vec4(0.0,0.0,2.0,1.0);return;}uvec4 covAndColorData=texelFetch(covAndColorTexture,texPos,0);vec2 cov3D_M11_M12=unpackInt16(covAndColorData.x)*centerAndScaleData.w;vec2 cov3D_M13_M22=unpackInt16(covAndColorData.y)*centerAndScaleData.w;vec2 cov3D_M23_M33=unpackInt16(covAndColorData.z)*centerAndScaleData.w;mat3 Vrk=mat3(cov3D_M11_M12.x,cov3D_M11_M12.y,cov3D_M13_M22.x,cov3D_M11_M12.y,cov3D_M13_M22.y,cov3D_M23_M33.x,cov3D_M13_M22.x,cov3D_M23_M33.x,cov3D_M23_M33.y);vec4 covVectors=calcCovVectors(viewPos.xyz,Vrk);vPosition=position.xy;uint colorUint=covAndColorData.w;vColor=vec4(float(colorUint&uint(0xFF))/255.0,float((colorUint>>uint(8))&uint(0xFF))/255.0,float((colorUint>>uint(16))&uint(0xFF))/255.0,float(colorUint>>uint(24))/255.0);vec2 vCenter=vec2(pos2d)/pos2d.w;gl_Position=vec4(vCenter+(position.x*covVectors.zw+position.y*covVectors.xy)/viewport*2.0,pos2d.z/pos2d.w,1.0);}";var S="in vec4 vColor;in vec2 vPosition;void main(){float A=-dot(vPosition,vPosition);if(A<-4.0)discard;float B=exp(A)*vColor.a;gl_FragColor=vec4(vColor.rgb,B);}";var x=class{constructor(){this._intervalTime=200,this._stableStopTime=3e3,this._isSorting=!1,this._dirty=!0,this._lastMvMatrix=null,this._lastSortTime=0,this._stableSince=0}set isSorting(t){this._isSorting=t}get isSorting(){return this._isSorting}set dirty(t){this._dirty=t}get dirty(){return this._dirty}set intervalTime(t){this._intervalTime=t}get intervalTime(){return this._intervalTime}set stableStopTime(t){this._stableStopTime=t}get stableStopTime(){return this._stableStopTime}_isMatrixChanged(t,e){return t?!t.equals(e):!0}tick(t,e){let r=performance.now(),i=this._isMatrixChanged(this._lastMvMatrix,t);i?this._stableSince=r:this._stableSince===0&&(this._stableSince=r);let s=r-this._lastSortTime>=this._intervalTime,h=r-this._stableSince<this._stableStopTime;return!this._isSorting&&s&&(this._dirty||i||h)&&(this._isSorting=!0,this._dirty=!1,this._lastSortTime=r,e()),this._lastMvMatrix=t,this}},M=x;var F="0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz".split("");function y(o="D"){let t=[];t[8]=t[13]=t[18]=t[23]="-",t[14]="4";let e;for(let r=0;r<36;r++)t[r]||(e=0|Math.random()*16,t[r]=F[r===19?e&3|8:e]);return o+"-"+t.join("")}function k(){let o=document.createElement("canvas"),t=o.getContext("webgl2")||o.getContext("webgl");return t.getParameter(t.MAX_TEXTURE_SIZE)}var f=k(),O=f*f,C=new P;C.setAttribute("position",new I(new Float32Array([-2,-2,0,2,2,0,-2,2,0,2,-2,0,2,2,0,-2,-2,0]),3));var v=class extends R{constructor(){super(),this._meshId=y(),this._vertexCount=0,this._textureWidth=f,this._textureHeight=1,this._threshold=-1e-5,this._centerAndScaleData=null,this._centerAndScaleTexture=null,this._rotationAndColorData=null,this._rotationAndColorTexture=null,this.geometry=new B().copy(C),this.geometry.instanceCount=1,this.material=new E({uniforms:{viewport:{value:new Float32Array([1980,1080])},centerAndScaleTexture:{value:null},covAndColorTexture:{value:null},gsModelViewMatrix:{value:null}},vertexShader:g,fragmentShader:S,blending:W,blendSrcAlpha:j,depthTest:!0,depthWrite:!1,transparent:!0}),this.material.onBeforeRender=this._onMaterialBeforeRender.bind(this),this.frustumCulled=!1,this._bounds=null,this._sortScheduler=new M,this._worker=null,this._useFrustumCulled=!1}get isSplatMesh(){return!0}set threshold(t){this._threshold=t}get threshold(){return this._threshold}set worker(t){this._worker=t}get worker(){return this._worker}get bounds(){return this._bounds}set vertexCount(t){this.setVertexCount(t)}get vertexCount(){return this._vertexCount}set useFrustumCulled(t){this._useFrustumCulled=t}get useFrustumCulled(){return this._useFrustumCulled}get sortScheduler(){return this._sortScheduler}_getModelViewMatrix(t){return t.matrixWorld.clone().invert().multiply(this.matrixWorld)}_updateTexture(t,e){t&&this._meshId===t.meshId&&(this._centerAndScaleData.set(t.out_cs),this._rotationAndColorData.set(t.out_rc),this._centerAndScaleTexture.image.data.set(this._centerAndScaleData.subarray(0,e*4),0),this._rotationAndColorTexture.image.data.set(this._rotationAndColorData.subarray(0,e*4),0),this._centerAndScaleTexture.needsUpdate=!0,this._rotationAndColorTexture.needsUpdate=!0,this._sortScheduler.dirty=!0)}_onMaterialBeforeRender(t,e,r,i,s,h){let u=this._getModelViewMatrix(r),p=s.material;p.uniforms.gsModelViewMatrix.value=u;let m=new $;t.getCurrentViewport(m),p.uniforms.viewport.value[0]=m.z,p.uniforms.viewport.value[1]=m.w,this._sortScheduler&&this._sortScheduler.tick(u,()=>{let c=u.elements,D=new Float32Array([c[2],c[6],c[10],c[14]]),a=new Float32Array(0);if(this._useFrustumCulled){a=new Float32Array(24);let n=new L;n.multiplyMatrices(r.projectionMatrix,r.matrixWorldInverse);let l=new J;l.setFromProjectionMatrix(n),l.planes.forEach((d,_)=>{a[_*4+0]=d.normal.x,a[_*4+1]=d.normal.y,a[_*4+2]=d.normal.z,a[_*4+3]=d.constant})}this._worker&&this._worker.call("sort_splats",this._meshId,D,a,this._threshold).then(n=>{if(n||(this._sortScheduler.isSorting=!1),n&&this._meshId===n.meshId){let l=new Uint32Array(n.data);this.geometry.attributes.splatIndex.set(l),this.geometry.attributes.splatIndex.needsUpdate=!0,this.geometry.instanceCount=l.length,this._sortScheduler.isSorting=!1}})})}setVertexCount(t){if(t===this._vertexCount)return;this._vertexCount=Math.min(t,O),this._textureHeight=Math.floor((this._vertexCount-1)/this._textureWidth)+1,this._centerAndScaleData=new Float32Array(this._textureWidth*this._textureHeight*4),this._centerAndScaleTexture&&this._centerAndScaleTexture.dispose(),this._centerAndScaleTexture=new A(this._centerAndScaleData,this._textureWidth,this._textureHeight,G,U),this._rotationAndColorData=new Uint32Array(this._textureWidth*this._textureHeight*4),this._rotationAndColorTexture&&this._rotationAndColorTexture.dispose(),this._rotationAndColorTexture=new A(this._rotationAndColorData,this._textureWidth,this._textureHeight,H,q),this._rotationAndColorTexture.internalFormat="RGBA32UI";let e=new Uint32Array(this._textureWidth*this._textureHeight),r=new z(e,1,!1);return r.setUsage(V),this.geometry.setAttribute("splatIndex",r),this.material.uniforms.centerAndScaleTexture.value=this._centerAndScaleTexture,this.material.uniforms.covAndColorTexture.value=this._rotationAndColorTexture,this}attachWorker(t){return this._worker=t,this}detachWorker(){return this._worker&&this._worker.call("unregister_positions",this._bufferId),this._worker=null,this}async setDataFromBuffer(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromBuffer skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromBuffer skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_buffer",this._meshId,t,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async setDataFromGeometry(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromGeometry skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromGeometry skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_geometry",this._meshId,t.attributes.position.array,t.attributes._scale.array,t.attributes._rotation.array,t.attributes.color.array,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async setDataFromSpz(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromSpz skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromSpz skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_spz",this._meshId,t.positions,t.scales,t.rotations,t.colors,t.alphas,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async computeBounds(){if(!this._worker)return console.warn("[SplatMesh] computeBounds skipped: worker not ready"),this;if(!this._bounds){let t=await this._worker.call("compute_bounds",this._meshId);this._meshId===t.meshId&&(this._bounds=t.data)}return this}dispose(){this._bounds=null,this._centerAndScaleTexture&&(this._centerAndScaleTexture.dispose(),this._centerAndScaleData=null),this._rotationAndColorTexture&&(this._rotationAndColorTexture.dispose(),this._rotationAndColorData=null),this._worker&&this._worker.call("unregister_positions",this._bufferId),this._bufferId=null,this._sortScheduler=null,this.geometry.dispose(),this.material.dispose(),this.parent=null}},b=v;var w=class{constructor(t,{timeout:e=0}={}){this._workerUrl=t,this._worker=null,this._seq=0,this._pending=new Map,this._timeout=e,this._initPromise=null,this._ready=!1,this._createWorker()}_createWorker(){let t=e=>{e.onmessage=r=>this._handleMessage(r),e.onerror=r=>this._handleError(r),e.onmessageerror=r=>console.error("[SplatWorker] Worker message error:",r)};try{this._worker=new Worker(this._workerUrl,{type:"module"}),t(this._worker);return}catch(e){console.error("Worker error:",e)}fetch(this._workerUrl).then(e=>{if(!e.ok)throw new Error(`[SplatWorker] Failed to fetch worker: ${this._workerUrl}`);return e.text()}).then(e=>{let r=new Blob([e],{type:"application/javascript"}),i=URL.createObjectURL(r);this._worker=new Worker(i),t(this._worker)})}_handleMessage(t){let{id:e,result:r,error:i}=t.data,s=this._pending.get(e);s&&(this._pending.delete(e),i?s.reject(i):s.resolve(r))}_handleError(t){console.error("[SplatWorker] Worker error:",t);for(let[e,r]of this._pending)r.reject("[SplatWorker] Worker crashed");this._pending.clear()}_collectTransferables(t){let e=[];for(let r of t)r&&(ArrayBuffer.isView(r)?e.push(r.buffer):r instanceof ArrayBuffer&&e.push(r));return e}async init(){if(this._ready)return!0;if(this._initPromise)return this._initPromise;this._initPromise=new Promise((e,r)=>{let i=++this._seq;this._pending.set(i,{resolve:e,reject:r}),this._worker.postMessage({id:i,fn:"__init_wasm__",args:[]})});let t=await this._initPromise;return this._ready=!0,t}call(t,...e){return new Promise((r,i)=>{this._worker||i(`the worker ${this._workerUrl} does not exist`);let s=++this._seq;this._pending.set(s,{resolve:r,reject:i});let h=this._collectTransferables(e);this._worker.postMessage({id:s,fn:t,args:e},h),this._timeout>0&&setTimeout(()=>{this._pending.has(s)&&(this._pending.delete(s),i(`[SplatWorker] Task "${t}" timeout after ${this._timeout}ms`))},this._timeout)})}cancel(t,e="canceled"){let r=this._pending.get(t);return r&&(r.reject(e),this._pending.delete(t)),this}dispose(){return this._worker.terminate(),this._pending.clear(),this}},T=w;typeof window!="undefined"&&(window.SplatMesh=b,window.SplatWorker=T);export{b as SplatMesh,T as SplatWorker};
+import{BufferAttribute as P,BufferGeometry as V,CustomBlending as W,DataTexture as b,DynamicDrawUsage as z,FloatType as B,InstancedBufferAttribute as R,InstancedBufferGeometry as j,Mesh as G,OneFactor as H,RGBAFormat as E,RGBAIntegerFormat as L,ShaderMaterial as q,UnsignedIntType as J,Vector4 as O,Frustum as $,Matrix4 as X}from"three";var S="precision highp sampler2D;precision highp usampler2D;out vec4 vColor;out vec2 vPosition;uniform vec2 viewport;uniform mat4 gsModelViewMatrix;attribute uint splatIndex;uniform sampler2D centerAndScaleTexture;uniform usampler2D covAndColorTexture;vec2 unpackInt16(in uint value){int v=int(value);int v0=v>>16;int v1=(v&0xFFFF);if((v&0x8000)!=0)v1|=0xFFFF0000;return vec2(float(v1),float(v0));}vec4 calcCovVectors(vec3 viewPos,mat3 Vrk){float focal=(viewport.y/2.0)*abs(projectionMatrix[1][1]);mat3 J=mat3(focal/viewPos.z,0.,-(focal*viewPos.x)/(viewPos.z*viewPos.z),0.,focal/viewPos.z,-(focal*viewPos.y)/(viewPos.z*viewPos.z),0.,0.,0.);mat3 W=transpose(mat3(gsModelViewMatrix));mat3 T=W*J;mat3 cov=transpose(T)*Vrk*T;float diagonal1=cov[0][0]+0.3;float offDiagonal=cov[0][1];float diagonal2=cov[1][1]+0.3;float mid=0.5*(diagonal1+diagonal2);float radius=length(vec2((diagonal1-diagonal2)/2.0,offDiagonal));float lambda1=mid+radius;float lambda2=max(mid-radius,0.1);vec2 diagonalVector=normalize(vec2(offDiagonal,lambda1-diagonal1));vec2 v1=min(sqrt(2.0*lambda1),1024.0)*diagonalVector;vec2 v2=min(sqrt(2.0*lambda2),1024.0)*vec2(diagonalVector.y,-diagonalVector.x);return vec4(v1,v2);}void main(){ivec2 texSize=textureSize(centerAndScaleTexture,0);ivec2 texPos=ivec2(splatIndex % uint(texSize.x),splatIndex/uint(texSize.x));vec4 centerAndScaleData=texelFetch(centerAndScaleTexture,texPos,0);vec4 viewPos=gsModelViewMatrix*vec4(centerAndScaleData.xyz,1);vec4 pos2d=projectionMatrix*viewPos;float clip=1.2*pos2d.w;if(pos2d.z<-pos2d.w||pos2d.x<-clip||pos2d.x>clip||pos2d.y<-clip||pos2d.y>clip){gl_Position=vec4(0.0,0.0,2.0,1.0);return;}uvec4 covAndColorData=texelFetch(covAndColorTexture,texPos,0);vec2 cov3D_M11_M12=unpackInt16(covAndColorData.x)*centerAndScaleData.w;vec2 cov3D_M13_M22=unpackInt16(covAndColorData.y)*centerAndScaleData.w;vec2 cov3D_M23_M33=unpackInt16(covAndColorData.z)*centerAndScaleData.w;mat3 Vrk=mat3(cov3D_M11_M12.x,cov3D_M11_M12.y,cov3D_M13_M22.x,cov3D_M11_M12.y,cov3D_M13_M22.y,cov3D_M23_M33.x,cov3D_M13_M22.x,cov3D_M23_M33.x,cov3D_M23_M33.y);vec4 covVectors=calcCovVectors(viewPos.xyz,Vrk);vPosition=position.xy;uint colorUint=covAndColorData.w;vColor=vec4(float(colorUint&uint(0xFF))/255.0,float((colorUint>>uint(8))&uint(0xFF))/255.0,float((colorUint>>uint(16))&uint(0xFF))/255.0,float(colorUint>>uint(24))/255.0);vec2 vCenter=vec2(pos2d)/pos2d.w;gl_Position=vec4(vCenter+(position.x*covVectors.zw+position.y*covVectors.xy)/viewport*2.0,pos2d.z/pos2d.w,1.0);}";var y="in vec4 vColor;in vec2 vPosition;void main(){float A=-dot(vPosition,vPosition);if(A<-4.0)discard;float B=exp(A)*vColor.a;gl_FragColor=vec4(vColor.rgb,B);}";var f=class{constructor(){this._intervalTime=200,this._stableStopTime=3e3,this._isSorting=!1,this._dirty=!0,this._lastMvMatrix=null,this._lastSortTime=0,this._stableSince=0}set isSorting(t){this._isSorting=t}get isSorting(){return this._isSorting}set dirty(t){this._dirty=t}get dirty(){return this._dirty}set intervalTime(t){this._intervalTime=t}get intervalTime(){return this._intervalTime}set stableStopTime(t){this._stableStopTime=t}get stableStopTime(){return this._stableStopTime}_isMatrixChanged(t,e){return t?!t.equals(e):!0}tick(t,e){let r=performance.now(),i=this._isMatrixChanged(this._lastMvMatrix,t);i?this._stableSince=r:this._stableSince===0&&(this._stableSince=r);let s=r-this._lastSortTime>=this._intervalTime,u=r-this._stableSince<this._stableStopTime;return!this._isSorting&&s&&(this._dirty||i||u)&&(this._isSorting=!0,this._dirty=!1,this._lastSortTime=r,e()),this._lastMvMatrix=t,this}},M=f;var x={},I="0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz".split("");function A(o="D"){let t=[];t[8]=t[13]=t[18]=t[23]="-",t[14]="4";let e;for(let r=0;r<36;r++)t[r]||(e=0|Math.random()*16,t[r]=I[r===19?e&3|8:e]);return o+"-"+t.join("")}function C(){let o=document.createElement("canvas"),t=o.getContext("webgl2")||o.getContext("webgl");return t.getParameter(t.MAX_TEXTURE_SIZE)}function k(o){if(!o||/^[a-zA-Z]+:\/\//.test(o))return o;try{if(typeof x!="undefined"&&x.url)return new URL(o,x.url).href}catch{}if(typeof document!="undefined"&&document.currentScript&&document.currentScript.src)return new URL(o,document.currentScript.src).href;try{return new URL(o,typeof location!="undefined"?location.href:void 0).href}catch{return o}}var v=C(),Z=v*v,T=new V;T.setAttribute("position",new P(new Float32Array([-2,-2,0,2,2,0,-2,2,0,2,-2,0,2,2,0,-2,-2,0]),3));var w=class extends G{constructor(){super(),this._meshId=A(),this._vertexCount=0,this._textureWidth=v,this._textureHeight=1,this._threshold=-1e-5,this._centerAndScaleData=null,this._centerAndScaleTexture=null,this._rotationAndColorData=null,this._rotationAndColorTexture=null,this.geometry=new j().copy(T),this.geometry.instanceCount=1,this.material=new q({uniforms:{viewport:{value:new Float32Array([1980,1080])},centerAndScaleTexture:{value:null},covAndColorTexture:{value:null},gsModelViewMatrix:{value:null}},vertexShader:S,fragmentShader:y,blending:W,blendSrcAlpha:H,depthTest:!0,depthWrite:!1,transparent:!0}),this.material.onBeforeRender=this._onMaterialBeforeRender.bind(this),this.frustumCulled=!1,this._bounds=null,this._sortScheduler=new M,this._worker=null,this._useFrustumCulled=!1}get isSplatMesh(){return!0}set threshold(t){this._threshold=t}get threshold(){return this._threshold}set worker(t){this._worker=t}get worker(){return this._worker}get bounds(){return this._bounds}set vertexCount(t){this.setVertexCount(t)}get vertexCount(){return this._vertexCount}set useFrustumCulled(t){this._useFrustumCulled=t}get useFrustumCulled(){return this._useFrustumCulled}get sortScheduler(){return this._sortScheduler}_getModelViewMatrix(t){return t.matrixWorld.clone().invert().multiply(this.matrixWorld)}_updateTexture(t,e){t&&this._meshId===t.meshId&&(this._centerAndScaleData.set(t.out_cs),this._rotationAndColorData.set(t.out_rc),this._centerAndScaleTexture.image.data.set(this._centerAndScaleData.subarray(0,e*4),0),this._rotationAndColorTexture.image.data.set(this._rotationAndColorData.subarray(0,e*4),0),this._centerAndScaleTexture.needsUpdate=!0,this._rotationAndColorTexture.needsUpdate=!0,this._sortScheduler.dirty=!0)}_onMaterialBeforeRender(t,e,r,i,s,u){let h=this._getModelViewMatrix(r),p=s.material;p.uniforms.gsModelViewMatrix.value=h;let m=new O;t.getCurrentViewport(m),p.uniforms.viewport.value[0]=m.z,p.uniforms.viewport.value[1]=m.w,this._sortScheduler&&this._sortScheduler.tick(h,()=>{let c=h.elements,U=new Float32Array([c[2],c[6],c[10],c[14]]),a=new Float32Array(0);if(this._useFrustumCulled){a=new Float32Array(24);let n=new X;n.multiplyMatrices(r.projectionMatrix,r.matrixWorldInverse);let l=new $;l.setFromProjectionMatrix(n),l.planes.forEach((d,_)=>{a[_*4+0]=d.normal.x,a[_*4+1]=d.normal.y,a[_*4+2]=d.normal.z,a[_*4+3]=d.constant})}this._worker&&this._worker.call("sort_splats",this._meshId,U,a,this._threshold).then(n=>{if(n||(this._sortScheduler.isSorting=!1),n&&this._meshId===n.meshId){let l=new Uint32Array(n.data);this.geometry.attributes.splatIndex.set(l),this.geometry.attributes.splatIndex.needsUpdate=!0,this.geometry.instanceCount=l.length,this._sortScheduler.isSorting=!1}})})}setVertexCount(t){if(t===this._vertexCount)return;this._vertexCount=Math.min(t,Z),this._textureHeight=Math.floor((this._vertexCount-1)/this._textureWidth)+1,this._centerAndScaleData=new Float32Array(this._textureWidth*this._textureHeight*4),this._centerAndScaleTexture&&this._centerAndScaleTexture.dispose(),this._centerAndScaleTexture=new b(this._centerAndScaleData,this._textureWidth,this._textureHeight,E,B),this._rotationAndColorData=new Uint32Array(this._textureWidth*this._textureHeight*4),this._rotationAndColorTexture&&this._rotationAndColorTexture.dispose(),this._rotationAndColorTexture=new b(this._rotationAndColorData,this._textureWidth,this._textureHeight,L,J),this._rotationAndColorTexture.internalFormat="RGBA32UI";let e=new Uint32Array(this._textureWidth*this._textureHeight),r=new R(e,1,!1);return r.setUsage(z),this.geometry.setAttribute("splatIndex",r),this.material.uniforms.centerAndScaleTexture.value=this._centerAndScaleTexture,this.material.uniforms.covAndColorTexture.value=this._rotationAndColorTexture,this}attachWorker(t){return this._worker=t,this}detachWorker(){return this._worker&&this._worker.call("unregister_positions",this._bufferId),this._worker=null,this}async setDataFromBuffer(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromBuffer skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromBuffer skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_buffer",this._meshId,t,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async setDataFromGeometry(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromGeometry skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromGeometry skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_geometry",this._meshId,t.attributes.position.array,t.attributes._scale.array,t.attributes._rotation.array,t.attributes.color.array,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async setDataFromSpz(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromSpz skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromSpz skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_spz",this._meshId,t.positions,t.scales,t.rotations,t.colors,t.alphas,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async computeBounds(){if(!this._worker)return console.warn("[SplatMesh] computeBounds skipped: worker not ready"),this;if(!this._bounds){let t=await this._worker.call("compute_bounds",this._meshId);this._meshId===t.meshId&&(this._bounds=t.data)}return this}dispose(){this._bounds=null,this._centerAndScaleTexture&&(this._centerAndScaleTexture.dispose(),this._centerAndScaleData=null),this._rotationAndColorTexture&&(this._rotationAndColorTexture.dispose(),this._rotationAndColorData=null),this._worker&&this._worker.call("unregister_positions",this._bufferId),this._bufferId=null,this._sortScheduler=null,this.geometry.dispose(),this.material.dispose(),this.parent=null}},D=w;var g=class{constructor(t,{timeout:e=0}={}){this._baseUrl=k(t||"./"),this._workerUrl=new URL("wasm_splat.worker.min.js",this._baseUrl).href,this._wasmUrl=new URL("wasm_splat_bg.wasm",this._baseUrl).href,this._seq=0,this._pending=new Map,this._timeout=e,this._initPromise=null,this._ready=!1,this._createWorker()}_createWorker(){let t=i=>{i.onmessage=s=>this._handleMessage(s),i.onerror=s=>this._handleError(s),i.onmessageerror=s=>console.error("[SplatWorker] Worker message error:",s)};try{this._worker=new Worker(this._workerUrl,{type:"module"}),t(this._worker);return}catch{}let e=`
+      import ${JSON.stringify(this._workerUrl)};
+    `,r=URL.createObjectURL(new Blob([e],{type:"text/javascript"}));this._worker=new Worker(r,{type:"module"}),t(this._worker)}_handleMessage(t){let{id:e,result:r,error:i}=t.data,s=this._pending.get(e);s&&(this._pending.delete(e),i?s.reject(i):s.resolve(r))}_handleError(t){console.error("[SplatWorker] Worker error:",t);for(let[e,r]of this._pending)r.reject("[SplatWorker]Worker crashed");this._pending.clear()}_collectTransferables(t){let e=[];for(let r of t)r&&(ArrayBuffer.isView(r)?e.push(r.buffer):r instanceof ArrayBuffer&&e.push(r));return e}async init(){if(this._ready)return!0;if(this._initPromise)return this._initPromise;this._initPromise=new Promise((e,r)=>{let i=++this._seq;this._pending.set(i,{resolve:e,reject:r}),this._worker.postMessage({id:i,fn:"__init__",wasmUrl:this._wasmUrl,args:[]})});let t=await this._initPromise;return this._ready=!0,t}call(t,...e){return new Promise((r,i)=>{this._worker||i(`the worker ${this._workerUrl} does not exist`);let s=++this._seq;this._pending.set(s,{resolve:r,reject:i});let u=this._collectTransferables(e);this._worker.postMessage({id:s,fn:t,wasmUrl:this._wasmUrl,args:e},u),this._timeout>0&&setTimeout(()=>{this._pending.has(s)&&(this._pending.delete(s),i(`[SplatWorker] Task "${t}" timeout after ${this._timeout}ms`))},this._timeout)})}cancel(t,e="canceled"){let r=this._pending.get(t);return r&&(r.reject(e),this._pending.delete(t)),this}dispose(){return this._worker.terminate(),this._pending.clear(),this}},F=g;typeof window!="undefined"&&(window.SplatMesh=D,window.SplatWorker=F);export{D as SplatMesh,F as SplatWorker};

package/dist/splat-mesh.min.js

@@ -1 +1,3 @@
-(()=>{var W=Object.create;var S=Object.defineProperty;var V=Object.getOwnPropertyDescriptor;var U=Object.getOwnPropertyNames;var z=Object.getPrototypeOf,B=Object.prototype.hasOwnProperty;var R=(o,t)=>()=>(t||o((t={exports:{}}).exports,t),t.exports);var E=(o,t,e,r)=>{if(t&&typeof t=="object"||typeof t=="function")for(let s of U(t))!B.call(o,s)&&s!==e&&S(o,s,{get:()=>t[s],enumerable:!(r=V(t,s))||r.enumerable});return o};var H=(o,t,e)=>(e=o!=null?W(z(o)):{},E(t||!o||!o.__esModule?S(e,"default",{value:o,enumerable:!0}):e,o));var y=R(($,M)=>{M.exports=THREE});var i=H(y());var k="precision highp sampler2D;precision highp usampler2D;out vec4 vColor;out vec2 vPosition;uniform vec2 viewport;uniform mat4 gsModelViewMatrix;attribute uint splatIndex;uniform sampler2D centerAndScaleTexture;uniform usampler2D covAndColorTexture;vec2 unpackInt16(in uint value){int v=int(value);int v0=v>>16;int v1=(v&0xFFFF);if((v&0x8000)!=0)v1|=0xFFFF0000;return vec2(float(v1),float(v0));}vec4 calcCovVectors(vec3 viewPos,mat3 Vrk){float focal=(viewport.y/2.0)*abs(projectionMatrix[1][1]);mat3 J=mat3(focal/viewPos.z,0.,-(focal*viewPos.x)/(viewPos.z*viewPos.z),0.,focal/viewPos.z,-(focal*viewPos.y)/(viewPos.z*viewPos.z),0.,0.,0.);mat3 W=transpose(mat3(gsModelViewMatrix));mat3 T=W*J;mat3 cov=transpose(T)*Vrk*T;float diagonal1=cov[0][0]+0.3;float offDiagonal=cov[0][1];float diagonal2=cov[1][1]+0.3;float mid=0.5*(diagonal1+diagonal2);float radius=length(vec2((diagonal1-diagonal2)/2.0,offDiagonal));float lambda1=mid+radius;float lambda2=max(mid-radius,0.1);vec2 diagonalVector=normalize(vec2(offDiagonal,lambda1-diagonal1));vec2 v1=min(sqrt(2.0*lambda1),1024.0)*diagonalVector;vec2 v2=min(sqrt(2.0*lambda2),1024.0)*vec2(diagonalVector.y,-diagonalVector.x);return vec4(v1,v2);}void main(){ivec2 texSize=textureSize(centerAndScaleTexture,0);ivec2 texPos=ivec2(splatIndex % uint(texSize.x),splatIndex/uint(texSize.x));vec4 centerAndScaleData=texelFetch(centerAndScaleTexture,texPos,0);vec4 viewPos=gsModelViewMatrix*vec4(centerAndScaleData.xyz,1);vec4 pos2d=projectionMatrix*viewPos;float clip=1.2*pos2d.w;if(pos2d.z<-pos2d.w||pos2d.x<-clip||pos2d.x>clip||pos2d.y<-clip||pos2d.y>clip){gl_Position=vec4(0.0,0.0,2.0,1.0);return;}uvec4 covAndColorData=texelFetch(covAndColorTexture,texPos,0);vec2 cov3D_M11_M12=unpackInt16(covAndColorData.x)*centerAndScaleData.w;vec2 cov3D_M13_M22=unpackInt16(covAndColorData.y)*centerAndScaleData.w;vec2 cov3D_M23_M33=unpackInt16(covAndColorData.z)*centerAndScaleData.w;mat3 Vrk=mat3(cov3D_M11_M12.x,cov3D_M11_M12.y,cov3D_M13_M22.x,cov3D_M11_M12.y,cov3D_M13_M22.y,cov3D_M23_M33.x,cov3D_M13_M22.x,cov3D_M23_M33.x,cov3D_M23_M33.y);vec4 covVectors=calcCovVectors(viewPos.xyz,Vrk);vPosition=position.xy;uint colorUint=covAndColorData.w;vColor=vec4(float(colorUint&uint(0xFF))/255.0,float((colorUint>>uint(8))&uint(0xFF))/255.0,float((colorUint>>uint(16))&uint(0xFF))/255.0,float(colorUint>>uint(24))/255.0);vec2 vCenter=vec2(pos2d)/pos2d.w;gl_Position=vec4(vCenter+(position.x*covVectors.zw+position.y*covVectors.xy)/viewport*2.0,pos2d.z/pos2d.w,1.0);}";var A="in vec4 vColor;in vec2 vPosition;void main(){float A=-dot(vPosition,vPosition);if(A<-4.0)discard;float B=exp(A)*vColor.a;gl_FragColor=vec4(vColor.rgb,B);}";var f=class{constructor(){this._intervalTime=200,this._stableStopTime=3e3,this._isSorting=!1,this._dirty=!0,this._lastMvMatrix=null,this._lastSortTime=0,this._stableSince=0}set isSorting(t){this._isSorting=t}get isSorting(){return this._isSorting}set dirty(t){this._dirty=t}get dirty(){return this._dirty}set intervalTime(t){this._intervalTime=t}get intervalTime(){return this._intervalTime}set stableStopTime(t){this._stableStopTime=t}get stableStopTime(){return this._stableStopTime}_isMatrixChanged(t,e){return t?!t.equals(e):!0}tick(t,e){let r=performance.now(),s=this._isMatrixChanged(this._lastMvMatrix,t);s?this._stableSince=r:this._stableSince===0&&(this._stableSince=r);let n=r-this._lastSortTime>=this._intervalTime,u=r-this._stableSince<this._stableStopTime;return!this._isSorting&&n&&(this._dirty||s||u)&&(this._isSorting=!0,this._dirty=!1,this._lastSortTime=r,e()),this._lastMvMatrix=t,this}},C=f;var j="0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz".split("");function T(o="D"){let t=[];t[8]=t[13]=t[18]=t[23]="-",t[14]="4";let e;for(let r=0;r<36;r++)t[r]||(e=0|Math.random()*16,t[r]=j[r===19?e&3|8:e]);return o+"-"+t.join("")}function b(){let o=document.createElement("canvas"),t=o.getContext("webgl2")||o.getContext("webgl");return t.getParameter(t.MAX_TEXTURE_SIZE)}var v=b(),G=v*v,D=new i.BufferGeometry;D.setAttribute("position",new i.BufferAttribute(new Float32Array([-2,-2,0,2,2,0,-2,2,0,2,-2,0,2,2,0,-2,-2,0]),3));var w=class extends i.Mesh{constructor(){super(),this._meshId=T(),this._vertexCount=0,this._textureWidth=v,this._textureHeight=1,this._threshold=-1e-5,this._centerAndScaleData=null,this._centerAndScaleTexture=null,this._rotationAndColorData=null,this._rotationAndColorTexture=null,this.geometry=new i.InstancedBufferGeometry().copy(D),this.geometry.instanceCount=1,this.material=new i.ShaderMaterial({uniforms:{viewport:{value:new Float32Array([1980,1080])},centerAndScaleTexture:{value:null},covAndColorTexture:{value:null},gsModelViewMatrix:{value:null}},vertexShader:k,fragmentShader:A,blending:i.CustomBlending,blendSrcAlpha:i.OneFactor,depthTest:!0,depthWrite:!1,transparent:!0}),this.material.onBeforeRender=this._onMaterialBeforeRender.bind(this),this.frustumCulled=!1,this._bounds=null,this._sortScheduler=new C,this._worker=null,this._useFrustumCulled=!1}get isSplatMesh(){return!0}set threshold(t){this._threshold=t}get threshold(){return this._threshold}set worker(t){this._worker=t}get worker(){return this._worker}get bounds(){return this._bounds}set vertexCount(t){this.setVertexCount(t)}get vertexCount(){return this._vertexCount}set useFrustumCulled(t){this._useFrustumCulled=t}get useFrustumCulled(){return this._useFrustumCulled}get sortScheduler(){return this._sortScheduler}_getModelViewMatrix(t){return t.matrixWorld.clone().invert().multiply(this.matrixWorld)}_updateTexture(t,e){t&&this._meshId===t.meshId&&(this._centerAndScaleData.set(t.out_cs),this._rotationAndColorData.set(t.out_rc),this._centerAndScaleTexture.image.data.set(this._centerAndScaleData.subarray(0,e*4),0),this._rotationAndColorTexture.image.data.set(this._rotationAndColorData.subarray(0,e*4),0),this._centerAndScaleTexture.needsUpdate=!0,this._rotationAndColorTexture.needsUpdate=!0,this._sortScheduler.dirty=!0)}_onMaterialBeforeRender(t,e,r,s,n,u){let c=this._getModelViewMatrix(r),m=n.material;m.uniforms.gsModelViewMatrix.value=c;let x=new i.Vector4;t.getCurrentViewport(x),m.uniforms.viewport.value[0]=x.z,m.uniforms.viewport.value[1]=x.w,this._sortScheduler&&this._sortScheduler.tick(c,()=>{let d=c.elements,P=new Float32Array([d[2],d[6],d[10],d[14]]),l=new Float32Array(0);if(this._useFrustumCulled){l=new Float32Array(24);let a=new i.Matrix4;a.multiplyMatrices(r.projectionMatrix,r.matrixWorldInverse);let h=new i.Frustum;h.setFromProjectionMatrix(a),h.planes.forEach((_,p)=>{l[p*4+0]=_.normal.x,l[p*4+1]=_.normal.y,l[p*4+2]=_.normal.z,l[p*4+3]=_.constant})}this._worker&&this._worker.call("sort_splats",this._meshId,P,l,this._threshold).then(a=>{if(a||(this._sortScheduler.isSorting=!1),a&&this._meshId===a.meshId){let h=new Uint32Array(a.data);this.geometry.attributes.splatIndex.set(h),this.geometry.attributes.splatIndex.needsUpdate=!0,this.geometry.instanceCount=h.length,this._sortScheduler.isSorting=!1}})})}setVertexCount(t){if(t===this._vertexCount)return;this._vertexCount=Math.min(t,G),this._textureHeight=Math.floor((this._vertexCount-1)/this._textureWidth)+1,this._centerAndScaleData=new Float32Array(this._textureWidth*this._textureHeight*4),this._centerAndScaleTexture&&this._centerAndScaleTexture.dispose(),this._centerAndScaleTexture=new i.DataTexture(this._centerAndScaleData,this._textureWidth,this._textureHeight,i.RGBAFormat,i.FloatType),this._rotationAndColorData=new Uint32Array(this._textureWidth*this._textureHeight*4),this._rotationAndColorTexture&&this._rotationAndColorTexture.dispose(),this._rotationAndColorTexture=new i.DataTexture(this._rotationAndColorData,this._textureWidth,this._textureHeight,i.RGBAIntegerFormat,i.UnsignedIntType),this._rotationAndColorTexture.internalFormat="RGBA32UI";let e=new Uint32Array(this._textureWidth*this._textureHeight),r=new i.InstancedBufferAttribute(e,1,!1);return r.setUsage(i.DynamicDrawUsage),this.geometry.setAttribute("splatIndex",r),this.material.uniforms.centerAndScaleTexture.value=this._centerAndScaleTexture,this.material.uniforms.covAndColorTexture.value=this._rotationAndColorTexture,this}attachWorker(t){return this._worker=t,this}detachWorker(){return this._worker&&this._worker.call("unregister_positions",this._bufferId),this._worker=null,this}async setDataFromBuffer(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromBuffer skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromBuffer skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_buffer",this._meshId,t,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async setDataFromGeometry(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromGeometry skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromGeometry skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_geometry",this._meshId,t.attributes.position.array,t.attributes._scale.array,t.attributes._rotation.array,t.attributes.color.array,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async setDataFromSpz(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromSpz skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromSpz skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_spz",this._meshId,t.positions,t.scales,t.rotations,t.colors,t.alphas,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async computeBounds(){if(!this._worker)return console.warn("[SplatMesh] computeBounds skipped: worker not ready"),this;if(!this._bounds){let t=await this._worker.call("compute_bounds",this._meshId);this._meshId===t.meshId&&(this._bounds=t.data)}return this}dispose(){this._bounds=null,this._centerAndScaleTexture&&(this._centerAndScaleTexture.dispose(),this._centerAndScaleData=null),this._rotationAndColorTexture&&(this._rotationAndColorTexture.dispose(),this._rotationAndColorData=null),this._worker&&this._worker.call("unregister_positions",this._bufferId),this._bufferId=null,this._sortScheduler=null,this.geometry.dispose(),this.material.dispose(),this.parent=null}},F=w;var g=class{constructor(t,{timeout:e=0}={}){this._workerUrl=t,this._worker=null,this._seq=0,this._pending=new Map,this._timeout=e,this._initPromise=null,this._ready=!1,this._createWorker()}_createWorker(){let t=e=>{e.onmessage=r=>this._handleMessage(r),e.onerror=r=>this._handleError(r),e.onmessageerror=r=>console.error("[SplatWorker] Worker message error:",r)};try{this._worker=new Worker(this._workerUrl,{type:"module"}),t(this._worker);return}catch(e){console.error("Worker error:",e)}fetch(this._workerUrl).then(e=>{if(!e.ok)throw new Error(`[SplatWorker] Failed to fetch worker: ${this._workerUrl}`);return e.text()}).then(e=>{let r=new Blob([e],{type:"application/javascript"}),s=URL.createObjectURL(r);this._worker=new Worker(s),t(this._worker)})}_handleMessage(t){let{id:e,result:r,error:s}=t.data,n=this._pending.get(e);n&&(this._pending.delete(e),s?n.reject(s):n.resolve(r))}_handleError(t){console.error("[SplatWorker] Worker error:",t);for(let[e,r]of this._pending)r.reject("[SplatWorker] Worker crashed");this._pending.clear()}_collectTransferables(t){let e=[];for(let r of t)r&&(ArrayBuffer.isView(r)?e.push(r.buffer):r instanceof ArrayBuffer&&e.push(r));return e}async init(){if(this._ready)return!0;if(this._initPromise)return this._initPromise;this._initPromise=new Promise((e,r)=>{let s=++this._seq;this._pending.set(s,{resolve:e,reject:r}),this._worker.postMessage({id:s,fn:"__init_wasm__",args:[]})});let t=await this._initPromise;return this._ready=!0,t}call(t,...e){return new Promise((r,s)=>{this._worker||s(`the worker ${this._workerUrl} does not exist`);let n=++this._seq;this._pending.set(n,{resolve:r,reject:s});let u=this._collectTransferables(e);this._worker.postMessage({id:n,fn:t,args:e},u),this._timeout>0&&setTimeout(()=>{this._pending.has(n)&&(this._pending.delete(n),s(`[SplatWorker] Task "${t}" timeout after ${this._timeout}ms`))},this._timeout)})}cancel(t,e="canceled"){let r=this._pending.get(t);return r&&(r.reject(e),this._pending.delete(t)),this}dispose(){return this._worker.terminate(),this._pending.clear(),this}},I=g;typeof window!="undefined"&&(window.SplatMesh=F,window.SplatWorker=I);})();
+(()=>{var W=Object.create;var y=Object.defineProperty;var z=Object.getOwnPropertyDescriptor;var B=Object.getOwnPropertyNames;var R=Object.getPrototypeOf,j=Object.prototype.hasOwnProperty;var H=(s,t)=>()=>(t||s((t={exports:{}}).exports,t),t.exports);var E=(s,t,e,r)=>{if(t&&typeof t=="object"||typeof t=="function")for(let o of B(t))!j.call(s,o)&&o!==e&&y(s,o,{get:()=>t[o],enumerable:!(r=z(t,o))||r.enumerable});return s};var G=(s,t,e)=>(e=s!=null?W(R(s)):{},E(t||!s||!s.__esModule?y(e,"default",{value:s,enumerable:!0}):e,s));var A=H((O,M)=>{M.exports=THREE});var i=G(A());var C="precision highp sampler2D;precision highp usampler2D;out vec4 vColor;out vec2 vPosition;uniform vec2 viewport;uniform mat4 gsModelViewMatrix;attribute uint splatIndex;uniform sampler2D centerAndScaleTexture;uniform usampler2D covAndColorTexture;vec2 unpackInt16(in uint value){int v=int(value);int v0=v>>16;int v1=(v&0xFFFF);if((v&0x8000)!=0)v1|=0xFFFF0000;return vec2(float(v1),float(v0));}vec4 calcCovVectors(vec3 viewPos,mat3 Vrk){float focal=(viewport.y/2.0)*abs(projectionMatrix[1][1]);mat3 J=mat3(focal/viewPos.z,0.,-(focal*viewPos.x)/(viewPos.z*viewPos.z),0.,focal/viewPos.z,-(focal*viewPos.y)/(viewPos.z*viewPos.z),0.,0.,0.);mat3 W=transpose(mat3(gsModelViewMatrix));mat3 T=W*J;mat3 cov=transpose(T)*Vrk*T;float diagonal1=cov[0][0]+0.3;float offDiagonal=cov[0][1];float diagonal2=cov[1][1]+0.3;float mid=0.5*(diagonal1+diagonal2);float radius=length(vec2((diagonal1-diagonal2)/2.0,offDiagonal));float lambda1=mid+radius;float lambda2=max(mid-radius,0.1);vec2 diagonalVector=normalize(vec2(offDiagonal,lambda1-diagonal1));vec2 v1=min(sqrt(2.0*lambda1),1024.0)*diagonalVector;vec2 v2=min(sqrt(2.0*lambda2),1024.0)*vec2(diagonalVector.y,-diagonalVector.x);return vec4(v1,v2);}void main(){ivec2 texSize=textureSize(centerAndScaleTexture,0);ivec2 texPos=ivec2(splatIndex % uint(texSize.x),splatIndex/uint(texSize.x));vec4 centerAndScaleData=texelFetch(centerAndScaleTexture,texPos,0);vec4 viewPos=gsModelViewMatrix*vec4(centerAndScaleData.xyz,1);vec4 pos2d=projectionMatrix*viewPos;float clip=1.2*pos2d.w;if(pos2d.z<-pos2d.w||pos2d.x<-clip||pos2d.x>clip||pos2d.y<-clip||pos2d.y>clip){gl_Position=vec4(0.0,0.0,2.0,1.0);return;}uvec4 covAndColorData=texelFetch(covAndColorTexture,texPos,0);vec2 cov3D_M11_M12=unpackInt16(covAndColorData.x)*centerAndScaleData.w;vec2 cov3D_M13_M22=unpackInt16(covAndColorData.y)*centerAndScaleData.w;vec2 cov3D_M23_M33=unpackInt16(covAndColorData.z)*centerAndScaleData.w;mat3 Vrk=mat3(cov3D_M11_M12.x,cov3D_M11_M12.y,cov3D_M13_M22.x,cov3D_M11_M12.y,cov3D_M13_M22.y,cov3D_M23_M33.x,cov3D_M13_M22.x,cov3D_M23_M33.x,cov3D_M23_M33.y);vec4 covVectors=calcCovVectors(viewPos.xyz,Vrk);vPosition=position.xy;uint colorUint=covAndColorData.w;vColor=vec4(float(colorUint&uint(0xFF))/255.0,float((colorUint>>uint(8))&uint(0xFF))/255.0,float((colorUint>>uint(16))&uint(0xFF))/255.0,float(colorUint>>uint(24))/255.0);vec2 vCenter=vec2(pos2d)/pos2d.w;gl_Position=vec4(vCenter+(position.x*covVectors.zw+position.y*covVectors.xy)/viewport*2.0,pos2d.z/pos2d.w,1.0);}";var k="in vec4 vColor;in vec2 vPosition;void main(){float A=-dot(vPosition,vPosition);if(A<-4.0)discard;float B=exp(A)*vColor.a;gl_FragColor=vec4(vColor.rgb,B);}";var x=class{constructor(){this._intervalTime=200,this._stableStopTime=3e3,this._isSorting=!1,this._dirty=!0,this._lastMvMatrix=null,this._lastSortTime=0,this._stableSince=0}set isSorting(t){this._isSorting=t}get isSorting(){return this._isSorting}set dirty(t){this._dirty=t}get dirty(){return this._dirty}set intervalTime(t){this._intervalTime=t}get intervalTime(){return this._intervalTime}set stableStopTime(t){this._stableStopTime=t}get stableStopTime(){return this._stableStopTime}_isMatrixChanged(t,e){return t?!t.equals(e):!0}tick(t,e){let r=performance.now(),o=this._isMatrixChanged(this._lastMvMatrix,t);o?this._stableSince=r:this._stableSince===0&&(this._stableSince=r);let n=r-this._lastSortTime>=this._intervalTime,h=r-this._stableSince<this._stableStopTime;return!this._isSorting&&n&&(this._dirty||o||h)&&(this._isSorting=!0,this._dirty=!1,this._lastSortTime=r,e()),this._lastMvMatrix=t,this}},T=x;var v={},L="0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz".split("");function b(s="D"){let t=[];t[8]=t[13]=t[18]=t[23]="-",t[14]="4";let e;for(let r=0;r<36;r++)t[r]||(e=0|Math.random()*16,t[r]=L[r===19?e&3|8:e]);return s+"-"+t.join("")}function D(){let s=document.createElement("canvas"),t=s.getContext("webgl2")||s.getContext("webgl");return t.getParameter(t.MAX_TEXTURE_SIZE)}function F(s){if(!s||/^[a-zA-Z]+:\/\//.test(s))return s;try{if(typeof v!="undefined"&&v.url)return new URL(s,v.url).href}catch{}if(typeof document!="undefined"&&document.currentScript&&document.currentScript.src)return new URL(s,document.currentScript.src).href;try{return new URL(s,typeof location!="undefined"?location.href:void 0).href}catch{return s}}var w=D(),q=w*w,U=new i.BufferGeometry;U.setAttribute("position",new i.BufferAttribute(new Float32Array([-2,-2,0,2,2,0,-2,2,0,2,-2,0,2,2,0,-2,-2,0]),3));var g=class extends i.Mesh{constructor(){super(),this._meshId=b(),this._vertexCount=0,this._textureWidth=w,this._textureHeight=1,this._threshold=-1e-5,this._centerAndScaleData=null,this._centerAndScaleTexture=null,this._rotationAndColorData=null,this._rotationAndColorTexture=null,this.geometry=new i.InstancedBufferGeometry().copy(U),this.geometry.instanceCount=1,this.material=new i.ShaderMaterial({uniforms:{viewport:{value:new Float32Array([1980,1080])},centerAndScaleTexture:{value:null},covAndColorTexture:{value:null},gsModelViewMatrix:{value:null}},vertexShader:C,fragmentShader:k,blending:i.CustomBlending,blendSrcAlpha:i.OneFactor,depthTest:!0,depthWrite:!1,transparent:!0}),this.material.onBeforeRender=this._onMaterialBeforeRender.bind(this),this.frustumCulled=!1,this._bounds=null,this._sortScheduler=new T,this._worker=null,this._useFrustumCulled=!1}get isSplatMesh(){return!0}set threshold(t){this._threshold=t}get threshold(){return this._threshold}set worker(t){this._worker=t}get worker(){return this._worker}get bounds(){return this._bounds}set vertexCount(t){this.setVertexCount(t)}get vertexCount(){return this._vertexCount}set useFrustumCulled(t){this._useFrustumCulled=t}get useFrustumCulled(){return this._useFrustumCulled}get sortScheduler(){return this._sortScheduler}_getModelViewMatrix(t){return t.matrixWorld.clone().invert().multiply(this.matrixWorld)}_updateTexture(t,e){t&&this._meshId===t.meshId&&(this._centerAndScaleData.set(t.out_cs),this._rotationAndColorData.set(t.out_rc),this._centerAndScaleTexture.image.data.set(this._centerAndScaleData.subarray(0,e*4),0),this._rotationAndColorTexture.image.data.set(this._rotationAndColorData.subarray(0,e*4),0),this._centerAndScaleTexture.needsUpdate=!0,this._rotationAndColorTexture.needsUpdate=!0,this._sortScheduler.dirty=!0)}_onMaterialBeforeRender(t,e,r,o,n,h){let c=this._getModelViewMatrix(r),m=n.material;m.uniforms.gsModelViewMatrix.value=c;let f=new i.Vector4;t.getCurrentViewport(f),m.uniforms.viewport.value[0]=f.z,m.uniforms.viewport.value[1]=f.w,this._sortScheduler&&this._sortScheduler.tick(c,()=>{let d=c.elements,V=new Float32Array([d[2],d[6],d[10],d[14]]),l=new Float32Array(0);if(this._useFrustumCulled){l=new Float32Array(24);let a=new i.Matrix4;a.multiplyMatrices(r.projectionMatrix,r.matrixWorldInverse);let u=new i.Frustum;u.setFromProjectionMatrix(a),u.planes.forEach((_,p)=>{l[p*4+0]=_.normal.x,l[p*4+1]=_.normal.y,l[p*4+2]=_.normal.z,l[p*4+3]=_.constant})}this._worker&&this._worker.call("sort_splats",this._meshId,V,l,this._threshold).then(a=>{if(a||(this._sortScheduler.isSorting=!1),a&&this._meshId===a.meshId){let u=new Uint32Array(a.data);this.geometry.attributes.splatIndex.set(u),this.geometry.attributes.splatIndex.needsUpdate=!0,this.geometry.instanceCount=u.length,this._sortScheduler.isSorting=!1}})})}setVertexCount(t){if(t===this._vertexCount)return;this._vertexCount=Math.min(t,q),this._textureHeight=Math.floor((this._vertexCount-1)/this._textureWidth)+1,this._centerAndScaleData=new Float32Array(this._textureWidth*this._textureHeight*4),this._centerAndScaleTexture&&this._centerAndScaleTexture.dispose(),this._centerAndScaleTexture=new i.DataTexture(this._centerAndScaleData,this._textureWidth,this._textureHeight,i.RGBAFormat,i.FloatType),this._rotationAndColorData=new Uint32Array(this._textureWidth*this._textureHeight*4),this._rotationAndColorTexture&&this._rotationAndColorTexture.dispose(),this._rotationAndColorTexture=new i.DataTexture(this._rotationAndColorData,this._textureWidth,this._textureHeight,i.RGBAIntegerFormat,i.UnsignedIntType),this._rotationAndColorTexture.internalFormat="RGBA32UI";let e=new Uint32Array(this._textureWidth*this._textureHeight),r=new i.InstancedBufferAttribute(e,1,!1);return r.setUsage(i.DynamicDrawUsage),this.geometry.setAttribute("splatIndex",r),this.material.uniforms.centerAndScaleTexture.value=this._centerAndScaleTexture,this.material.uniforms.covAndColorTexture.value=this._rotationAndColorTexture,this}attachWorker(t){return this._worker=t,this}detachWorker(){return this._worker&&this._worker.call("unregister_positions",this._bufferId),this._worker=null,this}async setDataFromBuffer(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromBuffer skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromBuffer skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_buffer",this._meshId,t,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async setDataFromGeometry(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromGeometry skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromGeometry skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_geometry",this._meshId,t.attributes.position.array,t.attributes._scale.array,t.attributes._rotation.array,t.attributes.color.array,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async setDataFromSpz(t){if(this._vertexCount<=0)return console.warn("[SplatMesh] setDataFromSpz skipped: vertexCount is 0"),this;if(!this._worker)return console.warn("[SplatMesh] setDataFromSpz skipped: worker not ready"),this;let e=await this._worker.call("process_splats_from_spz",this._meshId,t.positions,t.scales,t.rotations,t.colors,t.alphas,this._vertexCount);return this._updateTexture(e,this._vertexCount),this._bounds=null,this}async computeBounds(){if(!this._worker)return console.warn("[SplatMesh] computeBounds skipped: worker not ready"),this;if(!this._bounds){let t=await this._worker.call("compute_bounds",this._meshId);this._meshId===t.meshId&&(this._bounds=t.data)}return this}dispose(){this._bounds=null,this._centerAndScaleTexture&&(this._centerAndScaleTexture.dispose(),this._centerAndScaleData=null),this._rotationAndColorTexture&&(this._rotationAndColorTexture.dispose(),this._rotationAndColorData=null),this._worker&&this._worker.call("unregister_positions",this._bufferId),this._bufferId=null,this._sortScheduler=null,this.geometry.dispose(),this.material.dispose(),this.parent=null}},I=g;var S=class{constructor(t,{timeout:e=0}={}){this._baseUrl=F(t||"./"),this._workerUrl=new URL("wasm_splat.worker.min.js",this._baseUrl).href,this._wasmUrl=new URL("wasm_splat_bg.wasm",this._baseUrl).href,this._seq=0,this._pending=new Map,this._timeout=e,this._initPromise=null,this._ready=!1,this._createWorker()}_createWorker(){let t=o=>{o.onmessage=n=>this._handleMessage(n),o.onerror=n=>this._handleError(n),o.onmessageerror=n=>console.error("[SplatWorker] Worker message error:",n)};try{this._worker=new Worker(this._workerUrl,{type:"module"}),t(this._worker);return}catch{}let e=`
+      import ${JSON.stringify(this._workerUrl)};
+    `,r=URL.createObjectURL(new Blob([e],{type:"text/javascript"}));this._worker=new Worker(r,{type:"module"}),t(this._worker)}_handleMessage(t){let{id:e,result:r,error:o}=t.data,n=this._pending.get(e);n&&(this._pending.delete(e),o?n.reject(o):n.resolve(r))}_handleError(t){console.error("[SplatWorker] Worker error:",t);for(let[e,r]of this._pending)r.reject("[SplatWorker]Worker crashed");this._pending.clear()}_collectTransferables(t){let e=[];for(let r of t)r&&(ArrayBuffer.isView(r)?e.push(r.buffer):r instanceof ArrayBuffer&&e.push(r));return e}async init(){if(this._ready)return!0;if(this._initPromise)return this._initPromise;this._initPromise=new Promise((e,r)=>{let o=++this._seq;this._pending.set(o,{resolve:e,reject:r}),this._worker.postMessage({id:o,fn:"__init__",wasmUrl:this._wasmUrl,args:[]})});let t=await this._initPromise;return this._ready=!0,t}call(t,...e){return new Promise((r,o)=>{this._worker||o(`the worker ${this._workerUrl} does not exist`);let n=++this._seq;this._pending.set(n,{resolve:r,reject:o});let h=this._collectTransferables(e);this._worker.postMessage({id:n,fn:t,wasmUrl:this._wasmUrl,args:e},h),this._timeout>0&&setTimeout(()=>{this._pending.has(n)&&(this._pending.delete(n),o(`[SplatWorker] Task "${t}" timeout after ${this._timeout}ms`))},this._timeout)})}cancel(t,e="canceled"){let r=this._pending.get(t);return r&&(r.reject(e),this._pending.delete(t)),this}dispose(){return this._worker.terminate(),this._pending.clear(),this}},P=S;typeof window!="undefined"&&(window.SplatMesh=I,window.SplatWorker=P);})();

package/dist/workers/wasm_splat.worker.min.js

@@ -1 +1,363 @@
-import t,{process_splats_from_buffer as s,append_data_from_buffer as e,process_splats_from_geometry as u,process_splats_from_spz as r,sort_splats as o,compute_bounds as l,unregister_positions as n}from"./wasm_splat.min.js";let a=!1,f=null,c={};function _(t){return{outCS:new Float32Array(new ArrayBuffer(4*t*4)),outRC:new Uint32Array(new ArrayBuffer(4*t*4))}}self.onmessage=async i=>{const{id:d,fn:p,args:m}=i.data;try{let i;if("__init_wasm__"===p)return a||(await t(),a=!0),void postMessage({id:d,result:!0});if(await async function(){return!!a||(f||(f=t().then(()=>(a=!0,!0)).catch(t=>{throw a=!1,f=null,t})),f)}(),"process_splats_from_buffer"===p){const[t,e,u]=m,r=new Uint8Array(e),{outCS:o,outRC:l}=_(u),n=s(r,u,o,l);c[t]={bufferId:n},i={meshId:t,out_cs:o,out_rc:l}}else if("append_data_from_buffer"===p){let[t,s,u]=m;const{bufferId:r}=c[t];let o=new Uint8Array(s);const{outCS:l,outRC:n}=_(u);e(r,o,u,l,n),i={out_cs:l,out_rc:n},s=null,o=null}else if("process_splats_from_geometry"===p){let[t,s,e,r,o,l]=m;const{outCS:n,outRC:a}=_(l),f=u(s,e,r,o,l,n,a);c[t]={bufferId:f},i={meshId:t,out_cs:n,out_rc:a},s=null,e=null,r=null,o=null}else if("process_splats_from_spz"===p){let[t,s,e,u,o,l,n]=m;const{outCS:a,outRC:f}=_(n),d=r(s,e,u,o,l,n,a,f);c[t]={bufferId:d},i={meshId:t,out_cs:a,out_rc:f},s=null,e=null,u=null,o=null,l=null}else if("sort_splats"===p){const[t,s,e,u]=m,{bufferId:r}=c[t];i={meshId:t,data:o(r,s,e,u)}}else if("compute_bounds"===p){const[t]=m,{bufferId:s}=c[t];i={meshId:t,data:l(s)}}else if("unregister_positions"===p){const[t]=m,{bufferId:s}=c[t];n(s),i=!0}i instanceof Uint32Array?(postMessage({id:d,result:i},[i.buffer]),i=null):i&&i.data?(postMessage({id:d,result:i},[i.data.buffer]),i.data=null,i=null):i&&i.out_cs&&i.out_rc?(postMessage({id:d,result:i},[i.out_cs.buffer,i.out_rc.buffer]),i.out_cs=null,i.out_rc=null,i=null):postMessage({id:d,result:i})}catch(t){console.error(t),postMessage({id:d,error:t?.message||String(t)})}};
+// pkg-web/wasm_splat.js
+var wasm;
+function addHeapObject(obj) {
+  if (heap_next === heap.length) heap.push(heap.length + 1);
+  const idx = heap_next;
+  heap_next = heap[idx];
+  heap[idx] = obj;
+  return idx;
+}
+function dropObject(idx) {
+  if (idx < 132) return;
+  heap[idx] = heap_next;
+  heap_next = idx;
+}
+function getArrayF32FromWasm0(ptr, len) {
+  ptr = ptr >>> 0;
+  return getFloat32ArrayMemory0().subarray(ptr / 4, ptr / 4 + len);
+}
+function getArrayU32FromWasm0(ptr, len) {
+  ptr = ptr >>> 0;
+  return getUint32ArrayMemory0().subarray(ptr / 4, ptr / 4 + len);
+}
+function getArrayU8FromWasm0(ptr, len) {
+  ptr = ptr >>> 0;
+  return getUint8ArrayMemory0().subarray(ptr / 1, ptr / 1 + len);
+}
+var cachedDataViewMemory0 = null;
+function getDataViewMemory0() {
+  if (cachedDataViewMemory0 === null || cachedDataViewMemory0.buffer.detached === true || cachedDataViewMemory0.buffer.detached === void 0 && cachedDataViewMemory0.buffer !== wasm.memory.buffer) {
+    cachedDataViewMemory0 = new DataView(wasm.memory.buffer);
+  }
+  return cachedDataViewMemory0;
+}
+var cachedFloat32ArrayMemory0 = null;
+function getFloat32ArrayMemory0() {
+  if (cachedFloat32ArrayMemory0 === null || cachedFloat32ArrayMemory0.byteLength === 0) {
+    cachedFloat32ArrayMemory0 = new Float32Array(wasm.memory.buffer);
+  }
+  return cachedFloat32ArrayMemory0;
+}
+var cachedUint32ArrayMemory0 = null;
+function getUint32ArrayMemory0() {
+  if (cachedUint32ArrayMemory0 === null || cachedUint32ArrayMemory0.byteLength === 0) {
+    cachedUint32ArrayMemory0 = new Uint32Array(wasm.memory.buffer);
+  }
+  return cachedUint32ArrayMemory0;
+}
+var cachedUint8ArrayMemory0 = null;
+function getUint8ArrayMemory0() {
+  if (cachedUint8ArrayMemory0 === null || cachedUint8ArrayMemory0.byteLength === 0) {
+    cachedUint8ArrayMemory0 = new Uint8Array(wasm.memory.buffer);
+  }
+  return cachedUint8ArrayMemory0;
+}
+function getObject(idx) {
+  return heap[idx];
+}
+var heap = new Array(128).fill(void 0);
+heap.push(void 0, null, true, false);
+var heap_next = heap.length;
+function passArray32ToWasm0(arg, malloc) {
+  const ptr = malloc(arg.length * 4, 4) >>> 0;
+  getUint32ArrayMemory0().set(arg, ptr / 4);
+  WASM_VECTOR_LEN = arg.length;
+  return ptr;
+}
+function passArray8ToWasm0(arg, malloc) {
+  const ptr = malloc(arg.length * 1, 1) >>> 0;
+  getUint8ArrayMemory0().set(arg, ptr / 1);
+  WASM_VECTOR_LEN = arg.length;
+  return ptr;
+}
+function passArrayF32ToWasm0(arg, malloc) {
+  const ptr = malloc(arg.length * 4, 4) >>> 0;
+  getFloat32ArrayMemory0().set(arg, ptr / 4);
+  WASM_VECTOR_LEN = arg.length;
+  return ptr;
+}
+function takeObject(idx) {
+  const ret = getObject(idx);
+  dropObject(idx);
+  return ret;
+}
+var WASM_VECTOR_LEN = 0;
+function append_data_from_buffer(buffer_id, u8buf, vertex_count, out_center_scale, out_cov_color) {
+  const ptr0 = passArray8ToWasm0(u8buf, wasm.__wbindgen_export);
+  const len0 = WASM_VECTOR_LEN;
+  var ptr1 = passArrayF32ToWasm0(out_center_scale, wasm.__wbindgen_export);
+  var len1 = WASM_VECTOR_LEN;
+  var ptr2 = passArray32ToWasm0(out_cov_color, wasm.__wbindgen_export);
+  var len2 = WASM_VECTOR_LEN;
+  wasm.append_data_from_buffer(buffer_id, ptr0, len0, vertex_count, ptr1, len1, addHeapObject(out_center_scale), ptr2, len2, addHeapObject(out_cov_color));
+}
+function compute_bounds(buffer_id) {
+  try {
+    const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
+    wasm.compute_bounds(retptr, buffer_id);
+    var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
+    var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
+    var v1 = getArrayF32FromWasm0(r0, r1).slice();
+    wasm.__wbindgen_export2(r0, r1 * 4, 4);
+    return v1;
+  } finally {
+    wasm.__wbindgen_add_to_stack_pointer(16);
+  }
+}
+function process_splats_from_buffer(u8buf, vertex_count, out_center_scale, out_cov_color) {
+  const ptr0 = passArray8ToWasm0(u8buf, wasm.__wbindgen_export);
+  const len0 = WASM_VECTOR_LEN;
+  var ptr1 = passArrayF32ToWasm0(out_center_scale, wasm.__wbindgen_export);
+  var len1 = WASM_VECTOR_LEN;
+  var ptr2 = passArray32ToWasm0(out_cov_color, wasm.__wbindgen_export);
+  var len2 = WASM_VECTOR_LEN;
+  const ret = wasm.process_splats_from_buffer(ptr0, len0, vertex_count, ptr1, len1, addHeapObject(out_center_scale), ptr2, len2, addHeapObject(out_cov_color));
+  return ret >>> 0;
+}
+function process_splats_from_geometry(positions, scales, rotations, colors, vertex_count, out_center_scale, out_cov_color) {
+  const ptr0 = passArrayF32ToWasm0(positions, wasm.__wbindgen_export);
+  const len0 = WASM_VECTOR_LEN;
+  const ptr1 = passArrayF32ToWasm0(scales, wasm.__wbindgen_export);
+  const len1 = WASM_VECTOR_LEN;
+  const ptr2 = passArrayF32ToWasm0(rotations, wasm.__wbindgen_export);
+  const len2 = WASM_VECTOR_LEN;
+  const ptr3 = passArray8ToWasm0(colors, wasm.__wbindgen_export);
+  const len3 = WASM_VECTOR_LEN;
+  var ptr4 = passArrayF32ToWasm0(out_center_scale, wasm.__wbindgen_export);
+  var len4 = WASM_VECTOR_LEN;
+  var ptr5 = passArray32ToWasm0(out_cov_color, wasm.__wbindgen_export);
+  var len5 = WASM_VECTOR_LEN;
+  const ret = wasm.process_splats_from_geometry(ptr0, len0, ptr1, len1, ptr2, len2, ptr3, len3, vertex_count, ptr4, len4, addHeapObject(out_center_scale), ptr5, len5, addHeapObject(out_cov_color));
+  return ret >>> 0;
+}
+function process_splats_from_spz(positions, scales, rotations, colors, alphas, vertex_count, out_center_scale, out_cov_color) {
+  const ptr0 = passArrayF32ToWasm0(positions, wasm.__wbindgen_export);
+  const len0 = WASM_VECTOR_LEN;
+  const ptr1 = passArrayF32ToWasm0(scales, wasm.__wbindgen_export);
+  const len1 = WASM_VECTOR_LEN;
+  const ptr2 = passArrayF32ToWasm0(rotations, wasm.__wbindgen_export);
+  const len2 = WASM_VECTOR_LEN;
+  const ptr3 = passArrayF32ToWasm0(colors, wasm.__wbindgen_export);
+  const len3 = WASM_VECTOR_LEN;
+  const ptr4 = passArrayF32ToWasm0(alphas, wasm.__wbindgen_export);
+  const len4 = WASM_VECTOR_LEN;
+  var ptr5 = passArrayF32ToWasm0(out_center_scale, wasm.__wbindgen_export);
+  var len5 = WASM_VECTOR_LEN;
+  var ptr6 = passArray32ToWasm0(out_cov_color, wasm.__wbindgen_export);
+  var len6 = WASM_VECTOR_LEN;
+  const ret = wasm.process_splats_from_spz(ptr0, len0, ptr1, len1, ptr2, len2, ptr3, len3, ptr4, len4, vertex_count, ptr5, len5, addHeapObject(out_center_scale), ptr6, len6, addHeapObject(out_cov_color));
+  return ret >>> 0;
+}
+function sort_splats(buffer_id, view, frustum, threshold) {
+  try {
+    const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
+    const ptr0 = passArrayF32ToWasm0(view, wasm.__wbindgen_export);
+    const len0 = WASM_VECTOR_LEN;
+    const ptr1 = passArrayF32ToWasm0(frustum, wasm.__wbindgen_export);
+    const len1 = WASM_VECTOR_LEN;
+    wasm.sort_splats(retptr, buffer_id, ptr0, len0, ptr1, len1, threshold);
+    var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
+    var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
+    var v3 = getArrayU32FromWasm0(r0, r1).slice();
+    wasm.__wbindgen_export2(r0, r1 * 4, 4);
+    return v3;
+  } finally {
+    wasm.__wbindgen_add_to_stack_pointer(16);
+  }
+}
+function unregister_positions(buffer_id) {
+  wasm.unregister_positions(buffer_id);
+}
+var EXPECTED_RESPONSE_TYPES = /* @__PURE__ */ new Set(["basic", "cors", "default"]);
+async function __wbg_load(module, imports) {
+  if (typeof Response === "function" && module instanceof Response) {
+    if (typeof WebAssembly.instantiateStreaming === "function") {
+      try {
+        return await WebAssembly.instantiateStreaming(module, imports);
+      } catch (e) {
+        const validResponse = module.ok && EXPECTED_RESPONSE_TYPES.has(module.type);
+        if (validResponse && module.headers.get("Content-Type") !== "application/wasm") {
+          console.warn("`WebAssembly.instantiateStreaming` failed because your server does not serve Wasm with `application/wasm` MIME type. Falling back to `WebAssembly.instantiate` which is slower. Original error:\n", e);
+        } else {
+          throw e;
+        }
+      }
+    }
+    const bytes = await module.arrayBuffer();
+    return await WebAssembly.instantiate(bytes, imports);
+  } else {
+    const instance = await WebAssembly.instantiate(module, imports);
+    if (instance instanceof WebAssembly.Instance) {
+      return { instance, module };
+    } else {
+      return instance;
+    }
+  }
+}
+function __wbg_get_imports() {
+  const imports = {};
+  imports.wbg = {};
+  imports.wbg.__wbg___wbindgen_copy_to_typed_array_db832bc4df7216c1 = function(arg0, arg1, arg2) {
+    new Uint8Array(getObject(arg2).buffer, getObject(arg2).byteOffset, getObject(arg2).byteLength).set(getArrayU8FromWasm0(arg0, arg1));
+  };
+  imports.wbg.__wbindgen_object_drop_ref = function(arg0) {
+    takeObject(arg0);
+  };
+  return imports;
+}
+function __wbg_finalize_init(instance, module) {
+  wasm = instance.exports;
+  __wbg_init.__wbindgen_wasm_module = module;
+  cachedDataViewMemory0 = null;
+  cachedFloat32ArrayMemory0 = null;
+  cachedUint32ArrayMemory0 = null;
+  cachedUint8ArrayMemory0 = null;
+  return wasm;
+}
+async function __wbg_init(module_or_path) {
+  if (wasm !== void 0) return wasm;
+  if (typeof module_or_path !== "undefined") {
+    if (Object.getPrototypeOf(module_or_path) === Object.prototype) {
+      ({ module_or_path } = module_or_path);
+    } else {
+      console.warn("using deprecated parameters for the initialization function; pass a single object instead");
+    }
+  }
+  if (typeof module_or_path === "undefined") {
+    module_or_path = new URL("wasm_splat_bg.wasm", import.meta.url);
+  }
+  const imports = __wbg_get_imports();
+  if (typeof module_or_path === "string" || typeof Request === "function" && module_or_path instanceof Request || typeof URL === "function" && module_or_path instanceof URL) {
+    module_or_path = fetch(module_or_path);
+  }
+  const { instance, module } = await __wbg_load(await module_or_path, imports);
+  return __wbg_finalize_init(instance, module);
+}
+var wasm_splat_default = __wbg_init;
+
+// workers/wasm_splat.worker.js
+var wasmReady = false;
+var initPromise = null;
+var meshCache = {};
+async function _ensureWasm(wasmUrl) {
+  if (wasmReady) return true;
+  if (!initPromise) {
+    initPromise = wasm_splat_default(wasmUrl).then(() => {
+      wasmReady = true;
+      return true;
+    }).catch((err) => {
+      wasmReady = false;
+      initPromise = null;
+      throw err;
+    });
+  }
+  return initPromise;
+}
+function _createOutputBuffers(count) {
+  const outCS = new Float32Array(new ArrayBuffer(count * 4 * 4));
+  const outRC = new Uint32Array(new ArrayBuffer(count * 4 * 4));
+  return { outCS, outRC };
+}
+self.onmessage = async (e) => {
+  const { id, fn, wasmUrl, args } = e.data;
+  try {
+    let result;
+    if (fn === "__init__") {
+      if (!wasmReady) {
+        await wasm_splat_default(wasmUrl);
+        wasmReady = true;
+      }
+      postMessage({ id, result: true });
+      return;
+    }
+    await _ensureWasm(wasmUrl);
+    if (fn === "process_splats_from_buffer") {
+      const [meshId, buffer, count] = args;
+      const u8 = new Uint8Array(buffer);
+      const { outCS, outRC } = _createOutputBuffers(count);
+      const bufferId = process_splats_from_buffer(u8, count, outCS, outRC);
+      meshCache[meshId] = { bufferId };
+      result = { meshId, out_cs: outCS, out_rc: outRC };
+    } else if (fn === "append_data_from_buffer") {
+      let [meshId, buffer, count] = args;
+      const { bufferId } = meshCache[meshId];
+      let u8 = new Uint8Array(buffer);
+      const { outCS, outRC } = _createOutputBuffers(count);
+      append_data_from_buffer(bufferId, u8, count, outCS, outRC);
+      result = { out_cs: outCS, out_rc: outRC };
+      buffer = null;
+      u8 = null;
+    } else if (fn === "process_splats_from_geometry") {
+      let [meshId, positions, scales, rotations, colors, count] = args;
+      const { outCS, outRC } = _createOutputBuffers(count);
+      const bufferId = process_splats_from_geometry(
+        positions,
+        scales,
+        rotations,
+        colors,
+        count,
+        outCS,
+        outRC
+      );
+      meshCache[meshId] = { bufferId };
+      result = { meshId, out_cs: outCS, out_rc: outRC };
+      positions = null;
+      scales = null;
+      rotations = null;
+      colors = null;
+    } else if (fn === "process_splats_from_spz") {
+      let [meshId, positions, scales, rotations, colors, alphas, count] = args;
+      const { outCS, outRC } = _createOutputBuffers(count);
+      const bufferId = process_splats_from_spz(
+        positions,
+        scales,
+        rotations,
+        colors,
+        alphas,
+        count,
+        outCS,
+        outRC
+      );
+      meshCache[meshId] = { bufferId };
+      result = { meshId, out_cs: outCS, out_rc: outRC };
+      positions = null;
+      scales = null;
+      rotations = null;
+      colors = null;
+      alphas = null;
+    } else if (fn === "sort_splats") {
+      const [meshId, view, planes, threshold] = args;
+      const { bufferId } = meshCache[meshId];
+      const data = sort_splats(bufferId, view, planes, threshold);
+      result = { meshId, data };
+    } else if (fn === "compute_bounds") {
+      const [meshId] = args;
+      const { bufferId } = meshCache[meshId];
+      const data = compute_bounds(bufferId);
+      result = { meshId, data };
+    } else if (fn === "unregister_positions") {
+      const [meshId] = args;
+      const { bufferId } = meshCache[meshId];
+      unregister_positions(bufferId);
+      result = true;
+    }
+    if (result instanceof Uint32Array) {
+      postMessage({ id, result }, [result.buffer]);
+      result = null;
+    } else if (result && result.data) {
+      postMessage({ id, result }, [result.data.buffer]);
+      result.data = null;
+      result = null;
+    } else if (result && result.out_cs && result.out_rc) {
+      postMessage({ id, result }, [result.out_cs.buffer, result.out_rc.buffer]);
+      result.out_cs = null;
+      result.out_rc = null;
+      result = null;
+    } else {
+      postMessage({ id, result });
+    }
+  } catch (err) {
+    console.error(err);
+    postMessage({ id, error: err?.message || String(err) });
+  }
+};

package/package.json

@@ -1,40 +1,40 @@
-{
-  "name": "@dvt3d/splat-mesh",
-  "version": "1.0.0",
-  "main": "dist/index.js",
-  "repository": "https://github.com/dvt3d/splat-mesh.git",
-  "author": "Caven Chen <cavencj@gmail.com>",
-  "license": "Apache-2.0",
-  "scripts": {
-    "dev": "rimraf dist && gulp dev",
-    "build": "rimraf dist && gulp build",
-    "build:node": "rimraf dist && gulp buildNode",
-    "build:iife": "rimraf dist && gulp buildIIFE",
-    "build:release": "rimraf dist && gulp buildRelease",
-    "prepublishOnly": "yarn run build:release",
-    "lint": "eslint --fix src"
-  },
-  "peerDependencies": {
-    "three": "^0.182.0"
-  },
-  "keywords": [
-    "Gaussian splatting",
-    "3dgs",
-    "three.js",
-    "splat-mesh"
-  ],
-  "files": [
-    "dist"
-  ],
-  "devDependencies": {
-    "chalk": "^5.6.2",
-    "esbuild": "^0.27.2",
-    "esbuild-plugin-globals": "^0.2.0",
-    "esbuild-plugin-glsl": "^1.4.1",
-    "fs-extra": "^11.3.3",
-    "gulp": "^5.0.1",
-    "prettier": "^3.7.4",
-    "rimraf": "^6.1.2",
-    "shelljs": "^0.10.0"
-  }
-}
+{
+  "name": "@dvt3d/splat-mesh",
+  "version": "1.1.0",
+  "main": "dist/index.js",
+  "repository": "https://github.com/dvt3d/splat-mesh.git",
+  "author": "Caven Chen <cavencj@gmail.com>",
+  "license": "Apache-2.0",
+  "scripts": {
+    "dev": "rimraf dist && gulp dev",
+    "build": "rimraf dist && gulp build",
+    "build:node": "rimraf dist && gulp buildNode",
+    "build:iife": "rimraf dist && gulp buildIIFE",
+    "build:release": "rimraf dist && gulp buildRelease",
+    "prepublishOnly": "yarn run build:release",
+    "lint": "eslint --fix src"
+  },
+  "peerDependencies": {
+    "three": "^0.182.0"
+  },
+  "keywords": [
+    "Gaussian splatting",
+    "3dgs",
+    "three.js",
+    "splat-mesh"
+  ],
+  "files": [
+    "dist"
+  ],
+  "devDependencies": {
+    "chalk": "^5.6.2",
+    "esbuild": "^0.27.2",
+    "esbuild-plugin-globals": "^0.2.0",
+    "esbuild-plugin-glsl": "^1.4.1",
+    "fs-extra": "^11.3.3",
+    "gulp": "^5.0.1",
+    "prettier": "^3.7.4",
+    "rimraf": "^6.1.2",
+    "shelljs": "^0.10.0"
+  }
+}

package/dist/workers/wasm_splat.min.js

@@ -1 +0,0 @@
-let e;function n(e){l===f.length&&f.push(f.length+1);const n=l;return l=f[n],f[n]=e,n}function t(e,n){return e>>>=0,i().subarray(e/4,e/4+n)}let r=null;function o(){return(null===r||!0===r.buffer.detached||void 0===r.buffer.detached&&r.buffer!==e.memory.buffer)&&(r=new DataView(e.memory.buffer)),r}let _=null;function i(){return null!==_&&0!==_.byteLength||(_=new Float32Array(e.memory.buffer)),_}let s=null;function a(){return null!==s&&0!==s.byteLength||(s=new Uint32Array(e.memory.buffer)),s}let u=null;function c(){return null!==u&&0!==u.byteLength||(u=new Uint8Array(e.memory.buffer)),u}function p(e){return f[e]}let f=new Array(128).fill(void 0);f.push(void 0,null,!0,!1);let l=f.length;function b(e,n){const t=n(4*e.length,4)>>>0;return a().set(e,t/4),y=e.length,t}function d(e,n){const t=n(1*e.length,1)>>>0;return c().set(e,t/1),y=e.length,t}function g(e,n){const t=n(4*e.length,4)>>>0;return i().set(e,t/4),y=e.length,t}function w(e){const n=p(e);return function(e){e<132||(f[e]=l,l=e)}(e),n}let y=0;export function append_data_from_buffer(t,r,o,_,i){const s=d(r,e.__wbindgen_export),a=y;var u=g(_,e.__wbindgen_export),c=y,p=b(i,e.__wbindgen_export),f=y;e.append_data_from_buffer(t,s,a,o,u,c,n(_),p,f,n(i))}export function compute_bounds(n){try{const s=e.__wbindgen_add_to_stack_pointer(-16);e.compute_bounds(s,n);var r=o().getInt32(s+0,!0),_=o().getInt32(s+4,!0),i=t(r,_).slice();return e.__wbindgen_export2(r,4*_,4),i}finally{e.__wbindgen_add_to_stack_pointer(16)}}export function get_positions(n){try{const s=e.__wbindgen_add_to_stack_pointer(-16);e.get_positions(s,n);var r=o().getInt32(s+0,!0),_=o().getInt32(s+4,!0),i=t(r,_).slice();return e.__wbindgen_export2(r,4*_,4),i}finally{e.__wbindgen_add_to_stack_pointer(16)}}export function process_splats_from_buffer(t,r,o,_){const i=d(t,e.__wbindgen_export),s=y;var a=g(o,e.__wbindgen_export),u=y,c=b(_,e.__wbindgen_export),p=y;return e.process_splats_from_buffer(i,s,r,a,u,n(o),c,p,n(_))>>>0}export function process_splats_from_geometry(t,r,o,_,i,s,a){const u=g(t,e.__wbindgen_export),c=y,p=g(r,e.__wbindgen_export),f=y,l=g(o,e.__wbindgen_export),w=y,m=d(_,e.__wbindgen_export),x=y;var h=g(s,e.__wbindgen_export),v=y,A=b(a,e.__wbindgen_export),W=y;return e.process_splats_from_geometry(u,c,p,f,l,w,m,x,i,h,v,n(s),A,W,n(a))>>>0}export function process_splats_from_spz(t,r,o,_,i,s,a,u){const c=g(t,e.__wbindgen_export),p=y,f=g(r,e.__wbindgen_export),l=y,d=g(o,e.__wbindgen_export),w=y,m=g(_,e.__wbindgen_export),x=y,h=g(i,e.__wbindgen_export),v=y;var A=g(a,e.__wbindgen_export),W=y,I=b(u,e.__wbindgen_export),k=y;return e.process_splats_from_spz(c,p,f,l,d,w,m,x,h,v,s,A,W,n(a),I,k,n(u))>>>0}export function sort_splats(n,t,r,_){try{const f=e.__wbindgen_add_to_stack_pointer(-16),l=g(t,e.__wbindgen_export),b=y,d=g(r,e.__wbindgen_export),w=y;e.sort_splats(f,n,l,b,d,w,_);var i=o().getInt32(f+0,!0),s=o().getInt32(f+4,!0),u=(c=i,p=s,c>>>=0,a().subarray(c/4,c/4+p)).slice();return e.__wbindgen_export2(i,4*s,4),u}finally{e.__wbindgen_add_to_stack_pointer(16)}var c,p}export function unregister_positions(n){e.unregister_positions(n)}const m=new Set(["basic","cors","default"]);function x(){const e={wbg:{}};return e.wbg.__wbg___wbindgen_copy_to_typed_array_db832bc4df7216c1=function(e,n,t){var r,o;new Uint8Array(p(t).buffer,p(t).byteOffset,p(t).byteLength).set((r=e,o=n,r>>>=0,c().subarray(r/1,r/1+o)))},e.wbg.__wbindgen_object_drop_ref=function(e){w(e)},e}function h(n,t){return e=n.exports,A.__wbindgen_wasm_module=t,r=null,_=null,s=null,u=null,e}function v(n){if(void 0!==e)return e;void 0!==n&&(Object.getPrototypeOf(n)===Object.prototype?({module:n}=n):console.warn("using deprecated parameters for `initSync()`; pass a single object instead"));const t=x();n instanceof WebAssembly.Module||(n=new WebAssembly.Module(n));return h(new WebAssembly.Instance(n,t),n)}async function A(n){if(void 0!==e)return e;void 0!==n&&(Object.getPrototypeOf(n)===Object.prototype?({module_or_path:n}=n):console.warn("using deprecated parameters for the initialization function; pass a single object instead")),void 0===n&&(n=new URL("wasm_splat_bg.wasm",import.meta.url));const t=x();("string"==typeof n||"function"==typeof Request&&n instanceof Request||"function"==typeof URL&&n instanceof URL)&&(n=fetch(n));const{instance:r,module:o}=await async function(e,n){if("function"==typeof Response&&e instanceof Response){if("function"==typeof WebAssembly.instantiateStreaming)try{return await WebAssembly.instantiateStreaming(e,n)}catch(n){if(!e.ok||!m.has(e.type)||"application/wasm"===e.headers.get("Content-Type"))throw n;console.warn("`WebAssembly.instantiateStreaming` failed because your server does not serve Wasm with `application/wasm` MIME type. Falling back to `WebAssembly.instantiate` which is slower. Original error:\n",n)}const t=await e.arrayBuffer();return await WebAssembly.instantiate(t,n)}{const t=await WebAssembly.instantiate(e,n);return t instanceof WebAssembly.Instance?{instance:t,module:e}:t}}(await n,t);return h(r,o)}export{v as initSync};export default A;