]> git.sesse.net Git - nageru/blobdiff - theme.lua
Do signal mapping in get_width/get_height.
[nageru] / theme.lua
index e5f902c9563785ba8c8326f2de4363fd0a30978d..130e74173598d4fbe49712144d3bec7c49747235 100644 (file)
--- a/theme.lua
+++ b/theme.lua
@@ -7,13 +7,15 @@
 -- where all the low-level details (such as texture formats) are handled by the
 -- C++ side and you generally just build chains.
 
+-- TODO: Deal with inputs that are different from our native 1280x720 resolution.
+
 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 fade_src_signal = 0
+local fade_dst_signal = 0
 
 local input0_neutral_color = {0.5, 0.5, 0.5}
 local input1_neutral_color = {0.5, 0.5, 0.5}
@@ -21,6 +23,16 @@ local input1_neutral_color = {0.5, 0.5, 0.5}
 local live_signal_num = 0
 local preview_signal_num = 1
 
+-- Valid values for live_signal_num and preview_signal_num.
+local INPUT0_SIGNAL_NUM = 0
+local INPUT1_SIGNAL_NUM = 1
+local SBS_SIGNAL_NUM = 2
+local STATIC_SIGNAL_NUM = 3
+
+-- “fake” signal number that signifies that we are fading from one input
+-- to the next.
+local FADE_SIGNAL_NUM = 4
+
 -- The main live chain.
 function make_sbs_chain(hq)
        local chain = EffectChain.new(16, 9)
@@ -76,16 +88,62 @@ 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_wb0_effect = fade_chain_hq:add_effect(WhiteBalanceEffect.new())
-local fade_chain_hq_input1 = fade_chain_hq:add_live_input(true)
-local fade_chain_hq_wb1_effect = fade_chain_hq:add_effect(WhiteBalanceEffect.new())
-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_wb0_effect, fade_chain_hq_wb1_effect)
-fade_chain_hq:finalize(true)
+-- A chain to fade between two inputs, of which either can be a picture
+-- or a live input. In practice only used live, but we still support the
+-- hq parameter.
+function make_fade_chain(input0_live, input1_live, hq)
+       local chain = EffectChain.new(16, 9)
+
+       local input0, wb0_effect, input0_last, input1, wb1_effect, input1_last
+
+       if input0_live then
+               input0 = chain:add_live_input(true)
+               wb0_effect = chain:add_effect(WhiteBalanceEffect.new())
+               input0:connect_signal(0)
+               input0_last = wb0_effect
+       else
+               input0 = chain:add_effect(ImageInput.new("bg.jpeg"))
+               input0_last = input0
+       end
+
+       if input1_live then
+               input1 = chain:add_live_input(true)
+               wb1_effect = chain:add_effect(WhiteBalanceEffect.new())
+               input1:connect_signal(1)
+               input1_last = wb1_effect
+       else
+               input1 = chain:add_effect(ImageInput.new("bg.jpeg"))
+               input1_last = input1
+       end
+
+       local mix_effect = chain:add_effect(MixEffect.new(), input0_last, input1_last)
+       chain:finalize(hq)
+
+       return {
+               chain = chain,
+               input0 = {
+                       input = input0,
+                       white_balance_effect = wb0_effect
+               },
+               input1 = {
+                       input = input1,
+                       white_balance_effect = wb1_effect
+               },
+               mix_effect = mix_effect
+       }
+end
+
+-- Chains to fade between two inputs, in various configurations.
+local fade_chains = {}
+for input0_type, input0_live in pairs({static = false, live = true}) do
+       fade_chains[input0_type] = {}
+       for input1_type, input1_live in pairs({static = false, live = true}) do
+               fade_chains[input0_type][input1_type] = {}
+               for _, hq in pairs({true, false}) do
+                       fade_chains[input0_type][input1_type][hq] = make_fade_chain(input0_live, input1_live, hq)
+               end
+       end
+end
 
 -- A chain to show a single input on screen (HQ version).
 local simple_chain_hq = EffectChain.new(16, 9)
@@ -101,10 +159,20 @@ simple_chain_lq_input:connect_signal(0)  -- First input card. Can be changed whe
 local simple_chain_lq_wb_effect = simple_chain_lq:add_effect(WhiteBalanceEffect.new())
 simple_chain_lq:finalize(false)
 
