o Nrfv@sLddlZddlZddlmmZd ddZd ddZddZ d d d ZdS) Nc Cspg}|D]1}|d}t|dkrtdt|d}t|d}t||||d}||t|fg7}q|S)N.zNot enough tile info present)splitlen IndexErrorinttilesreshapectfZformat_dense_tile) arrayZtile_ids not_nan_arrayZ tile_valuesZtile_idpartszx ret_arrayr_/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/clodius/tiles/npvector.py tiles_wrappers    rc Cst|j}tt||td}|dkrdn|}d||}||}|r=|dg}|dg}|d|d|}n dg}|jdg}t|jdkrStd|j||||||dS)z, Get the tileset info for the array rrrz$The array shape is not a vector type) max_widthmin_posmax_posmax_zoombins_per_dimensionZ tile_size)maxshapemathceillogr ValueError) rZboundsrmax_dimrrZscale_uprrrrr tileset_infos(      r$c Csn|jd}tt||td}|dkrdn|}d|||}||}t|jd||}|||fS)a Return the maximum zoom level and data corresponding to the zoom level and position of the array. Parameters ---------- array: np.array The array containing the raw numpy data z: int The zoom level x: int The x position rr)rrr r!min) rrrbin_sizer#rZ tile_widthx_startx_endrrrmax_zoom_and_data_bounds>s  r)cCst||||\}}}|||}d||} | t|jd| } | |jd} tj|d| ffdtjfd} tj| d| fdd} |durRt |||}n|||}tj|d| fdtjfd}tj|d| fdd}| |d} || jd}tj| d|fdtjfdS) a Return tiles at the given positions. Parameters ----------- array: np.array An nxn array containing values z: int The zoom level (0 corresponds to most zoomed out) x: int The x tile position not_nan_array: np.array An array storing the number of values which are not nan in the original array. Can be precalculated for speed. bin_size: int The number of values per bin rrZconstant)Zconstant_valuesrr)ZaxisN) r)rr rnppadnanZnansumr isnan)rrrrr&rr'r(dataZ num_to_sumZdivisible_x_widthZdivisible_x_padarZ not_nan_datanaZ norm_arrayZx_padrrrr ^s"    r )N)Nr) rnumpyr*Zclodius.tiles.formatr formatr rr$r)rrrrs  %