def saveas_callback(self):
full_filename = save_filename(caption="Choose base pathname",
directory="",
parent=self.parent)
self.saveas_filename = full_filename.split('.')[0]
self.cam.filename_base = self.saveas_filename
self.saveas_lbl.setText(self.saveas_filename)
def save_figure(self,
filename='*.png',
initialdir='~ppzmis/Documents',
dpi=80):
if filename == '*.png':
filename = fd.save_filename(caption='select filename',
file_filter='*.png;;*.jpg;;*.tiff')
self.fig.savefig(filename, dpi=dpi)
def save_config(self, filename=None, parent=None):
if filename is None:
filename = save_filename(directory=self.cam_config_dir,
file_filter='*.ccf',
parent=parent)
save_dict_to_file(filename, self.cam_dict)
SISO.Fg_saveConfig(self.fg, filename[:-3] + 'mcf')
print('config saved')
def single_pic_img(self, filename=None, show_pic=False):
if filename == None:
filename = save_filename()
'''writes single image to picture file '''
ret, frame = self.cam.read()
cv2.imwrite(filename, frame)
if show_pic:
cv2.imshow('frame', frame)
def save_mesh_constraints(known_displacements,known_forces,filename=None):
if filename is None:
filename = save_filename(directory='/media/ppzmis/data/FEM/', file_filter='*.h5')
op_filename = filename[:-3] + 'constraints.h5'
hf = h5py.File(op_filename, 'w')
hf.create_dataset('displacements', data=known_displacements)
hf.create_dataset('forces', data=known_forces)
hf.close()
return op_filename
def write_to_vid(self, filename=None, fps=30.0):
if filename is None:
filename = save_filename()
self.vid = video.WriteVideo(filename=filename,
frame_size=(self.frame_size[1],
self.frame_size[0],
self.frame_size[2]),
fps=self.fps)
def save_cam_settings(self, filename=None):
if filename is None:
filename = save_filename(file_filter='*.camlog')
self.get_cam_props(show=True)
settings = (self.brightness, self.contrast, self.gain, self.saturation,
self.hue, self.exposure)
with open(filename, 'w') as settings_file:
for item in settings:
settings_file.write("%s\n" % item)
def lut_file(lut_array, filename=None):
#lut_np array must be 1024 long with numbers between 0 and 255
if filename is None:
filename = save_filename(directory='/opt/Microscope/ConfigFiles/',
file_filter='*.lut')
with open(filename, mode="w") as fout:
fout.writelines('#N\n')
fout.writelines('#l\n')
for i in range(np.size(lut_array)):
fout.writelines(str(int(lut_array[i])) + '\n')
fout.writelines('##quit')
def save_mesh_to_file(node_pts,elements,filename=None, vals=None):
#http://christopherlovell.co.uk/blog/2016/04/27/h5py-intro.html
if filename is None:
filename = save_filename(directory='/media/ppzmis/data/FEM/', file_filter='*.h5')
hf = h5py.File(filename, 'w')
hf.create_dataset('node_pts', data=str(node_pts))
hf.create_dataset('elements', data=str(elements))
if vals != None:
hf.create_dataset('vals', data=vals)
hf.close()
return filename
def __init__(self,
cam_config_dir='/opt/Microscope/ConfigFiles/',
filename=None,
ccf_file=None):
self.cam_config_dir = cam_config_dir
self.fg = SISO.Fg_InitConfig(cam_config_dir + 'current.mcf', 0)
self.camset = CameraSettings(self.fg,
cam_config_dir,
ccf_file=ccf_file)
if filename is None:
filename = save_filename(caption='Select filename base',
directory='/home/ppzmis/Videos/')
self.filename_base = filename.split('.')[0]
print(self.filename_base)
self.autosave = False
def __init__(self, filename=None, frame_size=None, frame=None,
write_frame=False, fps=30.0, codec='XVID'):
#codec_list = ['XVID','mp4v']
codec_code = list(codec)
#Check inputs for errors
if (frame_size is None) and (frame is None):
raise ArgumentsMissing(('frame_size','frame'))
if (frame_size is not None) and (frame is not None):
if frame_size != np.shape(frame):
raise ImageShapeError(frame,frame_size)
#Optional arguments to get frame_size and set codec type
if frame is None:
self.frame_size = frame_size
elif frame_size is None:
frame_size = np.shape(frame)
self.frame_size = frame_size
fourcc=cv2.VideoWriter_fourcc(
codec_code[0], codec_code[1],
codec_code[2], codec_code[3]
)
if filename is None:
filename = fd.save_filename(caption='select movie filename', file_filter='*.avi;;*.mp4')
self.write_vid = cv2.VideoWriter(
filename, fourcc, fps,
(self.frame_size[1], self.frame_size[0]))
if (write_frame is not None) and (frame is not None):
self.add_frame(frame)
self.fps = fps
self.filename = filename
if self.write_vid.isOpened():
print('Video open for writing')
else:
print('Issue correctly opening video')
print('Prob a compatability problem between codec '
'and format. Grayscale images are more fussy')
#Use this file to save a dictionary of sim params to file.
from Generic.filedialogs import save_filename
if __name__ == '__main__':
filename = save_filename()
