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 )
