o tf~h@sddlmZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlZddlmZddlmZddlmZgdZerFddlmZGdd d ZGd d d ZGd d d ZGdddejeeeZGdddejeeeZGdddej eeeZ Gdddej!eeeZ!Gdddej"eeeZ#e ee ee!e#fZ$e de$dZ%Gdddej&e%eee%Z&e de&dZ'ded!d"Z(Gd#d$d$ej)e&e%eee%Z)e d%e$dZ*e d&e$dZ+dfd.d/Z,e-e,d0Z.e-e,d1Z/Gd2d3d3eZ0e dgdhd8d9Z1e dgdid;d9Z1e dgdjd=d9Z1 dgdkd@d9Z1dddAdBddddCdldPdQZ2dgdmdTdUZ3e dVe eefdZ4dndWdXZ5e ddYdod[d\Z6e ddd]dpdcd\Z6ddYdqddd\Z6dS)r) annotationsN) defaultdict) TYPE_CHECKINGClassVarGenericLiteralTypeVarUnioncastoverload) BaseModel) EnumTrack HeatmapTrack"IndependentViewportProjectionTrack CombinedTrack PluginTrackTrackTViewViewTViewconfconcathconcatvconcattrackviewcombinedividelock)TilesetResourcec@eZdZ d d ddZd S) _PropertiesMixinFself utils.ModelTinplaceboolreturncKs4|r|nt|}|D] \}}t|||q |S)acConfigures top-level properties. Updates top-level properties for a Track, View, or Viewconf. This is really a convenience to allow method chaining/derived objects to be created without multiple lines of mutating the classes. For example, >>> view = hg.view(hg.track("heatmap")) >>> updated_view = view.properties(zoomFixed=True) >>> assert view.uid != updated_view.uid >>> assert view.zoomFixed is None >>> assert updated_view.zoomFixed is True inplace : bool, optional Whether to modify the existing track in place or return a new track with the options applied (default: `False`). **fields : dict The updated properties for . Returns ------- track : A track with the the newly specified track options. )utils copy_uniqueitemssetattr)r!r#fieldsmodelkvr.T/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/higlass/api.py properties6sz_PropertiesMixin.propertiesNF)r!r"r#r$r%r")__name__ __module__ __qualname__r0r.r.r.r/r 5r c@r) _OptionsMixinFr!rr#r$r%cKs2|r|nt|}|jduri|_|j||S)aConfigures options for a Track. A convenience method to update `track.options`. Parameters ---------- inplace : bool, optional Whether to modify the existing track in place or return a new track with the options applied (default: `False`). **options : dict Options to pass down to the underlying track object. Returns ------- track : A track with the the newly specified track options. Examples -------- >>> track = hg.track("heatmap") >>> derived = track.opts(colorRange=["rgba(255,255,255,1)", "rgba(0,0,0,1)"]) >>> assert track.options is None >>> assert isinstance(derived.options["colorRange"], list) N)r&r'optionsupdate)r!r#r7rr.r.r/opts[s   z_OptionsMixin.optsNr1)r!rr#r$r%r)r2r3r4r9r.r.r.r/r6Zr5r6c@seZdZ d d d d Zd S) _TilesetMixinFr!rtilesetrr#r$r%cCs(|r|nt|}|j|_|jj|_|S)aReplace or add a tileset to a Track. A convenience method to update a track with a tileset. Parameters ---------- tileset : TilesetResource A tileset resource returned from `hg.server`. inplace : bool, optional Whether to modify the existing track in place or return a new track (default: `False`) Returns ------- track : A track with the bound tileset. )r&r'serverr;uid tilesetUid)r!r;r#rr.r.r/r;s z_TilesetMixin.tilesetNr1)r!rr;rr#r$r%r)r2r3r4r;r.r.r.r/r:sr:c@eZdZdZdS)r zRepresents a generic track.Nr2r3r4__doc__r.r.r.r/r r c@r?)rz&Represets a specialized heatmap track.Nr@r.r.r.r/rrBrc@r?)rz8Represents a view-independent viewport projection track.Nr@r.r.r.r/rsrc@r?)rz-Represents a track combining multiple tracks.Nr@r.r.r.r/rrBrc@eZdZUdZded<dS)rz#Represents an unknown plugin track.z ClassVar[str] plugin_urlNr2r3r4rA__annotations__r.r.r.r/rs  rr)boundc@sheZdZdZ   dd d d Zd!d dZd!ddZddZddZd"ddZ ddZ  d#d$ddZ dS)%rz?Represets a HiGlass View that is generic over the inner tracks.NFxhgs.Domain | Noneyr#r$cCs2|r|nt|}|dur||_|dur||_|S)aEConfigures the view x and/or y domain(s). Parameters ---------- x : tuple[float, float], optional The x domain for the view (default: `None`) y : tuple[float, float], optional The y domain for the view (default: `None`) inplace : bool, optional Whether to modify this view inplace. If `False` (default), a new view is created and returned with the domains applied (default: `False`) Returns ------- view : a View with the updated x & y domains N)r&r'ZinitialXDomainZinitialYDomain)r!rHrJr#rr.r.r/domains z View.domainotherView[TrackT] | Viewconf[TrackT]cC t||Sa8Horizontally concatenate with another view or viewconf. A convenience method for `hg.hconcat`. Parameters ---------- other : View | Viewconf The other view or viewconf to combine with. Returns ------- viewconf : A combined Viewconf rr!rLr.r.r/__or__ z View.__or__cCrN)a"Vertically concatenate two Views. A convenience method for `hg.vconcat`. Parameters ---------- other : View | Viewconf The other view or viewconf to combine with. Returns ------- viewconf : A combined Viewconf rrQr.r.r/ __truediv__rSzView.__truediv__cCs t|S)zClone the the View instance. Inserts a unique uuid so the view can be uniquely identified in the front end. Returns ------- view : A copy of the original view with a new uuid )r&r')r!r.r.r/clones z View.clonecKsttdd|gi|S)zConsumes the current View into a top-level view config. Returns ------- viewconf : A top-level HiGlass Viewconf. viewsNr.)rrr!kwargsr.r.r/viewconf!sz View.viewconfcCs|||S)z. "Displays the view in an IPython environment.)rZ_repr_mimebundle_)r!includeexcluder.r.r/r[+szView._repr_mimebundle_cKs|jdi|S)zCreate a Jupyter Widget display for this view. Casts the view into a top-level view config which can be displayed. Returns ------- widget : A HiGlassWidget instance Nr.)rZwidgetrXr.r.r/r^/s z View.widgetcenterron3Literal['center', 'top', 'bottom', 'left', 'right']cKs|r|nt|}|dkrd}n|dks|dkrd}n|dks#|dkr&d}ntd t|j|d ur9t|j|gtd ||jd |}t|j|||S) aProject a different viewport onto this view. Creates a viewport-projection track whos bounds are synchonrized with an existing, separate view. Parameters ---------- view : View The view with the desired viewport to project. on : Literal["center", "top", "bottom", "left", "right"], optional Where to project the viewport on this view (default: "center"). inplace : bool Whether to modify this view inplace, or alternatively return a new view (default: `False`). **kwargs : dict Additional parameters for the created viewport-projection track. Returns ------- view : A view with a new viewport-projection track. r_zviewport-projection-centertopbottomzviewport-projection-horizontalleftrightzviewport-projection-verticalz Not possibleN)type_Z fromViewUidr.) r&r' ValueErrorgetattrtracksr)rr=append)r!rr`r#rYZnew_viewZ track_typeZtrkr.r.r/project:sz View.project)NNF)rHrIrJrIr#r$)rLrMNN)r_F)rrr`rar#r$) r2r3r4rArKrRrUrVrZr[r^rkr.r.r.r/rs !  rrrW list[ViewT]r% list[str]cCs@i}|D]}|jD]\}}t|tr|j||j<q qt|S)z;Inspect tracks and extract plugin urls to inject into HTML.)ri isinstancerrDtypelistvalues)rW plugin_urlsr_rr.r.r/gather_plugin_urlsqs   ruc@sZeZdZdZd ddZddZed!d d Zdddd d d"ddZd#ddZ d#ddZ dS)$rzGRepresents a top-level viewconfig, or a complete HiGlass visualization.NcCs2tj}|jdur gnt|j}|||dS)z5 "Displays the view config in an IPython environment.N)rs)displayZ renderersgetrWrudict)r!r\r]rendererrsr.r.r/r[~s zViewconf._repr_mimebundle_cKs ddlm}||fi|S)z5Create a Jupyter Widget display for this view config.r) HiGlassWidget)Zhiglass._widgetrzrx)r!rYrzr.r.r/r^s zViewconf.widgeturlstrcKsZddlm}|j|fi|}|| }|}Wdn1s#wY||S)acLoad a viewconf via URL and construct a Viewconf. Makes an HTTP request. Parameters ---------- url : str The URL for a JSON HiGlass view config. **kwargs : dict Options for the `urllib.Request` instance. Returns ------- viewconf : The parsed view config rN)urllib.requestrequestRequesturlopenreadZ parse_raw)clsr{rYurllibr~responserawr.r.r/from_urls    zViewconf.from_urlF)zoomlocation value_scaler#lockshgs.Lock | hgs.ValueScaleLockr list[hgs.Lock] | hgs.Lock | Nonerr4list[hgs.ValueScaleLock] | hgs.ValueScaleLock | Noner#r$c Gs|r|nt|}t|}t|}t|}g}|D]}t|tjr*||q||q|||||jdurDt |_|D]}t|j t sPJ||jj |j <|D] \} } |j |jj | <qYqF|jdurpt|_|D]}t|j t s|J||jj |j <|D] \} } |j |jj | <qqr|jdurt|_|D]}t|j t sJ||jj |j <|D] \} } |j |jj | <qq|S)aSpecify view locks. Parameters ---------- *locks : hgs.Lock | hgs.ValueScaleLock, optional A set of either zoom/location or value scale locks to add. Instances of `hgs.Lock` will synchronize _both_ zoom and location for the views included in the lock. If you'd like to synchronize just zoom or location, use the explicit keyword arguments below. zoom : hgs.Lock | list[hgs.Lock], optional A lock or set of locks to synchronize only the zoom. location : hgs.Lock | list[hgs.Lock], optional A lock or set of locks to synchronize only the location. value_scale : hgs.ValueScaleLock | list[hgs.ValueScaleLock], optional A single or set of value-scale locks. inplace : bool, optional Whether to modify the viewconf in place or return a new viewconf with the locks applied (default: `False`) Returns ------- viewconf : A Viewconf with the synchronized views. Examples -------- >>> view1 = hg.view(hg.track("heatmap")) >>> view2 = hg.view(hg.track("heatmap")) >>> # create an abstract lock for two views >>> view_lock = hg.lock(view1, view2) >>> # lock location & zoom >>> (view1 | view2).lock(view_lock) >>> # lock only zoom >>> (view1 | view2).lock(zoom=view_lock) >>> # lock only location >>> (view1 | view2).lock(location=view_lock) N)r&r'Z ensure_listrohgsLockrjextend zoomLocksZ ZoomLocksr=r| locksDictlocksByViewUid locationLocksZ LocationLocksvalueScaleLocksZValueScaleLocks) r!rrrr#rconfZ shared_locksrZvuidrtr.r.r/rsH5                 zViewconf.locksrLrMr%Viewconf[TrackT]cCrNrOrPrQr.r.r/rRrSzViewconf.__or__cCrN)a6Vertically concatenate with another view or viewconf. A convenience method for `hg.hconcat`. Parameters ---------- other : View | Viewconf The other view or viewconf to combine with. Returns ------- viewconf : A combined Viewconf rTrQr.r.r/rUrSzViewconf.__truediv__rl)r{r|) rrrrrrrrr#r$)rLrMr%r) r2r3r4rAr[r^ classmethodrrrRrUr.r.r.r/r{s   brTrackTATrackTBmethod!Literal['horizontal', 'vertical']a!View[TrackTA] | Viewconf[TrackTA]b!View[TrackTB] | Viewconf[TrackTB]Viewconf[TrackTA | TrackTB]c CsXt|tr |n|}|jdusJt|tr|n|}|jdus$J|dkr/dd}d}n|dkr:dd}d}ntd d d |jD}|jdurMd ntt||j}|D]}t|j|}t |j|||qW|j |d D]/} t|| } | rt|| durt || | j ddqqt|| j | j t|| j | jqqtttttf|S)a Concatenate two views or separate viewconfs together. Uses the layout of one view or the total bounds of all views in one viewconf to offset the other view/viewconf. Parameters ---------- method : Literal["horizontal", "vertical"] How to concatenate views/viewconfs. a : View | Viewconf A view or viewconf to combine. b : View | Viewconf The other view or viewconf to combine. Returns ------- viewconf : A combined viewconf containing multiple views. NverticalcS|jj|jjSN)layoutrJhrr.r.r/mapperPzconcat..mapperrJ horizontalcSrr)rrHwrr.r.r/rVrrHz1concat method must be 'vertical' or 'horizontal'.cSsg|]}|jddqS)Tdeep)copy).0r-r.r.r/ ^szconcat..r)rrrTr)rorrZrWrgmaxmaprhrr)rrrr8rr rr rr) rrrrfieldrWoffsetrcurrZlockattrrr.r.r/r.s4   rrrc@rC) _TrackCreatoraCreate track instances from their track type. Used internally by `hg.track` to leverage pydantic's ability to get the appropriate base model by the track type. Example: ------- >>> assert isinstance(_TrackCreator(type="heatmap").__root__, HeatmapTrack) Track__root__NrEr.r.r.r/r}s  rrfhgs.EnumTrackTyper= str | NonecKdSrr.rfr=rYr.r.r/rrLiteral['heatmap']cKrrr.rr.r.r/rrr|cKrrr.rr.r.r/rr,hgs.EnumTrackType | Literal['heatmap'] | strrcKs0|durt}td||d|}t|jS)aiCreate a HiGlass track. Parameters ---------- type_ : str The track type to create uid: str, optional A unique id for the track. If unspecified (default), a unique id is given internally. **kwargs: dict Other top-level track properties. Returns ------- track : An instance of an hg.Track N)rpr=r.)r&r=rxrZ parse_objr)rfr=rYdatar.r.r/rs  )rHrJwidthheightrirr=_tracks@TrackT | tuple[TrackT, utils.TrackPosition] | hgs.Tracks[TrackT]rHintrJrrrihgs.Tracks[TrackT] | Nonerhgs.Layout | None View[TrackT]c Os |durtj||||d}n tjdi|}|dur!tt} ntt|} |D]=} t| tjrG| } | D] \} } | | | q:q*t| t rQ| \} } nt j | j } | dur`td| | | q*|durpt }ttd|tjtdi| |d|S)aHCreate a HiGlass view from multiple tracks. Parameters ---------- *_tracks : Track | tuple[Track, str] | hgs.Tracks[Track] The tracks to include in the view. Can be 1) separate track objects (position inferred), 2) explicit (Track, position) tuples, or 3.) a `higlass_schema.Tracks` Object specifying all tracks and positions. x : int, optional The x position of the view in the HiGlass grid (default: 0). Used to created a Layout if none is provided. y : int, optional The y position of the view in the HiGlass grid (default: 0). Used to created a Layout if none is provided. width : int, optional The width of the view (default: 12). Used to created a Layout if none is provided. height : int, optional The height of the view (default: 6). Used to created a Layout if none is provided. layout : hgs.Layout, optional An explicit layout for the view (default: None). If provided the `x`, `y`, `width`, and `height` parameters are ignored. uid: str, optional A unique id for the view. If unspecified (default), a unique id is automatically generated internally. **kwargs: dict Other top-level view properties. Returns ------- view : An instance of an hg.View N)rHrJrrzNo default track type)rrir=r.)rZLayoutrxrrqroZTracksr(rtupler&Ztrack_default_positionrwrprgrjr=rr) rHrJrrrirr=rrYrrpositionZ track_listr.r.r/rs63    rt1t2cKs|durt}t|tr5tdi|}|j|jdi||D] \}}t |||q(|Stdd|dd||fDd|S)aA utility to create a CombinedTrack from two other tracks. Parameters ---------- t1 : hg.Track The first of two tracks to combine. t2 : hg.Track The second of two tracks to combine. uid : str, optional A unique id for the newly created CombinedTrack. **kwargs : dict Additional properties to pass to the CombinedTrack constructor. Returns ------- combined_track : An CombinedTrack with two children tracks. NcombinedcSs g|] }|jdi|qS)r.) __class__rxrrr.r.r/r3s zcombine..)rpr=contentsr.) r&r=rorrxrrjrr(r))rrr=rYrkeyvalr.r.r/rs rTcKs|j|jks Jdt|jtsJt|jtsJt|jts"Jt|jts*Jt|}d|_d|_tjddd||fDd|_ | D] \}}t |||qH|S)aA utility to create a divided track. Tracks must the the same type. Parameters ---------- t1 : hg.EnumTrack | hg.HeatmapTrack The first track to divide by the other. t2 : hg.EnumTrack | hg.HeatmapTrack The other track. **kwargs : dict Additional properties to pass to the newly created track. Returns ------- divided_track : An identical track with "divided" data sources. z divided tracks must be same typeNZdividedcSsg|] }|j|jdqS)r>r<rrr.r.r/r^s zdivide..)rpchildren) rpror>r|r<r&r'rZDatarr(r))rrrYrrrr.r.r/r;s"  rr=hgs.LockcGdS)a<Create an abstract lock linking two or more views. Parameters ---------- *views : View Two or more views to synchronize with this lock. uid : str, optional An (optional) unique id for this lock. Returns ------- lock : An abstract lock linking two or more views. Nr.)r=rWr.r.r/rlsr)r=ignoreOffScreenValuespairstuple[View, Track]r bool | Nonehgs.ValueScaleLockcGr)aYCreate an abstract value-scale lock linking one or more (View, Track) pairs. Parameters ---------- *pairs : tuple[View, Track] One or more (View, Track) pairs specifying a Track whos values should be used to scale all tracks within the View. uid : str, optional An (optional) unique id for this lock. ignoreOffScreenValues : bool, optional Whether to ignore off screen values when scaling the view. Returns ------- lock : A value-scale lock defining how the tracks in a View should be scaled by by a particular track. Nr.)r=rrr.r.r/rscOst|dksJ|durt}t|dtr3tj|d}|D]}t|jts)Jt||jdq|Stj d d|i|}|D](\}}t|jtsKJt|jtsSJ|jd|j}t|||j|jdq?|S) aCreate an abstract lock or value-scale lock. Overloaded to either return a `hgs.Lock` or `hgs.ValueScaleLock` depending on the arguments. See `Viewconf.locks` for how to use the created locks in your top-level view config. Examples -------- Create an abstract lock linking two or more separate views. >>> view1 = hg.view(hg.track("heatmap")) >>> view2 = hg.view(hg.track("heatmap")) >>> lock = hg.lock(view1, view2) Create a value-scale lock, scaling two views by the values of a particular track. >>> t1 = hg.track("heatmap") >>> t2 = hg.track("heatmap") >>> view1 = hg.view(t1, t2) >>> view2 = hg.view(t1, t2) >>> value_scale_lock = hg.lock((view1, t2), (view2, t2)) Nrr)rrrr=.)rrr.) lenr&r=rorrrr|r)ZValueScaleLock)r=rrYZlckrrZvtuidr.r.r/rs   )rWrmr%rn)rrrrrrr%rr)rfrr=rr%r )rfrr=rr%r)rfr|r=rr%r)rfrr=rr%r)rrrHrrJrrrrrrirrrr=rr%r)rrrrr=rr%r)rrrrr%r)rWrr=rr%r)rrr=rrrr%r)r=r)7 __future__r functools collectionsrtypingrrrrrr r r Zhiglass_schemarZpydanticr Zhiglass._display_displayrvZhiglass._utils_utilsr&__all__Zhiglass.serverrr r6r:r rrrZ BaseTrackrrrrrrurrrrpartialrrrrrrrrrr.r.r.r/s  (     %&!    "  0   E   W* 1