class SheetArea(QAbstractScrollArea):
def __init__(self, config={}):
QAbstractScrollArea.__init__(self)
self.setFocusPolicy(Qt.NoFocus)
self._ui_model = None
self._updater = None
# Widgets
self.setViewport(View())
self._corner = QWidget()
self._corner.setStyleSheet('QWidget { background-color: #000 }')
self._ruler = Ruler()
self._header = Header()
# Config
self._config = None
self._set_config(config)
# Layout
g = QGridLayout()
g.setSpacing(0)
g.setMargin(0)
g.addWidget(self._corner, 0, 0)
g.addWidget(self._ruler, 1, 0)
g.addWidget(self._header, 0, 1)
g.addWidget(self.viewport(), 1, 1)
self.setLayout(g)
self.viewport().setFocusProxy(None)
self.setVerticalScrollBar(LongScrollBar())
self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
self.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
QObject.connect(
self.viewport(),
SIGNAL('heightChanged()'),
self._update_scrollbars)
QObject.connect(
self.viewport(),
SIGNAL('followCursor(QString, int)'),
self._follow_cursor)
def set_ui_model(self, ui_model):
self._ui_model = ui_model
self._updater = ui_model.get_updater()
self._updater.register_updater(self._perform_updates)
self._sheet_manager = ui_model.get_sheet_manager()
# Default zoom level
px_per_beat = self._config['trs_per_beat'] * self._config['tr_height']
self._zoom_levels = self._get_zoom_levels(1, px_per_beat, tstamp.BEAT)
self._default_zoom_index = self._zoom_levels.index(px_per_beat)
self._sheet_manager.set_zoom_range(
-self._default_zoom_index,
len(self._zoom_levels) - self._default_zoom_index - 1)
# Default column width
fm = self._config['font_metrics']
em_px = int(math.ceil(fm.tightBoundingRect('m').width()))
em_range = list(range(3, 41))
self._col_width_levels = [em_px * width for width in em_range]
self._default_col_width_index = em_range.index(self._config['col_width'])
self._sheet_manager.set_column_width_range(
-self._default_col_width_index,
len(self._col_width_levels) - self._default_col_width_index - 1)
self._set_px_per_beat(self._zoom_levels[self._default_zoom_index])
self._set_column_width(self._col_width_levels[self._default_col_width_index])
# Child widgets
self._ruler.set_ui_model(ui_model)
self.viewport().set_ui_model(ui_model)
def unregister_updaters(self):
self._updater.unregister_updater(self._perform_updates)
self._ruler.unregister_updaters()
self.viewport().unregister_updaters()
def _perform_updates(self, signals):
if 'signal_sheet_zoom' in signals:
self._update_zoom()
if 'signal_sheet_column_width' in signals:
self._update_column_width()
def _set_config(self, config):
self._config = DEFAULT_CONFIG.copy()
self._config.update(config)
for subcfg in ('ruler', 'header', 'trigger', 'edit_cursor'):
self._config[subcfg] = DEFAULT_CONFIG[subcfg].copy()
if subcfg in config:
self._config[subcfg].update(config[subcfg])
self._header.set_config(self._config)
self._ruler.set_config(self._config['ruler'])
header_height = self._header.minimumSizeHint().height()
ruler_width = self._ruler.sizeHint().width()
self.setViewportMargins(
ruler_width,
header_height,
0, 0)
self._corner.setFixedSize(
ruler_width,
header_height)
self._header.setFixedHeight(header_height)
self._ruler.setFixedWidth(ruler_width)
fm = QFontMetrics(self._config['font'], self)
self._config['font_metrics'] = fm
self._config['tr_height'] = fm.tightBoundingRect('Ag').height() + 1
self.viewport().set_config(self._config)
def _get_zoom_levels(self, min_val, default_val, max_val):
zoom_levels = [default_val]
# Fill zoom out levels until minimum
prev_val = zoom_levels[-1]
next_val = prev_val / self._config['zoom_factor']
while int(next_val) > min_val:
actual_val = int(next_val)
assert actual_val < prev_val
zoom_levels.append(actual_val)
prev_val = actual_val
next_val = prev_val / self._config['zoom_factor']
zoom_levels.append(min_val)
zoom_levels = list(reversed(zoom_levels))
# Fill zoom in levels until maximum
prev_val = zoom_levels[-1]
next_val = prev_val * self._config['zoom_factor']
while math.ceil(next_val) < tstamp.BEAT:
actual_val = int(math.ceil(next_val))
assert actual_val > prev_val
zoom_levels.append(actual_val)
prev_val = actual_val
next_val = prev_val * self._config['zoom_factor']
zoom_levels.append(tstamp.BEAT)
return zoom_levels
def _set_px_per_beat(self, px_per_beat):
self._ruler.set_px_per_beat(px_per_beat)
self.viewport().set_px_per_beat(px_per_beat)
def _set_column_width(self, width):
self._header.set_column_width(width)
self.viewport().set_column_width(width)
def _update_scrollbars(self):
if not self._ui_model:
return
self._total_height_px = (
self.viewport().get_total_height() + self._config['tr_height'])
vp_height = self.viewport().height()
vscrollbar = self.verticalScrollBar()
vscrollbar.setPageStep(vp_height)
vscrollbar.set_actual_range(0, self._total_height_px - vp_height)
vp_width = self.viewport().width()
cur_col_width_index = self._sheet_manager.get_column_width() + self._default_col_width_index
max_visible_cols = vp_width // self._col_width_levels[cur_col_width_index]
hscrollbar = self.horizontalScrollBar()
hscrollbar.setPageStep(max_visible_cols)
hscrollbar.setRange(0, COLUMN_COUNT - max_visible_cols)
def _follow_cursor(self, new_y_offset_str, new_first_col):
new_y_offset = long(new_y_offset_str)
vscrollbar = self.verticalScrollBar()
hscrollbar = self.horizontalScrollBar()
old_y_offset = vscrollbar.get_actual_value()
old_scaled_y_offset = vscrollbar.value()
old_first_col = hscrollbar.value()
self._update_scrollbars()
vscrollbar.set_actual_value(new_y_offset)
hscrollbar.setValue(new_first_col)
if (old_scaled_y_offset == vscrollbar.value() and
old_first_col == hscrollbar.value()):
if old_y_offset != vscrollbar.get_actual_value():
# Position changed slightly, but the QScrollBar doesn't know this
self.scrollContentsBy(0, 0)
else:
# Position not changed, so just update our viewport
self.viewport().update()
def _update_zoom(self):
zoom_level = self._sheet_manager.get_zoom()
cur_zoom_index = zoom_level + self._default_zoom_index
self._set_px_per_beat(self._zoom_levels[cur_zoom_index])
def _update_column_width(self):
column_width_level = self._sheet_manager.get_column_width()
cur_col_width_index = column_width_level + self._default_col_width_index
self._set_column_width(self._col_width_levels[cur_col_width_index])
def paintEvent(self, ev):
self.viewport().paintEvent(ev)
def resizeEvent(self, ev):
self._update_scrollbars()
def scrollContentsBy(self, dx, dy):
hvalue = self.horizontalScrollBar().value()
vvalue = self.verticalScrollBar().get_actual_value()
self._header.set_first_column(hvalue)
self._ruler.set_px_offset(vvalue)
vp = self.viewport()
vp.set_first_column(hvalue)
vp.set_px_offset(vvalue)
def mousePressEvent(self, event):
self.viewport().mousePressEvent(event)
