def save_jpg(self, file_name, quality=None, progressive=None):
"""Arguments: file_name if passed will be used, quality is the
quality of the JPEG(10-100) are valid, the progressive
arguments toggles progressive jpegs."""
if len(file_name) != 0:
if not quality and not progressive:
quality, progressive = self.jpeg_quality, self.jpeg_progressive
w2if = self._get_window_to_image()
ex = tvtk.JPEGWriter()
ex.quality = quality
ex.progressive = progressive
ex.file_name = file_name
configure_input(ex, w2if)
self._exporter_write(ex)
def _handle_failed_image(idiff, src_img, pngr, img_fname):
"""Writes all the necessary images when an image comparison
failed."""
f_base, f_ext = os.path.splitext(img_fname)
# write out the difference file in full.
pngw = tvtk.PNGWriter(file_name=f_base + ".diff.png")
configure_input(pngw, idiff)
pngw.write()
# write the difference image scaled and gamma adjusted for the
# dashboard.
sz = pngr.output.dimensions
if sz[1] <= 250.0:
mag = 1.0
else:
mag = 250.0/sz[1]
shrink = tvtk.ImageResample(interpolate=1)
configure_input(shrink, idiff.output)
shrink.set_axis_magnification_factor(0, mag)
shrink.set_axis_magnification_factor(1, mag)
gamma = tvtk.ImageShiftScale(shift=0, scale=10)
configure_input(gamma, shrink)
jpegw = tvtk.JPEGWriter(file_name=f_base + ".diff.small.jpg",
quality=85)
configure_input(jpegw, gamma)
jpegw.write()
# write out the image that was generated.
pngw.trait_set(file_name=f_base + ".test.png")
configure_input(pngw, src_img)
pngw.write()
# write out a smaller version of the image that was generated.
configure_input(shrink, idiff.input)
jpegw.trait_set(file_name=f_base + ".test.small.jpg")
configure_input(jpegw, shrink)
jpegw.write()
# write out the valid image that matched.
configure_input(shrink, idiff.image)
jpegw.trait_set(file_name=f_base + ".small.jpg")
configure_input(jpegw, shrink)
jpegw.write()
def save_jpg(self, file_name, quality=None, progressive=None):
"""Arguments: file_name if passed will be used, quality is the
quality of the JPEG(10-100) are valid, the progressive
arguments toggles progressive jpegs."""
if len(file_name) != 0:
if not quality and not progressive:
quality, progressive = self.jpeg_quality, self.jpeg_progressive
w2if = tvtk.WindowToImageFilter(read_front_buffer=True)
w2if.magnification = self.magnification
self._lift()
w2if.input = self._renwin
ex = tvtk.JPEGWriter()
ex.quality = quality
ex.progressive = progressive
ex.file_name = file_name
ex.input = w2if.output
self._exporter_write(ex)
def __jpg_writer_default(self):
return tvtk.JPEGWriter(quality=self.quality, write_to_memory=True)