playcanvas
Version:
PlayCanvas WebGL game engine
243 lines (240 loc) • 10.4 kB
JavaScript
import { PIXELFORMAT_RGBA8 } from '../constants.js';
import { getMultisampledTextureCache } from '../multi-sampled-texture-cache.js';
class FramebufferPair {
constructor(msaaFB, resolveFB){
this.msaaFB = msaaFB;
this.resolveFB = resolveFB;
}
destroy(gl) {
if (this.msaaFB) {
gl.deleteRenderbuffer(this.msaaFB);
this.msaaFB = null;
}
if (this.resolveFB) {
gl.deleteRenderbuffer(this.resolveFB);
this.resolveFB = null;
}
}
}
class WebglRenderTarget {
destroy(device) {
const gl = device.gl;
this._isInitialized = false;
if (this._glFrameBuffer) {
if (this._glFrameBuffer !== this.suppliedColorFramebuffer) {
gl.deleteFramebuffer(this._glFrameBuffer);
}
this._glFrameBuffer = null;
}
if (this._glDepthBuffer) {
gl.deleteRenderbuffer(this._glDepthBuffer);
this._glDepthBuffer = null;
}
if (this._glResolveFrameBuffer) {
if (this._glResolveFrameBuffer !== this.suppliedColorFramebuffer) {
gl.deleteFramebuffer(this._glResolveFrameBuffer);
}
this._glResolveFrameBuffer = null;
}
this._glMsaaColorBuffers.forEach((buffer)=>{
gl.deleteRenderbuffer(buffer);
});
this._glMsaaColorBuffers.length = 0;
this.colorMrtFramebuffers?.forEach((framebuffer)=>{
framebuffer.destroy(gl);
});
this.colorMrtFramebuffers = null;
if (this._glMsaaDepthBuffer) {
this._glMsaaDepthBuffer = null;
if (this.msaaDepthBufferKey) {
getMultisampledTextureCache(device).release(this.msaaDepthBufferKey);
}
}
this.suppliedColorFramebuffer = undefined;
}
get initialized() {
return this._isInitialized;
}
init(device, target) {
const gl = device.gl;
this._isInitialized = true;
const buffers = [];
if (this.suppliedColorFramebuffer !== undefined) {
this._glFrameBuffer = this.suppliedColorFramebuffer;
} else {
this._glFrameBuffer = gl.createFramebuffer();
device.setFramebuffer(this._glFrameBuffer);
const colorBufferCount = target._colorBuffers?.length ?? 0;
const attachmentBaseConstant = gl.COLOR_ATTACHMENT0;
for(let i = 0; i < colorBufferCount; ++i){
const colorBuffer = target.getColorBuffer(i);
if (colorBuffer) {
if (!colorBuffer.impl._glTexture) {
colorBuffer._width = Math.min(colorBuffer.width, device.maxRenderBufferSize);
colorBuffer._height = Math.min(colorBuffer.height, device.maxRenderBufferSize);
device.setTexture(colorBuffer, 0);
}
gl.framebufferTexture2D(gl.FRAMEBUFFER, attachmentBaseConstant + i, colorBuffer._cubemap ? gl.TEXTURE_CUBE_MAP_POSITIVE_X + target._face : gl.TEXTURE_2D, colorBuffer.impl._glTexture, target.mipLevel);
buffers.push(attachmentBaseConstant + i);
}
}
gl.drawBuffers(buffers);
const depthBuffer = target._depthBuffer;
if (depthBuffer || target._depth) {
const attachmentPoint = target._stencil ? gl.DEPTH_STENCIL_ATTACHMENT : gl.DEPTH_ATTACHMENT;
if (depthBuffer) {
if (!depthBuffer.impl._glTexture) {
depthBuffer._width = Math.min(depthBuffer.width, device.maxRenderBufferSize);
depthBuffer._height = Math.min(depthBuffer.height, device.maxRenderBufferSize);
device.setTexture(depthBuffer, 0);
}
gl.framebufferTexture2D(gl.FRAMEBUFFER, attachmentPoint, depthBuffer._cubemap ? gl.TEXTURE_CUBE_MAP_POSITIVE_X + target._face : gl.TEXTURE_2D, target._depthBuffer.impl._glTexture, target.mipLevel);
} else {
const willRenderMsaa = target._samples > 1;
if (!willRenderMsaa) {
if (!this._glDepthBuffer) {
this._glDepthBuffer = gl.createRenderbuffer();
}
const internalFormat = target._stencil ? gl.DEPTH24_STENCIL8 : gl.DEPTH_COMPONENT32F;
gl.bindRenderbuffer(gl.RENDERBUFFER, this._glDepthBuffer);
gl.renderbufferStorage(gl.RENDERBUFFER, internalFormat, target.width, target.height);
gl.framebufferRenderbuffer(gl.FRAMEBUFFER, attachmentPoint, gl.RENDERBUFFER, this._glDepthBuffer);
gl.bindRenderbuffer(gl.RENDERBUFFER, null);
}
}
}
}
if (target._samples > 1) {
this._glResolveFrameBuffer = this._glFrameBuffer;
this._glFrameBuffer = gl.createFramebuffer();
device.setFramebuffer(this._glFrameBuffer);
const colorBufferCount = target._colorBuffers?.length ?? 0;
if (this.suppliedColorFramebuffer !== undefined) {
const buffer = gl.createRenderbuffer();
this._glMsaaColorBuffers.push(buffer);
const internalFormat = device.backBufferFormat === PIXELFORMAT_RGBA8 ? gl.RGBA8 : gl.RGB8;
gl.bindRenderbuffer(gl.RENDERBUFFER, buffer);
gl.renderbufferStorageMultisample(gl.RENDERBUFFER, target._samples, internalFormat, target.width, target.height);
gl.framebufferRenderbuffer(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0, gl.RENDERBUFFER, buffer);
} else {
for(let i = 0; i < colorBufferCount; ++i){
const colorBuffer = target.getColorBuffer(i);
if (colorBuffer) {
const buffer = gl.createRenderbuffer();
this._glMsaaColorBuffers.push(buffer);
gl.bindRenderbuffer(gl.RENDERBUFFER, buffer);
gl.renderbufferStorageMultisample(gl.RENDERBUFFER, target._samples, colorBuffer.impl._glInternalFormat, target.width, target.height);
gl.framebufferRenderbuffer(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0 + i, gl.RENDERBUFFER, buffer);
}
}
}
if (target._depth) {
const internalFormat = target._stencil ? gl.DEPTH24_STENCIL8 : gl.DEPTH_COMPONENT32F;
const attachmentPoint = target._stencil ? gl.DEPTH_STENCIL_ATTACHMENT : gl.DEPTH_ATTACHMENT;
let key;
const depthBuffer = target._depthBuffer;
if (depthBuffer) {
key = `${depthBuffer.id}:${target.width}:${target.height}:${target._samples}:${internalFormat}:${attachmentPoint}`;
this._glMsaaDepthBuffer = getMultisampledTextureCache(device).get(key);
}
if (!this._glMsaaDepthBuffer) {
this._glMsaaDepthBuffer = gl.createRenderbuffer();
gl.bindRenderbuffer(gl.RENDERBUFFER, this._glMsaaDepthBuffer);
gl.renderbufferStorageMultisample(gl.RENDERBUFFER, target._samples, internalFormat, target.width, target.height);
this._glMsaaDepthBuffer.destroy = function() {
gl.deleteRenderbuffer(this);
};
if (depthBuffer) {
getMultisampledTextureCache(device).set(key, this._glMsaaDepthBuffer);
}
}
this.msaaDepthBufferKey = key;
gl.framebufferRenderbuffer(gl.FRAMEBUFFER, attachmentPoint, gl.RENDERBUFFER, this._glMsaaDepthBuffer);
}
if (colorBufferCount > 1) {
this._createMsaaMrtFramebuffers(device, target, colorBufferCount);
device.setFramebuffer(this._glFrameBuffer);
gl.drawBuffers(buffers);
}
}
}
_createMsaaMrtFramebuffers(device, target, colorBufferCount) {
const gl = device.gl;
this.colorMrtFramebuffers = [];
for(let i = 0; i < colorBufferCount; ++i){
const colorBuffer = target.getColorBuffer(i);
const srcFramebuffer = gl.createFramebuffer();
device.setFramebuffer(srcFramebuffer);
const buffer = this._glMsaaColorBuffers[i];
gl.bindRenderbuffer(gl.RENDERBUFFER, buffer);
gl.renderbufferStorageMultisample(gl.RENDERBUFFER, target._samples, colorBuffer.impl._glInternalFormat, target.width, target.height);
gl.framebufferRenderbuffer(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0, gl.RENDERBUFFER, buffer);
gl.drawBuffers([
gl.COLOR_ATTACHMENT0
]);
const dstFramebuffer = gl.createFramebuffer();
device.setFramebuffer(dstFramebuffer);
gl.framebufferTexture2D(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0, colorBuffer._cubemap ? gl.TEXTURE_CUBE_MAP_POSITIVE_X + target._face : gl.TEXTURE_2D, colorBuffer.impl._glTexture, 0);
this.colorMrtFramebuffers[i] = new FramebufferPair(srcFramebuffer, dstFramebuffer);
}
}
_checkFbo(device, target, type = '') {
const gl = device.gl;
const status = gl.checkFramebufferStatus(gl.FRAMEBUFFER);
switch(status){
case gl.FRAMEBUFFER_INCOMPLETE_ATTACHMENT:
break;
case gl.FRAMEBUFFER_INCOMPLETE_MISSING_ATTACHMENT:
break;
case gl.FRAMEBUFFER_INCOMPLETE_DIMENSIONS:
break;
case gl.FRAMEBUFFER_UNSUPPORTED:
break;
}
}
loseContext() {
this._glFrameBuffer = null;
this._glDepthBuffer = null;
this._glResolveFrameBuffer = null;
this._glMsaaColorBuffers.length = 0;
this._glMsaaDepthBuffer = null;
this.msaaDepthBufferKey = undefined;
this.colorMrtFramebuffers = null;
this.suppliedColorFramebuffer = undefined;
this._isInitialized = false;
}
internalResolve(device, src, dst, target, mask) {
device.setScissor(0, 0, target.width, target.height);
const gl = device.gl;
gl.bindFramebuffer(gl.READ_FRAMEBUFFER, src);
gl.bindFramebuffer(gl.DRAW_FRAMEBUFFER, dst);
gl.blitFramebuffer(0, 0, target.width, target.height, 0, 0, target.width, target.height, mask, gl.NEAREST);
}
resolve(device, target, color, depth) {
const gl = device.gl;
if (this.colorMrtFramebuffers) {
if (color) {
for(let i = 0; i < this.colorMrtFramebuffers.length; i++){
const fbPair = this.colorMrtFramebuffers[i];
this.internalResolve(device, fbPair.msaaFB, fbPair.resolveFB, target, gl.COLOR_BUFFER_BIT);
}
}
if (depth) {
this.internalResolve(device, this._glFrameBuffer, this._glResolveFrameBuffer, target, gl.DEPTH_BUFFER_BIT);
}
} else {
this.internalResolve(device, this._glFrameBuffer, this._glResolveFrameBuffer, target, (color ? gl.COLOR_BUFFER_BIT : 0) | (depth ? gl.DEPTH_BUFFER_BIT : 0));
}
gl.bindFramebuffer(gl.FRAMEBUFFER, this._glFrameBuffer);
}
constructor(){
this._glFrameBuffer = null;
this._glDepthBuffer = null;
this._glResolveFrameBuffer = null;
this.colorMrtFramebuffers = null;
this._glMsaaColorBuffers = [];
this._glMsaaDepthBuffer = null;
this._isInitialized = false;
}
}
export { WebglRenderTarget };