py4dgeo 0.7.0__cp39-cp39-macosx_14_0_arm64.whl

py4dgeo/UpdateableZipFile.py

@@ -0,0 +1,81 @@
+import operator
+import zipfile
+
+
+class UpdateableZipFile(zipfile.ZipFile):
+    """A patched version of Python's zipfile that supports removing files in 'a' mode
+
+    Python does not implement this although the issue has been
+    known for over a decade and a fix exists:
+    https://github.com/python/cpython/pull/19358
+
+    This class ports the CPython fix forward.
+    """
+
+    def remove(self, member):
+        """Remove a file from the archive. The archive must be open with mode 'a'"""
+
+        if self.mode != "a":
+            raise RuntimeError("remove() requires mode 'a'")
+        if not self.fp:
+            raise ValueError("Attempt to write to ZIP archive that was already closed")
+        if self._writing:
+            raise ValueError(
+                "Can't write to ZIP archive while an open writing handle exists."
+            )
+
+        # Make sure we have an info object
+        if isinstance(member, zipfile.ZipInfo):
+            # 'member' is already an info object
+            zinfo = member
+        else:
+            # get the info object
+            zinfo = self.getinfo(member)
+
+        return self._remove_member(zinfo)
+
+    def _remove_member(self, member):
+        # get a sorted filelist by header offset, in case the dir order
+        # doesn't match the actual entry order
+        fp = self.fp
+        entry_offset = 0
+        filelist = sorted(self.filelist, key=operator.attrgetter("header_offset"))
+        for i in range(len(filelist)):
+            info = filelist[i]
+            # find the target member
+            if info.header_offset < member.header_offset:
+                continue
+
+            # get the total size of the entry
+            entry_size = None
+            if i == len(filelist) - 1:
+                entry_size = self.start_dir - info.header_offset
+            else:
+                entry_size = filelist[i + 1].header_offset - info.header_offset
+
+            # found the member, set the entry offset
+            if member == info:
+                entry_offset = entry_size
+                continue
+
+            # Move entry
+            # read the actual entry data
+            fp.seek(info.header_offset)
+            entry_data = fp.read(entry_size)
+
+            # update the header
+            info.header_offset -= entry_offset
+
+            # write the entry to the new position
+            fp.seek(info.header_offset)
+            fp.write(entry_data)
+            fp.flush()
+
+        # update state
+        self.start_dir -= entry_offset
+        self.filelist.remove(member)
+        del self.NameToInfo[member.filename]
+        self._didModify = True
+
+        # seek to the start of the central dir
+        fp.seek(self.start_dir)

py4dgeo/__init__.py

@@ -0,0 +1,32 @@
+from py4dgeo.logger import set_py4dgeo_logfile
+from py4dgeo.cloudcompare import CloudCompareM3C2
+from py4dgeo.epoch import (
+    Epoch,
+    read_from_las,
+    read_from_xyz,
+    save_epoch,
+    load_epoch,
+)
+from py4dgeo.m3c2 import M3C2, write_m3c2_results_to_las
+from py4dgeo.m3c2ep import M3C2EP
+from py4dgeo.registration import (
+    iterative_closest_point,
+    point_to_plane_icp,
+    icp_with_stable_areas,
+)
+from py4dgeo.segmentation import (
+    RegionGrowingAlgorithm,
+    SpatiotemporalAnalysis,
+    regular_corepoint_grid,
+    temporal_averaging,
+)
+from py4dgeo.util import (
+    __version__,
+    find_file,
+    MemoryPolicy,
+    set_memory_policy,
+    get_num_threads,
+    set_num_threads,
+)
+
+from py4dgeo.pbm3c2 import *

py4dgeo/cloudcompare.py

@@ -0,0 +1,32 @@
+from py4dgeo.m3c2 import M3C2
+
+
+_cloudcompare_param_mapping = {
+    "normalscale": "normal_radii",
+    "registrationerror": "reg_error",
+    "searchdepth": "max_distance",
+    "searchscale": "cyl_radius",
+    "usemedian": "robust_aggr",
+}
+
+
+class CloudCompareM3C2(M3C2):
+    def __init__(self, **params):
+        """An M3C2 implementation that uses parameter names from CloudCompare"""
+        # Remap parameters using above mapping
+        py4dgeo_params = {
+            _cloudcompare_param_mapping.get(k, k): v for k, v in params.items()
+        }
+
+        # Apply changes that are not a mere renaming
+
+        # Scale parameters are diameters in CloudCompare and radii in py4dgeo
+        if "cyl_radius" in py4dgeo_params:
+            py4dgeo_params["cyl_radius"] = py4dgeo_params["cyl_radius"] * 0.5
+        if "normal_radii" in py4dgeo_params:
+            py4dgeo_params["normal_radii"] = tuple(
+                0.5 * r for r in py4dgeo_params["normal_radii"]
+            )
+
+        # Intialize base class with remapped parameters
+        super().__init__(**py4dgeo_params)