Fix randomizer example

tmontaigu · Oct 4, 2020 · eb54a46 · eb54a46
1 parent c4ab401
commit eb54a46
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/examples/field-randomizer.py b/examples/field-randomizer.py
@@ -7,8 +7,6 @@
 
 
 def main(args):
-    print(args.exclude)
-
     files = [args.in_path] if args.in_path.is_file() else list(args.in_path.glob("*.la[s-z]"))
 
     if args.out_path.suffix and len(files) > 1:
@@ -19,7 +17,7 @@ def main(args):
 
         las = pylas.read(path)
 
-        dimensions = (dim for dim in las.point_format.dimensions if dim not in args.exclude)
+        dimensions = (dim for dim in las.point_format.dimensions if dim.name not in args.exclude)
         for dimension in dimensions:
             print(f"\t{dimension.name}", end='')
             if dimension.kind == pylas.DimensionKind.FloatingPoint:
@@ -34,7 +32,10 @@ def main(args):
             las[dimension.name] = np.random.randint(dimension.min, dimension.max + 1, len(las.points), type_str)
             print()
 
-        las.write(args.out_path / path.name)
+        if args.out_path.suffix:
+            las.write(args.out_path)
+        else:
+            las.write(args.out_path / path.name)
 
 
 if __name__ == '__main__':

diff --git a/examples/recursive-split.py b/examples/recursive-split.py
@@ -1,5 +1,6 @@
 import argparse
 import sys
+from pathlib import Path
 from typing import List
 from typing import Optional
 
@@ -36,7 +37,7 @@ def main():
     parser.add_argument("input_file")
     parser.add_argument("output_dir")
     parser.add_argument("size", type=tuple_size, help="eg: 50x64.17")
-    parser.add_argument("--points-per-iter", default=10**6)
+    parser.add_argument("--points-per-iter", default=10**6, type=int)
 
     args = parser.parse_args()
 
@@ -57,7 +58,7 @@ def main():
                 print(f"{count / file.header.point_count * 100}%")
 
                 # For performance we need to use copy
-                # so that the arrays underlying arrays are contiguous
+                # so that the underlying arrays are contiguous
                 x, y = points.x.copy(), points.y.copy()
 
                 point_piped = 0
@@ -67,7 +68,8 @@ def main():
 
                     if np.any(mask):
                         if writers[i] is None:
-                            writers[i] = pylas.open(f"{sys.argv[2]}/output_{i}.laz",
+                            output_path = Path(sys.argv[2]) / f"output_{i}.laz"
+                            writers[i] = pylas.open(output_path,
                                                     mode='w',
                                                     header=file.header)
                         sub_points = points[mask]
@@ -77,6 +79,7 @@ def main():
                     if point_piped == len(points):
                         break
                 count += len(points)
+            print(f"{count / file.header.point_count * 100}%")
         finally:
             for writer in writers:
                 if writer is not None: