Fixing Graphsjl driver

Author: GearsAD

src/CloudGraphsDFG/services/CloudGraphsDFG.jl

@@ -12,7 +12,7 @@ function _getname(t::T) where T
   T.name.name
 end
 
-# Simply for convenience -don't export
+# Simply for convenience - don't export
 const PackedFunctionNodeData{T} = GenericFunctionNodeData{T, <: AbstractString}
 PackedFunctionNodeData(x1, x2, x3, x4, x5::S, x6::T, x7::String="", x8::Vector{Int}=Int[]) where {T <: PackedInferenceType, S <: AbstractString} = GenericFunctionNodeData(x1, x2, x3, x4, x5, x6, x7, x8)
 const FunctionNodeData{T} = GenericFunctionNodeData{T, Symbol}

src/GraphsDFG/services/GraphsDFG.jl

@@ -209,14 +209,12 @@ Delete the referened DFGFactor from the DFG.
 """
 deleteFactor!(dfg::GraphsDFG, factor::DFGFactor)::DFGFactor = deleteFactor!(dfg, factor.label)
 
-# # Returns a flat vector of the vertices, keyed by ID.
-# # Assuming only variables here for now - think maybe not, should be variables+factors?
 """
     $(SIGNATURES)
 List the DFGVariables in the DFG.
 Optionally specify a label regular expression to retrieves a subset of the variables.
 """
-function ls(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing; tags::Vector{Symbol}=Symbol[])::Vector{DFGVariable}
+function getVariables(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing; tags::Vector{Symbol}=Symbol[])::Vector{DFGVariable}
     variables = map(v -> v.dfgNode, filter(n -> n.dfgNode isa DFGVariable, vertices(dfg.g)))
     if regexFilter != nothing
         variables = filter(v -> occursin(regexFilter, String(v.label)), variables)
@@ -228,14 +226,6 @@ function ls(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing; tags::Ve
 	return variables
 end
 
-# Alias
-"""
-    $(SIGNATURES)
-List the DFGVariables in the DFG.
-Optionally specify a label regular expression to retrieves a subset of the variables.
-"""
-getVariables(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing; tags::Vector{Symbol}=Symbol[])::Vector{DFGVariable} = ls(dfg, regexFilter, tags=tags)
-
 """
     $(SIGNATURES)
 Get a list of IDs of the DFGVariables in the DFG.
@@ -251,42 +241,54 @@ Related
 ls
 """
 function getVariableIds(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing; tags::Vector{Symbol}=Symbol[])::Vector{Symbol}
-  vars = ls(dfg, regexFilter, tags=tags)
+  vars = getVariables(dfg, regexFilter, tags=tags)
   # mask = map(v -> length(intersect(v.tags, tags)) > 0, vars )
   map(v -> v.label, vars)
 end
 
+# Alias
+"""
+    $(SIGNATURES)
+List the DFGVariables in the DFG.
+Optionally specify a label regular expression to retrieves a subset of the variables.
+"""
+ls(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing; tags::Vector{Symbol}=Symbol[])::Vector{Symbol} = getVariableIds(dfg, regexFilter, tags=tags)
 
 """
     $(SIGNATURES)
 List the DFGFactors in the DFG.
 Optionally specify a label regular expression to retrieves a subset of the factors.
 """
-function lsf(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing)::Vector{DFGFactor}
-    factors = map(v -> v.dfgNode, filter(n -> n.dfgNode isa DFGFactor, vertices(dfg.g)))
-    if regexFilter != nothing
-        factors = filter(f -> occursin(regexFilter, String(f.label)), factors)
-    end
-    return factors
-end
-function lsf(dfg::GraphsDFG, label::Symbol)::Vector{Symbol}
-  return getNeighbors(dfg, label)
+function getFactors(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing)::Vector{DFGFactor}
+	factors = map(v -> v.dfgNode, filter(n -> n.dfgNode isa DFGFactor, vertices(dfg.g)))
+	if regexFilter != nothing
+		factors = filter(f -> occursin(regexFilter, String(f.label)), factors)
+	end
+	return factors
 end
 
-# Alias
 """
     $(SIGNATURES)
