PyDMXControl.design.parts._Pipe

Complexity

Total Complexity

2

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	Pipe.__init__()	0	5	1
A	Pipe.design_render()	0	10	1

"""
 *  PyDMXControl: A Python 3 module to control DMX using uDMX.
 *                Featuring fixture profiles, built-in effects and a web control panel.
 *  <https://github.com/MattIPv4/PyDMXControl/>
 *  Copyright (C) 2018 Matt Cowley (MattIPv4) (me@mattcowley.co.uk)
"""

import pygame

from ._Part import Part
from .._screen import Screen


class Pipe(Part):

    def __init__(self, size: int, x: int, y: int, rotation: int = 0):
        super().__init__()
        self.__rotation = rotation
        self.__length = size
        self.set_pos(x, y)

    def design_render(self, screen: Screen):
        # Create the surface
        surface = pygame.Surface((self.__length * screen.block_size, screen.block_size))
        surface.fill([80] * 3)

        # Rotate
        surface = pygame.transform.rotate(surface, self.__rotation)

        # Render
        screen.screen.blit(surface, (self._x * screen.block_size, self._y * screen.block_size))