Fix: Handle signals properly also in servo_map bin

coord-e · coord-e · commit 45948977d750 · 2019-02-11T12:29:02.000Z
diff --git a/src/bin/servo_map.cpp b/src/bin/servo_map.cpp
@@ -14,6 +14,7 @@
 // along with servoarray.  If not, see <http://www.gnu.org/licenses/>.
 
 #include "servoarray/servomap.h"
+#include "./util.h"
 
 #include <iostream>
 
@@ -23,6 +24,10 @@ int main(int argc, char **argv) {
   auto sa = ServoArray::ServoArray(driver);
   auto map = ServoArray::ServoMap(sa);
 
+  util::register_signal(SIGINT);
+  util::register_signal(SIGQUIT);
+  util::register_signal(SIGTERM);
+
   while(true) {
     std::string name;
     std::cout << "(name) > ";
@@ -34,5 +39,7 @@ int main(int argc, char **argv) {
 
     map[name] = rad;
     std::cout << name << " -> " << map[name] << std::endl;
+
+    if(util::should_exit()) break;
   }
 }
