Exemple #1
0
    def testMultiPointsOneBuffer(self):
        """Test writing/reading several points at once.
        """
        if not numpy_available:
            return
        
        buffer = numpy.zeros(shape=(2,8), dtype=numpy.float32)
        buffer[0] = (1,2,3,1,0,0,7,8)
        buffer[1] = (-1,-2,-3,0,0,1,2,-8)
        ptc = pointcloud.open("tmp/pointcloud3.ptc", "w", self.libName, vars=[("float", "spam")], world2eye=mat4(1), world2ndc=mat4(1), format=(640,480,1))
        ptc.writeDataPoints(2, buffer)
        ptc.close()

        buf = numpy.zeros(shape=(3,8), dtype=numpy.float32)
        ptc = pointcloud.open("tmp/pointcloud3.ptc", "r", self.libName)
        n = ptc.readDataPoints(3, buf)
        self.assertEqual(2, n)
        # Round the normals
        buf[0][3] = round(buf[0][3],self.accuracy)
        buf[0][4] = round(buf[0][4],self.accuracy)
        buf[0][5] = round(buf[0][5],self.accuracy)
        buf[1][3] = round(buf[1][3],self.accuracy)
        buf[1][4] = round(buf[1][4],self.accuracy)
        buf[1][5] = round(buf[1][5],self.accuracy)
        self.assertEqual([1,2,3,1,0,0,7,8], list(list(buf)[0]))
        self.assertEqual([-1,-2,-3,0,0,1,2,-8], list(list(buf)[1]))
        n = ptc.readDataPoints(3, buf)
        self.assertEqual(0, n)
        ptc.close()
Exemple #2
0
def writePtc(fileName, libName):
    """Write a test point cloud file.
    """
    print ('Writing file "%s"...'%fileName) 
    vars = [("float", "spam"), ("vector", "dir")]
    ptc = pointcloud.open(fileName, "w", libName, vars=vars, world2eye=mat4(2), world2ndc=mat4(1), format=(320,240,1.333))
    ptc.writeDataPoint((0.5, 0.6, 0.7), (0,1,0), 0.2, {"spam":0.5})
    ptc.writeDataPoint((1, 2, 3), (1,0,0), 0.3, {"spam":0.2})
    ptc.writeDataPoint((-1, 2.5, 17), (0,1,0), 0.2, {"spam":0.5})
    ptc.close()
Exemple #3
0
def readPtc(fileName, libName):
    """Read a point cloud file and dump it to stdout.
    """
    print ('Reading file "%s"...'%fileName) 
    ptc = pointcloud.open(fileName, "r", libName)
    print ("variables: %s"%ptc.variables)
    print ("npoints: %s"%ptc.npoints)
    print ("datasize: %s"%ptc.datasize)
    print ("bbox: %s"%ptc.bbox)
    print ("format %sx%s, %s"%ptc.format)
    print ("world2eye %s"%ptc.world2eye)
    print ("world2ndc %s"%ptc.world2ndc)
    print ("")
    
    for i in range(ptc.npoints):
        pos,normal,radius,data = ptc.readDataPoint()
        print ("%s %s %s %s"%(pos,normal,radius,data))
    ptc.close()
Exemple #4
0
    def testSinglePoints(self):
        """Test writing/reading individual points.
        """
        world2eye = mat4(1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16)
        world2ndc = mat4(0.1,0.2,0.3,0.4,0.5,0.6,0.7,0.8,0.9,0.10,0.11,0.12,0.13,0.14,0.15,0.16)
        ptc = pointcloud.open("tmp/pointcloud.ptc", "w", self.libName, vars=[], world2eye=world2eye, world2ndc=world2ndc, format=(640,480,1))
        ptc.writeDataPoint((0.1,0.2,0.3), (0,1,0), 0.5, {})
        ptc.writeDataPoint((1,2,3), (1,0,0), 1.0, {})
        ptc.writeDataPoint((-1,0.17,42), (0,0,1), 1.5, {})
        ptc.close()
        
        ptc = pointcloud.open("tmp/pointcloud.ptc", "r", self.libName)
        self.assertEqual(3, ptc.npoints)
        self.assertEqual([], ptc.variables)
        self.assertEqual(0, ptc.datasize)
