Spaces:
Running
Running
File size: 8,621 Bytes
d4a90dc 4c4106b d4a90dc 78c9607 d4a90dc 4313cb9 7a6ace8 4c4106b 7a6ace8 4c4106b 4313cb9 8ab0d15 aaffc16 8e21624 d48f607 8e21624 4c4106b 8e21624 d48f607 8e21624 d48f607 8e21624 b7e38c4 d48f607 8e21624 8ab0d15 d48f607 8e21624 8ab0d15 d48f607 8e21624 8ab0d15 b7e38c4 8e21624 4c4106b d48f607 8e21624 78c9607 4c4106b 4313cb9 4c4106b 78c9607 57af44f 78c9607 4c4106b 8e21624 d48f607 8e21624 8ab0d15 78c9607 8e21624 4c4106b 8e21624 4c4106b 8ab0d15 78c9607 4c4106b 8ab0d15 78c9607 4c4106b a319f30 78c9607 a319f30 4c4106b 78c9607 4c4106b 8e21624 4c4106b ce4f9e0 8e21624 d48f607 8e21624 d48f607 8e21624 4c4106b d4a90dc 4c4106b 8e21624 |
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 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 |
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Plasma-Arc: WGPU Experiment</title>
</head>
<body>
<canvas></canvas>
<script type="module">
import { mat4 } from 'https://webgpufundamentals.org/3rdparty/wgpu-matrix.module.js';
import { fetchShaderCode, generateGlyphTextureAtlas } from './utility.js';
import { CONFIG } from './config.js';
import { CANVAS, CTX, COLORS, RENDER_PASS_DESCRIPTOR } from './constants.js';
const canvas = document.querySelector('canvas');
const context = canvas.getContext('webgpu');
const presentationFormat = navigator.gpu.getPreferredCanvasFormat();
let device, pipeline, vertexBuffer, indexBuffer, uniformBuffer, texture, sampler, bindGroup;
async function main() {
const adapter = await navigator.gpu?.requestAdapter();
device = await adapter?.requestDevice();
if (!device) return alert('WebGPU support required');
context.configure({ device, format: presentationFormat });
const shaderCode = await fetchShaderCode('shaders.wgsl');
const module = device.createShaderModule({ label: 'textured quad shaders', code: shaderCode });
const glyphCanvas = generateGlyphTextureAtlas(CANVAS, CTX, CONFIG);
document.body.appendChild(glyphCanvas).style.backgroundColor = '#222';
const vertexSize = CONFIG.floatsPerVertex * 4;
const vertexBufferSize = CONFIG.maxGlyphs * CONFIG.vertsPerGlyph * vertexSize;
vertexBuffer = device.createBuffer({ label: 'vertices', size: vertexBufferSize, usage: GPUBufferUsage.VERTEX | GPUBufferUsage.COPY_DST });
indexBuffer = device.createBuffer({ label: 'indices', size: CONFIG.maxGlyphs * CONFIG.vertsPerGlyph * 4, usage: GPUBufferUsage.INDEX | GPUBufferUsage.COPY_DST });
const indices = Array.from({ length: CONFIG.maxGlyphs * 6 }, (_, i) => {
const ndx = Math.floor(i / 6) * 4;
return (i % 6 < 3 ? [ndx, ndx + 1, ndx + 2] : [ndx + 2, ndx + 1, ndx + 3])[i % 3];
});
device.queue.writeBuffer(indexBuffer, 0, new Uint32Array(indices));
const { vertexData, numGlyphs, width, height } = generateGlyphVerticesForText('Hello\nworld!\nText in\nWebGPU!', COLORS, glyphCanvas);
device.queue.writeBuffer(vertexBuffer, 0, vertexData);
pipeline = device.createRenderPipeline({
label: 'textured quad pipeline',
layout: 'auto',
vertex: {
module,
entryPoint: 'vs',
buffers: [
{
arrayStride: vertexSize,
attributes: [
{ shaderLocation: 0, offset: 0, format: 'float32x2' }, // pos
{ shaderLocation: 1, offset: 8, format: 'float32x2' }, // tex
{ shaderLocation: 2, offset: 16, format: 'float32x4' } // col
],
},
],
},
fragment: {
module,
entryPoint: 'fs',
targets: [{
format: presentationFormat,
blend: {
color: { srcFactor: 'one', dstFactor: 'one-minus-src-alpha', operation: 'add' },
alpha: { srcFactor: 'one', dstFactor: 'one-minus-src-alpha', operation: 'add' }
},
}],
},
});
texture = createTextureFromSource(device, glyphCanvas);
sampler = device.createSampler({ minFilter: 'linear', magFilter: 'linear' });
uniformBuffer = device.createBuffer({ label: 'uniforms for quad', size: CONFIG.uniformBufferSize, usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST });
const uniformValues = new Float32Array(CONFIG.uniformBufferSize / 4);
const matrix = uniformValues.subarray(0, 16);
bindGroup = device.createBindGroup({
layout: pipeline.getBindGroupLayout(0),
entries: [
{ binding: 0, resource: sampler },
{ binding: 1, resource: texture.createView() },
{ binding: 2, resource: { buffer: uniformBuffer } },
],
});
function render(time) {
time *= CONFIG.time.phase;
const fov = 60 * Math.PI / 180;
const aspect = canvas.clientWidth / canvas.clientHeight;
const projectionMatrix = mat4.perspective(fov, aspect, CONFIG.render.zNear, CONFIG.render.zFar);
const viewMatrix = mat4.lookAt([0, 0, 5], [0, 0, 0], [0, 1, 0]);
const viewProjectionMatrix = mat4.multiply(projectionMatrix, viewMatrix);
RENDER_PASS_DESCRIPTOR.colorAttachments[0].view = context.getCurrentTexture().createView();
const encoder = device.createCommandEncoder();
const pass = encoder.beginRenderPass(RENDER_PASS_DESCRIPTOR);
pass.setPipeline(pipeline);
mat4.rotateY(viewProjectionMatrix, time, matrix);
mat4.translate(matrix, [-width / 2, -height / 2, 0], matrix);
device.queue.writeBuffer(uniformBuffer, 0, uniformValues);
pass.setBindGroup(0, bindGroup);
pass.setVertexBuffer(0, vertexBuffer);
pass.setIndexBuffer(indexBuffer, 'uint32');
pass.drawIndexed(numGlyphs * 6);
pass.end();
device.queue.submit([encoder.finish()]);
requestAnimationFrame(render);
}
requestAnimationFrame(render);
}
function generateGlyphVerticesForText(s, COLORS, glyphCanvas) {
const vertexData = new Float32Array(CONFIG.maxGlyphs * CONFIG.floatsPerVertex * CONFIG.vertsPerGlyph);
const glyphUVWidth = CONFIG.glyphWidth / glyphCanvas.width;
const glyphUVHeight = CONFIG.glyphHeight / glyphCanvas.height;
let offset = 0, x0 = 0, y0 = 0, x1 = 1, y1 = 1, width = 0;
let colorNdx = 0;
const addVertex = (x, y, u, v, color) => {
vertexData.set([x, y, u, v, ...color], offset);
offset += 8;
};
for (let i = 0; i < s.length; ++i) {
const c = s.charCodeAt(i);
if (c >= 33) {
const cIndex = c - 33;
const glyphX = cIndex % CONFIG.glyphsAcrossTexture;
const glyphY = Math.floor(cIndex / CONFIG.glyphsAcrossTexture);
const u0 = glyphX * CONFIG.glyphWidth / glyphCanvas.width;
const v1 = glyphY * CONFIG.glyphHeight / glyphCanvas.height;
const u1 = u0 + glyphUVWidth;
const v0 = v1 + glyphUVHeight;
width = Math.max(x1, width);
addVertex(x0, y0, u0, v0, COLORS[colorNdx]);
addVertex(x1, y0, u1, v0, COLORS[colorNdx]);
addVertex(x0, y1, u0, v1, COLORS[colorNdx]);
addVertex(x1, y1, u1, v1, COLORS[colorNdx]);
} else {
colorNdx = (colorNdx + 1) % COLORS.length;
if (c === 10) { // Newline
x0 = 0; x1 = 1; y0--; y1 = y0 + 1;
continue;
}
}
x0 += 0.55; x1 = x0 + 1;
}
return { vertexData, numGlyphs: offset / CONFIG.floatsPerVertex, width, height: y1 };
}
function createTextureFromSource(device, source) {
const texture = device.createTexture({
format: 'rgba8unorm',
size: [source.width, source.height],
usage: GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_DST | GPUTextureUsage.RENDER_ATTACHMENT,
});
device.queue.copyExternalImageToTexture(
{ source, flipY: true },
{ texture, premultipliedAlpha: true },
{ width: source.width, height: source.height }
);
return texture;
}
main();
</script>
</body>
</html>
|