DM_numpy_export.py

## @package python.demo.DM_numpy_export
# "
# example script showing how to write a point cloud file from numpy dictionary
# 
 
from opals import pyDM
import numpy as np
 
f = pyDM.AddInfoLayoutFactory()
f.addColumn(pyDM.ColumnSemantic.Amplitude)
layout = f.getLayout()
 
# 4 example points including a Amplitude attribute
pts = {}
pts["x"] = np.array([0.1,  0.2, 10, 10])
pts["y"] = np.array([0, 10, 10,  0])
pts["z"] = np.array([0,  1,  2,  3])
pts[layout.name(0)] = np.array([5.343,  30.234,  22.32,  30])
 
# filename stem and output format
filename_stem = "pts"
oformat = pyDM.DataFormat.las
 
exp = None
# some export formats require specific initialisation for attributes
if oformat == pyDM.DataFormat.shp:
    dbfTable = pyDM.ShapeTableDefinition(layout)
    exp = pyDM.ExportShape(filename_stem+".shp", dbfTable=dbfTable,layout=layout)
    exp.setShapeGeometryType(pyDM.ShapeGeometry.point)    #write single points
elif oformat == pyDM.DataFormat.odm:
    exp = pyDM.Export(filename_stem+".odm")
elif oformat == pyDM.DataFormat.las:
    exp = pyDM.ExportLAS(filename_stem+".laz")
    exp.exportExtraBytes(layout.semantic(0))
 
if not exp:
    Exception("Unable to create export object")
 
print(f"Writing file '{exp.getFilename()}'...")
pyDM.NumpyConverter.export(exp, pts, noDataObj=np.nan)    # if a value is nan it is treated as null
exp.close()
print("done")