src/Tools/geom_plugins.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2010-2014  CEA/DEN, EDF R&D, OPEN CASCADE
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License, or (at your option) any later version.
   8 #
   9 # This library is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20 # Author : Renaud Nédélec (OpenCascade S.A.S)
  21
  22 import salome_pluginsmanager
  23
  24 def t_shape_fluid(context):
  25     import t_shape_builder
  26     import t_shape_dialog
  27     import xalome
  28     from PyQt4.QtGui import QMessageBox
  29     activeStudy = context.study
  30
  31     dialog = t_shape_dialog.TShapeDialog()
  32
  33     # Get the parameter values from a gui dialog box. If the dialog is
  34     # closed using the Ok button, then the data are requested from the
  35     # gui and used to create the shape of the tube.
  36     dialog.exec_()
  37     if dialog.wasOk():
  38       r1, r2, h1, h2, thickness = dialog.getData()
  39       #QMessageBox.about(None, "Building in progress", "building shape, please be patient")
  40       shape = t_shape_builder.build_shape(activeStudy, r1, r2, h1, h2, thickness)
  41       entry = xalome.addToStudy(activeStudy, shape, "T_shape_fluid" )
  42       xalome.displayShape(entry)
  43     #if dialog.wasOk():
  44         #radius, length, width = dialog.getData()
  45         #shape = tubebuilder.createGeometry(activeStudy, radius, length, width)
  46         #entry = xalome.addToStudy(activeStudy, shape, "Tube" )
  47         #xalome.displayShape(entry)
  48
  49
  50 salome_pluginsmanager.AddFunction('T shape fluid',
  51                                   'Creates the fluid part of a pipe T-shape',
  52                                   t_shape_fluid)