o Nrf@szddlZddlZddlmZddZddZdd Zd d Zdd e d e de fddZ ddZ ddZ dddZ ddZdS)Ntiles_wrapper_2dc Csvt|}|}|d}|d|d|d|d|d|d|d|d ddg|d|dgd }||S) NzSELECT * from tileset_inforr) Z zoom_step max_lengthZassemblyZ chrom_namesZ chrom_sizesZ tile_sizeZmax_zoom max_widthZmin_posZmax_pos)sqlite3connectcursorexecuteZfetchoneclose)filepathconncrow tileset_infor^/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/clodius/tiles/bed2ddb.pyrs  rcCst|SN)rrrrrget_2d_tileset_inforcCs>t|dkrgSt|dddk}|rt||St||S)Nr.r)lensplittiles_1dtiles_2d)rtile_idsZis_1drrrtiles"s   r#cCsHg}|D]}|d\}}}||t|t|t|t|fg7}q|S)a< Generate 1D tiles from this dataset. Parameters ---------- filepath: str The filename of the sqlite db file tile_ids: [str...] A list of tile ids of the form Returns ------- tiles: [(tile_id, tile_value),...] A list of values indexed by the tile position r)r get_1d_tilesint)rr"Z to_returnZtile_id_zxrrrr .s (r zoom tile_x_pos num_tilescCsRt|}t|}|}|dd|}||}|||} d|d|d| d|d|d| d } || } tt} | D]`} z | d  d }Wn t yZ| d }Ynw| d }| d }| d}| d}t |||D]/}||}|d |}||kr||kr| |||||| d| d|| d ddg7<qrqB| | S)a Retrieve a contiguous set of tiles from a 2D db tile file. Parameters ---------- filepath: str The filename of the sqlite db file zoom: int The zoom level tile_x_pos: int The x position of the first tile num_tiles: int The number of tiles to retrieve Returns ------- tiles: {pos: tile_value} A set of tiles, indexed by position r rz SELECT fromX, toX, fromY, toY, chrOffset, importance, fields, uid FROM intervals, position_index WHERE intervals.id=position_index.id AND zoomLevel <= z AND rToX >= z AND rFromX <= z UNION SELECT fromX, toX, fromY, toY, chrOffset, importance, fields, uid FROM intervals, position_index WHERE intervals.id=position_index.id AND zoomLevel <= z AND rToY >= z AND rFromY <= z r utf-8rrrrrr  ZxStartZxEndZyStartZyEndZ chrOffsetZ importanceuidfields)rr rrrfetchallcol defaultdictlistdecodeAttributeErrorrangerr)rr)r*r+Zts_inforr tile_widthtile_x_start_postile_x_end_posqueryrowsnew_rowsrr/x_startx_endy_starty_endi tile_x_start tile_x_endrrrr$Esd        r$cGt|Sr)r$argsrrr get_1D_tilesrrIcst|fddS)Ncst|||||fSr get_2d_tiles)r'r(yrrrsztiles_2d..r)rr"rrrr!s r!cCst|}t|}|}|dd|} | |} | || } | |} | || } d|| | | | }||}tt }|D]}z |d d}Wn t yX|d}Ynw|d}|d}|d}|d}t |||D]U}t |||D]K}|| }|d| }|| }|d| }||kr||kr||kr||kr|||f|d|d|d|d|d |d ||d  d d g7<qyqpq@||S)a% Retrieve a contiguous set of tiles from a 2D db tile file. Parameters ---------- db_file: str The filename of the sqlite db file zoom: int The zoom level tile_x_pos: int The x position of the first tile tile_y_pos: int The y position of the first tile numx: int The width of the block of tiles to retrieve numy: int The height of the block of tiles to retrieve Returns ------- tiles: {pos: tile_value} A set of tiles, indexed by position r ra SELECT fromX, toX, fromY, toY, chrOffset, importance, fields, uid FROM intervals,position_index WHERE intervals.id=position_index.id AND zoomLevel <= {} AND rToX >= {} AND rFromX <= {} AND rToY >= {} AND rFromY <= {} r r,rrrrrr r-r.)rr rrformatrr1r2r3r4r5r6r7rr)Zdb_filer)r*Z tile_y_posZnumxZnumyrrrr8r9r:Ztile_y_start_posZtile_y_end_posr;r<r=r>r/r?r@rArBrCjrDrEZ tile_y_startZ tile_y_endrrrrKsf           rKcGrFrrJrGrrr get_2D_tilesrrP)r)rr) collectionsr2r utilsrrrr#r r%r$rIr!rKrPrrrrs  V  b