SwapChainWebGPU.cpp

#include "SwapChainWebGPU.h"
 
#include "Foundation/Logging/Logger.h"
 
namespace{
  Cogs::Logging::Log logger = Cogs::Logging::getLogger("SwapChainWebGPU");
}
 
namespace Cogs{
 
  SwapChainWebGPU::SwapChainWebGPU()
  {
  }
 
  SwapChainWebGPU::~SwapChainWebGPU()
  {
    // destroy(); // Done in GraphicsDevice::destroy();
  }
 
  void SwapChainWebGPU::initialize(WGPUDevice device_in, WGPUSurface surface_in)
  {
    device = device_in;
    surface = surface_in;
    create();
  }
 
  void SwapChainWebGPU::create()
  {
    WGPUTextureFormat formats[] = {
      color_format,
    };
 
    WGPUSurfaceConfiguration surfaceConf = {};
    surfaceConf.nextInChain = nullptr;
    surfaceConf.width = set_width;
    surfaceConf.height = set_height;
    surfaceConf.format = swap_chain_format;
    surfaceConf.viewFormatCount = sizeof(formats) / sizeof(WGPUTextureFormat);
    surfaceConf.viewFormats = formats;
    surfaceConf.usage = WGPUTextureUsage_RenderAttachment | WGPUTextureUsage_CopyDst;
    surfaceConf.device = device;
    surfaceConf.presentMode = WGPUPresentMode_Fifo;
    surfaceConf.alphaMode = WGPUCompositeAlphaMode_Auto;
 
    wgpuSurfaceConfigure(surface, &surfaceConf);
 
    if(samples > 1){
      WGPUTextureDescriptor desc = {};
      desc.label = {"colorBufferTexture", WGPU_STRLEN};
      desc.usage = WGPUTextureUsage_RenderAttachment;
      desc.dimension = WGPUTextureDimension_2D;
      desc.size.width = set_width;
      desc.size.height = set_height;
      desc.size.depthOrArrayLayers = 1;
      desc.format = color_format;
      desc.mipLevelCount = 1;
      desc.sampleCount = samples;
      desc.viewFormatCount = sizeof(formats)/sizeof(formats[0]);
      desc.viewFormats = formats;
 
      colorBufferTexture = wgpuDeviceCreateTexture(device, &desc);
      // colorBufferTexture = wgpuDeviceCreateErrorTexture(device, &desc);
 
      WGPUTextureViewDescriptor view_desc = {};
      view_desc.label = {"colorBufferView", WGPU_STRLEN};
      view_desc.format = color_format;
      view_desc.dimension = WGPUTextureViewDimension_2D;
      view_desc.baseMipLevel = 0;
      view_desc.mipLevelCount = 1;
      view_desc.baseArrayLayer = 0;
      view_desc.arrayLayerCount = 1;
      view_desc.aspect = WGPUTextureAspect_All;
      colorBufferView = wgpuTextureCreateView(colorBufferTexture, &view_desc);
    }
 
#ifndef EMSCRIPTEN
    {
      WGPUTextureDescriptor desc = {};
      desc.label = {"resolveTexture", WGPU_STRLEN};
      desc.usage = WGPUTextureUsage_RenderAttachment | WGPUTextureUsage_CopySrc;
      desc.dimension = WGPUTextureDimension_2D;
      desc.size.width = set_width;
      desc.size.height = set_height;
      desc.size.depthOrArrayLayers = 1;
      desc.format = color_format;
      desc.mipLevelCount = 1;
      desc.sampleCount = 1;
      desc.viewFormatCount = sizeof(formats)/sizeof(formats[0]);
      desc.viewFormats = formats;
 
      resolveTexture = wgpuDeviceCreateTexture(device, &desc);
      // resolveTexture = wgpuDeviceCreateErrorTexture(device, &desc);
 
      WGPUTextureViewDescriptor view_desc = {};
      view_desc.label = {"resolveView", WGPU_STRLEN};
      view_desc.format = color_format;
      view_desc.dimension = WGPUTextureViewDimension_2D;
      view_desc.baseMipLevel = 0;
      view_desc.mipLevelCount = 1;
      view_desc.baseArrayLayer = 0;
      view_desc.arrayLayerCount = 1;
      view_desc.aspect = WGPUTextureAspect_All;
      resolveView = wgpuTextureCreateView(resolveTexture, &view_desc);
    }
#endif
 
    {
      WGPUTextureDescriptor desc = {};
      desc.label = {"depthBufferTexture", WGPU_STRLEN};
      desc.usage = WGPUTextureUsage_RenderAttachment;
      desc.dimension = WGPUTextureDimension_2D;
      desc.size.width = set_width;
      desc.size.height = set_height;
      desc.size.depthOrArrayLayers = 1;
      desc.format = depth_format;
      desc.mipLevelCount = 1;
      desc.sampleCount = samples;
      desc.viewFormatCount = 1;
      desc.viewFormats = &depth_format;
 
      depthBufferTexture = wgpuDeviceCreateTexture(device, &desc);
      // depthBufferTexture = wgpuDeviceCreateErrorTexture(device, &desc);
 
      WGPUTextureViewDescriptor view_desc = {};
      view_desc.label = {"depthBufferView", WGPU_STRLEN};
      view_desc.format = depth_format;
      view_desc.dimension = WGPUTextureViewDimension_2D;
      view_desc.baseMipLevel = 0;
      view_desc.mipLevelCount = 1;
      view_desc.baseArrayLayer = 0;
      view_desc.arrayLayerCount = 1;
      view_desc.aspect = WGPUTextureAspect_DepthOnly;
      depthBufferView = wgpuTextureCreateView(depthBufferTexture, &view_desc);
    }
  }
 
