#------------------------------------------------------------------------------ # This file is part of the OpenStructure project # # Copyright (C) 2008-2020 by the OpenStructure authors # # This library is free software; you can redistribute it and/or modify it under # the terms of the GNU Lesser General Public License as published by the Free # Software Foundation; either version 3.0 of the License, or (at your option) # any later version. # This library is distributed in the hope that it will be useful, but WITHOUT # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS # FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more # details. # # You should have received a copy of the GNU Lesser General Public License # along with this library; if not, write to the Free Software Foundation, Inc., # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA #------------------------------------------------------------------------------ # -*- coding: utf-8 -*- import math from ost import gui from ost import gfx try: from ost import img _img_present=True except ImportError: _img_present=False pass from PyQt5 import QtCore, QtWidgets from .preset_widget import PresetWidget class AdditionalSettingsWidget(QtWidgets.QStackedWidget): def __init__(self, parent=None): QtWidgets.QStackedWidget.__init__(self, parent) self.map_widget_ = MapLevelWidget(self) self.preset_widget_ = PresetWidget(self) self.addWidget(self.preset_widget_); self.addWidget(self.map_widget_); self.setContentsMargins(0,0,0,0) self.setMinimumSize(self.preset_widget_.minimumSize()) def Update(self): self.setEnabled(True) scene_selection = gui.SceneSelection.Instance() all_img = True all_entity = True for i in range(0,scene_selection.GetActiveNodeCount()): node = scene_selection.GetActiveNode(i) if not (isinstance(node, gfx.Entity) or isinstance(node, gfx.Surface)): all_entity = False if (not _img_present) or (not isinstance(node, gfx.MapIso)): all_img = False if all_img: self.map_widget_.Update() self.setMinimumSize(self.map_widget_.minimumSize()) self.resize(self.map_widget_.minimumSize()) self.setMinimumSize(self.map_widget_.minimumSize()) self.setCurrentWidget(self.map_widget_) elif all_entity: self.preset_widget_.Update() self.setMinimumSize(self.preset_widget_.minimumSize()) self.resize(self.preset_widget_.minimumSize()) self.setMinimumSize(self.preset_widget_.minimumSize()) self.setCurrentWidget(self.preset_widget_) else: self.setEnabled(False) #Map Level Widget class MapLevelWidget(QtWidgets.QWidget): def __init__(self, parent=None): QtWidgets.QWidget.__init__(self, parent) #Create Ui elements self.map_level_label_ = QtWidgets.QLabel("Map Contour Level") self.font = self.map_level_label_.font() self.font.setBold(True) self.level_preview_ = LevelPreview() self.level_spinbox_ = QtWidgets.QDoubleSpinBox() self.level_spinbox_.setDecimals(3) self.level_spinbox_.setSingleStep(0.05) grid = QtWidgets.QGridLayout() grid.setContentsMargins(0,5,0,0) grid.addWidget(self.level_preview_, 0, 0, 1, 4) grid.addWidget(self.map_level_label_, 1, 0, 1, 3) grid.addWidget(self.level_spinbox_,1,3,1,1) grid.setRowStretch(3, 1) self.setLayout(grid) self.level_preview_.levelUpdated.connect(self.UpdateLevel) self.level_preview_.levelModified.connect(self.ModifySpinBox) self.level_spinbox_.valueChanged.connect(self.UpdateLevel) self.setMinimumSize(250,200) def Update(self): scene_selection = gui.SceneSelection.Instance() if(scene_selection.GetActiveNodeCount()==1): node = scene_selection.GetActiveNode(0) if _img_present and isinstance(node, gfx.MapIso): try: self.level_preview_.SetBins(node.GetHistogram()) self.level_preview_.SetMinimum(node.GetMinLevel()) self.level_spinbox_.setMinimum(node.GetMinLevel()) self.level_preview_.SetMaximum(node.GetMaxLevel()) self.level_spinbox_.setMaximum(node.GetMaxLevel()) self.level_preview_.SetLevel(node.GetLevel()) self.level_spinbox_.show() self.font.setBold(True) self.map_level_label_.setText("Map Contour Level") self.setEnabled(True) except UserWarning: self.font.setBold(False) self.map_level_label_.setText("Map uniformly filled with level %s"%node.GetLevel()) self.level_spinbox_.hide() self.setEnabled(False) else: self.setEnabled(False) else: self.setEnabled(False) def UpdateLevel(self, level): scene_selection = gui.SceneSelection.Instance() if(scene_selection.GetActiveNodeCount()==1): node = scene_selection.GetActiveNode(0) node.SetLevel(level) def ModifySpinBox(self, level): self.level_spinbox_.valueChanged.disconnect(self.UpdateLevel) self.level_spinbox_.setValue(level) self.level_spinbox_.valueChanged.connect(self.UpdateLevel) #Level Preview class LevelPreview(QtWidgets.QWidget): levelUpdated = QtCore.pyqtSignal(int, name="levelUpdated") levelModified = QtCore.pyqtSignal(int, name="levelModified") def __init__(self, parent=None): QtWidgets.QWidget.__init__(self, parent) #Defaults self.border_offset_ = 3 self.preview_height_ = 150 QtWidgets.QWidget.__init__(self, parent) #Ui self.setMinimumSize(0, self.preview_height_ + 4) self.bins_ = None self.level_ = 0 self.minimum_ = 0 self.maximum_ = 0 self.paint_mouse_=False def SetBins(self, bins): self.bins_ = bins self.update() def SetMaximum(self, max): self.maximum_ = max def SetMinimum(self, min): self.minimum_ = min def SetLevel(self, level): self.level_ = level def GetLevel(self): return self.level_ def paintEvent(self, event): if self.isEnabled() and self.bins_ is not None: painter = QtWidgets.QPainter() if painter.begin(self): self.PaintBackground(painter) self.PaintBins(painter) self.PaintLevel(painter) if(self.paint_mouse_): self.PaintMouse(painter) painter.end() def PaintBackground(self,painter): size = self.size() painter.setBrush(QtCore.Qt.white) painter.setPen(QtCore.Qt.white) painter.drawRect(self.border_offset_, self.border_offset_, size.width() - 2 * self.border_offset_, self.preview_height_) def PaintBins(self,painter): size = self.size() bin_cnt = len(self.bins_) bin_width = (size.width()-2* self.border_offset_) / float(bin_cnt) max=0 for b in self.bins_: if(b>max): max = b max = math.log(max) if(max > 0): painter.setBrush(QtCore.Qt.black) painter.setPen(QtCore.Qt.black) for i in range(0,bin_cnt): bin_height = self.bins_[i] if(bin_height>0): bin_height = math.floor((math.log(bin_height)/max)*(self.preview_height_-2*(self.border_offset_))) painter.drawRect(self.border_offset_ + (i*bin_width), self.preview_height_ - bin_height, bin_width, bin_height) def PaintLevel(self,painter): size = self.size() width = size.width()-(2* self.border_offset_) tot_len = self.maximum_-self.minimum_ if(tot_len>0): cur_len = self.level_-self.minimum_ painter.setBrush(QtCore.Qt.red) painter.setPen(QtCore.Qt.red) painter.drawRect((width / tot_len) * cur_len, self.border_offset_, 1, self.preview_height_) def PaintMouse(self,painter): size = self.size() width = size.width()-(2* self.border_offset_) painter.setBrush(QtCore.Qt.gray) painter.setPen(QtCore.Qt.gray) pos=self.mapFromGlobal(QtWidgets.QCursor.pos()) painter.drawRect(pos.x(), self.border_offset_, 1, self.preview_height_) def mouseReleaseEvent(self, event): self.paint_mouse_=False size = self.size() width = size.width()-(2* self.border_offset_) tot_len = self.maximum_-self.minimum_ self.level_ = self.minimum_ + float(event.x())/width * tot_len self.update() self.levelUpdated.emit(self.level_) def mousePressEvent(self,event): self.paint_mouse_=True def mouseMoveEvent(self, event): size = self.size() width = size.width()-(2* self.border_offset_) tot_len = self.maximum_-self.minimum_ level = self.minimum_ + float(event.x())/width * tot_len self.levelModified.emit(level) self.update()