Fix miscellaneous bugs and enhance colors (#49)

* Add colors for resources
* Fix arbitrary radius calculation bug
* Fix issue when topology is not defined properly
* Fix missing investments in transmission modes in the investment overview
* Fix formatting, update NEWS.md and figures.

Author: Zetison

NEWS.md

@@ -1,5 +1,17 @@
 # Release notes
 
+## Version 0.7.1 (2026-04-24)
+
+### Bugfix
+
+* Fix bug that resulted in arbitrary evaluated lat-lon coordinates for children of `Area`s due to the arbitrary default radius of 1.0 if not provided by the design `.yml` input files. The bug is resolved by first computing minimal distance between areas and letting the new radius be a third of this distance.
+* Fix issue when the transmissions/links is not properly defined (e.g., `from` or `to` objects not sharing resource).
+* Fix missing investments in transmission modes in the Investment overview.
+
+### Enhancements
+
+* Add more colors for resources aliases.
+
 ## Version 0.7.0 (2026-04-17)
 
 ### Breaking changes

Project.toml

@@ -1,6 +1,6 @@
 name = "EnergyModelsGUI"
 uuid = "737a7361-d3b7-40e9-b1ac-59bee4c5ea2d"
-version = "0.7.0"
+version = "0.7.1"
 authors = ["Jon Vegard Venås <JonVegard.Venas@sintef.no>", "Dimitri Pinel <Dimitri.Pinel@sintef.no>", "Magnus Askeland <Magnus.Askeland@sintef.no>", "Shweta Tiwari <Shweta.Tiwari@sintef.no>"]
 
 [deps]

docs/src/figures/EMI_geography.png

@@ -189,6 +189,14 @@ Map types to header symbols for saving results.
 """
 EMGUI._type_to_header(::Type{<:TransmissionMode}) = :element
 
+"""
+    EMGUI.get_inv_objs(obj::Transmission)
+
+Get the objects for which investment information should be stored for a given `Transmission`. 
+This includes obj itself but also its modes.
+"""
+EMGUI.get_inv_objs(obj::Transmission) = [obj, modes(obj)...]
+
 ############################################################################################
 ## From info_axis_utils.jl
 """

docs/src/figures/EMI_geography_Oslo.png

@@ -4,6 +4,7 @@ Power:           "#045993" # blue
 power:           "#045993" # blue
 Hydrogen:        "#009DAD" # cyan
 H2:              "#009DAD" # cyan
+H₂:              "#009DAD" # cyan
 Cooling:         "#00DBDB" # aqua
 Resource:        "#C059A1" # pink
 Waste:           "#DB6000" # orange
@@ -12,11 +13,17 @@ WarmWater:       "#75499C" # purple
 ResourceCarrier: "#C059A1" # pink
 Biomass:         "#118011" # green
 NG:              "#B88505" # goldenrod
+CH₄:             "#B88505" # goldenrod
+CH4:             "#B88505" # goldenrod
+Methane:         "#B88505" # goldenrod
 Oil:             "#DBB500" # gold
 ResourceEmit:    "#606060" # gray
 CO2:             "#606060" # gray
 CO₂:             "#606060" # gray
 Coal:            "#6D392E" # brown
 Water:           "#bfefff" # sky blue
 HeatLT:          "#fbdfb0" # :navajowhite
-HeatHT:          "#e78582" # :lightcoral
+HeatHT:          "#e78582" # :lightcoral
+Ammonia:         "#007C79" # dark teal
+NH3:             "#007C79" # dark teal
+NH₃:             "#007C79" # dark teal

ext/EMGExt/EMGExt.jl

@@ -32,6 +32,7 @@ function EnergySystemDesign(
     icon::String = "",
     parent::AbstractGUIObj = NothingDesign(),
     level::Int64 = 0,
+    radius::Float32 = 0.1f0,
 )
     # Create the path to the file where existing design is stored (if any)
     file::String = design_file(system, design_path)
@@ -43,6 +44,28 @@ function EnergySystemDesign(
         Dict()
     end
 
+    if isa(system, SystemGeo)
+        # Collect all (lon, lat) coordinates from elements that have them
+        coords = [
+            Point2f(element.lon, element.lat)
+            for element ∈ get_children(system)
+        ]
+
+        # Compute all pairwise distances
+        min_dist = Inf
+        for i ∈ 1:(length(coords)-1)
+            for j ∈ (i+1):length(coords)
+                dist = l2_norm(coords[i] - coords[j])
+                if dist < min_dist
+                    min_dist = dist
+                end
+            end
+        end
+
+        # Set radius to a third of the minimal distance such that expanded nodes do not overlap
+        radius = min_dist / 3
+    end
+
     # Complete the `id_to_color_map` if some products are lacking (this is done by choosing
     # colors for the lacking `Resource`s that are most distinct to the existing set of colors)
     if !issubset(get_products(system), keys(id_to_color_map))
@@ -98,7 +121,7 @@ function EnergySystemDesign(
                     xy = xy_parent
                 else # place nodes in a circle around the parents availability node
                     xy = place_nodes_in_circle(
-                        nodes_count, current_node, 1.0f0, xy_parent,
+                        nodes_count, current_node, radius, xy_parent,
                     )
                     current_node += 1
                 end
@@ -119,6 +142,7 @@ function EnergySystemDesign(
                     icon = find_icon(this_sys, id_to_icon_map),
                     parent = design,
                     level = level + 1,
+                    radius = radius,
                 ),
             )
         end

src/colors.yml

@@ -501,6 +501,15 @@ get_values(vals::JuMP.Containers.DenseAxisArray, ts::Vector) = Array(value.(vals
 get_values(vals::DataFrame, ts::Vector) = vals[in.(vals.t, Ref(ts)), :val]
 get_values(vals::TimeProfile, ts::Vector) = vals[ts]
 
+"""
+    get_inv_objs(obj::AbstractElement)
+
+Get the objects for which investment information should be stored for a given `AbstractElement`. 
+By default, this is just the element itself, but for some elements (e.g., `Transmission`) it can 
+be relevant to also store the investment information of related objects (e.g., its modes).
+"""
+get_inv_objs(obj::AbstractElement) = [obj]
+
 """
     get_investment_times(gui::GUI, max_inst::Float64)
 
@@ -516,29 +525,32 @@ function get_investment_times(gui::GUI, max_inst::Float64)
     period_labels = get_var(gui, :periods_labels)
     model = get_model(gui)
     for component ∈ get_root_design(gui)
-        element = get_element(component)
+        # Ensure to include both the component itself and its modes (in case of a transmission) when checking for investments
+        elements = get_inv_objs(get_element(component))
         investment_times = String[]
         investment_capex = Float64[]
-        for (i, t) ∈ enumerate(𝒯ᴵⁿᵛ)
-            for investment_indicator ∈ investment_indicators # important not to use shorthand loop syntax here due to the break command (exiting both loops in that case)
-                sym = Symbol(investment_indicator)
-                if haskey(model, sym) &&
-                   !isempty(model[sym]) &&
-                   element ∈ axes(model[sym])[1]
-                    val = value(model[sym][element, t])
-                    if val > get_var(gui, :tol) * max_inst
-                        capex::Float64 = 0.0
-                        for capex_field ∈ capex_fields
-                            capex_key = Symbol(capex_field[1])
-                            if haskey(model, capex_key) &&
-                               element ∈ axes(model[capex_key])[1]
-                                capex += value(model[capex_key][element, t])
+        for element ∈ elements
+            for (i, t) ∈ enumerate(𝒯ᴵⁿᵛ)
+                for investment_indicator ∈ investment_indicators # important not to use shorthand loop syntax here due to the break command (exiting both loops in that case)
+                    sym = Symbol(investment_indicator)
+                    if haskey(model, sym) &&
+                       !isempty(model[sym]) &&
+                       element ∈ axes(model[sym])[1]
+                        val = value(model[sym][element, t])
+                        if val > get_var(gui, :tol) * max_inst
+                            capex::Float64 = 0.0
+                            for capex_field ∈ capex_fields
+                                capex_key = Symbol(capex_field[1])
+                                if haskey(model, capex_key) &&
+                                   element ∈ axes(model[capex_key])[1]
+                                    capex += value(model[capex_key][element, t])
+                                end
                             end
+                            t_str = split(period_labels[i], " ")[1]
+                            push!(investment_times, t_str)
+                            push!(investment_capex, capex)
+                            break # Do not add the capex again for other elements in investment_indicators
                         end
-                        t_str = split(period_labels[i], " ")[1]
-                        push!(investment_times, t_str)
-                        push!(investment_capex, capex)
-                        break # Do not add the capex again for other elements in investment_indicators
                     end
                 end
             end

src/setup_topology.jl

@@ -229,6 +229,10 @@ function connect!(gui::GUI, connection::Connection, two_way::Bool)
             any(style == inv_linestyle for style ∈ linestyles) ? [inv_linestyle] : [:solid]
     else
         colors = get_colors(connection)
+        if isempty(colors)
+            @warn "No resources transmitted for connection $(get_element(connection).id). Skipping."
+            return
+        end
         no_colors = length(colors)
         linestyle = linestyles
     end