]> git.sesse.net Git - nageru/blobdiff - theme.lua
Make some labels and a white balance button per preview (the latter is not hooked...
[nageru] / theme.lua
index c7ce264017cfd9bde2572bf74e9c9e8096898ecf..b91e0fadfe8240c924d8f462380b0c053a206f20 100644 (file)
--- a/theme.lua
+++ b/theme.lua
@@ -6,12 +6,14 @@
 -- Themes are written in Lua, which reflects a simplified form of the Movit API
 -- where all the low-level details (such as texture formats) are handled by the
 -- C++ side and you generally just build chains.
-io.write("hello from lua\n")
 
-local zoom_start = -2.0
-local zoom_end = -1.0
+local transition_start = -2.0
+local transition_end = -1.0
 local zoom_src = 0.0
 local zoom_dst = 1.0
+local zoom_poi = 0   -- which input to zoom in on
+local fade_src = 0.0
+local fade_dst = 1.0
 
 local live_signal_num = 0
 local preview_signal_num = 1
@@ -19,9 +21,9 @@ local preview_signal_num = 1
 -- The main live chain.
 function make_sbs_chain(hq)
        local chain = EffectChain.new(16, 9)
-       local input0 = chain:add_live_input()
+       local input0 = chain:add_live_input(true)
        input0:connect_signal(0)
-       local input1 = chain:add_live_input()
+       local input1 = chain:add_live_input(true)
        input1:connect_signal(1)
 
        local resample_effect = nil
@@ -71,16 +73,25 @@ end
 local main_chain_hq = make_sbs_chain(true)
 local main_chain_lq = make_sbs_chain(false)
 
+-- A chain to fade between two inputs (live chain only)
+local fade_chain_hq = EffectChain.new(16, 9)
+local fade_chain_hq_input0 = fade_chain_hq:add_live_input(true)
+local fade_chain_hq_input1 = fade_chain_hq:add_live_input(true)
+fade_chain_hq_input0:connect_signal(0)
+fade_chain_hq_input1:connect_signal(1)
+local fade_chain_mix_effect = fade_chain_hq:add_effect(MixEffect.new(), fade_chain_hq_input0, fade_chain_hq_input1)
+fade_chain_hq:finalize(true)
+
 -- A chain to show a single input on screen (HQ version).
 local simple_chain_hq = EffectChain.new(16, 9)
-local simple_chain_hq_input = simple_chain_hq:add_live_input()
-simple_chain_hq_input:connect_signal(0);  -- First input card. Can be changed whenever you want.
+local simple_chain_hq_input = simple_chain_hq:add_live_input(true)
+simple_chain_hq_input:connect_signal(0)  -- First input card. Can be changed whenever you want.
 simple_chain_hq:finalize(true)
 
 -- A chain to show a single input on screen (LQ version).
 local simple_chain_lq = EffectChain.new(16, 9)
-local simple_chain_lq_input = simple_chain_lq:add_live_input()
-simple_chain_lq_input:connect_signal(0);  -- First input card. Can be changed whenever you want.
+local simple_chain_lq_input = simple_chain_lq:add_live_input(true)
+simple_chain_lq_input:connect_signal(0)  -- First input card. Can be changed whenever you want.
 simple_chain_lq:finalize(false)
 
 -- Returns the number of outputs in addition to the live (0) and preview (1).
@@ -89,35 +100,61 @@ function num_channels()
        return 3
 end
 
--- Called every frame.
-function get_transitions(t)
+-- Returns the name for each additional channel (starting from 2).
+-- Called only once for each channel, at the start of the program.
+function channel_name(channel)
+       if channel == 2 then
+               return "Input 1"
+       elseif channel == 3 then
+               return "Input 2"
+       else
+               return "Side-by-side"
+       end
+end
+
+-- Returns if a given channel supports setting white balance (starting from 2).
+-- Called only once for each channel, at the start of the program.
+function supports_set_wb(channel)
+       return channel == 2 or channel == 3
+end
+
+function finish_transitions(t)
        -- If live is 2 (SBS) but de-facto single, make it so.
-       if live_signal_num == 2 and t >= zoom_end and zoom_dst == 1.0 then
+       if live_signal_num == 2 and t >= transition_end and zoom_dst == 1.0 then
+               live_signal_num = zoom_poi
+       end
+
+       -- If live is 3 (fade) but de-facto single, make it so.
+       if live_signal_num == 3 and t >= transition_end and fade_dst == 1.0 then
                live_signal_num = 0
        end
+       if live_signal_num == 3 and t >= transition_end and fade_dst == 0.0 then
+               live_signal_num = 1
+       end
+end
+
+-- Called every frame.
+function get_transitions(t)
+       finish_transitions(t)
 
        if live_signal_num == preview_signal_num then
                return {}
        end
 
-       if live_signal_num == 2 and t >= zoom_start and t <= zoom_end then
+       if live_signal_num == 2 and t >= transition_start and t <= transition_end then
                -- Zoom in progress.
                return {"Cut"}
        end
 
        if (live_signal_num == 0 and preview_signal_num == 1) or
           (live_signal_num == 1 and preview_signal_num == 0) then
-               return {"Cut"}
+               return {"Cut", "", "Fade"}
        end
 
-       if live_signal_num == 2 and preview_signal_num == 1 then
-               -- Zoom-out not supported here yet.
-               return {"Cut"}
-       end
-
-       if live_signal_num == 2 and preview_signal_num == 0 then
+       -- Various zooms.
+       if live_signal_num == 2 and (preview_signal_num == 0 or preview_signal_num == 1) then
                return {"Cut", "Zoom in"}
-       elseif live_signal_num == 0 and preview_signal_num == 2 then
+       elseif (live_signal_num == 0 or live_signal_num == 1) and preview_signal_num == 2 then
                return {"Cut", "Zoom out"}
        end
 
@@ -127,6 +164,11 @@ end
 function transition_clicked(num, t)
        if num == 0 then
                -- Cut.
+               if live_signal_num == 3 then
+                       -- Ongoing fade; finish it immediately.
+                       finish_transitions(transition_end)
+               end
+
                local temp = live_signal_num
                live_signal_num = preview_signal_num
                preview_signal_num = temp
@@ -135,16 +177,13 @@ function transition_clicked(num, t)
                        -- Just cut to SBS, we need to reset any zooms.
                        zoom_src = 1.0
                        zoom_dst = 0.0
-                       zoom_start = -2.0
-                       zoom_end = -1.0
+                       transition_start = -2.0
+                       transition_end = -1.0
                end
        elseif num == 1 then
                -- Zoom.
 
-               -- If live is 2 (SBS) but de-facto single, make it so.
-               if live_signal_num == 2 and t >= zoom_end and zoom_dst == 1.0 then
-                       live_signal_num = 0
-               end
+               finish_transitions(t)
 
                if live_signal_num == preview_signal_num then
                        -- Nothing to do.
@@ -161,27 +200,45 @@ function transition_clicked(num, t)
                        return
                end
 
-               if live_signal_num == 2 and preview_signal_num == 1 then
-                       io.write("NOT SUPPORTED YET\n")
-                       return
-               end
-
-               if live_signal_num == 2 and preview_signal_num == 0 then
+               if live_signal_num == 2 and (preview_signal_num == 0 or preview_signal_num == 1) then
                        -- Zoom in from SBS to single.
-                       zoom_start = t
-                       zoom_end = t + 1.0
+                       transition_start = t
+                       transition_end = t + 1.0
                        zoom_src = 0.0
                        zoom_dst = 1.0
+                       zoom_poi = preview_signal_num
                        preview_signal_num = 2
-               elseif live_signal_num == 0 and preview_signal_num == 2 then
+               elseif (live_signal_num == 0 or live_signal_num == 1) and preview_signal_num == 2 then
                        -- Zoom out from single to SBS.
-                       zoom_start = t
-                       zoom_end = t + 1.0
+                       transition_start = t
+                       transition_end = t + 1.0
                        zoom_src = 1.0
                        zoom_dst = 0.0
-                       preview_signal_num = 0
+                       preview_signal_num = live_signal_num
+                       zoom_poi = live_signal_num
                        live_signal_num = 2
                end
+       elseif num == 2 then
+               finish_transitions(t)
+
+               -- Fade.
+               if live_signal_num == 0 and preview_signal_num == 1 then
+                       transition_start = t
+                       transition_end = t + 1.0
+                       fade_src = 1.0
+                       fade_dst = 0.0
+                       preview_signal_num = 0
+                       live_signal_num = 3
+               elseif live_signal_num == 1 and preview_signal_num == 0 then
+                       transition_start = t
+                       transition_end = t + 1.0
+                       fade_src = 0.0
+                       fade_dst = 1.0
+                       preview_signal_num = 1
+                       live_signal_num = 3
+               else
+                       -- Fades involving SBS are ignored (we have no chain for it).
+               end
        end
 end
 
@@ -214,10 +271,32 @@ function get_chain(num, t, width, height)
                                simple_chain_hq_input:connect_signal(live_signal_num)
                        end
                        return simple_chain_hq, prepare
+               elseif live_signal_num == 3 then  -- Fade.
+                       prepare = function()
+                               fade_chain_hq_input0:connect_signal(0)
+                               fade_chain_hq_input1:connect_signal(1)
+                               local tt = (t - transition_start) / (transition_end - transition_start)
+                               if tt < 0.0 then
+                                       tt = 0.0
+                               elseif tt > 1.0 then
+                                       tt = 1.0
+                               end
+
+                               tt = fade_src + tt * (fade_dst - fade_src)
+
+                               -- Make the fade look maybe a tad more natural, by pumping it
+                               -- through a sigmoid function.
+                               tt = 10.0 * tt - 5.0
+                               tt = 1.0 / (1.0 + math.exp(-tt))
+
+                               fade_chain_mix_effect:set_float("strength_first", tt)
+                               fade_chain_mix_effect:set_float("strength_second", 1.0 - tt)
+                       end
+                       return fade_chain_hq, prepare
                end
 
                -- SBS code (live_signal_num == 2).
-               if t > zoom_end and zoom_dst == 1.0 then
+               if t > transition_end and zoom_dst == 1.0 then
                        -- Special case: Show only the single image on screen.
                        prepare = function()
                                simple_chain_hq_input:connect_signal(0)
@@ -225,12 +304,12 @@ function get_chain(num, t, width, height)
                        return simple_chain_hq, prepare
                end
                prepare = function()
-                       if t < zoom_start then
+                       if t < transition_start then
                                prepare_sbs_chain(main_chain_hq, zoom_src, width, height)
-                       elseif t > zoom_end then
+                       elseif t > transition_end then
                                prepare_sbs_chain(main_chain_hq, zoom_dst, width, height)
                        else
-                               local tt = (t - zoom_start) / (zoom_end - zoom_start)
+                               local tt = (t - transition_start) / (transition_end - transition_start)
                                -- Smooth it a bit.
                                tt = math.sin(tt * 3.14159265358 * 0.5)
                                prepare_sbs_chain(main_chain_hq, zoom_src + (zoom_dst - zoom_src) * tt, width, height)
@@ -261,7 +340,7 @@ function get_chain(num, t, width, height)
        end
 end
 
-function place_rectangle(resample_effect, resize_effect, padding_effect, x0, y0, x1, y1, screen_width, screen_height)
+function place_rectangle(resample_effect, resize_effect, padding_effect, x0, y0, x1, y1, screen_width, screen_height, input_width, input_height)
        local srcx0 = 0.0
        local srcx1 = 1.0
        local srcy0 = 0.0
@@ -278,7 +357,15 @@ function place_rectangle(resample_effect, resize_effect, padding_effect, x0, y0,
                return
        end
 
-       -- Clip. (TODO: Clip on upper/left sides, too.)
+       -- Clip.
+       if x0 < 0 then
+               srcx0 = -x0 / (x1 - x0)
+               x0 = 0
+       end
+       if y0 < 0 then
+               srcy0 = -y0 / (y1 - y0)
+               y0 = 0
+       end
        if x1 > screen_width then
                srcx1 = (screen_width - x0) / (x1 - x0)
                x1 = screen_width
@@ -314,8 +401,8 @@ function place_rectangle(resample_effect, resize_effect, padding_effect, x0, y0,
                padding_effect:set_int("top", math.floor(y0))
 
                -- Correct _that_ discrepancy by subpixel offset in the resampling.
-               resample_effect:set_float("left", -x_subpixel_offset / zoom_x)
-               resample_effect:set_float("top", -y_subpixel_offset / zoom_y)
+               resample_effect:set_float("left", srcx0 * input_width - x_subpixel_offset / zoom_x)
+               resample_effect:set_float("top", srcy0 * input_height - y_subpixel_offset / zoom_y)
 
                -- Finally, adjust the border so it is exactly where we want it.
                padding_effect:set_float("border_offset_left", x_subpixel_offset)
@@ -340,6 +427,10 @@ function round(x)
        return math.floor(x + 0.5)
 end
 
+function lerp(a, b, t)
+       return a + (b - a) * t
+end
+
 function prepare_sbs_chain(chain, t, screen_width, screen_height)
        chain.input0.input:connect_signal(0)
        chain.input1.input:connect_signal(1)
@@ -354,8 +445,8 @@ function prepare_sbs_chain(chain, t, screen_width, screen_height)
        local right0 = left0 + width0
 
        -- Second input is positioned (16,48) from the bottom-right.
-       local width1 = 384 * screen_width/1280.0
-       local height1 = 216 * screen_height/720.0
+       local width1 = round(384 * screen_width/1280.0)
+       local height1 = round(216 * screen_height/720.0)
 
        local bottom1 = screen_height - 48 * screen_height/720.0
        local right1 = screen_width - 16 * screen_width/1280.0
@@ -363,9 +454,26 @@ function prepare_sbs_chain(chain, t, screen_width, screen_height)
        local left1 = right1 - width1
 
        -- Interpolate between the fullscreen and side-by-side views.
-       local scale0 = 1.0 + t * (1280.0 / 848.0 - 1.0)
-       local tx0 = 0.0 + t * (-left0 * scale0)
-       local ty0 = 0.0 + t * (-top0 * scale0)
+       local scale0, tx0, tx0
+       if zoom_poi == 0 then
+               local new_left0 = lerp(left0, 0, t)
+               local new_right0 = lerp(right0, screen_width, t)
+               local new_top0 = lerp(top0, 0, t)
+               local new_bottom0 = lerp(bottom0, screen_height, t)
+
+               scale0 = (new_right0 - new_left0) / width0  -- Same vertically and horizonally.
+               tx0 = new_left0 - left0 * scale0
+               ty0 = new_top0 - top0 * scale0
+       else
+               local new_left1 = lerp(left1, 0, t)
+               local new_right1 = lerp(right1, screen_width, t)
+               local new_top1 = lerp(top1, 0, t)
+               local new_bottom1 = lerp(bottom1, screen_height, t)
+
+               scale0 = (new_right1 - new_left1) / width1  -- Same vertically and horizonally.
+               tx0 = new_left1 - left1 * scale0
+               ty0 = new_top1 - top1 * scale0
+       end
 
        top0 = top0 * scale0 + ty0
        bottom0 = bottom0 * scale0 + ty0
@@ -376,6 +484,6 @@ function prepare_sbs_chain(chain, t, screen_width, screen_height)
        bottom1 = bottom1 * scale0 + ty0
        left1 = left1 * scale0 + tx0
        right1 = right1 * scale0 + tx0
-       place_rectangle(chain.input0.resample_effect, chain.input0.resize_effect, chain.input0.padding_effect, left0, top0, right0, bottom0, screen_width, screen_height)
-       place_rectangle(chain.input1.resample_effect, chain.input1.resize_effect, chain.input1.padding_effect, left1, top1, right1, bottom1, screen_width, screen_height)
+       place_rectangle(chain.input0.resample_effect, chain.input0.resize_effect, chain.input0.padding_effect, left0, top0, right0, bottom0, screen_width, screen_height, 1280, 720)
+       place_rectangle(chain.input1.resample_effect, chain.input1.resize_effect, chain.input1.padding_effect, left1, top1, right1, bottom1, screen_width, screen_height, 1280, 720)
 end