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 )
class TestSphereRenderer(unittest.TestCase): def setUp(self): self.renderer = SphereRenderer(511) def test_mapXYToPixel(self): xyCoord = [0,1] pixIdx = self.renderer.mapXYToPixel(xyCoord) self.assertTrue( np.allclose( [255, 0], pixIdx)) def test_ShadeSphereAtPoint(self): pt = [pi/2, 0] val = [3,4,5] self.renderer.shadeSphereAtPoint(pt, val) self.assertTrue(np.allclose(self.renderer.data[510,255,:], val, atol=tol))
def setUp(self): self.renderer = SphereRenderer(511)