class ImageScene2D_RenderTest( ut.TestCase ):
@classmethod
def setUpClass(cls):
cls.app = None
if QApplication.instance():
cls.app = QApplication.instance()
else:
cls.app = QApplication([], False)
@classmethod
def tearDownClass(cls):
del cls.app
def setUp( self ):
self.layerstack = LayerStackModel()
self.sims = StackedImageSources( self.layerstack )
self.GRAY = 201
self.ds = ConstantSource(self.GRAY)
self.layer = GrayscaleLayer( self.ds )
self.layer.set_normalize(0, False)
self.layerstack.append(self.layer)
self.ims = imsfac.createImageSource( self.layer, [self.ds] )
self.sims.register(self.layer, self.ims)
self.scene = ImageScene2D(PositionModel(), (0,3,4), preemptive_fetch_number=0)
self.scene.stackedImageSources = self.sims
self.scene.dataShape = (310,290)
def tearDown( self ):
if self.scene._tileProvider:
self.scene._tileProvider.notifyThreadsToStop()
self.scene._tileProvider.joinThreads()
def renderScene( self, s, exportFilename=None):
img = QImage(310,290,QImage.Format_ARGB32_Premultiplied)
img.fill(0)
p = QPainter(img)
s.render(p)
s.joinRendering()
s.render(p)
p.end()
if exportFilename is not None:
img.save(exportFilename)
return byte_view(img)
def testBasicImageRenderingCapability( self ):
aimg = self.renderScene(self.scene)
self.assertTrue(np.all(aimg[:,:,0:3] == self.GRAY))
self.assertTrue(np.all(aimg[:,:,3] == 255))
def testToggleVisibilityOfOneLayer( self ):
aimg = self.renderScene(self.scene)
self.assertTrue(np.all(aimg[:,:,0:3] == self.GRAY))
self.assertTrue(np.all(aimg[:,:,3] == 255))
self.layer.visible = False
aimg = self.renderScene(self.scene)
self.assertTrue(np.all(aimg[:,:,0:3] == 0)) # all white
self.assertTrue(np.all(aimg[:,:,3] == 0))
self.layer.visible = True
aimg = self.renderScene(self.scene)
self.assertTrue(np.all(aimg[:,:,0:3] == self.GRAY))
self.assertTrue(np.all(aimg[:,:,3] == 255))
class ImageScene2D_RenderTest(ut.TestCase):
@classmethod
def setUpClass(cls):
cls.app = None
if QApplication.instance():
cls.app = QApplication.instance()
else:
cls.app = QApplication([])
@classmethod
def tearDownClass(cls):
del cls.app
def setUp(self):
self.layerstack = LayerStackModel()
self.sims = StackedImageSources(self.layerstack)
self.GRAY = 201
self.ds = ConstantSource(self.GRAY)
self.layer = GrayscaleLayer(self.ds)
self.layer.set_normalize(0, False)
self.layerstack.append(self.layer)
self.ims = self.layer.createImageSource([self.ds])
self.sims.register(self.layer, self.ims)
self.scene = ImageScene2D(PositionModel(), (0, 3, 4),
preemptive_fetch_number=0)
self.scene.stackedImageSources = self.sims
self.scene.dataShape = (310, 290)
def renderScene(self, s, exportFilename=None):
img = QImage(310, 290, QImage.Format_ARGB32_Premultiplied)
img.fill(0)
p = QPainter(img)
s.render(p)
s.joinRenderingAllTiles(viewport_only=False)
s.render(p)
p.end()
if exportFilename is not None:
img.save(exportFilename)
return byte_view(img)
def testBasicImageRenderingCapability(self):
aimg = self.renderScene(self.scene)
self.assertTrue(np.all(aimg[:, :, 0:3] == self.GRAY))
self.assertTrue(np.all(aimg[:, :, 3] == 255))
def testToggleVisibilityOfOneLayer(self):
aimg = self.renderScene(self.scene)
self.assertTrue(np.all(aimg[:, :, 0:3] == self.GRAY))
self.assertTrue(np.all(aimg[:, :, 3] == 255))
self.layer.visible = False
aimg = self.renderScene(self.scene)
self.assertTrue(np.all(aimg[:, :, 0:3] == 0)) # all white
self.assertTrue(np.all(aimg[:, :, 3] == 0))
self.layer.visible = True
aimg = self.renderScene(self.scene)
self.assertTrue(np.all(aimg[:, :, 0:3] == self.GRAY))
self.assertTrue(np.all(aimg[:, :, 3] == 255))
