#------------------------------------------------------------------------------ # 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 -*- from ost import gui from ost import gfx from PyQt5 import QtCore, QtWidgets from .render_mode_widget import RenderModeWidget #Tube Render Options class TubeWidget(RenderModeWidget): def __init__(self, parent=None): RenderModeWidget.__init__(self, parent) #Title self.text_ = "Smooth Tube" #Set Render Mode self.mode_ = gfx.RenderMode.TUBE #Defaults min_spline_detail = 1 max_spline_detail = 20 min_arc_detail = 1 max_arc_detail = 20 min_radius = 0.1 max_radius = 2 min_ratio = 0.2 max_ratio = 4 #UI tube_label = QtWidgets.QLabel("Tube Settings") font = tube_label.font() font.setBold(True) #Poly Mode poly_mode_label = QtWidgets.QLabel("Poly Mode") self.poly_mode_cb_ = QtWidgets.QComboBox() self.poly_mode_cb_.addItem("Points") self.poly_mode_cb_.addItem("Wireframe") self.poly_mode_cb_.addItem("Surface") #Sphere Label spline_label = QtWidgets.QLabel("Spline Detail") self.spline_spinbox_ = QtWidgets.QSpinBox() self.spline_spinbox_.setRange(min_spline_detail, max_spline_detail) #Arc Label arc_label = QtWidgets.QLabel("Arc Detail") self.arc_spinbox_ = QtWidgets.QSpinBox() self.arc_spinbox_.setRange(min_arc_detail, max_arc_detail) #Radius radius_label = QtWidgets.QLabel("Tube radius") self.radius_spinbox_ = QtWidgets.QDoubleSpinBox() self.radius_spinbox_.setRange(min_radius, max_radius) self.radius_spinbox_.setDecimals(1) self.radius_spinbox_.setSingleStep(0.1) self.radius_slider_ = QtWidgets.QSlider(QtCore.Qt.Horizontal, self) self.radius_slider_.setRange(int(min_radius*10.0), int(max_radius*10.0)) self.radius_slider_.setTickPosition(QtWidgets.QSlider.NoTicks) self.radius_slider_.setTickInterval(1) #Ratio ratio_label = QtWidgets.QLabel("Tube ratio") self.ratio_spinbox_ = QtWidgets.QDoubleSpinBox() self.ratio_spinbox_.setRange(min_ratio, max_ratio) self.ratio_spinbox_.setDecimals(1) self.ratio_spinbox_.setSingleStep(0.1) self.ratio_slider_ = QtWidgets.QSlider(QtCore.Qt.Horizontal, self) self.ratio_slider_.setRange(int(min_ratio*10.0), int(max_ratio*10.0)) self.ratio_slider_.setTickPosition(QtWidgets.QSlider.NoTicks) self.ratio_slider_.setTickInterval(2) grid = QtWidgets.QGridLayout() grid.addWidget(tube_label,0,0,1,3) grid.addWidget(poly_mode_label,1,0,1,3) grid.addWidget(self.poly_mode_cb_,1,3,1,2) grid.addWidget(spline_label, 2, 0, 1, 3) grid.addWidget(self.spline_spinbox_, 2, 4, 1, 1) grid.addWidget(arc_label,3,0,1,3) grid.addWidget(self.arc_spinbox_,3,4,1,1) grid.addWidget(radius_label,4,0,1,1) grid.addWidget(self.radius_slider_,4,1,1,3) grid.addWidget(self.radius_spinbox_,4,4,1,1) grid.addWidget(ratio_label,5,0,1,1) grid.addWidget(self.ratio_slider_,5,1,1,3) grid.addWidget(self.ratio_spinbox_,5,4,1,1) grid.setRowStretch(6,1) self.setLayout(grid) self.spline_spinbox_.valueChanged.connect(self.UpdateSplineDetail) self.arc_spinbox_.valueChanged.connect(self.UpdateArcDetail) self.poly_mode_cb_.currentIndexChanged.connect(self.UpdatePolyMode) self.radius_spinbox_.valueChanged.connect(self.UpdateRadius) self.radius_slider_.valueChanged.connect(self.UpdateSliderRadius) self.ratio_spinbox_.valueChanged.connect(self.UpdateRatio) self.ratio_slider_.valueChanged.connect(self.UpdateSliderRatio) self.setMinimumSize(250,200) def UpdateGui(self,options): self.poly_mode_cb_.setCurrentIndex(options.GetPolyMode()) self.spline_spinbox_.setValue(options.GetSplineDetail()) self.arc_spinbox_.setValue(options.GetArcDetail()) self.UpdateRadiusGui(options.GetTubeRadius()) self.UpdateRatioGui(options.GetTubeRatio()) def UpdatePolyMode(self, value): self.GetOptions().SetPolyMode(value) self.ApplyOptions() def UpdateSplineDetail(self, value): self.GetOptions().SetSplineDetail(value) self.ApplyOptions() def UpdateArcDetail(self, value): self.GetOptions().SetArcDetail(value) self.ApplyOptions() def UpdateRadius(self, value): self.GetOptions().SetTubeRadius(value) self.ApplyOptions() def UpdateSliderRadius(self, value): self.GetOptions().SetTubeRadius(value/10.0) def UpdateRatio(self, value): self.GetOptions().SetTubeRatio(value) def UpdateSliderRatio(self, value): self.GetOptions().SetTubeRatio(value/10.0) def UpdateRadiusGui(self,value): value = round(value, 2) if(abs(value*10.0 - self.radius_slider_.value())>=self.radius_slider_.singleStep()): self.radius_slider_.setValue(int(value*10.0)) if(abs(value - self.radius_spinbox_.value())>=self.radius_spinbox_.singleStep()): self.radius_spinbox_.setValue(value) def UpdateRatioGui(self,value): value = round(value, 2) if(abs(value*10.0 - self.ratio_slider_.value())>=self.ratio_slider_.singleStep()): self.ratio_slider_.setValue(int(value*10.0)) if(abs(value - self.ratio_spinbox_.value())>=self.ratio_spinbox_.singleStep()): self.ratio_spinbox_.setValue(value) def GetText(self): return self.text_ def GetRenderMode(self): return self.mode_