add terrain cpp version check

move version assert inside

infinigen/terrain/core.py

@@ -6,6 +6,7 @@
 
 import logging
 import os
+from ctypes import c_int32
 from pathlib import Path
 
 import bpy
@@ -46,6 +47,7 @@
     Mesh,
     Vars,
     get_caminfo,
+    load_cdll,
     move_modifier,
     write_attributes,
 )
@@ -123,6 +125,20 @@ def __init__(
         populated_bounds=(-75, 75, -75, 75, -25, 55),
         bounds=(-500, 500, -500, 500, -500, 500),
     ):
+        dll = load_cdll(
+            str(
+                Path(__file__).parent.resolve()
+                / "lib"
+                / "cpu"
+                / "elements"
+                / "waterbody.so"
+            )
+        )
+        func = dll.get_version
+        func.argtypes = []
+        func.restype = c_int32
+        terrain_element_version = func()
+        assert terrain_element_version == 1
         self.seed = seed
         self.device = device
         self.surface_registry = surface_registry

infinigen/terrain/source/cpu/elements/core.cpp

@@ -43,5 +43,8 @@ extern "C" {
     void cleanup() {
     }
 
-
+    int get_version() {
+        return 1;
+    }
+
 }