(e .ddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z mZddlmZmZ dde d ee d ee d ed eef d ZdZGddeZej.fdZedk(r#ej4eej.yy)N)Optional)gdal) PathLikeOrStr)get_color_table)GDALArgumentParser GDALScript)GetOutputDriverForopen_ds src_filename pct_filename dst_filename color_count driver_namect|}|td||jdkrt|d|jd|s t|}t j |}|td|d|kt j }t j|jd|jd|jd||tj }n t|}|jd k(r|} nd dl } | jd \} } t j d} | j| |j|j d} | jdj#|| j%|j'| j)|j+|j-d kDr.| j/|j1|j3t j4|jd|jd|jd| jd|tj }|tj6k7r td| |k(r| }|S|j9|xsd| }d} t;j< | j?| |S)NzUnable to open z has z0 band(s), need 3 for inputs red, green and blue."z" driver not registered.)callbackgtiffrz.tif)suffixGTiffzDitherRGB2PCT failed) r Exception RasterCountr rGetDriverByName ColorTableComputeMedianCutPCT GetRasterBandTermProgress_nocbrlowertempfilemkstempCreate RasterXSize RasterYSizeSetRasterColorTable SetProjection GetProjectionSetGeoTransformGetGeoTransform GetGCPCountSetGCPsGetGCPsGetGCPProjection DitherRGB2PCTCE_None CreateCopyoscloseDelete)r r r rrsrc_ds dst_drivercterr tif_filenamer" tif_filedesc gtiff_drivertif_dsdst_dss 5/usr/lib/python3/dist-packages/osgeo_utils/rgb2pct.pyrgb2pctr@-s\ "F ~/,899 AnE&"4"4!55e f  (6 %%k2J!K=(@ABB __ &&   #   #   #  ++  \ * g%# %-%5%5V%5%D" l''0L  f((&*<* instead of computing it. Can be used to have a consistent color table for multiple files. The palette file must be either a raster file in a GDAL supported format with a palette or a color file in a supported format (txt, qml, qlr).)rUrZrWr zThe input RGB file.)rZrWr z4The output pseudo-colored file that will be created.)parser add_argumentadd_mutually_exclusive_groupinttuplerangestr)rPargvr]groups r? get_parserzRGB2PCT.get_parsers !)  335  %3-(:    M   N;PQ G   rAc tdi|S)NrC)r@)rPrDs r?rFz RGB2PCT.doits   rA)__name__ __module__ __qualname__rKrrfrF __classcell__)rQs@r?rHrHs  *"4*X!rArHc4tj|S)N)rHmain)rds r?rmrms 9>>$ rA__main__)NNrXN)os.pathr3sysrMtypingrosgeorosgeo_utils.auxiliary.baser!osgeo_utils.auxiliary.color_tabler#osgeo_utils.auxiliary.gdal_argparserrosgeo_utils.auxiliary.utilr r r`rcr@rFrHrdrmrhexitrCrAr?rxsB 4=NB -1,0!% WW=)W=)W W # Wt:!j:!zhh  z CHHT#((^rA