HYTOS

# coding: utf-8

""" This is rotate command module """

import os.path

import sys

from SymbolSvgItem import SymbolSvgItem

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

class RotateCommand(QUndoCommand):

    def __init__(self, scene, items, reverse=False, angles=None, parent=None):

        from SymbolSvgItem import SymbolSvgItem

        from EngineeringEqpDescTextItem import QEngineeringEqpDescTextItem

        super(RotateCommand, self).__init__(parent)

        self._reverse = reverse

        self._scene = scene

        self._items = [item for item in items if issubclass(type(item), SymbolSvgItem) or

                       type(item) is QEngineeringEqpDescTextItem]

        self._params = [item.rotation() for item in self._items] if angles is None else angles

        self._rotated = True if angles is not None  else False

    def undo(self):

        """undo"""

        import math

        for idx, item in enumerate(self._items):

            item.rotate(self._params[idx])

        self._scene.update()

        self._rotated = False

    def redo(self):

        """redo"""

        import math

        if not self._rotated:

            _params = []

            for idx, item in enumerate(self._items):

                angle = item.rotation()

                _params.append(angle)

                if self._reverse:

                    item.rotate((angle - 90) % 360 if angle in [0, 90, 180, 270, 360] else 0)

                else:

                    item.rotate((angle + 90) % 360 if angle in [0, 90, 180, 270, 360] else 0)

            self._params = _params

            self._scene.update()