Multiple output textures from the same program

前端 未结 1 954
终归单人心
终归单人心 2021-01-16 12:33

I\'m trying to learn how to do multiple outputs from the same program in WebGL2 leveraging gl.drawBuffer() capabilities.

I looked at the book \"

相关标签:
1条回答
  • 2021-01-16 12:40

    You only need one framebuffer object. You attach all the textures to it. So your steps would be

    1. Create a framebuffer object and BindFramebuffer
    2. Create output textures
    3. For each texture:
    4. Associate texture with FBO: frameBufferTexture2D(...)
    5. Create a Color attachment array GL_COLOR_ATTACHMENT0, ...
    6. call drawBuffers passing the color attachment array

    function main() {
      const gl = document.querySelector('canvas').getContext('webgl2');
      if (!gl) {
        return alert("need WebGL2");
      }
      const vs = `
      #version 300 es
      void main() {
        gl_PointSize = 300.0;
        gl_Position = vec4(0, 0, 0, 1);
      }
      `;
    
      const fs = `
      #version 300 es
      precision mediump float;
    
      layout(location = 0) out vec4 outColor0;
      layout(location = 1) out vec4 outColor1;
      layout(location = 2) out vec4 outColor2;
      layout(location = 3) out vec4 outColor3;
    
      void main() {
        outColor0 = vec4(1, .5, .3, .7);   // orange
        outColor1 = vec4(.6, .5, .4, .3);  // brown
        outColor2 = vec4(.2, .8, .0,  1);  // green
        outColor3 = vec4(.3, .4, .9, .6);  // blue
      } 
      `
    
      const program = twgl.createProgram(gl, [vs, fs]);
    
      const textures = [];
      const fb = gl.createFramebuffer();
      gl.bindFramebuffer(gl.FRAMEBUFFER, fb);
      for (let i = 0; i < 4; ++i) {
        const tex = gl.createTexture();
        textures.push(tex);
        gl.bindTexture(gl.TEXTURE_2D, tex);
        const width = 1;
        const height = 1;
        const level = 0;
        gl.texImage2D(gl.TEXTURE_2D, level, gl.RGBA, width, height, 0, 
                      gl.RGBA, gl.UNSIGNED_BYTE, null);
        // attach texture to framebuffer
        gl.framebufferTexture2D(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0 + i,
                                gl.TEXTURE_2D, tex, level);
      }
    
      // our framebuffer textures are only 1x1 pixels
      gl.viewport(0, 0, 1, 1);
    
      // tell it we want to draw to all 4 attachments
      gl.drawBuffers([
        gl.COLOR_ATTACHMENT0,
        gl.COLOR_ATTACHMENT1, 
        gl.COLOR_ATTACHMENT2,
        gl.COLOR_ATTACHMENT3,
      ]);
    
      // draw a single point
      gl.useProgram(program);
      {
        const offset = 0;
        const count = 1
        gl.drawArrays(gl.POINT, 0, 1);
      }
    
      // --- below this is not relevant to the question but just so we
      // --- we can see it's working
    
      // render the 4 textures
      const fs2 = `
      #version 300 es
      precision mediump float;
      uniform sampler2D tex[4];
      out vec4 outColor;
      void main() {
        vec4 colors[4];
    
        // you can't index textures with non-constant integer expressions
        // in WebGL2 (you can in WebGL1 lol)
        colors[0] = texture(tex[0], vec2(0));
        colors[1] = texture(tex[1], vec2(0));
        colors[2] = texture(tex[2], vec2(0));
        colors[3] = texture(tex[3], vec2(0));
        
        vec4 color = vec4(0);
        for (int i = 0; i < 4; ++i) { 
          float x = gl_PointCoord.x * 4.0;
          float amount = step(float(i), x) * step(x, float(i + 1));
          color = mix(color, colors[i], amount);
        }
        outColor = vec4(color.rgb, 1);
      }
      `;
      const prgInfo2 = twgl.createProgramInfo(gl, [vs, fs2]);
      gl.bindFramebuffer(gl.FRAMEBUFFER, null);
      gl.viewport(0, 0, gl.canvas.width, gl.canvas.height);
      gl.useProgram(prgInfo2.program);
      // binds all the textures and set the uniforms
      twgl.setUniforms(prgInfo2, {
        tex: textures,
      });
      gl.drawArrays(gl.POINTS, 0, 1);
    }
    main();
    <script src="https://twgljs.org/dist/4.x/twgl.min.js"></script>
    <canvas></canvas>

    0 讨论(0)
提交回复
热议问题