@@ -69,11 +69,9 @@ def __init__(self, **kwargs):
6969 if '_xobject' not in kwargs :
7070 try : # TODO: small workaround until PR
7171 self .compile_kernels (use_prebuilt_kernels = use_prebuilt_kernels ,
72- particles_class = xp .Particles ,
73- only_if_needed = True )
72+ only_if_needed = True )
7473 except TypeError :
75- self .compile_kernels (particles_class = xp .Particles ,
76- only_if_needed = True )
74+ self .compile_kernels (only_if_needed = True )
7775 self ._context .kernels .EverestBlock_set_material (el = self )
7876
7977
@@ -88,7 +86,7 @@ def material(self, material):
8886 raise ValueError ("Invalid material!" )
8987 if not xt .line ._dicts_equal (self .material .to_dict (), material .to_dict ()):
9088 self ._material = material
91- self .compile_kernels (particles_class = xp . Particles , only_if_needed = True )
89+ self .compile_kernels (only_if_needed = True )
9290 self ._context .kernels .EverestBlock_set_material (el = self )
9391
9492 def get_backtrack_element (self , _context = None , _buffer = None , _offset = None ):
@@ -141,11 +139,9 @@ def __init__(self, **kwargs):
141139 if '_xobject' not in kwargs :
142140 try : # TODO: small workaround until PR
143141 self .compile_kernels (use_prebuilt_kernels = use_prebuilt_kernels ,
144- particles_class = xp .Particles ,
145- only_if_needed = True )
142+ only_if_needed = True )
146143 except TypeError :
147- self .compile_kernels (particles_class = xp .Particles ,
148- only_if_needed = True )
144+ self .compile_kernels (only_if_needed = True )
149145 self ._context .kernels .EverestCollimator_set_material (el = self )
150146
151147 @property
@@ -159,7 +155,7 @@ def material(self, material):
159155 raise ValueError ("Invalid material!" )
160156 if not xt .line ._dicts_equal (self .material .to_dict (), material .to_dict ()):
161157 self ._material = material
162- self .compile_kernels (particles_class = xp . Particles , only_if_needed = True )
158+ self .compile_kernels (only_if_needed = True )
163159 self ._context .kernels .EverestCollimator_set_material (el = self )
164160
165161 def get_backtrack_element (self , _context = None , _buffer = None , _offset = None ):
@@ -242,11 +238,9 @@ def __init__(self, **kwargs):
242238 self ._bending_radius = self .active_length / np .sin (bending_angle )
243239 try : # TODO: small workaround until PR
244240 self .compile_kernels (use_prebuilt_kernels = use_prebuilt_kernels ,
245- particles_class = xp .Particles ,
246- only_if_needed = True )
241+ only_if_needed = True )
247242 except TypeError :
248- self .compile_kernels (particles_class = xp .Particles ,
249- only_if_needed = True )
243+ self .compile_kernels (only_if_needed = True )
250244 self ._context .kernels .EverestCrystal_set_material (el = self )
251245
252246
@@ -296,7 +290,7 @@ def material(self, material):
296290 raise ValueError ("Invalid material!" )
297291 if not xt .line ._dicts_equal (self .material .to_dict (), material .to_dict ()):
298292 self ._material = material
299- self .compile_kernels (particles_class = xp . Particles , only_if_needed = True )
293+ self .compile_kernels (only_if_needed = True )
300294 self ._context .kernels .EverestCrystal_set_material (el = self )
301295
302296
@@ -313,18 +307,3 @@ def _lattice_setter(lattice):
313307 else :
314308 raise ValueError (f"Illegal value { lattice } for 'lattice'! "
315309 + "Only use 'strip' (110) or 'quasi-mosaic' (111)." )
316-
317-
318- # TODO: We want this in the HybridClass to get Kernels attached automatically,
319- # like the PerParticlePyMethod in BeamElement
320- # def _exec_kernel(el, kernel_name, **kwargs):
321- # # context = el._context
322- # # desired_classes = tuple(a.atype for a in el._kernels[kernel_name].args)
323- # # if (kernel_name, desired_classes) not in context.kernels:
324- # # el.compile_kernels(particles_class=xp.Particles)
325- # # kern = context.kernels[(kernel_name, desired_classes)]
326- # # return kern(el=el._xobject, **kwargs)
327- # el.compile_kernels(particles_class=xp.Particles, only_if_needed=True)
328- # return getattr(el._context.kernels, kernel_name)(el=el, **kwargs)
329-
330-
0 commit comments