Salome HOME
817a73082603e2dfdb693aee3d83751e91895e4f
[modules/geom.git] / src / GEOM_SWIG / GEOM_TestMeasures.py
1 #  -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2014  CEA/DEN, EDF R&D, OPEN CASCADE
3 #
4 # Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 #
7 # This library is free software; you can redistribute it and/or
8 # modify it under the terms of the GNU Lesser General Public
9 # License as published by the Free Software Foundation; either
10 # version 2.1 of the License, or (at your option) any later version.
11 #
12 # This library is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15 # Lesser General Public License for more details.
16 #
17 # You should have received a copy of the GNU Lesser General Public
18 # License along with this library; if not, write to the Free Software
19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
20 #
21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 #
23
24 def TestMeasureOperations (geompy, math):
25
26   p0   = geompy.MakeVertex(0 ,  0,  0)
27   p137 = geompy.MakeVertex(10, 30, 70)
28
29   box = geompy.MakeBoxTwoPnt(p0, p137)
30
31   p678 = geompy.MakeVertex(60, 70, 80)
32   p789 = geompy.MakeVertex(70, 80, 90)
33
34   cube = geompy.MakeBoxTwoPnt(p678, p789)
35
36   ####### PointCoordinates #######
37
38   Coords = geompy.PointCoordinates(p137)
39   if Coords[0] != 10 or Coords[1] != 30 or Coords[2] != 70:
40     print "Coordinates of p137 must be (10, 30, 70), but returned (", Coords[0], ", ", Coords[1], ", ", Coords[2], ")"
41
42   ####### CheckShape #######
43
44   IsValid = geompy.CheckShape(box)
45   if IsValid == 0:
46     raise RuntimeError, "Invalid box created"
47   else:
48     print "\nBox is valid"
49
50   ####### Detect Self-intersections #######
51
52   [Face_1,Face_2] = geompy.SubShapes(box, [33, 23])
53   Translation_1 = geompy.MakeTranslation(Face_1, 5, -15, -40)
54   Compound_1 = geompy.MakeCompound([Face_2, Translation_1])
55   if geompy.CheckSelfIntersections(Compound_1) == True:
56     raise RuntimeError, "Existing self-intersection is not detected"
57
58   ####### WhatIs #######
59
60   Descr = geompy.WhatIs(box)
61   print "\nBox 10x30x70 description:"
62   print Descr
63
64   ####### NbShapes #######
65
66   NbSolids = geompy.NbShapes(box, geompy.ShapeType["SOLID"])
67   print "\nBox 10x30x70 quantity of solids:", NbSolids
68
69   ####### ShapeInfo #######
70
71   BoxInfo = geompy.ShapeInfo(box)
72   print "\nBox 10x30x70 shapes:"
73   print BoxInfo
74
75   ####### BasicProperties #######
76
77   Props = geompy.BasicProperties(box)
78   print "\nBox 10x30x70 Basic Properties:"
79   print " Wires length: ", Props[0]
80   print " Surface area: ", Props[1]
81   print " Volume      : ", Props[2]
82
83   dl = math.sqrt((Props[0] -   880)*(Props[0] -   880))
84   da = math.sqrt((Props[1] -  6200)*(Props[1] -  6200))
85   dv = math.sqrt((Props[2] - 21000)*(Props[2] - 21000))
86   #print "|Props[0] - 880| = ", dl
87
88   if dl > 1e-7 or da > 1e-7 or dv > 1e-7:
89     print "While must be:"
90     print " Wires length: ", 880
91     print " Surface area: ", 6200
92     print " Volume      : ", 21000
93
94   ####### BoundingBox #######
95
96   BB = geompy.BoundingBox(box)
97   print "\nBounding Box of box 10x30x70:"
98   print " Xmin = ", BB[0], ", Xmax = ", BB[1]
99   print " Ymin = ", BB[2], ", Ymax = ", BB[3]
100   print " Zmin = ", BB[4], ", Zmax = ", BB[5]
101   BB = geompy.MakeBoundingBox(box)
102   geompy.addToStudy(BB, "BoundingBox")
103
104   ####### Inertia #######
105
106   In = geompy.Inertia(box)
107   print "\nInertia matrix of box 10x30x70:"
108   print " (", In[0], ", ", In[1], ", ", In[2], ")"
109   print " (", In[3], ", ", In[4], ", ", In[5], ")"
110   print " (", In[6], ", ", In[7], ", ", In[8], ")"
111   print "Main moments of inertia of box 10x30x70:"
112   print " Ix = ", In[9], ", Iy = ", In[10], ", Iz = ", In[11]
113
114   ####### Tolerance #######
115
116   Toler = geompy.Tolerance(box)
117   print "\nBox 10x30x70 tolerance:"
118   print " Face min. tolerance: ", Toler[0]
119   print " Face max. tolerance: ", Toler[1]
120   print " Edge min. tolerance: ", Toler[2]
121   print " Edge max. tolerance: ", Toler[3]
122   print " Vertex min. tolerance: ", Toler[4]
123   print " Vertex max. tolerance: ", Toler[5]
124
125   ####### MakeCDG #######
126
127   pcdg = geompy.MakeCDG(box)
128   if pcdg is None:
129     raise RuntimeError, "MakeCDG(box) failed"
130   else:
131     print "\nCentre of gravity of box has been successfully obtained:"
132     Coords = geompy.PointCoordinates(pcdg)
133     print "(", Coords[0], ", ", Coords[1], ", ", Coords[2], ")"
134     if Coords[0] != 5 or Coords[1] != 15 or Coords[2] != 35:
135       print "But must be (5, 15, 35)"
136
137   ####### GetNormal #######
138
139   faces = geompy.SubShapeAllSortedCentres(box, geompy.ShapeType["FACE"])
140   face0 = faces[0]
141   vnorm = geompy.GetNormal(face0)
142   if vnorm is None:
143     raise RuntimeError, "GetNormal(face0) failed"
144   else:
145     geompy.addToStudy(face0, "Face0")
146     geompy.addToStudy(vnorm, "Normale to Face0")
147     print "\nNormale of face has been successfully obtained:"
148     #Coords = geompy.PointCoordinates(pcdg)
149     #print "(", Coords[0], ", ", Coords[1], ", ", Coords[2], ")"
150     #if Coords[0] != 5 or Coords[1] != 15 or Coords[2] != 35:
151     #  print "But must be (5, 15, 35)"
152
153   ####### MinDistance #######
154
155   MinDist = geompy.MinDistance(box, cube)
156
157   #print "\nMinimal distance between Box and Cube = ", MinDist[0]
158   #print "It is reached at points:"
159   #print " On Box  (", MinDist[1], ", ", MinDist[2], ", ", MinDist[3], ")"
160   #print " On Cube (", MinDist[4], ", ", MinDist[5], ", ", MinDist[6], ")"
161
162   print "\nMinimal distance between Box and Cube = ", MinDist
163
164   MinDistComps = geompy.MinDistanceComponents(box, cube)
165   print "\nMinimal distance between Box and Cube = ", MinDistComps[0]
166   print "Its components are  (", MinDistComps[1], ", ", MinDistComps[2], ", ", MinDistComps[3], ")"
167
168   # Get all closest points
169   [nbSols, listCoords] = geompy.ClosestPoints(box, cube)
170   for i in range(nbSols):
171     v1 = geompy.MakeVertex(listCoords[i*6 + 0], listCoords[i*6 + 1], listCoords[i*6 + 2])
172     v2 = geompy.MakeVertex(listCoords[i*6 + 3], listCoords[i*6 + 4], listCoords[i*6 + 5])
173
174     geompy.addToStudy(v1, 'MinDist_%d_on_Box'%(i+1))
175     geompy.addToStudy(v2, 'MinDist_%d_on_Cube'%(i+1))
176     pass
177
178   ####### Angle #######
179
180   OX  = geompy.MakeVectorDXDYDZ(10, 0,0)
181   OXY = geompy.MakeVectorDXDYDZ(10,10,0)
182
183   # in one plane
184   Angle = geompy.GetAngle(OX, OXY)
185
186   print "\nAngle between OX and OXY = ", Angle
187   if math.fabs(Angle - 45.0) > 1e-05:
188     print "  Error: returned angle is", Angle, "while must be 45.0"
189
190   Angle = geompy.GetAngleRadians(OX, OXY)
191
192   print "\nAngle between OX and OXY in radians = ", Angle
193   if math.fabs(Angle - math.pi/4) > 1e-05:
194     print "  Error: returned angle is", Angle, "while must be pi/4"
195     pass
196
197   # not in one plane
198   OXY_shift = geompy.MakeTranslation(OXY,10,-10,20)
199   Angle = geompy.GetAngle(OX, OXY_shift)
200
201   print "Angle between OX and OXY_shift = ", Angle
202   if math.fabs(Angle - 45.0) > 1e-05:
203     print "  Error: returned angle is", Angle, "while must be 45.0"
204
205   ####### Position (LCS) #######
206
207   Pos = geompy.GetPosition(box)
208   print "\nPosition(LCS) of box 10x30x70:"
209   print "Origin: (", Pos[0], ", ", Pos[1], ", ", Pos[2], ")"
210   print "Z axis: (", Pos[3], ", ", Pos[4], ", ", Pos[5], ")"
211   print "X axis: (", Pos[6], ", ", Pos[7], ", ", Pos[8], ")"
212
213   ####### KindOfShape #######
214
215   Kind = geompy.KindOfShape(box)
216   print "\nKindOfShape(box 10x30x70):", Kind
217   #if Kind[0] != geompy.kind.BOX:
218   #  print "Error: returned type is", Kind[0], "while must be", geompy.kind.BOX
219
220   Kind = geompy.KindOfShape(p137)
221   print "\nKindOfShape(p137):", Kind
222   if Kind[0] != geompy.kind.VERTEX:
223     print "  Error: returned type is", Kind[0], "while must be", geompy.kind.VERTEX
224   else:
225     dx = math.fabs(Kind[1] - 10)
226     dy = math.fabs(Kind[2] - 30)
227     dz = math.fabs(Kind[3] - 70)
228     if (dx + dy + dz) > 1e-5:
229       print "  Error: coordinates are (", Kind[1], ",", Kind[2], ",", Kind[3], ") while must be (10, 20, 30)"
230
231   pass