Make enum more like struct

JuliaInterop · Aug 31, 2021 · 756d2fe · 756d2fe
1 parent aaeb1d3
commit 756d2fe
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/generator/jltypes.jl b/src/generator/jltypes.jl
@@ -26,11 +26,12 @@ end
 JuliaCrecord() = JuliaCrecord(Symbol())
 JuliaCrecord(x::AbstractString) = JuliaCrecord(Symbol(x), getNullCursor())
 
-struct JuliaCenum <: AbstractJuliaType
+struct JuliaCenum{T<:CLCursor} <: AbstractJuliaType
     sym::Symbol
+    cursor::T
 end
 JuliaCenum() = JuliaCenum(Symbol())
-JuliaCenum(x::AbstractString) = JuliaCenum(Symbol(x))
+JuliaCenum(x::AbstractString) = JuliaCenum(Symbol(x), getNullCursor())
 
 struct JuliaCtypedef <: AbstractJuliaType
     sym::Symbol
@@ -151,7 +152,10 @@ end
 tojulia(x::CLInvalid) = JuliaUnknown(x)
 tojulia(x::CLFunctionProto) = JuliaCfunction(spelling(getTypeDeclaration(x)))
 tojulia(x::CLFunctionNoProto) = JuliaCfunction(spelling(getTypeDeclaration(x)))
-tojulia(x::CLEnum) = JuliaCenum(spelling(getTypeDeclaration(x)))
+function tojulia(x::CLEnum)
+    c = getTypeDeclaration(x)
+    JuliaCenum(Symbol(spelling(c)), c)
+end
 function tojulia(x::CLRecord)
     c = getTypeDeclaration(x)
     JuliaCrecord(Symbol(spelling(c)), c)