import unittest

from common import get_context


class TestCase(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.ctx = get_context()

    def test_1(self):
        rbo1 = self.ctx.renderbuffer((4, 4), dtype='f1')
        rbo2 = self.ctx.renderbuffer((4, 4))
        rbo3 = self.ctx.renderbuffer((4, 4))
        fbo1 = self.ctx.framebuffer(rbo1)
        fbo2 = self.ctx.framebuffer(rbo2)
        fbo3 = self.ctx.framebuffer(rbo3)
        fbo4 = self.ctx.framebuffer([rbo1, rbo2, rbo3])

        fbo4.clear(0.0, 1.0, 0.0, 1.0)

        pixels1 = fbo1.read(components=4, dtype='f1')
        self.assertEqual(pixels1, b'\x00\xff\x00\xff' * 16)

        pixels2 = fbo2.read(components=4, dtype='f1')
        self.assertEqual(pixels2, b'\x00\xff\x00\xff' * 16)

        pixels3 = fbo3.read(components=4, dtype='f1')
        self.assertEqual(pixels3, b'\x00\xff\x00\xff' * 16)

    def test_2(self):
        rbo1 = self.ctx.renderbuffer((4, 4), dtype='f1')
        rbo2 = self.ctx.renderbuffer((4, 4))
        rbo3 = self.ctx.renderbuffer((4, 4))
        fbo1 = self.ctx.framebuffer(rbo1)
        fbo2 = self.ctx.framebuffer(rbo2)
        fbo3 = self.ctx.framebuffer(rbo3)
        fbo4 = self.ctx.framebuffer([rbo1, rbo2, rbo3])

        fbo1.clear(1.0, 0.0, 0.0, 1.0)
        fbo2.clear(0.0, 1.0, 0.0, 1.0)
        fbo3.clear(0.0, 0.0, 1.0, 1.0)

        pixels1 = fbo4.read(attachment=0, components=4, dtype='f1')
        self.assertEqual(pixels1, b'\xff\x00\x00\xff' * 16)

        pixels2 = fbo4.read(attachment=1, components=4, dtype='f1')
        self.assertEqual(pixels2, b'\x00\xff\x00\xff' * 16)

        pixels3 = fbo4.read(attachment=2, components=4, dtype='f1')
        self.assertEqual(pixels3, b'\x00\x00\xff\xff' * 16)

    def test_3(self):
        rbo1 = self.ctx.renderbuffer((4, 4), dtype='f1')
        rbo2 = self.ctx.renderbuffer((4, 4))
        rbo3 = self.ctx.renderbuffer((4, 4))
        fbo1 = self.ctx.framebuffer(rbo1)
        fbo2 = self.ctx.framebuffer(rbo2)
        fbo3 = self.ctx.framebuffer(rbo3)
        fbo4 = self.ctx.framebuffer([rbo1, rbo2, rbo3])

        fbo1.clear(1.0, 0.0, 0.0, 1.0)
        fbo2.clear(0.0, 1.0, 0.0, 1.0)
        fbo3.clear(0.0, 0.0, 1.0, 1.0)

        pixels1 = bytearray(64)
        fbo4.read_into(pixels1, attachment=0, components=4, dtype='f1')
        self.assertEqual(bytes(pixels1), b'\xff\x00\x00\xff' * 16)

        pixels2 = bytearray(64)
        fbo4.read_into(pixels2, attachment=1, components=4, dtype='f1')
        self.assertEqual(bytes(pixels2), b'\x00\xff\x00\xff' * 16)

        pixels3 = bytearray(64)
        fbo4.read_into(pixels3, attachment=2, components=4, dtype='f1')
        self.assertEqual(bytes(pixels3), b'\x00\x00\xff\xff' * 16)


if __name__ == '__main__':
    unittest.main()