ScannerInterfaceExample/example_widget.py at main · ProstateBRP/ScannerInterfaceExample · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
from PyQt5 import QtCore, QtGui, QtWidgets


from mrigtlbridge.widget_base  import WidgetBase
import example_listener

class ExampleWidget(WidgetBase):

  def __init__(self, *args):
    super().__init__(*args)
    self.listener_class = ['example_listener', 'ExampleListener']
    self.parent = None
    self.listenerParameter['socketIP']    = '192.168.2.1'
    self.listenerParameter['socketPort']  = '7787'


  def buildGUI(self, parent):

    self.parent = parent

    layout = QtWidgets.QGridLayout()
    parent.setLayout(layout)

    self.ExampleConnectButton = QtWidgets.QPushButton("Connect to Example")
    self.ExampleConnectButton.setSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
    layout.addWidget(self.ExampleConnectButton, 0, 0, 1, 3)
    self.ExampleConnectButton.setEnabled(True)
    #self.ExampleConnectButton.clicked.connect(self.connectExample)
    self.ExampleConnectButton.clicked.connect(self.startListener)
    self.ExampleDisconnectButton = QtWidgets.QPushButton("Disconnect from Example")
    self.ExampleDisconnectButton.setSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
    layout.addWidget(self.ExampleDisconnectButton, 0, 3, 1, 3)
    self.ExampleDisconnectButton.setEnabled(False)
    #self.ExampleDisconnectButton.clicked.connect(self.disconnectExample)
    self.ExampleDisconnectButton.clicked.connect(self.stopListener)

    self.Example_IpEdit = QtWidgets.QLineEdit(self.listenerParameter['socketIP'])
    self.Example_IpEdit.textChanged.connect(self.updateListenerParameter)
    layout.addWidget(self.Example_IpEdit, 1, 0, 1, 4)

    self.Example_PortEdit = QtWidgets.QLineEdit(self.listenerParameter['socketPort'])
    self.Example_PortEdit.textChanged.connect(self.updateListenerParameter)
    layout.addWidget(self.Example_PortEdit, 1, 4, 1, 2)

    hline1 = QtWidgets.QFrame()
    hline1.setFrameShape(QtWidgets.QFrame.HLine)
    hline1.setFrameShadow(QtWidgets.QFrame.Sunken)
    layout.addWidget(hline1, 4, 0, 1, 6)

    self.ExampleStartSequenceButton = QtWidgets.QPushButton("Start sequence")
    layout.addWidget(self.ExampleStartSequenceButton, 6, 0, 1, 3)
    self.ExampleStartSequenceButton.clicked.connect(self.startSequenceExample)
    self.ExampleStartSequenceButton.setEnabled(False)
    self.ExampleStopSequenceButton = QtWidgets.QPushButton("Stop sequence")
    layout.addWidget(self.ExampleStopSequenceButton, 6, 3, 1, 3)
    self.ExampleStopSequenceButton.clicked.connect(self.stopSequenceExample)
    self.ExampleStopSequenceButton.setEnabled(False)

    hline2 = QtWidgets.QFrame()
    hline2.setFrameShape(QtWidgets.QFrame.HLine)
    hline2.setFrameShadow(QtWidgets.QFrame.Sunken)
    layout.addWidget(hline2, 7, 0, 1, 6)

    self.Example_textBox = QtWidgets.QTextEdit()
    self.Example_textBox.setReadOnly(True)
    self.Example_textBox.setSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Expanding)
    layout.addWidget(self.Example_textBox, 10, 0, 6, 6)

  def setSignalManager(self, sm):
    super().setSignalManager(sm)
    self.signalManager.connectSlot('consoleTextMR', self.updateExampleBox)
    self.signalManager.connectSlot('hostConnected', self.onHostConnected) # former connectExample()
    self.signalManager.connectSlot('hostDisconnected', self.onHostDisconnected) # former disconnectExample()


  def updateGUI(self, state):
    if state == 'listenerConnected':
      pass
    elif state == 'listenerDisconnected':
      pass


  def updateListenerParameter(self):

    self.listenerParameter['socketIP']    = self.Example_IpEdit.text()
    self.listenerParameter['socketPort']  = self.Example_PortEdit.text()

  def onHostConnected(self):
    try:
      self.updateTemplates()

      self.ExampleConnectButton.setEnabled(False)
      self.ExampleDisconnectButton.setEnabled(True)
      self.Example_IpEdit.setEnabled(False)
      self.Example_PortEdit.setEnabled(False)
      self.ExampleStartSequenceButton.setEnabled(True)
      self.ExampleStopSequenceButton.setEnabled(True)
    except:
      print("Failed to connect to Example!")


  def onHostDisconnected(self):
    self.ExampleConnectButton.setEnabled(True)
    self.ExampleDisconnectButton.setEnabled(False)
    self.Example_IpEdit.setEnabled(True)
    self.Example_PortEdit.setEnabled(True)
    self.ExampleStartSequenceButton.setEnabled(False)
    self.ExampleStopSequenceButton.setEnabled(False)


  def onSequenceStarted(self):
    print('onSequenceStarted()')
    self.ExampleStartSequenceButton.setEnabled(False)
    self.ExampleStopSequenceButton.setEnabled(True)

  def updateExampleBox(self, text):
    self.Example_textBox.append(text)

  def startSequenceExample(self):
    self.signalManager.emitSignal('startSequence')

  def stopSequenceExample(self):
    self.signalManager.emitSignal('stopSequence')