DM_numpy_pointfile_read.py

## @package python.demo.DM_numpy_pointfile_read
# 
# Small example showing how to the read a point cloud file as numpy dict using the import functionality of the pyDM
# 
from __future__ import print_function
 
import opals, sys, os
 
from opals import pyDM
 
 
 
def DM_read_pointfile(filename):
  ptsToOutput = 100  # for limiting point output. use -1 for output all points
 
  # if unspecified or 'auto' is set, import creator automatically detects file format based on its content
  imp = pyDM.Import.create(filename, pyDM.DataFormat.auto)
 
  if not imp:
    print("Unable to create import object for '" + filename + "'")
    return
 
  # read header
  imp.readHeaderSeparately()
  header = imp.getHeader()
 
  # output header details
  print("File {} details:".format(filename))
  print("\tFile format:", str(imp.getFileFormat()))
  print("\tPoint count:", str(header.getPointCount()))
  print("\tCRS        :", str(header.getCRS()))
 
  if len(header.attributes()) > 0:
    print("\tAttributes:")
    formatStr = "\t\t{:<25}{:<10}"
    print(formatStr.format("Name","Type"))
    for attr in header.attributes():
      print(formatStr.format(attr.getName(), str(attr.getType())))
 
  print() # print empty line
 
  overallPtCount = header.getPointCount()
  layout = header.getDataLayout()
 
  ptsRead = 0
  ptsOutput = 0
  for obj in imp:
    # points are always read in chunks of 1000 points (can be change when creating the import object)
    pts = pyDM.NumpyConverter.get(obj, layout)  # get points as numpy dict
    rows = pts["x"].shape[0]
    ptsRead += rows
 
    for i in range(rows):
      if ptsToOutput != -1 and ptsOutput >= ptsToOutput:
        print("\t...")
        break
      print(f'\t{ptsOutput:5d}  { pts["x"][i]:12.3f} {pts["y"][i]:12.3f} {pts["z"][i]:10.3f}')
      ptsOutput += 1
 
 
    # statistic on how much points have been imported already
    #print("{0:.1f} pts read".format((ptsRead/float(overallPtCount)*100)))
 
  # close file handle
  del imp
 
if len(sys.argv) < 1:
    print("filename parameter missing")
    sys.exit(-1)
 
DM_read_pointfile(sys.argv[1])