|
|
from PyQt5.QtCore import pyqtSignal, QObject
|
|
|
from codes.show import Camera
|
|
|
from codes.calibration import Clb
|
|
|
from codes.sampling import Smp
|
|
|
from codes.tune_models_params import Tuning
|
|
|
from codes.eye_track import EyeTrack
|
|
|
from codes.see_data import See
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Worker(QObject, Camera, Clb, Smp, Tuning, EyeTrack, See):
|
|
|
num = 0
|
|
|
camera_id = 0
|
|
|
thb = 0.0
|
|
|
ss = 0
|
|
|
mfr = 0.0
|
|
|
dft = 0.0
|
|
|
st = 0.0
|
|
|
|
|
|
cam = False
|
|
|
clb = False
|
|
|
smp = False
|
|
|
acc = False
|
|
|
tbt = False
|
|
|
mdl = False
|
|
|
uio = False
|
|
|
gps = False
|
|
|
gpa = False
|
|
|
sps = False
|
|
|
spa = False
|
|
|
gfx = False
|
|
|
|
|
|
running = True
|
|
|
|
|
|
cam_started = pyqtSignal()
|
|
|
clb_started = pyqtSignal()
|
|
|
smp_started = pyqtSignal()
|
|
|
acc_started = pyqtSignal()
|
|
|
tbt_started = pyqtSignal()
|
|
|
mdl_started = pyqtSignal()
|
|
|
gps_started = pyqtSignal()
|
|
|
gpa_started = pyqtSignal()
|
|
|
sps_started = pyqtSignal()
|
|
|
spa_started = pyqtSignal()
|
|
|
gfx_started = pyqtSignal()
|
|
|
finished = pyqtSignal()
|
|
|
|
|
|
def __init__(self):
|
|
|
super().__init__()
|
|
|
|
|
|
def do_work(self):
|
|
|
if self.cam and self.running:
|
|
|
print("\nCamera")
|
|
|
self.cam_started.emit()
|
|
|
self.features(camera_id=self.camera_id)
|
|
|
if self.clb and self.running:
|
|
|
print("\nCalibration")
|
|
|
self.clb_started.emit()
|
|
|
self.et(
|
|
|
num=self.num,
|
|
|
camera_id=self.camera_id,
|
|
|
info=self.info,
|
|
|
clb_grid=self.clb_grid
|
|
|
)
|
|
|
if self.smp and self.running:
|
|
|
print("\nSampling")
|
|
|
self.smp_started.emit()
|
|
|
self.sampling(
|
|
|
num=self.num,
|
|
|
camera_id=self.camera_id,
|
|
|
gui=True
|
|
|
)
|
|
|
if self.acc and self.running:
|
|
|
print("\nTesting")
|
|
|
self.acc_started.emit()
|
|
|
self.accuracy(
|
|
|
num=self.num,
|
|
|
camera_id=self.camera_id,
|
|
|
clb_grid=(5, 7, 20)
|
|
|
)
|
|
|
if self.tbt and self.running:
|
|
|
print("\nSee user blinking")
|
|
|
self.user_face(
|
|
|
num=self.num,
|
|
|
threshold=self.thb,
|
|
|
save_threshold=True
|
|
|
)
|
|
|
if self.mdl and self.running:
|
|
|
print("\nTuning params")
|
|
|
self.mdl_started.emit()
|
|
|
self.et_mdl(
|
|
|
subjects=[self.num],
|
|
|
shift_samples=[self.ss],
|
|
|
delete_files=False
|
|
|
)
|
|
|
if self.gps and self.running:
|
|
|
print("\nGetting pixels")
|
|
|
self.gps_started.emit()
|
|
|
self.get_pixels(
|
|
|
subjects=[self.num],
|
|
|
shift_samples=[self.ss],
|
|
|
use_io=self.uio,
|
|
|
delete_files=False
|
|
|
)
|
|
|
if self.gpa and self.running:
|
|
|
print("\nGetting test pixels")
|
|
|
self.gpa_started.emit()
|
|
|
self.get_pixels(
|
|
|
subjects=[self.num],
|
|
|
target_fol="acc",
|
|
|
shift_samples=[self.ss],
|
|
|
use_io=True,
|
|
|
delete_files=False
|
|
|
)
|
|
|
if self.sps and self.running:
|
|
|
print("\nSeeing sampling data")
|
|
|
self.sps_started.emit()
|
|
|
self.pixels_smp(num=self.num, show_in_all_monitors=True)
|
|
|
if self.spa and self.running:
|
|
|
print("\nSeeing testing data")
|
|
|
self.spa_started.emit()
|
|
|
self.pixels_acc(
|
|
|
num=self.num,
|
|
|
show_in_all_monitors=True
|
|
|
)
|
|
|
if self.gfx and self.running:
|
|
|
print("\nGetting fixations")
|
|
|
self.gfx_started.emit()
|
|
|
self.get_fixations(
|
|
|
subjects=[self.num],
|
|
|
t_discard=self.dft,
|
|
|
x_merge=self.mfr[0],
|
|
|
y_merge=self.mfr[1],
|
|
|
vx_thr=self.st,
|
|
|
vy_thr=self.st
|
|
|
)
|
|
|
|
|
|
|
|
|
print("\nEye Tracking finished!")
|
|
|
self.finished.emit()
|
|
|
self.running = True
|
|
|
|
