Hide implementation of MessageHandlerMap.

firmware/application/main.cpp

@@ -184,10 +184,7 @@ private:
 		while( !shared_memory.application_queue.is_empty() ) {
 			auto message = shared_memory.application_queue.pop();
 
-			auto& fn = context.message_map[message->id];
+			context.message_map.send(message);
-			if( fn ) {
-				fn(message);
-			}
 
 			message->state = Message::State::Free;
 		}

firmware/application/ui_audio.cpp

@@ -26,13 +26,15 @@
 namespace ui {
 
 void Audio::on_show() {
-	context().message_map[Message::ID::AudioStatistics] = [this](const Message* const p) {
+	context().message_map.register_handler(Message::ID::AudioStatistics,
+		[this](const Message* const p) {
 			this->on_statistics_update(static_cast<const AudioStatisticsMessage*>(p)->statistics);
-	};
+		}
+	);
 }
 
 void Audio::on_hide() {
-	context().message_map[Message::ID::AudioStatistics] = nullptr;
+	context().message_map.unregister_handler(Message::ID::AudioStatistics);
 }
 
 void Audio::paint(Painter& painter) {

firmware/application/ui_channel.cpp

@@ -26,13 +26,15 @@
 namespace ui {
 
 void Channel::on_show() {
-	context().message_map[Message::ID::ChannelStatistics] = [this](const Message* const p) {
+	context().message_map.register_handler(Message::ID::ChannelStatistics,
+		[this](const Message* const p) {
 			this->on_statistics_update(static_cast<const ChannelStatisticsMessage*>(p)->statistics);
-	};
+		}
+	);
 }
 
 void Channel::on_hide() {
-	context().message_map[Message::ID::ChannelStatistics] = nullptr;
+	context().message_map.unregister_handler(Message::ID::ChannelStatistics);
 }
 
 void Channel::paint(Painter& painter) {

firmware/application/ui_debug.cpp

@@ -40,13 +40,15 @@ BasebandStatsView::BasebandStatsView() {
 }
 
 void BasebandStatsView::on_show() {
-	context().message_map[Message::ID::BasebandStatistics] = [this](const Message* const p) {
+	context().message_map.register_handler(Message::ID::BasebandStatistics,
+		[this](const Message* const p) {
 			this->on_statistics_update(static_cast<const BasebandStatisticsMessage*>(p)->statistics);
-	};
+		}
+	);
 }
 
 void BasebandStatsView::on_hide() {
-	context().message_map[Message::ID::BasebandStatistics] = nullptr;
+	context().message_map.unregister_handler(Message::ID::BasebandStatistics);
 }
 
 

firmware/application/ui_receiver.cpp

@@ -481,14 +481,16 @@ ReceiverView::ReceiverView(
 
 	receiver_model.enable();
 
-	context().message_map[Message::ID::FSKPacket] = [](const Message* const p) {
+	context().message_map.register_handler(Message::ID::FSKPacket,
+		[](const Message* const p) {
 			const auto message = static_cast<const FSKPacketMessage*>(p);
 			(void)message;
-	};
+		}
+	);
 }
 
 ReceiverView::~ReceiverView() {
-	context().message_map[Message::ID::FSKPacket] = nullptr;
+	context().message_map.unregister_handler(Message::ID::FSKPacket);
 
 	// TODO: Manipulating audio codec here, and in ui_receiver.cpp. Good to do
 	// both?

firmware/application/ui_rssi.cpp

@@ -26,13 +26,15 @@
 namespace ui {
 
 void RSSI::on_show() {
-	context().message_map[Message::ID::RSSIStatistics] = [this](const Message* const p) {
+	context().message_map.register_handler(Message::ID::RSSIStatistics,
+		[this](const Message* const p) {
 			this->on_statistics_update(static_cast<const RSSIStatisticsMessage*>(p)->statistics);
-	};
+		}
+	);
 }
 
 void RSSI::on_hide() {
-	context().message_map[Message::ID::RSSIStatistics] = nullptr;
+	context().message_map.unregister_handler(Message::ID::RSSIStatistics);
 }
 
 void RSSI::paint(Painter& painter) {

firmware/application/ui_spectrum.hpp

@@ -233,13 +233,15 @@ public:
 	}
 
 	void on_show() override {
-		context().message_map[Message::ID::ChannelSpectrum] = [this](const Message* const p) {
+		context().message_map.register_handler(Message::ID::ChannelSpectrum,
+			[this](const Message* const p) {
 				this->on_channel_spectrum(reinterpret_cast<const ChannelSpectrumMessage*>(p)->spectrum);
-		};
+			}
+		);
 	}
 
 	void on_hide() override {
-		context().message_map[Message::ID::ChannelSpectrum] = nullptr;
+		context().message_map.unregister_handler(Message::ID::ChannelSpectrum);
 	}
 
 	void set_parent_rect(const Rect new_parent_rect) override {

firmware/baseband/main.cpp

@@ -481,14 +481,16 @@ public:
 		MessageHandlerMap& message_handlers
 	) : message_handlers(message_handlers)
 	{
-		message_handlers[Message::ID::FSKConfiguration] = [this](const Message* const p) {
+		message_handlers.register_handler(Message::ID::FSKConfiguration,
+			[this](const Message* const p) {
 				auto m = reinterpret_cast<const FSKConfigurationMessage*>(p);
 				this->configure(m->configuration);
-		};
+			}
+		);
 	}
 
 	~FSKProcessor() {
-		message_handlers[Message::ID::FSKConfiguration] = nullptr;
+		message_handlers.unregister_handler(Message::ID::FSKConfiguration);
 	}
 
 	void configure(const FSKConfiguration new_configuration) {
@@ -736,10 +738,7 @@ private:
 		while( !shared_memory.baseband_queue.is_empty() ) {
 			auto message = shared_memory.baseband_queue.pop();
 
-			auto& fn = message_map[message->id];
+			message_map.send(message);
-			if( fn ) {
-				fn(message);
-			}
 
 			message->state = Message::State::Free;
 		}
@@ -804,7 +803,8 @@ int main(void) {
 	EventDispatcher event_dispatcher;
 	auto& message_handlers = event_dispatcher.message_handlers();
 
-	message_handlers[Message::ID::BasebandConfiguration] = [&message_handlers](const Message* const p) {
+	message_handlers.register_handler(Message::ID::BasebandConfiguration,
+		[&message_handlers](const Message* const p) {
 			auto message = reinterpret_cast<const BasebandConfigurationMessage*>(p);
 			if( message->configuration.mode != baseband_configuration.mode ) {
 
@@ -846,7 +846,8 @@ int main(void) {
 			}
 
 			baseband_configuration = message->configuration;
-	};
+		}
+	);
 
 	/* TODO: Ensure DMAs are configured to point at first LLI in chain. */
 

firmware/common/message.hpp

@@ -246,12 +246,19 @@ class MessageHandlerMap {
 public:
 	using MessageHandler = std::function<void(const Message* const p)>;
 
-	MessageHandler& operator[](Message::ID n) {
+	void register_handler(const Message::ID id, MessageHandler&& handler) {
-		return map_[toUType(n)];
+		map_[toUType(id)] = std::move(handler);
 	}
 
-	const MessageHandler& operator[](Message::ID n) const {
+	void unregister_handler(const Message::ID id) {
-		return map_[toUType(n)];
+		map_[toUType(id)] = nullptr;
+	}
+
+	void send(const Message* const message) {
+		auto& fn = map_[toUType(message->id)];
+		if( fn ) {
+			fn(message);
+		}
 	}
 
 private: