def test_StaticModel(self):
myContext = gem.initialize()
gem.setAttribute(myContext, "CONTEXT", 0, "Modeler", "OpenCASCADE")
myModel = gem.loadModel(myContext, sample_file)
server, filename, modeler, uptodate, myBReps, nparam, nbranch, nattr = gem.getModel(myModel)
newModel = gem.staticModel(myContext)
gem.add2Model(newModel, myBReps[0])
gem.add2Model(newModel, myBReps[0], (0.5, 0, 0, 2, 0, 0.5, 0, 0, 0, 0, 0.5, 0))
server, filename, modeler, uptodate, myBReps, nparam, nbranch, nattr = gem.getModel(newModel)
massProps1 = gem.getMassProps(myBReps[0], "BREP", 0)
massProps2 = gem.getMassProps(myBReps[1], "BREP", 0)
self.assertAlmostEqual(massProps1[0], 8 * massProps2[0]) # volume
self.assertAlmostEqual(massProps1[1], 4 * massProps2[1]) # surface area
self.assertAlmostEqual(massProps1[2], 2 * massProps2[2] - 4) # xcg
self.assertAlmostEqual(massProps1[3], 2 * massProps2[3]) # ycg
self.assertAlmostEqual(massProps1[4], 2 * massProps2[4]) # zcg
self.assertAlmostEqual(massProps1[5], 32 * massProps2[5]) # Ixx
self.assertAlmostEqual(massProps1[6], 32 * massProps2[6]) # Ixy
self.assertAlmostEqual(massProps1[7], 32 * massProps2[7]) # Ixz
self.assertAlmostEqual(massProps1[8], 32 * massProps2[8]) # Iyx
self.assertAlmostEqual(massProps1[9], 32 * massProps2[9]) # Iyy
self.assertAlmostEqual(massProps1[10], 32 * massProps2[10]) # Iyz
self.assertAlmostEqual(massProps1[11], 32 * massProps2[11]) # Izx
self.assertAlmostEqual(massProps1[12], 32 * massProps2[12]) # Izy
self.assertAlmostEqual(massProps1[13], 32 * massProps2[13]) # Izz
gem.releaseModel(newModel)
gem.releaseModel(myModel)
gem.terminate(myContext)
def test_BRep(self):
myContext = gem.initialize()
gem.setAttribute(myContext, "CONTEXT", 0, "Modeler", "OpenCASCADE")
myModel = gem.loadModel(myContext, sample_file)
server, filename, modeler, uptodate, myBReps, nparam, nbranch, nattr = gem.getModel(myModel)
self.assertEqual(nattr, 0)
for myBRep in myBReps:
box, type, nnode, nedge, nloop, nface, nshell, nattr = gem.getBRepInfo(myBRep)
# check that all nodes are in box
for inode in range(1, nnode + 1):
xyz, nattr = gem.getNode(myBRep, inode)
self.assertEqual(nattr, 0)
self.assertTrue(xyz[0] >= box[0], "x[%d] < box[0]" % inode)
self.assertTrue(xyz[1] >= box[1], "y[%d] < box[1]" % inode)
self.assertTrue(xyz[2] >= box[2], "z[%d] < box[2]" % inode)
self.assertTrue(xyz[0] <= box[3], "x[%d] > box[3]" % inode)
self.assertTrue(xyz[1] <= box[4], "y[%d] > box[4]" % inode)
self.assertTrue(xyz[2] <= box[5], "z[%d] > box[5]" % inode)
# check consistency of all loops
for iloop in range(1, nloop + 1):
iface, type, edges, nattr = gem.getLoop(myBRep, iloop)
self.assertEqual(nattr, 0, "iloop=%d" % iloop)
# loop and face point to each other
ID, uvbox, norm, loops, nattr = gem.getFace(myBRep, iface)
self.assertTrue(iloop in loops, "iloop=%d, iface=%d" % (iloop, iface))
# edge and face point to each other
for i in range(len(edges)):
tlimit, nodes, faces, nattr = gem.getEdge(myBRep, edges[i])
self.assertTrue(iface in faces, "iloop-%d, i=%d" % (iloop, i))
# edges link end-to-end in loop
tlimit, nodes, faces, nattr = gem.getEdge(myBRep, edges[0])
iend = nodes[1]
for i in range(1, len(edges)):
tlimit, nodes, faces, nattr = gem.getEdge(myBRep, edges[i])
self.assertEqual(iend, nodes[0], "iloop=%d, i=%d" % (iloop, i))
iend = nodes[1]
# check consistency of all shells
for ishell in range(1, nshell + 1):
type, faces, nattr = gem.getShell(myBRep, ishell)
self.assertEqual(nattr, 0)
gem.releaseModel(myModel)
gem.terminate(myContext)
def test_DRep(self):
myContext = gem.initialize()
gem.setAttribute(myContext, "CONTEXT", 0, "Modeler", "OpenCASCADE")
myModel = gem.loadModel(myContext, sample_file)
myDRep = gem.newDRep(myModel)
gem.tesselDRep(myDRep, 0, 0, 0, 0)
iface = 1
triArray, xyzArray = gem.getTessel(myDRep, 1, 1)
npnt = (xyzArray.shape)[0]
ntri = (triArray.shape)[0]
# make sure that triangle pointers are consistent
for itri in range(1, ntri + 1):
for iside in [0, 1, 2]:
jtri = triArray[itri - 1, iside]
if not ((jtri > 0) and (jtri <= npnt)):
self.fail("jtri not in (0 to %d)" % npnt)
gem.destroyDRep(myDRep)
gem.releaseModel(myModel)
gem.terminate(myContext)
def test_LoadModel(self):
myContext = gem.initialize()
gem.setAttribute(myContext, "CONTEXT", 0, "Modeler", "OpenCASCADE")
myModel = gem.loadModel(myContext, sample_file)
gem.setAttribute(myModel, "MODEL", 0, "s_attr", "model attribute")
gem.setAttribute(myModel, "MODEL", 0, "i_attrs", (2222222, 3333333, 4444444, 5555555))
gem.setAttribute(myModel, "MODEL", 0, "r_attrs", (0.2345678901234567, 0.3456789012345678))
copyModel = gem.copyModel(myModel)
gem.releaseModel(myModel)
server, filename, modeler, uptodate, myBReps, nparam, nbranch, nattr = gem.getModel(copyModel)
self.assertEqual(filename, sample_file)
self.assertEqual(modeler, "OpenCASCADE")
self.assertEqual(uptodate, -1)
self.assertEqual(nattr, 3)
aname, values = gem.getAttribute(copyModel, "MODEL", 0, 1)
self.assertEqual(aname, "s_attr")
self.assertEqual(values, "model attribute")
aname, values = gem.getAttribute(copyModel, "MODEL", 0, 2)
self.assertEqual(aname, "i_attrs")
self.assertEqual(values[0], 2222222)
self.assertEqual(values[1], 3333333)
self.assertEqual(values[2], 4444444)
self.assertEqual(values[3], 5555555)
aname, values = gem.getAttribute(copyModel, "MODEL", 0, 3)
self.assertEqual(aname, "r_attrs")
self.assertEqual(values[0], 0.2345678901234567)
self.assertEqual(values[1], 0.3456789012345678)
gem.releaseModel(copyModel)
gem.terminate(myContext)
def test_ContextAttributes(self):
myContext = gem.initialize()
gem.setAttribute(myContext, "CONTEXT", 0, "s_attr", "context attribute")
gem.setAttribute(myContext, "CONTEXT", 0, "i_attrs", (1111111, 2222222, 3333333, 4444444))
gem.setAttribute(myContext, "CONTEXT", 0, "r_attrs", (0.1234567890123456, 0.2345678901234567))
aindex, values = gem.retAttribute(myContext, "CONTEXT", 0, "s_attr")
self.assertEqual(aindex, 1)
self.assertEqual(values, "context attribute")
aindex, values = gem.retAttribute(myContext, "CONTEXT", 0, "i_attrs")
self.assertEqual(aindex, 2)
self.assertEqual(values[0], 1111111)
self.assertEqual(values[1], 2222222)
self.assertEqual(values[2], 3333333)
self.assertEqual(values[3], 4444444)
aindex, values = gem.retAttribute(myContext, "CONTEXT", 0, "r_attrs")
self.assertEqual(aindex, 3)
self.assertEqual(values[0], 0.1234567890123456)
self.assertEqual(values[1], 0.2345678901234567)
gem.terminate(myContext)
uptodate = foo[3] ;print "uptodate ->", uptodate
myBReps = foo[4] ;print "myBReps ->", myBReps
nparam = foo[5] ;print "nparam ->", nparam
nbranch = foo[6] ;print "nbranch ->", nbranch
nattr = foo[7]
for iattr in range(1, nattr+1):
dum = 0 ;print "iattr ->", iattr
foo = gem.getAttribute(newModel, "MODEL", 0, iattr)
aname = foo[0] ;print ". aname ->", aname
values = foo[1] ;print ". values ->", values
print "plotting newDRep..."
newDRep = gem.newDRep(newModel)
gem.tesselDRep(newDRep, 0, 0, 0, 0)
gem.plotDRep(newDRep);
gem.destroyDRep(newDRep)
print "releasing newModel"
gem.releaseModel(newModel);
# release the Model and terminate
print "releasing myModel"
gem.releaseModel(myModel)
# terminate GEM
print "terminating myContext"
gem.terminate(myContext)
print "SUCCESSFUL completion of 'test_gem.py'"