class BMI_RASTER_GUI(QWidget):
def __init__(self, prb, fet_file):
QWidget.__init__(self)
self.view_timer = QtCore.QTimer(self)
self.view_timer.timeout.connect(self.view_update)
self.update_interval = 60
self.bmi = BMI(prb, fet_file)
self.setAutoFillBackground(True)
p = self.palette()
p.setColor(self.backgroundRole(), Qt.darkGray)
p.setColor(self.foregroundRole(), Qt.white)
self.setPalette(p)
self.bmiBtn = QPushButton("BMI Stream Off", self)
self.bmiBtn.setCheckable(True)
self.bmiBtn.setStyleSheet("background-color: darkgrey")
self.bmiBtn.toggled.connect(self.bmi_process_toggle)
self.rsview = raster_view(t_window=25e-3)
# self.rsview.load_raster('./fet.bin')
# self.rsview = scatter_3d_view()
# self.rsview.set_data(np.random.randn(1000,4))
layout = QVBoxLayout()
layout.addWidget(self.bmiBtn)
layout.addWidget(self.rsview.native)
self.setLayout(layout)
def bmi_process_toggle(self, checked):
if checked:
self.bmi_process_start()
else:
self.bmi_process_stop()
def bmi_process_start(self, gui_queue=False):
self.bmiBtn.setText('BMI Stream ON')
self.bmiBtn.setStyleSheet("background-color: green")
self.bmi.start(gui_queue)
self.view_timer.start(self.update_interval)
def bmi_process_stop(self):
self.bmiBtn.setText('BMI Stream Off')
self.bmiBtn.setStyleSheet("background-color: darkgrey")
self.bmi.stop()
self.view_timer.stop()
def view_update(self):
self.rsview.update_fromfile('./fet.bin', last_N=8000, view_window=10)
def keyPressEvent(self, e):
print("event", e)
class BMI_GUI(QWidget):
def __init__(self, prb, fet_file, show=True):
QWidget.__init__(self)
self.fet_view_timer = QtCore.QTimer(self)
self.fet_view_timer.timeout.connect(self.update_fet_views) # fet_view_update
self.current_group = 0
self.init_BMI(prb, fet_file)
self.fresh_interval = 30
self._frame = 0
self._fet = {}
self._clu = {}
for i in range(40):
self._fet[i] = np.zeros((40, 4))
self._clu[i] = np.zeros((40,), dtype=np.int64)
if show:
self.init_UI()
def init_BMI(self, prb, fet_file):
print('---init FPGA BMI---')
print('---download probe file into FPGA---')
try:
self.bmi = BMI(prb, fet_file)
except Exception as e:
raise
def init_UI(self, keys='interactive'):
self.setAutoFillBackground(True)
p = self.palette()
p.setColor(self.backgroundRole(), Qt.darkGray)
p.setColor(self.foregroundRole(), Qt.white)
self.setPalette(p)
self.bmiBtn = QPushButton("BMI Stream Off",self)
self.bmiBtn.setCheckable(True)
self.bmiBtn.setStyleSheet("background-color: darkgrey")
self.bmiBtn.toggled.connect(self.bmi_process_toggle)
rows,cols=6,7
self.fet_grid = grid_scatter3d(rows, cols)
self.N = 5000
for i in range(rows*cols):
if i in self.bmi.fpga.configured_groups:
self.fet_grid.fet_view[i].info = i
self.fet_grid.fet_view[i].set_data(np.zeros((self.N,4), dtype=np.float32))
layout = QVBoxLayout()
layout.addWidget(self.bmiBtn)
layout.addWidget(self.fet_grid)
self.setLayout(layout)
def bmi_process_toggle(self, checked):
if checked:
self.bmi_process_start()
else:
self.bmi_process_stop()
def bmi_process_start(self, gui_queue=False):
self.bmiBtn.setText('BMI Stream ON')
self.bmiBtn.setStyleSheet("background-color: green")
self.bmi.start(gui_queue=False)
self.fet_view_timer.start(self.fresh_interval)
def bmi_process_stop(self):
self.bmiBtn.setText('BMI Stream Off')
self.bmiBtn.setStyleSheet("background-color: darkgrey")
self.bmi.stop()
self.fet_view_timer.stop()
def update_fet_views(self):
self._frame += 1
# print('frame',self._frame%2)
N = 5000
try:
# fet = np.fromfile('./fet.bin', dtype=np.int32).reshape(-1, 7)
fet = np.memmap('./fet.bin', dtype=np.int32)
ngrp = len(self.bmi.fpga.configured_groups)
if fet.shape[0] > 0:
fet = fet.reshape(-1, 7)
fet_info = fet[:,:2]
fet_val = fet[:,2:6]
labels = fet[:, -1]
labels[labels==101] = self.bmi.fpga.target_unit # it is 101 because FPGA output triggered TTL
# get idx of fet from current selected group
if self._frame % 2 == 0:
for grp_id in self.bmi.fpga.configured_groups[:int(ngrp/2)]:
idx = np.where(fet_info[:,1]==grp_id)[0]
if len(idx)>N: idx = idx[-N:]
fet = fet_val[idx, :]/float(2**16)
clu = labels[idx]
# self.log.info('get_fet{}'.format(idx.shape))
if len(fet)>0:
self.fet_grid.fet_view[grp_id].stream_in(fet, clu, highlight_no=30)
elif self._frame % 2 == 1:
for grp_id in self.bmi.fpga.configured_groups[int(ngrp/2):]:
idx = np.where(fet_info[:,1]==grp_id)[0]
if len(idx)>N: idx = idx[-N:]
fet = fet_val[idx, :]/float(2**16)
clu = labels[idx]
# self.log.info('get_fet{}'.format(idx.shape))
if len(fet)>0:
self.fet_grid.fet_view[grp_id].stream_in(fet, clu, highlight_no=30)
except:
pass
class BMI_GUI(QWidget):
def __init__(self, fet_file, prb=None, show=True):
QWidget.__init__(self)
self.fet_view_timer = QtCore.QTimer(self)
self.fet_view_timer.timeout.connect(
self.update_fet_views) # fet_view_update
self.current_group = 0
self.init_BMI(fet_file, prb)
self.fresh_interval = 30
self._fet = {}
self._clu = {}
for i in range(40):
self._fet[i] = np.zeros((40, 4))
self._clu[i] = np.zeros((40, ), dtype=np.int64)
if show:
self.init_UI()
def init_BMI(self, fet_file, prb=None):
print('---init FPGA BMI---')
print('---download probe file into FPGA---')
try:
self.bmi = BMI(fetfile=fet_file, prb=prb)
except Exception as e:
raise
def init_UI(self, keys='interactive'):
self.setAutoFillBackground(True)
p = self.palette()
p.setColor(self.backgroundRole(), Qt.darkGray)
p.setColor(self.foregroundRole(), Qt.white)
self.setPalette(p)
self.bmiBtn = QPushButton("BMI Stream Off", self)
self.bmiBtn.setCheckable(True)
self.bmiBtn.setStyleSheet("background-color: darkgrey")
self.bmiBtn.toggled.connect(self.bmi_process_toggle)
rows, cols = 6, 7
self.fet_grid = grid_scatter3d(rows, cols)
self.N = 2000
for i in range(rows * cols):
if i in self.bmi.fpga.configured_groups:
self.fet_grid.fet_view[i].info = i
self.fet_grid.fet_view[i].set_data(
np.zeros((self.N, 4), dtype=np.float32))
layout = QVBoxLayout()
layout.addWidget(self.bmiBtn)
layout.addWidget(self.fet_grid)
self.setLayout(layout)
def bmi_process_toggle(self, checked):
if checked:
self.bmi_process_start()
else:
self.bmi_process_stop()
def bmi_process_start(self, gui_queue=True):
self.bmiBtn.setText('BMI Stream ON')
self.bmiBtn.setStyleSheet("background-color: green")
self.bmi.start(gui_queue)
if gui_queue:
self.fet_view_timer.start(self.fresh_interval)
def bmi_process_stop(self):
self.bmiBtn.setText('BMI Stream Off')
self.bmiBtn.setStyleSheet("background-color: darkgrey")
self.bmi.stop()
self.fet_view_timer.stop()
def update_fet_views(self):
group_need_update = []
_nspk, limit = 0, 40
while not self.bmi.gui_queue.empty():
with Timer('receive', verbose=False):
_timestamp, _grp_id, _fet0, _fet1, _fet2, _fet3, _spk_id = self.bmi.gui_queue.get(
)
# print(_grp_id, _spk_id)
if _grp_id in self.bmi.fpga.configured_groups and _spk_id > 0 and _spk_id != 101: ## 101 is for target TTL neuron
group_need_update.append(_grp_id)
incoming_fet = np.array([_fet0, _fet1, _fet2, _fet3
]) / float(2**16)
incoming_clu = int(_spk_id)
self._fet[_grp_id] = np.roll(self._fet[_grp_id], 1, axis=0)
self._fet[_grp_id][0] = incoming_fet
self._clu[_grp_id] = np.roll(self._clu[_grp_id], 1)
self._clu[_grp_id][0] = incoming_clu
_nspk += 1
if _nspk > limit:
while not self.bmi.gui_queue.empty():
self.bmi.gui_queue.get()
break
with Timer('update', verbose=False):
for grp_id in group_need_update:
with Timer('render', verbose=False):
self.fet_grid.fet_view[grp_id].stream_in(
self._fet[grp_id][:_nspk],
self._clu[grp_id][:_nspk],
rho=1,
highlight_no=10)