(self)
| 399 | pygame.display.flip() |
| 400 | |
| 401 | def __processUserInput(self): |
| 402 | if not self.displayInitialized: return |
| 403 | for event in pygame.event.get(): |
| 404 | if event.type == pygame.QUIT: |
| 405 | self.shouldQuit = True |
| 406 | elif event.type == pygame.KEYDOWN: |
| 407 | if event.key in self.args.customKeyDownCallbacks: |
| 408 | self.args.customKeyDownCallbacks[event.key]() |
| 409 | elif event.key == pygame.K_q: self.shouldQuit = True |
| 410 | elif event.key == pygame.K_SPACE: self.shouldPause = not self.shouldPause |
| 411 | elif event.key == pygame.K_c: |
| 412 | self.cameraMode = self.cameraMode.next() |
| 413 | self.cameraControls2D.reset() |
| 414 | self.cameraControls3D.reset() |
| 415 | if self.cameraSmooth: self.cameraSmooth.reset() |
| 416 | elif event.key == pygame.K_PLUS: |
| 417 | self.map.setPointSize(np.clip(self.map.pointSize*1.05, 0.0, 10.0)) |
| 418 | elif event.key == pygame.K_MINUS: |
| 419 | self.map.setPointSize(np.clip(self.map.pointSize*0.95, 0.0, 10.0)) |
| 420 | elif event.key == pygame.K_0: |
| 421 | self.map.setColorMode(ColorMode.ORIGINAL.value) |
| 422 | elif event.key == pygame.K_1: |
| 423 | self.map.setColorMode(ColorMode.COORDINATE_X.value) |
| 424 | elif event.key == pygame.K_2: |
| 425 | self.map.setColorMode(ColorMode.COORDINATE_Y.value) |
| 426 | elif event.key == pygame.K_3: |
| 427 | self.map.setColorMode(ColorMode.COORDINATE_Z.value) |
| 428 | elif event.key == pygame.K_4: |
| 429 | self.map.setColorMode(ColorMode.DEPTH.value) |
| 430 | elif event.key == pygame.K_5: |
| 431 | self.map.setColorMode(ColorMode.NORMAL.value) |
| 432 | elif event.key == pygame.K_m: |
| 433 | self.map.setRenderMesh(not self.map.renderMesh) |
| 434 | elif event.key == pygame.K_n: |
| 435 | self.map.setRenderPointCloud(not self.map.renderPointCloud) |
| 436 | elif event.key == pygame.K_k: |
| 437 | self.map.setRenderKeyFrames(not self.map.renderKeyFrames) |
| 438 | elif event.key == pygame.K_g: |
| 439 | self.showGrid = not self.showGrid |
| 440 | elif event.key == pygame.K_p: |
| 441 | self.showPoseTrail = not self.showPoseTrail |
| 442 | elif event.key == pygame.K_f: |
| 443 | from pygame.locals import DOUBLEBUF, OPENGL, FULLSCREEN |
| 444 | w, h = self.adjustedResolution[0], self.adjustedResolution[1] |
| 445 | self.fullScreen = not self.fullScreen |
| 446 | if self.fullScreen: pygame.display.set_mode((w, h), DOUBLEBUF | OPENGL | FULLSCREEN) |
| 447 | else: pygame.display.set_mode((w, h), DOUBLEBUF | OPENGL) |
| 448 | elif event.key == pygame.K_h: |
| 449 | self.printHelp() |
| 450 | else: |
| 451 | if self.cameraMode is CameraMode.THIRD_PERSON: self.cameraControls3D.update(event) |
| 452 | if self.cameraMode is CameraMode.TOP_VIEW: self.cameraControls2D.update(event) |
| 453 | |
| 454 | def onVioOutput(self, cameraPose, image=None, width=None, height=None, colorFormat=None, status=None): |
| 455 | if self.shouldQuit: return |
no test coverage detected