Fixed a segfault on shutdown

include/Pub_Sub.h

@@ -3,6 +3,7 @@
 #include <iostream>
 #include "Event.hpp"
 #include <memory>
+#include <set>
 
 
 class VrEventPublisher;
@@ -13,14 +14,6 @@ class VrEventPublisher;
  *
  * VrEventSubscriber
  *
- * When inherited, the user must impliment a
- *
- *
- *
- *
- *
- *
- *
  */
 
 
@@ -28,16 +21,25 @@ class VrEventPublisher;
 class VrEventSubscriber {
 public:
 	virtual ~VrEventSubscriber();
+
+    // Recieve an event from a publisher, event must be cast to it's respective event type
 	virtual void event_handler(VrEventPublisher *publisher, std::unique_ptr<vr::Event> event) = 0;
+
+    // Subscribes to the publisher, keeps track of the ptr and the relevent event types
 	void subscribe_to_publisher(VrEventPublisher* publisher, vr::Event::EventType type);
 	void subscribe_to_publisher(VrEventPublisher* publisher, std::vector<vr::Event::EventType> type);
-	void unsubscribe(VrEventPublisher* publisher, vr::Event::EventType type);
+
+    // Looks for the publisher ptr and event type in the subscriptions map. If there, Removes them
+    void unsubscribe(VrEventPublisher* publisher, vr::Event::EventType type);
+    void unsubscribe_all(VrEventPublisher* publisher);
+    void unsubscribe_all();
+
 protected:
 
     // When we destroy a subscriber we need to be able to notify the publishers
     // We have to keep track of every EventType because of the way EventTypes
     // are mapped to subscribers in the publisher
-	std::map<VrEventPublisher*, std::vector<vr::Event::EventType>> subscriptions;
+	std::map<VrEventPublisher*, std::set<vr::Event::EventType>> subscriptions;
 };
 
 
@@ -45,11 +47,20 @@ class VrEventPublisher {
 public:
 
 	virtual ~VrEventPublisher();
+
+    // Adds the subscriber ptr to the bucket[event_type]
 	virtual void subscribe(VrEventSubscriber *subscriber, vr::Event::EventType type);
 	virtual void subscribe(VrEventSubscriber *subscriber, std::vector<vr::Event::EventType> type);
-	virtual void unsubscribe(VrEventSubscriber *s, vr::Event::EventType c);
+
+    // Removes the subscriber ptr from the specified bucket[type]
+    // If subscribed to multiple events, unsubscribe must be called for each event
+    virtual void unsubscribe(VrEventSubscriber *s, vr::Event::EventType c);
+
+    // Trigger the publisher to notify it's subscribers to the specified event
 	virtual void notify_subscribers(std::unique_ptr<vr::Event> event);
+
 private:
+
 	std::map<vr::Event::EventType, std::vector<VrEventSubscriber*>> subscribers;
 
 };