| 172 | self.setCentralWidget(mainwidget) |
| 173 | |
| 174 | def setupCustomMenuBar(self): |
| 175 | menuBar = QWidget() |
| 176 | menuBar.setStyleSheet( |
| 177 | "QPushButton {background-color: rgb(72,61,179); color: white;} \ |
| 178 | QPushButton:checked { background-color: rgb(72,61,139); }" |
| 179 | ) |
| 180 | layout = QGridLayout() |
| 181 | self.startBtn = QPushButton("Choose Start Lane") |
| 182 | self.startBtn.setCheckable(True) |
| 183 | self.startBtn.toggled.connect(self.chooseEdgeS) |
| 184 | layout.addWidget(self.startBtn, 0, 0) |
| 185 | self.destBtn = QPushButton("Choose Arrival Lane") |
| 186 | self.destBtn.setCheckable(True) |
| 187 | self.destBtn.toggled.connect(self.chooseEdgeD) |
| 188 | layout.addWidget(self.destBtn, 0, 1) |
| 189 | self.okBtn = QPushButton("Confirm") |
| 190 | self.okBtn.clicked.connect(self.startSimulate) |
| 191 | layout.addWidget(self.okBtn, 0, 2) |
| 192 | menuBar.setLayout(layout) |
| 193 | self.setMenuWidget(menuBar) |
| 194 | |
| 195 | def chooseEdgeS(self, checked): |
| 196 | global CHOOSE_STATE |