SuperStrict

Framework sedm.simpleirr

Type TPostProcessing_SetShaderConstants Extends IShaderConstantSetCallBack
   Global shaderparameters:Float[8]
   
   Method OnSetConstants(services:IMaterialRendererServices, userdata:Int)
      services.setPixelShaderConstantFromName("vecValues", Self.shaderparameters, 8)
   
      If userdata = 1
         'Set Textures For openGL Shaders
         Local texture1:Float = 0.0
         Local texture2:Float = 0.0
         services.setPixelShaderConstantFromName("texture1", VarPtr(texture1), 1)
         services.setPixelShaderConstantFromName("texture2", VarPtr(texture2), 1)
      EndIf
   End Method
   
   Function setShaderParameters(paras:Float[])
      shaderparameters = paras
   End Function
   
   Function generate:IShaderConstantSetCallBack()
      Return New TPostProcessing_SetShaderConstants
   EndFunction
End Type

rem
   Class which manages postprocessing effects
   
   To apply the effect, run
   driver->setRenderTarget(postprocessing->getFirstMap(), true, true, video::SColor(255,150,180,255));
   
   or similar before you render anything and run
   
   driver->setRenderTarget(0);
   postprocessing->renderEffect();
   
   to actually run the shaders and render the result to the screen.
end rem
Type TPostProcessing
   Field shadercallback:TPostProcessing_SetShaderConstants
   Field scenemgr:ISceneManager
   Field Driver:IVideoDriver
   Field vertices:Array_S3DVertex
   Field indices:Array_u16
   Field material:SMaterial
   Field matid:Int
   Field shaderparameters:Float[8]
   Field prevstage:TPostProcessing
   Field nextstage:TPostProcessing
   Field firstmap:ITexture
   Field secondmap:ITexture
   Field gpu:IGPUProgrammingServices
   
   rem
      Constructor
      smgr      : Scene manager which is used for the post progressing
      filename_gl   : Path to the GLSL script
      filename_dx   : Path to the HLSL script
      type_ps      : Type of the pixel shader
      res_x      : Horizontal resolution of the used texture
      res_y      : Vertical resolution of the used texture
   end rem
   Function PostProcessing:TPostProcessing(smgr:ISceneManager, filename_gl:String, filename_dx:String, type_ps:Int, res_x:Int, res_y:Int)
      Local pp:TPostProcessing = New TPostProcessing
      
      pp.Driver         = smgr.getVideoDriver()
      pp.shadercallback   = TPostProcessing_SetShaderConstants(IShaderConstantSetCallBack.Create(TPostProcessing_SetShaderConstants.generate))
      pp.vertices         = Array_S3DVertex.createWithSize(4)
      pp.indices         = Array_u16.createWithSize(6)
      pp.material         = SMaterial.Create()
      
      If pp.Driver.getDriverType() = EDT_OPENGL Or pp.Driver.getDriverType() = EDT_DIRECT3D9
         pp.vertices.Insert(S3DVertexDefault.createFromVals(-1.0, -1.0, 0.0, 1.0, 1.0, 0.0, _SCOLOR(0,0,0,0), 0.0, 1.0), 0)
         pp.vertices.Insert(S3DVertexDefault.createFromVals(-1.0,  1.0, 0.0, 1.0, 1.0, 0.0, _SCOLOR(0,0,0,0), 0.0, 0.0), 1)
         pp.vertices.Insert(S3DVertexDefault.createFromVals( 1.0,  1.0, 0.0, 1.0, 1.0, 0.0, _SCOLOR(0,0,0,0), 1.0, 0.0), 2)
         pp.vertices.Insert(S3DVertexDefault.createFromVals( 1.0, -1.0, 0.0, 1.0, 1.0, 0.0, _SCOLOR(0,0,0,0), 1.0, 1.0), 3)
         
         pp.indices.Insert(0, 0)
         pp.indices.Insert(1, 1)
         pp.indices.Insert(2, 2)
         pp.indices.Insert(0, 3)
         pp.indices.Insert(2, 4)
         pp.indices.Insert(3, 5)
         
         pp.gpu      = pp.Driver.getGPUProgrammingServices()
         pp.scenemgr   = smgr

         pp.prevstage = Null
         pp.nextstage = Null
         
         Local para:Float[] = [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]
         pp.setShaderParameters(para)
         pp.shadercallback.setShaderParameters(pp.shaderparameters)

         If pp.Driver.getDriverType() = EDT_OPENGL
            pp.matid = pp.gpu.addHighLevelShaderMaterialFromFiles( ..
               "Shaders/PP_GL_Vertex.fx", ..
               "main", ..
               EVST_VS_1_1, ..
               filename_gl, ..
               "main", ..
               type_ps, ..
               pp.shadercallback, ..
               EMT_SOLID, ..
               1)
         Else
            pp.matid = pp.gpu.addHighLevelShaderMaterialFromFiles( ..
               "Shaders/PP_DX_Vertex.fx", ..
               "main", ..
               EVST_VS_1_1, ..
               filename_dx, ..
               "main", ..
               type_ps, ..
               pp.shadercallback, ..
               EMT_SOLID, ..
               0)
         EndIf
   
         pp.firstmap = pp.Driver.addRenderTargetTexture(_DIMENSION2DI(res_x, res_y))
         pp.secondmap = Null
         pp.material.setWireframe(False)
         pp.material.setLighting(False)
         pp.material.setTexture(0,pp.firstmap);
         'material.TextureLayer[0].TextureWrap = video::ETC_CLAMP;
         pp.material.setMaterialType(pp.matid)
      EndIf
      
      Return pp
   End Function

   rem
      Destructor
   endrem
   Method Delete()
      Self.free()
   End Method
   
   Method free()
      Self.shadercallback = Null
      Self.scenemgr.Deconstructor()
      Self.Driver.Deconstructor()
      Self.vertices.Deconstructor()
      Self.material.Deconstructor()
      Self.shaderparameters = Null
      Self.firstmap.Deconstructor()
      Self.secondmap.Deconstructor()
      Self.gpu.Deconstructor()
      If Self.prevstage <> Null
         Self.prevstage.free()
         Self.prevstage = Null
      End If
      If Self.nextstage <> Null
         Self.nextstage.free()
         Self.nextstage = Null
      EndIf
   End Method

   rem
      Adds another stage and inserts it after this stage
      filename_gl Path to the GLSL script
      filename_dx Path to the HLSL script
      type_ps Type of the pixel shader
      res_x Horizontal resolution of the used texture
      res_y Vertical resolution of the used texture
   end rem
   Method addMaterial:TPostProcessing(filename_gl:String, filename_dx:String, type_ps:Int, res_x:Int, res_y:Int)
      Self.nextstage = TPostProcessing.PostProcessing(Self.scenemgr, filename_gl, filename_dx, type_ps, res_x, res_y)
      Return nextstage
   End Method

   rem
      Renders this postprocessing chain
   end rem
   Method renderEffect()
      Self.Driver.setMaterial(Self.material)

      If Self.nextstage <> Null
         Self.Driver.setRenderTarget(Self.nextstage.getFirstMap(), True, True, _SCOLOR(255,150,180,255))
         Self.Driver.drawIndexedTriangleList(Self.vertices, Self.indices)
         Self.Driver.setRenderTarget()
         Self.nextstage.renderEffect()
      Else
         Self.Driver.setMaterial(Self.material)
         Self.Driver.drawIndexedTriangleList(Self.vertices, Self.indices)
      EndIf
   End Method

   rem
      Sets the second texture
   end rem
   Method setSecondMap(tex:ITexture, mode:Int = ETC_CLAMP)
      Self.secondmap = tex
      Self.material.setTexture(1, Self.secondmap)
      'self.material.TextureLayer[1].TextureWrap = mode
   End Method

   rem
      Sets the parameters of the shader
   endrem
   Method setShaderParameters(para:Float[])
      Self.shaderparameters = para
   End Method

   rem
      Returns a pointer to the material
   end rem
   Method getMaterial:SMaterial()
      Return Self.material
   End Method

   rem
      Returns a pointer to the first texture
   end rem
   Method getFirstMap:ITexture()
      Return Self.firstmap
   End Method
   
   rem
      Returns a pointer to the second texture
   end rem
   Method getSecondMap:ITexture()
      Return Self.secondmap
   End Method
