RemoveMcu.py

import json
import numpy as np
from pathlib import Path
import laspy


def get_mcu(json_path):
    with open(json_path) as J:
        meta_data = json.load(J)
    return np.float(meta_data["VDatum region MCU"]) / 100


if __name__ == "__main__":

    las_dir = Path(r"D:\RSD_PROJECTS\mcu_tpu_las")
    json_dir = las_dir

    for las in las_dir.glob("*.las"):
        json_path = json_dir / las.name.replace(".las", ".json")
        out_las_path = str(las).replace(".las", "_no_mcu.las")
        print("-" * 50)
        print(out_las_path)
        in_las = laspy.file.File(las, mode="r")
        out_las = laspy.file.File(out_las_path, mode="w", header=in_las.header)

        for field in in_las.point_format:
            print(f"writing {field.name}...")
            las_data = in_las.reader.get_dimension(field.name)
            if field.name == "total_tvu":
                mcu = get_mcu(json_path)
                print(f"removing MCU of {mcu}...")
                print(f"with MCU:  {las_data}")
                las_data = np.sqrt(las_data ** 2 - mcu ** 2)
                print(f"without MCU:  {las_data}")
            out_las.writer.set_dimension(field.name, las_data)
# dummy comment