working on primitive basis of a monomial

aplK.py

@@ -544,7 +544,7 @@ class SimplicialAplN(AplN, UniqueRepresentation):
             terms = []
             apl = self.parent()
             for b in basis:
-                dim = len(monomial)//2 - 1
+                dim = len(b)//2 - 1
                 terms.append(self._get_term(1, b, apl.gens()))
             terms.sort(reverse=False)
             return terms

test_primitive.py

@@ -0,0 +1,74 @@
+from operator import itemgetter
+load('aplK.py')
+
+s3 = SimplicialAlgebraplN(3)
+s3.inject_variables()
+q = y1*y2*y3
+p = q.propagate([0,1,2,3],4)
+p = p.reduce()
+s4 = SimplicialAlgebraplN(4)
+s4.inject_variables()
+#p = -t3*y0*y1*y2 + t2*y0*y1*y3 - t1*y0*y2*y3 + t0*y1*y2*y3
+#p = p.reduce()
+
+## def _get_primitive_basis(self):
+basis = []
+for monomial in p.dict().keys():
+    basis += p._get_primitive_basis_of_a_monomial(monomial)
+basis = list(set(basis))
+terms = []
+apl = p.parent()
+for b in basis:
+    dim = len(b)//2 - 1
+    terms.append(p._get_term(1, b, apl.gens()))
+terms.sort(reverse=False)
+
+with open("output/primitive_basis_without_kernel:terms.txt", 'w') as fp:
+    print("original polynomial: " + str(p),file=fp)
+    print(terms,file=fp)
+            
+#print("primitive basis without kernel terms")
+#print(terms)
+
+kernel_terms = []
+kernel_terms += p._add_kernel_terms_to_primitive_basis(basis)
+kernel_terms = list(set(kernel_terms))
+terms = []
+apl = p.parent()
+for b in kernel_terms:
+    dim = len(b)//2 - 1
+    terms.append(p._get_term(1, b, apl.gens()))
+terms.sort(reverse=False)
+
+#print("kernel terms")
+#print(terms)
+
+def get_kernel_terms(p, kernel_basis, parent, degree):
+
+    kernel_terms = []
+    for k in kernel_basis:
+        dim = len(k)//2 - 1
+        term = p._get_term(1, k, parent.gens())
+        e = k[5:]
+        gens_idx = []
+        for idx,ex in enumerate(e):
+            if ex:
+                gens_idx.append(idx)
+        list_ring_gens = [itemgetter(*gens_idx)(parent.gens())]*degree
+        sets = list_ring_gens+[[term]]
+        cp = cartesian_product(sets)
+        kernel_terms += [prod(e) for e in cp.list()]
+    return list(set(kernel_terms))
+        
+        
+
+basis = []
+
+for degree in range(5):
+    basis = get_kernel_terms(p, kernel_terms, s4, degree)    
+    with open("output/kernel_terms_degree_"+str(degree)+".txt", 'w') as fp:
+        for b in basis:
+            print(b,file=fp)
+#implicit fp.close() here as "with" scope ends.
+
+