# https://pyrocko.org - GPLv3
#
# The Pyrocko Developers, 21st Century
# ---|P------/S----------~Lg----------
import vtk
from pyrocko.guts import Bool, String, StringChoice, Float
from pyrocko.gui.qt_compat import qw, qc
from pyrocko.gui.talkie import TalkieConnectionOwner, TalkieStringer
from .base import Element, ElementState
from .. import common
guts_prefix = 'sparrow'
[docs]class HudPositionChoice(StringChoice):
choices = ['bottom', 'bottom-left', 'bottom-right',
'top', 'top-left', 'top-right']
[docs]class HudState(ElementState):
visible = Bool.T(default=True)
template = String.T()
position = HudPositionChoice.T(default='bottom')
lightness = Float.T(default=1.0)
fontsize = Float.T(default=0.05)
def create(self):
element = HudElement()
return element
class HudElement(Element):
def __init__(self):
Element.__init__(self)
self._controls = None
self._actor = None
self._connections2 = TalkieConnectionOwner()
def get_name(self):
return 'HUD'
def bind_state(self, state):
Element.bind_state(self, state)
self.talkie_connect(
state,
['visible', 'lightness', 'fontsize', 'template', 'position'],
self.update)
self.talkie_connect(state, 'template', self.update_bindings)
def unbind_state(self):
self._connections2.talkie_disconnect_all()
Element.unbind_state(self)
def set_parent(self, parent):
self._parent = parent
self._parent.add_panel(
self.get_title_label(),
self._get_controls(),
visible=True,
title_controls=[
self.get_title_control_remove(),
self.get_title_control_visible()])
self.talkie_connect(
self._parent.gui_state, 'size', self.update)
self.update_bindings()
self.update()
def unset_parent(self):
self.unbind_state()
if self._parent:
if self._actor is not None:
self._parent.remove_actor_2d(self._actor)
if self._controls is not None:
self._parent.remove_panel(self._controls)
self._controls = None
self._parent.update_view()
self._parent = None
def update_bindings(self, *args):
self._connections2.talkie_disconnect_all()
s = TalkieStringer(self._parent.state)
self._state.template.format_map(s)
for (state, path) in s.get_paths():
self._connections2.talkie_connect(state, path, self.update)
def update(self, *args):
state = self._state
if not self._state:
return
pstate = self._parent.state
if self._actor is None:
self._actor = vtk.vtkTextActor()
actor = self._actor
s = TalkieStringer(pstate)
actor.SetInput(self._state.template.format_map(s))
sx, sy = self._parent.gui_state.size
cx = 0.5 * sx
# cy = 0.5 * sy
off = 0.1 * sy
pos = {
'top': (cx, sy - off, 1, 2),
'top-left': (off, sy - off, 0, 2),
'top-right': (sx - off, sy - off, 2, 2),
'bottom': (cx, off, 1, 0),
'bottom-left': (off, off, 0, 0),
'bottom-right': (sx - off, off, 2, 0)}
x, y, hj, vj = pos[state.position]
actor.SetPosition(x, y)
# actor.SetPosition2(200, 100)
prop = actor.GetTextProperty()
prop.SetFontSize(int(round(state.fontsize*sy)))
lightness = state.lightness
prop.SetColor(lightness*0.8, lightness*0.8, lightness*0.7)
prop.SetJustification(hj)
prop.SetVerticalJustification(vj)
if state.visible:
self._parent.add_actor_2d(actor)
else:
self._parent.remove_actor_2d(actor)
self._parent.update_view()
def _get_controls(self):
if not self._controls:
from ..state import state_bind_lineedit, \
state_bind_combobox, state_bind_slider
frame = qw.QFrame()
layout = qw.QGridLayout()
frame.setLayout(layout)
layout.addWidget(qw.QLabel('Template'), 0, 0)
le = qw.QLineEdit()
layout.addWidget(le, 0, 1)
state_bind_lineedit(self, self._state, 'template', le)
cb = common.string_choices_to_combobox(HudPositionChoice)
layout.addWidget(qw.QLabel('Position'), 1, 0)
layout.addWidget(cb, 1, 1)
state_bind_combobox(self, self._state, 'position', cb)
layout.addWidget(qw.QLabel('Lightness'), 2, 0)
slider = qw.QSlider(qc.Qt.Horizontal)
slider.setSizePolicy(
qw.QSizePolicy(
qw.QSizePolicy.Expanding, qw.QSizePolicy.Fixed))
slider.setMinimum(0)
slider.setMaximum(1000)
layout.addWidget(slider, 2, 1)
state_bind_slider(
self, self._state, 'lightness', slider, factor=0.001)
layout.addWidget(qw.QLabel('Fontsize'), 3, 0)
slider = qw.QSlider(qc.Qt.Horizontal)
slider.setSizePolicy(
qw.QSizePolicy(
qw.QSizePolicy.Expanding, qw.QSizePolicy.Fixed))
slider.setMinimum(0)
slider.setMaximum(1000)
layout.addWidget(slider, 3, 1)
state_bind_slider(
self, self._state, 'fontsize', slider, factor=0.001)
layout.addWidget(qw.QFrame(), 4, 0)
self._controls = frame
return self._controls
__all__ = [
'HudElement',
'HudState']
