wip

2 files changed, 23 insertions, 3 deletions

diff --git a/lib/felflame/scene_manager.rb b/lib/felflame/scene_manager.rb
index a9d3ac7..06a0dfa 100644
--- a/lib/felflame/scene_manager.rb
+++ b/lib/felflame/scene_manager.rb
@@ -34,8 +34,11 @@ module FelFlame
     def add(*systems_to_add)
       self.systems |= systems_to_add
       self.systems = systems.sort_by(&:priority)
-      FelFlame::Stage.update_systems_list if FelFlame::Stage.scenes.include? self
-      true
+      systems_to_add.each do |system|
+        system.scenes |= [self]
+      end
+      #FelFlame::Stage.update_systems_list if FelFlame::Stage.scenes.include? self
+      #true
     end
 
     # Removes any number of Systems from this Scene
@@ -43,6 +46,9 @@ module FelFlame
     def remove(*systems_to_remove)
       self.systems -= systems_to_remove
       self.systems = systems.sort_by(&:priority)
+      systems_to_remove.each do |system|
+        system.scenes.delete system
+      end
       FelFlame::Stage.update_systems_list if FelFlame::Stage.scenes.include? self
       true
     end
@@ -50,6 +56,9 @@ module FelFlame
     # Removes all Systems from this Scene
     # @return [Boolean] +true+
     def clear
+      systems.each do |system|
+        system.scenes.delete self
+      end
       systems.clear
       FelFlame::Stage.update_systems_list if FelFlame::Stage.scenes.include? self
       true
diff --git a/lib/felflame/system_manager.rb b/lib/felflame/system_manager.rb
index d936e8e..183b359 100644
--- a/lib/felflame/system_manager.rb
+++ b/lib/felflame/system_manager.rb
@@ -12,10 +12,20 @@ module FelFlame
     # @!visibility private
     attr_writer :addition_triggers, :removal_triggers, :attr_triggers
 
+    # Stores all the scenes this system is a part of.
+    attr_writer :scenes
+
+    def scenes
+      @scenes ||= []
+    end
+
     def priority=(priority)
       @priority = priority
-      FelFlame::Stage.systems = FelFlame::Stage.systems.sort_by(&:priority)
+      scenes.each do |scene|
+        scene.systems = scene.systems.sort_by(&:priority)
+      end
     end
+
     # Stores references to components or their managers that trigger
     # this component when a component or component from that manager
     # is added to an entity.
@@ -73,6 +83,7 @@ module FelFlame
       @const_name = name
       @priority = priority
       @block = block
+      @scenes = []
     end
 
     # Manually execute the system a single time