ign-pdal-tools 1.15.4__py3-none-any.whl → 1.15.5__py3-none-any.whl

{ign_pdal_tools-1.15.4.dist-info → ign_pdal_tools-1.15.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ign-pdal-tools
-Version: 1.15.4
+Version: 1.15.5
 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.15.4.dist-info → ign_pdal_tools-1.15.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
-ign_pdal_tools-1.15.4.dist-info/licenses/LICENSE.md,sha256=iVzCFZTUXeiqP8bP474iuWZiWO_kDCD4SPh1Wiw125Y,1120
-pdaltools/_version.py,sha256=v1-P-Ov-24Bdx7-zaJ7QUAVXDpuCxPkaM1l4VOXLOSA,75
+ign_pdal_tools-1.15.5.dist-info/licenses/LICENSE.md,sha256=iVzCFZTUXeiqP8bP474iuWZiWO_kDCD4SPh1Wiw125Y,1120
+pdaltools/_version.py,sha256=PMKGb38n4U-Bt7iqMv25IaET5dM26eUnM9VEKomGNDI,75
 pdaltools/add_points_in_pointcloud.py,sha256=UJTwoOjC0WKnp_ynNHpwUh1fmbIgw7hq5xoNN8_FxQQ,12965
 pdaltools/color.py,sha256=s-_rmLK6fIK3UwkUzHVZPEkm6r1LliG5ftGr-jkqyjM,9549
 pdaltools/create_random_laz.py,sha256=XuHH4G8Nrs8DB-F8bkcIeto7JtmrlrNGF_R66oxGCbQ,6069
@@ -12,11 +12,11 @@ pdaltools/las_merge.py,sha256=tcFVueV9X9nNEaoAl5zCduY5DETlBg63MAgP2SuKiNo,4121
 pdaltools/las_remove_dimensions.py,sha256=f8imGhN6LNTuQ1GMJQRzIIV3Wab_oRPOyEnKi1CgfiM,2318
 pdaltools/las_rename_dimension.py,sha256=FEWIcq0ZZiv9xWbCLDRE9Hzb5K0YYfoi3Z8IZFEs-uU,2887
 pdaltools/pcd_info.py,sha256=NIAH5KGikVDQLlbCcw9FuaPqe20UZvRfkHsDZd5kmZA,3210
-pdaltools/replace_area_in_pointcloud.py,sha256=4JyWWDtPUqtzE3zMh7eHwjGAupLclvjxKg8ScgIC4-4,8714
+pdaltools/replace_area_in_pointcloud.py,sha256=J3az8_Z7nNsGKqns9MNHrWRLc5yss7DKZvmYinlweEE,10539
 pdaltools/replace_attribute_in_las.py,sha256=MHpIizSupgWtbizteoRH8FKDE049hrAh4v_OhmRmSPU,4318
 pdaltools/standardize_format.py,sha256=I2oNiwhSMtr4e5ZK9qbB_yKmy3twOoO6QLiSFu4_AaI,3905
 pdaltools/unlock_file.py,sha256=G2odk0cpp_X9r49Y90oK88v3qlihaMfg6acwmWqblik,1958
-ign_pdal_tools-1.15.4.dist-info/METADATA,sha256=kBr1Y0CwOciH9Gjw4HxpYiYy5eQpcHqbSxQ_zXnZr3M,6146
-ign_pdal_tools-1.15.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ign_pdal_tools-1.15.4.dist-info/top_level.txt,sha256=KvGW0ZzqQbhCKzB5_Tp_buWMZyIgiO2M2krWF_ecOZc,10
-ign_pdal_tools-1.15.4.dist-info/RECORD,,
+ign_pdal_tools-1.15.5.dist-info/METADATA,sha256=6L_lOss5aYUBg8p3MAZNH2_xyaZpwidS9yCms9CCzcA,6146
+ign_pdal_tools-1.15.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ign_pdal_tools-1.15.5.dist-info/top_level.txt,sha256=KvGW0ZzqQbhCKzB5_Tp_buWMZyIgiO2M2krWF_ecOZc,10
+ign_pdal_tools-1.15.5.dist-info/RECORD,,

pdaltools/_version.py

@@ -1,4 +1,4 @@
-__version__ = "1.15.4"
+__version__ = "1.15.5"
 
 
 if __name__ == "__main__":

pdaltools/replace_area_in_pointcloud.py

@@ -1,12 +1,22 @@
 import argparse
+import os
+import shutil
+import tempfile
+import time
 import warnings
 
+import geopandas as gpd
 import numpy as np
 import pdal
 from numpy.lib import recfunctions as rfn
 from osgeo import gdal
+from shapely.geometry import box
 
-from pdaltools.las_info import get_writer_parameters_from_reader_metadata
+from pdaltools.las_info import (
+    get_bounds_from_header_info,
+    get_writer_parameters_from_reader_metadata,
+    las_info_metadata,
+)
 
 
 def argument_parser():
@@ -139,6 +149,18 @@ def pipeline_read_from_DSM(dsm, ground_mask, classification):
     return pipeline
 
 
+def clip_area(area_input, area_output, las_file):
+    start = time.time()
+    gdf = gpd.read_file(area_input)
+    minx, maxx, miny, maxy = get_bounds_from_header_info(las_info_metadata(las_file))
+    selection = gdf[gdf.intersects(box(minx, miny, maxx, maxy))]
+    num_polygon = len(selection)
+    selection.to_file(area_output)
+    end = time.time()
+    print(f"Create replacement area cropped: {num_polygon} polygons in {end-start:.2f} seconds")
+    return num_polygon
+
+
 def replace_area(
     target_cloud, pipeline_source, replacement_area, output_cloud, source_pdal_filter="", target_pdal_filter=""
 ):
@@ -147,6 +169,24 @@ def replace_area(
     print("output cloud: ", output_cloud)
     print("source pdal filter: ", source_pdal_filter)
     print("target pdal filter: ", target_pdal_filter)
+
+    start = time.time()
+    tmpdir = tempfile.mkdtemp()
+    replacement_name, ext = os.path.splitext(os.path.basename(replacement_area))
+    las_name, _ = os.path.splitext(os.path.basename(target_cloud))
+    replacement_area_crop = os.path.join(tmpdir, f"{replacement_name}_{las_name}{ext}")
+    print("replacement area crop: ", replacement_area_crop)
+
+    num_polygon = clip_area(replacement_area, replacement_area_crop, target_cloud)
+    if num_polygon == 0:
+        print("No polygon inside target cloud, output file is target cloud. We copy target in output.")
+
+        shutil.copy2(src=target_cloud, dst=output_cloud)
+        end = time.time()
+        print("all steps done in ", f"{end-start:.2f}", " seconds")
+        return
+
+    replacement_area = replacement_area_crop
     crops = []
     # pipeline to read target_cloud and remove points inside the polygon
     pipeline_target = pdal.Pipeline()
@@ -160,7 +200,9 @@ def replace_area(
     # Keep only points out of the area
     pipeline_target |= pdal.Filter.expression(expression="geometryFid==-1", tag="A")
     target_count = pipeline_target.execute()
-    print("Step 1: target count: ", target_count)
+
+    t1 = time.time()
+    print(f"Step 1: target count: {target_count} points in {t1-start:.2f} seconds")
 
     # get input dimensions dtype from target
     if pipeline_target.arrays:
@@ -171,7 +213,10 @@ def replace_area(
         pipeline_target2 |= pdal.Reader.las(filename=target_cloud)
         pipeline_target2.execute()
         input_dim_dtype = pipeline_target2.arrays[0].dtype
+        t1_bis = time.time()
+        print(f"Step 1-bis: re-read to have dimensions: {t1_bis-t1:.2f} seconds")
 
+    t2 = time.time()
     # get input dimensions names
     input_dimensions = list(input_dim_dtype.fields.keys())
 
@@ -179,7 +224,7 @@ def replace_area(
     output_dimensions = [dim for dim in input_dimensions if dim not in "geometryFid"]
 
     # add target to the result after keeping only the expected dimensions
-    if pipeline_target.arrays:
+    if target_count:
         target_cloud_pruned = pipeline_target.arrays[0][output_dimensions]
         crops.append(target_cloud_pruned)
 
@@ -192,7 +237,9 @@ def replace_area(
     # Keep only points in the area
     pipeline_source |= pdal.Filter.expression(expression="geometryFid>=0", tag="B")
     source_count = pipeline_source.execute()
-    print("Step 2: source count: ", source_count)
+
+    t3 = time.time()
+    print(f"Step 2: source count: {source_count} points in {t3-t2:.2f} seconds")
 
     # add source to the result
     if source_count:
@@ -221,7 +268,11 @@ def replace_area(
 
     writer_params = get_writer_params(target_cloud)
     pipeline |= pdal.Writer.las(filename=output_cloud, **writer_params)
-    pipeline.execute()
+    points = pipeline.execute()
+
+    end = time.time()
+    print(f"Step 3: merge: { points }, points in {end-t3:.2f} seconds")
+    print("all steps done in ", f"{end-start:.2f}", " seconds")
 
 
 if __name__ == "__main__":