Make rerun_demo default to the cube

.github/workflows/python.yml

@@ -185,8 +185,8 @@ jobs:
         # If you change the line below you should almost definitely change the `key:` line above by giving it a new, unique name
         run: |
           pip install -r examples/python/colmap/requirements.txt
-          python examples/python/colmap/main.py --dataset colmap_fiat --resize 800x600 --save colmap_fiat.rrd
-          cp colmap_fiat.rrd rerun_py/rerun_sdk/rerun_demo/demo.rrd
+          python examples/python/colmap/main.py --dataset colmap_fiat --resize 800x600 --save colmap.rrd
+          cp colmap.rrd rerun_py/rerun_sdk/rerun_demo/colmap.rrd
 
       - name: Build Wheel
         uses: PyO3/maturin-action@v1

rerun_py/pyproject.toml

@@ -44,6 +44,6 @@ target-version = ["py38"]
 # with the other `rerun` pypi package. The rerun_sdk.pth adds this to the pythonpath
 # which then allows `import rerun` to work as expected.
 # See https://github.com/rerun-io/rerun/pull/1085 for more details
-include = ["rerun_sdk.pth", "rerun_sdk/rerun_demo/demo.rrd"]
+include = ["rerun_sdk.pth", "rerun_sdk/rerun_demo/colmap.rrd"]
 locked = true
 python-packages = ["rerun_sdk/rerun", "rerun_sdk/rerun_demo"]

rerun_py/rerun_sdk/rerun_demo/__main__.py

@@ -1,22 +1,62 @@
 """Demo program which loads an rrd file built into the package."""
 
+import argparse
 import pathlib
 import sys
 
-from rerun import bindings, unregister_shutdown  # type: ignore[attr-defined]
 
+def run_cube():
+    import rerun as rr
+
+    rr.init("Cube", spawn=True, default_enabled=True)
+    from rerun_demo.data import color_grid
+
+    rr.log_points("cube", positions=color_grid.positions, colors=color_grid.colors, radii=0.5)
+
+
+def run_colmap():
+    from rerun import bindings, unregister_shutdown  # type: ignore[attr-defined]
 
-def main() -> None:
     # We don't need to call shutdown in this case. Rust should be handling everything
     unregister_shutdown()
 
-    rrd_file = pathlib.Path(__file__).parent.joinpath("demo.rrd").resolve()
+    rrd_file = pathlib.Path(__file__).parent.joinpath("colmap.rrd").resolve()
     if not rrd_file.exists():
         print("No demo file found at {}. Package was built without demo support".format(rrd_file), file=sys.stderr)
         exit(1)
     else:
         exit(bindings.main([sys.argv[0], str(rrd_file)]))
 
 
+def main() -> None:
+
+    parser = argparse.ArgumentParser(description="Run rerun example programs")
+
+    group = parser.add_mutually_exclusive_group()
+
+    group.add_argument(
+        "--cube",
+        action="store_true",
+        help="Run the color grid cube demo",
+    )
+
+    group.add_argument(
+        "--colmap",
+        action="store_true",
+        help="Run the COLMAP data demo",
+    )
+
+    args = parser.parse_args()
+
+    if not any([args.cube, args.colmap]):
+        args.cube = True
+
+    if args.cube:
+        run_cube()
+
+    elif args.colmap:
+        run_colmap()
+
+
 if __name__ == "__main__":
     main()

rerun_py/rerun_sdk/rerun_demo/data.py

@@ -10,7 +10,7 @@
 ColorGrid = namedtuple("ColorGrid", ["positions", "colors"])
 
 
-def build_color_grid(x_count=6, y_count=6, z_count=6):
+def build_color_grid(x_count=10, y_count=10, z_count=10):
     """
     Create a cube of points with colors.