UNPKG

@animech-public/playcanvas

Version:
131 lines (128 loc) 4.79 kB
import { SEMANTIC_POSITION, TYPE_FLOAT32, SEMANTIC_TEXCOORD0, BUFFER_STREAM, INDEXFORMAT_UINT16, BUFFER_STATIC, PRIMITIVE_TRIANGLES, CULLFACE_NONE, BLENDEQUATION_ADD, BLENDMODE_SRC_ALPHA, BLENDMODE_ONE_MINUS_SRC_ALPHA, BLENDMODE_ONE } from '../../platform/graphics/constants.js'; import { DepthState } from '../../platform/graphics/depth-state.js'; import { BlendState } from '../../platform/graphics/blend-state.js'; import { GraphNode } from '../../scene/graph-node.js'; import { MeshInstance } from '../../scene/mesh-instance.js'; import { Material } from '../../scene/materials/material.js'; import { Mesh } from '../../scene/mesh.js'; import { IndexBuffer } from '../../platform/graphics/index-buffer.js'; import { VertexBuffer } from '../../platform/graphics/vertex-buffer.js'; import { VertexFormat } from '../../platform/graphics/vertex-format.js'; import { shaderChunks } from '../../scene/shader-lib/chunks/chunks.js'; const vertexShader = ` attribute vec3 vertex_position; attribute vec4 vertex_texCoord0; varying vec4 uv0; varying float wordFlag; void main(void) { gl_Position = vec4(vertex_position.xy * 2.0 - 1.0, 0.5, 1.0); uv0 = vertex_texCoord0; wordFlag = vertex_position.z; }`; const fragmentShader = ` varying vec4 uv0; varying float wordFlag; uniform vec4 clr; uniform sampler2D graphTex; uniform sampler2D wordsTex; void main (void) { vec4 graphSample = texture2D(graphTex, uv0.xy); vec4 graph; if (uv0.w < graphSample.r) graph = vec4(0.7, 0.2, 0.2, 1.0); else if (uv0.w < graphSample.g) graph = vec4(0.2, 0.7, 0.2, 1.0); else if (uv0.w < graphSample.b) graph = vec4(0.2, 0.2, 0.7, 1.0); else graph = vec4(0.0, 0.0, 0.0, 1.0 - 0.25 * sin(uv0.w * 3.14159)); vec4 words = texture2D(wordsTex, vec2(uv0.x, 1.0 - uv0.y)); gl_FragColor = mix(graph, words, wordFlag) * clr; }`; class Render2d { constructor(device, maxQuads = 512) { const format = new VertexFormat(device, [{ semantic: SEMANTIC_POSITION, components: 3, type: TYPE_FLOAT32 }, { semantic: SEMANTIC_TEXCOORD0, components: 4, type: TYPE_FLOAT32 }]); const indices = new Uint16Array(maxQuads * 6); for (let i = 0; i < maxQuads; ++i) { indices[i * 6 + 0] = i * 4; indices[i * 6 + 1] = i * 4 + 1; indices[i * 6 + 2] = i * 4 + 2; indices[i * 6 + 3] = i * 4; indices[i * 6 + 4] = i * 4 + 2; indices[i * 6 + 5] = i * 4 + 3; } const shader = shaderChunks.createShaderFromCode(device, vertexShader, fragmentShader, 'mini-stats'); this.device = device; this.buffer = new VertexBuffer(device, format, maxQuads * 4, { usage: BUFFER_STREAM }); this.data = new Float32Array(this.buffer.numBytes / 4); this.indexBuffer = new IndexBuffer(device, INDEXFORMAT_UINT16, maxQuads * 6, BUFFER_STATIC, indices); this.prim = { type: PRIMITIVE_TRIANGLES, indexed: true, base: 0, count: 0 }; this.quads = 0; this.mesh = new Mesh(device); this.mesh.vertexBuffer = this.buffer; this.mesh.indexBuffer[0] = this.indexBuffer; this.mesh.primitive = [this.prim]; const material = new Material(); this.material = material; material.cull = CULLFACE_NONE; material.shader = shader; material.depthState = DepthState.NODEPTH; material.blendState = new BlendState(true, BLENDEQUATION_ADD, BLENDMODE_SRC_ALPHA, BLENDMODE_ONE_MINUS_SRC_ALPHA, BLENDEQUATION_ADD, BLENDMODE_ONE, BLENDMODE_ONE); material.update(); this.meshInstance = new MeshInstance(this.mesh, material, new GraphNode('MiniStatsMesh')); this.uniforms = { clr: new Float32Array(4) }; this.targetSize = { width: device.width, height: device.height }; } quad(x, y, w, h, u, v, uw, uh, texture, wordFlag = 0) { const rw = this.targetSize.width; const rh = this.targetSize.height; const x0 = x / rw; const y0 = y / rh; const x1 = (x + w) / rw; const y1 = (y + h) / rh; const tw = texture.width; const th = texture.height; const u0 = u / tw; const v0 = v / th; const u1 = (u + (uw != null ? uw : w)) / tw; const v1 = (v + (uh != null ? uh : h)) / th; this.data.set([x0, y0, wordFlag, u0, v0, 0, 0, x1, y0, wordFlag, u1, v0, 1, 0, x1, y1, wordFlag, u1, v1, 1, 1, x0, y1, wordFlag, u0, v1, 0, 1], 4 * 7 * this.quads); this.quads++; this.prim.count += 6; } startFrame() { this.quads = 0; this.prim.count = 0; this.targetSize.width = this.device.canvas.scrollWidth; this.targetSize.height = this.device.canvas.scrollHeight; } render(app, layer, graphTexture, wordsTexture, clr, height) { this.buffer.setData(this.data.buffer); this.uniforms.clr.set(clr, 0); this.material.setParameter('clr', this.uniforms.clr); this.material.setParameter('graphTex', graphTexture); this.material.setParameter('wordsTex', wordsTexture); app.drawMeshInstance(this.meshInstance, layer); } } export { Render2d };