U e^ @sdZddlZddlZddlZddlZddlmZddlZddlZddl Z ddl Z ddl m Zdejd<ddZdd d Zd d Zd dZdddZdddZdddZdddZdddZdd d!Zd"d#Zd$d%Zdd'd(Zd)d*Zd+d,Zdd-d.Zd/d0Z de!e!e!e"d2d3d4Z#de!e"e!e"d5d6d7Z$dd8d9Z%dd;d<Z&de!e"e!e"d5d=d>Z'de!e!e!e"d?d@dAZ(dBdCZ)dDdEZ*dFdGZ+dHdIZ,ddJdKZ-dLdMZ.ddNdOZ/ddPdQZ0e,dRddSddTdUfdVdWZ1ddYdZZ2e,dRddSfd[d\Z3dd]d^Z4d_d`Z5ddadbZ6ddcddZ7ddedfZ8ddgdhZ9ddidjZ:ddkdlZ;ddmdnZddwdxZ?dydzZ@d{d|d}d&giifd~dZAdddZBdddZCdddZDdddZEdddZFdddZGddZHdddZIdddZJdddZKddddgfddZLdS)z The source code is adapted from https://github.com/aliaksandr960/segment-anything-eo. Credit to the author Aliaksandr Hancharenka. N)tqdmzAPI-KEYZGOOGLE_MAPS_API_KEYcCsddl}d|jkrdSdSdS)z8Tests if the code is being executed within Google Colab.rN google.colabTF)sysmodules)rrP/home/ankuromar296_gmail_com/.local/lib/python3.8/site-packages/samgeo/common.pyis_colabs rTcCsdt|trX|dr"tj|}n tj|}tj|}tj|sT|rTt ||St ddS)aGets the absolute file path. Args: file_path (str): The path to the file. make_dirs (bool, optional): Whether to create the directory if it does not exist. Defaults to True. Raises: FileNotFoundError: If the directory could not be found. TypeError: If the input directory path is not a string. Returns: str: The absolute path to the file. ~z(The provided file path must be a string.N) isinstancestr startswithospath expanduserabspathdirnameexistsmakedirs TypeError) file_pathZ make_dirsZfile_dirrrrcheck_file_paths     rcCsLddl}ddl}|ds"d|}t|}tj|||}|S)zReturns a temporary file path. Args: extension (str): The file extension. Returns: str: The temporary file path. rN.) tempfileuuidr r uuid4r rjoin gettempdir) extensionrrZfile_idrrrrtemp_file_path<s   rcCs4t|tr0|dr0d|kr0|dddd}|S)z|Get the raw URL for a GitHub file. Args: url (str): The GitHub URL. Returns: str: The raw URL. zhttps://github.com/Zblobz github.comzraw.githubusercontent.comzblob/)r r r replace)urlrrrgithub_raw_urlQs  r"Fc  Csddl} z ddl}Wntk r2tdYdSX|dkr\t|tr\|dr\tj |}tj tj |}tj |st |t|trtj tj |r| st|dtj |St|}d|krd}||||||||||| }| r|dr| |d v}|s td | rptjtj |d}tj||}tj |sdt |||n|tj |W5QRXtj |S) apDownload a file from URL, including Google Drive shared URL. Args: url (str, optional): Google Drive URL is also supported. Defaults to None. output (str, optional): Output filename. Default is basename of URL. quiet (bool, optional): Suppress terminal output. Default is False. proxy (str, optional): Proxy. Defaults to None. speed (float, optional): Download byte size per second (e.g., 256KB/s = 256 * 1024). Defaults to None. use_cookies (bool, optional): Flag to use cookies. Defaults to True. verify (bool | str, optional): Either a bool, in which case it controls whether the server's TLS certificate is verified, or a string, in which case it must be a path to a CA bundle to use. Default is True.. Defaults to True. id (str, optional): Google Drive's file ID. Defaults to None. fuzzy (bool, optional): Fuzzy extraction of Google Drive's file Id. Defaults to False. resume (bool, optional): Resume the download from existing tmp file if possible. Defaults to False. unzip (bool, optional): Unzip the file. Defaults to True. overwrite (bool, optional): Overwrite the file if it already exists. Defaults to False. subfolder (bool, optional): Create a subfolder with the same name as the file. Defaults to False. Returns: str: The output file path. rNzWThe gdown package is required for this function. Use `pip install gdown` to install it.httpzC already exists. Skip downloading. Set overwrite=True to overwrite.z https://drive.google.com/file/d/Tz.ziprzExtracting files...)zipfilegdown ImportErrorprintr r r r rbasenamerrrrr"downloadendswithZipFilesplitextr extractall)r!outputquietproxyspeedZ use_cookiesverifyidZfuzzyresumeZunzip overwriteZ subfolderr%r&out_dirZzip_refr)rrr download_file`s^$        r8vit_hcCs|s$dddddddddd}n&d d dd d dd dddddd}||krltd|dd||dkrtjdtjd}tj|||d}tj|std|d||d}t |||S)aDownload the SAM model checkpoint. Args: model_type (str, optional): The model type. Can be one of ['vit_h', 'vit_l', 'vit_b']. Defaults to 'vit_h'. See https://bit.ly/3VrpxUh for more details. checkpoint_dir (str, optional): The checkpoint_dir directory. Defaults to None, "~/.cache/torch/hub/checkpoints". hq (bool, optional): Whether to use HQ-SAM model (https://github.com/SysCV/sam-hq). Defaults to False. sam_vit_h_4b8939.pthDhttps://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth)namer!sam_vit_l_0b3195.pthDhttps://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pthsam_vit_b_01ec64.pthDhttps://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth)r9vit_lvit_bzsam_hq_vit_h.pthzRhttps://drive.google.com/file/d/1qobFYrI4eyIANfBSmYcGuWRaSIXfMOQ8/view?usp=sharingzsam_hq_vit_l.pthzRhttps://drive.google.com/file/d/1Uk17tDKX1YAKas5knI4y9ZJCo0lRVL0G/view?usp=sharingzsam_hq_vit_b.pthzRhttps://drive.google.com/file/d/11yExZLOve38kRZPfRx_MRxfIAKmfMY47/view?usp=sharingzsam_hq_vit_tiny.pthzDhttps://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_tiny.pth)r9rArBZvit_tinyzInvalid model_type: z. It must be one of z, NZ TORCH_HOMEz~/.cache/torch/hub/checkpointsr<zModel checkpoint for z not found.r!) ValueErrorrr environgetrrrr(r8)Z model_typeZcheckpoint_dirZhqZ model_types checkpointr!rrrdownload_checkpointsN     rGcKs`dddd}t|tr&||kr&||}|dkr6|d}|dkrJtj|}t||fd|i|S)aDownload a checkpoint from URL. It can be one of the following: sam_vit_h_4b8939.pth, sam_vit_l_0b3195.pth, sam_vit_b_01ec64.pth. Args: url (str, optional): The checkpoint URL. Defaults to None. output (str, optional): The output file path. Defaults to None. overwrite (bool, optional): Overwrite the file if it already exists. Defaults to False. Returns: str: The output file path. r;r>r@)r:r=r?Nr:r6)r r r rr)r8)r!r/r6kwargsZ checkpointsrrrdownload_checkpoint_legacys  rIdeflatecKszddlm}ddlm}Wntk r8tdYnX|ds`t|}tj |s`t d|dkr|dstj |dd}n t d d }t|}| |}||||f|dS) aConverts an image to a COG file. Args: source (str): A dataset path, URL or rasterio.io.DatasetReader object. dst_path (str, optional): An output dataset path or or PathLike object. Defaults to None. profile (str, optional): COG profile. More at https://cogeotiff.github.io/rio-cogeo/profile. Defaults to "deflate". Raises: ImportError: If rio-cogeo is not installed. FileNotFoundError: If the source file could not be found. r) cog_translate) cog_profileszThe rio-cogeo package is not installed. Please install it with `pip install rio-cogeo` or `conda install rio-cogeo -c conda-forge`.r#z+The provided input file could not be found.Nz_cog.tif.tif)r)Zrio_cogeo.cogeorKZrio_cogeo.profilesrLr'r rr rrFileNotFoundErrorr-rrE)sourceZdst_pathprofilerHrKrLZ dst_profilerrr image_to_cogs$        rQ EPSG:4326nearestc Ks@ddl}ddlm}m}m} t|tr0t| |}tj |}tj |}tj tj |snt tj ||j|f|} || j|| j| jf| j\} } } | j}||| | | d|j|df|N}td| jdD]6}|f|| ||||| j| j| ||d|qW5QRXW5QRX|r.resolution_to_zoom_levelzNbbox must be a list of 4 coordinates in the format of [xmin, ymin, xmax, ymax]z*Either zoom or resolution must be providedz.Only one of zoom or resolution can be providedg@TXAic s8|}d|d}||fS)N-g@)radianslogtan)latlonxtileytile)EARTH_EQUATORIAL_RADIUSr{rrfrom4326_to3857s z'tms_to_geotiff..from4326_to3857csZ|}d|}|dd|}d|d|jd|}||fS)Nihr\)rrrcospi)rrzoomZlat_rnrrrzrrdeg2nums  0ztms_to_geotiff..deg2numcSsd|}t|dkrTt|dkr0|ddkr0dS|ddD]}|dkr<dSq.is_emptyc s|dkr |S|}|jdkr*dnd}|j}|dkr|d|d<|d|d<||d|d|d|d|d|df}n|}t|d|d} t|d|d} |d| |d| f} |dkr||| n*|j|kr||}|s||| ||S)NRGBRGBArr\rr) rcBytesIOmodesizenewabsZpasteconvertclose) bigim base_sizetile corner_xybboxrrrZnewimZdxZdyZxy0)rsiorrr paste_tiles0  .    z"tms_to_geotiff..paste_tilecSs||d}||d}t|d|} t|d|} t|d||} t|d||} || | | | | | f} | jdkr| ddkr| d} || S)Nrr\rr)rr)roundcroprrrr)rrrx0y0x1y1ZxfracZyfracZx2y2ZimgwZimghZretimrrrfinish_picture!s   z&tms_to_geotiff..finish_picturecsfd}zj|dd}Wq@Wqtk r<|d8}|s8YqXq|jdkrNdS|jsXdS||jS)Nr<)timeoutr\i)rE Exception status_codecontentraise_for_status)r!retryr$)SESSIONrrget_tile.s  z tms_to_geotiff..get_tileFc% s|||\} } |||\} } t| | g\} } t| | g\} } tt| | t| | } t| }g}jd}| D]&\}}| | |j |||dq| | | | f}d}ddg}t t || dD]J\}\}} |||||}|stdt|tt|d|qW5QRX|s^td|||| | | | }t|}d}d |krd d d d g|d <|j||jd|jd|jf|}||\}}||\}} t|||jd}!t| ||jd}"|t|||!dt|| d|" f|t|D]L}# j||#dd}$|$|jd|jdf}$||#d}#|# |$qD|!|std||S)N)zxyr\zDownloaded image /zSaving GeoTIFF. Please wait...GTiffoptionsCOMPRESS=DEFLATEz PREDICTOR=2zZLEVEL=9z TILED=YESru8)dtypezImage saved to )"sortedtupleproductrhfloorceilrfuturesThreadPoolExecutorappendsubmitformat enumeratezipresultr(r zfillZgetbandsGetDriverByNameCreaterZGDT_ByterSetGeoTransformminmax SetProjectionarrayZgetdatareshape GetRasterBand WriteArrayZ FlushCache)%rOlat0lon0lat1lon1rfilenamer0rHrrrrZcornersZtotalnumrexecutorrrrrrkfutrimgZimgbandsdriverZgtiffZxp0Zyp0Zxp1Zyp1ZpwidthZpheightrjr) WKT_3857 concurrentrrrrtr itertoolsr{numpyrrr draw_tile?sz $    $  z!tms_to_geotiff..draw_tilerq)rl)F)#r rr{rconcurrent.futuresrPILrsosgeortrur'httpxZClientrequestsSessionrrr( get_basemapsr r upperr rClistrZMAX_IMAGE_PIXELSZ UseExceptionsZSpatialReferenceZImportFromEPSGZ ExportToWktrUrQr)r/rrryrOrWrl return_imager6r0rHr rurrZ xyz_tilesbasemapsr|ZwestZsouthZeastZnorthZ web_mercatorrrker)rrsrrrrrrrtrrrrr{rrrtms_to_geotiffss               "Erc Cs(t|}|jW5QRSQRXdSN)rarcrPsrc_fprmrrr get_profiles rc Cs(t|}|jW5QRSQRXdSr)rarcrWrrrrget_crss rr\c Csdddlm}t|D}|j||ddddddd}tj|}||j|W5QRSQRXdS)Nrfeaturesr\TFr)bidxZsamplingrjZas_maskZ with_nodataZ geographic precision) rarrcZdataset_featuresgpd GeoDataFrame from_featuresset_crsrW)rrrrmgdfrrr get_featuress    rcCstjj|||fSr)rarXZ from_bounds)Zgeo_boxrYrZrrr set_transformsrcKs&tjj||fddi|}|||S)aQTransform coordinates from one CRS to another. Args: x (float): The x coordinate. y (float): The y coordinate. src_crs (str): The source CRS, e.g., "EPSG:4326". dst_crs (str): The destination CRS, e.g., "EPSG:3857". Returns: dict: The transformed coordinates in the format of (x, y) Z always_xyT)pyprojZ TransformerZfrom_crsrX)rrr^r_rHZ transformerrrrtransform_coordss rcKs@|dst|}tj|f|}|dkr.|jS|j|dddS)zConverts a vector file to a geojson file. Args: filename (str): The vector file path. output (str, optional): The output geojson file path. Defaults to None. Returns: dict: The geojson dictionary. r#NGeoJSONr)r r8r read_fileZ__geo_interface__to_file)rr/rHrrrrvector_to_geojsons r cKs4tj|f|}|j}|dkr&|jSd|SdS)zGets the CRS of a vector file. Args: filename (str): The vector file path. Returns: str: The CRS of the vector file. NzEPSG:)rr rWto_epsg)rrHrZepsgrrrget_vector_crss  r epsg:4326)geojsonr^r_returncCsddl}ddl}|dt|tr.||}tj|dd}|jj }dd|D}||krt dd|Dd d|D||}d dt |d|d D}|S) aConverts a geojson file or a dictionary of feature collection to a list of centroid coordinates. Args: geojson (str | dict): The geojson file path or a dictionary of feature collection. src_crs (str, optional): The source CRS. Defaults to "epsg:4326". dst_crs (str, optional): The destination CRS. Defaults to "epsg:4326". Returns: list: A list of centroid coordinates in the format of [[x1, y1], [x2, y2], ...] rNignorer r cSsg|]}|j|jgqSrrr.0pointrrr sz%geojson_to_coords..cSsg|] }|dqSrrrrrrrrscSsg|] }|dqSr\rrrrrrscSsg|]\}}||gqSrrrrrrrrr sr\) jsonwarningsfilterwarningsr dictdumpsrr geometrycentroidrr)rr^r_rrr centroidsZ centroid_listrrrgeojson_to_coordss"     r%)rcoords coord_crsrc st|tjr|}t|\}}t|N}|j|j||j krZt ||||j f|\}}tj j |j ||f|\}}W5QRXddt||D} fdd| D} t | dkrtdnt | t |krtd| S)aConverts a list of coordinates to pixel coordinates, i.e., (col, row) coordinates. Args: src_fp: The source raster file path. coords: A list of coordinates in the format of [[x1, y1], [x2, y2], ...] coord_crs: The coordinate CRS of the input coordinates. Defaults to "epsg:4326". **kwargs: Additional keyword arguments to pass to rasterio.transform.rowcol. Returns: A list of pixel coordinates in the format of [[x1, y1], [x2, y2], ...] cSsg|]\}}||gqSrr)rcolrowrrrr&sz coords_to_xy..cs8g|]0\}}|dkr|dkr|kr|kr||gqSrrrrZrYrrr(s r!No valid pixel coordinates found./Some coordinates are out of the image boundary.)r npndarraytolistrrarcrYrZrWrrXrowcolrr() rr&r'rHxsysrmrowscolsrrr*r coords_to_xys$    $   r5c sZddlmfdd|D}tjd|i|d}||}|dk rR|j|f|n|SdS)a9 Convert a list of bounding box coordinates to vector data. Args: coords (list): A list of bounding box coordinates in the format [[left, top, right, bottom], [left, top, right, bottom], ...]. src_crs (int or str): The EPSG code or proj4 string representing the source coordinate reference system (CRS) of the input coordinates. dst_crs (int or str, optional): The EPSG code or proj4 string representing the destination CRS to reproject the data (default is "EPSG:4326"). output (str or None, optional): The full file path (including the directory and filename without the extension) where the vector data should be saved. If None (default), the function returns the GeoDataFrame without saving it to a file. **kwargs: Additional keyword arguments to pass to geopandas.GeoDataFrame.to_file() when saving the vector data. Returns: geopandas.GeoDataFrame or None: The GeoDataFrame with the converted vector data if output is None, otherwise None if the data is saved to a file. rboxcsg|] }|qSrr)rcoordr6rrrFsz#boxes_to_vector..r")rWN)Zshapely.geometryr7rrto_crsr )r&r^r_r/rHpolygonsrZgdf_reprojectedrr6rboxes_to_vector3s  r;centerc s|dk rRg}g}|D]<} || d|| d|| d|| dq|dksb|dkrjtdt|*} tjj| j||||f|\| j} W5QRX|dkrddtDSfd dtdt dD} |dk rt | | ||n| SdS) aConverts a list of (row, col) coordinates to (x, y) coordinates. Args: src_fp (str): The source raster file path. rows (list, optional): A list of row coordinates. Defaults to None. cols (list, optional): A list of col coordinates. Defaults to None. boxes (list, optional): A list of (row, col) coordinates in the format of [[left, top, right, bottom], [left, top, right, bottom], ...] zs: zs (list or float, optional): Height associated with coordinates. Primarily used for RPC based coordinate transformations. offset (str, optional): Determines if the returned coordinates are for the center of the pixel or for a corner. output (str, optional): The output vector file path. Defaults to None. dst_crs (str, optional): The destination CRS. Defaults to "EPSG:4326". **kwargs: Additional keyword arguments to pass to rasterio.transform.xy. Returns: A list of (x, y) coordinates. Nr\rrrzrows and cols must be provided.cSsg|]\}}||gqSrrrrrrrsz rowcol_to_xy..cs0g|](}||d|d|gqSrrrror1r2rrrs) rrCrarcrXZxyrWrrhrr;) rr3r4ZboxesZzsoffsetr/r_rHr7rmr^rrr>r rowcol_to_xyTs&  r@c KsPt|tr>t|}|jjj}|jdk rRd|j }nt|t j rR|}t|t rddl }||}tj|dd}|jjj}nt|tstdt|dts|g}g}t|}|j} |j} |D]} | \} } }}||jkrft| | ||jf|\} } t||||jf|\}}tjj|j| | f|\}}tjj|j||f|\}}|||||gq|| | ||gqW5QRXg}|D]t} | \} } }}| dkr| dkr|dkr|dkr| | kr| | kr|| kr|| kr|| ||| gqt|dkrtddSt|dkr2|dSt|t|krLtd |S) aConverts a list of coordinates to pixel coordinates, i.e., (col, row) coordinates. Note that map bbox coords is [minx, miny, maxx, maxy] from bottomleft to topright While rasterio bbox coords is [minx, max, maxx, min] from topleft to bottomright Args: src_fp (str): The source raster file path. coords (list): A list of coordinates in the format of [[minx, miny, maxx, maxy], [minx, miny, maxx, maxy], ...] coord_crs (str, optional): The coordinate CRS of the input coordinates. Defaults to "epsg:4326". Returns: list: A list of pixel coordinates in the format of [[minx, maxy, maxx, miny], ...] from top left to bottom right. Nzepsg:rr r z%coords must be a list of coordinates.r+r\r,)r r rr r"rdvaluesr/rWrr-r.r rr!rrCrarcrYrZrrXr0rrr()rr&r'rHrrrZ new_coordsrmrYrZr8ZminxZminyZmaxxZmaxyZrows1Zcols1Zrows2Zcols2rrrr bbox_to_xys            rB)rrr'rc Ks8t| }|j}W5QRXt|||}t|||f|S)aConverts a geojson file or a dictionary of feature collection to a list of pixel coordinates. Args: src_fp: The source raster file path. geojson: The geojson file path or a dictionary of feature collection. coord_crs: The coordinate CRS of the input coordinates. Defaults to "epsg:4326". **kwargs: Additional keyword arguments to pass to rasterio.transform.rowcol. Returns: A list of pixel coordinates in the format of [[x1, y1], [x2, y2], ...] )rarcrWr%r5)rrr'rHrmr^r&rrr geojson_to_xys  rCc CsRt|>}tj|j||\}}tt|t|t|t|g}W5QRX|Sr)rarcrXr0r-rrr)rr1r2rmr3r4r7rrrget_pixel_coordss ,rDcCs||dSr)r )rdst_fprrrwrite_featuressrFc Cst|dddtj|df|J}t|jdkr>|tjdf}t|jdD]}||||dqLW5QRXdS) Nr0)rnodatar[r.rr\) rgrarcrshaper-Znewaxisrhwrite) rEZdst_arrrPrYrZrXrWrnrorrr write_rasters rKcCs |dddf}t|d}|S)Nr.)r\rr)r-Z transpose)blockrrr chw_to_hwcs rMcCs|d|ftj}|S)N.)astyper-uint8)rLZchannelrrr hwc_to_hwsrPc Cs||}}g}|d|}|d|} t| ||D]^} t| ||D]J} t|| | |} t|| ||} || | || || g|| ggdqFq4|S)Nr)rrrZrYrd)rhrr)raster_hraster_wsample_hsample_wboundhr[blocksrZrYrrZ right_x_boundZbottom_y_boundrrrcalculate_sample_grids"    rXcKs$|j|||f|||ffd|dS)NT)windowZ boundlessZ fill_value)read)rmrrrZrYrHrHrrr read_block4s r[cCs|r||dd|jd|dd|dd|jd|ddf}||dd7}||dd7}||dd|dd}||dd|dd}|j|d|||f|||ffddS)Nrr\)rY)rIrJ)rnZrasterrrrZrYrdrrr write_block:s  r\)r])rrrc  Ks"t| } | j} | d| d} }|\}}|}|}t| ||||d}d| d<d| d<|dk rnt|tj}tj|df| }t|D]}t| f|}||}|j dd }|dk rt j ||t j d }||f||| d | }|dk rt j ||t j d }t||f|qW5QRXW5QRXdS) NrZrY)rQrRrSrTrUr\rirOrr[r) interpolation) foregrounderosion_kernelmask_multiplier)rarcrPrXr-ZonesrOrr[rIcv2resizeZ INTER_LINEARZ INTER_NEARESTr\)rrEfunc data_to_rgb sample_size sample_resizerUr`rarbrHrmrPZrhrwshswZ resize_hwZ sample_gridrnbr$Z uint8_rgb_inZ orig_sizeZuin8_outrrr tiff_to_tiffGsX   rmroc Csztd}td|\}}|||j}t,} | j} t|| |t|||dt | } W5QRXW5QRX| dS)NrMrfrgrhrU.r) rNamedTemporaryFilercZimencoderJtobytesr<rmrMimread) rkrergrhrUZ src_tmpfilesrlr dst_tmpfilerErrrrimage_to_images"   rwc CsBt,}|j}t|||||||dt|}W5QRX|dS)Nrprq)rrrr<rmrcrt) rrerfrgrhrUrvrErrrr tiff_to_images  rxc spddlm}t|(}|}|dk}|j|||jd}W5QRXdd|D}dk rlfdd|D}|S)NrrmaskrXcSsg|]\}}tj|qSrshapelyr"rI)rrI_rrrrsz"tiff_to_shapes..csg|]}|jdqS)Z tolerance)simplify)rrIsimplify_tolerancerrrs)rarrcrZshapesrX) tiff_pathrrrmrjrzrrrrrtiff_to_shapess  rc Ks0||||g}t||f|d|dddd|} | S)NFT)rryrOrlrr0)r) rOrrrrrrrHrrkrrrrs   rc Ksddlm}t|(}|}|dk}|j|||jd} W5QRXdd| D} |dk rx| D]} | dj|d| d<q^tj | } |j dk r| j |j d d |dk r| |} | j |f|dS) aIVectorize a raster dataset. Args: source (str): The path to the tiff file. output (str): The path to the vector file. simplify_tolerance (float, optional): The maximum allowed geometry displacement. The higher this value, the smaller the number of vertices in the resulting geometry. rrrycSs&g|]\}}tj|d|idqS)value)r" propertiesr{)rrIrrrrrsz$raster_to_vector..Nr"r~T)rWinplace)rarrcrZrrXrrrrrWrr9r ) rOr/rr_rHrrmrjrzrfcrorrrrraster_to_vectors"     rcKs,|ds|d7}t||fd|i|dS)aRConvert a tiff file to a gpkg file. Args: tiff_path (str): The path to the tiff file. output (str): The path to the gpkg file. simplify_tolerance (float, optional): The maximum allowed geometry displacement. The higher this value, the smaller the number of vertices in the resulting geometry. .gpkgrNr+rrr/rrHrrrraster_to_gpkgs rcKs,|ds|d7}t||fd|i|dS)aRConvert a tiff file to a shapefile. Args: tiff_path (str): The path to the tiff file. output (str): The path to the shapefile. simplify_tolerance (float, optional): The maximum allowed geometry displacement. The higher this value, the smaller the number of vertices in the resulting geometry. z.shprNrrrrr raster_to_shps rcKs,|ds|d7}t||fd|i|dS)aXConvert a tiff file to a GeoJSON file. Args: tiff_path (str): The path to the tiff file. output (str): The path to the GeoJSON file. simplify_tolerance (float, optional): The maximum allowed geometry displacement. The higher this value, the smaller the number of vertices in the resulting geometry. z.geojsonrNrrrrrraster_to_geojsons rc Csddl}ddlm}dd}i}|D]}z<|d}|||}|||r^|rTqf|||<n|||<Wq(tk r|D]@}||d}|||}|||r|rq|||<q||||<q|Yq(Xq(|t|}|S)zReturns a dictionary of xyz services. Args: free_only (bool, optional): Whether to return only free xyz tile services that do not require an access token. Defaults to True. Returns: dict: A dictionary of xyz services. rNcSs"|}|dD]}t||}q|S)Nr)splitrb)varparamtempZ sub_paramrrr_unpack_sub_parameters0s z,get_xyz_dict.._unpack_sub_parametersr<) collectionsZxyzservices.providers providersrAZrequires_tokenr OrderedDictritems) free_onlyrZxyzrxyz_dictitemr<rZsub_itemrrr get_xyz_dict$s0         rcCs:i}t|d}|D]"}||j}||}|||<q|S)zReturns a dictionary of xyz basemaps. Args: free_only (bool, optional): Whether to return only free xyz tile services that do not require an access token. Defaults to True. Returns: dict: A dictionary of xyz basemaps. )r)rr<Z build_url)rrrrr<r!rrrrSs     rc KsBddlm}t|tr:tj|r:t|}t |tj }| dr&|dk r&t | }|j}|j} |dkrv|j}W5QRXt|} t|} |dkr | dkr| dkrtj}nj| dkr| dkrtj}nR| dkr| dkrtj}n:| dkr| d krtj}n"| d kr| d krtj}ntj}||}|jd krZd |jd|jdd|j|| d} n4|jdkrd |jd|jd|jd |j|| d} |dk r|| d<ntdt j |df| d} |jd kr| |dnD|jdkrt |jd D](}| |dddd|f|dqW5QRXn|!|}|j"|f|dS)aNSave a NumPy array as a GeoTIFF using the projection information from an existing GeoTIFF file. Args: array (np.ndarray): The NumPy array to be saved as a GeoTIFF. output (str): The path to the output image. source (str, optional): The path to an existing GeoTIFF file with map projection information. Defaults to None. dtype (np.dtype, optional): The data type of the output array. Defaults to None. compress (str, optional): The compression method. Can be one of the following: "deflate", "lzw", "packbits", "jpeg". Defaults to "deflate". rrrrMNr\riiiirr)rrZrYrirrWrXrcompresszArray must be 2D or 3D.r[)#rrsr r r rrrcrtcvtColor COLOR_BGR2RGBr+rarcrWrX compressionr-rrfloat32rOZint8Zuint16Zint16Zfloat64rNndimrIrrCrJrhZ fromarraysave)rr/rOrrrHrsrmrWrXZ min_valueZ max_valuemetadatarnrorrrrarray_to_imagegsl             2 r offcKst|trP|drt|}tj|s8td|dt |}t |tj }t j fd|i|t j|fd|i|t |t jf|dS)Nr# Input path  does not exist.figsizecmap)r r r r8r rrrCrcrtrrpltfigureimshowaxisshow)rOrrrZfig_argsZ show_argsrHrrr show_images     rcCs~t}|r0tjtjdtdggdd}ntddddg}|jdd\}}|||d |d d d }||dS) Nrg333333?rrg?g?g?r\r) rgcar-Z concatenaterandomrrIrr)rzZ random_coloraxcolorrVr[Z mask_imagerrr show_masks$rwonc Kst|trP|drt|}tj|s8td|dt |}t |tj }t j |dt |t } ||dk} ||dk} | j| dddf| dddfdd|d d d | j| dddf| dddfd d|d d d |dk rt |t |t dS) Nr#rrrr\rgreen*whiteg?)rmarkerru edgecolorZ linewidthred)r r r r8r rrrCrcrtrrrrrrZscattertitlerr) rkr&labelsZ marker_sizerrrrzrHrZ pos_pointsZ neg_pointsrrr show_pointssD            rc Cs`t}|d|d}}|d|d|d|d}}|tj||f||dddddS)Nrr\rrr)rrrr)rZ facecolorZlw)rrZ add_patchZ Rectangle)r7rrrr[rVrrrshow_boxs "r?ZTkAggrcs<ddl}ddl}ddlm} d|jkr6d}tddS||t|trx| dr\t |}t j |sxtd|dt|tr| drt |}t j |std|dt|} t|} tjd d d |id \\} } | j| f|}| j| fd |i|fdd}| j| d dd |d}||tdS)a]Overlays two images using a slider to control the opacity of the top image. Args: image1 (str | np.ndarray): The first input image at the bottom represented as a NumPy array or the path to the image. image2 (_type_): The second input image on top represented as a NumPy array or the path to the image. alpha (float, optional): The alpha value of the top image. Defaults to 0.5. backend (str, optional): The backend of the matplotlib plot. Defaults to "TkAgg". height_ratios (list, optional): The height ratios of the two subplots. Defaults to [10, 1]. show_args1 (dict, optional): The keyword arguments to pass to the imshow() function for the first image. Defaults to {}. show_args2 (dict, optional): The keyword arguments to pass to the imshow() function for the second image. Defaults to {}. rNrinlinezgThe TkAgg backend is not supported in Google Colab. The overlay_images function will not work on Colab.r#rrrr\ height_ratios)Z gridspec_kwalphacs|jdSr)Z set_alphaZcanvasZ draw_idle)rZfigimg1rrrgSs zoverlay_images..update)rlabelZvalminZvalmaxZvalinit)r matplotlibZmatplotlib.widgetswidgetsrr(Zuser r r r8r rrrCrrtZsubplotsrZSliderZ on_changedr)Zimage1Zimage2rbackendrZ show_args1Z show_args2rrZ mpwidgetsrrZax0Zax1Zimg0rgZslider0rrroverlay_imagess:            rc KsXt|trB|drt|}tj|s8td|dt |}t|tr|dr^t|}tj|sztd|dt |}|j t j kr|d t j}|j t j kr|d t j}|j |j kr| |j }t||jd|jdf}d|}tj||||df|} |r t| |||rPtj|dt| t|tn| SdS) at Blends two images together using the addWeighted function from the OpenCV library. Args: img1 (numpy.ndarray): The first input image on top represented as a NumPy array. img2 (numpy.ndarray): The second input image at the bottom represented as a NumPy array. alpha (float): The weighting factor for the first image in the blend. By default, this is set to 0.5. output (str, optional): The path to the output image. Defaults to False. show (bool, optional): Whether to display the blended image. Defaults to True. figsize (tuple, optional): The size of the figure. Defaults to (12, 10). axis (str, optional): The axis of the figure. Defaults to "off". **kwargs: Additional keyword arguments to pass to the cv2.addWeighted() function. Returns: numpy.ndarray: The blended image as a NumPy array. r#rrrr\rrN)r r r r8r rrrCrcrtrr-rrNrOrdrIZ addWeightedrrrrrr) rZimg2rr/rrrrHbetaZ blend_imgrrr blend_images_s:                 rc Ksddl}z|dkrt}d}d}t||f|tj|d}t}t||ddkrdd}nd}t|t||s| |zt |Wn YnXt d Wn*t k r} z t | W5d} ~ XYnXdS) a_Updates the package from the GitHub repository without the need to use pip or conda. Args: out_dir (str, optional): The output directory. Defaults to None. keep (bool, optional): Whether to keep the downloaded package. Defaults to False. **kwargs: Additional keyword arguments to pass to the download_file() function. rNzJhttps://github.com/opengeos/segment-geospatial/archive/refs/heads/main.zipzsegment-geospatial-main.zipzsegment-geospatial-mainpipzpip3 install .z pip install .zPackage updated successfully.) shutilr getcwdr8rrchdirwhichsystemrmtreeremover(r) r7ZkeeprHrr!rZpkg_dirZwork_dircmdrrrrupdate_packages2      rrvc sz@ddlddl}ddl}ddlddl}ddlddlmWntk r\tdYnX dkrn| |j fd|i| ddi _ j |dd z j jd d Wn YnXg _g _j jd d j jdd    _ _d}d} d} ddi} jdddjdd| ddjddddjdd| ddjddd djdd| ddjdd!d"djdd| dd jd dgd#d| j|| d$d%jdd&d| jd'| d$d(jdd)d| jd'| d$d(jd*ddj| d+d,jd-ddd.jd/ddd.| j_| j_| j_jd0dd1d2ddj|| d$| d3 jdd4jd5| d$| d6jdd7d8jd9| d$| d:g gjgjd;d+d<g} fd=d>} | d?j!j|| |d@d< g_"| g_"g_"|j#dAdBgdC} fdDdE  fdFdG} |d? |d?fdHdI}|$|fdJdK} |d? fdLdM} |d? fdNdO} %| fdPdQ} |d? fdRdS fdTdU} |d? f dVdW} |d?j&dXdY} '|| _( S)ZaCDisplay the SAM Map GUI. Args: sam (SamGeo): basemap (str, optional): The basemap to use. Defaults to "SATELLITE". repeat_mode (bool, optional): Whether to use the repeat mode for the draw control. Defaults to True. out_dir (str, optional): The output directory. Defaults to None. rN FileChooserKThe sam_map function requires the leafmap package. Please install it first. repeat_modecursor crosshairFrrs layer_name Foreground)Zlayersr< Background280px90px0px 0px 0px 4pxdescription_widthinitialTToolbargear28pxrYrZpaddingrtooltipiconlayoutClose the tooltimesprimaryrZrYrrrr button_stylerzLoad foreground points plus-circlezLoad background points minus-circlez Class Type:rYr)r descriptiondisabledstylerz Foreground #:Z135px)rrrrrz Background #:SegmentrrrrrSaverrrResetz Mask opacity:r\rrrrrreadoutcontinuous_updaterr Regularize130pxrrrrColor#ffff00140pxZconciserrrr0px 4px 0px 4pxrcs,|dr(d}|dk r(|jjddS)NrMasksopacity) find_layerinteractrchangeZ mask_layermopacity_sliderrropacity_changedvs z$sam_map_gui..opacity_changedrrYr max_width mouseenter mouseleaverOZwatched_eventsc s|jdk r`zt|j}|j}dd|D}|D]}jrtrlj|d|dfdddd}n&j|d|dgjddd d d }j |j |t j _q<jr.marker_callback..r\rrrlocationradiusrZ fill_colorrdarkredr<Z marker_colorZ icon_colorr(rrrmarker_controlF)selectedrr r#rr CircleMarkerMarker AwesomeIconfg_layeradd fg_markersrrbg_layer bg_markersrr(r-controlsremove_controldelattr)chooserrr$r&r8rr)bg_countfg_count ipyleafletr minus_buttonr/ plus_buttonrrmarker_callbacksh        z$sam_map_gui..marker_callbackcs|drjtjd}t|jddd}d|_ddd g|_|j|d d } ||_ n,t d rj j kr j j dS) Nr SANDBOX_PATH454pxrY)r sandbox_pathrTz*.shpz *.geojsonz*.gpkgtoprightZwidgetpositionr-)r rDrErLayout use_dir_iconsfilter_patternregister_callback WidgetControl add_controlr-hasattrr7r8r)rrD filechooserr-)rr=rr@rrrmarker_button_clicks&      z(sam_map_gui..marker_button_clickcsB|ddkrg_n&|ddkr>js>g_d_d_dSNtyper$r%Fchildrenrevent close_buttontoolbar_buttontoolbar_footertoolbar_headertoolbar_widgetrrhandle_toolbar_events   z)sam_map_gui..handle_toolbar_eventcs,|drd_g_njs(g_dSNrFrrTrrWrrtoolbar_btn_clicks  z&sam_map_gui..toolbar_btn_clickcs2|dr.d_jjkr&jdSr^rtoolbar_controlr7r8rr`rrYr\rrclose_btn_clicks   z$sam_map_gui..close_btn_clickc s"z|ddkr|d}jdkrtrBjt|dddd}nj|jddd d d }|j |t j_ndjd krtrjt|dd d d}nj|jdd d d d }|j |t j _Wn6t t fk r}ztd|W5d}~XYnXdS)NrRZclick coordinatesrrrr'rr*r+r,rrrz Error handling map interaction: )rErrr/rr0r1r3r4rrr6rKeyErrorr()rHZlatlonrr)r;r5r<r2r=r radio_buttonsrrhandle_map_interactionsV       z+sam_map_gui..handle_map_interactionc s |drd_tjdkrJtdd_W5QRdSz~ddjD}ddjD}||}dgt|dgt|}d td }tj |}j ||d |d  d dk rވ  d  ddk r  dt dr8tjjr8ztjWn YnXznj|ddjd ddjr|d d}|d d}t||t||dji}j|d|dddWn YnXd_|_Wn2tk r} zd_t| W5d} ~ XYnXW5QRXdS)NrFrz#Please add some foreground markers.cSs g|]}|jd|jdgqSr\rr(rrrrrrAsz=sam_map_gui..segment_button_click..cSs g|]}|jd|jdgqSrjrkrlrrrrEsr\Zmasks_rMrR) point_coords point_labelsZ point_crsr/r Regularized prediction_fpZBlues)rHrrr zoom_to_layerr _rect.gpkgrrrZ info_moderq)r clear_outputrr4r(r6 random_stringr rrpredictr remove_layerrNrrpr add_rasterr r regularize add_vectorr) r fg_points bg_pointsrmrnrvector vector_rec vector_styler) colorpickerrr r7r/ rectangularsamsegment_buttonrrsegment_button_click5s         z)sam_map_gui..segment_button_clickc s |jdk rz|j}j||dd}t||jrX|dd}t||ddjD}ddjD}t ||ddt ||ddWn*t k r}z t |W5d}~XYnXt d rj jkrj td d _W5QRXdS) NrMrrrcSs g|]}|jd|jdgqSrjrkrlrrrrsz=sam_map_gui..filechooser_callback..cSs g|]}|jd|jdgqSrjrkrlrrrrsz_fg_markers.geojsonz_bg_markers.geojson save_controlF)r.rfrpr rrryr4r6coords_to_geojsonrr(rNrr7r8r9)r:rr}r~r{r|rrr/rr save_buttonrrrfilechooser_callbacks<         z)sam_map_gui..filechooser_callbackc s|drxdtjd}td|jddd}d|_dg|_|j|d d } ||_ W5QRXn,t d rj j kr j td dS) NrrAz masks.tifrBrCrrrDrT*.tifrErFr)r rDrErrHrIrJrKrLrMrrNr7r8r9rrDrOrrrr=rr/rrrsave_button_clicks*     z&sam_map_gui..save_button_clickcs|drd _d_d_d_d_zdddk r^dtdrd_g_g_j j d_d_zt jWn YnXWn YnXdS) NrFrrrror4r)rrtrwrclear_drawingsrNZ user_roisr4r6r2Z clear_layersr5r rrpr`) r;rr<rr r/r reset_buttonrrrrreset_button_clicks4    z'sam_map_gui..reset_button_clickrErF))rrleafmapr= ipyevents ipywidgetsipyfilechooserrr'rMap default_style add_basemaprxrOr4r6Z LayerGroupr3r2r5 ToggleButtonrHZ RadioButtonsZIntTextrrY FloatSliderCheckbox ColorPickerVBoxHBoxobserveOutputrTEvent on_dom_eventZon_interactionrLrMrc)rbasemaprr7rHrrr widget_width button_widthrrZbuttonsr! toolbar_eventrPr]rarerirrrrcr)rr;r5rXrr<r2rr=rr@r>r r7r/r?rhrrrrrrrYrZr[r\rr sam_map_guisj                   ;     0 N $   rcs4ddlddl}|jdfddt|DS)zGenerates a random string of fixed length. Args: string_length (int, optional): Fixed length. Defaults to 3. Returns: str: A random string rNrc3s|]}VqdSr)choicer=lettersrrr sz random_string..)rstringascii_lowercaserrh)Z string_lengthrrrrrus ruc Csddl}t|dkrdSdgd}|D]$}dd|did}|d |q&||}|dk rt|d }||W5QRXn|SdS) aConvert a list of coordinates (lon, lat) to a GeoJSON string or file. Args: coords (list): A list of coordinates (lon, lat). output (str, optional): The output file path. Defaults to None. Returns: dict: A GeoJSON dictionary. rNZFeatureCollection)rRrZFeatureZPoint)rRrf)rRr"rrr[)rrrr!rcrJ)r&r/rZfeature_collectionr8featureZ geojson_strfrrrrs     rrrrrrrrcsttr(drttnttjr6ntdggfdd}t dt d|t dt dt fS)aShow a canvas to collect foreground and background points. Args: image (str | np.ndarray): The input image. fg_color (tuple, optional): The color for the foreground points. Defaults to (0, 255, 0). bg_color (tuple, optional): The color for the background points. Defaults to (0, 0, 255). radius (int, optional): The radius of the points. Defaults to 5. Returns: tuple: A tuple of two lists of foreground and background points. r#z+Input image must be a URL or a NumPy array.csz|tjkr<||ft||fdtdn:|tjkrv||ft||fdtddS)Nrrs)rcZEVENT_LBUTTONDOWNrcirclerZEVENT_RBUTTONDOWN)rVrrbg_colorfg_colorrkZ left_clicksr)Z right_clicksrrget_mouse_coordinates: s  z*show_canvas..get_mouse_coordinatesrsr)r r r r8rcrtr-r.rCZ namedWindowZsetMouseCallbackrZwaitKeyZdestroyAllWindows)rkrrr)rrrr show_canvas s        rcCsddl}t|tr|g}|D]v}|dr4d|}d|}|j||jd}|j}|dkrr| dk rrq|rRt | d qR| qdS) zInstall a Python package. Args: package (str | list): The package name or a GitHub URL or a list of package names or GitHub URLs. rNzhttps://github.comzgit+z pip install )stdoutzutf-8) subprocessr r r PopenrPIPErreadlinepollr(decodestripwait)packagerpackagescommandprocessr/rrrinstall_packagea s     r?viridisc szLddlddl}ddl} ddlddl} ddlddlm} ddlm Wnt k rht dYnX dkrz| | j f|ddi_ j|ddzj jd d Wn YnXd } d } d }ddi}jdddjdddddjddddjdddddjd|j| |ddjddd |d!ddj| |d|d" jd#dd d!|ddj| |d|d$ jd%| ||j| |dd&jd'dd |ddj| |d|d(fd)d*}|d+jdd,jd-|d|d. jdd/d0jd1|d|d2jd3ddj|d4d5jd6ddd7jd8ddd7 | j_| j_| j_jj| || d9d: g_ gj gjd d4d: g_g_| j d;dd?}|!|fd@dA}|d+fdBdC}|d+ f dDdE}|d+ fdFdG fdHdI}|d+ f dJdK} |d+j"dLdM}#||_$S)NzDisplay the SAM Map GUI. Args: sam (SamGeo): basemap (str, optional): The basemap to use. Defaults to "SATELLITE". out_dir (str, optional): The output directory. Defaults to None. rNrrrrFrrsrrrrrrTrrrrrrrrrrrz Text prompt:r)rrrzBox threshold:r\g{Gz?) rrrrstepr r rrzText threshold:) rrrrrr r rrzPalette:)rrrrrzOpacity:r cs8|dr4tdr4j}|dk r4|jjddS)Nrrr)rNrrrrrrrrr! s   z%text_sam_gui..opacity_changedrrrrrrrrrrrrr r r"rr$r%r&csB|ddkrg_n&|ddkr>js>g_d_d_dSrQrSrUrWrrr]9 s   z*text_sam_gui..handle_toolbar_eventcs,|drd_g_njs(g_dSr^r_r`rWrrraD s  z'text_sam_gui..toolbar_btn_clickcs2|dr.d_jjkr&jdSr^rbr`rdrrreN s   z%text_sam_gui..close_btn_clickc s|drd _t jdkr:tdnjdkrPtdntd jdd}tj|dt d }znj j jj j|d |_  |dk rĈ  | |d dk r  |d Wn4tk r$}zt|W5d}~XYnXtj|rzj||jjddd |_jr|d d}|d d}t||t||dji}j||d |dddWn,tk r}z t|W5d}~XYnXW5QRXdS)NrFrz!Please enter a text prompt first.z!Please run sam.set_image() first.z Segmenting... r}rM)r/Z_rectz Regularized)rpaletterrHrqrrrrrs)rrtrr(rOr r rrrurvr/rrwrrrxrrryrz)rrrrr}r~r) box_slider cmap_dropdownrrr r7r/rrr text_prompt text_sliderrrrW sr          z*text_sam_gui..segment_button_clickc s|jdk rzJ|j}j||dd}t||jrX|dd}t||Wn*tk r}z t|W5d}~XYnXt drj j kr j t dd_W5QRXdS)NrMrrrrF)r.rfr/r rrryrr(rNrr7r8r9)r:rr}r~rrrrr s       z*text_sam_gui..filechooser_callbackc s|drtds*tdnftjd}tjd|jddd}d |_ d g|_ | j |d d } ||_W5QRXn,td rȈjjkrȈjtd dS)Nrrz&Please click the Segment button first.rArMrBrCrTrrErFr)rtrNr(r rDrErrrHrIrJrKrLrMrr7r8r9rrrrr s0        z'text_sam_gui..save_button_clickcs|drd_d_d_d_d_d _d_d_z8tdrnjdk rnjWn YnXdS)NrFrrrrr)rrtrNrrrwrr`) rrrr r/rrrrrrrr s  z(text_sam_gui..reset_button_clickrErF)%rrrr=rrZleafmap.colormapsZ colormapsrrr'rrrrrxrOrrHTextrZDropdownZlist_colormapsrrrrrYrrrTrrrrLrMrc)rrr7Z box_thresholdZtext_thresholdrrrHrrrcmrrrrr!rr]rarerrrrcr)rrrXrrrr=rr r7r/rrrrrrrrrYrZr[r\rr text_sam_gui sJ                        "=    rcKst|trt|}nt|tjr(|}ntd|jdd}tj||jdddd}|dk rn|j |d d |dk r|j |f|n|SdS) a Regularize a polygon GeoDataFrame. Args: source (str | gpd.GeoDataFrame): The input file path or a GeoDataFrame. output (str, optional): The output file path. Defaults to None. Returns: gpd.GeoDataFrame: The output GeoDataFrame. z7The input source must be a GeoDataFrame or a file path.cSs|jSr)Zminimum_rotated_rectangle)Zgeomrrr rzregularize..r"r\r)r"dataNT)r) r r rr rrCr"applyZdropr9r )rOr/rWrHrr:rrrrry s   ryrc Cszddlm}Wntk r,tdYnXt|tr^|dr^|dd}t|||}||}t j |s~t |t|t r|}|}nt|tr|d}|d}|j} |j} | |||t | |||dk} | |||t | |||dk} |} t| D]l}t| D]Z}|||}|||}t||| }t||| }|| dkr|t||d}|| dkrt||d}||}||}|d|d |d }|d }|||||j|dj}| d|| d| dd| d || d d| d f}||||t|jD]<}||d}||d}|||||}||q>d}q&qd}dS)aSplit a raster into tiles. Args: filename (str): The path or http URL to the raster file. out_dir (str): The path to the output directory. tile_size (int | tuple, optional): The size of the tiles. Can be an integer or a tuple of (width, height). Defaults to 256. overlap (int, optional): The number of pixels to overlap between tiles. Defaults to 0. Raises: ImportError: Raised if GDAL is not installed. rrtZGDAL is required to use this function. Install it with `conda install gdal -c conda-forge`r#rrr\z/tile_r}rMrrrN)rrtr'r r r rr8ZOpenr rrrrxrZ RasterXSizeZ RasterYSizeZGetGeoTransformrhrrrrZ RasterCountrZDataTyperrZ GetProjectionZ ReadAsArrayr)rr7Z tile_sizeoverlaprtr/ZdsZ tile_widthZ tile_heightrYrZZ num_tiles_xZ num_tiles_yZ geotransformrojZx_minZy_minZx_maxZy_maxZ output_filerZtile_dsZtile_geotransformrrjZ tile_bandZ tile_datarrr split_raster s                 rrrrc Csbddl}zddlm}Wntk r4tdYnX|tj||}|j|||||ddS)a^Merge a directory of rasters into a single raster. Args: input_dir (str): The path to the input directory. output (str): The path to the output raster. input_pattern (str, optional): The pattern to match the input files. Defaults to "*.tif". output_format (str, optional): The output format. Defaults to "GTiff". output_nodata (float, optional): The output nodata value. Defaults to None. output_options (list, optional): A list of output options. Defaults to ["COMPRESS=DEFLATE"]. Raises: ImportError: Raised if GDAL is not installed. rNrr)rZ dstNodatar)globrrtr'r rrZWarp) Z input_dirr/Z input_patternZ output_formatZ output_nodataZoutput_optionsrrtZ input_filesrrr merge_rasterss s r)T) NNFNNTTNFFTFF)r9NF)NNF)NrJ)rRrST)NNrprqFFFF)r\)N)rr)r)rRN)NNNNr<NrR)r)r)r)r)N)rnNr^)N)NN)N)N)N)T)T)NNrJ)F)rrrNN)rFTrr)NF)rvTN)r)N)rrr)rvNrrrr)NrR)rr)M__doc__r rrcrr-rr|rraZ geopandasrZmatplotlib.pyplotZpyplotrrDrrrr"r8rGrIrQrUrrrrrrr rr rr%r5r;r@rBrCrDrFrKrMrPrXr[r\rmrwrxrrrrrrrrrrrrrrrrrrurrrrryrrrrrrsd     W ?  ( < !   $ % # 9 \      ?    #    /  Z   2  I H -   % B! f  u