forked from mb003/HomeTest
-
Notifications
You must be signed in to change notification settings - Fork 0
/
paintmod.py
134 lines (107 loc) · 5.41 KB
/
paintmod.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
# -*- coding: utf-8 -*-
import pygame
import sys
from human import human
import time
class paintMod:
def __init__(self, screenSize = (1280, 720), backgroundColor = (255, 255, 255)):
self.floorColor = {
"dormitory" : (189, 33, 25),
"bathroom" : (41, 181, 206),
"diningRoom" : (115, 123, 181),
"outside" : (34, 139, 34)
}
self.wallColor = {
"dormitory" : (164, 8, 0),
"bathroom" : (16, 156, 181),
"diningRoom" : (90, 98, 156),
"outside" : (107, 142, 35)
}
self.isRunning = True
pygame.init()
self.ratio = 25
# self.manColor = (255, 0, 0)
# self.textFont = pygame.font.SysFont('arial', 22)
# self.textFont = pygame.font.Font(None, 20)
# self.text = self.textFont.render('', 1, (0, 0, 0) )
# self.textPos = text.get_rect()
self.screenSize = screenSize
self.backgroundColor = backgroundColor
# pygame初始化
# 创建一个窗口
self.screen = pygame.display.set_mode(self.screenSize, 0, 32)
pygame.display.set_caption('Draw rect and circle')
#背景填充
self.screen.fill(self.backgroundColor)
self.manImage = pygame.image.load('./pic/head_mini.png')
# self.temp = 0
def drawHousePicture(self, house, humanlist):
for pygameEvent in pygame.event.get():
# 按下关闭按钮，退出程序
if(pygameEvent.type == pygame.QUIT):
sys.exit()
if(pygameEvent.type == pygame.KEYDOWN):
self.isRunning = not self.isRunning
# 清屏
from light import lightSimulator
outDoorLight = lightSimulator()
self.screen.fill( self.changeColor( outDoorLight.getPercentage( house.getCurrentTime() ) , self.backgroundColor ) )
# 根据房间信息绘制图形
for tempRoom in house.getRoomList():
tempLeft = tempRoom.getLeft() * self.ratio + 100
tempTop = tempRoom.getTop() * self.ratio + 100
tempWidth = (tempRoom.getRight() - tempRoom.getLeft() + 1) * self.ratio
tempHeight = (tempRoom.getBottom() - tempRoom.getTop() + 1) * self.ratio
tempFloorColor = self.floorColor.get(tempRoom.getType())
tempFloorColor = self.changeColor(tempRoom.getLightPercentage(house.getCurrentTime()), tempFloorColor)
tempWallColor = self.wallColor.get(tempRoom.getType())
pygame.draw.rect(self.screen, tempFloorColor, [tempLeft, tempTop, tempWidth, tempHeight])
pygame.draw.rect(self.screen, tempWallColor, [tempLeft, tempTop, tempWidth, tempHeight], 2)
self.printOnScreen(str = tempRoom.getType(), pos = ( tempLeft+tempWidth/2, tempTop+tempHeight/2 ), color = (200, 200, 200))
# 绘制设备
for tempRoom in house.getRoomList():
for tempDevice in tempRoom.getDeviceList():
tempImage = tempDevice.getImage()
tempPos = tempImage.get_rect()
tempPos.centerx = tempDevice.getPosX()*self.ratio + 100
tempPos.centery = tempDevice.getPosY()*self.ratio + 100
self.screen.blit(tempImage, tempPos)
dayStr = time.strftime("%Y %m %d %A", time.localtime( house.getCurrentTime() ) )
self.printOnScreen(str = u'Day: %s' %dayStr, font = 'arial', pos = (40*self.ratio , 15), mode = 'LEFT_TOP', color = (150,150,150) )
self.printOnScreen(str = u'Time: %s' %house.getCurrentTimeStr(), font = 'arial', pos = (40*self.ratio , 35), mode = 'LEFT_TOP', color = (150,150,150) )
self.printOnScreen(str = u'Temp: %.2f' %house.getSimT().getCurrentTemperature(), font = 'arial', pos = (40*self.ratio,55), mode = 'LEFT_TOP', color = (150,150,150))
# 绘制人
for man in humanlist:
tempPos = self.manImage.get_rect()
tempPos.centerx = man.getPosX()*self.ratio + 100
tempPos.centery = man.getPosY()*self.ratio + 100
self.screen.blit(self.manImage, tempPos )
# pygame.draw.circle(self.screen, self.manColor, (int(self.temp), int(self.temp)), 5)
# self.temp = self.temp + 0.1
# print(man.getCurrentTimeStr(), man.getPos())
# self.screen.blint(text_screen)
# 重画屏幕
pygame.display.flip()
#time.sleep(0.00001)
# print('paint picture')
def printOnScreen(self, str = '', font = None, size = 20, color = (0, 0, 0), pos = (0, 0), mode = 'CENTER'):
textFont = pygame.font.SysFont(font, size)
text = textFont.render(str, 1, color )
textPos = text.get_rect()
if( mode == 'LEFT_TOP' ):
textPos.left = pos[0]
textPos.top = pos[1]
elif( mode == 'CENTER' ):
textPos.centerx = pos[0]
textPos.centery = pos[1]
else:
textPos.left = pos[0]
textPos.top = pos[1]
self.screen.blit(text, textPos)
def changeColor(self, percentage = 1.00, color = (255, 255, 255) ):
r = color[0] * percentage
g = color[1] * percentage
b = color[2] * percentage
return (r, g, b)
def isRun(self):
return self.isRunning