python pygame study 5 实现一个画板

# -*- coding:UTF-8 -*-
#这个是按照教程写的,教程敲http://eyehere.net/2011/python-pygame-novice-professional-index/
#用来学Python挺好的,尤其是有了基础不知道该写点啥的孩砸
import pygame
from pygame.locals import *
from sys import exit
from math import sqrt

class Brush():
    def __init__(self,screen):
        self.screen=screen
        self.color=(255,255,255)
        self.size=1
        self.drawing=False
        self.style=False
        self.last_pos=None
        self.brush=pygame.image.load("images/brush.png")
        self.brush_now=self.brush.subsurface((0,0),(1,1))
    def get_style(self):
        return self.style

    def set_style(self,style):
        self.style=style

    def set_color(self,color):
        self.color=color
        for i in xrange(self.brush.get_width()):
            for j in xrange(self.brush.get_height()):
                self.brush.set_at((i,j),color+(self.brush.get_at((i,j)).a,))
                # self.brush.set_at((i, j),
                #                   color + (self.brush.get_at((i, j)).a,))

    def get_color(self):
        return  self.color

    def set_size(self,size):
        if size<1:
            size=0.5
        if size>32:
            size=32
        self.size=size
        self.brush_now = self.brush.subsurface((0, 0), (size * 2, size * 2))
        print ("set size to %f.2:"%size)

    def get_size(self):
        return self.size

    def start_drawing(self,pos):
        self.drawing=True
        self.last_pos=pos

    def end_drawing(self):
        self.drawing=False

    def get_current_brush(self):
        return self.brush_now


    def draw(self,pos):
        if self.drawing:
            for p in self._get_position(pos):
                if self.style:
                    self.screen.blit(self.brush_now,p)
                else:
                    pygame.draw.circle(self.screen,self.color,p,int(self.size))

    def _get_position(self,pos):
        points = [(self.last_pos[0], self.last_pos[1])]
        len_x=pos[0]-self.last_pos[0]
        len_y=pos[1]-self.last_pos[1]
        length=sqrt(len_x**2+len_y**2)
        step_x=len_x/length
        step_y=len_y/length
        for i in range(int(length)):
            points.append((points[-1][0]+step_x,points[-1][1]+step_y))
        points=map(lambda x:(int(x[0]+0.5),int(x[1]+0.5)),points)
        self.last_pos=pos
        return list(set(points))
class Menu():
    def __init__(self,screen):
        self.screen=screen
        self.brush=None
        self.colors=[
            (0xff, 0x00, 0xff), (0x80, 0x00, 0x80),
            (0x00, 0x00, 0xff), (0x00, 0x00, 0x80),
            (0x00, 0xff, 0xff), (0x00, 0x80, 0x80),
            (0x00, 0xff, 0x00), (0x00, 0x80, 0x00),
            (0xff, 0xff, 0x00), (0x80, 0x80, 0x00),
            (0xff, 0x00, 0x00), (0x80, 0x00, 0x00),
            (0xc0, 0xc0, 0xc0), (0xff, 0xff, 0xff),
            (0x00, 0x00, 0x00), (0x80, 0x80, 0x80),
        ]
        self.colors_rect=[]
        for (i,rgb) in enumerate(self.colors):
            rect=pygame.Rect(10 + i % 2 * 32, 254 + i / 2 * 32, 32, 32)
            self.colors_rect.append(rect)
        self.sizes_rect=[]
        self.sizes=[
            pygame.image.load("images/big.png"),
            pygame.image.load("images/small.png"),
        ]
        for (i,size) in enumerate(self.sizes):
            rect=pygame.Rect(10 + i * 32, 138, 32, 32)
            self.sizes_rect.append(rect)
        self.pens=[
            pygame.image.load("images/pen1.png"),
            pygame.image.load("images/pen2.png"),
        ]
        self.pens_rect=[]
        for (i,img) in enumerate(self.pens):
            rect=pygame.Rect(10, 10 + i * 64, 64, 64)
            self.pens_rect.append(rect)

    def set_brush(self,brush):
        self.brush=brush

    def draw(self):
        #draw pen
        for (i,img) in enumerate(self.pens):
            self.screen.blit(img,self.pens_rect[i].topleft)
        #draw big or small
        for (i,img) in enumerate(self.sizes):
            self.screen.blit(img,self.sizes_rect[i].topleft)
        #draw xianshi
        self.screen.fill((255,255,255),(10,180,64,64))
        pygame.draw.rect(self.screen, (0, 0, 0), (10, 180, 64, 64), 1)
        size=self.brush.get_size()
        x=10+32
        y=180+32
        if self.brush.get_style():
            new_x=x-size
            new_y=y-size
            self.screen.blit(self.brush.get_current_brush(),(new_x,new_y))
        else:
            pygame.draw.circle(self.screen,self.brush.get_color(),(x,y),int(size))
        #draw color
        for (i,rgb) in enumerate(self.colors):
            pygame.draw.rect(self.screen,rgb,self.colors_rect[i])


    def click_button(self,pos):
        for (i,rect) in enumerate(self.colors_rect):
            if rect.collidepoint(pos):
                self.brush.set_color(self.colors[i])
                return True
        size=self.brush.get_size
        for (i,rect) in enumerate(self.sizes_rect):
            if rect.collidepoint(pos):
                if i:  # i == 1, size down
                    self.brush.set_size(self.brush.get_size() - 0.5)
                else:
                    self.brush.set_size(self.brush.get_size() + 0.5)
                return True
        for (i,rect) in enumerate(self.pens_rect):
            if rect.collidepoint(pos):
                self.brush.set_style(bool(i))
                return True
        return False
class Painter():
    def __init__(self):
        pygame.init()
        pygame.display.set_caption("Painter")
        self.clock=pygame.time.Clock()
        self.screen=pygame.display.set_mode((800,600),0,32)
        self.brush=Brush(self.screen)
        self.menu=Menu(self.screen)
        self.menu.brush=self.brush

    def run(self):
        self.screen.fill((255, 255, 255))
        while True:
            self.clock.tick(30)
            for event in pygame.event.get():
                if event.type == QUIT:
                    exit()
                elif event.type == KEYDOWN:
                    # press esc to clear screen
                    if event.key == K_ESCAPE:
                        self.screen.fill((255, 255, 255))
                elif event.type == MOUSEBUTTONDOWN:
                    if (event.pos)[0] <= 74 and self.menu.click_button(event.pos):
                        pass
                    else:
                        self.brush.start_drawing(event.pos)
                elif event.type == MOUSEMOTION:
                    self.brush.draw(event.pos)

                elif event.type == MOUSEBUTTONUP:
                    self.brush.end_drawing()
            self.menu.draw()
            pygame.display.update()

painter=Painter()
painter.run()