Spaces:
Sleeping
Sleeping
File size: 3,032 Bytes
e6b949c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
import * as SPLAT from "gsplat";
const axisVertexShader = /*glsl*/ `#version 300 es
uniform mat4 projection, view;
in vec3 position;
void main() {
gl_Position = projection * view * vec4(position, 1.0);
}
`;
const axisFragmentShader = /*glsl*/ `#version 300 es
precision mediump float;
uniform vec4 axisColor;
out vec4 outColor;
void main() {
outColor = axisColor;
}
`;
class AxisProgram extends SPLAT.ShaderProgram {
protected _initialize: () => void;
protected _resize: () => void;
protected _render: () => void;
protected _dispose: () => void;
constructor(renderer: SPLAT.WebGLRenderer, passes: SPLAT.ShaderPass[]) {
super(renderer, passes);
const gl = renderer.gl;
let vertexBuffer: WebGLBuffer;
let positionAttribute: number;
let u_projection: WebGLUniformLocation;
let u_view: WebGLUniformLocation;
let u_color: WebGLUniformLocation;
const xVertices = new Float32Array([-50, 0, 0, 50, 0, 0]);
const yVertices = new Float32Array([0, -50, 0, 0, 50, 0]);
const zVertices = new Float32Array([0, 0, -50, 0, 0, 50]);
const xColor = new Float32Array([1, 0, 0, 0.5]);
const yColor = new Float32Array([0, 1, 0, 0.5]);
const zColor = new Float32Array([0, 0.5, 1, 0.5]);
this._initialize = () => {
vertexBuffer = gl.createBuffer() as WebGLBuffer;
positionAttribute = gl.getAttribLocation(this.program, "position");
gl.enableVertexAttribArray(positionAttribute);
u_projection = gl.getUniformLocation(this.program, "projection") as WebGLUniformLocation;
u_view = gl.getUniformLocation(this.program, "view") as WebGLUniformLocation;
u_color = gl.getUniformLocation(this.program, "axisColor") as WebGLUniformLocation;
};
const drawAxis = (vertices: Float32Array, color: Float32Array) => {
gl.uniform4fv(u_color, color);
gl.bindBuffer(gl.ARRAY_BUFFER, vertexBuffer);
gl.bufferData(gl.ARRAY_BUFFER, vertices, gl.STATIC_DRAW);
gl.vertexAttribPointer(positionAttribute, 3, gl.FLOAT, false, 0, 0);
gl.drawArrays(gl.LINES, 0, 2);
};
this._resize = () => {};
this._render = () => {
if (!this._camera) {
throw new Error("Camera not set");
}
gl.enable(gl.BLEND);
gl.blendFunc(gl.SRC_ALPHA, gl.ONE_MINUS_SRC_ALPHA);
gl.uniformMatrix4fv(u_projection, false, this._camera.data.projectionMatrix.buffer);
gl.uniformMatrix4fv(u_view, false, this._camera.data.viewMatrix.buffer);
drawAxis(xVertices, xColor);
drawAxis(yVertices, yColor);
drawAxis(zVertices, zColor);
};
this._dispose = () => {};
}
protected _getVertexSource() {
return axisVertexShader;
}
protected _getFragmentSource() {
return axisFragmentShader;
}
}
export { AxisProgram };
|