Custom label not found error

Author: Affie

src/DataBlobs/services/BlobEntry.jl

@@ -67,18 +67,14 @@ Also see: [`addBlobentry!`](@ref), [`getBlob`](@ref), [`listBlobentries`](@ref)
 """
 function getBlobentry(var::AbstractDFGVariable, key::Symbol)
     if !hasBlobentry(var, key)
-        throw(
-            KeyError(
-                "No dataEntry label $(key) found in variable $(getLabel(var)). Available keys: $(keys(var.dataDict))",
-            ),
-        )
+        throw(DFGLabelError(key, keys(var.dataDict)))
     end
     return var.dataDict[key]
 end
 
 function getBlobentry(var::VariableDFG, key::Symbol)
     if !hasBlobentry(var, key)
-        throw(KeyError(key))
+        throw(DFGLabelError(key))
     end
     return var.blobEntries[findfirst(x -> x.label == key, var.blobEntries)]
 end

src/DistributedFactorGraphs.jl

@@ -354,6 +354,7 @@ export plotDFG
 ##==============================================================================
 
 # Entities
+include("errors.jl")
 
 include("entities/AbstractDFG.jl")
 

src/errors.jl

@@ -0,0 +1,19 @@
+"""
+    DFGLabelError(label)
+
+Label not found.
+"""
+struct DFGLabelError <: Exception
+    label::Any
+    available::Any
+end
+
+DFGLabelError(key::T) where {T} = DFGLabelError(key, T[])
+
+function Base.showerror(io::IO, ex::DFGLabelError)
+    print(io, "DFGLabelError: label ", ex.label, " not found.")
+    if !isempty(ex.available)
+        println(io, " Available labels:")
+        show(io, ex.available)
+    end
+end