Scripts: surfval.py

# Output map values at surface points.
#
# Example command, #0 is surface, #1 is volume data:
#
#  surfvalues #0 #1 ~/Desktop/values.txt
#
def surface_values(surfaces, volume, path):
    from os.path import expanduser
    f = open(expanduser(path), 'w')
    f.write('# %8s %10s %10s %10s %10s\n' % ('x', 'y', 'z', 'area', 'value'))
    for p in surfaces:
        sxf = p.model.openState.xform
        va, ta = p.geometry
        vals = volume.interpolated_values(va, sxf)
        from _surface import vertex_areas
        areas = vertex_areas(va, ta)
        for i in range(len(va)):
            x,y,z = va[i]
            f.write('%10.5g %10.5g %10.5g %10.5g %10.5g\n' % (x,y,z, areas[i], vals[i]))
    f.close()

def surfvalue_command(cmd_name, args):
    from Commands import parse_arguments, surface_pieces_arg, volume_arg, string_arg
    req_args = (('surfaces', surface_pieces_arg),
                ('volume', volume_arg),
                ('path', string_arg))
    kw = parse_arguments(cmd_name, args, req_args)
    surface_values(**kw)

from Midas.midas_text import addCommand
addCommand('surfvalues', surfvalue_command)