Merge main

Author: GDeLaurentis

lips/algebraic_geometry/covariant_ideal.py

@@ -1,6 +1,9 @@
 import re
 import numpy
 import sympy
+import functools
+
+from collections import defaultdict
 
 from lips.tools import flatten
 from lips.algebraic_geometry.tools import lips_covariant_symbols, lips_invariant_symbols, conversionIdeal
@@ -30,7 +33,7 @@ def __init__(self, ring_or_multiplicity, generators_or_covariants, momentum_cons
             oParticles.make_analytical_d()
             generators = []
             for covariant in generators_or_covariants:
-                poly_or_polys = 4 * oParticles(covariant)
+                poly_or_polys = 4 * oParticles(covariant)  # TODO: remove 4 * when https://github.com/sympy/sympy/pull/28139 is accepted
                 if hasattr(poly_or_polys, 'shape'):
                     polys = flatten(poly_or_polys)
                     for poly in polys:
@@ -57,7 +60,7 @@ def __contains__(self, covariant):
         oParticles = Particles(self.multiplicity)
         oParticles.make_analytical_d()
         try:
-            poly_or_polys = 4 * oParticles(covariant)
+            poly_or_polys = 4 * oParticles(covariant)  # TODO: remove 4 * when https://github.com/sympy/sympy/pull/28139 is accepted
         except TypeError:
             poly_or_polys = covariant
         if isinstance(poly_or_polys, numpy.ndarray):
@@ -76,6 +79,16 @@ def image(self, rule):
         newIdeal.rule = rule
         return newIdeal
 
+    @functools.cached_property
+    def equivalenceClass(self):
+        from ..symmetries import all_symmetries
+        all_perms_self = {symmetry: self(*symmetry) for symmetry in all_symmetries(self.multiplicity)}
+        value_to_keys = defaultdict(list)
+        for k, v in all_perms_self.items():
+            value_to_keys[v].append(k)
+        distinct_perms_self = {keys[0]: val for val, keys in value_to_keys.items()}
+        return distinct_perms_self
+
     def to_mom_cons_qring(self):
         oZeroIdeal = LipsIdeal(self.multiplicity, ())
         self.to_qring(oZeroIdeal)

lips/particle.py

@@ -51,6 +51,12 @@ def __init__(self, kinematics=None, real_momentum=False, field=Field('mpc', 0, 3
         else:
             raise Exception('Bad Particle Constructor')
 
+    def to_field(self, field):
+        self._r_sp_d = numpy.vectorize(field)(self.r_sp_d)
+        self._r_sp_d_to_r_sp_u()
+        self.l_sp_d = numpy.vectorize(field)(self.l_sp_d)
+        self.field = field
+
     def __eq__(self, other):
         """Equality: checks equality of four momenta."""
         if isinstance(other, Particle):

lips/particles.py

@@ -73,6 +73,11 @@ def __init__(self, number_of_particles_or_particles=None, seed=None, real_moment
         else:
             raise Exception(f"Internal masses not understood, received {internal_masses} of type {type(internal_masses)}.")
 
+    def to_field(self, field):
+        for oP in self:
+            oP.to_field(field)
+        self.field = field
+
     def __setattr__(self, name, value):
         if pMVar.match(name):
             self.internal_masses.add(name)

lips/symmetries.py

@@ -29,3 +29,7 @@ def inverse(permutation_or_symmetry):
 
 def identity(n):
     return (''.join(str(i) for i in range(1, n + 1)), False)
+
+
+def all_symmetries(multiplicity):
+    return phase_weights_compatible_symmetries([0] * multiplicity)

tests/test_particles.py

@@ -151,3 +151,13 @@ def test_hashes_does_not_change_under_identity_mapping():
     # cache_key_hash2 = hashlib.sha256(cache_key_bytes).hexdigest()
     assert hash(oPs) == hash(oPsMapped)
     # assert cache_key_hash1 == cache_key_hash2  # this fails and I'm unsure how to fix it atm
+
+
+def test_field_change():
+    Fp = Field("finite field", 2 ** 31 - 1, 1)
+    oPsFp = Particles(6, field=Field("rational", 0, 0), seed=0)
+    oPsFp.to_field(Fp)
+    Qp = Field("padic", 2 ** 31 - 1, 4)
+    oPsQp = Particles(6, field=Field("rational", 0, 0), seed=0)
+    oPsQp.to_field(Qp)
+    assert oPsFp("⟨1|2⟩") == oPsQp("⟨1|2⟩").as_tuple[0]