RemoteRenderingEventReplay.py

Go to the documentation of this file.

#
# Copyright 2013, MeVis Medical Solutions AG
#
# The user may use this file in accordance with the license agreement provided with
# the Software or, alternatively, in accordance with the terms contained in a
# written agreement between the user and MeVis Medical Solutions AG.
#
# For further information use the contact form at https://www.mevislab.de/contact
#
 
from TestSupport import Base, Logging
from TestSupport.Macros import EXPECT_TRUE
from mevis import MLAB, MLABFileManager
from PythonQt import QtCore
import json
import os.path
 
# A helper class to replay events recorded by the RemoteRenderingEventRecorder
class EventReplay(object):
  
  # init the replay with a given remote rendering slave and a *.events filename
  def __init__(self, slave, filename, resizeHandler = None):
    self._baseName_baseName = os.path.splitext(os.path.basename(filename))[0]
    self._slave_slave = slave
    f = open(filename, "rb")
    self._events_events = json.load(f)
    f.close()
    self.screenshotCountscreenshotCount = 0
    self.sceneHasChangedsceneHasChanged = False
    self.resizeHandlerresizeHandler = resizeHandler
    self._slave_slave.sceneChanged.connect(self.sceneChangedsceneChanged)
 
  # create an additional screenshot
  def createScreenshot(self):
    self._slave_slave.imageReceived.connect(self.imageReceivedimageReceived)
    self._slave_slave.sendRenderRequest(self._slave_slave.PNG)
    self._slave_slave.imageReceived.disconnect(self.imageReceivedimageReceived)
 
  # notice if the scene has changed
  def sceneChanged(self):
    self.sceneHasChangedsceneHasChanged = True
 
  # handle the receiving of an image  
  def imageReceived(self, img):
    name = self._baseName_baseName + str(self.screenshotCountscreenshotCount)
    filename = Base.getResultDirectory() + "/" + name + ".png"
    EXPECT_TRUE(img.save(filename))
 
    referenceImage = Base.getDataDirectory() + "/" + name + ".png"
    if not MLABFileManager.exists(referenceImage):
      EXPECT_TRUE(img.save(referenceImage))
 
    Logging.showImage(name + " (reference image on the right)", filename, referenceImage)
    self.screenshotCountscreenshotCount += 1
 
  def doPause(self, delay):
    time = QtCore.QTime()
    time.start()
    while time.elapsed() < delay:
      MLAB.msleep(10)
      MLAB.processEvents()
    
  # replay all events
  def replay(self, redrawAfterResize=False, redrawAfterSceneChanged=False):
    for event in self._events_events:
      shouldRedraw = False
      eventType = event["event"]
      hasPosition = "x" in event and "y" in event
      if eventType == "Resize":
        w = event["width"]
        h = event["height"]
        if self.resizeHandlerresizeHandler:
          self.resizeHandlerresizeHandler(w,h)
        self._slave_slave.resize(w, h)
        if redrawAfterResize:
          shouldRedraw = True
      elif eventType == "Pause":
        self.doPausedoPause(event["delay"])
      elif eventType == "Enter":
        self._slave_slave.enter()
      elif eventType == "Leave":
        self._slave_slave.leave()
      elif eventType == "MouseMove":
        self._slave_slave.mouseMoveTo(event["x"],event["y"])
      elif eventType == "MouseButtonPress":
        if hasPosition:
          self._slave_slave.mousePress(event["button"],event["x"],event["y"])
        else:
          self._slave_slave.mousePress(event["button"])
      elif eventType == "MouseButtonRelease":
        if hasPosition:
          self._slave_slave.mouseRelease(event["button"],event["x"],event["y"])
        else:
          self._slave_slave.mouseRelease(event["button"])
      elif eventType == "MouseButtonDblClick":
        if hasPosition:
          self._slave_slave.mouseDoubleClickPress(event["button"],event["x"],event["y"])
        else:
          self._slave_slave.mouseDoubleClickPress(event["button"])
      elif eventType == "Wheel":
        if hasPosition:
          self._slave_slave.mouseWheel(event["delta"],event["x"],event["y"])
        else:
          self._slave_slave.mouseWheel(event["delta"])
      elif eventType == "KeyPress":
        self._slave_slave.keyPress(event["key"])
      elif eventType == "KeyRelease":
        self._slave_slave.keyRelease(event["key"])
      elif eventType == "Screenshot":
        self.createScreenshotcreateScreenshot()
      else:
        MLAB.logWarning("Unknown event: " + event)
 
      # it was required to use QCoreApplication.processEvents instead of MLAB.processEvent()
      # so that events are reliably delivered to QGraphicsView/RemotePanelRendering:
      QtCore.QCoreApplication.processEvents()
      
      if redrawAfterSceneChanged and self.sceneHasChangedsceneHasChanged:
        shouldRedraw = True
        self.sceneHasChangedsceneHasChanged = False
      if shouldRedraw:
        self._slave_slave.sendRenderRequest(self._slave_slave.PNG)
        MLAB.processEvents()