+-- A chain to show a single static picture on screen (HQ version).
+local static_chain_hq = EffectChain.new(16, 9)
+local static_chain_hq_input = static_chain_hq:add_effect(ImageInput.new("bg.jpeg"))
+static_chain_hq:finalize(true)
+
+-- A chain to show a single static picture on screen (LQ version).
+local static_chain_lq = EffectChain.new(16, 9)
+local static_chain_lq_input = static_chain_lq:add_effect(ImageInput.new("bg.jpeg"))
+static_chain_lq:finalize(false)
+
 -- Returns the number of outputs in addition to the live (0) and preview (1).
 -- Called only once, at the start of the program.
 function num_channels()
-       return 3
+       return 4
 end
 
 -- Returns the name for each additional channel (starting from 2).
@@ -114,8 +182,10 @@ function channel_name(channel)
                return "Input 1"
        elseif channel == 3 then
                return "Input 2"
-       else
+       elseif channel == 4 then
                return "Side-by-side"
+       elseif channel == 5 then
+               return "Static picture"
        end
 end
 
@@ -136,17 +206,14 @@ function set_wb(channel, red, green, blue)
 end
 
 function finish_transitions(t)
-       -- If live is 2 (SBS) but de-facto single, make it so.
-       if live_signal_num == 2 and t >= transition_end and zoom_dst == 1.0 then
+       -- If live is SBS but de-facto single, make it so.
+       if live_signal_num == SBS_SIGNAL_NUM 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
+       -- If live is fade but de-facto single, make it so.
+       if live_signal_num == FADE_SIGNAL_NUM and t >= transition_end then
+               live_signal_num = fade_dst_signal
        end
 end
 
@@ -158,20 +225,26 @@ function get_transitions(t)
                return {}
        end
 
-       if live_signal_num == 2 and t >= transition_start and t <= transition_end then
+       if live_signal_num == SBS_SIGNAL_NUM 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
+       if (live_signal_num == INPUT0_SIGNAL_NUM or
+           live_signal_num == INPUT1_SIGNAL_NUM or
+           live_signal_num == STATIC_SIGNAL_NUM) and
+          (preview_signal_num == INPUT0_SIGNAL_NUM or
+           preview_signal_num == INPUT1_SIGNAL_NUM or
+           preview_signal_num == STATIC_SIGNAL_NUM) then
                return {"Cut", "", "Fade"}
        end
 
        -- Various zooms.
-       if live_signal_num == 2 and (preview_signal_num == 0 or preview_signal_num == 1) then
+       if live_signal_num == SBS_SIGNAL_NUM and
+          (preview_signal_num == INPUT0_SIGNAL_NUM or preview_signal_num == INPUT1_SIGNAL_NUM) then
                return {"Cut", "Zoom in"}
-       elseif (live_signal_num == 0 or live_signal_num == 1) and preview_signal_num == 2 then
+       elseif (live_signal_num == INPUT0_SIGNAL_NUM or live_signal_num == INPUT1_SIGNAL_NUM) and
+              preview_signal_num == SBS_SIGNAL_NUM then
                return {"Cut", "Zoom out"}
        end
 
@@ -181,7 +254,7 @@ end
 function transition_clicked(num, t)
        if num == 0 then
                -- Cut.
-               if live_signal_num == 3 then
+               if live_signal_num == FADE_SIGNAL_NUM then
                        -- Ongoing fade; finish it immediately.
                        finish_transitions(transition_end)
                end
@@ -190,7 +263,7 @@ function transition_clicked(num, t)
                live_signal_num = preview_signal_num
                preview_signal_num = temp
 
-               if live_signal_num == 2 then
+               if live_signal_num == SBS_SIGNAL_NUM then
                        -- Just cut to SBS, we need to reset any zooms.
                        zoom_src = 1.0
                        zoom_dst = 0.0
@@ -207,8 +280,8 @@ function transition_clicked(num, t)
                        return
                end
 
-               if (live_signal_num == 0 and preview_signal_num == 1) or
-                  (live_signal_num == 1 and preview_signal_num == 0) then
+               if (live_signal_num == INPUT0_SIGNAL_NUM and preview_signal_num == INPUT1_SIGNAL_NUM) or
+                  (live_signal_num == INPUT1_SIGNAL_NUM and preview_signal_num == INPUT0_SIGNAL_NUM) then
                        -- We can't zoom between these. Just make a cut.
                        io.write("Cutting from " .. live_signal_num .. " to " .. live_signal_num .. "\n")
                        local temp = live_signal_num
@@ -217,15 +290,17 @@ function transition_clicked(num, t)
                        return
                end
 
-               if live_signal_num == 2 and (preview_signal_num == 0 or preview_signal_num == 1) then
+               if live_signal_num == SBS_SIGNAL_NUM and
+                  (preview_signal_num == INPUT0_SIGNAL_NUM or preview_signal_num == INPUT1_SIGNAL_NUM) then
                        -- Zoom in from SBS to single.
                        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 or live_signal_num == 1) and preview_signal_num == 2 then
+                       preview_signal_num = SBS_SIGNAL_NUM
+               elseif (live_signal_num == INPUT0_SIGNAL_NUM or live_signal_num == INPUT1_SIGNAL_NUM) and
+                      preview_signal_num == SBS_SIGNAL_NUM then
                        -- Zoom out from single to SBS.
                        transition_start = t
                        transition_end = t + 1.0
@@ -233,26 +308,25 @@ function transition_clicked(num, t)
                        zoom_dst = 0.0
                        preview_signal_num = live_signal_num
                        zoom_poi = live_signal_num
-                       live_signal_num = 2
+                       live_signal_num = SBS_SIGNAL_NUM
                end
        elseif num == 2 then
                finish_transitions(t)
 
                -- Fade.
-               if live_signal_num == 0 and preview_signal_num == 1 then
+               if (live_signal_num ~= preview_signal_num) and
+                  (live_signal_num == INPUT0_SIGNAL_NUM or
+                   live_signal_num == INPUT1_SIGNAL_NUM or
+                   live_signal_num == STATIC_SIGNAL_NUM) and
+                  (preview_signal_num == INPUT0_SIGNAL_NUM or
+                   preview_signal_num == INPUT1_SIGNAL_NUM or
+                   preview_signal_num == STATIC_SIGNAL_NUM) 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
+                       fade_src_signal = live_signal_num
+                       fade_dst_signal = preview_signal_num
+                       preview_signal_num = live_signal_num
+                       live_signal_num = FADE_SIGNAL_NUM
                else
                        -- Fades involving SBS are ignored (we have no chain for it).
                end
@@ -270,6 +344,11 @@ end
 -- the output, although you can ignore them if you don't need them
 -- (they're useful if you want to e.g. know what to resample by).
 --
+-- <signals> is basically an exposed InputState, which you can use to
+-- query for information about the signals at the point of the current
+-- frame. In particular, you can call get_width() and get_height()
+-- for any signal number, and use that to e.g. assist in chain selection.
+--
 -- You should return two objects; the chain itself, and then a
 -- function (taking no parameters) that is run just before rendering.
 -- The function needs to call connect_signal on any inputs, so that
@@ -281,49 +360,44 @@ end
 --
 -- NOTE: The chain returned must be finalized with the Y'CbCr flag
 -- if and only if num==0.
-function get_chain(num, t, width, height)
+function get_chain(num, t, width, height, signals)
        if num == 0 then  -- Live.
-               if live_signal_num == 0 or live_signal_num == 1 then  -- Plain inputs.
+               if live_signal_num == INPUT0_SIGNAL_NUM or live_signal_num == INPUT1_SIGNAL_NUM then  -- Plain input.
                        prepare = function()
                                simple_chain_hq_input:connect_signal(live_signal_num)
-                               if live_signal_num == 0 then
-                                       set_neutral_color(simple_chain_hq_wb_effect, input0_neutral_color)
-                               else
-                                       set_neutral_color(simple_chain_hq_wb_effect, input1_neutral_color)
-                               end
+                               set_neutral_color_from_signal(simple_chain_hq_wb_effect, live_signal_num)
                        end
                        return simple_chain_hq, prepare
-               elseif live_signal_num == 3 then  -- Fade.
+               elseif live_signal_num == STATIC_SIGNAL_NUM then  -- Static picture.
                        prepare = function()
-                               fade_chain_hq_input0:connect_signal(0)
-                               set_neutral_color(fade_chain_hq_wb0_effect, input0_neutral_color)
-                               fade_chain_hq_input1:connect_signal(1)
-                               set_neutral_color(fade_chain_hq_wb1_effect, input1_neutral_color)
-                               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
+                       return static_chain_hq, prepare
+               elseif live_signal_num == FADE_SIGNAL_NUM then  -- Fade.
+                       local input0_type = (fade_src_signal == STATIC_SIGNAL_NUM) and "static" or "live"
+                       local input1_type = (fade_dst_signal == STATIC_SIGNAL_NUM) and "static" or "live"
+                       local chain = fade_chains[input0_type][input1_type][true]
+                       prepare = function()
+                               if input0_type == "live" then
+                                       chain.input0.input:connect_signal(fade_src_signal)
+                                       set_neutral_color_from_signal(chain.input0.white_balance_effect, fade_src_signal)
                                end
+                               if input1_type == "live" then
+                                       chain.input1.input:connect_signal(fade_dst_signal)
+                                       set_neutral_color_from_signal(chain.input1.white_balance_effect, fade_dst_signal)
+                               end
+                               local tt = calc_fade_progress(t, transition_start, transition_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)
+                               chain.mix_effect:set_float("strength_first", 1.0 - tt)
+                               chain.mix_effect:set_float("strength_second", tt)
                        end
-                       return fade_chain_hq, prepare
+                       return chain.chain, prepare
                end
 
-               -- SBS code (live_signal_num == 2).
+               -- SBS code (live_signal_num == SBS_SIGNAL_NUM).
                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)
