X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Ftheme.cpp;h=162c3bc5b7f5a90ba3fd54ed047e82abdb0d6daa;hb=f9024d141398e69e7b4011becd3ebbe37eaa1776;hp=f3e1aa1c64bb53cdc913e7f5fdb6d326362ff860;hpb=6cf87ad853439f8565c575fb29dc539a15fdba87;p=nageru

diff --git a/nageru/theme.cpp b/nageru/theme.cpp
index f3e1aa1..162c3bc 100644
--- 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);