Salome HOME
2ad4dc6e7c79805609d904b6078d016319e467ba
[modules/smesh.git] / src / Tools / blocFissure / gmu / facesVolumesToriques.py
1 # -*- coding: utf-8 -*-
2
3 import logging
4 from geomsmesh import geompy
5 from extractionOrientee import extractionOrientee
6 from getSubshapeIds import getSubshapeIds
7
8 # -----------------------------------------------------------------------------
9 # --- TORE
10 # --- faces toriques  et volumes du tore
11
12 def facesVolumesToriques(tore, plan, facesDefaut):
13   """
14   Extraction des deux faces  et volumes du tore partitionné, qui suivent la génératrice elliptique.
15   @param tore : le tore partitionné et coupé.
16   @param plan : le plan de coupe
17   @return (facetore1,facetore2) les 2 faces selon la génératrice
18   """
19   logging.info("start")
20
21   centre = geompy.MakeVertexOnSurface(plan, 0.5, 0.5)
22   normal = geompy.GetNormal(plan, centre)
23   reference = geompy.MakeTranslationVector(centre, normal)
24
25   [facesInPlan, facesOutPlan, facesOnPlan] = extractionOrientee(plan, tore, reference, "FACE", 1.e-2, "faceTorePlan_")
26   [facesInSide, facesOutSide, facesOnSide] = extractionOrientee(facesDefaut, tore, reference, "FACE", 1.e-2, "faceTorePeau_")
27   facesIdInPlan = getSubshapeIds(tore, facesInPlan)
28   facesIdOutPlan = getSubshapeIds(tore, facesOutPlan)
29   facesIdOnSide = getSubshapeIds(tore, facesOnSide)
30   facesIdInSide = getSubshapeIds(tore, facesInSide)
31   facesIdOutSide = getSubshapeIds(tore, facesOutSide)
32   #facesIdInOutSide = facesIdInSide + facesIdOutSide
33   facetore1 = None
34   faceTore2 = None
35   for i, faceId in enumerate(facesIdInPlan):
36     if faceId not in facesIdOnSide:
37       facetore1 = facesInPlan[i]
38       break
39   for i, faceId in enumerate(facesIdOutPlan):
40     if faceId not in facesIdOnSide:
41       facetore2 = facesOutPlan[i]
42       break
43
44   #[facetore1,facetore2] = geompy.GetShapesOnShape(pipe0, tore, geompy.ShapeType["FACE"], GEOM.ST_ON)
45
46   geompy.addToStudyInFather( tore, facetore1, 'facetore1' )
47   geompy.addToStudyInFather( tore, facetore2, 'facetore2' )
48
49   [volumeTore1, volumeTore2] = geompy.ExtractShapes(tore, geompy.ShapeType["SOLID"], True)
50   geompy.addToStudyInFather( tore, volumeTore1, 'volumeTore1' )
51   geompy.addToStudyInFather( tore, volumeTore2, 'volumeTore2' )
52
53   return facetore1, facetore2, volumeTore1, volumeTore2