src/rendering/renderers/gpu/shader/GpuShaderSystem.ts - dev - githubmirror/pixijs

src/rendering/renderers/gpu/shader/GpuShaderSystem.ts

76 строк2 KB

Zyie

chore: add standard/advanced tags to documentation (#11448)

03 июн 2025, 10:12

03 июн 2025, 10:127d91234

import { ExtensionType } from '../../../../extensions/Extensions';

import type { GPU } from '../GpuDeviceSystem';
import type { GpuProgram } from './GpuProgram';

/**
 * Data structure for GPU program layout.
 * Contains bind group layouts and pipeline layout.
 * @category rendering
 * @advanced
 */
export interface GPUProgramData
{
    bindGroups: GPUBindGroupLayout[]
    pipeline: GPUPipelineLayout
}

/**
 * A system that manages the rendering of GpuPrograms.
 * @category rendering
 * @advanced
 */
export class GpuShaderSystem
{
    /** @ignore */
    public static extension = {
        type: [
            ExtensionType.WebGPUSystem,
        ],
        name: 'shader',
    } as const;

    private _gpu: GPU;

    private readonly _gpuProgramData: Record<number, GPUProgramData> = Object.create(null);

    protected contextChange(gpu: GPU): void
    {
        this._gpu = gpu;
    }

    public getProgramData(program: GpuProgram)
    {
        return this._gpuProgramData[program._layoutKey] || this._createGPUProgramData(program);
    }

    private _createGPUProgramData(program: GpuProgram)
    {
        const device = this._gpu.device;

        const bindGroups = program.gpuLayout.map((group) => device.createBindGroupLayout({ entries: group }));

        const pipelineLayoutDesc = { bindGroupLayouts: bindGroups };

        this._gpuProgramData[program._layoutKey] = {
            bindGroups,
            pipeline: device.createPipelineLayout(pipelineLayoutDesc),
        };

        // generally we avoid having to make this automatically
        // keeping this for a reminder, if any issues popup
        // program._gpuLayout = {
        //     bindGroups: null,
        //     pipeline: 'auto',
        // };

        return this._gpuProgramData[program._layoutKey];
    }

    public destroy(): void
    {
        // TODO destroy the _gpuProgramData
        this._gpu = null;
        (this._gpuProgramData as null) = null;
    }
}