CSG/tests/CSGFoundryTest.py

0001 #!/usr/bin/env python
0002 import os, codecs, numpy as np
0003
0004 class Solid(object):
0005     def __init__(self, a, i):
0006         b = a.tobytes()
0007         self.i = i
0008         self.label = a[i,0].tobytes().decode("utf-8")  # py3?
0009         self.numPrim = a[i,1]
0010         self.primOffset = a[i,2]
0011         self.extent = a[i,3].view(np.float32)
0012
0013     def __repr__(self):
0014         return "Solid(%d) %10s numPrim:%3d primOffset:%4d extent:%10.4f " % ( self.i, self.label, self.numPrim, self.primOffset, self.extent )
0015
0016 class Foundry(object):
0017     def __init__(self, base):
0018         solid = np.load(os.path.join(base, "solid.npy"))
0019         prim = np.load(os.path.join(base, "prim.npy"))
0020         node = np.load(os.path.join(base, "node.npy"))
0021         plan = np.load(os.path.join(base, "plan.npy"))
0022         tran = np.load(os.path.join(base, "tran.npy"))
0023
0024         self.solid = solid
0025         self.prim = prim
0026         self.node = node
0027         self.plan = plan
0028         self.tran = tran
0029
0030
0031
0032 if __name__ == '__main__':
0033     fd = Foundry("/tmp/FoundryTest_")
0034
0035
0036
0037