new post | browse code | authors | help | about

LuaBin 2.0

Viewing file root / lua / postprocess / bloom.lua

  1.  
  2.  
  3. local mat_Downsample    = Material( "pp/downsample" )
  4. local mat_Bloom                 = Material( "pp/bloom" )
  5. local mat_BlurX                 = Material( "pp/blurx" )
  6. local mat_BlurY                 = Material( "pp/blury" )
  7.  
  8. local tex_Bloom0                = render.GetBloomTex0()
  9. local tex_Bloom1                = render.GetBloomTex1()
  10.  
  11. mat_Downsample:SetMaterialTexture( "$fbtexture", render.GetScreenEffectTexture() )
  12.  
  13. /*---------------------------------------------------------
  14.    Register the convars that will control this effect
  15. ---------------------------------------------------------*/  
  16. local pp_bloom                  = CreateClientConVar( "pp_bloom",                       "0",    true,   false )                         // On/Off
  17. local pp_bloom_darken   = CreateClientConVar( "pp_bloom_darken",        "0.65", false,  false )         // Decides the strength of the bloom
  18. local pp_bloom_multiply = CreateClientConVar( "pp_bloom_multiply",      "2.0",  false,  false ) // Decides the strength of the bloom
  19. local pp_bloom_sizex    = CreateClientConVar( "pp_bloom_sizex",         "9.0",  false,  false )         // Horizontal blur size
  20. local pp_bloom_sizey    = CreateClientConVar( "pp_bloom_sizey",         "9.0",  false,  false )         // Vertical blur size
  21. local pp_bloom_color    = CreateClientConVar( "pp_bloom_color",         "1.0",  false,  false )
  22. local pp_bloom_color_r  = CreateClientConVar( "pp_bloom_color_r",       "255",  false,  false )
  23. local pp_bloom_color_g  = CreateClientConVar( "pp_bloom_color_g",       "255",  false,  false )
  24. local pp_bloom_color_b  = CreateClientConVar( "pp_bloom_color_b",       "255",  false,  false )
  25. local pp_bloom_passes   = CreateClientConVar( "pp_bloom_passes",        "1",    false,  false )
  26.  
  27. /*---------------------------------------------------------
  28.    Can be called from engine or hooks using bloom.Draw
  29. ---------------------------------------------------------*/
  30. function DrawBloom( darken, multiply, sizex, sizey, passes, color, colr, colg, colb )
  31.  
  32.         // No bloom for crappy gpus
  33.         if ( !render.SupportsPixelShaders_2_0() ) then return end
  34.        
  35.         // Todo: Does this stop it working properly when rendered to a rect?
  36.         local w = ScrW()
  37.         local h = ScrH()
  38.        
  39.         // Copy the backbuffer to the screen effect texture
  40.         render.UpdateScreenEffectTexture()
  41.        
  42.         // Store the render target so we can swap back at the end
  43.         local OldRT = render.GetRenderTarget();
  44.        
  45.         // The downsample material adjusts the contrast
  46.         mat_Downsample:SetMaterialFloat( "$darken", darken )
  47.         mat_Downsample:SetMaterialFloat( "$multiply", multiply  )
  48.        
  49.                
  50.         // Downsample to BloomTexture0
  51.         render.SetRenderTarget( tex_Bloom0 )
  52.        
  53.         render.SetMaterial( mat_Downsample )
  54.         render.DrawScreenQuad()                  
  55.                                          
  56.         mat_BlurX:SetMaterialTexture( "$basetexture", tex_Bloom0 )
  57.         mat_BlurY:SetMaterialTexture( "$basetexture", tex_Bloom1  )
  58.         mat_BlurX:SetMaterialFloat( "$size", sizex )
  59.         mat_BlurY:SetMaterialFloat( "$size", sizey )
  60.        
  61.         for i=1, passes do
  62.  
  63.                 render.SetRenderTarget( tex_Bloom1 )
  64.                 render.SetMaterial( mat_BlurX )
  65.                 render.DrawScreenQuad()
  66.  
  67.                 render.SetRenderTarget( tex_Bloom0 )
  68.                 render.SetMaterial( mat_BlurY )
  69.                 render.DrawScreenQuad()
  70.  
  71.         end
  72.                          
  73.         render.SetRenderTarget( OldRT );
  74.        
  75.         mat_Bloom:SetMaterialFloat( "$levelr", colr )
  76.         mat_Bloom:SetMaterialFloat( "$levelg", colg )
  77.         mat_Bloom:SetMaterialFloat( "$levelb", colb )
  78.         mat_Bloom:SetMaterialFloat( "$colormul", color )
  79.         mat_Bloom:SetMaterialTexture( "$basetexture", tex_Bloom0 )
  80.        
  81.         render.SetMaterial( mat_Bloom )
  82.         render.DrawScreenQuad()
  83.        
  84. end
  85.  
  86.  
  87. /*---------------------------------------------------------
  88.    The function to draw the bloom (called from the hook)
  89. ---------------------------------------------------------*/
  90. local function DrawInternal()
  91.  
  92.         // No bloom for crappy gpus
  93.        
  94.         if ( !render.SupportsPixelShaders_2_0() ) then return end      
  95.         if ( !pp_bloom:GetBool() ) then return end
  96.         if ( !GAMEMODE:PostProcessPermitted( "bloom" ) ) then return end
  97.        
  98.         DrawBloom( pp_bloom_darken:GetFloat(),
  99.                                 pp_bloom_multiply:GetFloat(),
  100.                                 pp_bloom_sizex:GetFloat(),
  101.                                 pp_bloom_sizey:GetFloat(),
  102.                                 pp_bloom_passes:GetFloat(),
  103.                                 pp_bloom_color:GetFloat(),
  104.                                 pp_bloom_color_r:GetFloat()/255,
  105.                                 pp_bloom_color_g:GetFloat()/255,
  106.                                 pp_bloom_color_b:GetFloat()/255 )
  107.        
  108.  
  109. end
  110.  
  111.  
  112.  
  113. hook.Add( "RenderScreenspaceEffects", "RenderBloom", DrawInternal )
  114.  
  115.  
  116.  
  117. /*
  118. // Control Panel
  119. */
  120. local function BuildControlPanel( CPanel )
  121.  
  122.         CPanel:AddControl( "Header", { Text = "#Bloom", Description = "#Bloom_Information" }  )
  123.         CPanel:AddControl( "CheckBox", { Label = "#Bloom_Toggle", Command = "pp_bloom" }  )
  124.        
  125.         local params = { Options = {}, CVars = {}, Label = "#Presets", MenuButton = "1", Folder = "bloom" }
  126.         params.Options[ "#Default" ] = {pp_bloom_passes =               "1",
  127.                                                                         pp_bloom_darken =               "0.65",
  128.                                                                         pp_bloom_multiply =             "2.0",
  129.                                                                         pp_bloom_sizex =                "9",
  130.                                                                         pp_bloom_sizey =                "9",
  131.                                                                         pp_bloom_color =                "1.0",
  132.                                                                         pp_bloom_color_r =              "255",
  133.                                                                         pp_bloom_color_g =              "255",
  134.                                                                         pp_bloom_color_b =              "255" }
  135.                                                                        
  136.         params.CVars = { "pp_bloom_passes",
  137.                                                 "pp_bloom_darken",
  138.                                                 "pp_bloom_multiply",
  139.                                                 "pp_bloom_sizex",
  140.                                                 "pp_bloom_sizey",
  141.                                                 "pp_bloom_color",
  142.                                                 "pp_bloom_color_r",
  143.                                                 "pp_bloom_color_g",
  144.                                                 "pp_bloom_color_b" }
  145.                                                
  146.         CPanel:AddControl( "ComboBox",  params )
  147.        
  148.         CPanel:AddControl( "Slider", { Label = "#Bloom_Passes", Command = "pp_bloom_passes", Type = "Integer", Min = "0", Max = "30" }  )
  149.         CPanel:AddControl( "Slider", { Label = "#Bloom_Darken", Command = "pp_bloom_darken", Type = "Float", Min = "0", Max = "1" }  )
  150.         CPanel:AddControl( "Slider", { Label = "#Bloom_Multiply", Command = "pp_bloom_multiply", Type = "Float", Min = "0", Max = "5" }  )     
  151.         CPanel:AddControl( "Slider", { Label = "#Bloom_BlurX", Command = "pp_bloom_sizex", Type = "Float", Min = "0", Max = "50" }  )  
  152.         CPanel:AddControl( "Slider", { Label = "#Bloom_BlurY", Command = "pp_bloom_sizey", Type = "Float", Min = "0", Max = "50" }  )  
  153.         CPanel:AddControl( "Slider", { Label = "#Bloom_Color_Multiplier", Command = "pp_bloom_color", Type = "Float", Min = "0", Max = "20" }  )       
  154.  
  155.         CPanel:AddControl( "Color", { Label = "#Bloom_Color", Red = "pp_bloom_color_r", Green = "pp_bloom_color_g", Blue = "pp_bloom_color_b", ShowAlpha = "0", ShowHSV = "1", ShowRGB = "1" }  )                      
  156.        
  157. end
  158.  
  159. /*
  160. // Tool Menu
  161. */
  162. local function AddPostProcessMenu()
  163.  
  164.         spawnmenu.AddToolMenuOption( "PostProcessing", "PPShader", "Bloom", "#Bloom", "", "", BuildControlPanel, { SwitchConVar = "pp_bloom" } )
  165.  
  166. end
  167.  
  168. hook.Add( "PopulateToolMenu", "AddPostProcessMenu_Bloom", AddPostProcessMenu )
  169.