feat(cli): expose the greedy assignment algorithm in the CLI

closes #35

README.md

@@ -61,7 +61,8 @@ usage: phomo [-h] [-o OUTPUT] [-c MASTER_CROP_RATIO]
              [-S TILE_SIZE [TILE_SIZE ...]] [-n N_APPEARANCES] [-b] [-g]
              [-d SUBDIVISIONS [SUBDIVISIONS ...]] [-G]
              [-m {greyscale,norm,luv_approx}] [-j WORKERS] [-e]
-             [--match-master-to-tiles] [--match-tiles-to-master] [-v]
+             [--match-master-to-tiles] [--match-tiles-to-master] [--greedy]
+             [-v]
              master tile_dir
 
 positional arguments:
@@ -102,6 +103,8 @@ options:
   --match-tiles-to-master
                         Match the tile images' colour distribution with the
                         master image colours.
+  --greedy              Use a greedy tile assignment algorithm. Should improve
+                        performance at the expense of accuracy.
   -v, --verbose         Verbosity.
 ```
 

phomo/__main__.py

@@ -130,6 +130,11 @@ def parse_args(args: List[str]) -> argparse.Namespace:
         help="Match the tile images' colour distribution with the master image colours.",
         action="store_true",
     )
+    parser.add_argument(
+        "--greedy",
+        help="Use a greedy tile assignment algorithm. Should improve performance at the expense of accuracy.",
+        action="store_true",
+    )
     parser.add_argument(
         "-v",
         "--verbose",
@@ -213,7 +218,9 @@ def main():
             if not args.gpu
             else mosaic.d_matrix_cuda(metric=args.metric)
         )
-        mosaic_im = mosaic.build(d_matrix)
+        mosaic_im = (
+            mosaic.build_greedy(d_matrix) if args.greedy else mosaic.build(d_matrix)
+        )
         if args.output is None:
             mosaic_im.show()
         else: