threepipe/examples/normal-buffer-plugin/script.ts

import {
    _testFinish,
    downloadBlob,
    HalfFloatType,
    NoColorSpace,
    NormalBufferPlugin,
    RenderTargetPreviewPlugin,
    SRGBColorSpace,
    ThreeViewer,
} from 'threepipe'
import {createSimpleButtons} from '../examples-utils/simple-bottom-buttons.js'

const viewer = new ThreeViewer({
    canvas: document.getElementById('mcanvas') as HTMLCanvasElement,
})

async function init() {

    const normalPlugin = viewer.addPluginSync(new NormalBufferPlugin(HalfFloatType))

    await viewer.setEnvironmentMap('https://threejs.org/examples/textures/equirectangular/venice_sunset_1k.hdr')
    await viewer.load('https://threejs.org/examples/models/gltf/kira.glb', {
        autoCenter: true,
        autoScale: true,
    })

    const normalTarget = normalPlugin.target
    if (!normalTarget) {
        throw new Error('normalPlugin.target returned undefined')
    }

    viewer.renderManager.screenPass.overrideReadBuffer = normalTarget
    viewer.renderManager.screenPass.outputColorSpace = NoColorSpace // default is SRGBColorSpace


    const targetPreview = await viewer.addPlugin(RenderTargetPreviewPlugin)
    targetPreview.addTarget(()=>normalTarget, 'normal')

    createSimpleButtons({
        ['Toggle Normal rendering']: () => {
            const state = !!viewer.renderManager.screenPass.overrideReadBuffer
            viewer.renderManager.screenPass.overrideReadBuffer = state ? null : normalTarget
            viewer.renderManager.screenPass.outputColorSpace = state ? SRGBColorSpace : NoColorSpace
            viewer.setDirty()
        },
        ['Download snapshot']: async(btn: HTMLButtonElement) => {
            btn.disabled = true
            const blob = await viewer.getScreenshotBlob({mimeType: 'image/png'})
            if (blob) downloadBlob(blob, 'file.png')
            else console.error('Unable to get screenshot')
            btn.disabled = false
        },
    })

}

init().finally(_testFinish)