o tf @sNddlmZddlZddlmZddlmZmZeee fZ GdddZ dS)) annotationsN) bisect_right)IterableTuplec@sveZdZdZ d!d"ddZed#d d Zed$d dZd$ddZd%ddZ d&ddZ d'ddZ d(ddZ d)ddZ d S)*Scalea A bidirectional mapping between a composite genomic coordinate system and a partition of that coordinate system into a sequence of bins. The partition is a sequence of bins of a fixed size, with the exception of the last bin in each chromosome, which may be smaller than the fixed size. Bins do not cross chromosome boundaries. The scale provides a mapping from a genomic coordinate to the index of the bin in which it falls, and an inverse mapping from a bin index to the genomic coordinate of the bin's start. Parameters ---------- chromsizes : dict | list[tuple] A dictionary of chromosome names and lengths or a list of tuples of chromosome names and lengths. binsize : int, optional The size of each bin in the partition in bp (default: 1). Notes ----- The genomic coordinates are 0-based and the bins of the partition are half-open intervals, i.e. the start coordinate of a bin is included in the bin, but the end is not.  chromsizes*dict[str, int] | Iterable[tuple[str, int]]binsizeintcspt|}t|\}}fdd|D}ttj|dd}||_||_||_tt||dd|_ |_ dS)Ncsg|] }|dqSr).0lengthr r W/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/higlass/_scale.py ,sz"Scale.__init__..r)initial) dictzipitemslist itertools accumulate _chrom_names_chrom_offsets_chrom_lengths_map_chrom_offsets_map_binsize)selfrr nameslengthsZlengths_binnedZ chrom_offsetsr rr__init__%s zScale.__init__returndict[str, int]cC|jS)zK A dictionary of the ordered chromosome names and lengths. )rr r r rr4zScale.chromsizescCr&)z> The size of each bin in the partition in bp. )rr'r r rr ;r(z Scale.binsizecCs |jdS)Nr)rr'r r r__len__Bs z Scale.__len__strcCsd|jd|jdS)NzScale(chromsizes=z , binsize=)rr r'r r r__repr__EszScale.__repr__Iterable[tuple]ccsd|jfVd|jfVdS)Nrr r,r'r r r __rich_repr__Hs zScale.__rich_repr__gposGenomicPositioncCs>|\}}|j|}|j|}tdt||d}|||jS)zY Returns the index of the bin in which the given genomic position falls. rr)rrmaxminr)r r0chromposZ chrom_offsetZclenr r r__call__Ls   zScale.__call__indexcCsX|jd}tdt||d}t|j|}|j|d}||j|d}|||jfS)zZ Returns the genomic position of the start of the bin at the given index. rrr)rr2r3rrr)r r7Zn_binsir4Z rel_offsetr r rinvertVs  z Scale.invertcCs t|j|S)zo Returns a new scale with the same genomic coordinate system but a different bin size. )rr)r r r r rrebinas z Scale.rebinNr )rr r r )r$r%)r$r )r$r*)r$r.)r0r1r$r )r7r r$r1)r r r$r)__name__ __module__ __qualname____doc__r#propertyrr r)r-r/r6r9r:r r r rr s         r) __future__rrbisectrtypingrrr*r r1rr r r rs