Salome HOME
remove std::cout.
[modules/geom.git] / src / GEOM_SWIG / GEOM_TestMeasures.py
1 #  -*- coding: iso-8859-1 -*-
2 #  Copyright (C) 2007-2010  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.
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   ####### WhatIs #######
51
52   Descr = geompy.WhatIs(box)
53   print "\nBox 10x30x70 description:"
54   print Descr
55
56   ####### NbShapes #######
57
58   NbSolids = geompy.NbShapes(box, geompy.ShapeType["SOLID"])
59   print "\nBox 10x30x70 quantity of solids:", NbSolids
60
61   ####### ShapeInfo #######
62
63   BoxInfo = geompy.ShapeInfo(box)
64   print "\nBox 10x30x70 shapes:"
65   print BoxInfo
66
67   ####### BasicProperties #######
68
69   Props = geompy.BasicProperties(box)
70   print "\nBox 10x30x70 Basic Properties:"
71   print " Wires length: ", Props[0]
72   print " Surface area: ", Props[1]
73   print " Volume      : ", Props[2]
74
75   dl = math.sqrt((Props[0] -   880)*(Props[0] -   880))
76   da = math.sqrt((Props[1] -  6200)*(Props[1] -  6200))
77   dv = math.sqrt((Props[2] - 21000)*(Props[2] - 21000))
78   #print "|Props[0] - 880| = ", dl
79
80   if dl > 1e-7 or da > 1e-7 or dv > 1e-7:
81     print "While must be:"
82     print " Wires length: ", 880
83     print " Surface area: ", 6200
84     print " Volume      : ", 21000
85
86   ####### BoundingBox #######
87
88   BB = geompy.BoundingBox(box)
89   print "\nBounding Box of box 10x30x70:"
90   print " Xmin = ", BB[0], ", Xmax = ", BB[1]
91   print " Ymin = ", BB[2], ", Ymax = ", BB[3]
92   print " Zmin = ", BB[4], ", Zmax = ", BB[5]
93
94   ####### Inertia #######
95
96   In = geompy.Inertia(box)
97   print "\nInertia matrix of box 10x30x70:"
98   print " (", In[0], ", ", In[1], ", ", In[2], ")"
99   print " (", In[3], ", ", In[4], ", ", In[5], ")"
100   print " (", In[6], ", ", In[7], ", ", In[8], ")"
101   print "Main moments of inertia of box 10x30x70:"
102   print " Ix = ", In[9], ", Iy = ", In[10], ", Iz = ", In[11]
103
104   ####### Tolerance #######
105
106   Toler = geompy.Tolerance(box)
107   print "\nBox 10x30x70 tolerance:"
108   print " Face min. tolerance: ", Toler[0]
109   print " Face max. tolerance: ", Toler[1]
110   print " Edge min. tolerance: ", Toler[2]
111   print " Edge max. tolerance: ", Toler[3]
112   print " Vertex min. tolerance: ", Toler[4]
113   print " Vertex max. tolerance: ", Toler[5]
114
115   ####### MakeCDG #######
116
117   pcdg = geompy.MakeCDG(box)
118   if pcdg is None:
119     raise RuntimeError, "MakeCDG(box) failed"
120   else:
121     print "\nCentre of gravity of box has been successfully obtained:"
122     Coords = geompy.PointCoordinates(pcdg)
123     print "(", Coords[0], ", ", Coords[1], ", ", Coords[2], ")"
124     if Coords[0] != 5 or Coords[1] != 15 or Coords[2] != 35:
125       print "But must be (5, 15, 35)"
126
127   ####### GetNormal #######
128
129   faces = geompy.SubShapeAllSortedCentres(box, geompy.ShapeType["FACE"])
130   face0 = faces[0]
131   vnorm = geompy.GetNormal(face0)
132   if vnorm is None:
133     raise RuntimeError, "GetNormal(face0) failed"
134   else:
135     geompy.addToStudy(face0, "Face0")
136     geompy.addToStudy(vnorm, "Normale to Face0")
137     print "\nNormale of face has been successfully obtained:"
138     #Coords = geompy.PointCoordinates(pcdg)
139     #print "(", Coords[0], ", ", Coords[1], ", ", Coords[2], ")"
140     #if Coords[0] != 5 or Coords[1] != 15 or Coords[2] != 35:
141     #  print "But must be (5, 15, 35)"
142
143   ####### MinDistance #######
144
145   MinDist = geompy.MinDistance(box, cube)
146
147   #print "\nMinimal distance between Box and Cube = ", MinDist[0]
148   #print "It is reached at points:"
149   #print " On Box  (", MinDist[1], ", ", MinDist[2], ", ", MinDist[3], ")"
150   #print " On Cube (", MinDist[4], ", ", MinDist[5], ", ", MinDist[6], ")"
151
152   print "\nMinimal distance between Box and Cube = ", MinDist
153
154   MinDistComps = geompy.MinDistanceComponents(box, cube)
155   print "\nMinimal distance between Box and Cube = ", MinDistComps[0]
156   print "Its components are  (", MinDistComps[1], ", ", MinDistComps[2], ", ", MinDistComps[3], ")"
157
158   ####### Angle #######
159
160   OX  = geompy.MakeVectorDXDYDZ(10, 0,0)
161   OXY = geompy.MakeVectorDXDYDZ(10,10,0)
162
163   # in one plane
164   Angle = geompy.GetAngle(OX, OXY)
165
166   print "\nAngle between OX and OXY = ", Angle
167   if math.fabs(Angle - 45.0) > 1e-05:
168     print "  Error: returned angle is", Angle, "while must be 45.0"
169
170   Angle = geompy.GetAngleRadians(OX, OXY)
171
172   print "\nAngle between OX and OXY in radians = ", Angle
173   if math.fabs(Angle - math.pi/4) > 1e-05:
174     print "  Error: returned angle is", Angle, "while must be pi/4"
175     pass
176
177   # not in one plane
178   OXY_shift = geompy.MakeTranslation(OXY,10,-10,20)
179   Angle = geompy.GetAngle(OX, OXY_shift)
180
181   print "Angle between OX and OXY_shift = ", Angle
182   if math.fabs(Angle - 45.0) > 1e-05:
183     print "  Error: returned angle is", Angle, "while must be 45.0"
184
185   ####### Position (LCS) #######
186
187   Pos = geompy.GetPosition(box)
188   print "\nPosition(LCS) of box 10x30x70:"
189   print "Origin: (", Pos[0], ", ", Pos[1], ", ", Pos[2], ")"
190   print "Z axis: (", Pos[3], ", ", Pos[4], ", ", Pos[5], ")"
191   print "X axis: (", Pos[6], ", ", Pos[7], ", ", Pos[8], ")"
192
193   ####### KindOfShape #######
194
195   Kind = geompy.KindOfShape(box)
196   print "\nKindOfShape(box 10x30x70):", Kind
197   #if Kind[0] != geompy.kind.BOX:
198   #  print "Error: returned type is", Kind[0], "while must be", geompy.kind.BOX
199
200   Kind = geompy.KindOfShape(p137)
201   print "\nKindOfShape(p137):", Kind
202   if Kind[0] != geompy.kind.VERTEX:
203     print "  Error: returned type is", Kind[0], "while must be", geompy.kind.VERTEX
204   else:
205     dx = math.fabs(Kind[1] - 10)
206     dy = math.fabs(Kind[2] - 30)
207     dz = math.fabs(Kind[3] - 70)
208     if (dx + dy + dz) > 1e-5:
209       print "  Error: coordinates are (", Kind[1], ",", Kind[2], ",", Kind[3], ") while must be (10, 20, 30)"
210
211   pass