first working version of the plugin

This commit is contained in:
Renaud Nédélec 2014-06-20 11:57:06 +02:00
parent c07b5ec312
commit 4d7ec7e96b
3 changed files with 215 additions and 189 deletions

View File

@ -22,15 +22,21 @@
import salome_pluginsmanager
def t_shape_fluid(context):
activeStudy = context.study
import t_shape_builder
import t_shape_dialog
import xalome
activeStudy = context.study
dialog = t_shape_dialog.TShapeDialog()
# Get the parameter values from a gui dialog box. If the dialog is
# closed using the Ok button, then the data are requested from the
# gui and used to create the shape of the tube.
dialog.exec_()
dialog.getData()
r1, r2, h1, h2 = dialog.getData()
shape = t_shape_builder.build_shape(activeStudy, r1, r2, h1, h2)
entry = xalome.addToStudy(activeStudy, shape, "T_shape_fluid" )
xalome.displayShape(entry)
#if dialog.wasOk():
#radius, length, width = dialog.getData()
#shape = tubebuilder.createGeometry(activeStudy, radius, length, width)

View File

@ -3,38 +3,35 @@
import sys
import salome
salome.salome_init()
theStudy = salome.myStudy
import GEOM
from salome.geom import geomBuilder
import math
import SALOMEDS
geompy = geomBuilder.New(theStudy)
O = geompy.MakeVertex(0, 0, 0)
OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
#O = geompy.MakeVertex(0, 0, 0)
#OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
#OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
#OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
geompy.addToStudy( O, 'O' )
geompy.addToStudy( OX, 'OX' )
geompy.addToStudy( OY, 'OY' )
geompy.addToStudy( OZ, 'OZ' )
#geompy.addToStudy( O, 'O' )
#geompy.addToStudy( OX, 'OX' )
#geompy.addToStudy( OY, 'OY' )
#geompy.addToStudy( OZ, 'OZ' )
r1 = 100.0
r2 = 50.0
h1 = 200.0
h2 = 200.0
#r1 = 100.0
#r2 = 50.0
#h1 = 200.0
#h2 = 200.0
a1 = 45.0
seuilmax = 0.1
ratio = float(r2)/float(r1)
if ratio > (1.0 -seuilmax):
a1 = 45.0*(1.0 -ratio)/seuilmax
def demidisk(r1, a1, roty=0):
def demidisk(study, r1, a1, roty=0):
geompy = geomBuilder.New(study)
O = geompy.MakeVertex(0, 0, 0)
OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
v=range(8)
l=range(8)
v0 = geompy.MakeVertex(0, 0, 0)
@ -71,7 +68,8 @@ def demidisk(r1, a1, roty=0):
else:
return v, l, arc1, part1
def pointsProjetes(vref, face):
def pointsProjetes(study, vref, face):
geompy = geomBuilder.New(study)
vface = geompy.ExtractShapes(face, geompy.ShapeType["VERTEX"], True)
vord = range(len(vref))
plan = geompy.MakePlaneThreePnt(vref[0], vref[1], vref[-1], 10000)
@ -84,7 +82,8 @@ def pointsProjetes(vref, face):
vord[dist[0][1]] = vface[i]
return vord
def arcsProjetes(vf, face):
def arcsProjetes(study, vf, face):
geompy = geomBuilder.New(study)
lface = geompy.ExtractShapes(face, geompy.ShapeType["EDGE"], True)
lord = range(3)
ends = [vf[1], vf[6], vf[7], vf[3]]
@ -98,170 +97,186 @@ def arcsProjetes(vf, face):
break
pass
return lord
def build_shape(study, r1, r2, h1, h2):
geompy = geomBuilder.New(study)
# --- sections droites des deux demi cylindres avec le partionnement
v1, l1, arc1, part1 = demidisk(r1, a1)
v2, l2, arc2, part2 = demidisk(r2, a1, 90.0)
geompy.addToStudy(part1, 'part1')
geompy.addToStudy(part2, 'part2')
# --- extrusion des sections --> demi cylindres de travail, pour en extraire les sections utilisées au niveau du Té
# et enveloppe cylindrique du cylindre principal
demicyl1 = geompy.MakePrismVecH(part1, OX, h1)
demicyl2 = geompy.MakePrismVecH(part2, OZ, h2)
arcextru = geompy.MakePrismVecH(arc1, OX, h1)
geompy.addToStudy(demicyl1, 'demicyl1')
geompy.addToStudy(demicyl2, 'demicyl2')
geompy.addToStudy(arcextru, 'arcextru')
# --- plan de coupe à 45° sur le cylindre principal,
# section à 45° du cylndre principal,
# section du cylindre secondaire par l'enveloppe cylindique du cylindre principal
plan1 = geompy.MakePlane(O, OX, 4*r1)
planr = geompy.MakeRotation(plan1, OY, 45*math.pi/180.0)
geompy.addToStudy(planr, 'planr')
sect45 = geompy.MakeCommonList([demicyl1, planr], True)
geompy.addToStudy(sect45, 'sect45')
sect90 = geompy.MakeCommonList([demicyl2, arcextru], True)
geompy.addToStudy(sect90, 'sect90')
# --- liste ordonnée des points projetés sur les deux sections
vord45 = pointsProjetes(v1, sect45)
vord90 = pointsProjetes(v2, sect90)
for i,v in enumerate(vord45):
geompy.addToStudyInFather(sect45, v, 'v%d'%i)
for i,v in enumerate(vord90):
geompy.addToStudyInFather(sect90, v, 'v%d'%i)
# --- identification des projections des trois arcs de cercle, sur les deux sections.
lord45 = arcsProjetes(vord45, sect45)
lord90 = arcsProjetes(vord90, sect90)
for i,l in enumerate(lord45):
geompy.addToStudyInFather(sect45, l, 'l%d'%i)
for i,l in enumerate(lord90):
geompy.addToStudyInFather(sect90, l, 'l%d'%i)
# --- abaissement des quatre points centraux de la section du cylindre secondaire
dz = -r2/2.0
for i in (0, 2, 4, 5):
vord90[i] = geompy.TranslateDXDYDZ(vord90[i], 0, 0, dz, True)
geompy.addToStudyInFather(sect90, vord90[i], 'vm%d'%i)
O = geompy.MakeVertex(0, 0, 0)
OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
# --- création des deux arêtes curvilignes sur l'enveloppe cylindrique du cylindre principal, à la jonction
a1 = 45.0
seuilmax = 0.1
ratio = float(r2)/float(r1)
if ratio > (1.0 -seuilmax):
a1 = 45.0*(1.0 -ratio)/seuilmax
# --- sections droites des deux demi cylindres avec le partionnement
curv = [None for i in range(4)] # liaisons entre les points 1, 3, 6 et 7 des 2 sections
v1, l1, arc1, part1 = demidisk(study, r1, a1)
v2, l2, arc2, part2 = demidisk(study, r2, a1, 90.0)
curv[0] = geompy.MakeArcCenter(O, vord90[1] , vord45[1], False)
curv[1] = geompy.MakeArcCenter(O, vord90[3] , vord45[3], False)
geompy.addToStudy(part1, 'part1')
geompy.addToStudy(part2, 'part2')
lipts = ((6, 6, 4), (7, 7, 5))
for i, ipts in enumerate(lipts):
print i, ipts
p0 = vord90[ipts[0]]
p1 = vord45[ipts[1]]
p2 = vord45[ipts[2]]
plan = geompy.MakePlaneThreePnt(p0, p1, p2, 10000)
#geompy.addToStudy(plan, "plan%d"%i)
section = geompy.MakeSection(plan, arcextru, True)
secpart = geompy.MakePartition([section], [sect45, sect90], [], [], geompy.ShapeType["EDGE"], 0, [], 0, True)
#geompy.addToStudy(secpart, "secpart%d"%i)
lsec = geompy.ExtractShapes(secpart, geompy.ShapeType["EDGE"], True)
#print "len(lsec)", len(lsec)
for l in lsec:
pts = geompy.ExtractShapes(l, geompy.ShapeType["VERTEX"], True)
if (((geompy.MinDistance(pts[0], p0) < 0.001) and (geompy.MinDistance(pts[1], p1) < 0.001)) or
((geompy.MinDistance(pts[1], p0) < 0.001) and (geompy.MinDistance(pts[0], p1) < 0.001))):
curv[i+2] =l
#print "curv_%d OK"%i
break
for i,l in enumerate(curv):
geompy.addToStudyInFather(arcextru, l, "curv%d"%i)
# --- extrusion des sections --> demi cylindres de travail, pour en extraire les sections utilisées au niveau du Té
# et enveloppe cylindrique du cylindre principal
demicyl1 = geompy.MakePrismVecH(part1, OX, h1)
demicyl2 = geompy.MakePrismVecH(part2, OZ, h2)
arcextru = geompy.MakePrismVecH(arc1, OX, h1)
geompy.addToStudy(demicyl1, 'demicyl1')
geompy.addToStudy(demicyl2, 'demicyl2')
geompy.addToStudy(arcextru, 'arcextru')
# --- plan de coupe à 45° sur le cylindre principal,
# section à 45° du cylndre principal,
# section du cylindre secondaire par l'enveloppe cylindique du cylindre principal
plan1 = geompy.MakePlane(O, OX, 4*r1)
planr = geompy.MakeRotation(plan1, OY, 45*math.pi/180.0)
geompy.addToStudy(planr, 'planr')
sect45 = geompy.MakeCommonList([demicyl1, planr], True)
geompy.addToStudy(sect45, 'sect45')
sect90 = geompy.MakeCommonList([demicyl2, arcextru], True)
geompy.addToStudy(sect90, 'sect90')
# --- liste ordonnée des points projetés sur les deux sections
vord45 = pointsProjetes(study, v1, sect45)
vord90 = pointsProjetes(study, v2, sect90)
for i,v in enumerate(vord45):
geompy.addToStudyInFather(sect45, v, 'v%d'%i)
for i,v in enumerate(vord90):
geompy.addToStudyInFather(sect90, v, 'v%d'%i)
# --- identification des projections des trois arcs de cercle, sur les deux sections.
lord45 = arcsProjetes(study, vord45, sect45)
lord90 = arcsProjetes(study, vord90, sect90)
for i,l in enumerate(lord45):
geompy.addToStudyInFather(sect45, l, 'l%d'%i)
for i,l in enumerate(lord90):
geompy.addToStudyInFather(sect90, l, 'l%d'%i)
# --- abaissement des quatre points centraux de la section du cylindre secondaire
dz = -r2/2.0
for i in (0, 2, 4, 5):
vord90[i] = geompy.TranslateDXDYDZ(vord90[i], 0, 0, dz, True)
geompy.addToStudyInFather(sect90, vord90[i], 'vm%d'%i)
# --- création des deux arêtes curvilignes sur l'enveloppe cylindrique du cylindre principal, à la jonction
curv = [None for i in range(4)] # liaisons entre les points 1, 3, 6 et 7 des 2 sections
curv[0] = geompy.MakeArcCenter(O, vord90[1] , vord45[1], False)
curv[1] = geompy.MakeArcCenter(O, vord90[3] , vord45[3], False)
lipts = ((6, 6, 4), (7, 7, 5))
for i, ipts in enumerate(lipts):
print i, ipts
p0 = vord90[ipts[0]]
p1 = vord45[ipts[1]]
p2 = vord45[ipts[2]]
plan = geompy.MakePlaneThreePnt(p0, p1, p2, 10000)
#geompy.addToStudy(plan, "plan%d"%i)
section = geompy.MakeSection(plan, arcextru, True)
secpart = geompy.MakePartition([section], [sect45, sect90], [], [], geompy.ShapeType["EDGE"], 0, [], 0, True)
#geompy.addToStudy(secpart, "secpart%d"%i)
lsec = geompy.ExtractShapes(secpart, geompy.ShapeType["EDGE"], True)
#print "len(lsec)", len(lsec)
for l in lsec:
pts = geompy.ExtractShapes(l, geompy.ShapeType["VERTEX"], True)
if (((geompy.MinDistance(pts[0], p0) < 0.001) and (geompy.MinDistance(pts[1], p1) < 0.001)) or
((geompy.MinDistance(pts[1], p0) < 0.001) and (geompy.MinDistance(pts[0], p1) < 0.001))):
curv[i+2] =l
#print "curv_%d OK"%i
break
for i,l in enumerate(curv):
geompy.addToStudyInFather(arcextru, l, "curv%d"%i)
# --- creation des arêtes droites manquantes, des faces et volumes pour les quatre volumes de la jonction
edges = [None for i in range(8)]
edges[0] = geompy.MakeLineTwoPnt(vord45[0], vord90[0])
edges[1] = curv[0]
edges[2] = geompy.MakeLineTwoPnt(vord45[2], vord90[2])
edges[3] = curv[1]
edges[4] = geompy.MakeLineTwoPnt(vord45[4], vord90[4])
edges[5] = geompy.MakeLineTwoPnt(vord45[5], vord90[5])
edges[6] = curv[2]
edges[7] = curv[3]
for i,l in enumerate(edges):
geompy.addToStudy( l, "edge%d"%i)
ed45 = [None for i in range(8)]
ed45[0] = geompy.MakeLineTwoPnt(vord45[0], vord45[2])
ed45[1] = geompy.MakeLineTwoPnt(vord45[0], vord45[1])
ed45[2] = geompy.MakeLineTwoPnt(vord45[4], vord45[6])
ed45[3] = geompy.MakeLineTwoPnt(vord45[2], vord45[3])
ed45[4] = geompy.MakeLineTwoPnt(vord45[5], vord45[7])
ed45[5] = geompy.MakeLineTwoPnt(vord45[4], vord45[5])
ed45[6] = geompy.MakeLineTwoPnt(vord45[0], vord45[4])
ed45[7] = geompy.MakeLineTwoPnt(vord45[2], vord45[5])
for i,l in enumerate(ed45):
geompy.addToStudyInFather(sect45, l, "ed45_%d"%i)
ed90 = [None for i in range(8)]
ed90[0] = geompy.MakeLineTwoPnt(vord90[0], vord90[2])
ed90[1] = geompy.MakeLineTwoPnt(vord90[0], vord90[1])
ed90[2] = geompy.MakeLineTwoPnt(vord90[4], vord90[6])
ed90[3] = geompy.MakeLineTwoPnt(vord90[2], vord90[3])
ed90[4] = geompy.MakeLineTwoPnt(vord90[5], vord90[7])
ed90[5] = geompy.MakeLineTwoPnt(vord90[4], vord90[5])
ed90[6] = geompy.MakeLineTwoPnt(vord90[0], vord90[4])
ed90[7] = geompy.MakeLineTwoPnt(vord90[2], vord90[5])
for i,l in enumerate(ed90):
geompy.addToStudyInFather(sect90, l, "ed90_%d"%i)
faci = []
faci.append(geompy.MakeFaceWires([ed45[6], edges[0], ed90[6], edges[4]], 0))
faci.append(geompy.MakeFaceWires([ed45[7], edges[2], ed90[7], edges[5]], 0))
faci.append(geompy.MakeFaceWires([ed45[2], edges[4], ed90[2], edges[6]], 0))
faci.append(geompy.MakeFaceWires([ed45[5], edges[4], ed90[5], edges[5]], 0))
faci.append(geompy.MakeFaceWires([ed45[4], edges[5], ed90[4], edges[7]], 0))
faci.append(geompy.MakeFaceWires([ed90[0], ed90[6], ed90[5], ed90[7]], 0))
faci.append(geompy.MakeFaceWires([ed90[1], ed90[6], ed90[2], lord90[0]], 0))
faci.append(geompy.MakeFaceWires([ed90[2], ed90[5], ed90[4], lord90[1]], 0))
faci.append(geompy.MakeFaceWires([ed90[3], ed90[7], ed90[4], lord90[2]], 0))
for i,f in enumerate(faci):
geompy.addToStudy(f, "faci_%d"%i)
# --- extrusion droite des faces de jonction, pour reconstituer les demi cylindres
extru1 = geompy.MakePrismVecH(sect45, OX, h1+10)
geompy.addToStudy(extru1, "extru1")
base2 = geompy.MakePartition(faci[5:], [], [], [], geompy.ShapeType["FACE"], 0, [], 0, True)
extru2 = geompy.MakePrismVecH(base2, OZ, h2)
geompy.addToStudy(extru2, "extru2")
# --- partition et coupe
demiDisque = geompy.MakeFaceWires([arc1, l1[0]], 1)
demiCylindre = geompy.MakePrismVecH(demiDisque, OX, h1)
#geompy.addToStudy(demiCylindre, "demiCylindre")
box = geompy.MakeBox(0, -2*(r1+h1), -2*(r1+h1), 2*(r1+h1), 2*(r1+h1), 2*(r1+h1))
rot = geompy.MakeRotation(box, OY, 45*math.pi/180.0)
#geompy.addToStudy(rot, "rot")
garder = geompy.MakeCutList(demiCylindre, [extru2, rot], True)
geompy.addToStudy(garder, "garder")
raccord = geompy.MakePartition([garder], faci, [], [], geompy.ShapeType["SOLID"], 0, [], 0, True)
assemblage = geompy.MakePartition([raccord, extru1, extru2], [], [], [], geompy.ShapeType["SOLID"], 0, [], 0, True)
geompy.addToStudy(assemblage, "assemblage")
box = geompy.MakeBox(-1, -(r1+r2), -1, h1, r1+r2, h2)
geompy.addToStudy(box, "box")
final = geompy.MakeCommonList([box, assemblage], True)
# --- creation des arêtes droites manquantes, des faces et volumes pour les quatre volumes de la jonction
edges = [None for i in range(8)]
edges[0] = geompy.MakeLineTwoPnt(vord45[0], vord90[0])
edges[1] = curv[0]
edges[2] = geompy.MakeLineTwoPnt(vord45[2], vord90[2])
edges[3] = curv[1]
edges[4] = geompy.MakeLineTwoPnt(vord45[4], vord90[4])
edges[5] = geompy.MakeLineTwoPnt(vord45[5], vord90[5])
edges[6] = curv[2]
edges[7] = curv[3]
for i,l in enumerate(edges):
geompy.addToStudy( l, "edge%d"%i)
ed45 = [None for i in range(8)]
ed45[0] = geompy.MakeLineTwoPnt(vord45[0], vord45[2])
ed45[1] = geompy.MakeLineTwoPnt(vord45[0], vord45[1])
ed45[2] = geompy.MakeLineTwoPnt(vord45[4], vord45[6])
ed45[3] = geompy.MakeLineTwoPnt(vord45[2], vord45[3])
ed45[4] = geompy.MakeLineTwoPnt(vord45[5], vord45[7])
ed45[5] = geompy.MakeLineTwoPnt(vord45[4], vord45[5])
ed45[6] = geompy.MakeLineTwoPnt(vord45[0], vord45[4])
ed45[7] = geompy.MakeLineTwoPnt(vord45[2], vord45[5])
for i,l in enumerate(ed45):
geompy.addToStudyInFather(sect45, l, "ed45_%d"%i)
ed90 = [None for i in range(8)]
ed90[0] = geompy.MakeLineTwoPnt(vord90[0], vord90[2])
ed90[1] = geompy.MakeLineTwoPnt(vord90[0], vord90[1])
ed90[2] = geompy.MakeLineTwoPnt(vord90[4], vord90[6])
ed90[3] = geompy.MakeLineTwoPnt(vord90[2], vord90[3])
ed90[4] = geompy.MakeLineTwoPnt(vord90[5], vord90[7])
ed90[5] = geompy.MakeLineTwoPnt(vord90[4], vord90[5])
ed90[6] = geompy.MakeLineTwoPnt(vord90[0], vord90[4])
ed90[7] = geompy.MakeLineTwoPnt(vord90[2], vord90[5])
for i,l in enumerate(ed90):
geompy.addToStudyInFather(sect90, l, "ed90_%d"%i)
faci = []
faci.append(geompy.MakeFaceWires([ed45[6], edges[0], ed90[6], edges[4]], 0))
faci.append(geompy.MakeFaceWires([ed45[7], edges[2], ed90[7], edges[5]], 0))
faci.append(geompy.MakeFaceWires([ed45[2], edges[4], ed90[2], edges[6]], 0))
faci.append(geompy.MakeFaceWires([ed45[5], edges[4], ed90[5], edges[5]], 0))
faci.append(geompy.MakeFaceWires([ed45[4], edges[5], ed90[4], edges[7]], 0))
faci.append(geompy.MakeFaceWires([ed90[0], ed90[6], ed90[5], ed90[7]], 0))
faci.append(geompy.MakeFaceWires([ed90[1], ed90[6], ed90[2], lord90[0]], 0))
faci.append(geompy.MakeFaceWires([ed90[2], ed90[5], ed90[4], lord90[1]], 0))
faci.append(geompy.MakeFaceWires([ed90[3], ed90[7], ed90[4], lord90[2]], 0))
for i,f in enumerate(faci):
geompy.addToStudy(f, "faci_%d"%i)
# --- extrusion droite des faces de jonction, pour reconstituer les demi cylindres
extru1 = geompy.MakePrismVecH(sect45, OX, h1+10)
geompy.addToStudy(extru1, "extru1")
base2 = geompy.MakePartition(faci[5:], [], [], [], geompy.ShapeType["FACE"], 0, [], 0, True)
extru2 = geompy.MakePrismVecH(base2, OZ, h2)
geompy.addToStudy(extru2, "extru2")
# --- partition et coupe
demiDisque = geompy.MakeFaceWires([arc1, l1[0]], 1)
demiCylindre = geompy.MakePrismVecH(demiDisque, OX, h1)
#geompy.addToStudy(demiCylindre, "demiCylindre")
box = geompy.MakeBox(0, -2*(r1+h1), -2*(r1+h1), 2*(r1+h1), 2*(r1+h1), 2*(r1+h1))
rot = geompy.MakeRotation(box, OY, 45*math.pi/180.0)
#geompy.addToStudy(rot, "rot")
garder = geompy.MakeCutList(demiCylindre, [extru2, rot], True)
geompy.addToStudy(garder, "garder")
raccord = geompy.MakePartition([garder], faci, [], [], geompy.ShapeType["SOLID"], 0, [], 0, True)
assemblage = geompy.MakePartition([raccord, extru1, extru2], [], [], [], geompy.ShapeType["SOLID"], 0, [], 0, True)
geompy.addToStudy(assemblage, "assemblage")
box = geompy.MakeBox(-1, -(r1+r2), -1, h1, r1+r2, h2)
geompy.addToStudy(box, "box")
final = geompy.MakeCommonList([box, assemblage], True)
geompy.addToStudy(final, "final")
return final

View File

@ -40,7 +40,12 @@ class TShapeDialog(QtGui.QDialog):
QtGui.QDialog.accept(self)
def getData(self):
print "DATA : %f"%(self.ui.doubleSpinBox.value())
r1 = self.ui.doubleSpinBox.value()
r2 = self.ui.doubleSpinBox_2.value()
h1 = self.ui.doubleSpinBox_3.value()
h2 = self.ui.doubleSpinBox_4.value()
return r1, r2, h1, h2
#def setupUi(self):
#TShapeDialog_UI.setupUi(self)