Для события отслеживания мыши вы можете использовать оператор if event.type == pygame.MOUSEBUTTONDOWN:
.
Полный код ниже:
import pygame,sys,numpy
pygame.init()
display = (1500, 900)
screen = pygame.display.set_mode(display)
pygame.display.set_caption("Shape")
draw = False
size = (15,15)
run = True
shape = []
while run:
for event in pygame.event.get():
if event.type == pygame.QUIT:
run = False
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_ESCAPE:
run = False
if event.type == pygame.MOUSEBUTTONDOWN and event.button == 3: #Detect right-click; left-click is 1
draw = True
if event.type == pygame.MOUSEBUTTONUP and event.button == 3: #Detect release right-click
draw = False
#Here to save the file of 2d shape
shape = []
screen.fill((0,0,0))
#Draw the shape
if draw == True:
shape.append(pygame.mouse.get_pos())
for i in shape:
screen.fill((255,255,255), (i, size))
pygame.display.flip()
pygame.quit()
sys.exit()
Я думаю, что есть способ сохранить список как 2dфайл.Проверьте https://gis.stackexchange.com/questions/52705/how-to-write-shapely-geometries-to-shapefiles, и это может помочь вам.Просто добавьте процесс сохранения файла после комментария #Here to save the file of 2d shape
.
Я буду работать над частью файла сохранения, но это лучшее, что я могу получить на данный момент.