]> git.sesse.net Git - nageru/blobdiff - theme.lua
Do signal mapping in get_width/get_height.
[nageru] / theme.lua
index 93f79ff80df21c096926fe06eabe218f2140a27d..130e74173598d4fbe49712144d3bec7c49747235 100644 (file)
--- a/theme.lua
+++ b/theme.lua
@@ -7,6 +7,8 @@
 -- 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
@@ -25,10 +27,11 @@ local preview_signal_num = 1
 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
+-- “fake” signal number that signifies that we are fading from one input
 -- to the next.
-local FADE_SIGNAL_NUM = 3
+local FADE_SIGNAL_NUM = 4
 
 -- The main live chain.
 function make_sbs_chain(hq)
@@ -85,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)
@@ -110,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).
@@ -125,6 +184,8 @@ function channel_name(channel)
                return "Input 2"
        elseif channel == 4 then
                return "Side-by-side"
+       elseif channel == 5 then
+               return "Static picture"
        end
 end
 
@@ -169,8 +230,12 @@ function get_transitions(t)
                return {"Cut"}
        end
 
-       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
+       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
 
@@ -249,8 +314,13 @@ function transition_clicked(num, t)
                finish_transitions(t)
 
                -- Fade.
-               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
+               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_signal = live_signal_num
@@ -274,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
@@ -285,7 +360,7 @@ 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 == INPUT0_SIGNAL_NUM or live_signal_num == INPUT1_SIGNAL_NUM then  -- Plain input.
                        prepare = function()
@@ -293,28 +368,29 @@ function get_chain(num, t, width, height)
                                set_neutral_color_from_signal(simple_chain_hq_wb_effect, live_signal_num)
                        end
                        return simple_chain_hq, prepare
+               elseif live_signal_num == STATIC_SIGNAL_NUM then  -- Static picture.
+                       prepare = function()
+                       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()
-                               fade_chain_hq_input0:connect_signal(fade_src_signal)
-                               set_neutral_color_from_signal(fade_chain_hq_wb0_effect, fade_src_signal)
-                               fade_chain_hq_input1:connect_signal(fade_dst_signal)
-                               set_neutral_color_from_signal(fade_chain_hq_wb1_effect, fade_dst_signal)
-                               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
+                               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)
 
-                               -- 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", 1.0 - tt)
-                               fade_chain_mix_effect:set_float("strength_second", 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 == SBS_SIGNAL_NUM).
@@ -363,6 +439,11 @@ function get_chain(num, t, 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)
@@ -526,3 +607,19 @@ function set_neutral_color_from_signal(effect, signal)
                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