add listSolvekeys listSupersolves

Author: dehann

src/DistributedFactorGraphs.jl

@@ -68,6 +68,7 @@ export exists,
        updateVariable!, updateFactor!,
        deleteVariable!, deleteFactor!,
        listVariables, listFactors,
+       listSolvekeys, listSupersolves,
        getVariables, getFactors,
        isVariable, isFactor
 

src/services/AbstractDFG.jl

@@ -493,6 +493,24 @@ function listFactors(dfg::G, regexFilter::Union{Nothing, Regex}=nothing; tags::V
     return map(f -> f.label, getFactors(dfg, regexFilter, tags=tags, solvable=solvable))
 end
 
+"""
+    $SIGNATURES
+List all the solvekeys used amongst all variables in the distributed factor graph object.
+
+Related
+
+listSupersolves, getSolverDataDict, listVariables
+"""
+function listSolvekeys(dfg::AbstractDFG)
+  skeys = Set{Symbol}()
+  for vs in listVariables(dfg), ky in keys(getSolverDataDict(getVariable(dfg, vs)))
+    push!(skeys, ky)
+  end
+  return skeys
+end
+const listSupersolves = listSolvekeys
+
+
 ##------------------------------------------------------------------------------
 ## Aliases and Other filtered lists
 ##------------------------------------------------------------------------------

test/testBlocks.jl

@@ -507,6 +507,11 @@ end
 @test ls(fg) == listVariables(fg)
 @test lsf(fg) == listFactors(fg)
 
+if getVariable(fg, ls(fg)[1]) isa DFGVariable
+  @test :default in listSolvekeys(fg)
+  @test :default in listSupersolves(fg)
+end
+
 # simple broadcast test
 if f0 isa DFGFactor
     @test issetequal(getFactorType.(fg, lsf(fg)),  [TestFunctorInferenceType1(), TestFunctorSingleton()])