def save_as(self, event):
if self.f_name_tb.GetValue() != '':
sets = self.img_panel.get_ranges()
if (len(sets[1]) + len(sets[2]) + len(sets[3]) != 6):
erwr = wx.MessageDialog(self,'Ranges are not properly selected.' 'Ranges Malformed', wx.OK)
erwr.ShowModal()
erwr.Destroy()
return
#write an entry to the replay log
replay_file = open('./replay.txt','a')
replay_file.write(self.img_file + '\n')
replay_file.write('1,' + str(sets[1][0][0])+','+str(sets[1][0][1])+','+
str(sets[1][1][0]-sets[1][0][0])+','+str(sets[1][1][1]-sets[1][0][1])+'\n')
replay_file.write('2,' + str(sets[2][0][0])+','+str(sets[2][0][1])+','+
str(sets[2][1][0]-sets[2][0][0])+','+str(sets[2][1][1]-sets[2][0][1])+'\n')
replay_file.write('3,' + str(sets[3][0][0])+','+str(sets[3][0][1])+','+
str(sets[3][1][0]-sets[3][0][0])+','+str(sets[3][1][1]-sets[3][0][1])+'\n')
replay_file.write(str(self.typebox.GetSelection()) + '\n')
replay_file.write(self.f_name_tb.GetValue() + '\n')
replay_file.close()
f_s = face_sampler()
left_eye_range = (sets[1][0][0],sets[1][0][1],sets[1][1][0]-sets[1][0][0],sets[1][1][1]-sets[1][0][1])
right_eye_range = (sets[2][0][0],sets[2][0][1],sets[2][1][0]-sets[2][0][0],sets[2][1][1]-sets[2][0][1])
mouth_range = (sets[3][0][0],sets[3][0][1],sets[3][1][0]-sets[3][0][0],sets[3][1][1]-sets[3][0][1])
p_s = f_s.gen_pixel_set(self.img,left_eye_range,right_eye_range,mouth_range)
f_s.write_to_csv(p_s,self.typebox.GetSelection(),self.f_name_tb.GetValue())
def __init__(self,replay_file):
self.f_s = face_sampler()
self.replays = []
rf = open(replay_file,'r')
while True:
rp_file = rf.readline().strip()
if rp_file == None or rp_file == '':
break
l_e = rf.readline().strip().split(',')
r_e = rf.readline().strip().split(',')
mouth = rf.readline().strip().split(',')
type = rf.readline().strip()
ofile = rf.readline().strip()
self.replays.append((rp_file,(int(l_e[1]),int(l_e[2]),int(l_e[3]),int(l_e[4])),
(int(r_e[1]),int(r_e[2]),int(r_e[3]),int(r_e[4])),
(int(mouth[1]),int(mouth[2]),int(mouth[3]),int(mouth[4])),
type,ofile))
def save_as(self, event):
if self.f_name_tb.GetValue() != '':
sets = self.img_panel.get_ranges()
if (len(sets[1]) + len(sets[2]) + len(sets[3]) != 6):
erwr = wx.MessageDialog(
self, 'Ranges are not properly selected.'
'Ranges Malformed', wx.OK)
erwr.ShowModal()
erwr.Destroy()
return
#write an entry to the replay log
replay_file = open('./replay.txt', 'a')
replay_file.write(self.img_file + '\n')
replay_file.write('1,' + str(sets[1][0][0]) + ',' +
str(sets[1][0][1]) + ',' +
str(sets[1][1][0] - sets[1][0][0]) + ',' +
str(sets[1][1][1] - sets[1][0][1]) + '\n')
replay_file.write('2,' + str(sets[2][0][0]) + ',' +
str(sets[2][0][1]) + ',' +
str(sets[2][1][0] - sets[2][0][0]) + ',' +
str(sets[2][1][1] - sets[2][0][1]) + '\n')
replay_file.write('3,' + str(sets[3][0][0]) + ',' +
str(sets[3][0][1]) + ',' +
str(sets[3][1][0] - sets[3][0][0]) + ',' +
str(sets[3][1][1] - sets[3][0][1]) + '\n')
replay_file.write(str(self.typebox.GetSelection()) + '\n')
replay_file.write(self.f_name_tb.GetValue() + '\n')
replay_file.close()
f_s = face_sampler()
left_eye_range = (sets[1][0][0], sets[1][0][1],
sets[1][1][0] - sets[1][0][0],
sets[1][1][1] - sets[1][0][1])
right_eye_range = (sets[2][0][0], sets[2][0][1],
sets[2][1][0] - sets[2][0][0],
sets[2][1][1] - sets[2][0][1])
mouth_range = (sets[3][0][0], sets[3][0][1],
sets[3][1][0] - sets[3][0][0],
sets[3][1][1] - sets[3][0][1])
p_s = f_s.gen_pixel_set(self.img, left_eye_range, right_eye_range,
mouth_range)
f_s.write_to_csv(p_s, self.typebox.GetSelection(),
self.f_name_tb.GetValue())
