Source code for napari.components.dims

from numbers import Integral
from typing import Sequence, Tuple, Union

import numpy as np
from pydantic import root_validator, validator
from typing_extensions import Literal  # Added to typing in 3.8

from ..utils.events import EventedModel
from ..utils.translations import trans


[docs]class Dims(EventedModel): """Dimensions object modeling slicing and displaying. Parameters ---------- ndim : int Number of dimensions. ndisplay : int Number of displayed dimensions. last_used : int Dimension which was last used. range : tuple of 3-tuple of float List of tuples (min, max, step), one for each dimension. In a world coordinates space. As with Python's `range` and `slice`, max is not included. current_step : tuple of int Tuple of the slider position for each dims slider, in slider coordinates. order : tuple of int Tuple of ordering the dimensions, where the last dimensions are rendered. axis_labels : tuple of str Tuple of labels for each dimension. Attributes ---------- ndim : int Number of dimensions. ndisplay : int Number of displayed dimensions. last_used : int Dimension which was last used. range : tuple of 3-tuple of float List of tuples (min, max, step), one for each dimension. In a world coordinates space. As with Python's `range` and `slice`, max is not included. current_step : tuple of int Tuple the slider position for each dims slider, in slider coordinates. order : tuple of int Tuple of ordering the dimensions, where the last dimensions are rendered. axis_labels : tuple of str Tuple of labels for each dimension. nsteps : tuple of int Number of steps available to each slider. These are calculated from the ``range``. point : tuple of float List of floats setting the current value of the range slider when in POINT mode, one for each dimension. In a world coordinates space. These are calculated from the ``current_step`` and ``range``. displayed : tuple of int List of dimensions that are displayed. These are calculated from the ``order`` and ``ndisplay``. not_displayed : tuple of int List of dimensions that are not displayed. These are calculated from the ``order`` and ``ndisplay``. displayed_order : tuple of int Order of only displayed dimensions. These are calculated from the ``displayed`` dimensions. """ # fields ndim: int = 2 ndisplay: Literal[2, 3] = 2 last_used: int = 0 range: Tuple[Tuple[float, float, float], ...] = () current_step: Tuple[int, ...] = () order: Tuple[int, ...] = () axis_labels: Tuple[str, ...] = () # private vars _scroll_progress: int = 0 # validators @validator('axis_labels', pre=True) def _string_to_list(v): if isinstance(v, str): return list(v) return v @root_validator def _check_dims(cls, values): """Check the consitency of dimensionaity for all attributes Parameters ---------- values : dict Values dictionary to update dims model with. """ ndim = values['ndim'] # Check the range tuple has same number of elements as ndim if len(values['range']) < ndim: values['range'] = ((0, 2, 1),) * ( ndim - len(values['range']) ) + values['range'] elif len(values['range']) > ndim: values['range'] = values['range'][-ndim:] # Check the current step tuple has same number of elements as ndim if len(values['current_step']) < ndim: values['current_step'] = (0,) * ( ndim - len(values['current_step']) ) + values['current_step'] elif len(values['current_step']) > ndim: values['current_step'] = values['current_step'][-ndim:] # Check the order tuple has same number of elements as ndim if len(values['order']) < ndim: values['order'] = tuple( range(ndim - len(values['order'])) ) + tuple(o + ndim - len(values['order']) for o in values['order']) elif len(values['order']) > ndim: values['order'] = reorder_after_dim_reduction( values['order'][-ndim:] ) # Check the order is a permutation of 0, ..., ndim - 1 if not set(values['order']) == set(range(ndim)): raise ValueError( trans._( "Invalid ordering {order} for {ndim} dimensions", deferred=True, order=values['order'], ndim=ndim, ) ) # Check the axis labels tuple has same number of elements as ndim if len(values['axis_labels']) < ndim: # Append new "default" labels to existing ones if values['axis_labels'] == tuple( map(str, range(len(values['axis_labels']))) ): values['axis_labels'] = tuple(map(str, range(ndim))) else: values['axis_labels'] = ( tuple(map(str, range(ndim - len(values['axis_labels'])))) + values['axis_labels'] ) elif len(values['axis_labels']) > ndim: values['axis_labels'] = values['axis_labels'][-ndim:] return values @property def nsteps(self) -> Tuple[int, ...]: """Tuple of int: Number of slider steps for each dimension.""" return tuple( int((max_val - min_val) // step_size) for min_val, max_val, step_size in self.range ) @property def point(self) -> Tuple[int, ...]: """Tuple of float: Value of each dimension.""" # The point value is computed from the range and current_step point = tuple( min_val + step_size * value for (min_val, max_val, step_size), value in zip( self.range, self.current_step ) ) return point @property def displayed(self) -> Tuple[int, ...]: """Tuple: Dimensions that are displayed.""" return self.order[-self.ndisplay :] @property def not_displayed(self) -> Tuple[int, ...]: """Tuple: Dimensions that are not displayed.""" return self.order[: -self.ndisplay] @property def displayed_order(self) -> Tuple[int, ...]: displayed = self.displayed # equivalent to: order = np.argsort(self.displayed) order = sorted(range(len(displayed)), key=lambda x: displayed[x]) return tuple(order)
[docs] def set_range( self, axis: Union[int, Sequence[int]], _range: Union[ Sequence[Union[int, float]], Sequence[Sequence[Union[int, float]]] ], ): """Sets ranges (min, max, step) for the given dimensions. Parameters ---------- axis : int or sequence of int Dimension index or a sequence of axes whos range will be set. _range : tuple or sequence of tuple Range specified as (min, max, step) or a sequence of these range tuples. """ if isinstance(axis, Integral): axis = assert_axis_in_bounds(axis, self.ndim) # type: ignore if self.range[axis] != _range: full_range = list(self.range) full_range[axis] = _range self.range = full_range else: full_range = list(self.range) # cast range to list for list comparison below _range = list(_range) # type: ignore axis = tuple(axis) # type: ignore if len(axis) != len(_range): raise ValueError( trans._("axis and _range sequences must have equal length") ) if _range != full_range: for ax, r in zip(axis, _range): ax = assert_axis_in_bounds(int(ax), self.ndim) full_range[ax] = r self.range = full_range
[docs] def set_point( self, axis: Union[int, Sequence[int]], value: Union[Union[int, float], Sequence[Union[int, float]]], ): """Sets point to slice dimension in world coordinates. The desired point gets transformed into an integer step of the slider and stored in the current_step. Parameters ---------- axis : int or sequence of int Dimension index or a sequence of axes whos point will be set. value : scalar or sequence of scalars Value of the point for each axis. """ if isinstance(axis, Integral): axis = assert_axis_in_bounds(axis, self.ndim) # type: ignore (min_val, max_val, step_size) = self.range[axis] raw_step = (value - min_val) / step_size self.set_current_step(axis, raw_step) else: value = tuple(value) # type: ignore axis = tuple(axis) # type: ignore if len(axis) != len(value): raise ValueError( trans._("axis and value sequences must have equal length") ) raw_steps = [] for ax, val in zip(axis, value): ax = assert_axis_in_bounds(int(ax), self.ndim) min_val, _, step_size = self.range[ax] raw_steps.append((val - min_val) / step_size) self.set_current_step(axis, raw_steps)
[docs] def set_current_step( self, axis: Union[int, Sequence[int]], value: Union[Union[int, float], Sequence[Union[int, float]]], ): """Set the slider steps at which to slice this dimension. The position of the slider in world coordinates gets calculated from the current_step of the slider. Parameters ---------- axis : int or sequence of int Dimension index or a sequence of axes whos step will be set. value : scalar or sequence of scalars Value of the step for each axis. """ if isinstance(axis, Integral): axis = assert_axis_in_bounds(axis, self.ndim) step = round(min(max(value, 0), self.nsteps[axis] - 1)) if self.current_step[axis] != step: full_current_step = list(self.current_step) full_current_step[axis] = step self.current_step = full_current_step else: full_current_step = list(self.current_step) # cast value to list for list comparison below value = list(value) # type: ignore axis = tuple(axis) # type: ignore if len(axis) != len(value): raise ValueError( trans._("axis and value sequences must have equal length") ) if value != full_current_step: # (computed) nsteps property outside of the loop for efficiency nsteps = self.nsteps for ax, val in zip(axis, value): ax = assert_axis_in_bounds(int(ax), self.ndim) step = round(min(max(val, 0), nsteps[ax] - 1)) full_current_step[ax] = step self.current_step = full_current_step
[docs] def set_axis_label( self, axis: Union[int, Sequence[int]], label: Union[str, Sequence[str]], ): """Sets new axis labels for the given axes. Parameters ---------- axis : int or sequence of int Dimension index or a sequence of axes whos labels will be set. label : str or sequence of str Given labels for the specified axes. """ if isinstance(axis, Integral): axis = assert_axis_in_bounds(axis, self.ndim) if self.axis_labels[axis] != str(label): full_axis_labels = list(self.axis_labels) full_axis_labels[axis] = str(label) self.axis_labels = full_axis_labels self.last_used = axis else: full_axis_labels = list(self.axis_labels) # cast label to list for list comparison below label = list(label) # type: ignore axis = tuple(axis) # type: ignore if len(axis) != len(label): raise ValueError( trans._("axis and label sequences must have equal length") ) if label != full_axis_labels: for ax, val in zip(axis, label): ax = assert_axis_in_bounds(int(ax), self.ndim) full_axis_labels[ax] = val self.axis_labels = full_axis_labels
[docs] def reset(self): """Reset dims values to initial states.""" # Don't reset axis labels self.range = ((0, 2, 1),) * self.ndim self.current_step = (0,) * self.ndim self.order = tuple(range(self.ndim))
def _increment_dims_right(self, axis: int = None): """Increment dimensions to the right along given axis, or last used axis if None Parameters ---------- axis : int, optional Axis along which to increment dims, by default None """ if axis is None: axis = self.last_used self.set_current_step(axis, self.current_step[axis] + 1) def _increment_dims_left(self, axis: int = None): """Increment dimensions to the left along given axis, or last used axis if None Parameters ---------- axis : int, optional Axis along which to increment dims, by default None """ if axis is None: axis = self.last_used self.set_current_step(axis, self.current_step[axis] - 1) def _focus_up(self): """Shift focused dimension slider to be the next slider above.""" sliders = [d for d in self.not_displayed if self.nsteps[d] > 1] if len(sliders) == 0: return index = (sliders.index(self.last_used) + 1) % len(sliders) self.last_used = sliders[index] def _focus_down(self): """Shift focused dimension slider to be the next slider bellow.""" sliders = [d for d in self.not_displayed if self.nsteps[d] > 1] if len(sliders) == 0: return index = (sliders.index(self.last_used) - 1) % len(sliders) self.last_used = sliders[index] def _roll(self): """Roll order of dimensions for display.""" order = np.array(self.order) nsteps = np.array(self.nsteps) order[nsteps > 1] = np.roll(order[nsteps > 1], 1) self.order = order.tolist() def _transpose(self): """Transpose displayed dimensions.""" order = list(self.order) order[-2], order[-1] = order[-1], order[-2] self.order = order
def reorder_after_dim_reduction(order): """Ensure current dimension order is preserved after dims are dropped. Parameters ---------- order : tuple The data to reorder. Returns ------- arr : tuple The original array with the unneeded dimension thrown away. """ arr = sorted(range(len(order)), key=lambda x: order[x]) return tuple(arr) def assert_axis_in_bounds(axis: int, ndim: int) -> int: """Assert a given value is inside the existing axes of the image. Returns ------- axis : int The axis which was checked for validity. ndim : int The dimensionality of the layer. Raises ------ ValueError The given axis index is out of bounds. """ if axis not in range(-ndim, ndim): msg = trans._( 'Axis {axis} not defined for dimensionality {ndim}. Must be in [{ndim_lower}, {ndim}).', deferred=True, axis=axis, ndim=ndim, ndim_lower=-ndim, ) raise ValueError(msg) return axis % ndim