This PR predates #2405 and may need to be updated to reflect the VectorArray interface changes in #2405.

This pull request is for issue #2146, which wants to have an option to store visualization results as image files.

Features

For both PatchVisualizer and OnedVisualizer, the visualization result could be saved as image files by specifying the img_filename argument as calling the function visualize, as shown below:

visualize(U, img_filename="IMG_NAME")

1. Single image output: a single JPG file named with the argument will be stored in the working directory.
2. Multiple image output: a folder named with the argument will be created in the working directory first. The frames will be stored as sequentially numbered JPG files inside the folder (e.g. 0.jpg, 1.jpg, etc.).

Testing

I have tested the code with :

1. The example provided in the issue for PatchVisualizer.
2. The following code for OnedVisualizer:

from pymor.models.examples import heat_equation_1d_example

fom_tb = heat_equation_1d_example()
mu = 0.1
U = fom_tb.solve(mu)
fom_tb.visualize(U, img_filename="1D_test")

If there is any problems or feedbacks, please let me know.