From 0571986059f24946a823e2a31d93d4b8d8f88a05 Mon Sep 17 00:00:00 2001
From: Brychan Dempsey <brychand@hotmail.com>
Date: Wed, 9 Jun 2021 12:51:50 +1200
Subject: [PATCH] Changed: Version; System Registration behaviour

---
 javaecs/pom.xml                                          | 2 +-
 .../src/main/java/nz/ac/massey/javaecs/ECSSystem.java    | 6 +++++-
 .../main/java/nz/ac/massey/javaecs/SystemManager.java    | 9 ++++-----
 3 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/javaecs/pom.xml b/javaecs/pom.xml
index 1f1f28b..01a9425 100644
--- a/javaecs/pom.xml
+++ b/javaecs/pom.xml
@@ -2,7 +2,7 @@
   <modelVersion>4.0.0</modelVersion>
   <groupId>nz.ac.massey.javaecs</groupId>
   <artifactId>javaecs</artifactId>
-  <version>1.2-PRERELEASE-v2</version>
+  <version>0.9.2-PRERELEASE</version>
   <properties>
     <maven.compiler.source>1.8</maven.compiler.source>
     <maven.compiler.target>1.8</maven.compiler.target>
diff --git a/javaecs/src/main/java/nz/ac/massey/javaecs/ECSSystem.java b/javaecs/src/main/java/nz/ac/massey/javaecs/ECSSystem.java
index 60b27ef..47e76ae 100644
--- a/javaecs/src/main/java/nz/ac/massey/javaecs/ECSSystem.java
+++ b/javaecs/src/main/java/nz/ac/massey/javaecs/ECSSystem.java
@@ -13,10 +13,14 @@ package nz.ac.massey.javaecs;
  */
 
 import java.util.Set;
+import java.util.BitSet;
 import java.util.HashSet;
 
 public abstract class ECSSystem{
-    Set<Entity> entities = new HashSet<>();
+    protected Set<Entity> entities = new HashSet<>();
+    protected BitSet registrationSet;
+
+    public BitSet getRegistrationSet;
 
     /**
      * Implement additional parameterised init() functions as required.
diff --git a/javaecs/src/main/java/nz/ac/massey/javaecs/SystemManager.java b/javaecs/src/main/java/nz/ac/massey/javaecs/SystemManager.java
index 0b00a35..529dff1 100644
--- a/javaecs/src/main/java/nz/ac/massey/javaecs/SystemManager.java
+++ b/javaecs/src/main/java/nz/ac/massey/javaecs/SystemManager.java
@@ -15,7 +15,7 @@ import java.util.Map;
 import java.util.HashMap;
 
 class SystemManager{
-    private Map<Type, BitSet> registrationSignatures = new HashMap<>();
+    //private Map<Type, BitSet> registrationSignatures = new HashMap<>();
     private Map<Type, ECSSystem> systems = new HashMap<>();
 
     /**
@@ -40,8 +40,8 @@ class SystemManager{
             // Check if the signature is null
             if (!entityRegistrations.equals(null)){
                 BitSet srcCpy = (BitSet)entityRegistrations.clone();
-                srcCpy.and(registrationSignatures.get(key));
-                if (srcCpy.equals(registrationSignatures.get(key))){ // Bitwise check if the entity is subscribed to this system
+                srcCpy.and(systems.get(key).registrationSet);
+                if (srcCpy.equals(systems.get(key).registrationSet)){ // Bitwise check if the entity is subscribed to this system
                     systems.get(key).entities.add(entity);
                 }
                 else{
@@ -63,7 +63,6 @@ class SystemManager{
             return false;
         }
         systems.put(systemType, system);
-        registrationSignatures.put(systemType, new BitSet());
         return true;
     }
 
@@ -73,7 +72,7 @@ class SystemManager{
      * @param registrations the BitSet containing the required registrations set to true
      */
     protected void setRegistrationSignature(Type systemType, BitSet registrations){
-        registrationSignatures.put(systemType, registrations);
+        systems.get(systemType).registrationSet = registrations;
     }