refactored code for improved structure and reduced memory allocations

mcosovic · Sep 30, 2024 · b69ac95 · b69ac95
1 parent 86ce441
commit b69ac95
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/powerSystem/generator.jl b/src/powerSystem/generator.jl
@@ -303,8 +303,12 @@ function updateGenerator!(system::PowerSystem; label::IntStrMiss, kwargs...)
         end
         if statusOld == 0
             gen.layout.inservice += 1
-            position = searchsortedfirst(bus.supply.generator[idxBus], idx)
-            insert!(bus.supply.generator[idxBus], position, idx)
+            if haskey(bus.supply.generator, idxBus)
+                position = searchsortedfirst(bus.supply.generator[idxBus], idx)
+                insert!(bus.supply.generator[idxBus], position, idx)
+            else
+                bus.supply.generator[idxBus] = [idx]
+            end
         end
     end
     gen.layout.status[idx] = statusNew