def testTileSize(self): p = IECore.DeepPixel("A") p.addSample(1, [ 1, ]) writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) writer.parameters()['resolution'].setTypedValue(IECore.V2i(2, 2)) self.assertRaises(RuntimeError, IECore.curry(writer.writePixel, 0, 0, p)) writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) writer.parameters()['tileSize'].setTypedValue(IECore.V2i(2, 2)) writer.writePixel(0, 0, p) writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) writer.parameters()['resolution'].setTypedValue(IECore.V2i(127, 127)) writer.writePixel(0, 0, p) writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) writer.parameters()['resolution'].setTypedValue(IECore.V2i(127, 127)) writer.parameters()['tileSize'].setTypedValue(IECore.V2i(128, 128)) self.assertRaises(RuntimeError, IECore.curry(writer.writePixel, 0, 0, p)) writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) writer.parameters()['tileSize'].setTypedValue(IECore.V2i(30, 30)) self.assertRaises(RuntimeError, IECore.curry(writer.writePixel, 0, 0, p))
def testConstructor(self): self.failUnless("shw" in IECore.DeepImageWriter.supportedExtensions()) self.failUnless("shw" in IECore.DeepImageWriter.supportedExtensions( IECore.TypeId.DeepImageWriter)) writer = IECoreRI.SHWDeepImageWriter() self.failUnless(isinstance(writer, IECoreRI.SHWDeepImageWriter)) self.assertEqual(writer.typeId(), IECoreRI.SHWDeepImageWriter.staticTypeId()) writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) self.failUnless(isinstance(writer, IECoreRI.SHWDeepImageWriter)) self.assertEqual(writer.typeId(), IECoreRI.SHWDeepImageWriter.staticTypeId()) writer = IECore.DeepImageWriter.create(TestSHWDeepImageWriter.__output) self.failUnless(isinstance(writer, IECoreRI.SHWDeepImageWriter)) self.assertEqual(writer.typeId(), IECoreRI.SHWDeepImageWriter.staticTypeId())
def testDefaultWriter(self): writer = IECoreRI.SHWDeepImageWriter() self.assertEqual(writer.parameters()['fileName'].getTypedValue(), "") self.assertEqual(writer.parameters()['channelNames'].getValue(), IECore.StringVectorData(["A"])) self.assertEqual(writer.parameters()['resolution'].getTypedValue(), IECore.V2i(2048, 1556)) self.assertEqual(writer.parameters()['tileSize'].getTypedValue(), IECore.V2i(32, 32)) p = IECore.DeepPixel("RGBA") p.addSample(1, [1, 0, 0, 1]) self.assertRaises(RuntimeError, IECore.curry(writer.writePixel, 0, 0, p))
def testWriteSimplePixel(self): writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) writer.parameters()['channelNames'].setValue( IECore.StringVectorData(["A"])) writer.parameters()['resolution'].setTypedValue(IECore.V2i(2, 2)) writer.parameters()['tileSize'].setTypedValue(IECore.V2i(2, 2)) p = IECore.DeepPixel("A") p.addSample(1, (0.25, )) p2 = IECore.DeepPixel("A") p2.addSample(2, (0.5, )) p3 = IECore.DeepPixel("A") p3.addSample(1, (0.25, )) p3.addSample(2, (0.5, )) writer.writePixel(0, 0, p) writer.writePixel(0, 1, p2) writer.writePixel(1, 1, p3) del writer reader = IECoreRI.SHWDeepImageReader(TestSHWDeepImageWriter.__output) self.assertEqual(reader.dataWindow().size() + IECore.V2i(1), IECore.V2i(2, 2)) self.assertEqual(reader.channelNames(), IECore.StringVectorData(["A"])) rp = reader.readPixel(0, 0) self.assertEqual(rp.channelNames(), tuple(reader.channelNames())) self.assertEqual(rp.numSamples(), 1) self.assertEqual(rp.getDepth(0), 1) self.assertEqual(rp[0], (0.25, )) rp2 = reader.readPixel(0, 1) self.assertEqual(rp2.channelNames(), tuple(reader.channelNames())) self.assertEqual(rp2.numSamples(), 1) self.assertEqual(rp2.getDepth(0), 2) self.assertEqual(rp2[0], (0.5, )) rp3 = reader.readPixel(1, 1) self.assertEqual(rp3.channelNames(), tuple(reader.channelNames())) self.assertEqual(rp3.numSamples(), 2) self.assertEqual(rp3.getDepth(0), 1) self.assertEqual(rp3.getDepth(1), 2) self.assertEqual(rp3[0], (0.25, )) self.assertEqual(rp3[1], (0.5, )) self.assertEqual(rp3.composite(), [0.625]) self.failUnless(reader.readPixel(1, 0) is None)
def testParameters(self): writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) self.assertEqual(writer.parameters()['fileName'].getTypedValue(), TestSHWDeepImageWriter.__output) self.assertEqual(writer.parameters()['channelNames'].getValue(), IECore.StringVectorData(["A"])) self.assertEqual(writer.parameters()['resolution'].getTypedValue(), IECore.V2i(2048, 1556)) self.assertEqual(writer.parameters()['tileSize'].getTypedValue(), IECore.V2i(32, 32)) self.assertEqual( writer.parameters()['worldToCameraMatrix'].getTypedValue(), IECore.M44f()) self.assertEqual( writer.parameters()['worldToNDCMatrix'].getTypedValue(), IECore.M44f())
def testEmptyPixel(self): writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) p = IECore.DeepPixel("A") writer.writePixel(0, 0, p) writer.writePixel(0, 50, None) writer.writePixel(0, 100, p) p.addSample(1, [0.5]) writer.writePixel(0, 1, p) del writer reader = IECoreRI.SHWDeepImageReader(TestSHWDeepImageWriter.__output) self.failUnless(reader.readPixel(0, 0) is None) self.failUnless(reader.readPixel(0, 50) is None) self.failUnless(reader.readPixel(0, 100) is None) rp = reader.readPixel(0, 1) self.failUnless(isinstance(rp, IECore.DeepPixel)) self.assertEqual(rp[0], (0.5, ))
def testExtraChannels(self): writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) writer.parameters()['channelNames'].setValue( IECore.StringVectorData(list("RGBAST"))) p = IECore.DeepPixel("RGBAST", 2) p.addSample(1.0, [0.25, 0.5, 0.75, 0.5, 0.25, 0.5]) p.addSample(1.5, [1.0, 0.5, 0.25, 0.75, 0.5, 0.75]) writer.writePixel(0, 0, p) del writer reader = IECore.DeepImageReader.create(TestSHWDeepImageWriter.__output) self.assertEqual(reader.channelNames(), IECore.StringVectorData(["A"])) rp = reader.readPixel(0, 0) self.assertEqual(rp.channelNames(), ("A", )) self.assertEqual(rp.numSamples(), 2) self.assertEqual(rp.getDepth(0), 1.0) self.assertEqual(rp.getDepth(1), 1.5) self.assertEqual(rp[0], (0.5, )) self.assertEqual(rp[1], (0.75, ))
def testStrictChannels(self): writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) p = IECore.DeepPixel("RGBA") p.addSample(1, [1, 0, 0, 1]) self.assertRaises(RuntimeError, IECore.curry(writer.writePixel, 0, 0, p)) p = IECore.DeepPixel("RGB") p.addSample(1, [1, 0, 0]) self.assertRaises(RuntimeError, IECore.curry(writer.writePixel, 0, 1, p)) p = IECore.DeepPixel("A") p.addSample(1, [1]) writer.writePixel(0, 2, p) p = IECore.DeepPixel("RGBAST") p.addSample(1, [1, 0, 0, 1, 0, 1]) self.assertRaises(RuntimeError, IECore.curry(writer.writePixel, 0, 3, p))
def testReadWritePixel(self): reader = IECoreRI.SHWDeepImageReader(TestSHWDeepImageWriter.__shw) writer = IECoreRI.SHWDeepImageWriter(TestSHWDeepImageWriter.__output) writer.parameters()['resolution'].setTypedValue( reader.dataWindow().size() + IECore.V2i(1)) self.assertEqual(writer.parameters()['resolution'].getTypedValue(), IECore.V2i(512, 512)) self.assertEqual(writer.parameters()['channelNames'].getValue(), IECore.StringVectorData(["A"])) writer.parameters()['resolution'].setTypedValue(IECore.V2i(2, 2)) self.assertEqual(writer.parameters()['resolution'].getTypedValue(), IECore.V2i(2, 2)) writer.parameters()['tileSize'].setTypedValue(IECore.V2i(2, 2)) wToC = IECore.M44f(1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9, 10.1, 11.11, 12.12, 13.13, 14.14, 15.15, 16.16) cToS = IECore.M44f(10.1, 20.2, 30.3, 40.4, 50.5, 60.6, 7.7, 80.8, 90.9, 100.1, 110.11, 120.12, 130.13, 140.14, 150.15, 160.16) writer.parameters()['worldToCameraMatrix'].setTypedValue(wToC) writer.parameters()['worldToNDCMatrix'].setTypedValue(cToS) self.assertEqual( writer.parameters()['worldToCameraMatrix'].getTypedValue(), wToC) self.assertEqual( writer.parameters()['worldToNDCMatrix'].getTypedValue(), cToS) # hits ground plane only p = reader.readPixel(100, 100) self.assertEqual(p.channelNames(), ("A", )) self.assertEqual(p.numSamples(), 1) self.assertAlmostEqual(p.getDepth(0), 102.17636108, 6) self.assertAlmostEqual(p[0][0], 1.0, 6) # hits one box then ground plane p2 = reader.readPixel(256, 256) self.assertEqual(p2.channelNames(), tuple(reader.channelNames())) self.assertEqual(p2.numSamples(), 3) self.assertAlmostEqual(p2.getDepth(0), 72.6087493, 6) self.assertAlmostEqual(p2.getDepth(1), 77.7387313, 6) self.assertAlmostEqual(p2.getDepth(2), 85.6622314, 6) expected = (0.5, 0.5, 1.0) for i in range(0, len(expected)): self.assertEqual(p2[i][0], expected[i]) # hits 2 boxes then ground plane p3 = reader.readPixel(195, 225) self.assertEqual(p3.channelNames(), tuple(reader.channelNames())) self.assertEqual(p3.numSamples(), 5) self.assertAlmostEqual(p3.getDepth(0), 68.6177368, 6) self.assertAlmostEqual(p3.getDepth(1), 75.3023605, 6) self.assertAlmostEqual(p3.getDepth(2), 77.4083328, 6) self.assertAlmostEqual(p3.getDepth(3), 80.0680771, 6) self.assertAlmostEqual(p3.getDepth(4), 88.8455811, 6) expected = (0.5, 0.75, 0.5, 0.5, 1.0) for i in range(0, len(expected)): self.assertEqual(p3[i][0], expected[i]) writer.writePixel(0, 0, p) writer.writePixel(0, 1, p2) writer.writePixel(1, 1, p3) del writer reader = IECoreRI.SHWDeepImageReader(TestSHWDeepImageWriter.__output) self.assertEqual(reader.dataWindow().size() + IECore.V2i(1), IECore.V2i(2, 2)) self.assertEqual(reader.channelNames(), IECore.StringVectorData(["A"])) self.failUnless(reader.worldToCameraMatrix().equalWithAbsError( wToC, 1e-6)) self.failUnless(reader.worldToNDCMatrix().equalWithAbsError( cToS, 1e-6)) # hits ground plane only rp = reader.readPixel(0, 0) self.assertEqual(rp.channelNames(), ("A", )) self.assertEqual(rp.numSamples(), 1) self.assertAlmostEqual(rp.getDepth(0), 102.17636108, 4) self.assertAlmostEqual(rp[0][0], 1.0, 6) # hits one box then ground plane rp2 = reader.readPixel(0, 1) self.assertEqual(rp2.channelNames(), tuple(reader.channelNames())) self.assertEqual(rp2.numSamples(), 3) self.assertAlmostEqual(rp2.getDepth(0), 72.6087493, 4) self.assertAlmostEqual(rp2.getDepth(1), 77.7387313, 4) self.assertAlmostEqual(rp2.getDepth(2), 85.6622314, 4) expected = (0.5, 0.5, 1.0) for i in range(0, len(expected)): self.assertEqual(rp2[i][0], expected[i]) # hits 2 boxes then ground plane rp3 = reader.readPixel(1, 1) self.assertEqual(rp3.channelNames(), tuple(reader.channelNames())) self.assertEqual(rp3.numSamples(), 5) self.assertAlmostEqual(rp3.getDepth(0), 68.6177368, 6) self.assertAlmostEqual(rp3.getDepth(1), 75.3023605, 6) self.assertAlmostEqual(rp3.getDepth(2), 77.4083328, 6) self.assertAlmostEqual(rp3.getDepth(3), 80.0680771, 6) self.assertAlmostEqual(rp3.getDepth(4), 88.8455811, 6) expected = (0.5, 0.75, 0.5, 0.5, 1.0) for i in range(0, len(expected)): self.assertEqual(rp3[i][0], expected[i]) self.failUnless(reader.readPixel(1, 0) is None)