Adapt to introduction of tracking header

Author: szymonlopaciuk

xfields/beam_elements/temp_slicer.py

@@ -2,6 +2,7 @@
 from scipy import special
 import xobjects as xo
 import xpart as xp
+import xtrack as xt
 from ..general import _pkg_root
 
 _digitize_kernel = xo.Kernel(
@@ -106,6 +107,13 @@ def __init__(self, _context=None,
 
         self.compile_kernels(only_if_needed=False)
 
+    def compile_kernels(self, *args, **kwargs):
+        extra_compile_args = kwargs.pop('extra_compile_args', [])
+        extra_compile_args.append(f'-I{xt.__path__[0]}')
+        kwargs['extra_compile_args'] = extra_compile_args
+
+        return super().compile_kernels(*args, **kwargs)
+
     def rho(self, z):
         """
         Gaussian charge density.

xfields/fieldmaps/interpolated.py

@@ -215,7 +215,10 @@ def __init__(self,
                  dphi_dy = nelem,
                  dphi_dz = nelem)
 
-        self.compile_kernels(only_if_needed=True)
+        self.compile_kernels(
+            only_if_needed=True,
+            extra_compile_args=[f"-I{xt.__path__[0]}"],
+        )
 
         if isinstance(solver, str):
             self.solver = self.generate_solver(solver, fftplan)
@@ -237,7 +240,6 @@ def __init__(self,
     def _assert_updatable(self):
         assert self.updatable, 'This FieldMap is not updatable!'
 
-    #@profile
     def get_values_at_points(self,
             x, y, z,
             return_rho=True,

xfields/fieldmaps/tricubicinterpolated.py

@@ -233,6 +233,13 @@ def __init__(self,
                 if solver is not None and rho is not None:
                     self.update_phi_from_rho()
 
+    def compile_kernels(self, *args, **kwargs):
+        extra_compile_args = kwargs.pop('extra_compile_args', [])
+        extra_compile_args.append(f'-I{xt.__path__[0]}')
+        kwargs['extra_compile_args'] = extra_compile_args
+
+        return super().compile_kernels(*args, **kwargs)
+
     def _assert_updatable(self):
         assert self.updatable, 'This FieldMap is not updatable!'