schubmult 2.0.0__py3-none-any.whl

schubmult/__init__.py

@@ -0,0 +1 @@
+__version__ = "2.0.0"

schubmult/_base_argparse.py

@@ -0,0 +1,174 @@
+from argparse import ArgumentParser, SUPPRESS, RawDescriptionHelpFormatter
+
+
+def schub_argparse(prog_name, description, quantum=False, yz=False):
+    parser = ArgumentParser(
+        prog=prog_name,
+        description=description,
+        epilog=f"""Example:
+        {prog_name} 5 1 7 3 2 6 4 - 2 1 6 3 5 4 {"" if not yz else "[ --display-positive]"}
+            or equivalently
+        {prog_name} --code 4 0 4 1 0 1 - 1 0 3 0 1 {"" if not yz else "[ --display-positive]"}
+        {"    or alternatively" if not quantum else ""}
+        {prog_name if not quantum else ""} {"--coprod --code 2 0 3 0 1 - 2 4 " if not quantum else ""} {"" if not yz or quantum else "[ --display-positive]"}
+    """,
+        formatter_class=RawDescriptionHelpFormatter,
+    )
+
+    parser.add_argument(
+        "perms",
+        nargs="+",
+        action="append",
+        help="Space-delimited permutations separated by hyphens, e. g. 3 4 1 2 - 5 1 2 4 3",
+        metavar="hyphen-separated list of perms",
+    )
+    parser.add_argument("-", nargs="+", action="append", dest="perms", help=SUPPRESS)
+
+    parser.add_argument(
+        "-np",
+        "--no-print",
+        action="store_false",
+        default=True,
+        dest="pr",
+        help="Compute the result but do not print it",
+    )
+
+    parser.add_argument(
+        "--code",
+        action="store_true",
+        default=False,
+        dest="ascode",
+        help="Permutations represented by the Lehmer code",
+    )
+
+    parser.add_argument(
+        "--mult",
+        nargs="+",
+        required=False,
+        default=None,
+        help="Some additional terms in the ring to multiply by",
+    )
+
+    if not quantum:
+        parser.add_argument(
+            "--coprod",
+            action="store_true",
+            default=False,
+            help="Compute the coproduct (different syntax: one permutation, then a hyphen, the variable index positions to split on)",
+        )
+
+    if yz:
+        parser.add_argument(
+            "--display-positive",
+            action="store_true",
+            default=False,
+            dest="display_positive",
+            help="Display the result in terms of the positive roots, or if mixed variable attempt to display the result as a positive algebraic combination of terms of the form y_i - z_j",
+        )
+
+        parser.add_argument(
+            "--optimizer-message",
+            action="store_true",
+            default=False,
+            dest="msg",
+            help="Display debug output during integer optimization for --display-positive",
+        )
+
+        parser.add_argument(
+            "--down", action="store_true", default=False, help="Reverse multiplication"
+        )
+
+        parser.add_argument(
+            "-nc",
+            "--no-check",
+            action="store_false",
+            default=True,
+            dest="check",
+            help="Do not check if positive result matches the original",
+        )
+
+        parser.add_argument(
+            "--mixed-var",
+            action="store_false",
+            dest="same",
+            help="Used mixed variables y and z",
+        )
+
+        parser.add_argument(
+            "--expand",
+            action="store_true",
+            default=False,
+            dest="expa",
+            help="Expand the output rather than leaving it as originally computed (slow)",
+        )
+
+    if quantum:
+        parser.add_argument(
+            "--parabolic",
+            nargs="+",
+            required=False,
+            default=[],
+            help="Generators of the parabolic subgroup to compute quantum coeffs for",
+        )
+
+        parser.add_argument(
+            "--basic-pieri",
+            action="store_true",
+            default=False,
+            dest="slow",
+            help="Do not apply conjectural computation optimization to quantum",
+        )
+        if yz:
+            parser.add_argument(
+                "--nil-hecke",
+                type=int,
+                required=False,
+                default=None,
+                dest="nilhecke",
+                metavar="N",
+                help="Substitute up to N of Fomin-Gelfand-Postnikov commuting difference operators",
+            )
+
+            parser.add_argument(
+                "--nil-hecke-apply",
+                type=int,
+                required=False,
+                default=None,
+                dest="nilhecke_apply",
+                metavar="N",
+                help="Substitute commuting difference operators for perm1, then apply to Schub indexed by perm2",
+            )
+
+    parser.add_argument(
+        "--display-mode",
+        type=str,
+        required=False,
+        choices=["basic", "pretty", "latex"],
+        default="basic",
+        dest="disp_mode",
+        help="Method of displaying the output. Default basic",
+    )
+
+    args = parser.parse_args()
+    args.mulstring = ""
+
+    if args.mult is not None:
+        args.mulstring = " ".join(args.mult)
+        args.mult = True
+
+    for perm in args.perms:
+        try:
+            for i in range(len(perm)):
+                perm[i] = int(perm[i])
+        except Exception as e:
+            print("Permutations must have integer values")
+            raise e
+    import sympy
+
+    formatter = lambda bob: str(bob)  # noqa: E731
+    if args.disp_mode == "latex":
+        formatter = lambda bob: sympy.latex(sympy.sympify(bob)).replace("\\left", "").replace("\\right", "")  # noqa: E731
+    elif args.disp_mode == "pretty":
+        formatter = lambda bob: sympy.pretty(sympy.sympify(bob))  # noqa: E731
+
+    return args, formatter