Implement multiple radial distortion rendering

2021-07-10 14:04:01 +02:00
parent 1c519259a1
commit c63b293ea5
8 changed files with 75 additions and 22 deletions
@@ -142,15 +142,45 @@ doDrawing pdata w = do
    renderBlankWalls pdata windowPoints

    depthFunc $= Just Always
-    bindFramebuffer Framebuffer $= defaultFramebufferObject
-    clear [ColorBuffer]
-    activeTexture $= TextureUnit 1
-    textureBinding Texture2D $= Just (snd $ _fboBase pdata)
-    bufferUBO $ perspectiveMatrixb rot camzoom trans wins viewFroms
-    nraddist <- F.foldM (pokeShader $ _barrelShader pdata) (map Render2221 $ _radDistortion w)
-    bindShaderBuffers [_barrelShader pdata] [nraddist]
-    drawShader (_barrelShader pdata) nraddist
-    activeTexture $= TextureUnit 0
+    blendFunc $= (One,Zero)
+
+    case _radDistortion w of
+        [] -> do
+            bindTO $ snd $ _fboBase pdata
+            bindFramebuffer Framebuffer $= defaultFramebufferObject
+            --bufferUBO $ perspectiveMatrixb rot camzoom trans wins viewFroms
+            drawShader (_fullscreenShader pdata) 4
+        rds -> do
+            let bindDrawDist :: (Point2,Point2,Point2,Float) -> IO ()
+                bindDrawDist distParam = do
+                    _ <- F.foldM (pokeShader $ _barrelShader pdata) [Render2221 distParam]
+                    bindShaderBuffers [_barrelShader pdata] [1]
+                    drawShader (_barrelShader pdata) 1
+                fboList = take (length rds - 1) (concat (repeat [fst3 $ _fbo2 pdata,fst3 $ _fbo3 pdata]))
+                        ++ [defaultFramebufferObject]
+                toList = snd (_fboBase pdata) : concat (repeat [snd3 $ _fbo2 pdata,snd3 $ _fbo3 pdata])
+                bindings = zipWith (>>) (map bindFBO fboList) (map bindTO toList)
+            activeTexture $= TextureUnit 1
+            sequence_ $ zipWith (>>) bindings $ map bindDrawDist rds
+            activeTexture $= TextureUnit 0
+    --activeTexture $= TextureUnit 1
+    --bindTO $ snd $ _fboBase pdata
+
+    --let fbolist = 
+
+    --let bindDrawDist :: (Point2,Point2,Point2,Float) -> IO ()
+    --    bindDrawDist distParam = do
+    --        _ <- F.foldM (pokeShader $ _barrelShader pdata) [Render2221 distParam]
+    --        bindShaderBuffers [_barrelShader pdata] [1]
+    --        drawShader (_barrelShader pdata) 1
+    --sequence_ $ zipWith (>>) 
+    --    (map bindFBO $ concat $ repeat [fst3 $ _fbo2 pdata, fst3 $ _fbo3 pdata])
+    --    $ zipWith (>>) 
+    --        (map bindDrawDist (_radDistortion w) ++ [activeTexture $= TextureUnit 0])
+    --        (map bindTO $ concat $ repeat [snd3 $ _fbo2 pdata, snd3 $ _fbo3 pdata]) 
+    --bindFramebuffer Framebuffer $= defaultFramebufferObject
+    ----bufferUBO $ perspectiveMatrixb rot camzoom trans wins viewFroms
+    --drawShader (_fullscreenShader pdata) 4

    -- draw overlay
    bufferUBO $ isoMatrix 0 1 (0,0) (2,2)
@@ -163,6 +193,17 @@ doDrawing pdata w = do
    eTicks <- SDL.ticks
    return (eTicks - sTicks)

+bindTO :: TextureObject -> IO ()
+bindTO t = textureBinding Texture2D $= Just t
+
+bindFBO :: FramebufferObject -> IO ()
+bindFBO fb = bindFramebuffer Framebuffer $= fb
+
+fst3 :: (a,b,c) -> a
+fst3 (x,_,_) = x
+snd3 :: (a,b,c) -> b
+snd3 (_,x,_) = x
+
 --------------------------------------------------------------------------------
 -- note: currently assume there is only one UBO, we only bind it once at setup
 bufferUBO :: [Float] -> IO ()