Merge pull request #1432 from NREL-Sienna/feat/get-num-components

Add get_num_components for System

Author: jd-lara

src/PowerSystems.jl

@@ -382,6 +382,7 @@ export ReservoirConnectedDevicesKey
 export ReservoirConnectedDevicesMapping
 export get_component
 export get_components
+export get_num_components
 export get_associated_components
 export show_components
 export get_subcomponents
@@ -631,6 +632,7 @@ import InfrastructureSystems:
     get_resolution,
     get_window,
     get_name,
+    get_num_components,
     get_component_uuids,
     get_supplemental_attribute,
     get_supplemental_attributes,

src/base.jl

@@ -2960,3 +2960,5 @@ attributes.
 function get_forecast_summary_table(sys::System)
     return IS.get_forecast_summary_table(sys.data)
 end
+
+IS.get_base_component_type(sys::System) = Component

src/utils/print.jl

@@ -7,7 +7,7 @@ end
 function Base.show(io::IO, sys::System)
     show_system_table(io, sys; backend = Val(:auto))
 
-    if IS.get_num_components(sys.data.components) > 0
+    if IS.get_num_components(sys) > 0
         show_components_table(io, sys; backend = Val(:auto))
     end
 
@@ -21,7 +21,7 @@ Base.show(io::IO, ::MIME"text/plain", sys::System) = show(io, sys)
 function Base.show(io::IO, ::MIME"text/html", sys::System)
     show_system_table(io, sys; backend = Val(:html), standalone = false)
 
-    if IS.get_num_components(sys.data.components) > 0
+    if get_num_components(sys) > 0
         show_components_table(
             io,
             sys;
@@ -44,7 +44,7 @@ end
 
 function show_system_table(io::IO, sys::System; kwargs...)
     header = ["Property", "Value"]
-    num_components = IS.get_num_components(sys.data.components)
+    num_components = get_num_components(sys)
     table = [
         "Name" isnothing(get_name(sys)) ? "" : get_name(sys)
         "Description" isnothing(get_description(sys)) ? "" : get_description(sys)

test/test_parse_psse.jl

@@ -339,8 +339,8 @@ end
 
     @test get_base_power(original_sys) == get_base_power(deserialized_sys)
     @test get_frequency(original_sys) == get_frequency(deserialized_sys)
-    @test IS.get_num_components(original_sys.data.components) ==
-          IS.get_num_components(deserialized_sys.data.components)
+    @test get_num_components(original_sys) ==
+          get_num_components(deserialized_sys)
 
     gen1_names = sort(get_name.(get_components(ThermalStandard, original_sys)))
     gen2_names = sort(get_name.(get_components(ThermalStandard, deserialized_sys)))