#        print ptc.bbox
        self.assertAlmostEqual(-1.0, ptc.bbox[0], 1)
        self.assertAlmostEqual(0.17, ptc.bbox[1], 1)
        self.assertAlmostEqual( 0.3, ptc.bbox[2], 1)
        self.assertAlmostEqual( 1.0, ptc.bbox[3], 1)
        self.assertAlmostEqual( 2.0, ptc.bbox[4], 1)
        self.assertAlmostEqual(42.0, ptc.bbox[5], 1)
        self.assertEqual(16, len(ptc.world2eye))
        self.assertAlmostEqual(1, ptc.world2eye[0], self.accuracy)
        self.assertAlmostEqual(2, ptc.world2eye[1], self.accuracy)
        self.assertAlmostEqual(3, ptc.world2eye[2], self.accuracy)
        self.assertAlmostEqual(4, ptc.world2eye[3], self.accuracy)
        self.assertAlmostEqual(5, ptc.world2eye[4], self.accuracy)
        self.assertAlmostEqual(6, ptc.world2eye[5], self.accuracy)
        self.assertAlmostEqual(7, ptc.world2eye[6], self.accuracy)
        self.assertAlmostEqual(8, ptc.world2eye[7], self.accuracy)
        self.assertAlmostEqual(9, ptc.world2eye[8], self.accuracy)
        self.assertAlmostEqual(10, ptc.world2eye[9], self.accuracy)
        self.assertAlmostEqual(11, ptc.world2eye[10], self.accuracy)
        self.assertAlmostEqual(12, ptc.world2eye[11], self.accuracy)
        self.assertAlmostEqual(13, ptc.world2eye[12], self.accuracy)
        self.assertAlmostEqual(14, ptc.world2eye[13], self.accuracy)
        self.assertAlmostEqual(15, ptc.world2eye[14], self.accuracy)
        self.assertAlmostEqual(16, ptc.world2eye[15], self.accuracy)
        self.assertEqual(16, len(ptc.world2ndc))
        self.assertAlmostEqual(0.1, ptc.world2ndc[0], self.accuracy)
        self.assertAlmostEqual(0.2, ptc.world2ndc[1], self.accuracy)
        self.assertAlmostEqual(0.3, ptc.world2ndc[2], self.accuracy)
        self.assertAlmostEqual(0.4, ptc.world2ndc[3], self.accuracy)
        self.assertAlmostEqual(0.5, ptc.world2ndc[4], self.accuracy)
        self.assertAlmostEqual(0.6, ptc.world2ndc[5], self.accuracy)
        self.assertAlmostEqual(0.7, ptc.world2ndc[6], self.accuracy)
        self.assertAlmostEqual(0.8, ptc.world2ndc[7], self.accuracy)
        self.assertAlmostEqual(0.9, ptc.world2ndc[8], self.accuracy)
        self.assertAlmostEqual(0.10, ptc.world2ndc[9], self.accuracy)
        self.assertAlmostEqual(0.11, ptc.world2ndc[10], self.accuracy)
        self.assertAlmostEqual(0.12, ptc.world2ndc[11], self.accuracy)
        self.assertAlmostEqual(0.13, ptc.world2ndc[12], self.accuracy)
        self.assertAlmostEqual(0.14, ptc.world2ndc[13], self.accuracy)
        self.assertAlmostEqual(0.15, ptc.world2ndc[14], self.accuracy)
        self.assertAlmostEqual(0.16, ptc.world2ndc[15], self.accuracy)
        self.assertEqual((640.0, 480.0, 1.0), ptc.format)

        pos,norm,rad,data = ptc.readDataPoint()
        self.assertAlmostEqual(0.1, pos[0], self.accuracy)
        self.assertAlmostEqual(0.2, pos[1], self.accuracy)
        self.assertAlmostEqual(0.3, pos[2], self.accuracy)
        self.assertAlmostEqual(0, norm[0], self.accuracy)
        self.assertAlmostEqual(1, norm[1], self.accuracy)
        self.assertAlmostEqual(0, norm[2], self.accuracy)
        self.assertAlmostEqual(0.5, rad, self.accuracy)
        self.assertEqual({}, data)

        pos,norm,rad,data = ptc.readDataPoint()
        self.assertAlmostEqual(1.0, pos[0], self.accuracy)
        self.assertAlmostEqual(2.0, pos[1], self.accuracy)
        self.assertAlmostEqual(3.0, pos[2], self.accuracy)
        self.assertAlmostEqual(1, norm[0], self.accuracy)
        self.assertAlmostEqual(0, norm[1], self.accuracy)
        self.assertAlmostEqual(0, norm[2], self.accuracy)
        self.assertAlmostEqual(1.0, rad, self.accuracy)
        self.assertEqual({}, data)

        pos,norm,rad,data = ptc.readDataPoint()
        self.assertAlmostEqual(-1.0, pos[0], self.accuracy)
        self.assertAlmostEqual(0.17, pos[1], self.accuracy)
        self.assertAlmostEqual(42.0, pos[2], self.accuracy)
        self.assertAlmostEqual(0, norm[0], self.accuracy)
        self.assertAlmostEqual(0, norm[1], self.accuracy)
        self.assertAlmostEqual(1, norm[2], self.accuracy)
        self.assertAlmostEqual(1.5, rad, self.accuracy)
        self.assertEqual({}, data)

        self.assertRaises(EOFError, lambda: ptc.readDataPoint())
        
        ptc.close()
