map_level_widget.py

Go to the documentation of this file.

#------------------------------------------------------------------------------
# This file is part of the OpenStructure project <www.openstructure.org>
#
# Copyright (C) 2008-2011 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()