class DataManagementScreen(AbstractDrawable):
def __init__(self,
x=None,
y=None,
w=None,
h=None,
vec2_pos=None,
size=None,
parent=None):
super(DataManagementScreen, self).__init__(x, y, w, h, vec2_pos, size)
self.radiobuttons = []
self.button_ok = Button(self.x + 325,
self.y + 450,
75,
75,
keycode=FONT_AWESOME.CHECK)
self.button_ok.SetTextIconColor(COLOR.GREEN)
self.button_cancel = Button(self.x + 500,
self.y + 450,
75,
75,
keycode=FONT_AWESOME.CLOSE,
padding=8)
self.button_cancel.SetTextIconColor(COLOR.RED)
self.main_rect = Rect(self.x,
self.y,
self.w,
self.h,
color=COLOR.BLACK,
width=1,
transparent=False)
self.PrepareCheckboxes()
self.text = None
self.font = None
self.text_load = None
self.text_save = None
self.selected = None
self.parent = parent
self.timer = 0
def IsInside(self, position):
if self.x <= position[0] and self.x + self.w >= position[
0] and self.y <= position[1] and self.y + self.h >= position[1]:
for elem in self.radiobuttons:
if elem.IsInside(position):
self.selected = elem
return True
if self.button_ok.IsInside(position):
self.selected = self.button_ok
return True
if self.button_cancel.IsInside(position):
self.selected = self.button_cancel
return True
return False
else:
return False
def DrawObject(self, screen):
self.main_rect.DrawObject(screen)
for elem in self.radiobuttons:
elem.DrawObject(screen)
screen.blit(self.text, (self.x + 415, self.y + 10))
self.button_ok.DrawObject(screen)
self.button_cancel.DrawObject(screen)
if self.timer:
self.timer -= 1
def PrepareCheckboxes(self):
for i in range(3):
for j in range(7):
pos_x = self.main_rect.x + 25 + j * 125
pos_y = self.main_rect.y + 75 + i * 125
checkbox = CheckboxRect(pos_x,
pos_y,
size=(100, 100),
index=int(i * 7 + j))
self.radiobuttons.append(checkbox)
def SetMode(self, mode):
self.timer = 0
self.mode = mode
if not self.font:
self.LoadFont()
if mode == "Save":
self.text = self.text_save
elif mode == "Load":
self.text = self.text_load
self.LoadAllImages()
self.DeselectAllRadioButtons()
def LoadFont(self):
font = os.path.join("Resources", "nimbus-roman-no9-l.regular.otf")
self.font = pygame.font.Font(font, 38)
self.text_save = self.font.render("Save", 1, COLOR.BLACK)
self.text_load = self.font.render("Load", 1, COLOR.BLACK)
def OnClick(self, event):
if isinstance(self.selected, CheckboxRect):
for elem in self.radiobuttons:
if elem is self.selected:
elem.SetSelected(True)
if self.timer:
self.ProcessData(elem.index)
self.parent.CloseDataManagementWindow()
self.timer = 60
else:
elem.SetSelected(False)
elif isinstance(self.selected, Button):
if self.selected is self.button_ok:
index = -1
for elem in self.radiobuttons:
if elem.GetSelected():
index = elem.index
if index > -1:
self.ProcessData(index)
else:
print "no elem selected"
else:
print "cancel"
self.parent.CloseDataManagementWindow()
def DeselectAllRadioButtons(self):
for elem in self.radiobuttons:
elem.SetSelected(False)
def LoadAllImages(self):
for elem in self.radiobuttons:
elem.LoadImage()
def ProcessData(self, index):
if self.mode == "Save":
self.parent.CloseDataManagementWindow()
self.parent.SaveData(index)
print "Saved", index
else:
self.parent.LoadData(index)
print "Loaded", index
