ign-pdal-tools 1.7.10__py3-none-any.whl → 1.7.11__py3-none-any.whl

{ign_pdal_tools-1.7.10.dist-info → ign_pdal_tools-1.7.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ign-pdal-tools
-Version: 1.7.10
+Version: 1.7.11
 Summary: Library for common LAS files manipulation with PDAL
 Author-email: Guillaume Liegard <guillaume.liegard@ign.fr>
 Description-Content-Type: text/markdown

{ign_pdal_tools-1.7.10.dist-info → ign_pdal_tools-1.7.11.dist-info}/RECORD

@@ -1,7 +1,7 @@
-ign_pdal_tools-1.7.10.dist-info/licenses/LICENSE.md,sha256=iVzCFZTUXeiqP8bP474iuWZiWO_kDCD4SPh1Wiw125Y,1120
-pdaltools/_version.py,sha256=2hMdkmuq8cBrwb79NJB3d8dhjK2v9egMpm_KrdD0S3o,75
+ign_pdal_tools-1.7.11.dist-info/licenses/LICENSE.md,sha256=iVzCFZTUXeiqP8bP474iuWZiWO_kDCD4SPh1Wiw125Y,1120
+pdaltools/_version.py,sha256=lXuB5yCoqB_XdA0YVaGjTGc2J_sFF0RE_IJePTMnyjI,75
 pdaltools/add_points_in_las.py,sha256=TGbt5JUkszjmbQiA2LCUntsjz6A8DHb7QPIXGDuEgWA,3643
-pdaltools/add_points_in_pointcloud.py,sha256=lstU-iO4fkHAm8gqNSEyc3sELApqMLLvtJQ9IDQmUrM,12328
+pdaltools/add_points_in_pointcloud.py,sha256=UHdC8tFagsehGggkLJCcmBCi8gknJOL4s99f_2TJjsA,12538
 pdaltools/color.py,sha256=-a0ramyLMkZX4-M3siPq1zIio5u-a0p8UzvgcfoU6zU,8313
 pdaltools/las_add_buffer.py,sha256=sBpTywlfsHHS8KuCUa-eydB2hylshEvjrMQt5TrqXb8,11275
 pdaltools/las_clip.py,sha256=GvEOYu8RXV68e35kU8i42GwSkbo4P9TvmS6rkrdPmFM,1034
@@ -12,7 +12,7 @@ pdaltools/pcd_info.py,sha256=NIAH5KGikVDQLlbCcw9FuaPqe20UZvRfkHsDZd5kmZA,3210
 pdaltools/replace_attribute_in_las.py,sha256=po1F-fi8s7iilqKWaryW4JRbsmdMOUe0yGvG3AEKxtk,4771
 pdaltools/standardize_format.py,sha256=gqm2GJbtDkT4k4oC_NX2LIPh9R2BLh4sMHLKYgfKrMc,3973
 pdaltools/unlock_file.py,sha256=pIThdWMNkTph0xgJVVRaM1o9aUMQhM6804PscScB3JI,1963
-ign_pdal_tools-1.7.10.dist-info/METADATA,sha256=8JpuSz74R6IiEbQ1XvonH1en-FI1gdzMNTzp-8Li1yY,5745
-ign_pdal_tools-1.7.10.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-ign_pdal_tools-1.7.10.dist-info/top_level.txt,sha256=KvGW0ZzqQbhCKzB5_Tp_buWMZyIgiO2M2krWF_ecOZc,10
-ign_pdal_tools-1.7.10.dist-info/RECORD,,
+ign_pdal_tools-1.7.11.dist-info/METADATA,sha256=Bh4cFjnGLLt4DpiPw8GQuknhDTrLQqx3H5EjkFzV1n4,5745
+ign_pdal_tools-1.7.11.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+ign_pdal_tools-1.7.11.dist-info/top_level.txt,sha256=KvGW0ZzqQbhCKzB5_Tp_buWMZyIgiO2M2krWF_ecOZc,10
+ign_pdal_tools-1.7.11.dist-info/RECORD,,

pdaltools/_version.py

@@ -1,4 +1,4 @@
-__version__ = "1.7.10"
+__version__ = "1.7.11"
 
 
 if __name__ == "__main__":

pdaltools/add_points_in_pointcloud.py

@@ -1,5 +1,5 @@
 import argparse
-import shutil
+from shutil import copy2
 
 import geopandas as gpd
 import laspy
@@ -127,25 +127,22 @@ def add_points_to_las(
         crs (str): CRS of the data.
         virtual_points_classes (int): The classification value to assign to those virtual points (default: 66).
     """
-
     if input_points_with_z.empty:
         print(
             "No points to add. All points of the geojson file are outside the tile. Copying the input file to output"
         )
-        shutil.copy(input_las, output_las)
-
         return
 
     # Extract XYZ coordinates and additional attribute (classification)
+    nb_points = len(input_points_with_z.geometry.x)
     x_coords = input_points_with_z.geometry.x
     y_coords = input_points_with_z.geometry.y
     z_coords = input_points_with_z.geometry.z
-    classes = virtual_points_classes * np.ones(len(input_points_with_z.index))
+    classes = virtual_points_classes * np.ones(nb_points)
 
-    with laspy.open(input_las, mode="r") as las:
-        las_data = las.read()
+    # Open the input LAS file to check and possibly update the header of the output
+    with laspy.open(input_las) as las:
         header = las.header
-
         if not header:
             header = laspy.LasHeader(point_format=8, version="1.4")
         if crs:
@@ -155,20 +152,21 @@ def add_points_to_las(
                 raise ValueError(f"Invalid CRS: {crs}")
             header.add_crs(crs_obj)
 
-        # Append new points
-        new_x = np.concatenate([las_data.x, x_coords])
-        new_y = np.concatenate([las_data.y, y_coords])
-        new_z = np.concatenate([las_data.z, z_coords])
-        new_classes = np.concatenate([las_data.classification, classes])
-
-        updated_las = laspy.LasData(header)
-        updated_las.x = new_x
-        updated_las.y = new_y
-        updated_las.z = new_z
-        updated_las.classification = new_classes
-
-        with laspy.open(output_las, mode="w", header=header, do_compress=True) as writer:
-            writer.write_points(updated_las.points)
+    # Copy data pointcloud
+    copy2(input_las, output_las)
+
+    # Add the new points with 3D points
+    nb_points = len(x_coords)
+    with laspy.open(output_las, mode="a", header=header) as output_las:  # mode `a` for adding points
+        # create nb_points points with "0" everywhere
+        new_points = laspy.ScaleAwarePointRecord.zeros(nb_points, header=header)  # use header for input_las
+        # then fill in the gaps (X, Y, Z an classification)
+        new_points.x = x_coords.astype(new_points.x.dtype)
+        new_points.y = y_coords.astype(new_points.y.dtype)
+        new_points.z = z_coords.astype(new_points.z.dtype)
+        new_points.classification = classes.astype(new_points.classification.dtype)
+
+        output_las.append_points(new_points)
 
 
 def line_to_multipoint(line, spacing: float, z_value: float = None):