LipsIdeal equivalenceClass cached property

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
@@ -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/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)