def test_haldclut_graph(self): raise SkipTest() graph = Graph() img = Image.open(fate_suite("png1/lena-rgb24.png")) frame = VideoFrame.from_image(img) img_source = graph.add_buffer(frame) hald_img = Image.open("hald_7.png") hald_frame = VideoFrame.from_image(hald_img) hald_source = graph.add_buffer(hald_frame) hald_filter = graph.add("haldclut") sink = graph.add("buffersink") img_source.link(0, hald_filter, 0) hald_source.link(0, hald_filter, 1) hald_filter.link(0, sink, 0) graph.config() self.assertIs(img_source.outputs[0].linked_to, hald_filter.inputs[0]) self.assertIs(hald_source.outputs[0].linked_to, hald_filter.inputs[1]) self.assertIs(hald_filter.outputs[0].linked_to, sink.inputs[0]) hald_source.push(hald_frame) img_source.push(frame) frame = sink.pull() self.assertIsInstance(frame, VideoFrame) frame.to_image().save(self.sandboxed("filtered.png"))
def test_haldclut_graph(self): raise SkipTest() graph = Graph() img = Image.open(fate_suite('png1/lena-rgb24.png')) frame = VideoFrame.from_image(img) img_source = graph.add_buffer(frame) hald_img = Image.open('hald_7.png') hald_frame = VideoFrame.from_image(hald_img) hald_source = graph.add_buffer(hald_frame) try: hald_filter = graph.add('haldclut') except ValueError: # Not in Libav. raise SkipTest() sink = graph.add('buffersink') img_source.link(0, hald_filter, 0) hald_source.link(0, hald_filter, 1) hald_filter.link(0, sink, 0) graph.config() self.assertIs(img_source.outputs[0].linked_to, hald_filter.inputs[0]) self.assertIs(hald_source.outputs[0].linked_to, hald_filter.inputs[1]) self.assertIs(hald_filter.outputs[0].linked_to, sink.inputs[0]) hald_source.push(hald_frame) img_source.push(frame) frame = sink.pull() self.assertIsInstance(frame, VideoFrame) frame.to_image().save('sandbox/filtered.png')