xcoll 0.6.5__py3-none-any.whl → 0.6.7__py3-none-any.whl

xcoll/colldb.py

@@ -36,7 +36,7 @@ def _dict_keys_to_lower(dct):
     else:
         return dct
 
-
+# TODO: need better handling of beam argument (not hardcoded 'b1' and 'b2' strings)
 def _get_coll_dct_by_beam(coll, beam):
     # The dictionary can be a CollimatorDatabase for a single beam (beam=None)
     # or for both beams (beam='b1' or beam='b2)
@@ -51,7 +51,7 @@ def _get_coll_dct_by_beam(coll, beam):
             raise ValueError("Need to specify a beam, because the given dict is for both beams!")
         return coll[beam]
 
-    elif len(beam_in_db) == 1:
+    elif len(beam_in_db) == 1 and beam_in_db[0].lower() in ['b1','b2']:
         if beam is None:
             beam = beam_in_db[0].lower()
         elif beam != beam_in_db[0].lower():
@@ -551,7 +551,8 @@ class CollimatorDatabase:
         prop_dict.update(kwargs)
         el = cls(**prop_dict)
         el.emittance = [self.nemitt_x, self.nemitt_y]
-        if 'family' in self[name] and self[name]['family'].lower() != 'unknown':
+        if 'family' in self[name] and self[name]['family'] \
+        and self[name]['family'].lower() != 'unknown':
             if self[name]['family'] == name:
                 raise ValueError(f"Collimator {name} has the same name as its family!")
             el.family = self[name]['family']

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.6.5'
+__version__ = '0.6.7'
 # ======================

{xcoll-0.6.5.dist-info → xcoll-0.6.7.dist-info}/METADATA

@@ -1,8 +1,10 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: xcoll
-Version: 0.6.5
+Version: 0.6.7
 Summary: Xsuite collimation package
 License: Apache-2.0
+License-File: LICENSE
+License-File: NOTICE
 Author: Frederik F. Van der Veken
 Author-email: frederik@cern.ch
 Requires-Python: >=3.10
@@ -12,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Provides-Extra: tests
 Requires-Dist: numpy (>=1.0)
 Requires-Dist: pandas (>=1.4)
@@ -20,9 +23,9 @@ Requires-Dist: pytest-html ; extra == "tests"
 Requires-Dist: pytest-xdist ; extra == "tests"
 Requires-Dist: ruamel-yaml (>=0.17.31,<0.18.15)
 Requires-Dist: xdeps (>=0.10.6)
-Requires-Dist: xobjects (>=0.5.2)
+Requires-Dist: xobjects (>=0.5.3)
 Requires-Dist: xpart (>=0.23.1)
-Requires-Dist: xtrack (>=0.90.1)
+Requires-Dist: xtrack (>=0.93.0)
 Project-URL: Bug Tracker, https://github.com/xsuite/xsuite/issues
 Project-URL: Documentation, https://xsuite.readthedocs.io/
 Project-URL: Homepage, https://github.com/xsuite/xcoll

{xcoll-0.6.5.dist-info → xcoll-0.6.7.dist-info}/RECORD

@@ -10,17 +10,13 @@ xcoll/beam_elements/elements_src/emittance_monitor.h,sha256=vlHzQFoUjKQHWBmzpaSz
 xcoll/beam_elements/elements_src/everest_block.h,sha256=ITSzfiqwm8ZNIkAj_odQkAjKhPcdT8b0AUlq_uxtsqI,6562
 xcoll/beam_elements/elements_src/everest_collimator.h,sha256=fkD5GazVv2zlmqyb6s9zVNMq6baXeFb_pf7gg9TKjNc,9787
 xcoll/beam_elements/elements_src/everest_crystal.h,sha256=LISp_9NEWMIi5GvZ3LwPfeSbmsbXvHWcTqxW6eCCsiE,11951