End Type

Local irr:TIrrlicht = TIrrlicht.Init()
irr.Graphics3D(1024, 768)

Local para:Float[8]
Local PP_Test:TPostProcessing = TPostProcessing.PostProcessing(irr.irr_scene, "Shaders/PP_GL_Bloom1.fx","Shaders/PP_DX_Bloom1.fx", EPST_PS_1_4, 512,256)
para[0] = 0.5; PP_Test.setShaderParameters(para)

Local Test2:TPostProcessing = PP_test.addMaterial("Shaders/PP_GL_Bloom2.fx", "Shaders/PP_DX_Bloom2.fx", EPST_PS_2_0, 128, 128)
para[0] = 0.01; Test2.setShaderParameters(para)

Test2 = Test2.addMaterial("Shaders/PP_GL_Bloom3.fx", "Shaders/PP_DX_Bloom3.fx", EPST_PS_2_0, 128, 128)
Test2.setShaderParameters(para)

Test2 = Test2.addMaterial("Shaders/PP_GL_Bloom4.fx", "Shaders/PP_DX_Bloom4.fx", EPST_PS_2_0, 512, 256)
para[0] = 0.7; Test2.setShaderParameters(para)

Test2.setSecondMap(PP_Test.getFirstMap(), ETC_CLAMP)

