o fF'@sddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z d6d d Zgd Zd7ddZd8ddZGddde j Zd9d:ddZd;dd Zd23frz<23f)openreadstructunpackr%)r&fpr rhdrlenrrr isSpiderImageUs    r0c@sxeZdZdZdZdZdddZedd d Zed d d Z dddZ d!ddZ d"d#ddZ e r5ddlmZd$ddZdS)%SpiderImageFileSPIDERzSpider 2D imageFr Nonec Csd}|j|}z*d|_td|}t|}|dkr'd|_td|}t|}|dkr1d}t|WntjyF}zd}t||d}~wwd|}t|d}|dkr[d }t|t|d t|d f|_ t|d |_ t|d |_ |j dkr|j dkr|} d|_ nD|j dkr|j dkrt|d t|d d|_ ||_t|d|_ |d } d|_ n|j dkr|j dkr||j} d |_ nd}t||jrd|_nd|_d|_dd|j| |jddffg|_|j|_dS)Nlrz>27frz<27fznot a valid Spider filerrznot a Spider 2D imagerrz inconsistent stack header valueszF;32BFzF;32FFrawrr)r.r+ bigendianr,r-r% SyntaxErrorerrorr _sizeistack imgnumber_nimagesimgbytesr/ stkoffsetrawmode_modesizetile_fp) selfnr rr/msger r!offsetrrr_openes^       zSpiderImageFile._openr cCs|jSNrBrJrrrn_framesszSpiderImageFile.n_framesboolcCs |jdkSNrrQrRrrr is_animateds zSpiderImageFile.is_animatedcCs|jdkrdS|jdS)Nrr)rArRrrrtells  zSpiderImageFile.tellframecCs^|jdkr d}t|||sdS|j||j|j|_|j|_|j|j| dS)Nrz#attempt to seek in a non-stack file) r@EOFError _seek_checkr/rCrDrIr.seekrO)rJrXrLrrrr[s   zSpiderImageFile.seekdepth Image.Imagecsl|}t|dts Jttttf|\}}d||kr$||| ||fdddS)Nrrcs |SrPr)rbmrrs z.SpiderImageFile.convert2byte..L) getextrema isinstancefloatrrpointconvert)rJr]extremaminimummaximumrr_r convert2bytes  zSpiderImageFile.convert2byterImageTkImageTk.PhotoImagecCsddlm}|j|ddS)Nrrm)palette)rn PhotoImagerl)rJrnrrr tkPhotoImages zSpiderImageFile.tkPhotoImageN)r r3)r r )r rT)rXr r r3)r\)r]r r r^)r ro)__name__ __module__ __qualname__formatformat_description!_close_exclusive_fp_after_loadingrOpropertyrSrVrWr[rlrrrrnrtrrrrr1`s ;      r1filelistlist[str] | Nonelist[SpiderImageFile] | Nonec Cs|dus t|dkr dSg}|D]J}tj|s td|qzt| }|}Wdn1s5wYWntyOt |sMt|dYqw||j d<| |q|S)zJcreate a list of :py:class:`~PIL.Image.Image` objects for use in a montageNrzunable to find z is not a Spider image filer&) lenospathexistsprintr r*rl Exceptionr0infoappend)r|imglistimgimrrrloadImageSeriess(      rrr^ list[bytes]cCs|j\}}|d}td|}d|dkr|d7}||}t|d}|dkr)gSdg|}d|d<t||d<t||d <d|d <t||d <t||d <t||d <t||d<|dd}|ddd|DS)Nr7irrrgg?rrrrrrcSsg|]}td|qS)r )r,pack).0vrrr sz$makeSpiderHeader..)rGr rfr)rnsamnrowr$r"r#nvalueshdrrrrmakeSpiderHeaders*             rr. IO[bytes] str | bytesr3c Csn|jddkr |d}t|}t|dkrd}t|||d}t||dd|jd|ddffgdS) Nrr9rpzError creating Spider headerzF;32NFr:r;r) moderhrrOSError writelinesr _saverG)rr.r&rrLrErrrr s   (rcCsDtj|d}t|tr|n|}ttj |t |||dSrU) rrsplitextrebytesdecoder register_extensionr1rxr)rr.r& filename_extextrrr _save_spidersr__main__rz7Syntax: python3 SpiderImagePlugin.py [infile] [outfile]z$input image must be in Spider formatzimage: zformat: zsize: zmode: z max, min:  )endzsaving a flipped version of z as )r rr r )rrr r )r&r'r r rP)r|r}r r~)rr^r r)rr^r.rr&rr r3), __future__rrr,systypingrrrrrrrr r rrr%r0r1rrrr register_openrx register_saverurargvrexitr&r*rrGrrdoutfile transpose TransposeFLIP_LEFT_RIGHTrbasenamesaverrrrsZ "   t         "