source: issm/trunk/src/m/solve/marshall.py@ 26744

import subprocess

from WriteData import WriteData


def marshall(md):
    """MARSHALL - outputs a compatible binary file from @model md, for certain solution type.

    The routine creates a compatible binary file from @model md
    his binary file will be used for parallel runs in JPL-package

    Usage:
        marshall(md)
    """

    if md.verbose.solution:
        print("marshalling file {}.bin".format(md.miscellaneous.name))

    # Open file for binary writing
    try:
        fid = open(md.miscellaneous.name + '.bin', 'wb')
    except IOError as e:
        raise IOError("marshall error message: could not open '%s.bin' file for binary writing. Due to: ".format(md.miscellaneous.name), e)

    fields = md.properties()
    fields.sort() # sort fields so that comparison of binary files is easier
    for field in fields:
        # Some properties do not need to be marshalled
        if field in ['results', 'radaroverlay', 'toolkits', 'cluster', 'private']:
            continue

        # Check that current field is an object
        if not hasattr(getattr(md, field), 'marshall'):
            raise TypeError("field '{}' is not an object.".format(field))

        # Marshall current object
        #print('marshalling {} ...'.format(field) # Uncomment for debugging
        exec('md.{}.marshall(\'md.{}\', md, fid)'.format(field, field))

    #Last, write "md.EOF" to make sure that the binary file is not corrupt
    WriteData(fid, 'XXX', 'name', 'md.EOF', 'data', True, 'format', 'Boolean')

    #close file
    try:
        fid.close()

    except IOError as e:
        print('marshall error message: could not close file \'{}.bin\' due to:'.format(md.miscellaneous.name), e)

    # Uncomment the following to make a copy of the binary input file for 
    # debugging purposes (can be fed into scripts/BinRead.py).
    # copy_cmd = 'cp {}.bin {}.py.bin'.format(md.miscellaneous.name, md.miscellaneous.name)
    # subprocess.call(copy_cmd, shell=True)