+                               simple_chain_hq_input:connect_signal(INPUT0_SIGNAL_NUM)
                                set_neutral_color(simple_chain_hq_wb_effect, input0_neutral_color)
                        end
                        return simple_chain_hq, prepare
@@ -345,26 +419,31 @@ function get_chain(num, t, width, height)
        if num == 1 then  -- Preview.
                num = preview_signal_num + 2
        end
-       if num == 2 then
+       if num == INPUT0_SIGNAL_NUM + 2 then
                prepare = function()
                        simple_chain_lq_input:connect_signal(0)
                        set_neutral_color(simple_chain_lq_wb_effect, input0_neutral_color)
                end
                return simple_chain_lq, prepare
        end
-       if num == 3 then
+       if num == INPUT1_SIGNAL_NUM + 2 then
                prepare = function()
                        simple_chain_lq_input:connect_signal(1)
                        set_neutral_color(simple_chain_lq_wb_effect, input1_neutral_color)
                end
                return simple_chain_lq, prepare
        end
-       if num == 4 then
+       if num == SBS_SIGNAL_NUM + 2 then
                prepare = function()
                        prepare_sbs_chain(main_chain_lq, 0.0, width, height)
                end
                return main_chain_lq.chain, prepare
        end
+       if num == STATIC_SIGNAL_NUM + 2 then
+               prepare = function()
+               end
+               return static_chain_lq, prepare
+       end
 end
 
 function place_rectangle(resample_effect, resize_effect, padding_effect, x0, y0, x1, y1, screen_width, screen_height, input_width, input_height)
@@ -484,7 +563,7 @@ function prepare_sbs_chain(chain, t, screen_width, screen_height)
 
        -- Interpolate between the fullscreen and side-by-side views.
        local scale0, tx0, tx0
-       if zoom_poi == 0 then
+       if zoom_poi == INPUT0_SIGNAL_NUM then
                local new_left0 = lerp(left0, 0, t)
                local new_right0 = lerp(right0, screen_width, t)
                local new_top0 = lerp(top0, 0, t)
@@ -520,3 +599,27 @@ end
 function set_neutral_color(effect, color)
        effect:set_vec3("neutral_color", color[1], color[2], color[3])
 end
+
+function set_neutral_color_from_signal(effect, signal)
+       if signal == INPUT0_SIGNAL_NUM then
+               set_neutral_color(effect, input0_neutral_color)
+       else
+               set_neutral_color(effect, input1_neutral_color)
+       end
+end
+
+function calc_fade_progress(t, transition_start, transition_end)
+       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
+
+       -- 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))
+
+       return tt
+end