Add format option to c2e cli

sunset1995 · Dec 16, 2024 · 7435e6c · 7435e6c
1 parent 030a426
commit 7435e6c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/py360convert/__main__.py b/py360convert/__main__.py
@@ -42,6 +42,7 @@ def main():
     c2e_parser.add_argument("--help", action="help", help="Show this help message and exit")
     c2e_parser.add_argument("input", type=Path, help="Path to input image.")
     c2e_parser.add_argument("output", type=Path, help="Path to output image.")
+    c2e_parser.add_argument("--format", "-f", choices=["horizon", "dice"], default="dice", help="Input image layout.")
     c2e_parser.add_argument("--height", "-h", type=int, required=True, help="Output image height in pixels.")
     c2e_parser.add_argument("--width", "-w", type=int, required=True, help="Output image width in pixels.")
     c2e_parser.add_argument(
@@ -98,7 +99,7 @@ def main():
     elif args.command == "c2e":
         _assert_height_width(args)
         img = np.array(Image.open(args.input))
-        out = py360convert.c2e(img, h=args.height, w=args.width, mode=args.mode)  # pyright: ignore[reportCallIssue]
+        out = py360convert.c2e(img, h=args.height, w=args.width, mode=args.mode, cube_format=args.format)  # pyright: ignore[reportCallIssue]
         Image.fromarray(out).save(args.output)
     elif args.command == "e2c":
         _size_to_dims(args)