DM_pointfile_read.py

## @package python.demo.DM_pointfile_read
# 
# Small example showing how to the read a point cloud file 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()
  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)
    ptsRead += obj.sizePoint()
 
    #loop over points
    for i in range(obj.sizePoint()):
      if ptsToOutput != -1 and ptsOutput >= ptsToOutput:
        print("\t...")
        break
      pt = obj[i]
      print("\t{0:5d}  {1:12.3f} {2:12.3f} {3:10.3f}".format(ptsOutput, pt.x,pt.y,pt.z))
      ptsOutput += 1
 
    if ptsToOutput != -1 and ptsOutput >= ptsToOutput:
      break
 
    # 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])