-
Notifications
You must be signed in to change notification settings - Fork 4
/
train-UI.py
92 lines (81 loc) · 3.8 KB
/
train-UI.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
from functools import partial
import wx
import random
class MyFrame(wx.Frame):
def __init__(self, parent, title):
super(MyFrame, self).__init__(parent, title=title)
self.panel = wx.Panel(self)
self.coloredpan = wx.Panel(self.panel)
self.coloredpan.SetSize(1, 110, 300, 100)
self.currentR = random.randint(0, 255)
self.currentG = random.randint(0, 255)
self.currentB = random.randint(0, 255)
self.coloredpan.SetBackgroundColour(wx.Colour( self.currentR, self.currentG, self.currentB))
self.panelCtrl = wx.Panel(self.panel)
self.panel1 = wx.Panel(self.panel)
self.buttonValues={}
self.buttonValues["DARK"] = [0,0,0,0]
self.buttonValues["PURPLE"] = [0,0,0,1]
self.buttonValues["BLUE"] = [0,0,1,0]
self.buttonValues["GREEN"] = [0,0,1,1]
self.buttonValues["YELLOW"] = [0,1,0,0]
self.buttonValues["ORANGE"] = [0,1,0,1]
self.buttonValues["RED"] = [0,1,1,0]
self.buttonValues["WHITE"] = [0,1,1,1]
self.buttonValues["GREY"] = [1,0,0,0]
self.UI()
self.Centre()
def UI(self):
boxPanels= wx.BoxSizer(wx.VERTICAL)
new_button = wx.Button(self.panelCtrl,wx.ID_ANY,"NEW COLOR")
new_button.Bind(wx.EVT_BUTTON, self.generateNewColor)
dark_button = wx.Button(self.panel1, 3,"DARK" )
dark_button.Bind(wx.EVT_BUTTON, partial(self.choiceColor, button_label="DARK"))
purple_button = wx.Button(self.panel1, 3, "PURPLE")
purple_button.Bind(wx.EVT_BUTTON, partial(self.choiceColor, button_label="PURPLE"))
blue_button = wx.Button(self.panel1,3,"BLUE")
blue_button.Bind(wx.EVT_BUTTON, partial(self.choiceColor, button_label="BLUE"))
green_button = wx.Button(self.panel1, 3, "GREEN")
green_button.Bind(wx.EVT_BUTTON, partial(self.choiceColor, button_label="GREEN"))
yellow_button = wx.Button(self.panel1, 3, "YELLOW")
yellow_button.Bind(wx.EVT_BUTTON, partial(self.choiceColor, button_label="YELLOW"))
orange_button = wx.Button(self.panel1, 3, "ORANGE")
orange_button.Bind(wx.EVT_BUTTON, partial(self.choiceColor, button_label="ORANGE"))
red_button = wx.Button(self.panel1, wx.ID_ANY, "RED")
red_button.Bind(wx.EVT_BUTTON, partial(self.choiceColor, button_label="RED"))
white_button = wx.Button(self.panel1, 3, "WHITE")
white_button.Bind(wx.EVT_BUTTON, partial(self.choiceColor, button_label="WHITE"))
grey_button = wx.Button(self.panel1, 3, "GREY")
grey_button.Bind(wx.EVT_BUTTON, partial(self.choiceColor, button_label="GREY"))
boxcolors = wx.BoxSizer(wx.HORIZONTAL)
boxcolors.Add(dark_button, 0)
boxcolors.Add(purple_button, 0)
boxcolors.Add(blue_button, 0)
boxcolors.Add(green_button, 0)
boxcolors.Add(yellow_button, 0)
boxcolors.Add(orange_button, 0)
boxcolors.Add(red_button, 0)
boxcolors.Add(white_button, 0)
boxcolors.Add(grey_button, 0)
self.panel1.SetSizer(boxcolors);
boxPanels.Add(self.panelCtrl)
boxPanels.Add(self.panel1)
self.panel.SetSizer(boxPanels)
def generateNewColor(self,e):
print("CLICKED")
self.currentR = random.randint(0,255)
self.currentG = random.randint(0, 255)
self.currentB = random.randint(0, 255)
self.coloredpan.SetBackgroundColour(wx.Colour( self.currentR, self.currentG, self.currentB))
self.coloredpan.Refresh()
def choiceColor(self, Event, button_label):
print ("In OnButton:", button_label)
print (self.buttonValues[button_label])
self.generateNewColor(Event)
def main():
app = wx.App()
mf = MyFrame(None,"TRAINER NN COLOR RECOGNITION")
mf.Show()
app.MainLoop()
if __name__ == '__main__':
main()