WebGPUUtils.js 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. import { GPUPrimitiveTopology, GPUTextureFormat } from './WebGPUConstants.js';
  2. class WebGPUUtils {
  3. constructor( backend ) {
  4. this.backend = backend;
  5. }
  6. getCurrentDepthStencilFormat( renderContext ) {
  7. let format;
  8. if ( renderContext.depthTexture !== null ) {
  9. format = this.getTextureFormatGPU( renderContext.depthTexture );
  10. } else if ( renderContext.depth && renderContext.stencil ) {
  11. format = GPUTextureFormat.Depth24PlusStencil8;
  12. } else if ( renderContext.depth ) {
  13. format = GPUTextureFormat.Depth24Plus;
  14. }
  15. return format;
  16. }
  17. getTextureFormatGPU( texture ) {
  18. return this.backend.get( texture ).texture.format;
  19. }
  20. getCurrentColorFormat( renderContext ) {
  21. let format;
  22. if ( renderContext.texture !== null ) {
  23. format = this.getTextureFormatGPU( renderContext.texture );
  24. } else {
  25. format = GPUTextureFormat.BGRA8Unorm; // default context format
  26. }
  27. return format;
  28. }
  29. getCurrentColorSpace( renderContext ) {
  30. if ( renderContext.texture !== null ) {
  31. return renderContext.texture.colorSpace;
  32. }
  33. return this.backend.renderer.outputColorSpace;
  34. }
  35. getPrimitiveTopology( object, material ) {
  36. if ( object.isPoints ) return GPUPrimitiveTopology.PointList;
  37. else if ( object.isLineSegments || ( object.isMesh && material.wireframe === true ) ) return GPUPrimitiveTopology.LineList;
  38. else if ( object.isLine ) return GPUPrimitiveTopology.LineStrip;
  39. else if ( object.isMesh ) return GPUPrimitiveTopology.TriangleList;
  40. }
  41. getSampleCount( renderContext ) {
  42. if ( renderContext.texture !== null ) {
  43. return 1;
  44. }
  45. return this.backend.parameters.sampleCount;
  46. }
  47. }
  48. export default WebGPUUtils;