def main(): renderer = SphereRenderer(511) latlongPath = '/Users/paul/github/hdr/GraceCathedral/grace_latlong.pfm' writedir = '/Users/paul/github/hdr/cw2/' graceLatlong = imageIO.loadPFMFile( latlongPath ) sphereMapped_pfm = renderer.mapLatlong( graceLatlong ) imageIO.writePPMFile( latlongPath + 'graceSphere.pfm', sphereMapped_pfm ) sphereMapped_ppm = imageIO.toPPM( sphereMapped_pfm ) imageIO.writePPMFile( latlongPath + 'graceSphere.ppm', sphereMapped_ppm )
def __init__(self, filepath): self.filepath = filepath self.npData = loadPFMFile( filepath ) self.height = self.npData.shape[0] self.width = self.npData.shape[1]
def test_writePFM(self): d = np.random.uniform( size = (512, 1024, 3) ) imageIO.writePFMFile( testdir + 'writing.pfm', d ) loadedData = imageIO.loadPFMFile( testdir + 'writing.pfm' ) self.assertTrue( np.allclose( d, loadedData ) )
def test_writePPM(self): d = np.random.uniform( size = (512, 1024, 3) ) dppm = imageIO.toPPM(d) imageIO.writePFMFile( testdir + 'writing.ppm', dppm ) loadedData = imageIO.loadPFMFile( testdir + 'writing.ppm' ) self.assertTrue( np.array_equal( dppm, loadedData ) )
def test_readPFM(self): d = imageIO.loadPFMFile( self.pfmFilepath ) self.assertTrue( d.shape == (512, 1024, 3) ) self.assertTrue( np.average(d) >= 0.01 and np.average(d) <= 0.99 )