o e+@szdZddlmZmZmZddlZddlZddlm Z ddlm Z ddl mZd d Zd d Zd dZddZddZddZddZdOddZdOddZdOddZdOddZdOdd ZdOd!d"Z dOd#d$Zd%d&Zd'd(Zd)d*Zd+d,Z d-d.Z!d/d0Z"d1d2Z#d3d4Z$d5d6Z%d7d8Z&d9d:Z'd;d<Z(d=d>Z)d?d@Z*dAdBZ+dCdDZ,dEdFZ-dGdHZ.dIdJZ/dOdKdLZ0dMdNZ1dS)PzJFunctions dealing with normalization of user input data to imgaug classes.)print_functiondivisionabsolute_importN)imgaug)dtypes) IAugmentablecCs|durdSt|r!|jdvsJd|j|jfdd|DSt|ts/Jdt|fg}|D]!}t|tr@||q3t|sNJdt|f||jq3|S)N)zZExpected array 'shapes' to be 3- or 4-dimensional, got %d dimensions and shape %s instead.cSg|]}|jqSshape).0imager r QD:\Projects\ConvertPro\env\Lib\site-packages\imgaug/augmentables/normalization.py z&_preprocess_shapes..zEExpected 'shapes' to be None or ndarray or list, got type %s instead.zUExpected each entry in list 'shapes' to be either a tuple or an ndarray, got type %s.) ia is_np_arrayndimr isinstancelisttypetupleappend)shapesresultshape_ir r r_preprocess_shapes s2       r cCsB|dur td|||ft||krtd|||t|fdS)NaTried to convert data of form '%s' to '%s'. This required %d corresponding image shapes, but argument 'shapes' was set to None. This can happen e.g. if no images were provided in a Batch, as these would usually be used to automatically derive image shapes.aTried to convert data of form '%s' to '%s'. This required exactly %d corresponding image shapes, but instead %d were provided. This can happen e.g. if more images were provided than corresponding augmentables, e.g. 10 images but only 5 segmentation maps. It can also happen if there was a misunderstanding about how an augmentable input would be parsed. E.g. if a list of N (x,y)-tuples was provided as keypoints and the expectation was that this would be parsed as one keypoint per image for N images, but instead it was parsed as N keypoints on 1 image (i.e. 'shapes' would have to contain 1 shape, but N would be provided). To avoid this, it is recommended to provide imgaug standard classes, e.g. KeypointsOnImage for keypoints instead of lists of tuples.) ValueErrorlen)rn from_ntypeto_ntyper r r_assert_exactly_n_shapes$s  r&cCs$|j|krtd||||jfdS)NzTried to convert an array to list of %s. Expected that array to be of shape %s, i.e. %d-dimensional, but got %d dimensions instead.)rr!)arrr shape_strr%r r r_assert_single_array_ndim?s  r)c sd}t|dkr g}nt|dr|}n d}dd|D}tfdd|DrEtd|||dd d|Ddd d|DfdS) Niterableriterable of iterablecSg|] }|D]}|qqSr r rZ arrs_sublistr'r r rrTz,_assert_many_arrays_ndim..csg|]}|jkqSr rrr'r/r rrVzTried to convert an %s of arrays to a list of %s. Expected each array to be of shape %s, i.e. to be %d-dimensional, but got dimensions %s instead (array shapes: %s)., cSg|]}t|jqSr )strrr0r r rr]r1cSr3r r4rr0r r rr^r1r"rranyr!join)arrsrr(r%iterable_type_str arrs_flatr r/r_assert_many_arrays_ndimHs" r<cCs2|jd|krtd|||jdt|jfdS)NzTried to convert an array to a list of %s. Expected the array's last dimension to have size %d, but got %d instead (array shape: %s).)rr!r4)r'sizer%r r r%_assert_single_array_last_dim_exactlyasr?c sd}t|dkr g}nt|dr|}n d}dd|D}tfdd|DrDtd||dd d|Ddd d|DfdS) Nr*rr+cSr,r r r-r r rrvr.z8_assert_many_arrays_last_dim_exactly..csg|] }|jdkqSr=rr0r>r rrxr.zTried to convert an %s of array to a list of %s. Expected the arrays' last dimensions to have size %d, but got %s instead (array shapes: %s).r2cSsg|] }t|jdqSr@r5r0r r rr~r.cSr3r r5r0r r rrr1r6)r9r>r%r:r;r rAr$_assert_many_arrays_last_dim_exactlyjs" rBcCs|durdSt|r'|jdkr|tjdtjfS|jdkr%|dtjfS|St|rWg}|D]$}|jdvs?Jd|jf|jdkrO||dtjfq0||q0|Stdt|f)Nr.r )rr zGot a list of arrays as argument 'images'. Expected each array in that list to have 2 or 3 dimensions, i.e. shape (H,W) or (H,W,C). Got %d dimensions instead.zhExpected argument 'images' to be any of the following: None or array or iterable of array. Got type: %s.) rrrnpZnewaxis is_iterablerr!r)imagesrrr r rnormalize_imagess4       rFcsddlmt|}t|}tjt|d|d}|dkrdS|dkr;t|ddd |t|d fd d t ||DS|d krB|gS|d krHdS|dkrft |ddd |t|d fdd t ||DS|dksqJd|f|S)NrHeatmapsOnImagezList[HeatmapsOnImage]r$r%rNone array[float]r z (N,H,W,C)rHr#cg|] \}}||dqSrr rattr_irrGr rrz&normalize_heatmaps..iterable[empty]iterable-array[float]r z(H,W,C)crMrNr rOrGr rrrQiterable-HeatmapsOnImage$Got unknown normalization type '%s'.) Zimgaug.augmentables.heatmapsrHr estimate_heatmaps_norm_type functoolspartialr&r)r"zipr<inputsrntype _assert_exactly_n_shapes_partialr rGrnormalize_heatmapss:    r^cs&ddlmt|}t|}tjt|d|d}|dkrdS|dvrKt|ddd |t|d |d kr?fd d t ||DSfdd t ||DS|d krR|gS|dkrXdS|dvrt |ddd |t|d |dkrzfdd t ||DSfdd t ||DS|dksJd|f|S)NrSegmentationMapsOnImagezList[SegmentationMapsOnImage]rIrJ array[int] array[uint] array[bool]r z(N,H,W,#SegmapsPerImage)r`rLrdcrMrNr rrPrr_r rrrQz/normalize_segmentation_maps..crMrNr rer_r rrrQrRiterable-array[int]iterable-array[uint]iterable-array[bool]r z(H,W,#SegmapsPerImage)ricrMrNr rer_r rrrQcrMrNr rer_r rrrQ iterable-SegmentationMapsOnImagerU) Zimgaug.augmentables.segmapsr`r estimate_segmaps_norm_typerWrXr&r)r"rYr<rZr r_rnormalize_segmentation_mapssT      rlcsddlmmt|}t|}tjt|d|d}|dkr|S|dvrCt|ddd t |d d |t |d fd d t ||DS|dkr^|dd |d|ddg|ddgS|dkrq|dd |g|ddgS|d krx|gS|dkr~dS|dvrt |d dd t |d d |t |d fdd t ||DS|dkr|dd fdd |D|ddgS|dkr|dd ||ddgS|dkr|S|dkrdS|dkr|t |d fdd t ||DS|dksJd|f|t |d fd d t ||DS)!NrKeypointKeypointsOnImagezList[KeypointsOnImage]rIrJrKrbrcr z(N,K,2)rorrLcg|] \}}j||dqSrN from_xy_arrayreror rr z'normalize_keypoints..tuple[number,size=2]rxyrrnrRrSrgrhz(K,2)crqrNrrrertr rrruiterable-tuple[number,size=2]crM)rwr )rrxryrnr rriterable-Keypointiterable-KeypointsOnImageiterable-iterable[empty]&iterable-iterable-tuple[number,size=2]c*g|]\}}jtj|tjd|dqS)dtyper)rsrCarrayfloat32rertr rr"iterable-iterable-KeypointrUcrMrNr rertr rr-rQ)imgaug.augmentables.kpsrnror estimate_keypoints_norm_typerWrXr&r)r?r"rYr<rBrZr rmrnormalize_keypointssx           rcs*ddlmmt|}t|}tjt|d|d}|dkrdS|dvrCt|ddd t |d d |t |d fd d t ||DS|dkrd|dd |d|d|d|ddg|ddgS|dkrw|dd |g|ddgS|d kr~|gS|dkrdS|dvrt |ddd t |d d |t |d fdd t ||DS|dkr|dd fdd |D|ddgS|dkr|dd ||ddgS|dkr|S|dkrdS|dkr|t |d fdd t ||DS|dksJd |f|t |d fd!d t ||DS)"Nr BoundingBoxBoundingBoxesOnImagezList[BoundingBoxesOnImage]rIrJrpr z(N,B,4)rr rLcrqrNfrom_xyxy_arrayrerr rrDruz,normalize_bounding_boxes..tuple[number,size=4]rrx1y1x2y2rrrRrzz(B,4)crqrNrrerr rr_ruiterable-tuple[number,size=4]cs$g|]\}}}}||||dqS)rr )rrrrr)rr rrhs iterable-BoundingBoxiterable-BoundingBoxesOnImager&iterable-iterable-tuple[number,size=4]crr)rrCrrrerr rruriterable-iterable-BoundingBoxrUcrMrNr rerr rrrQ)Zimgaug.augmentables.bbsrrr !estimate_bounding_boxes_norm_typerWrXr&r)r?r"rYr<rBrZr rrnormalize_bounding_boxes2s             rcC(ddlm}m}t||ddgt||dS)Nr)PolygonPolygonsOnImagez#polys#points cls_singlecls_oi axis_namesestimate_ntype_funcr[r)Zimgaug.augmentables.polysrr$_normalize_polygons_and_line_stringsestimate_polygons_norm_type)r[rrrr r rnormalize_polygonsrcCr)Nr) LineStringLineStringsOnImagez#linesrr)Zimgaug.augmentables.linesrrrestimate_line_strings_norm_type)r[rrrr r rnormalize_line_stringsrrc sj}j}d|d|df}d|d|df} d|df} t|}||} tjt| d|f|d} | dkr;dS| d vr`t|d ||t|d || t|d fd dt||DS| |krs| dd |g|ddgS| |krz|gS| dkrdS| dvrt |d| |t |d || t|d fddt||DS| dkr| dd |g|ddgS| dkr| dd |g|ddgS| d|fkr| dd ||ddgS| d|fkr|S| dkrdS| dvrt |d | |t |d || t|d fddt||DS| dkr3| dd fdd|D|ddgS| dkrM| dd fdd|D|ddgS| d|fkrh| t|d fddt||DS| d krodS| d!vs{Jd"| f| t|d fd#dt||DS)$Nz (N,%s,%s,2)rrz (%s,%s,2)z(%s,2)zList[%s]rIrJrpr rrLc*g|]\}}fdd|D|dqS)cg|]}|qSr r rZpointsrr rrC_normalize_polygons_and_line_strings...rr rerrr rrz8_normalize_polygons_and_line_strings..rrRrzr cr)crr r rrr rrrrrr rerr rrsr{r~ iterable-%sriterable-iterable-array[float]iterable-iterable-array[int]iterable-iterable-array[uint]cr)crr r rrr rrrrrr rerr rrrrcrr r rrPrr rrrrcrr r rrr rrriterable-iterable-%scrMrNr re)rr rrs !iterable-iterable-iterable[empty])/iterable-iterable-iterable-tuple[number,size=2]#iterable-iterable-iterable-KeypointrUcr)crr r rrr rrrrrr rerr rrr) __name__r rWrXr&r)r?r"rYr<rB) rrrrr[rcls_single_name cls_oi_nameZaxis_names_4_strZaxis_names_3_strZaxis_names_2_strr\r]r rrrs                    rcCs|dur|dusJdt|fdSt|r|t|s/|jdkr(t||gSt|t|S|jdkr^|jddksGJd|jd|jf|jddksZJd|jd|jf|dS|jdkrz|jddksvJd |jd|jf|d S|St|rg}t||D]?\}}|jdkr|jddksJd |jd|jf| |dddddfq|jdksJd |j|jf| |q|St d t|f)NzmExpected (normalized) 'images' to be None due to (unnormalized) 'images_old' being None. Got type %s instead.rrrzExpected normalized images of shape (N,H,W,C) to have N=1 due to the unnormalized images being a single 2D image. Got instead N=%d and shape %s.r zExpected normalized images of shape (N,H,W,C) to have C=1 due to the unnormalized images being a single 2D image. Got instead C=%d and shape %s.)r.rzExpected normalized images of shape (N,H,W,C) to have C=1 due to unnormalized images being a single 3D image. Got instead C=%d and shape %s).rzExpected each image of shape (H,W,C) to have C=1 due to the corresponding unnormalized image being a 2D image. Got instead C=%d and shape %s.zMExpected 'image_old' to be three-dimensional, got %d dimensions and shape %s.zlExpected argument 'images_old' to be any of the following: None or array or iterable of array. Got type: %s.) rrrrinvert_normalize_imagesrrrDrYrr!)rEZ images_oldrrZ image_oldr r rr sx               rcst|}|dkr|dusJdt|f|S|dkr>t||jdks1Jdt||jdf|jtdd|DS|dkrVt|d ksRJdt|d f|dS|d kri|dusgJd t|fgS|d krt|\}}}|jfd d|DS|dksJd|f|S)NrJznExpected (normalized) 'heatmaps' to be None due (unnormalized) 'heatmaps_old' being None. Got type %s instead.rKrzfExpected as many heatmaps after normalization as before normalization. Got %d (after) and %d (before).cSr r )arr_0to1rZhm_ir r rr\rz-invert_normalize_heatmaps..rHrrRzExpected heatmaps after normalization to be None, due to the heatmaps before normalization being an empty iterable. Got type %s instead.rScsg|]}t|jqSr )restore_dtype_and_mergerr input_dtyper rrmsrTrU)rVrr"rrrfind_first_nonempty)heatmapsZ heatmaps_oldr\nonempty_r rrinvert_normalize_heatmapsMs\     rcst|}|dkr|dusJdt|f|S|dvr>t||jdks1Jdt||jdf|jtdd|DS|dkrVt|d ksRJdt|d f|dS|d kri|dusgJd t|fgS|d vrt|\}}}|jfd d|DS|dksJd|f|S)NrJzExpected (normalized) 'segmentation_maps' to be None due (unnormalized) 'segmentation_maps_old' being None. Got type %s instead.rarzoExpected as many segmentation maps after normalization as before normalization. Got %d (after) and %d (before).cSg|]}|qSr )get_arrrZsegmap_ir r rrrz6invert_normalize_segmentation_maps..r`rrRzExpected segmentation maps after normalization to be None, due to the segmentation maps before normalization being an empty iterable. Got type %s instead.rfcg|] }t|qSr )rrrrr rrrjrU)rkrr"rrrr)segmentation_mapsZsegmentation_maps_oldr\rrr rr"invert_normalize_segmentation_mapsusZ     rcst|}|dkr|dusJdt|f|S|dvr7t|dks*Jdt|f|jtdd|DS|dkrnt|dksJJd t|ft|d jdks^Jd t|d j|d jd j|d jd jfS|d krt|dksJd t|ft|d jdksJdt|d j|d jd S|dkrt|dksJdt|f|d S|dkr|dusJdt|fgS|dvrt|\}}}|jfdd|DS|dkrt|dksJdt|fdd|d jDS|dkrt|dksJdt|f|d jS|dkr|S|dkr4|dus.Jdt|f|ddS|dkr@dd|DS|dksLJd |fd!d|DS)"NrJzpExpected (normalized) 'keypoints' to be None due (unnormalized) 'keypoints_old' being None. Got type %s instead.rprzExpected a single KeypointsOnImage instance after normalization due to getting a single ndarray before normalization. Got %d instances instead.cSrr ) to_xy_arrayrZkpsoir r rrrz.invert_normalize_keypoints..rvzExpected a single KeypointsOnImage instance after normalization due to getting a single (x,y) tuple before normalization. Got %d instances instead.rzExpected a KeypointsOnImage instance containing a single Keypoint after normalization due to getting a single (x,y) tuple before normalization. Got %d keypoints instead.rnzExpected a single KeypointsOnImage instance after normalization due to getting a single Keypoint before normalization. Got %d instances instead.zExpected a KeypointsOnImage instance containing a single Keypoint after normalization due to getting a single Keypoint before normalization. Got %d keypoints instead.rozExpected a single KeypointsOnImage instance after normalization due to getting a single KeypointsOnImage before normalization. Got %d instances instead.rRzExpected keypoints after normalization to be None, due to the keypoints before normalization being an empty iterable. Got type %s instead.rzcrr )rr)rZkps_irr rr r{zExpected a single KeypointsOnImage instance after normalization due to getting an iterable of (x,y) tuples before normalization. Got %d instances instead.cSg|]}|j|jfqSr rwrkpr r rrs r~zExpected a single KeypointsOnImage instance after normalization due to getting an iterable of Keypoint before normalization. Got %d instances instead.rrzExpected keypoints after normalization to be None, due to the keypoints before normalization being an empty iterable of iterables. Got type %s instead.rcSg|] }dd|jDqS)cSrr rwrr r rrsz9invert_normalize_keypoints... keypointsrr r rrrrUcSg|] }|jddqSNrrr r rrr.) rrr"rrrrxryr)rZ keypoints_oldr\rrr rrinvert_normalize_keypointss              rcst|}|dkr|dusJdt|f|S|dvr7t|dks*Jdt|f|jtdd|DS|dkrot|dksJJd t|ft|d jdks^Jd t|d j|d jd }|j|j|j|j fS|d krt|dksJd t|ft|d jdksJdt|d j|d jd S|dkrt|dksJdt|f|d S|dkr|dusJdt|fgS|dvrt |\}}}|jfdd|DS|dkrt|dksJdt|fdd|d jDS|dkrt|dksJdt|f|d jS|dkr|S|dkr5|dus/Jdt|f|ddS|dkrAdd|DS|dksMJd |fd!d|DS)"NrJzzExpected (normalized) 'bounding_boxes' to be None due (unnormalized) 'bounding_boxes_old' being None. Got type %s instead.rprzExpected a single BoundingBoxesOnImage instance after normalization due to getting a single ndarray before normalization. Got %d instances instead.cSrr ) to_xyxy_arrayrZbbsoir r rr sz3invert_normalize_bounding_boxes..rzExpected a single BoundingBoxesOnImage instance after normalization due to getting a single (x1,y1,x2,y2) tuple before normalization. Got %d instances instead.rzExpected a BoundingBoxesOnImage instance containing a single BoundingBox after normalization due to getting a single (x1,y1,x2,y2) tuple before normalization. Got %d bounding boxes instead.rzExpected a single BoundingBoxesOnImage instance after normalization due to getting a single BoundingBox before normalization. Got %d instances instead.zExpected a BoundingBoxesOnImage instance containing a single BoundingBox after normalization due to getting a single BoundingBox before normalization. Got %d bounding boxes instead.rzExpected a single BoundingBoxesOnImage instance after normalization due to getting a single BoundingBoxesOnImage before normalization. Got %d instances instead.rRzExpected bounding boxes after normalization to be None, due to the bounding boxes before normalization being an empty iterable. Got type %s instead.rzcrr )rrrrr rr:rrzExpected a single BoundingBoxesOnImage instance after normalization due to getting a an iterable of (x1,y1,x2,y2) tuples before normalization. Got %d instances instead.cS g|] }|j|j|j|jfqSr rrbbr r rrCrzExpected a single BoundingBoxesOnImage instance after normalization due to getting an iterable of BoundingBox before normalization. Got %d instances instead.rrzExpected bounding boxes after normalization to be None, due to the bounding boxes before normalization being an empty iterable of iterables. Got type %s instead.rcSr)cSrr rrr r rrXs z>invert_normalize_bounding_boxes...bounding_boxesrr r rrWrrrUcSrrrrr r rr]r.) estimate_normalization_typerr"rrrrrrrr)rZbounding_boxes_oldr\rrrr rrinvert_normalize_bounding_boxess             rc Ct||tddddddS)NrrcS|jSrpolygons)Zpsoir r rez+invert_normalize_polygons..cSrr)Zexterior)Zpolyr r rrfr)+_invert_normalize_polygons_and_line_stringsr)rZ polygons_oldr r rinvert_normalize_polygons`rc Cr)NrrcSrr line_strings)Zlsoir r rrnrz/invert_normalize_line_strings..cSrr)coords)Zlsr r rror)rr)rZline_strings_oldr r rinvert_normalize_line_stringsirrc sLddlm||}|dkr|dusJdt|f|S|dvr1|jtfdd|DS||krht|dksFJd ||t|ft|ddks`Jd |||t|df|ddS||krt|dks}Jd ||t|f|dS|d kr|dusJd t|fgS|d vrt|\}} } |jfdd|DS|dkrt|dksJd|t|ft|ddksJd||||t|fdd|ddDS|dkr%t|dksJd|t|ft|ddksJd||||t|ffdd|ddDS|d|fkr`t|dks?Jd||t|ft|dt|ksZJd|||||t|f|dS|d|fkrj|S|dkr|dus}Jdt|f|ddS|dvrt|\}} } |jfdd|DS|dkrt|dksJd|t|ffd d|dDS|d!krt|dksJd"|t|ffd#d|dDS|d$|fkrfd%d|DS|d&kr|ddS|d'krfd(d|DS|d)ksJd*|ffd+d|DS),Nrr|rJzExpected (normalized) polygons/line strings to be None due (unnormalized) polygons/line strings being None. Got type %s instead.rpc"g|] }fdd|DqS)crr r rentityget_points_funcr rrrJ_invert_normalize_polygons_and_line_strings...r rZoiget_entities_funcrr rrsz?_invert_normalize_polygons_and_line_strings..rz|Expected a single %s instance after normalization due to getting a single %s before normalization. Got %d instances instead.zExpected a %s instance containing a single %s after normalization due to getting a single %s before normalization. Got %d instances instead.rRzExpected polygons/line strings after normalization to be None, due to the polygons/line strings before normalization being an empty iterable. Got type %s instead.rzcs(g|]}tfdd|DqS)crr r rrr rrrrrrrrrr rrsr{zExpected a single %s instance after normalization due to getting an iterable of (x,y) tuples before normalization. Got %d instances instead.zExpected a %s instance after normalization containing a single %s instance due to getting an iterable of (x,y) tuples before normalization. Got a %s with %d %s instances instead.cSg|] }|d|dfqSrrr rpointr r rrsr~zExpected a single %s instance after normalization due to getting an iterable of Keypoint before normalization. Got %d instances instead.zExpected a %s instance after normalization containing a single %s instance due to getting an iterable of Keypoint before normalization. Got a %s with %d %s instances instead.c g|] }|d|ddqSrrrwr rr|r rrrzExpected a single %s instance after normalization due to getting an iterable of %s before normalization. Got %d instances instead.zExpected a %s instance after normalization containing a single %s instance due to getting an iterable of %s before normalization. Got a %s with %d %s instances instead.rzExpected polygons/line strings after normalization to be None, due to the polygons/line strings before normalization being an empty iterable of iterables. Got type %s instead.rcs$g|]}fdd|DqS)csg|] }t|qSr rr)rrr rrrrr rrr rrs  rzExpected a single %s instance after normalization due to getting an iterable of iterables of (x,y) tuples before normalization. Got %d instances instead.cg|] }dd|DqS)cSrrr rr r rrr}rr rrr rrsrzExpected a single %s instance after normalization due to getting an iterable of iterables of Keypoint before normalization. Got %d instances instead.c"g|] }fdd|DqS)cr r r rr|r rrr rr rrnrr rrs  rcrr r r)rr rrrrrcr)cr )cSrrr rr r rrrU_invert_normalize_polygons_and_line_strings....r rrr rrs rr rrr rrs  rrUcs$g|]}fdd|DqS)cr)cr r r rr|r rrrrr rrr rrs  rr r)rnrrr rrs  )rrnrrrr"r) r[Z inputs_oldrrrrrr\rrr )rnrrrrrrs.                        rcCs$||vsJd|d||fdS)NzPGot an unknown datatype for argument '%s'. Expected datatypes were: %s. Got: %s.r2)r8)type_strvalid_type_strsZarg_namer r r_assert_is_of_norm_types  rcC t|}gd}t||d|S)N)rJrKrHrRrSrTrrr)rrrr r rrV"s rVcCr)N) rJrbrcrdr`rRrgrhrirjrr)rrrr r rrk0s  rkcCr)N)rJrKrbrcrvrnrorRrSrgrhr{r~rrrrrr)rrrr r rrCs rcCr)N)rJrKrbrcrrrrRrSrgrhrrrrrrrr)rrrr r rr\s rcCt|dddS)Nrrr._estimate_polygons_and_line_segments_norm_typerr r rrvrcCr)Nrrrrrr r rr{rrcCs^t|}dddd||ddddd d d |fd |fd d ddddd|fdddg}t||||S)NrJrKrbrcrRrSrgrhr{r~rrrrrrrrrrrr)r[rrZaugmentable_namerrr r rrs8 rcCst|\}}}t|||}|Sr)r_nonempty_info_to_type_str)r[rsuccessparentsrr r rrs rcs^t|tr"fdd|D}dd|D}tt|dkr"t|}t|r-t |}|S)Ncsg|]}t|qSr rrZarr_irr rrsz+restore_dtype_and_merge..cSr r rrr r rrrr) rrr"setrCrrriadtZrestore_dtypes_)r'rrr rrrs     rcCs.t|ot|t ot|d ot| S)Nr)rrDrr hasattrZ is_string)objr r r _is_iterables    r"c Cs|durg}|dust|r|d|fSt|rat|dkr#dd|fS|d|ur.|d|fSd}d}|}|D]%}t|||gd\}}}|rM|||fSt|t|kr[|}|}|}q6|||fS|d|fS)NTrF)r)rrr"r"r) attrrZnonempty_deepestZsuccess_deepestZparents_deepestrPrrZ parents_foundr r rrs2         rc CsTt|dksJdt|fd}t|dkr$dddgt|f}|s+d|fSt|d ko7t|d t}|rnt|d dkoLtd d |d D}|rnddgt|d }d t|d f}d||gdS|durtdSt|r|jj }ddddd}d|||vr||fS|fSt|t sJdt |fd||j j fS)Nr z%Expected 'parents' to be <=4, got %d.rz%s--r*z%siterable[empty]rr=cSsg|]}t|qSr )rZis_single_number)rvalr r rrr1z._nonempty_info_to_type_str..ztuple[number,size=%d]rJfloatZuintintbool)fuibz %sarray[%s]z1Expected 'nonempty' to be an object, got type %s.z%s%s)r"r8rralllstriprrrkindobjectr __class__r) rrrZ parent_itersZis_parent_tupleZis_only_numbers_in_tupleZtpl_namer0Zkind_mapr r rrsF      rr)2__doc__ __future__rrrrWnumpyrCr$rrrrbaser r r&r)r<r?rBrFr^rlrrrrrrrrrrrrrrrVrkrrrrrrrr"rrr r r rsX        + I S  n@(+]c  *" +