irr.AddZip("data/data.zip")

Local scene:TScene = TScene.Load("scene.irr")

Local sky:ISceneNode = irr.irr_scene.addSkyBoxSceneNode(irr.irr_video.getTexture("sky/morning_up.jpg"), ..
                                          irr.irr_video.getTexture("sky/morning_down.jpg"), ..
                                          irr.irr_video.getTexture("sky/morning_west.jpg"), ..
                                          irr.irr_video.getTexture("sky/morning_east.jpg"), ..
                                          irr.irr_video.getTexture("sky/morning_north.jpg"), ..
                                          irr.irr_video.getTexture("sky/morning_south.jpg"))
                                          
' die Würfel aus der Scene holen
Local cb:ISceneNode = TScene.FindByName("Cube")
Local ce:TEntity = New TEntity
ce.node=cb
cb.getMaterial(0).setShininess(200.0)

Local cam:ISceneNode = ICameraSceneNode(TScene.FindByName("Camera"))
If Not cam Then RuntimeError "no camera found"
cam.remove()

Local cam2:ICameraSceneNode = irr.irr_scene.addCameraSceneNodeFPS(Null, 100.0, 0.05)
irr.irr_scene.setActiveCamera(cam2)
cam2.setPosition(_VECTOR3DF(0.0, 3.0, -10.0))

'the main loop
Repeat
   If irr.irr_device.run()
      irr.irr_video.BeginScene(True, True, _SCOLOR(255,150,180,255))
            
      'render the scene into the postprocessing texture
      irr.irr_video.setRenderTarget(PP_Test.getFirstMap(), True, True, _SCOLOR(255,150,180,255))
      irr.irr_scene.drawAll()
      irr.irr_video.setRenderTarget()
         
      'render the effect chain
      PP_Test.renderEffect()

      irr.irr_video.EndScene()
   EndIf
Until irr.IsKeyDown(KEY_ESCAPE)
End

      'render the scene into the postprocessing texture
      irr.irr_video.setRenderTarget(PP_Test.getFirstMap(), True, True, _SCOLOR(255,150,180,255))
      irr.irr_scene.drawAll()