  void SwapChainWebGPU::destroy()
  {
    if(samples > 1){
      wgpuTextureViewRelease(colorBufferView);
      wgpuTextureDestroy(colorBufferTexture);
      wgpuTextureRelease(colorBufferTexture);
    }
#ifndef EMSCRIPTEN
    wgpuTextureViewRelease(resolveView);
    wgpuTextureDestroy(resolveTexture);
    wgpuTextureRelease(resolveTexture);
#endif
    wgpuTextureViewRelease(depthBufferView);
    wgpuTextureDestroy(depthBufferTexture);
    wgpuTextureRelease(depthBufferTexture);
 
    wgpuSurfaceUnconfigure(surface);
  }
 
  void SwapChainWebGPU::beginFrame()
  {
    if(resize){
      destroy();
      create();
      resize = false;
    }
    #ifdef EMSCRIPTEN
    WGPUSurfaceTexture surfaceTexture = getTextureData();
    WGPUTextureViewDescriptor view_desc = {};
    view_desc.label = { "resolveView", WGPU_STRLEN };
    view_desc.format = color_format;
    view_desc.dimension = WGPUTextureViewDimension_2D;
    view_desc.baseMipLevel = 0;
    view_desc.mipLevelCount = 1;
    view_desc.baseArrayLayer = 0;
    view_desc.arrayLayerCount = 1;
    view_desc.aspect = WGPUTextureAspect_All;
    resolveView = wgpuTextureCreateView(surfaceTexture.texture, &view_desc);
    #endif
  }
 
  void SwapChainWebGPU::endFrame(uint32_t syncInterval, PresentFlags presentFlags)
  {
    (void)syncInterval; // TODO
    (void)presentFlags; // TODO
#ifndef EMSCRIPTEN
    wgpuSurfacePresent(surface);
#endif
  }
 
  void SwapChainWebGPU::setSize(int width, int height, bool /*forceIt*/)
  {
    LOG_DEBUG(logger, "setSize %d %d", width, height);
    set_width = width;
    set_height = height;
    resize = true;
  }
 
  void SwapChainWebGPU::setFullscreen(bool /*enabled*/)
  {
  }
 
  bool SwapChainWebGPU::isFullscreen()
  {
    return false;
  }
 
  const RenderTargetHandle& SwapChainWebGPU::getRenderTarget() const
  {
    return RenderTargetHandle::InvalidHandle;
  }
 
  const DepthStencilHandle& SwapChainWebGPU::getDepthStencil() const
  {
    return DepthStencilHandle::InvalidHandle;
  }
 
  WGPUSurfaceTexture SwapChainWebGPU::getTextureData()
  {
    WGPUSurfaceTexture surfaceTexture;
    wgpuSurfaceGetCurrentTexture(surface, &surfaceTexture);
 
    if(surfaceTexture.status == WGPUSurfaceGetCurrentTextureStatus_Error) {
      LOG_ERROR(logger, "failed to retrieve image from swapchain");
    }
 
 
    return surfaceTexture;
  }
}