Exemple #5
0
    def testMultiPoints(self):
        """Test writing/reading several points at once.
        """
        pnts = (6*ctypes.c_float)(0.4, 0.8, 1.0,  0.9, 0.7, 0.6)
        norms = (6*ctypes.c_float)(1,0,0,  0,0,1)
        rads = (2*ctypes.c_float)(0.4, 0.5)
        data = (8*ctypes.c_float)(12.0, 1,2,3,   42.0, -1,-2,-3)
        ptc = pointcloud.open("tmp/pointcloud2.ptc", "w", self.libName, vars=[("float", "fspam"), ("vector", "vspam")], world2eye=mat4(1), world2ndc=mat4(1), format=(640,480,1))
        ptc.writeDataPoints(2, (pnts,norms,rads,data))
        ptc.close()
        
        ptc = pointcloud.open("tmp/pointcloud2.ptc", "r", self.libName)
        self.assertEqual(2, ptc.npoints)
        self.assertEqual([("float","fspam"), ("vector","vspam")], ptc.variables)
        self.assertEqual(4, ptc.datasize)
        ps = (6*ctypes.c_float)()
        ns = (6*ctypes.c_float)()
        rs = (2*ctypes.c_float)()
        ds = (8*ctypes.c_float)()
        ptc.readDataPoints(2, (ps,ns,rs,ds))
        self.assertAlmostEqual(0.4, ps[0], self.accuracy)
        self.assertAlmostEqual(0.8, ps[1], self.accuracy)
        self.assertAlmostEqual(1.0, ps[2], self.accuracy)
        self.assertAlmostEqual(0.9, ps[3], self.accuracy)
        self.assertAlmostEqual(0.7, ps[4], self.accuracy)
        self.assertAlmostEqual(0.6, ps[5], self.accuracy)
        self.assertAlmostEqual(1, ns[0], self.accuracy)
        self.assertAlmostEqual(0, ns[1], self.accuracy)
        self.assertAlmostEqual(0, ns[2], self.accuracy)
        self.assertAlmostEqual(0, ns[3], self.accuracy)
        self.assertAlmostEqual(0, ns[4], self.accuracy)
        self.assertAlmostEqual(1, ns[5], self.accuracy)
        self.assertAlmostEqual(0.4, rs[0], self.accuracy)
        self.assertAlmostEqual(0.5, rs[1], self.accuracy)
        self.assertAlmostEqual(12.0, ds[0], self.accuracy)
        self.assertAlmostEqual(1, ds[1], self.accuracy)
        self.assertAlmostEqual(2, ds[2], self.accuracy)
        self.assertAlmostEqual(3, ds[3], self.accuracy)
        self.assertAlmostEqual(42.0, ds[4], self.accuracy)
        self.assertAlmostEqual(-1, ds[5], self.accuracy)
        self.assertAlmostEqual(-2, ds[6], self.accuracy)
        self.assertAlmostEqual(-3, ds[7], self.accuracy)
        ptc.close()

        ptc = pointcloud.open("tmp/pointcloud2.ptc", "r", self.libName)
        pos,norm,rad,data = ptc.readDataPoint()
        self.assertAlmostEqual(0.4, pos[0], self.accuracy)
        self.assertAlmostEqual(0.8, pos[1], self.accuracy)
        self.assertAlmostEqual(1.0, pos[2], self.accuracy)
        self.assertAlmostEqual(1, norm[0], self.accuracy)
        self.assertAlmostEqual(0, norm[1], self.accuracy)
        self.assertAlmostEqual(0, norm[2], self.accuracy)
        self.assertAlmostEqual(0.4, rad, self.accuracy)
        self.assertAlmostEqual(12, data["fspam"], self.accuracy)
        self.assertAlmostEqual(1, data["vspam"][0], self.accuracy)
        self.assertAlmostEqual(2, data["vspam"][1], self.accuracy)
        self.assertAlmostEqual(3, data["vspam"][2], self.accuracy)
        ptc.close()
        
        ptc = pointcloud.open("tmp/pointcloud2.ptc", "r", self.libName)
        for buf in ptc.iterBatches(10, numpyArray=False, combinedBuffer=True):
            self.assertEqual(22, len(buf))
            self.assertAlmostEqual(0.4, buf[0], self.accuracy)
            self.assertAlmostEqual(0.8, buf[1], self.accuracy)
            self.assertAlmostEqual(1.0, buf[2], self.accuracy)
            self.assertAlmostEqual(1, buf[3], self.accuracy)
            self.assertAlmostEqual(0, buf[4], self.accuracy)
            self.assertAlmostEqual(0, buf[5], self.accuracy)
            self.assertAlmostEqual(0.4, buf[6], self.accuracy)
            self.assertAlmostEqual(12, buf[7], self.accuracy)
            self.assertAlmostEqual(1, buf[8], self.accuracy)
            self.assertAlmostEqual(2, buf[9], self.accuracy)
            self.assertAlmostEqual(3, buf[10], self.accuracy)
        ptc.close()