def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
#ETC
self.output = False
#Connect signals
self.ui.horizontalSlider_general_blur.sliderMoved.connect(self._blurSlider)
self.ui.horizontalSlider_general_zoom.sliderMoved.connect(self._zoomSlider)
self.ui.horizontalSlider_led_binary.sliderMoved.connect(self._ledBinarySlider)
self.ui.horizontalSlider_led_min.sliderMoved.connect(self._ledMinSlider)
self.ui.horizontalSlider_led_max.sliderMoved.connect(self._ledMaxSlider)
self.ui.horizontalSlider_pupil_binary.sliderMoved.connect(self._pupilBinarySlider)
self.ui.horizontalSlider_pupil_min.sliderMoved.connect(self._pupilMinSlider)
self.ui.horizontalSlider_pupil_max.sliderMoved.connect(self._pupilMaxSlider)
self.ui.horizontalSlider_camera_hue.sliderMoved.connect(self._cameraHueSlider)
self.ui.horizontalSlider_camera_saturation.sliderMoved.connect(self._cameraSaturationSlider)
self.ui.horizontalSlider_camera_brightness.sliderMoved.connect(self._cameraBrightnessSlider)
self.ui.horizontalSlider_camera_gain.sliderMoved.connect(self._cameraGainSlider)
self.ui.horizontalSlider_camera_contrast.sliderMoved.connect(self._cameraContrastSlider)
self.ui.horizontalSlider_camera_exposure.sliderMoved.connect(self._cameraExposureSlider)
self.ui.pushButton_output.clicked.connect(self._outputToggle)
#Create Eyetracker
self.et = Eyetracker()
#Set initial slider bar states (might should be before signals)
self.ui.horizontalSlider_general_blur.setValue(self.et.blur)
self.ui.horizontalSlider_general_zoom.setValue(self.et.zoom)
self.ui.horizontalSlider_led_binary.setValue(self.et.ledthresh)
self.ui.horizontalSlider_led_min.setValue(self.et.ledsize[0])
self.ui.horizontalSlider_led_max.setValue(self.et.ledsize[1])
self.ui.horizontalSlider_pupil_binary.setValue(self.et.pupilthresh)
self.ui.horizontalSlider_pupil_min.setValue(self.et.pupilsize[0])
self.ui.horizontalSlider_pupil_max.setValue(self.et.pupilsize[1])
#and for the camera...
#self.ui.horizontalSlider_camera_hue.setValue(
# self.et.camproperties['hue'])
self.ui.horizontalSlider_camera_saturation.setValue(
self.et.camproperties['saturation'])
self.ui.horizontalSlider_camera_brightness.setValue(
self.et.camproperties['brightness'])
#self.ui.horizontalSlider_camera_gain.setValue(
# self.et.camproperties['gain'])
self.ui.horizontalSlider_camera_contrast.setValue(
self.et.camproperties['contrast'])
self.ui.horizontalSlider_camera_exposure.setValue(
self.et.camproperties['exposure'])
#Timer setup
self.ctimer = QtCore.QTimer()
self.ctimer.timeout.connect(self._tick)
self.ctimer.start(1) #run as fast as possible
def main():
try:
ser = serial.Serial('/dev/ttyACM0',115200)
except serial.serialutil.SerialException as e:
pass
tracker = Eyetracker()
tracker.getBoundingRectangle()
tracker.calibrate()
maxRotationVal = 40
minRotationVal = -40
start = time.time()
end = time.time()
blinktimer = 0
forwardmode = False
while True:
end = time.time()
dt = end-start
if tracker.detectBlink():
blinktimer += dt
else:
blinktimer = 0
start = time.time()
if blinktimer > 0.5:
if forwardmode:
forwardmode = False
else:
forwardmode = True
blinktimer = -9999
r = 192
f = 64
if not tracker.blink:
tracker.trackPupil()
dist = tracker.pupil[0]- tracker.center[0]
if dist > maxRotationVal:
dist = maxRotationVal
if dist < minRotationVal:
dist = minRotationVal
if dist < 0:
r = r - 63*dist/minRotationVal
if dist > 0:
r = r + 63*dist/maxRotationVal
if forwardmode:
print("FORWAAAAAARD")
f = 80
#ser.write(chr(r))
#ser.write(chr(f))
#print(r)
tracker.draw()
if cv2.waitKey(1) & 0xFF == ord('q'):
break
tracker.cam.release()
class MyForm(QtGui.QMainWindow):
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
#ETC
self.output = False
#Connect signals
self.ui.horizontalSlider_general_blur.sliderMoved.connect(self._blurSlider)
self.ui.horizontalSlider_general_zoom.sliderMoved.connect(self._zoomSlider)
self.ui.horizontalSlider_led_binary.sliderMoved.connect(self._ledBinarySlider)
self.ui.horizontalSlider_led_min.sliderMoved.connect(self._ledMinSlider)
self.ui.horizontalSlider_led_max.sliderMoved.connect(self._ledMaxSlider)
self.ui.horizontalSlider_pupil_binary.sliderMoved.connect(self._pupilBinarySlider)
self.ui.horizontalSlider_pupil_min.sliderMoved.connect(self._pupilMinSlider)
self.ui.horizontalSlider_pupil_max.sliderMoved.connect(self._pupilMaxSlider)
self.ui.horizontalSlider_camera_hue.sliderMoved.connect(self._cameraHueSlider)
self.ui.horizontalSlider_camera_saturation.sliderMoved.connect(self._cameraSaturationSlider)
self.ui.horizontalSlider_camera_brightness.sliderMoved.connect(self._cameraBrightnessSlider)
self.ui.horizontalSlider_camera_gain.sliderMoved.connect(self._cameraGainSlider)
self.ui.horizontalSlider_camera_contrast.sliderMoved.connect(self._cameraContrastSlider)
self.ui.horizontalSlider_camera_exposure.sliderMoved.connect(self._cameraExposureSlider)
self.ui.pushButton_output.clicked.connect(self._outputToggle)
#Create Eyetracker
self.et = Eyetracker()
#Set initial slider bar states (might should be before signals)
self.ui.horizontalSlider_general_blur.setValue(self.et.blur)
self.ui.horizontalSlider_general_zoom.setValue(self.et.zoom)
self.ui.horizontalSlider_led_binary.setValue(self.et.ledthresh)
self.ui.horizontalSlider_led_min.setValue(self.et.ledsize[0])
self.ui.horizontalSlider_led_max.setValue(self.et.ledsize[1])
self.ui.horizontalSlider_pupil_binary.setValue(self.et.pupilthresh)
self.ui.horizontalSlider_pupil_min.setValue(self.et.pupilsize[0])
self.ui.horizontalSlider_pupil_max.setValue(self.et.pupilsize[1])
#and for the camera...
#self.ui.horizontalSlider_camera_hue.setValue(
# self.et.camproperties['hue'])
self.ui.horizontalSlider_camera_saturation.setValue(
self.et.camproperties['saturation'])
self.ui.horizontalSlider_camera_brightness.setValue(
self.et.camproperties['brightness'])
#self.ui.horizontalSlider_camera_gain.setValue(
# self.et.camproperties['gain'])
self.ui.horizontalSlider_camera_contrast.setValue(
self.et.camproperties['contrast'])
self.ui.horizontalSlider_camera_exposure.setValue(
self.et.camproperties['exposure'])
#Timer setup
self.ctimer = QtCore.QTimer()
self.ctimer.timeout.connect(self._tick)
self.ctimer.start(1) #run as fast as possible
def _blurSlider(self):
self.et.blur = int(self.ui.horizontalSlider_general_blur.value())
def _zoomSlider(self):
self.et.zoom = int(self.ui.horizontalSlider_general_zoom.value())
def _ledBinarySlider(self):
self.et.ledthresh = int(self.ui.horizontalSlider_led_binary.value())
def _ledMinSlider(self):
newval = int(self.ui.horizontalSlider_led_min.value())
if newval <= self.et.ledsize[1]:
self.et.ledsize[0] = newval
def _ledMaxSlider(self):
newval = int(self.ui.horizontalSlider_led_max.value())
if newval >= self.et.ledsize[0]:
self.et.ledsize[1] = newval
def _pupilBinarySlider(self):
self.et.pupilthresh = int(self.ui.horizontalSlider_pupil_binary.value())
def _pupilMinSlider(self):
newval = int(self.ui.horizontalSlider_pupil_min.value())
if newval <= self.et.pupilsize[1]:
self.et.pupilsize[0] = newval
def _pupilMaxSlider(self):
newval = int(self.ui.horizontalSlider_pupil_max.value())
if newval >= self.et.pupilsize[0]:
self.et.pupilsize[1] = newval
def _cameraHueSlider(self):
pass
def _cameraSaturationSlider(self):
self.et.camproperties['saturation'] = int(
self.ui.horizontalSlider_camera_saturation.value())
self.et.setCamProp('saturation')
def _cameraBrightnessSlider(self):
self.et.camproperties['brightness'] = int(
self.ui.horizontalSlider_camera_brightness.value())
self.et.setCamProp('brightness')
def _cameraGainSlider(self):
pass
def _cameraContrastSlider(self):
self.et.camproperties['contrast'] = int(
self.ui.horizontalSlider_camera_contrast.value())
self.et.setCamProp('contrast')
def _cameraExposureSlider(self):
self.et.camproperties['exposure'] = int(
self.ui.horizontalSlider_camera_exposure.value())
self.et.setCamProp('exposure')
def _tick(self):
try:
self.et.nextFrame()
if self.output:
self._output()
except:
self.et.getNewCam()
def _output(self):
if self.ui.checkBox_toFile.isChecked():
out = str(self.et.getGaze())+'\n'
self.outputfile.write(out)
if self.ui.checkBox_toVideo.isChecked():
pass #grabs frames automatically
if self.ui.checkBox_toConsole.isChecked():
print self.et.getGaze()
def _outputToggle(self):
if self.output:
self.output=False
try:
self.outputfile.close()
except Exception, e:
pass
try:
self.et.stopVideo()
except Exception, e:
print e
def main():
try:
ser = serial.Serial('/dev/ttyACM0', 115200)
except serial.serialutil.SerialException as e:
pass
tracker = Eyetracker()
tracker.getBoundingRectangle()
tracker.calibrate()
maxRotationVal = 40
minRotationVal = -40
start = time.time()
end = time.time()
blinktimer = 0
forwardmode = False
while True:
end = time.time()
dt = end - start
if tracker.detectBlink():
blinktimer += dt
else:
blinktimer = 0
start = time.time()
if blinktimer > 0.5:
if forwardmode:
forwardmode = False
else:
forwardmode = True
blinktimer = -9999
r = 192
f = 64
if not tracker.blink:
tracker.trackPupil()
dist = tracker.pupil[0] - tracker.center[0]
if dist > maxRotationVal:
dist = maxRotationVal
if dist < minRotationVal:
dist = minRotationVal
if dist < 0:
r = r - 63 * dist / minRotationVal
if dist > 0:
r = r + 63 * dist / maxRotationVal
if forwardmode:
print("FORWAAAAAARD")
f = 80
#ser.write(chr(r))
#ser.write(chr(f))
#print(r)
tracker.draw()
if cv2.waitKey(1) & 0xFF == ord('q'):
break
tracker.cam.release()