Refactor entity storage registration

TorchAPI · Jun 17, 2017 · 3a5431f · 3a5431f
1 parent 041b190
commit 3a5431f
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/Concealment/ConcealmentPlugin.cs b/Concealment/ConcealmentPlugin.cs
@@ -62,16 +62,18 @@ public override void Init(ITorchBase torch)
             base.Init(torch);
             Settings = Persistent<Settings>.Load(Path.Combine(StoragePath, "Concealment.cfg"));
             _concealedAabbTree = new MyDynamicAABBTreeD(MyConstants.GAME_PRUNING_STRUCTURE_AABB_EXTENSION);
-            torch.SessionUnloading += Torch_SessionUnloading;
+            //torch.SessionUnloading += Torch_SessionUnloading;
+            RegisterEntityStorage("Concealment", Id);
+        }
 
-            //Init storage component.
+        private void RegisterEntityStorage(string name, Guid id)
+        {
             var comp = new MyModStorageComponentDefinition
-            { 
-                Id = new MyDefinitionId(typeof(MyObjectBuilder_ModStorageComponent), "Concealment"),
-                RegisteredStorageGuids = new[] {Id}
+            {
+                Id = new MyDefinitionId(typeof(MyObjectBuilder_ModStorageComponent), name),
+                RegisteredStorageGuids = new[] { id }
             };
             MyDefinitionManager.Static.Definitions.AddDefinition(comp);
-
         }
 
         private void Torch_SessionUnloading()