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