scale_line.py

import pyqtgraph as pg
from PyQt5 import QtCore

class Plot_Scale_Line(object):
    def __init__(self, plot):
        self.v_grid_lines = list()
        for i in range(35):
            grid_lines = pg.InfiniteLine(pos=(i, 0), pen=pg.mkPen(color=(192, 192, 192), style=QtCore.Qt.DotLine))
            self.v_grid_lines.append(grid_lines)
            plot.addItem(grid_lines)

        self.zero_base_lines = list()
        self.neg_base_lines = list()
        self.pos_base_lines = list()
        for i in range(1,65):
            zero_line = pg.InfiniteLine(pos=(0, i*10), angle=180, pen=pg.mkPen(color=(128, 128, 128), style=QtCore.Qt.DotLine))
            self.zero_base_lines.append(zero_line)
            plot.addItem(zero_line)

    def update(self, selected_channels, axises_tick):
        for idx, (val, label) in enumerate(axises_tick):
            self.v_grid_lines[idx].setValue(val)
        for i in range(64):
            if i < len(selected_channels):
                self.zero_base_lines[i].show()
            else:
                self.zero_base_lines[i].hide()
                continue