Commit ddd01088 authored by Martin Drechsler's avatar Martin Drechsler

image save button improved, so that it wont overwrite

parent 7861a4d6
......@@ -18,6 +18,7 @@ from subclasses.controllayouts import LaserControlLayout, LaserScanLayout
from subclasses.cameraParameterFrame import CameraParameterFrame
from resources.scanFunctions import create_measurement_array
from resources.storage import get_date_string
from resources.messages import show_warning
from appsettings import WORKING_DIR
from measurement import MeasurementFrame
import sys, os
......@@ -316,7 +317,7 @@ class CameraGuiMainWindow(QMainWindow):
measure_params['comment'] = self.measurementFrame.comments.toPlainText()
for l in LaserControlLayout._registry:
for sb_name, sb in l.spinboxes.items():
measure_params['sb_name'] = sb.value()
measure_params[sb_name] = sb.value()
for roi in self.rois:
measure_params['Roi Pos'] = roi.pos()
......@@ -324,11 +325,22 @@ class CameraGuiMainWindow(QMainWindow):
self.signalWithMeasurementParameters.emit(measure_params)
def save_current_image(self):
if self.cameraParamFrame.imagePrefix.text()=='':
show_warning('You have to write a filename')
return
if '.' in self.cameraParamFrame.imagePrefix.text():
show_warning('No points in your filename please')
return
saving_directory = WORKING_DIR + get_date_string()
if not os.path.exists(saving_directory):
os.makedirs(saving_directory)
os.makedirs(saving_directory + '/images')
np.save(saving_directory+'/images/'+self.cameraParamFrame.imagePrefix.text(), self.current_image)
filename = saving_directory+'/images/'+self.cameraParamFrame.imagePrefix.text()
if os.path.isfile(filename+'.npy'):
show_warning('File already exists!')
else:
np.save(filename, self.current_image)
def __del__(self):
print("chau gui")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment