3
dt                 @   sN   d dl Z d dlZd dlmZ dd Zdd Zdd Zd	d
 ZedkrJe  dS )    N)rangec               C   s&   t d t d t d tjd d S )Nz1Usage: $0 [-extension/extensions] <list of names>z3       -extension for one image with full extensionz6       -extensions for image list with full extensions   )printsysexit r   r   </tmp/pip-build-7vycvbft/nipype/nipype/external/fsl_imglob.pyusageG   s    r	   c             C   sF   x:|D ]2}| t | d  |krd| d t |  |fS qW d| dfS )NTF )len)inputallExtensions	extensionr   r   r   isImageP   s    
r   c             C   s   t | |d S )Nr   )r   )r   r   r   r   r   removeImageExtensionW   s    r   c              C   s  t tjdkrt  tjdk r8dd l} ddlm} d}nd}d}dd	d
dg}ddg}|| }|}d}tjd dkr~|}d}d}tjd dkrd}d}g }	xFt|t tjD ]2}
x,|D ]$}|	jt	j	t
tj|
 ||  qW qW |rx(tdt |	D ]}t
|	| ||	|< qW |r$tt|	}	nt||	}	|	j  xFtdt |	D ]4}t|	| dd |t |	d k rHtddd qHW d S )Nr         r   )SetFTz.nii.gzz.niiz.hdr.gzz.hdrz.img.gzz.imgz-extensionsz
-extension )end)r   r   )r   r   argvr	   version_infosetsr   r   extendglobr   listsetsortr   )r   r   ZsetAvailableZdeleteExtensionsZprimaryExtensionsZsecondaryExtensionsr   ZvalidExtensionsZstartingArgfilelistargZcurrentExtensionfiler   r   r   main[   sL    

r!   __main__)	r   r   builtinsr   r	   r   r   r!   __name__r   r   r   r   <module>B   s   	: