3 `JZ@s|ddlZddlmZmZmZmZGdddeZddZdd Z d d Z dd dZ dddZ dddZ dddZddZdS)N) ImageColorGimpPaletteFileGimpGradientFile PaletteFilec@sFeZdZdZdddZddZd d Zd d ZeZd dZ ddZ dS) ImagePaletteas Color palette for palette mapped images :param mode: The mode to use for the Palette. See: :ref:`concept-modes`. Defaults to "RGB" :param palette: An optional palette. If given, it must be a bytearray, an array or a list of ints between 0-255 and of length ``size`` times the number of colors in ``mode``. The list must be aligned by channel (All R values must be contiguous in the list before G and B values.) Defaults to 0 through 255 per channel. :param size: An optional palette size. If given, it cannot be equal to or greater than 256. Defaults to 0. RGBNrcCsv||_d|_|p$ttdt|j|_i|_d|_|dkrTt|jdt|jksj|dkrr|t|jkrrtddS)Nrzwrong palette size) moderawmode bytearrayrangelenpalettecolorsdirty ValueError)selfr rsizer2/usr/lib/python3/dist-packages/PIL/ImagePalette.py__init__&s zImagePalette.__init__cCsHt}|j|_|j|_|jdk r0|jdd|_|jj|_|j|_|S)N)rr r rrcopyr)rnewrrrr0s  zImagePalette.copycCs$|jr|j|jfS|jd|jfS)z Get palette contents in format suitable # for the low-level ``im.putpalette`` primitive. .. warning:: This method is experimental. z;L)r rr tobytes)rrrrgetdata<s zImagePalette.getdatacCsH|jrtdt|jtr |jStjd|j}t|dr@|jS|jS)zUConvert palette to bytes. .. warning:: This method is experimental. z!palette contains raw palette dataBr) r r isinstancerbytesarrayhasattrrtostring)rZarrrrrrGs  zImagePalette.tobytesc Cs|jrtdt|try |j|Stk rt|jtrJt|j|_t |j}|dkrdtd||j|<|d|j|<|d|j|d<|d|j|d<d|_ |SXn td|d S) zgGiven an rgb tuple, allocate palette entry. .. warning:: This method is experimental. z!palette contains raw palette datar z$cannot allocate more than 256 colorsrrizunknown color specifier: %rN) r rrtuplerKeyErrorrrr rr)rZcolorindexrrrgetcolorXs$      zImagePalette.getcolorc Cs|jrtdt|tr"t|d}|jd|jd|jxtdD]}|jd|xbt|t|j|dt|jD]<}y|jd|j |Wq|t k r|jd Yq|Xq|W|jd qFW|j d S) zVSave palette to text file. .. warning:: This method is experimental. z!palette contains raw palette datawz # Palette z # Mode: %s r z%drz %dz 0 N) r rrstropenwriter r rr IndexErrorclose)rfpijrrrsavers   (zImagePalette.save)rNr) __name__ __module__ __qualname____doc__rrrrr!r&r1rrrrrs    rcCst}||_||_d|_|S)Nr)rr rr)r datarrrrraws r7cCs:g}|dkr2x(tdD]}|j||dqWnt|S)Nrr )r appendNotImplementedError)Zblackwhitelutr/rrrmake_linear_luts r=cCs8g}x.tdD]"}|jt|d|ddqW|S)Nr go@g?)r r9int)Zexpr<r/rrrmake_gamma_luts"r?rcCs&ttd}|jt||t|S)Nr )listr reverserr)r rrrrnegatives rBcCsDddlm}g}x(tdt|D]}|j|ddq"Wt||S)Nr)randintr r8)randomrCr rr9r)r rCrr/rrrrDs  rD#fff0c0cCs@tj|\}}}td|}td|}td|}td|||S)Nrr)rZgetrgbr=r)r;rgbrrrsepias    rIcCst|ttdt|S)Nr )rr@r r)r rrrwedgesrJcCsvt|db}xZtjtjtjgD]>}y |jd||j}|r@PWqttfk rZYqXqWtdWdQRX|S)Nrbrzcannot load palette) r*rrrseekZ getpalette SyntaxErrorrIOError)filenamer.ZpaletteHandlerr<rrrloads     rP)r)r)rE)r)rrrrrobjectrr7r=r?rBrDrIrJrPrrrrst