Implement head detection setting.

1 files changed, 17 insertions, 0 deletions

diff --git a/parrot_zik/interface/version2.py b/parrot_zik/interface/version2.py
index ff21004..c348bb4 100644
--- a/parrot_zik/interface/version2.py
+++ b/parrot_zik/interface/version2.py
@@ -98,14 +98,23 @@ class ParrotZikVersion2Interface(ParrotZikBaseInterface):
 
         self.flight_mode = MenuItem("Flight Mode", self.toggle_flight_mode,
                                     checkitem=True, visible=False)
+        self.settings = MenuItem("Settings", None, visible=False)
+        self.settings_submenu = Menu()
+        self.settings.set_submenu(self.settings_submenu)
+
+        self.head_detection = MenuItem("Head Detection", self.toggle_head_detection, checkitem=True)
+        self.settings_submenu.append(self.head_detection)
+
         self.indicator.menu.append(self.room_sound_effect)
         self.indicator.menu.append(self.noise_cancelation)
         self.indicator.menu.append(self.flight_mode)
+        self.indicator.menu.append(self.settings)
 
     def activate(self, manager):
         self.noise_cancelation.show()
         self.flight_mode.show()
         self.room_sound_effect.show()
+        self.settings.show()
         super(ParrotZikVersion2Interface, self).activate(manager)
         self._read_noise_cancelation()
         self.flight_mode.set_active(self.parrot.flight_mode)
@@ -130,6 +139,7 @@ class ParrotZikVersion2Interface(ParrotZikBaseInterface):
         self.noise_cancelation.hide()
         self.flight_mode.hide()
         self.room_sound_effect.hide()
+        self.settings.hide()
         super(ParrotZikVersion2Interface, self).deactivate()
 
     def toggle_flight_mode(self, widget):
@@ -222,3 +232,10 @@ class ParrotZikVersion2Interface(ParrotZikBaseInterface):
         )
         for noise_control, menu_item in noise_control_to_menuitem_map:
             menu_item.set_active(active_noise_control == noise_control)
+
+    def toggle_head_detection(self, widget):
+        try:
+            self.parrot.head_detection = self.head_detection.get_active()
+            self.head_detection.set_active(self.parrot.head_detection)
+        except resource_manager.DeviceDisconnected:
+            self.deactivate()
\ No newline at end of file