xtrack 0.69.2__tar.gz → 0.69.3__tar.gz

{xtrack-0.69.2/xtrack.egg-info → xtrack-0.69.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xtrack
-Version: 0.69.2
+Version: 0.69.3
 Summary: Tracking library for particle accelerators
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xtrack

{xtrack-0.69.2 → xtrack-0.69.3}/tests/test_environment.py

@@ -1782,3 +1782,74 @@ def test_select_in_multiline():
             == "((-vars['acbch7.r1b1']) + vars['aaa'])")
     assert line_sel.get('mcbch.7r1.b1').knl[0] == 1e-6
     assert line.get('mcbch.7r1.b1').knl[0] == 1e-6
+
+@pytest.mark.parametrize('container_type', ['env', 'line'])
+def test_inpection_methods(container_type):
+
+    env = xt.Environment()
+
+    env.vars({
+        'k.1': 1.,
+        'a': 2.,
+        'b': '2 * a + k.1',
+    })
+
+    line = env.new_line([
+        env.new('bb', xt.Bend, k0='2 * b', length=3+env.vars['a'] + env.vars['b'],
+            h=5., ksl=[0, '3*b']),
+    ])
+
+    ee = {'env': env, 'line': line}[container_type]
+
+    # Line/Env methods (get, set, eval, get_expr, new_expr, info)
+    assert ee.get('b') == 2 * 2 + 1
+    assert ee.get('bb') is env.element_dict['bb']
+
+    assert str(ee.get_expr('b')) == "((2.0 * vars['a']) + vars['k.1'])"
+
+    assert ee.eval('3*a - sqrt(k.1)') == 5
+
+    ne = ee.new_expr('sqrt(3*a + 3)')
+    assert xd.refs.is_ref(ne)
+    assert str(ne) == "f.sqrt(((3.0 * vars['a']) + 3.0))"
+
+    ee.info('bb') # Check that it works
+    ee.info('b')
+    ee.info('a')
+
+    ee.set('c', '6*a')
+    assert ee.get('c') == 6 * 2
+
+    # Line/Env containers (env[...], env.ref[...]
+    assert ee['b'] == 2 * 2 + 1
+    assert type(ee['bb']).__name__ == 'View'
+    assert ee['bb'].__class__.__name__ == 'Bend'
+
+    # Vars methods (get, set, eval, get_expr, new_expr, info, get_table)
+    assert ee.vars.get('b') == 2 * 2 + 1
+
+    assert str(ee.vars.get_expr('b')) == "((2.0 * vars['a']) + vars['k.1'])"
+
+    assert ee.vars.eval('3*a - sqrt(k.1)') == 5
+
+    ne = ee.vars.new_expr('sqrt(3*a + 3)')
+    assert xd.refs.is_ref(ne)
+    assert str(ne) == "f.sqrt(((3.0 * vars['a']) + 3.0))"
+
+    ee.vars.info('b')
+    ee.vars.info('a')
+
+    ee.vars.set('d', '7*a')
+    assert ee.vars.get('d') == 7 * 2
+
+    assert xd.refs.is_ref(ee.vars['b'])
+
+    # View methods get_expr, get_value, get_info, get_table (for now)
+    assert xd.refs.is_ref(ee['bb'].get_expr('k0'))
+    assert str(ee['bb'].get_expr('k0')) == "(2.0 * vars['b'])"
+    assert ee['bb'].get_expr('k0')._value == 2 * (2 * 2 + 1)
+    assert ee['bb'].get_value('k0') == 2 * (2 * 2 + 1)
+
+    tt = ee['bb'].get_table()
+    assert tt['value', 'k0'] == 2 * (2 * 2 + 1)
+    assert tt['expr', 'k0'] == "(2.0 * vars['b'])"

{xtrack-0.69.2 → xtrack-0.69.3}/tests/test_survey.py

@@ -12,6 +12,11 @@ tilted = True
 orientation = 'acw'
 transform_to_actual_elements = True
 
+if hasattr(np, 'trapezoid'): # numpy >= 2.0
+    trapz = np.trapezoid
+else:
+    trapz = np.trapz
+
 @for_all_test_contexts
 @pytest.mark.parametrize(
     'slice_mode',
@@ -103,17 +108,17 @@ def test_survey_slicing(test_context, slice_mode, tilted, orientation,
 
     if not tilted and orientation == 'acw':
         assert_allclose(np.abs(sv.Y), 0, rtol=0, atol=1e-14)
-        assert_allclose(np.trapezoid(sv.X, sv.Z), -4.818 , rtol=0, # anti-clockwise
+        assert_allclose(trapz(sv.X, sv.Z), -4.818 , rtol=0, # anti-clockwise
                         atol=(2e-3 if slice_mode is not None else 0.5))
     elif not tilted and orientation == 'cw':
         assert_allclose(np.abs(sv.Y), 0, rtol=0, atol=1e-14)
-        assert_allclose(np.trapezoid(sv.X, sv.Z), 4.818 , rtol=0, # clockwise
+        assert_allclose(trapz(sv.X, sv.Z), 4.818 , rtol=0, # clockwise
                         atol=(2e-3 if slice_mode is not None else 0.5))
     elif tilted and orientation == 'acw':
         assert_allclose(np.abs(sv.X), 0, rtol=0, atol=1e-14)
-        assert_allclose(np.trapezoid(sv.Y, sv.Z), -4.818 , rtol=0, # anti-clockwise
+        assert_allclose(trapz(sv.Y, sv.Z), -4.818 , rtol=0, # anti-clockwise
                         atol=(2e-3 if slice_mode is not None else 0.5))
     elif tilted and orientation == 'cw':
         assert_allclose(np.abs(sv.X), 0, rtol=0, atol=1e-14)
-        assert_allclose(np.trapezoid(sv.Y, sv.Z), 4.818 , rtol=0, # clockwise
+        assert_allclose(trapz(sv.Y, sv.Z), 4.818 , rtol=0, # clockwise
                         atol=(2e-3 if slice_mode is not None else 0.5))

xtrack-0.69.3/xtrack/_version.py

@@ -0,0 +1 @@
+__version__ = '0.69.3'

{xtrack-0.69.2 → xtrack-0.69.3}/xtrack/environment.py

@@ -40,6 +40,12 @@ class Environment:
         self._drift_counter = 0
         self.ref = EnvRef(self)
 
+    @property
+    def manager(self):
+        if not hasattr(self, '_var_management') or self._var_management is None:
+            self._init_var_management()
+        return self._var_management['manager']
+
     def new_line(self, components=None, name=None):
         out = xt.Line()
         out.particle_ref = self.particle_ref
@@ -198,6 +204,11 @@ Environment.replace_replica = xt.Line.replace_replica
 Environment.__getitem__ = xt.Line.__getitem__
 Environment.set = xt.Line.set
 Environment.get = xt.Line.get
+Environment.eval = xt.Line.eval
+Environment.info = xt.Line.info
+Environment.get_expr = xt.Line.get_expr
+Environment.new_expr = xt.Line.new_expr
+
 
 class Place:
 
@@ -356,7 +367,7 @@ def _resolve_s_positions(seq_all_places, env):
             aux_s_center.append(ss)
     aux_tt['s_center'] = np.concatenate([aux_s_center, [0]])
 
-    i_sorted = np.argsort(aux_s_center, stable=True)
+    i_sorted = np.argsort(aux_s_center, kind='stable')
 
     name_sorted = [str(aux_tt.name[ii]) for ii in i_sorted]
 

{xtrack-0.69.2 → xtrack-0.69.3}/xtrack/line.py

@@ -3405,7 +3405,8 @@ class Line:
 
     def __add__(self, other):
         self._env_if_needed
-        assert isinstance(other, Line), 'Only Line can be added to Line'
+        #assert isinstance(other, Line), 'Only Line can be added to Line'
+        assert other.__class__.__name__=="Line", 'Only Line can be added to Line'
         assert other.env is self.env, 'Lines must be in the same environment'
         out = self.env.new_line(
             components=list(self.element_names) + list(other.element_names))
@@ -3555,6 +3556,35 @@ class Line:
         else:
             raise KeyError(f'Element or variable {key} not found')
 
+    def info(self, key, limit=12):
+        if key in self.element_dict:
+            return self[key].get_info()
+        elif key in self.vars:
+            return self.vars.info(key, limit=limit)
+        else:
+            raise KeyError(f'Element or variable {key} not found')
+
+#    def get_value(self, key):
+#        if key in self.element_dict:
+#            return self.element_dict[key].get_value()
+#        elif key in self.vars:
+#            return self.vars.get_value(key)
+#        else:
+#            raise KeyError(f'Element or variable {key} not found')
+
+    @property
+    def manager(self):
+        return self._xdeps_manager
+
+    def eval(self, expr):
+        return self.vars.eval(expr)
+
+    def new_expr(self, expr):
+        return self.vars.new_expr(expr)
+
+    def get_expr(self, vars):
+        return self.vars.get_expr(vars)
+
     def _env_if_needed(self):
         if not hasattr(self, 'env') or self.env is None:
             self.env = xt.Environment(element_dict=self.element_dict,
@@ -4700,26 +4730,17 @@ class LineVars:
 
         return xd.Table({'name': name, 'value': value, 'expr': expr})
 
-    def expr(self, var):
-        raise NotImplementedError # Untested
-        if isinstance(var,str):
-            ref=self.line._xdeps_vref[var]
-        elif is_expr(var):
-            ref=var
-        else:
-            raise ValueError(f"`{var}` not valid, must be str or expr")
-        expr=ref._expr
-        if expr is None:
-            raise NameError(f"`{var}` does not have any expression")
-        return expr
+    def new_expr(self, expr):
+        return self.line._xdeps_eval.eval(expr)
 
     def eval(self, expr):
-        raise NotImplementedError # Untested
-        return self.line._xdeps_eval.eval(expr)
+        return self.new_expr(expr)._get_value()
 
-    def value(self, expr):
-        raise NotImplementedError # Untested
-        return eval(self)._get_value()
+    def info(self, var, limit=10):
+        return self[var]._info(limit=limit)
+
+    def get_expr(self, var):
+        return self[var]._expr
 
     def __contains__(self, key):
         if self.line._xdeps_vref is None:
@@ -4844,6 +4865,9 @@ class LineVars:
         else:
             self[name] = value
 
+    def get(self, name):
+        return self[name]._value
+
 class ActionVars(Action):
 
     def __init__(self, line):

{xtrack-0.69.2 → xtrack-0.69.3/xtrack.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xtrack
-Version: 0.69.2
+Version: 0.69.3
 Summary: Tracking library for particle accelerators
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xtrack

xtrack-0.69.2/xtrack/_version.py

@@ -1 +0,0 @@
-__version__ = '0.69.2'