Spaces:
Running
Running
File size: 9,174 Bytes
d4a90dc 4c4106b d4a90dc 78c9607 d4a90dc 4313cb9 7a6ace8 4c4106b 7a6ace8 4c4106b 4313cb9 8ab0d15 aaffc16 4c4106b b7e38c4 4c4106b b7e38c4 4c4106b 8ab0d15 4c4106b 8ab0d15 4c4106b 8ab0d15 4c4106b 8ab0d15 b7e38c4 ad1e7e0 8ab0d15 c3695de 78c9607 c3695de 4c4106b 78c9607 8ab0d15 4c4106b 78c9607 4c4106b ad1e7e0 4c4106b ad1e7e0 4313cb9 c3695de 4c4106b c3695de 4c4106b 8ab0d15 4c4106b ad1e7e0 4c4106b 78c9607 4c4106b 4313cb9 4c4106b 78c9607 57af44f 78c9607 4c4106b 78c9607 4c4106b ab2822f 4c4106b 78c9607 4c4106b 8ab0d15 78c9607 4c4106b 8ab0d15 78c9607 4c4106b 8ab0d15 78c9607 4c4106b 8ab0d15 78c9607 4c4106b a319f30 78c9607 a319f30 4c4106b 78c9607 4c4106b ce4f9e0 4c4106b d4a90dc 4c4106b |
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 183 184 |
<!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';
async function main() {
const adapter = await navigator.gpu?.requestAdapter();
const device = await adapter?.requestDevice();
if (!device) {
alert('need a browser that supports WebGPU');
return;
}
const canvas = document.querySelector('canvas');
const context = canvas.getContext('webgpu');
const presentationFormat = navigator.gpu.getPreferredCanvasFormat();
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);
glyphCanvas.style.backgroundColor = '#222';
const vertexSize = CONFIG.floatsPerVertex * 4;
const vertexBufferSize = CONFIG.maxGlyphs * CONFIG.vertsPerGlyph * vertexSize;
const vertexBuffer = device.createBuffer({
label: 'vertices',
size: vertexBufferSize,
usage: GPUBufferUsage.VERTEX | GPUBufferUsage.COPY_DST,
});
const 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));
function generateGlyphVerticesForText(s, COLORS = [[1, 1, 1, 1]]) {
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 };
}
const { vertexData, numGlyphs, width, height } = generateGlyphVerticesForText('Hello\nworld!\nText in\nWebGPU!', COLORS);
device.queue.writeBuffer(vertexBuffer, 0, vertexData);
const 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' }
},
}],
},
});
function createTextureFromSource(device, source, options = {}) {
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: options.flipY },
{ texture, premultipliedAlpha: true },
{ width: source.width, height: source.height }
);
return texture;
}
const texture = createTextureFromSource(device, glyphCanvas, { mips: true });
const sampler = device.createSampler({
minFilter: 'linear',
magFilter: 'linear'
});
const 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);
const 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);
}
main();
</script>
</body>
</html> |