-xcoll/beam_elements/elements_src/rf_sweep.h,sha256=eTPPSJvFrkoh9STI0hmmiozwMUL-TQVhz1mOZQhObWE,846
 xcoll/beam_elements/elements_src/transparent_collimator.h,sha256=pA_Fxx85N87iSbpHSjR2OSOx6JN9e38T1pOUKrC7zxs,4610
 xcoll/beam_elements/elements_src/transparent_crystal.h,sha256=DGX4uL0hpm4njOypoPuEWtMzYxnL9jQkf2ZrGaS0alI,4128
 xcoll/beam_elements/everest.py,sha256=iJva_S0OQnmG-qj7rIAiUCssu5WL_Qs5P_n6Qx-IWzo,9080
 xcoll/beam_elements/monitor.py,sha256=baMYT_PNya8KX35ReeMwk5Cm04ePUqiDF_s69L5GTX0,16968
-xcoll/beam_elements/sweep.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 xcoll/beam_elements/transparent.py,sha256=AZOWfH3dvsetL6DjQGejUgX7T18e_3omIIffG_1gwc0,2894
-xcoll/colldb.py,sha256=WaJC6gkDNVvxo65OhucTM6Fwo6pgRFHNnBnHSkiX7AQ,31054
-xcoll/general.py,sha256=RyE6HAGBJF0DRK5n4X3-2o4fbRQwBZIV-LRjUo5-n8A,534
-xcoll/general.py.orig,sha256=vIx4LhKIwbjcC8cyk6wtxrjuihDEF1THgzwb7_dpgJU,609
-xcoll/geometry.md,sha256=AARDOk5lZWFBAllX75UYMRywSYSB04FIlYWE0ra1t5c,1623
+xcoll/colldb.py,sha256=4QwvWH-KYTa6xyaVS7Y8S473T6RHXOER8B2xZjeNahI,31213
+xcoll/general.py,sha256=NYF0rBViyMS_GaNBuPF8_Me6rLrMZBwn6npYxpLSLG0,534
 xcoll/headers/checks.h,sha256=qdXsOTBOK1MwW6bdFF93j4yE648mcDtEv5rGN1w9sfk,1582
 xcoll/headers/particle_states.py,sha256=N8ehLtg99358I4RGCJ3sjuKCk2LVdGqgVx2xGoHrRdM,1959
 xcoll/initial_distribution.py,sha256=CHhqeCVjVOyaBWhfy2sDgLwIgtXZoqcTW-su-M2YByE,10989
@@ -61,9 +57,9 @@ xcoll/scattering_routines/geometry/rotation.h,sha256=lO3RaQBys9r0ROMjR8T8Rr7UsIE
 xcoll/scattering_routines/geometry/segments.h,sha256=7nKnnin2ByxkKyaYwGvFaqgLQg5uBba4CdLHL7L3iQs,7667
 xcoll/scattering_routines/geometry/sort.h,sha256=b1MkFO2ddzv1fWGeQzsLuz46qo2pKyRSXHjoAEVU7Ts,5763
 xcoll/xaux.py,sha256=mh4E_jhaUFm2LOKawJ0b6X4zuz8JiJhx6w7GcnRgIXQ,2291
-xcoll-0.6.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-xcoll-0.6.5.dist-info/METADATA,sha256=3HdRmdvs2oyskur291fR-oZf2lbRZAIRoX6fNyY0iuA,2953
-xcoll-0.6.5.dist-info/NOTICE,sha256=6DO_E7WCdRKc42vUoVVBPGttvQi4mRt9fAcxj9u8zy8,74
-xcoll-0.6.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-xcoll-0.6.5.dist-info/entry_points.txt,sha256=fC9oojAU6WjKujnGBodo7ZcEeBGfSfeLPO5287echdY,26
-xcoll-0.6.5.dist-info/RECORD,,
+xcoll-0.6.7.dist-info/METADATA,sha256=-8wXPMjf3s99eSTuM08jv9pYu1rkts0MRDrmgneUTRc,3047
+xcoll-0.6.7.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+xcoll-0.6.7.dist-info/entry_points.txt,sha256=fC9oojAU6WjKujnGBodo7ZcEeBGfSfeLPO5287echdY,26
+xcoll-0.6.7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+xcoll-0.6.7.dist-info/licenses/NOTICE,sha256=6DO_E7WCdRKc42vUoVVBPGttvQi4mRt9fAcxj9u8zy8,74
+xcoll-0.6.7.dist-info/RECORD,,

{xcoll-0.6.5.dist-info → xcoll-0.6.7.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any

xcoll/beam_elements/elements_src/rf_sweep.h

@@ -1,23 +0,0 @@
-// copyright ############################### #
-// This file is part of the Xcoll package.   #
-// Copyright (c) CERN, 2025.                 #
-// ######################################### #
-
-#ifndef XCOLL_RF_SWEEP_H
-#define XCOLL_RF_SWEEP_H
-
-/*gpufun*/
-void RFSweep_track_local_particle(RFSweepData el, LocalParticle* part0){
-    double  const step = RFSweepData_get__rf_sweep_df_step(el); // L*df/(f0 + df)
-    int64_t const start_turn = RFSweepData_get_start_turn(el);
-    int64_t const stop_turn = RFSweepData_get_stop_turn(el);
-
-    //start_per_particle_block (part0->part)
-        int64_t at_turn = LocalParticle_get_at_turn(part);
-        if (at_turn >= start_turn && at_turn <= stop_turn){
-            LocalParticle_add_to_zeta(part, -(at_turn - start_turn + 1)*step);
-        }
-    //end_per_particle_block
-}
-
-#endif /* XCOLL_RF_SWEEP_H */

xcoll/general.py.orig

@@ -1,21 +0,0 @@
-# copyright ############################### #
-# This file is part of the Xcoll package.   #
-# Copyright (c) CERN, 2025.                 #
-# ######################################### #
-
-from pathlib import Path
-
-_pkg_root = Path(__file__).parent.absolute()
-
-citation  = "F.F. Van der Veken, et al.: Recent Developments with the New Tools for Collimation "
-citation += "Simulations in Xsuite, Proceedings of HB2023, Geneva, Switzerland."
-
-# ======================
-# Do not change
-# ======================
-<<<<<<< HEAD
-__version__ = '0.7.0rc0'
-=======
-__version__ = '0.6.4'
->>>>>>> main
-# ======================

xcoll/geometry.md

@@ -1,42 +0,0 @@
-PSEUDOCODE for material interactions tracking in collimator
-===========================================================
-
-- create DriftTrajectory from particle
-- loop:
-    - find first crossing with collimator shape (segments)
-    - if distance larger than end of BeamElement -> move particle to exendit and break loop
-    - move particle to collimator impact point
-    - create MCSTrajectory from new particle
-    - loop:
-        - find first crossing with collimator shape (segments)
-        - do not move, but calculate path length
-        - compare to distance to nuclear interaction
-        - shortest wins (nuclear or exit)
-        - move to next point
-    - when at exit, create DriftTrajectory from particle
-
-
-
-PSEUDOCODE for material interactions tracking in crystal
-========================================================
-
-- create DriftTrajectory from particle
-- loop:
-    - find first crossing with crystal shape (segments)
-    - if distance larger than end of BeamElement -> move particle to end and break loop
-    - move particle to crystal impact point
-    - create MCSTrajectory (if amorphous) or CircularTrajectory (if channelling) from new particle
-    - loop:
-        - find first crossing with crystal shape (segments)
-        - do not move, but calculate path length
-        - compare to distance to nuclear interaction and dechannelling
-        - shortest wins (nuclear, dechannelling, or exit)
-        - move to next point
-    - when at exit, create DriftTrajectory from particle
-
-
-PSEUDOCODE for crossing finder
-==============================
-
-- if DriftTrajectory and any segment
-=> use analytical