schubmult 1.3.8__tar.gz → 1.3.9__tar.gz

{schubmult-1.3.8 → schubmult-1.3.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: schubmult
-Version: 1.3.8
+Version: 1.3.9
 Summary: Computing Littlewood-Richardson coefficients of Schubert polynomials
 Home-page: https://github.com/matthematics/schubmult
 Author: Matt Samuel

{schubmult-1.3.8 → schubmult-1.3.9}/schubmult/perm_lib.py

@@ -322,6 +322,13 @@ def add_perm_dict(d1,d2):
 zero = sympify(0)
 one = sympify(1)
 
+def elem_sym_poly_q(p,k,varl1,varl2):
+	if p == 0 and k>=0:
+		return one
+	if p<0 or p>k:
+		return zero
+	return (varl1[k-1] - varl2[k-p])*elem_sym_poly_q(p-1,k-1,varl1,varl2)+elem_sym_poly_q(p,k-1,varl1,varl2)+q_var[k-1]*elem_sym_poly_q(p-2,k-2,varl1,varl2)
+
 def elem_sym_poly(p,k,varl1,varl2,xstart=0,ystart=0):
 	global zero, one
 	if p>k:

{schubmult-1.3.8 → schubmult-1.3.9}/schubmult/schubmult_q_yz/schubmult_q_yz.py

@@ -10,6 +10,13 @@ var_y = var2.tolist()
 var_z = var3.tolist()
 var_x = symarray("x",100).tolist()
 
+x = var_x
+y = var_y
+z = var_z
+
+def E(p,k,varl=var_y[1:]):
+	return elem_sym_poly_q(p,k,var_x[1:],varl)
+
 def single_variable(coeff_dict,varnum):
 	ret = {}
 	for u in coeff_dict:
@@ -267,6 +274,8 @@ def main():
 		nil_N = 0
 		
 		mulstring = ""
+		norep = False
+		expa = False
 		
 		try:
 			for s in sys.argv[1:]:
@@ -274,6 +283,12 @@ def main():
 					just_nil = False
 					nil_N = int(s)
 					continue
+				if s == "--norep":
+					norep = True
+					continue
+				if s == "--expand":
+					expa = True
+					continue
 				if s == "-np" or s == "--no-print":
 					pr = False
 					continue
@@ -322,6 +337,8 @@ def main():
 				perms[i] = tuple(permtrim(uncode(perms[i])))
 		else:
 			for i in range(len(perms)):
+				if len(perms[i])<2 and (len(perms[i])==0 or perms[i][0]==1):
+					perms[i] = (1,2)
 				perms[i] = tuple(permtrim([*perms[i]]))
 		
 		size = 0
@@ -335,10 +352,15 @@ def main():
 			for perm in perms[1:]:
 				coeff_dict = schubmult(coeff_dict,perm)
 			if mult:
-				mul_exp = sympify(mulstring)
+				for v in var2:
+					globals()[str(v)] = v
+				for v in var3:
+					globals()[str(v)] = v	
+				for v in var_x:
+					globals()[str(v)] = v	
+				mul_exp = eval(mulstring)
 				coeff_dict = mult_poly(coeff_dict,mul_exp)
-			rep = ("","")
-			rep = ("","")
+			rep = ("","")			
 		
 		if pr:
 			if ascode:
@@ -398,11 +420,19 @@ def main():
 									print(f"error: value not equal; write to schubmult@gmail.com with the case {perms=} {perm=} {val=} {coeff_dict.get(perm,0)=}")
 									exit(1)
 							val = val2
+					if expa:
+						val = expand(val)
 					if val!=0:
 						if ascode:
-							print(f"{str(trimcode(perm)):>{width}}  {str(val).replace('**','^').replace('*',' ').replace(*rep)}")	
+							if norep:
+								print(f"{str(trimcode(perm)):>{width}}  {str(val).replace(*rep)}")	
+							else:
+								print(f"{str(trimcode(perm)):>{width}}  {str(val).replace('**','^').replace('*',' ').replace(*rep)}")	
 						else:
-							print(f"{str(perm):>{width}}  {str(val).replace('**','^').replace('*',' ').replace(*rep)}")	
+							if norep:
+								print(f"{str(perm):>{width}}  {str(val).replace(*rep)}")	
+							else:
+								print(f"{str(perm):>{width}}  {str(val).replace('**','^').replace('*',' ').replace(*rep)}")	
 	except BrokenPipeError:
 		pass
 		

{schubmult-1.3.8 → schubmult-1.3.9}/schubmult/schubmult_yz/schubmult_yz.py

@@ -28,6 +28,15 @@ var2 = tuple(symarray('y',n).tolist())
 var3 = tuple(symarray('z',n).tolist())
 
 var_x = symarray("x",100).tolist()
+var_y = var2
+var_z = var3
+
+x = var_x
+y = var_y
+z = var_z
+
+def E(p,k,varl=var_y[1:]):
+	return elem_sym_poly(p,k,var_x[1:],varl)
 
 def single_variable(coeff_dict,varnum):
 	ret = {}
@@ -1600,6 +1609,8 @@ def main():
 					perms[i] = tuple(permtrim(uncode(perms[i])))
 			else:
 				for i in range(len(perms)):
+					if len(perms[i])<2 and (len(perms[i])==0 or perms[i][0]==1):
+						perms[i] = (1,2)
 					perms[i] = tuple(permtrim([*perms[i]]))
 			
 			size = 0
@@ -1612,17 +1623,25 @@ def main():
 			coeff_dict = {perms[0]: 1}
 			check_coeff_dict = {perms[0]: 1}
 			
+			if mult:
+				for v in var2:
+					globals()[str(v)] = v
+				for v in var3:
+					globals()[str(v)] = v	
+				for v in var_x:
+					globals()[str(v)] = v	
+			
 			if down:
 				for perm in orig_perms[1:]:
 					check_coeff_dict = schubmult_down(check_coeff_dict,perm)
 				if mult:
-					mul_exp = sympify(mulstring)
+					mul_exp = eval(mulstring)
 					check_coeff_dict = mult_poly_down(check_coeff_dict,mul_exp)				
 			else:
 				for perm in orig_perms[1:]:
 					check_coeff_dict = schubmult(check_coeff_dict,perm)
 				if mult:
-					mul_exp = sympify(mulstring)
+					mul_exp = eval(mulstring)
 					check_coeff_dict = mult_poly(check_coeff_dict,mul_exp)
 			
 			

{schubmult-1.3.8 → schubmult-1.3.9}/schubmult.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: schubmult
-Version: 1.3.8
+Version: 1.3.9
 Summary: Computing Littlewood-Richardson coefficients of Schubert polynomials
 Home-page: https://github.com/matthematics/schubmult
 Author: Matt Samuel

{schubmult-1.3.8 → schubmult-1.3.9}/setup.py

@@ -6,7 +6,7 @@ long_description = (this_directory / "README.md").read_text()
 
 setup(
     name="schubmult",
-    version="1.3.8",
+    version="1.3.9",
     description="Computing Littlewood-Richardson coefficients of Schubert polynomials",
 	long_description=long_description,
 	long_description_content_type='text/markdown',