from PyQt5.QtWidgets import QWidget
from PyQt5.Qt import QPixmap, QPainter, QPoint, QPen, QColor, QSize
from PyQt5.QtCore import Qt


class PaintBoard(QWidget):
    def __init__(self, Parent=None):
        super().__init__(Parent)

        # 先初始化数据，再初始化界面
        self.__InitData()
        self.__InitView()


    def __InitData(self):
        # 画板尺寸
        self.__size = QSize(560, 560)

        # 新建QPixmap作为画板，尺寸为__size
        self.__board = QPixmap(self.__size)

        # 画板背景填充色（默认为白色）
        self.__board.fill(Qt.white)

        # 画板内容（默认为空）
        self.__IsEmpty = True

        # 橡皮擦模式（默认关闭）
        self.EraserMode = False

        # 上一次鼠标位置
        self.__lastPos = QPoint(0, 0)

        # 当前的鼠标位置
        self.__currentPos = QPoint(0, 0)

        # 新建绘图工具
        self.__painter = QPainter()

        # 画笔粗细（默认为 8）
        self.__thickness = 20

        # 画笔颜色（默认为黑色）
        self.__penColor = QColor("black")

        # 获取颜色列表
        self.__colorList = QColor.colorNames()


    def __InitView(self):
        # 设置界面的尺寸
        self.setFixedSize(self.__size)


    def Clear(self):
        # 清空画板
        self.__board.fill(Qt.white)
        self.update()
        self.__IsEmpty = True


    def ChangePenColor(self, color="black"):
        # 改变画笔颜色
        self.__penColor = QColor(color)


    def ChangePenThickness(self, thickness=20):
        # 改变画笔粗细
        self.__thickness = thickness


    def IsEmpty(self):
        # 返回画板是否为空
        return self.__IsEmpty


    def SetContentFromQPixmap(self, pixmap):
        # 设置画板内容为给定的QPixmap
        if pixmap.size() != self.__size:
            # 如果传入的pixmap尺寸与画板尺寸不一致，则调整其尺寸
            self.__board = pixmap.scaled(self.__size)
        else:
            self.__board = pixmap
        self.__IsEmpty = False  # 画板不再为空
        self.update()  # 更新显示


    def GetContentAsQImage(self):
        # 获取画板内容（返回QImage）
        image = self.__board.toImage()
        return image


    def paintEvent(self, paintEvent):
        # 绘图事件
        # 绘图时必须使用QPainter的实例，此处为__painter
        # 绘图在begin()函数与end()函数间进行
        # begin(param)的参数要指定绘图设备，即把图画在哪里
        # drawPixmap用于绘制QPixmap类型的对象
        self.__painter.begin(self)

        # 0,0为绘图的左上角起点的坐标，__board即要绘制的图
        self.__painter.drawPixmap(0, 0, self.__board)
        self.__painter.end()


    def mousePressEvent(self, mouseEvent):
        # 鼠标按下时，获取鼠标的当前位置保存为上一次位置
        self.__currentPos = mouseEvent.pos()
        self.__lastPos = self.__currentPos


    def mouseMoveEvent(self, mouseEvent):
        # 鼠标移动时，更新当前位置，并在上一个位置和当前位置间画线
        self.__currentPos = mouseEvent.pos()
        self.__painter.begin(self.__board)

        if self.EraserMode == False:
            # 非橡皮擦模式
            self.__painter.setPen(QPen(self.__penColor, self.__thickness))  # 设置画笔颜色，粗细
        else:
            # 橡皮擦模式下画笔为纯白色，粗细不变
            self.__painter.setPen(QPen(Qt.white, self.__thickness))

        # 画线
        self.__painter.drawLine(self.__lastPos, self.__currentPos)
        self.__painter.end()
        self.__lastPos = self.__currentPos

        self.update()  # 更新显示


    def mouseReleaseEvent(self, mouseEvent):
        self.__IsEmpty = False  # 画板不再为空