new post | browse code | authors | help | about

LuaBin 2.0

Viewing file root / lua / postprocess / super_dof.lua

  1.  
  2. PANEL = {}
  3.  
  4. local Distance                  = 256
  5. local BlurSize                  = 0.5
  6. local pnlWindow                 = nil
  7.  
  8. local Window                    = nil
  9. local Status                    = "Preview"
  10.  
  11.  
  12. local sldDistance       = nil
  13. local lblDistance               = nil
  14. local lblSize                   = nil
  15. local FocusGrabber      = false
  16. local ScreenshotTimer   = 0
  17.  
  18. local strTitle          = Localize( "SuperDOF_WindowTitle", "Super DOF" )
  19. local strBlurSize       = Localize( "SuperDOF_BlurSize", "Blur Size:" )
  20. local strDistance       = Localize( "SuperDOF_Distance", "Focus Distance:" )
  21. local strRender         = Localize( "SuperDOF_Render", "Render" )
  22. local strScreenshot = Localize( "SuperDOF_Screenshot", "Save Screenshot" )
  23. local strOpenWindow = Localize( "SuperDoF_Open", "Open Window" )
  24. local strInformation =  Localize( "SuperDoF_Warning", "Warning: This is VERY experimental so it might not totally work on your graphics card. \n\nThis effect is not realtime. You render it and then save a screenshot of your render. Also, you will have low fps when previewing - that's normal.")
  25.  
  26.  
  27. function PANEL:Init()
  28.  
  29.         self:SetTitle( strTitle )
  30.         self:SetRenderInScreenshots( false )
  31.  
  32.         self.BlurSize = vgui.Create( "DNumSlider", self )
  33.                 self.BlurSize:SetMin( 0 )
  34.                 self.BlurSize:SetMax( 10 )
  35.                 self.BlurSize:SetDecimals( 3 )
  36.                 self.BlurSize:SetText( strBlurSize )
  37.                 self.BlurSize:SetValue( BlurSize )
  38.                 function self.BlurSize:OnValueChanged( val ) BlurSize = val end
  39.        
  40.         self.Distance = vgui.Create( "DNumSlider", self )
  41.                 self.Distance:SetMin( 0 )
  42.                 self.Distance:SetMax( 4096 )
  43.                 self.Distance:SetText( strDistance )
  44.                 self.Distance:SetValue( Distance )
  45.                 function self.Distance:OnValueChanged( val ) Distance = val end
  46.                
  47.  
  48.         self.Render = vgui.Create( "DButton", self )
  49.                 self.Render:SetText( strRender )
  50.                 function self.Render:DoClick() Status = "Render" end
  51.                
  52.         self.Screenshot = vgui.Create( "DButton", self )
  53.                 self.Screenshot:SetText( strScreenshot )
  54.                 function self.Screenshot:DoClick() RunConsoleCommand( "jpeg" ) end
  55.  
  56. end
  57.  
  58. function PANEL:ChangeDistanceTo( dist )
  59.  
  60.         self.Distance:SetValue( dist )
  61.  
  62. end
  63.  
  64. function PANEL:PerformLayout()
  65.  
  66.         self:SetSize( 300, 180 )
  67.        
  68.         DFrame.PerformLayout( self )
  69.        
  70.         self.BlurSize:SetPos( 5, 35 )
  71.         self.BlurSize:StretchToParent( 20, nil, 20, nil )
  72.         self.BlurSize:SetTall( 45 )
  73.        
  74.         self.Distance:StretchToParent( 20, nil, 20, nil )
  75.         self.Distance:SetTall( 45 )    
  76.         self.Distance:MoveBelow( self.BlurSize, 5 )
  77.        
  78.         self.Render:SetWide( 100 )
  79.         self.Render:MoveBelow( self.Distance, 10 )
  80.         self.Render:AlignRight( 20 )
  81.        
  82.         self.Screenshot:CopyBounds( self.Render )
  83.         self.Screenshot:MoveLeftOf( self.Render, 10 )
  84.  
  85. end
  86.  
  87. local paneltypeSuperDOF = vgui.RegisterTable( PANEL, "DFrame" )
  88.  
  89.  
  90. local texFSB = render.GetSuperFPTex()
  91. local matFSB = Material( "pp/motionblur" )
  92. local matFB      = Material( "pp/fb" )
  93.  
  94. local function RenderDoF( vOrigin, vAngle, vFocus, fAngleSize, radial_steps, passes, drawhud )
  95.  
  96.         local OldRT     = render.GetRenderTarget();
  97.         local view              = {  x = 0, y = 0, w = ScrW(), h = ScrH(), drawhud = drawhud }
  98.         local fDistance = vOrigin:Distance( vFocus )
  99.        
  100.         fAngleSize = fAngleSize * math.Clamp( 256/fDistance, 0.1, 1 ) * 0.5
  101.        
  102.         view.origin = vOrigin
  103.         view.angles = vAngle
  104.        
  105.         // Straight render (to act as a canvas)
  106.         render.RenderView( view )
  107.         render.UpdateScreenEffectTexture()
  108.        
  109.         render.SetRenderTarget( texFSB )
  110.                         matFB:SetMaterialFloat( "$alpha", 1  )
  111.                         render.SetMaterial( matFB )
  112.                         render.DrawScreenQuad()
  113.        
  114.         local Radials = (math.pi*2) / radial_steps
  115.        
  116.         for mul=(1 / passes), 1, (1 / passes) do
  117.        
  118.                 for i=0,(math.pi*2), Radials do
  119.                
  120.                         local VA = vAngle * 1 // hack - this makes it copy the angles instead of the reference
  121.                        
  122.                         // Rotate around the focus point
  123.                         VA:RotateAroundAxis( vAngle:Right(),    math.sin( i + (mul) ) * fAngleSize * mul )
  124.                         VA:RotateAroundAxis( vAngle:Up(),               math.cos( i + (mul) ) * fAngleSize * mul )
  125.                        
  126.                         ViewOrigin = vFocus - VA:Forward() * fDistance
  127.                        
  128.                         view.origin = ViewOrigin
  129.                         view.angles = VA
  130.                        
  131.                         // Render to the front buffer
  132.                         render.SetRenderTarget( OldRT )
  133.                         render.Clear( 0, 0, 0, 255, true )
  134.                         render.RenderView( view )
  135.                         render.UpdateScreenEffectTexture()
  136.                        
  137.                         // Copy it to our floating point buffer at a reduced alpha
  138.                         render.SetRenderTarget( texFSB )
  139.                         local alpha = (Radials/(math.pi*2))             // Divide alpha by number of radials
  140.                         alpha = alpha * (1-mul)                                 // Reduce alpha the further away from center we are
  141.                         matFB:SetMaterialFloat( "$alpha", alpha  )
  142.                        
  143.  
  144.                                 render.SetMaterial( matFB )
  145.                                 render.DrawScreenQuad()
  146.  
  147.                
  148.                 end
  149.                
  150.         end
  151.        
  152.         // Restore RT
  153.         render.SetRenderTarget( OldRT )
  154.        
  155.         // Render our result buffer to the screen
  156.         matFSB:SetMaterialFloat( "$alpha", 1 )
  157.         matFSB:SetMaterialTexture( "$basetexture", texFSB )
  158.                
  159.         render.SetMaterial( matFSB )
  160.         render.DrawScreenQuad()
  161.  
  162. end
  163.  
  164. function RenderSuperDoF( ViewOrigin, ViewAngles )
  165.  
  166.         if ( FocusGrabber ) then
  167.        
  168.                 tr = util.TraceLine( util.GetPlayerTrace( LocalPlayer(), LocalPlayer():GetCursorAimVector() ) )
  169.                 Distance = tr.HitPos:Distance( ViewOrigin )
  170.                 Status = "Preview"
  171.        
  172.                 pnlWindow:ChangeDistanceTo( Distance )
  173.        
  174.         end
  175.  
  176.         local FocusPoint = ViewOrigin + ViewAngles:Forward() * Distance
  177.        
  178.         if ( Status == "Preview" ) then
  179.                
  180.                 // A low quality, pretty quickly drawn rough outline
  181.                 RenderDoF( ViewOrigin, ViewAngles, FocusPoint, BlurSize, 2, 2, 1 )
  182.                
  183.         elseif ( Status == "Render" ) then
  184.                
  185.                 // A great quality render..
  186.                 RenderDoF( ViewOrigin, ViewAngles, FocusPoint, BlurSize, 16, 8, 1 )
  187.                 Status = "ViewShot"
  188.        
  189.         elseif ( Status == "ViewShot" ) then
  190.                
  191.                 matFSB:SetMaterialFloat( "$alpha", 1 )
  192.                 matFSB:SetMaterialTexture( "$basetexture", texFSB )
  193.                 render.SetMaterial( matFSB )
  194.                 render.DrawScreenQuad()
  195.                
  196.         end
  197.        
  198. end
  199.  
  200. local function RenderSceneHook( ViewOrigin, ViewAngles )
  201.  
  202.         if ( !ValidPanel( pnlWindow ) ) then return end
  203.        
  204.         // Don't render it when the console is up
  205.         if ( FrameTime() == 0 ) then return end
  206.        
  207.         RenderSuperDoF( ViewOrigin, ViewAngles );
  208.         return true;
  209.  
  210. end
  211.  
  212.  
  213. hook.Add( "RenderScene", "RenderSuperDoF", RenderSceneHook )
  214.  
  215.  
  216. local function OpenWindow()
  217.  
  218.         Status = "Preview"
  219.        
  220.         if ( ValidPanel( pnlWindow ) ) then
  221.        
  222.                 pnlWindow:MakePopup()
  223.                 pnlWindow:AlignBottom( 50 )
  224.                 pnlWindow:CenterHorizontal()
  225.                 pnlWindow:SetKeyboardInputEnabled( false )
  226.        
  227.         return end
  228.        
  229.         pnlWindow = vgui.CreateFromTable( paneltypeSuperDOF )
  230.        
  231.         pnlWindow:InvalidateLayout( true )
  232.         pnlWindow:MakePopup()
  233.         pnlWindow:AlignBottom( 50 )
  234.         pnlWindow:CenterHorizontal()
  235.         pnlWindow:SetKeyboardInputEnabled( false )
  236.  
  237. end
  238.  
  239. concommand.Add( "pp_superdof", OpenWindow )
  240.  
  241.  
  242. /*---------------------------------------------------------
  243.    Mouse button down
  244. ---------------------------------------------------------*/  
  245. local function MouseDown( mouse )
  246.  
  247.         if ( !ValidPanel(pnlWindow) ) then return end
  248.  
  249.         vgui.GetWorldPanel():MouseCapture( true )
  250.         FocusGrabber = true
  251.  
  252. end
  253.  
  254.  
  255. /*---------------------------------------------------------
  256.    Mouse button released
  257. ---------------------------------------------------------*/  
  258. local function MouseUp( mouse )
  259.  
  260.         if ( !ValidPanel(pnlWindow) ) then return end
  261.        
  262.         vgui.GetWorldPanel():MouseCapture( false )
  263.         FocusGrabber = false
  264.        
  265. end
  266.  
  267. hook.Add( "GUIMousePressed", "SuperDOFMouseDown", MouseDown )
  268. hook.Add( "GUIMouseReleased", "SuperDOFMouseUp", MouseUp )
  269.  
  270.  
  271. /*
  272. // Control Panel
  273. */
  274. local function BuildControlPanel( CPanel )
  275.  
  276.         CPanel:Help( strInformation )
  277.         CPanel:Button( strOpenWindow, "pp_superdof" )
  278.        
  279. end
  280.  
  281. /*
  282. // Tool Menu
  283. */
  284. local function AddPostProcessMenu()
  285.  
  286.         spawnmenu.AddToolMenuOption( "PostProcessing", "PPShader", "SuperDoF", "#Super DoF", "", "", BuildControlPanel )
  287.  
  288. end
  289.  
  290. hook.Add( "PopulateToolMenu", "AddPostProcessMenu_SuperDoF", AddPostProcessMenu )
  291.