src/Tools/geom_plugins.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2014-2016  EDF R&D
   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     #from salome.geom.t_shape import t_shape_builder
  26     from salome.geom.t_shape import t_shape_dialog
  27     from salome.geom.t_shape import t_shape_progress
  28     import xalome
  29     from PyQt4.QtGui import QMessageBox
  30     activeStudy = context.study
  31
  32     dialog = t_shape_dialog.TShapeDialog()
  33
  34     # Get the parameter values from a gui dialog box. If the dialog is
  35     # closed using the Ok button, then the data are requested from the
  36     # gui and used to create the shape of the tube.
  37     dialog.exec_()
  38     if dialog.wasOk():
  39       r1, r2, h1, h2, thickness = dialog.getData()
  40       #QMessageBox.about(None, "Building in progress", "building shape, please be patient")
  41       #shape = t_shape_builder.build_shape(activeStudy, r1, r2, h1, h2, thickness)
  42       shapeBuilder = t_shape_progress.t_shape_progress()
  43       shape = shapeBuilder.run(activeStudy, r1, r2, h1, h2, thickness)
  44       entry = xalome.addToStudy(activeStudy, shape, "T_shape_fluid" )
  45       xalome.displayShape(entry)
  46     #if dialog.wasOk():
  47         #radius, length, width = dialog.getData()
  48         #shape = tubebuilder.createGeometry(activeStudy, radius, length, width)
  49         #entry = xalome.addToStudy(activeStudy, shape, "Tube" )
  50         #xalome.displayShape(entry)
  51
  52
  53 salome_pluginsmanager.AddFunction('T shape fluid',
  54                                   'Creates the fluid part of a pipe T-shape',
  55                                   t_shape_fluid)