From ceaaad29a126181e2baf535f0c23c41ce7a43638 Mon Sep 17 00:00:00 2001
From: dehann <dehann.fourie@gmail.com>
Date: Thu, 14 Oct 2021 23:34:49 -0400
Subject: [PATCH] enh api on defVariable and typeModName

---
 src/services/DFGVariable.jl   | 2 ++
 src/services/Serialization.jl | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/services/DFGVariable.jl b/src/services/DFGVariable.jl
index e59c8e16..69200669 100644
--- a/src/services/DFGVariable.jl
+++ b/src/services/DFGVariable.jl
@@ -102,6 +102,8 @@ macro defVariable(structname, manifold, point_identity)
 
         DFG.getPointIdentity(::Type{$structname}) = $point_identity
 
+        DFG.getVariableType(::typeof($manifold)) = $structname
+
     end)
 end
 
diff --git a/src/services/Serialization.jl b/src/services/Serialization.jl
index 8b5522c1..caeed6e7 100644
--- a/src/services/Serialization.jl
+++ b/src/services/Serialization.jl
@@ -74,6 +74,8 @@ function typeModuleName(variableType::InferenceVariable)
     return String(take!(io))
 end
 
+typeModuleName(varT::Type{<:InferenceVariable}) = typeModuleName(varT())
+
 function getTypeFromSerializationModule(variableTypeString::String)
     try
         # split the type at last `.`