Salome HOME
debug
[modules/smesh.git] / src / Tools / blocFissure / gmu / whichSide.py
1 # -*- coding: utf-8 -*-
2
3 import logging
4 from geomsmesh import geompy
5
6 # -----------------------------------------------------------------------------
7 # --- calcul de la position d'une shape par rapport à une face (dessus, dessous, sur la surface même)
8
9 def whichSide(face, obj, tol = 1.e-3):
10   """
11   Position d'un objet par rapport a une face (non nécessairement plane).
12   L'objet est supposé situé entierement d'un coté de la face,
13   ou lui appartenant totalement (objets traversants non pris en compte)
14   renvoie 1 si 'objet est du coté de la normale à la face,
15   -1 de l'autre coté, 0 si il est sur la face
16   """
17   logging.debug('start')
18   side = 0
19   logging.debug("shape info %s", geompy.ShapeInfo(obj))
20   nbEdges = geompy.NbShapes(obj, geompy.ShapeType["EDGE"]) # --- attention ! pour une seule edge presente, renvoie 2
21   logging.debug("   nbEdges %s", nbEdges)
22   nbFaces = geompy.NbShapes(obj, geompy.ShapeType["FACE"]) # --- attention ! pour une seule face presente, renvoie 2
23   logging.debug("   nbFaces %s", nbFaces)
24   vertices = geompy.ExtractShapes(obj, geompy.ShapeType["VERTEX"], False)
25   if nbEdges > 0 and nbFaces == 0: # --- edges
26     if nbEdges <= 2:
27       point = geompy.MakeVertexOnCurve(obj, 0.5)
28       vertices.append(point)
29     else:
30       edges = geompy.ExtractShapes(obj, geompy.ShapeType["EDGE"], False)
31       for anEdge in edges:
32         point = geompy.MakeVertexOnCurve(anEdge, 0.5)
33         vertices.append(point)
34   elif nbFaces >0: # --- faces
35     if nbFaces <=2:
36       point = geompy.MakeVertexOnSurface(obj, 0.5, 0.5)
37       vertices.append(point)
38     if nbFaces > 2:
39       faces = geompy.ExtractShapes(obj, geompy.ShapeType["FACE"], False)
40       for aFace in faces:
41         point = geompy.MakeVertexOnSurface(aFace, 0.5, 0.5)
42         vertices.append(point)
43   else: # --- vertices
44     vertices = [obj]
45   for vertex in vertices:
46     distance = geompy.MinDistance(vertex, face)
47     logging.debug("    distance %s", distance)
48     if distance > tol:
49       projection = geompy.MakeProjection(vertex, face)
50       normal = geompy.GetNormal(face, projection)
51       vect = geompy.MakeVector(projection, vertex)
52       angle = geompy.GetAngle(normal, vect)
53       logging.debug("  angle %s", angle)
54       side = 1
55       if abs(angle) > 10:
56         side = -1
57       break
58   logging.debug("  side %s", side)
59   return side
60