def testSingleChannelImage(self):
r = GafferImage.ImageReader()
r["fileName"].setValue(
"${GAFFER_ROOT}/python/GafferImageTest/images/blurRange.exr")
self.assertEqual(r["out"]["channelNames"].getValue(),
IECore.StringVectorData(["R"]))
s = GafferImage.Shuffle()
s["in"].setInput(r["out"])
s["channels"].addChild(s.ChannelPlug("G", "R"))
s["channels"].addChild(s.ChannelPlug("B", "R"))
c1 = GafferImage.ColorSpace()
c1["in"].setInput(r["out"])
c1["inputSpace"].setValue("linear")
c1["outputSpace"].setValue("sRGB")
c2 = GafferImage.ColorSpace()
c2["in"].setInput(s["out"])
c2["inputSpace"].setValue("linear")
c2["outputSpace"].setValue("sRGB")
self.assertEqual(c2["out"].channelData("R", imath.V2i(0)),
c1["out"].channelData("R", imath.V2i(0)))
def testHashPassThrough( self ) : n = GafferImage.ImageReader() n["fileName"].setValue( self.fileName ) o = GafferImage.ColorSpace() o["in"].setInput( n["out"] ) self.assertImageHashesEqual( n["out"], o["out"] ) self.assertImagesEqual( n["out"], o["out"] ) o["inputSpace"].setValue( "linear" ) o["outputSpace"].setValue( "sRGB" ) self.assertNotEqual( n["out"].image(), o["out"].image() ) o["enabled"].setValue( False ) self.assertImageHashesEqual( n["out"], o["out"] ) self.assertImagesEqual( n["out"], o["out"] ) self.assertEqual( n["out"]['format'].hash(), o["out"]['format'].hash() ) self.assertEqual( n["out"]['dataWindow'].hash(), o["out"]['dataWindow'].hash() ) self.assertEqual( n["out"]["metadata"].getValue(), o["out"]["metadata"].getValue() ) self.assertEqual( n["out"]['channelNames'].hash(), o["out"]['channelNames'].hash() ) o["enabled"].setValue( True ) o["inputSpace"].setValue( "linear" ) o["outputSpace"].setValue( "linear" ) self.assertImageHashesEqual( n["out"], o["out"] ) self.assertImagesEqual( n["out"], o["out"] ) self.assertEqual( n["out"]['format'].hash(), o["out"]['format'].hash() ) self.assertEqual( n["out"]['dataWindow'].hash(), o["out"]['dataWindow'].hash() ) self.assertEqual( n["out"]["metadata"].getValue(), o["out"]["metadata"].getValue() ) self.assertEqual( n["out"]['channelNames'].hash(), o["out"]['channelNames'].hash() )
def testJpgRead( self ) : exrReader = GafferImage.OpenImageIOReader() exrReader["fileName"].setValue( self.circlesExrFileName ) jpgReader = GafferImage.OpenImageIOReader() jpgReader["fileName"].setValue( self.circlesJpgFileName ) jpgOCIO = GafferImage.ColorSpace() jpgOCIO["in"].setInput( jpgReader["out"] ) jpgOCIO["inputSpace"].setValue( "sRGB" ) jpgOCIO["outputSpace"].setValue( "linear" ) self.assertImagesEqual( exrReader["out"], jpgOCIO["out"], ignoreMetadata = True, maxDifference = 0.001 )
def testImageHashPassThrough(self):
i = GafferImage.ImageReader()
i["fileName"].setValue(self.fileName)
o = GafferImage.ColorSpace()
o["in"].setInput(i["out"])
self.assertEqual(i["out"].imageHash(), o["out"].imageHash())
o["inputSpace"].setValue("linear")
o["outputSpace"].setValue("sRGB")
self.assertNotEqual(i["out"].imageHash(), o["out"].imageHash())
def test(self):
n = GafferImage.ImageReader()
n["fileName"].setValue(self.fileName)
o = GafferImage.ColorSpace()
o["in"].setInput(n["out"])
self.assertEqual(n["out"].image(), o["out"].image())
o["inputSpace"].setValue("linear")
o["outputSpace"].setValue("sRGB")
self.assertNotEqual(n["out"].image(), o["out"].image())
def testPassThrough( self ) : i = GafferImage.ImageReader() i["fileName"].setValue( self.fileName ) o = GafferImage.ColorSpace() o["in"].setInput( i["out"] ) o["inputSpace"].setValue( "linear" ) o["outputSpace"].setValue( "sRGB" ) self.assertEqual( i["out"]["format"].hash(), o["out"]["format"].hash() ) self.assertEqual( i["out"]["dataWindow"].hash(), o["out"]["dataWindow"].hash() ) self.assertEqual( i["out"]["channelNames"].hash(), o["out"]["channelNames"].hash() ) self.assertEqual( i["out"]["format"].getValue(), o["out"]["format"].getValue() ) self.assertEqual( i["out"]["dataWindow"].getValue(), o["out"]["dataWindow"].getValue() ) self.assertEqual( i["out"]["channelNames"].getValue(), o["out"]["channelNames"].getValue() )
def testChannelsAreSeparate(self):
i = GafferImage.ImageReader()
i["fileName"].setValue(
os.path.expandvars(
"$GAFFER_ROOT/python/GafferImageTest/images/circles.exr"))
o = GafferImage.ColorSpace()
o["in"].setInput(i["out"])
o["inputSpace"].setValue("linear")
o["outputSpace"].setValue("sRGB")
self.assertNotEqual(o["out"].channelDataHash("R", imath.V2i(0)),
o["out"].channelDataHash("G", imath.V2i(0)))
self.assertNotEqual(o["out"].channelData("R", imath.V2i(0)),
o["out"].channelData("G", imath.V2i(0)))
def testContext(self):
scriptFileName = self.temporaryDirectory() + "/script.gfr"
contextImageFile = self.temporaryDirectory() + "/context.#.exr"
contextOverrideImageFile = self.temporaryDirectory(
) + "/context_override.#.exr"
s = Gaffer.ScriptNode()
s["reader"] = GafferImage.ImageReader()
s["reader"]["fileName"].setValue(self.fileName)
s["cs"] = GafferImage.ColorSpace()
s["cs"]["in"].setInput(s["reader"]["out"])
s["cs"]["inputSpace"].setValue("linear")
s["cs"]["outputSpace"].setValue("context")
s["writer"] = GafferImage.ImageWriter()
s["writer"]["fileName"].setValue(contextImageFile)
s["writer"]["in"].setInput(s["cs"]["out"])
s["writer"]["channels"].setValue("R G B A")
s["fileName"].setValue(scriptFileName)
s.save()
env = os.environ.copy()
env["OCIO"] = os.path.expandvars(
"$GAFFER_ROOT/python/GafferImageTest/openColorIO/context.ocio")
env["LUT"] = "srgb.spi1d"
env["CDL"] = "cineon.spi1d"
subprocess.check_call(
" ".join(["gaffer", "execute", scriptFileName, "-frames", "1"]),
shell=True,
stderr=subprocess.PIPE,
env=env,
)
i = GafferImage.ImageReader()
i["fileName"].setValue(
os.path.expandvars(
"$GAFFER_ROOT/python/GafferImageTest/images/checker_ocio_context.exr"
))
o = GafferImage.ImageReader()
o["fileName"].setValue(contextImageFile)
expected = i["out"]
context = o["out"]
# check against expected output
self.assertImagesEqual(expected, context, ignoreMetadata=True)
# override context
s["writer"]["fileName"].setValue(contextOverrideImageFile)
s["cs"]["context"].addChild(
Gaffer.NameValuePlug("LUT",
"cineon.spi1d",
True,
"LUT",
flags=Gaffer.Plug.Flags.Default
| Gaffer.Plug.Flags.Dynamic))
s["cs"]["context"].addChild(
Gaffer.NameValuePlug("CDL",
"rec709.spi1d",
True,
"CDL",
flags=Gaffer.Plug.Flags.Default
| Gaffer.Plug.Flags.Dynamic))
s.save()
subprocess.check_call(" ".join(
["gaffer", "execute", scriptFileName, "-frames", "1"]),
shell=True,
stderr=subprocess.PIPE,
env=env)
i = GafferImage.ImageReader()
i["fileName"].setValue(
os.path.expandvars(
"$GAFFER_ROOT/python/GafferImageTest/images/checker_ocio_context_override.exr"
))
o = GafferImage.ImageReader()
o["fileName"].setValue(contextOverrideImageFile)
expected = i["out"]
context = o["out"]
# check override produce expected output
self.assertImagesEqual(expected, context, ignoreMetadata=True)
