Fix auto-WB of certain kinds of input.

[nageru] / nageru / theme.cpp

diff --git a/nageru/theme.cpp b/nageru/theme.cpp
index f3e1aa1c64bb53cdc913e7f5fdb6d326362ff860..162c3bc5b7f5a90ba3fd54ed047e82abdb0d6daa 100644 (file)
--- a/nageru/theme.cpp
+++ b/nageru/theme.cpp
@@ -1686,9 +1686,11 @@ bool Theme::get_supports_set_wb(unsigned channel)
 
 void Theme::set_wb(unsigned channel, float r, float g, float b)
 {
+       int signal = get_channel_signal(channel);
+
        lock_guard<mutex> lock(m);
-       if (channel_signals.count(channel)) {
-               white_balance_for_signal[channel_signals[channel]] = RGBTriplet{ r, g, b };
+       if (signal != -1) {
+               white_balance_for_signal[signal] = RGBTriplet{ r, g, b };
        }
 
        call_lua_wb_callback(channel, r, g, b);