imcp.py

#!/usr/bin/env python
#
# imcp.py - Copy image files.
#
# Author: Paul McCarthy <paulmc@fmrib.ox.ac.uk>
#
"""This module defines the ``imcp`` application, for copying NIFTI image
files. 

The :func:`main` function is essentially a wrapper around the
:func:`fsl.utils.imcp.imcp` function - see its documentation for more details.
"""


from __future__ import print_function

import os.path        as op
import                   sys
import fsl.utils.path as fslpath
import fsl.utils.imcp as imcp
import fsl.data.image as fslimage


usage = """Usage:
  imcp <file1> <file2>
  imcp <file1> <directory>
  imcp <file1> <file2> ... <fileN> <directory>

Copy images from <file1> to <file2>, or copy all <file>s to <directory>

NB: filenames can be basenames or include an extension.

Recognised file extensions: {}
""".format(', '.join(fslimage.ALLOWED_EXTENSIONS))


def main(argv=None):
    """Parses CLI arguments (see the usage string), and calls the
    :func:`fsl.utils.imcp.imcp` function on each input.
    """

    if argv is None:
        argv = sys.argv[1:]

    if len(argv) < 2:
        raise RuntimeError(usage)

    srcs = argv[:-1]
    dest = argv[ -1]

    if len(srcs) > 1 and not op.isdir(dest):
        raise RuntimeError(usage)

    srcs = fslpath.removeDuplicates(srcs,
                                    allowedExts=fslimage.ALLOWED_EXTENSIONS,
                                    fileGroups=fslimage.FILE_GROUPS)

    for src in srcs:
        imcp.imcp(src, dest, useDefaultExt=True, overwrite=True) 


if __name__ == '__main__':

    try:
        main()
    except Exception as e:
        print(e)
        sys.exit(1)