Last active
          March 7, 2020 23:56 
        
      - 
      
- 
        Save spewil/c08c6da6c62243d832e94a942a86bec8 to your computer and use it in GitHub Desktop. 
    pyqt with a background process updating shared memory block thanks to python 3.8
  
        
  
    
      This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
      Learn more about bidirectional Unicode characters
    
  
  
    
  | import time | |
| from PyQt5.QtCore import QTimer | |
| from PyQt5.QtWidgets import QWidget, QVBoxLayout, QProgressBar, QPushButton | |
| import multiprocessing | |
| from multiprocessing import shared_memory | |
| import numpy as np | |
| DATA_SIZE = 1 | |
| DTYPE = np.int64 | |
| class DataPuller(multiprocessing.Process): | |
| def __init__( | |
| self, | |
| event, | |
| shared_mem_name, | |
| ): | |
| super().__init__() | |
| self.event = event | |
| self.counter = 0 | |
| self.shared_mem_name = shared_mem_name | |
| def run(self): | |
| self.shm = shared_memory.SharedMemory(name=self.shared_mem_name) | |
| self.sample = np.ndarray( | |
| shape=(DATA_SIZE, 1), dtype=DTYPE, buffer=self.shm.buf) | |
| print("child process started") | |
| while not self.event.is_set(): | |
| # time.sleep(0.001) | |
| self.sample[:, 0] = self.counter | |
| # print("child: ", self.sample[0, 0]) | |
| self.counter += 1 | |
| if self.counter >= 1000: | |
| self.counter = 0 | |
| self.shm.close() | |
| print("child process finished") | |
| class Window(QWidget): | |
| def __init__(self, *args, **kwargs): | |
| super(Window, self).__init__(*args, **kwargs) | |
| layout = QVBoxLayout(self) | |
| self.progressBar = QProgressBar(self) | |
| self.timer = QTimer(self) | |
| self.timer.timeout.connect(self.onTimer) | |
| self.progressBar.setRange(0, 1000) | |
| layout.addWidget(self.progressBar) | |
| layout.addWidget(QPushButton('开启线程', self, clicked=self.onStart)) | |
| self.counter = 0 | |
| self.shared_mem_name = "data" | |
| self.base_array = np.zeros((DATA_SIZE, 1), dtype=DTYPE) | |
| self.shm = shared_memory.SharedMemory( | |
| create=True, | |
| size=self.base_array.nbytes, | |
| name=self.shared_mem_name) | |
| self.sample = np.ndarray( | |
| shape=self.base_array.shape, | |
| dtype=self.base_array.dtype, | |
| buffer=self.shm.buf) | |
| self.event = multiprocessing.Event() | |
| self._process = DataPuller(self.event, self.shared_mem_name) | |
| def onStart(self): | |
| if not self._process.is_alive(): | |
| print("main starting process") | |
| self._process.start() | |
| self.timer.start(16) | |
| else: | |
| pass | |
| def onTimer(self): | |
| # print("main: ", self.sample[0, 0]) | |
| self.progressBar.setValue(self.sample[0, 0]) | |
| def closeEvent(self, event): | |
| if self._process.is_alive(): | |
| self.event.set() | |
| self._process.join() | |
| self.shm.close() | |
| self.shm.unlink() | |
| self.close() | |
| print("main process finished") | |
| if __name__ == '__main__': | |
| # need this to match OSX on Windows | |
| multiprocessing.set_start_method("spawn") | |
| import sys | |
| from PyQt5.QtWidgets import QApplication | |
| app = QApplication(sys.argv) | |
| w = Window() | |
| w.show() | |
| sys.exit(app.exec_()) | 
  
    Sign up for free
    to join this conversation on GitHub.
    Already have an account?
    Sign in to comment