Only include capnp pyuda files when capnp is enabled.

ukaea · Feb 6, 2024 · 6e04681 · 6e04681
1 parent 2c30f60
commit 6e04681
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/source/wrappers/python/CMakeLists.txt b/source/wrappers/python/CMakeLists.txt
@@ -1,6 +1,4 @@
-file( GLOB CYTHON_FILES
-  pyuda/cpyuda/capnp.pxd
-  pyuda/cpyuda/capnp_tree.pyx
+set( CYTHON_FILES
   pyuda/cpyuda/client.pyx
   pyuda/cpyuda/cpyuda.pyx
   pyuda/cpyuda/dim.pyx
@@ -12,6 +10,12 @@ file( GLOB CYTHON_FILES
   pyuda/cpyuda/types.pyx
   pyuda/cpyuda/uda.pxd
 )
+set( CYTHON_SOURCES pyuda/cpyuda/cpyuda.pyx )
+
+if( ENABLE_CAPNP )
+  list( APPEND CYTHON_FILES pyuda/cpyuda/capnp.pxd pyuda/cpyuda/capnp_tree.pyx  )
+  list( APPEND CYTHON_SOURCES pyuda/cpyuda/capnp_tree.pyx )
+endif( )
 
 install( FILES ${CYTHON_FILES} DESTINATION python_installer/pyuda/cpyuda )
 

diff --git a/source/wrappers/python/setup.py.in b/source/wrappers/python/setup.py.in
@@ -47,7 +47,7 @@ else:
 
 ext = Extension(
     'cpyuda',
-    ['pyuda/cpyuda/cpyuda.pyx', 'pyuda/cpyuda/capnp_tree.pyx'],
+    '@CYTHON_SOURCES@'.split(';'),
     include_dirs=list(filter(None, [
         '/usr/include/tirpc',
         '@TIRPC_PYUDA_INCLUDE_DIR@',