Fix focused tree graph spacing layout

Author: mcosovic

src/visualization/figure.jl

@@ -993,8 +993,14 @@ function treeGraphFigure(
         labelPlacement,
         showFactorLabels
     )
-    variableX = [levelX[depth] for depth in variableDepths]
-    factorX = [levelX[depth] for depth in factorDepths]
+    variableX = zeros(Float64, length(graph.variables))
+    factorX = zeros(Float64, length(graph.factors))
+    for index in view.variables
+        variableX[index] = levelX[variableDepths[index]]
+    end
+    for index in view.factors
+        factorX[index] = levelX[factorDepths[index]]
+    end
     variableY = zeros(Float64, length(graph.variables))
     factorY = zeros(Float64, length(graph.factors))
 
@@ -1298,8 +1304,14 @@ function verticalTreeGraphFigure(
     )
     variableX = zeros(Float64, length(graph.variables))
     factorX = zeros(Float64, length(graph.factors))
-    variableY = [levelY[depth] for depth in variableDepths]
-    factorY = [levelY[depth] for depth in factorDepths]
+    variableY = zeros(Float64, length(graph.variables))
+    factorY = zeros(Float64, length(graph.factors))
+    for index in view.variables
+        variableY[index] = levelY[variableDepths[index]]
+    end
+    for index in view.factors
+        factorY[index] = levelY[factorDepths[index]]
+    end
 
     for depth in 1:maxDepth
         nodes = Tuple{Symbol, Int}[]

test/graph_figure.jl

@@ -296,6 +296,13 @@ end
         @test occursin("variable-context", viewSvg)
         @test occursin("<title>Variable x1", viewSvg)
         @test occursin(r"<text class=\"label[^\"]*\"[^>]*>\s*<title>Variable x1", viewSvg)
+
+        spacedViewSvg = graphFigure(
+            tree;
+            layout = (rowSpacing = 80, columnSpacing = (90, 210)),
+            view = (variables = [:x1], hops = 2)
+        )
+        @test startswith(spacedViewSvg, "<svg")
     end
 
     @testset "File output" begin