DM_odm2numpy.py

## @package python.demo.DM_odm2numpy
# 
#   converting an odm including all attributes into numpy objects
# 
#   The script uses the NumpyConverter functionality of pyDM. Coordinates and attributes are converted as separate numpy array
#   and stored in a single dictionary object
#   (script works in python 2 and 3)
# 
from __future__ import print_function
 
from opals import pyDM
import sys
 
odm = "fullwave.odm"  # odm test file
 
# pyDM.Datamanager.load parameter: filename(string), readOnly(bool) threadSafety(bool)
dm = pyDM.Datamanager.load(odm, True, False)
if not dm:
    print("Unable to open ODM '" + odm + "'")
    sys.exit(1)
 
#query attribute layout that is stored within the odm
stat = dm.getAddInfoStatistics()
dmLayout = stat.layout()
 
#output odm attribute layout
print("Attribute list of the ODM:")
for i in range(dmLayout.columns()):
    name = dmLayout.name(i)
    print(i,"\t", dmLayout.name(i)," "*(25-len(name)), dmLayout.type(i))
 
# build layout for attributes of interest (subset of odm attributes)
lf = pyDM.AddInfoLayoutFactory()
type, inDM = lf.addColumn(dm, "GPSTime",   True); assert inDM == True
type, inDM = lf.addColumn(dm, "Amplitude", True); assert inDM == True
type, inDM = lf.addColumn(dm, "EchoWidth", True); assert inDM == True
layout = lf.getLayout()
 
 
print("Get odm points as numpy object...")
#create dictionary of numpy objects
numpyDict = pyDM.NumpyConverter.createNumpyDict(dm.sizePoint(),layout,True)
pointindex = dm.getPointIndex()
 
#fill numpy dictionary with all leafs of the ODM point index
rowIdx = 0
count = float(pointindex.sizeLeaf())
for idx,leaf in enumerate(pointindex.leafs()):
    print("%5.1f%% finished" % (idx/count*100.) )
    rowIdx += pyDM.NumpyConverter.fillNumpyDict(numpyDict,rowIdx,leaf)
 
print("100.0% finished.",rowIdx,"points have been converted")
 
# list all values
print("Output converted numpy arrays")
if sys.version_info >= (3, 0):
    for attr, value in numpyDict.items():
        print(attr,"->", value)
else:
    for attr, value in numpyDict.iteritems():
        print(attr,"->", value)