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

xcoll/beam_elements/base.py

@@ -629,7 +629,7 @@ class BaseCollimator(BaseBlock):
             raise ValueError("When using `twiss`, need to provide the name as well.")
         else:
             tw_up   = twiss.rows[name]
-            tw_down = twiss.rows[twiss.mask[[name]]+1]
+            tw_down = twiss.rows[twiss.rows.indices[[name]]+1]
         if not np.isclose(tw_up.s[0] + self.length, tw_down.s[0]):
             raise ValueError(f"Downstream twiss not compatible with length {self.length}m.")
         self._optics = {

xcoll/beam_elements/blowup.py

@@ -61,6 +61,8 @@ class BlowUp(InvalidXcoll):
     @classmethod
     def install(cls, line, name, *, at_s=None, at=None, need_apertures=True, aperture=None, s_tol=1.e-6, **kwargs):
         self = cls(**kwargs)
+        if name in line.element_names:
+            raise ValueError(f"Element {name} already exists in the line as {line[name].__class__.__name__}.")
         line.insert_element(element=self, name=name, at_s=at_s, at=at, s_tol=s_tol)
         self._name = name
         self._line = line

xcoll/beam_elements/monitor.py

@@ -137,6 +137,8 @@ class EmittanceMonitor(xt.BeamElement):
     @classmethod
     def install(cls, line, name, *, at_s=None, at=None, s_tol=1.e-6, **kwargs):
         self = cls(**kwargs)
+        if name in line.element_names:
+            raise ValueError(f"Element {name} already exists in the line as {line[name].__class__.__name__}.")
         line.insert_element(element=self, name=name, at_s=at_s, at=at, s_tol=s_tol)
         self._name = name
         self._line = line

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.1'
+__version__ = '0.5.3'
 # ======================

xcoll/initial_distribution.py

@@ -61,13 +61,13 @@ def generate_pencil_on_collimator(line, name, num_particles, *, side='+-', penci
     if is_converging:
         # pencil at front of jaw
         match_at_s = s_front
-        sigma = beam_sizes.rows[name:f'{name}%%1'][f'sigma_{plane}'][0]
-        sigma_transv = beam_sizes.rows[name:f'{name}%%1'][f'sigma_{transv_plane}'][0]
+        sigma = beam_sizes.rows[name:f'{name}>>1'][f'sigma_{plane}'][0]
+        sigma_transv = beam_sizes.rows[name:f'{name}>>1'][f'sigma_{transv_plane}'][0]
     else:
         # pencil at back of jaw
         match_at_s = s_back
-        sigma = beam_sizes.rows[name:f'{name}%%1'][f'sigma_{plane}'][1]
-        sigma_transv = beam_sizes.rows[name:f'{name}%%1'][f'sigma_{transv_plane}'][1]
+        sigma = beam_sizes.rows[name:f'{name}>>1'][f'sigma_{plane}'][1]
+        sigma_transv = beam_sizes.rows[name:f'{name}>>1'][f'sigma_{transv_plane}'][1]
     dr_sigmas = pencil_spread/sigma
 
     # Generate 4D coordinates
@@ -157,7 +157,7 @@ def generate_delta_from_dispersion(line, at_element, *, plane, position_mm, nemi
         twiss = line.twiss()
 
     beam_sizes = twiss.get_beam_covariance(nemitt_x=nemitt_x, nemitt_y=nemitt_y)
-    beam_sizes = beam_sizes.rows[at_element:f'{at_element}%%1'][f'sigma_{plane}']
+    beam_sizes = beam_sizes.rows[at_element:f'{at_element}>>1'][f'sigma_{plane}']
     sigma = beam_sizes[0] if match_at_front else beam_sizes[1]
     delta  = (position_mm - betatron_cut*sigma - twiss.rows[at_element][plane])
     delta /= twiss.rows[at_element][f'd{plane}']

xcoll/line_tools.py

@@ -26,9 +26,9 @@ def get_optics_at(names, *, twiss=None, line=None):
         twiss = line.twiss()
     if not hasattr(names, '__iter__') and not isinstance(names, str):
         names = [names]
-    coll_entry_mask = twiss.mask[names]
-    tw_entry = twiss.rows[coll_entry_mask]
-    tw_exit = twiss.rows[coll_entry_mask+1]
+    coll_entry_indices = twiss.rows.indices[names]
+    tw_entry = twiss.rows[coll_entry_indices]
+    tw_exit = twiss.rows[coll_entry_indices+1]
     tw_exit.name = tw_entry.name
     return tw_entry, tw_exit
 

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

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

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

@@ -3,8 +3,8 @@ NOTICE,sha256=6DO_E7WCdRKc42vUoVVBPGttvQi4mRt9fAcxj9u8zy8,74
 xcoll/__init__.py,sha256=dUv7QzbojU0SfOiKPe6-XThev2ir9jhqzRIUhTz8y84,1021
 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
+xcoll/beam_elements/base.py,sha256=vX3fvRgGa-penwdpUnyHIa8WMIwzcbFDx2S1Ezv7F7E,50922
+xcoll/beam_elements/blowup.py,sha256=XtcMNe_tLpa9-8ayHEM2hdbFf76In4bTljc5VrsKDhs,8029
 xcoll/beam_elements/elements_src/black_absorber.h,sha256=ERDortSn5pVFec-9b9DWkHQEkgfHKeUDNqeJ6-DU5Zg,5027
 xcoll/beam_elements/elements_src/black_crystal.h,sha256=rrrQ9RW8ltiZ47-w2ZZM91iLBEwn4C5SJrwQiiXBJ4M,4622
 xcoll/beam_elements/elements_src/blowup.h,sha256=kV3BTUJNy9Gt-pAEcUrueJWGL7x9YNHKSncTw0Bo8vg,1564
@@ -13,18 +13,18 @@ xcoll/beam_elements/elements_src/everest_block.h,sha256=qRU8g6Ux6Q6igWecSaRzclrp
 xcoll/beam_elements/elements_src/everest_collimator.h,sha256=JPf3LPAYuqFYcE7xC-Esntz3gWEIZLn1gO5e_y0pYj0,10365
 xcoll/beam_elements/elements_src/everest_crystal.h,sha256=UCLq62yKoKq-qrfrX8naWUf0TlU-KN-Nnk8w1ow1nzw,12524
 xcoll/beam_elements/everest.py,sha256=jlPDmUq11DFxJXuT1f-GVYI18aKo7PMwzGjnPa8Yqls,8367
-xcoll/beam_elements/monitor.py,sha256=yInXQYVDM_9opcp7CyywUBsAJbSr3B18ZjkKmlOK7b4,16778
+xcoll/beam_elements/monitor.py,sha256=l8zhnNngXf5QtxI6YXdXdzDYQrk9MIX25Ok1B9-kobk,16928
 xcoll/colldb.py,sha256=rSSR_j19tsPAWBGZYnFVgOMEP2fbywhfXA0WiVN8Wkw,31027
-xcoll/general.py,sha256=FKB0NMJHEYV4RVjv52MBDT0x0EhKFF2T817Ae3N1oOU,534
+xcoll/general.py,sha256=ALO_2bZQivwBSLcLyS54g6XXorOGFZcyseud1ycEGo4,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/initial_distribution.py,sha256=RWPwBqusqgdBxeeIgJiyZK4bfiOJ4KCmAalkx6zQxEY,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=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
+xcoll/line_tools.py,sha256=Zt1VpBwtF_5zm-OMaEYOg6zdxLgk7ZsnCXkalQ41pJQ,3270
 xcoll/lossmap.py,sha256=VOck3CFvra0BUgKzzP2ddmN8bSnSMUXQg01FCmeSBrU,8091
 xcoll/manager.py,sha256=Uy58saX2836q_CZjNa5at611k38Za_L9jQSTqBQXooA,343
 xcoll/rf_sweep.py,sha256=5ABhou9cZcLYQ8DSPYowPpLCea8zJQyhWgUa7JUHJIg,8936
@@ -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.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,,
+xcoll-0.5.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+xcoll-0.5.3.dist-info/METADATA,sha256=7kLj8aDAZNYLA3voys1yUBfS8C8NSBrd1tiq5AIxV6o,2709
+xcoll-0.5.3.dist-info/NOTICE,sha256=6DO_E7WCdRKc42vUoVVBPGttvQi4mRt9fAcxj9u8zy8,74
+xcoll-0.5.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+xcoll-0.5.3.dist-info/RECORD,,