feat: move getiftversion-to-__init__

WilhelmusLab · Aug 18, 2023 · 313e78d · 313e78d
1 parent 2b3d215
commit 313e78d
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 27 deletions.
diff --git a/src/IFTPipeline.jl b/src/IFTPipeline.jl
@@ -21,6 +21,37 @@ include("feature-extraction.jl")
 include("tracker.jl")
 include("h5.jl")
 
+const iftversion = []
+function __init__()
+    function getiftversion()
+        deps = Pkg.dependencies()
+        iftversion = []
+        for (_, dep) in deps
+            dep.is_direct_dep || continue
+            dep.version === nothing && continue
+            dep.name != "IceFloeTracker" && continue
+            push!(iftversion, dep.version)
+            break
+        end
+
+        # For CI tests where IceFloeTracker is not a dependency
+        try
+            ift = iftversion[1]
+        catch
+            return "unknown"
+        end
+
+        ift = iftversion[1]
+        maj = Int(ift.major)
+        min = Int(ift.minor)
+        patch = Int(ift.patch)
+        return "v$maj.$min.$patch"
+    end
+
+    push!(iftversion, getiftversion())
+    return nothing
+end
+
 export cache_vector, sharpen,
     sharpen_gray,
     preprocess,
@@ -40,7 +71,8 @@ export cache_vector, sharpen,
     mkclitrack!,
     mkfilenames,
     makeh5files,
-    getlatlon
+    getlatlon,
+    iftversion
 
 export IceFloeTracker
 end
diff --git a/src/h5.jl b/src/h5.jl
@@ -1,28 +1,3 @@
-function getiftversion()
-    deps = Pkg.dependencies()
-    iftversion = []
-    for (_, dep) in deps
-        dep.is_direct_dep || continue
-        dep.version === nothing && continue
-        dep.name != "IceFloeTracker" && continue
-        push!(iftversion, dep.version)
-        break
-    end
-
-    # For CI tests where IceFloeTracker is not a dependency
-    try
-        ift = iftversion[1]
-    catch
-        return "unknown"
-    end
-
-    ift = iftversion[1]
-    maj = Int(ift.major)
-    min = Int(ift.minor)
-    patch = Int(ift.patch)
-    return "v$maj.$min.$patch"
-end
-
 """
     makeh5filename(imgfname, ts)
 
@@ -125,7 +100,7 @@ The `index` group contains the spatial coordinates in the source image coordinat
 function makeh5files(; pathtosampleimg::String, resdir::String)
     latlondata = getlatlon(pathtosampleimg)
 
-    iftversion = getiftversion()
+    iftversion = iftversion[1]
 
     ptpath = joinpath(resdir, "passtimes.jls")
     passtimes = deserialize(ptpath)