Fixed installation and nasty exit bug

[pierogi] / pirkeysetwidgetitem.cpp

diff --git a/pirkeysetwidgetitem.cpp b/pirkeysetwidgetitem.cpp
index f2d1293..97d47b4 100644 (file)
--- a/pirkeysetwidgetitem.cpp
+++ b/pirkeysetwidgetitem.cpp
@@ -11,6 +11,13 @@ PIRKeysetWidgetItem::PIRKeysetWidgetItem(
 }
 
 
+void PIRKeysetWidgetItem::addDeviceType(
+  PIRDeviceTypeName dt)
+{
+  devices.insert(dt);
+}
+
+
 unsigned int PIRKeysetWidgetItem::getID()
 {
   return id;
@@ -21,3 +28,10 @@ PIRMakeName PIRKeysetWidgetItem::getMake()
 {
   return make;
 }
+
+
+bool PIRKeysetWidgetItem::supportsDeviceType(
+  PIRDeviceTypeName dt)
+{
+  return (devices.find(dt) != devices.end());
+}