xcoll 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

xcoll/beam_elements/__init__.py

@@ -10,7 +10,8 @@ from .blowup import BlowUp
 from .monitor import EmittanceMonitor
 
 block_classes = tuple(v for v in globals().values()
-                      if isinstance(v, type) and issubclass(v, BaseBlock) and v != BaseBlock)
+                      if isinstance(v, type) and issubclass(v, BaseBlock) and v != BaseBlock
+                      and v != BaseCollimator and v != BaseCrystal)
 # Includes crystals
 collimator_classes = tuple(v for v in globals().values()
                            if isinstance(v, type) and (issubclass(v, BaseCollimator) or issubclass(v, BaseCrystal))
@@ -18,7 +19,7 @@ collimator_classes = tuple(v for v in globals().values()
 crystal_classes = tuple(v for v in globals().values()
                         if isinstance(v, type) and issubclass(v, BaseCrystal) and v != BaseCrystal)
 
-element_classes = block_classes + collimator_classes + (BlowUp, EmittanceMonitor)
+element_classes = block_classes + (BlowUp, EmittanceMonitor)
 
 _all_collimator_classes = collimator_classes
 _all_crystal_classes    = crystal_classes

xcoll/general.py

@@ -12,5 +12,5 @@ citation = "F.F. Van der Veken, et al.: Recent Developments with the New Tools f
 # ======================
 # Do not change
 # ======================
-__version__ = '0.5.0'
+__version__ = '0.5.1'
 # ======================

xcoll/interaction_record/interaction_record.py

@@ -264,7 +264,7 @@ class InteractionRecord(xt.BeamElement):
 
 
 def _get_xcoll_elements(line=None, elements=None, names=None):
-    from xcoll import element_classes
+    from xcoll.beam_elements import block_classes
     if names is not None and names is not False and \
     (not hasattr(names, '__iter__') or isinstance(names, str)):
         names = [names]
@@ -278,7 +278,7 @@ def _get_xcoll_elements(line=None, elements=None, names=None):
         if elements is not None and elements is not False:
             raise ValueError("Cannot provide both line and elements!")
         if names is None or names is True:
-            elements, names = line.get_elements_of_type(element_classes)
+            elements, names = line.get_elements_of_type(block_classes)
             if len(names) == 0:
                 raise ValueError("No Xcoll elements in line!")
         elif names is False:
@@ -291,7 +291,7 @@ def _get_xcoll_elements(line=None, elements=None, names=None):
                     raise ValueError(f"Element {name} not found in line!")
             elements = [line[name] for name in names]
     for idx, element in enumerate(elements):
-        if not isinstance(element, element_classes):
+        if not isinstance(element, block_classes):
             name = name[idx] if names is not None else element.__class__.__name__
             raise ValueError(f"Element {name} not an Xcoll element!")
     return elements, names

{xcoll-0.5.0.dist-info → xcoll-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xcoll
-Version: 0.5.0
+Version: 0.5.1
 Summary: Xsuite collimation package
 Home-page: https://github.com/xsuite/xcoll
 License: Apache 2.0

{xcoll-0.5.0.dist-info → xcoll-0.5.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 NOTICE,sha256=6DO_E7WCdRKc42vUoVVBPGttvQi4mRt9fAcxj9u8zy8,74
 xcoll/__init__.py,sha256=dUv7QzbojU0SfOiKPe6-XThev2ir9jhqzRIUhTz8y84,1021
-xcoll/beam_elements/__init__.py,sha256=6DENPooYBVZfonhOoqSA5u22OF1B-07QHUcUa6y7ZQs,1164
+xcoll/beam_elements/__init__.py,sha256=gJRYzcDjoimFrHiLvYQYLDtP_5A97Qbg0XL6q-j2uC4,1210
 xcoll/beam_elements/absorber.py,sha256=teLGFiO22S8ItXa3ftsjg7oAN7Ypw0iJhkJmjbtGlL4,2502
 xcoll/beam_elements/base.py,sha256=UJHjC4-EQd7j_cfJZI7eJ9ADgQFZB1XPyrErHmchcVA,50914
 xcoll/beam_elements/blowup.py,sha256=0YdKTzMRt_ABGyFwHvnnYFBq0oHMs0ezHrU3hUTMp7Q,7879
@@ -15,13 +15,13 @@ xcoll/beam_elements/elements_src/everest_crystal.h,sha256=UCLq62yKoKq-qrfrX8naWU
 xcoll/beam_elements/everest.py,sha256=jlPDmUq11DFxJXuT1f-GVYI18aKo7PMwzGjnPa8Yqls,8367
 xcoll/beam_elements/monitor.py,sha256=yInXQYVDM_9opcp7CyywUBsAJbSr3B18ZjkKmlOK7b4,16778
 xcoll/colldb.py,sha256=rSSR_j19tsPAWBGZYnFVgOMEP2fbywhfXA0WiVN8Wkw,31027
-xcoll/general.py,sha256=NRBkZprYIqgiP1o6pgFimSu6bl_azPL_F5yXPsjoFsc,534
+xcoll/general.py,sha256=FKB0NMJHEYV4RVjv52MBDT0x0EhKFF2T817Ae3N1oOU,534
 xcoll/headers/checks.h,sha256=IMLhQMnh3t-t9UhVK-1hfS4GVgl2Mry69H_17By00M0,1582
 xcoll/headers/particle_states.h,sha256=WCzeHAniEK7sX7HTwhoIgewioQ7SibGn01RLrhVI98k,840
 xcoll/initial_distribution.py,sha256=RlBWGm1e_qUF3svz4id77aDxl4TFxMuP65LZcepxC8U,8404
 xcoll/install.py,sha256=v9TdrKa7hjwFH25WN8mQ21Yy6qV3jW3NAH4YhBvEibw,7685
 xcoll/interaction_record/__init__.py,sha256=UFoLiKa-z2oX7YoszP-7Vgdt1nM6kT382v1CaIu8_u0,50
-xcoll/interaction_record/interaction_record.py,sha256=MKibjRZYcLGYlY1wDSTtVKZV7m9Z8SK9p3K-dEk-U_Q,13184
+xcoll/interaction_record/interaction_record.py,sha256=QzkK27-KAztWD__3l97dB-GvNrB6RJgZnKmGVMppDQ4,13192
 xcoll/interaction_record/interaction_record_src/interaction_record.h,sha256=YCJLbk1dUcuj-I935K8iFwqDDja4afGB9bJ5jNWJSpo,6067
 xcoll/interaction_record/interaction_types.py,sha256=_UGq-PBq30hj1lp29CpNTmmwzsP7tuCng8ZIAALnzCc,2894
 xcoll/line_tools.py,sha256=MSNQly2_7SxXPM68qGGnGm_HGW78Llu_kqguE04P69o,3253
@@ -406,8 +406,8 @@ xcoll/scattering_routines/geometry/objects.h,sha256=lNFEfqrSDnJuB6sKh59bbg52bGqO
 xcoll/scattering_routines/geometry/rotation.h,sha256=AvNwcqAAlhjEUtSbTqi53BTvzR256gtfUYcM5jlXUco,701
 xcoll/scattering_routines/geometry/segments.h,sha256=IAz4IWvarIIpX7Waph-Hh2j6o6n_2uCPmHxxQp8Q3yg,7667
 xcoll/scattering_routines/geometry/sort.h,sha256=wNdQLwEn58jgeSG4B9x-jfm7vOjN4hnAlxZ3SG-SkWg,5763
-xcoll-0.5.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-xcoll-0.5.0.dist-info/METADATA,sha256=2cwlzCRry48_7NRFQF8i3ceGIzns1j-1dE4bMHIqDDQ,2709
-xcoll-0.5.0.dist-info/NOTICE,sha256=6DO_E7WCdRKc42vUoVVBPGttvQi4mRt9fAcxj9u8zy8,74
-xcoll-0.5.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-xcoll-0.5.0.dist-info/RECORD,,
+xcoll-0.5.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+xcoll-0.5.1.dist-info/METADATA,sha256=5m3ZpHN5fKSZ0iANDHgpi8N20s5C4xxkMpmP8weLuQw,2709
+xcoll-0.5.1.dist-info/NOTICE,sha256=6DO_E7WCdRKc42vUoVVBPGttvQi4mRt9fAcxj9u8zy8,74
+xcoll-0.5.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+xcoll-0.5.1.dist-info/RECORD,,