客户端.py

import pygame, sys
from pygame.locals import *
from math import sqrt
from functools import reduce
from copy import deepcopy

windowSurface = pygame.display.set_mode((480, 320))
pygame.display.set_caption('客户端')
GRAY = (197, 197, 197)
GREEN = (0, 255, 0)
BLACK = (0, 0, 0)
DARK_GRAY = (87, 84, 79)
WHITE = (255, 255, 255)
RED = (255,0,0)
pygame.init()
menu_font = pygame.font.Font("font/STFANGSO.ttf", 17)
menu_text = menu_font.render("客户端", True, BLACK)
clear_text = menu_font.render("清除", True, WHITE)
save_text = menu_font.render("写完", True, WHITE)
logo_text = menu_font.render("o(=•ェ•=)m", True, BLACK)

draw = False
brush_size = 2
brush_color = GREEN

menu_rect = pygame.Rect(0, 0, 100, 320)
screen_rect = pygame.Rect(100, 0, 380, 320)
clear_rect = pygame.Rect(5, 290, 60, 25)
save_rect = pygame.Rect(5, 260, 60, 25)
save_flag = False
file_number = 1

lst=[]
Lst=[]
ch='1'#input()
lbp=[]
breakPoint=True
f=open(ch+".dat")
comp=[]
for i in range(144):
    comp.append(float(f.readline()))
f.close()
print(comp)
while True:
    #Pygame 事件
    ex=False
    for event in pygame.event.get():
        if event.type == QUIT:
            pygame.quit()
            ex=True
            break
        if event.type == KEYDOWN:
            if event.key == K_ESCAPE:
                pygame.quit()
                ex=True
                break
        if event.type == MOUSEBUTTONDOWN:
            draw = True
        if event.type == MOUSEBUTTONUP:
            draw = False
    if ex : break
    #Draw
    mouse_pos = pygame.mouse.get_pos()
    if draw == True and mouse_pos[0] > 100:
        if not(lst) or mouse_pos!=lst[-1]:
            lst.append(mouse_pos)
            if breakPoint:lbp.append(len(lst))
            #pygame.draw.circle(windowSurface, brush_color, mouse_pos, brush_size)
            if not(breakPoint):pygame.draw.line(windowSurface, brush_color, lst[-2], lst[-1])
            save_flag = False
    #Clear
    if draw == True:
        if clear_rect.collidepoint(mouse_pos):
            lst=[]
            pygame.draw.rect(windowSurface, BLACK, screen_rect)
    
    pygame.draw.rect(windowSurface, GRAY, menu_rect)
    windowSurface.blit(menu_text, (10, 20))
    
    pygame.draw.line(windowSurface, BLACK, (0, 100), (100, 100))
    pygame.draw.line(windowSurface, BLACK, (0, 105), (100, 105))
    
    pygame.draw.rect(windowSurface, DARK_GRAY, clear_rect)
    windowSurface.blit(clear_text, (10, 290))


    pygame.draw.rect(windowSurface, DARK_GRAY, save_rect)
    windowSurface.blit(save_text, (10, 262))
    windowSurface.blit(logo_text, (10, 60))
    
    if draw == True and save_flag == False:
        if save_rect.collidepoint(mouse_pos) and lst:
            save_surface = pygame.Surface((380, 320))
            save_surface.blit(windowSurface, (0, 0), (100, 0, 380, 320))
            pygame.draw.rect(windowSurface, BLACK, screen_rect)
            ll=[lst[0]]
            i=1
            while i<len(lst):
                if i+1 not in lbp:
                    dist=round(sqrt((lst[i][0]-lst[i-1][0])**2+(lst[i][1]-lst[i-1][1])**2))
                    tr=[[lst[i-1][0]+(lst[i][0]-lst[i-1][0])/dist*j,lst[i-1][1]+(lst[i][1]-lst[i-1][1])/dist*j] for j in range(dist)]
                    ll.extend(tr)
                    ll.append(lst[i])
                i+=1

            llt=list(zip(*ll))
            #找重心
            cx,cy=sum(llt[0])/len(llt[0]),sum(llt[1])/len(llt[0])

            llt=[[llt[0][i]-cx for i in range(len(llt[0]))],[llt[1][i]-cy for i in range(len(llt[0]))]]
            
            maxLine=max(llt[0])
            minLine=min(llt[0])
            maxCol=max(llt[1])
            minCol=min(llt[1])
            ll=[[round(llt[0][i]/(maxLine-minLine)*300),round(llt[1][i]/(maxCol-minCol)*300)] for i in range(len(llt[0]))]
            #已处理
            #开始划分
            l=[]
            for j in range(12):
                l.append([])
                for k in range(12):
                    l[j].append(0)
            for k in range(len(ll)):
                x=ll[k][0]
                y=ll[k][1]
                #如果划分成9格
                l[(x+300)//50][(y+300)//50]+=1

            l=reduce(list.__add__,l)
            #print(l)
            print(sqrt(sum([(l[j]-comp[j])**2 for j in range(len(l))])))
            lst=[]
            
    pygame.display.update()
    breakPoint=not draw