3
XdJ                 @   s    d dl mZ dd Zdd ZdS )   )Registrationc           9   C   s>  t t ddt dddt dgddt dgddt d	ddt ddd
t ddt dd ddgdt ddgdt ddt ddgdt ddgdt dddt dddt  t dgdgdt ddt  t  t ddgddt ddt  t ddt d ddgdgdt ddgdt ddd
t  t ddgd t dddt dd!t ddt  t d"gddt d#d d$t  t d%gd&t  t  t d"gd&t  t  t d'd d$t ddt d(gd&t ddt  t ddd)t  t ddt d*ddd+t dddt dddt d,ddd-5} tj }xRt| j D ]B\}}x6t|j D ]&\}}t|j | ||ks
tq
W qW d S ).Nz%s)argstrz--collapse-output-transforms %dT)r   
usedefaultnumber_of_iterations)requiresr   convergence_thresholdz--dimensionality %d)nohashr   )	mandatoryz2.1.0fixed_image_masks)r   
extensionsmax_verxorz2.2.0fixed_image_mask)Zmin_verr   z
--float %dinitial_moving_transform_com)r   r   initial_moving_transformz$--initialize-transforms-per-stage %d)r   r   metric)r	   r   r   )r   moving_image_masks)r   r   r   r   moving_image_maskFoutput_warped_image)
hash_filesr   )r   metric_weightz--restore-state %s)r   r   sampling_strategy)r   z--save-state %ssmoothing_sigmas)r   r	   z-v)r   r   r   z--write-composite-transform %d)5argsZcollapse_output_transformsr   Zconvergence_window_sizeZ	dimensionenvironZfixed_imager   r
   floatr   r   Zinitialize_transforms_per_stageinterpolationZinterpolation_parametersZinvert_initial_moving_transformr   Zmetric_item_traitZmetric_stage_traitr   Zmetric_weight_item_traitZmetric_weight_stage_traitZmoving_imager   r   num_threadsr   Zoutput_inverse_warped_imageZoutput_transform_prefixr   Zradius_bins_item_traitZradius_bins_stage_traitZradius_or_number_of_binsZrestore_stateZrestrict_deformationZsampling_percentageZsampling_percentage_item_traitZsampling_percentage_stage_traitr   Zsampling_strategy_item_traitZsampling_strategy_stage_trait
save_stateZshrink_factorsZsigma_unitsr   Ztransform_parametersZ
transformsZuse_estimate_learning_rate_onceZuse_histogram_matchingverboseZwinsorize_lower_quantileZwinsorize_upper_quantileZwrite_composite_transform)dictr   Z
input_speclistitemsgetattrtraitsAssertionError)Z	input_mapinputskeymetadatametakeyvalue r+   e/var/www/html/virt/lib/python3.6/site-packages/nipype/interfaces/ants/tests/test_auto_Registration.pytest_Registration_inputs   s   r-   c              C   s   t t d dt  t  t  t d dt d dt  t  t  t  t  t d dt d dd} tj }xLt| j D ]<\}}x2t|j D ]"\}}t|j | ||ks|tq|W qfW d S )N)r   )Zcomposite_transformZelapsed_timeZforward_invert_flagsZforward_transformsZinverse_composite_transformZinverse_warped_imageZmetric_valueZreverse_forward_invert_flagsZreverse_forward_transformsZreverse_invert_flagsZreverse_transformsr   Zwarped_image)r    r   Zoutput_specr!   r"   r#   r$   r%   )Z
output_mapoutputsr'   r(   r)   r*   r+   r+   r,   test_Registration_outputs   s.    r/   N)Zregistrationr   r-   r/   r+   r+   r+   r,   <module>   s    0