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)
