Scripts: savemarkers.py

#
# Register the savemarkers command to save markers sets in a Chimera marker file (suffix .cmm).
# Example:
#
#       open savemarkers.py
#       savemarkers #1 ~/Desktop/mymarkers.cmm
#
def save_markers(marker_models, path):
    marker_sets = [m.marker_set for m in marker_models if hasattr(m, 'marker_set')]
    if len(marker_sets) == 0:
        from Commands import CommandError
        raise CommandError('No marker sets specified.')
    from os.path import expanduser
    f = open(expanduser(path), 'w')
    from VolumePath.markerset import save_marker_sets
    save_marker_sets(marker_sets, f)
    f.close()

def save_markers_command(cmdname, args):
    from Commands import parse_arguments, models_arg, string_arg
    req_args = (('marker_models', models_arg),
                ('path', string_arg))
    opt_args = ()
    kw_args = ()
    kw = parse_arguments(cmdname, args, req_args, opt_args, kw_args)
    save_markers(**kw)


from Midas.midas_text import addCommand
addCommand('savemarkers', save_markers_command)