def _save_gif(self):
# Get save file location
root = Tk()
fn = ''
if self._save_filename:
fn = '{}.gif'.format(self._save_filename)
filename = filedialog.asksaveasfilename(
initialfile=fn,
initialdir=(self._save_directory or os.getcwd()),
title='Select file save location',
filetypes=(('gif files', '*.gif'), ('all files', '*.*')))
root.destroy()
if filename == ():
self._saved_frames = []
return
else:
self._save_directory = os.path.dirname(filename)
imageio.mimwrite(filename,
self._saved_frames,
fps=30.0,
palettesize=128,
subrectangles=True)
self._saved_frames = []
def save(self):
"""
Save the output
"""
outpath = filedialog.asksaveasfilename()
print("Saving as:", outpath)
with open(outpath, "w+") as outfile:
# create list of field names
fieldnames = [
'Camera_id', 'Directory', 'Image_Name', 'Date', 'Processed',
'new_ROI', 'overall_B', 'overall_R', 'overall_G'
]
for stat in STATS:
for roi in ROI_TYPES:
fieldnames.append("_".join([roi, stat]))
writer = csv.DictWriter(outfile, fieldnames=fieldnames)
writer.writeheader()
for img_name in self.images:
# combine metadata and statistics
output = self.images[img_name].metadata.copy()
output.update(self.images[img_name].stats)
output['Processed'] = (img_name in self.done)
# write row
writer.writerow(output)
def _save_image(self):
# Get save file location
root = Tk()
fn = ''
if self._save_filename:
fn = '{}.png'.format(self._save_filename)
filename = filedialog.asksaveasfilename(
initialfile=fn,
initialdir=(self._save_directory or os.getcwd()),
title='Select file save location',
filetypes=(('png files', '*.png'), ('jpeg files', '*.jpg'),
('all files', '*.*')))
root.destroy()
if filename == ():
return
else:
self._save_directory = os.path.dirname(filename)
# Extract color image from frame buffer
width, height = self._size
glReadBuffer(GL_FRONT)
color_buf = glReadPixels(0, 0, width, height, GL_RGB, GL_UNSIGNED_BYTE)
# Re-format them into numpy arrays
color_im = np.frombuffer(color_buf, dtype=np.uint8).reshape(
(height, width, 3))
color_im = np.flip(color_im, axis=0)
imageio.imwrite(filename, color_im)
def asksaveasfilename(self):
"""Returns an opened file in write mode.
This time the dialog just returns a filename and the file is opened by your own code.
"""
# get filename
filename = tkFileDialog.asksaveasfilename(**self.file_opt)
# open file on your own
if filename:
return open(filename, 'w')
def saveas():
global text
t = text.get("1.0", "end-1c")
savelocation = tkFileDialog.asksaveasfilename()
file1 = open(savelocation, "w+")
file1.write(t)
file1.close()
def FileSave(self):
if sys.version_info[0] >= 3:
savefile = filedialog.asksaveasfilename(defaultextension=".ngc",title = "Select file",filetypes = (("gcode files","*.ngc"),("all files","*.*")))
else:
savefile = tkFileDialog.asksaveasfilename(defaultextension=".ngc",title = "Select file",filetypes = (("gcode files","*.ngc"),("all files","*.*")))
if savefile is None: # asksaveasfile return `None` if dialog closed with "cancel".
return
if savefile == '':
return
f = open(savefile, 'w')
for line in self.gcode:
f.write(line + '\n')
f.close() # `()` was missing.
def _get_save_filename(self, file_exts):
file_types = {
'png': ('png files', '*.png'),
'jpg': ('jpeg files', '*.jpg'),
'gif': ('gif files', '*.gif'),
'all': ('all files', '*'),
}
filetypes = [file_types[x] for x in file_exts]
try:
root = Tk()
save_dir = self.viewer_flags['save_directory']
if save_dir is None:
save_dir = os.getcwd()
filename = filedialog.asksaveasfilename(
initialdir=save_dir,
title='Select file save location',
filetypes=filetypes)
except Exception:
return None
root.destroy()
if filename == ():
return None
return filename
def _get_save_filename(self, file_exts):
file_types = {
"png": ("png files", "*.png"),
"jpg": ("jpeg files", "*.jpg"),
"gif": ("gif files", "*.gif"),
"all": ("all files", "*"),
}
filetypes = [file_types[x] for x in file_exts]
try:
root = Tk()
save_dir = self.viewer_flags["save_directory"]
if save_dir is None:
save_dir = os.getcwd()
filename = filedialog.asksaveasfilename(
initialdir=save_dir,
title="Select file save location",
filetypes=filetypes)
except Exception:
return None
root.destroy()
if filename == ():
return None
return filename