#!/usr/bin/python3
# SPDX-License-Identifier: GPL-2.0-or-later
# SPDX-FileCopyrightText: 2022 Renesas Electronics Corporation

import kmstest
import pykms

class PlaneAlphaTest(kmstest.KMSTest):
    """Test composition with multiple planes and alpha channels."""

    def handle_page_flip(self, frame, time):
        self.logger.log('Page flip complete')

    def find_pipeline(self):
        # Find a CRTC that has multiple planes with a connected connector
        for connector in self.output_connectors():
            # Skip disconnected connectors
            if not connector.connected():
                continue

            # Add the connector to the map
            for crtc in connector.get_possible_crtcs():
                planes = []
                for plane in self.card.planes:
                    if plane.supports_crtc(crtc) and plane != crtc.primary_plane:
                        planes.append(plane)

                if len(planes):
                    return crtc, connector, planes

        return None, None, None

    def main(self):
        self.start('composition with alpha control')

        crtc, connector, planes = self.find_pipeline()
        if crtc is None:
            self.skip('no suitable pipeline')
            return

        # Get the default mode
        try:
            mode = connector.get_default_mode()
        except KeyError:
            self.skip('no mode available')
            return

        self.logger.log(f'Testing connector {connector.fullname}, CRTC {crtc.id}, '
                        f'mode {mode.name} with {len(planes)} planes '
                        f'(P: {crtc.primary_plane.id}, O: {[plane.id for plane in planes]})')

        # Create a frame buffer
        fb = pykms.DumbFramebuffer(self.card, mode.hdisplay, mode.vdisplay, 'XR24')
        pykms.draw_test_pattern(fb)

        # Set the mode with a primary plane
        ret = self.atomic_crtc_mode_set(crtc, connector, mode, fb)
        if ret < 0:
            self.fail(f'atomic mode set failed with {ret}')
            return

        req = kmstest.AtomicRequest(self)
        req.add(crtc.primary_plane, 'alpha', '50%')
        ret = req.commit_sync(True)
        if ret < 0:
            self.fail(f'failed to set alpha for primary plane: {ret}')
            return

        self.run(3)

        # Add all other planes one by one
        alpha = 20
        offset = 100
        for plane in planes:
            source = kmstest.Rect(0, 0, fb.width, fb.height)
            destination = kmstest.Rect(offset, offset, fb.width, fb.height)
            ret = self.atomic_plane_set(plane, crtc, source, destination, fb, alpha=f'{alpha}%')
            if ret < 0:
                self.fail(f'atomic plane set failed with {ret}')
                break

            self.logger.log(f'Adding plane {plane.id}')
            self.run(1)

            if self.flips == 0:
                self.fail('No page flip registered')
                break

            alpha = min(alpha + 20, 100)
            offset += 50

        else:
            self.success()

        self.atomic_crtc_disable(crtc)


if __name__ == '__main__':
    PlaneAlphaTest().execute()