-List the DFGFactors in the DFG.
+Get a list of the IDs of the DFGFactors in the DFG.
 Optionally specify a label regular expression to retrieves a subset of the factors.
 """
-getFactors(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing)::Vector{DFGFactor} = lsf(dfg, regexFilter)
+getFactorIds(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing)::Vector{Symbol} = map(f -> f.label, getFactors(dfg, regexFilter))
 
 """
     $(SIGNATURES)
-Get a list of the IDs of the DFGFactors in the DFG.
+List the DFGFactors in the DFG.
 Optionally specify a label regular expression to retrieves a subset of the factors.
 """
-getFactorIds(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing)::Vector{Symbol} = map(f -> f.label, lsf(dfg, regexFilter))
+# Alias
+lsf(dfg::GraphsDFG, regexFilter::Union{Nothing, Regex}=nothing)::Vector{Symbol} = getFactorIds(dfg, regexFilter)
+
+"""
+	$(SIGNATURES)
+Alias for getNeighbors - returns neighbors around a given node label.
+"""
+function lsf(dfg::GraphsDFG, label::Symbol)::Vector{Symbol}
+  return getNeighbors(dfg, label)
+end
 
 """
     $(SIGNATURES)

test/interfaceTests.jl

@@ -17,8 +17,8 @@ addFactor!(dfg, [v1, v2], f1)
     @test symdiff([:a, :b], getVariableIds(dfg)) == []
     @test getFactorIds(dfg) == [:f1]
     # Regexes
-    @test ls(dfg, r"a") == [v1]
-    @test lsf(dfg, r"f*") == [f1]
+    @test ls(dfg, r"a") == [v1.label]
+    @test lsf(dfg, r"f*") == [f1.label]
     # Accessors
     @test getAddHistory(dfg) == [:a, :b] #, :f1
     @test getDescription(dfg) != nothing
@@ -118,13 +118,13 @@ end
     # Subgraphs
     dfgSubgraph = getSubgraphAroundNode(dfg, verts[1], 2)
     # Only returns x1 and x2
-    @test symdiff([:x1, :x1x2f1, :x2], map(n -> n.label, [ls(dfgSubgraph)..., lsf(dfgSubgraph)...])) == []
+    @test symdiff([:x1, :x1x2f1, :x2], [ls(dfgSubgraph)..., lsf(dfgSubgraph)...]) == []
     # Test include orphan factorsVoid
     dfgSubgraph = getSubgraphAroundNode(dfg, verts[1], 1, true)
-    @test symdiff([:x1, :x1x2f1], map(n -> n.label, [ls(dfgSubgraph)..., lsf(dfgSubgraph)...])) == []
+    @test symdiff([:x1, :x1x2f1], [ls(dfgSubgraph)..., lsf(dfgSubgraph)...]) == []
     # Test adding to the dfg
     dfgSubgraph = getSubgraphAroundNode(dfg, verts[1], 2, true, dfgSubgraph)
-    @test symdiff([:x1, :x1x2f1, :x2], map(n -> n.label, [ls(dfgSubgraph)..., lsf(dfgSubgraph)...])) == []
+    @test symdiff([:x1, :x1x2f1, :x2], [ls(dfgSubgraph)..., lsf(dfgSubgraph)...]) == []
 end
 
 @testset "Producing Dot Files" begin

test/runtests.jl

@@ -6,7 +6,7 @@ using DistributedFactorGraphs
 apis = [GraphsDFG]
 # apis = [graphsDFG, cgDFG]
 for api in apis
-    @testset "Testing Driver: $(typeof(api))" begin
+    @testset "Testing Driver: $(api)" begin
         global testDFGAPI = api
         include("interfaceTests.jl")
     end