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))
[docs] def transpose(self):
"""Transpose displayed dimensions.
This swaps the order of the last two displayed dimensions.
The order of the displayed is taken from Dims.order.
"""
order = list(self.order)
order[-2], order[-1] = order[-1], order[-2]
self.order = order
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 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