Skip to content
GitLab
Projects
Groups
Snippets
Help
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
P
pymor
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Requirements
Requirements
List
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Operations
Operations
Environments
Packages & Registries
Packages & Registries
Container Registry
Analytics
Analytics
CI / CD
Insights
Issue
Repository
Value Stream
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Jobs
Commits
Open sidebar
pyMOR
pymor
Commits
4195a51c
Commit
4195a51c
authored
Jun 23, 2020
by
René Fritze
Committed by
René Fritze
Jul 22, 2020
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[gui] prototype jupyter matplotlib output for 1D data
parent
a0c8bae8
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
69 additions
and
8 deletions
+69
-8
src/pymor/discretizers/builtin/gui/jupyter/patch.py
src/pymor/discretizers/builtin/gui/jupyter/patch.py
+7
-3
src/pymor/discretizers/builtin/gui/matplotlib.py
src/pymor/discretizers/builtin/gui/matplotlib.py
+46
-0
src/pymor/discretizers/builtin/gui/visualizers.py
src/pymor/discretizers/builtin/gui/visualizers.py
+16
-5
No files found.
src/pymor/discretizers/builtin/gui/jupyter/patch.py
View file @
4195a51c
...
...
@@ -4,7 +4,7 @@
import
numpy
as
np
from
pymor.core.config
import
config
from
pymor.discretizers.builtin.gui.matplotlib
import
MatplotlibPatchAxes
from
pymor.discretizers.builtin.gui.matplotlib
import
MatplotlibPatchAxes
,
Matplotlib1DAxes
from
pymor.vectorarrays.interface
import
VectorArray
...
...
@@ -88,8 +88,12 @@ def visualize_patch(grid, U, bounding_box=([0, 0], [1, 1]), codim=2, title=None,
for
i
,
(
vmin
,
vmax
)
in
enumerate
(
zip
(
self
.
vmins
,
self
.
vmaxs
)):
ax
=
figure
.
add_subplot
(
rows
,
columns
,
i
+
1
)
axes
.
append
(
ax
)
plots
.
append
(
MatplotlibPatchAxes
(
figure
,
grid
,
bounding_box
=
bounding_box
,
vmin
=
vmin
,
vmax
=
vmax
,
codim
=
codim
,
colorbar
=
separate_colorbars
))
if
grid
.
dim
==
2
:
plots
.
append
(
MatplotlibPatchAxes
(
figure
,
grid
,
bounding_box
=
bounding_box
,
vmin
=
vmin
,
vmax
=
vmax
,
codim
=
codim
,
colorbar
=
separate_colorbars
))
else
:
plots
.
append
(
Matplotlib1DAxes
(
figure
,
grid
,
bounding_box
=
bounding_box
,
vmin
=
vmin
,
vmax
=
vmax
,
codim
=
codim
,
colorbar
=
separate_colorbars
))
if
legend
:
ax
.
set_title
(
legend
[
i
])
...
...
src/pymor/discretizers/builtin/gui/matplotlib.py
View file @
4195a51c
...
...
@@ -57,6 +57,52 @@ class MatplotlibPatchAxes:
p
.
set_clim
(
self
.
vmin
,
self
.
vmax
)
class
Matplotlib1DAxes
:
def
__init__
(
self
,
figure
,
grid
,
bounding_box
=
None
,
vmin
=
None
,
vmax
=
None
,
codim
=
1
,
colorbar
=
True
):
assert
isinstance
(
grid
,
OnedGrid
)
assert
codim
in
(
0
,
1
)
self
.
codim
=
codim
self
.
grid
=
grid
centers
=
grid
.
centers
(
1
)
if
grid
.
identify_left_right
:
centers
=
np
.
concatenate
((
centers
,
[[
grid
.
domain
[
1
]]]),
axis
=
0
)
self
.
periodic
=
True
else
:
self
.
periodic
=
False
if
codim
==
1
:
xs
=
centers
else
:
xs
=
np
.
repeat
(
centers
,
2
)[
1
:
-
1
]
a
=
figure
.
gca
()
self
.
lines
,
=
a
.
plot
(
xs
,
np
.
zeros_like
(
xs
))
# TODO
import
matplotlib.pyplot
as
plt
self
.
p
=
plt
.
cm
.
ScalarMappable
(
cmap
=
plt
.
get_cmap
(
"viridis"
),
norm
=
plt
.
Normalize
(
vmin
=
vmin
or
0
,
vmax
=
vmax
or
1
))
if
colorbar
:
figure
.
colorbar
(
self
.
p
,
ax
=
a
)
def
set
(
self
,
U
,
vmin
=
None
,
vmax
=
None
):
self
.
vmin
=
self
.
vmin
if
vmin
is
None
else
vmin
self
.
vmax
=
self
.
vmax
if
vmax
is
None
else
vmax
u
=
np
.
array
(
U
)
if
self
.
codim
==
1
:
if
self
.
periodic
:
self
.
lines
.
set_ydata
(
np
.
concatenate
((
u
,
[
u
[
0
]])))
else
:
self
.
lines
.
set_ydata
(
u
)
else
:
self
.
lines
.
set_ydata
(
np
.
repeat
(
u
,
2
))
self
.
p
.
set_clim
(
self
.
vmin
,
self
.
vmax
)
if
config
.
HAVE_QT
and
config
.
HAVE_MATPLOTLIB
:
from
Qt.QtWidgets
import
QSizePolicy
...
...
src/pymor/discretizers/builtin/gui/visualizers.py
View file @
4195a51c
...
...
@@ -111,12 +111,14 @@ class OnedVisualizer(BasicObject):
If `True`, block execution until the plot window is closed.
"""
def
__init__
(
self
,
grid
,
codim
=
1
,
block
=
False
):
def
__init__
(
self
,
grid
,
codim
=
1
,
block
=
False
,
backend
=
None
):
assert
isinstance
(
grid
,
OnedGrid
)
assert
codim
in
(
0
,
1
)
backend
=
backend
or
(
'jupyter'
if
is_jupyter
()
else
None
)
self
.
__auto_init
(
locals
())
def
visualize
(
self
,
U
,
m
,
title
=
None
,
legend
=
None
,
block
=
None
):
def
visualize
(
self
,
U
,
m
,
title
=
None
,
legend
=
None
,
separate_colorbars
=
False
,
rescale_colorbars
=
False
,
block
=
None
,
filename
=
None
,
columns
=
2
):
"""Visualize the provided data.
Parameters
...
...
@@ -137,6 +139,15 @@ class OnedVisualizer(BasicObject):
If `True`, block execution until the plot window is closed. If `None`, use the
default provided during instantiation.
"""
block
=
self
.
block
if
block
is
None
else
block
from
pymor.discretizers.builtin.gui.qt
import
visualize_matplotlib_1d
visualize_matplotlib_1d
(
self
.
grid
,
U
,
codim
=
self
.
codim
,
title
=
title
,
legend
=
legend
,
block
=
block
)
if
filename
is
not
None
:
raise
NotImplementedError
if
self
.
backend
==
'jupyter'
:
from
pymor.discretizers.builtin.gui.jupyter
import
get_visualizer
return
get_visualizer
()(
self
.
grid
,
U
,
bounding_box
=
self
.
grid
.
domain
,
codim
=
self
.
codim
,
title
=
title
,
legend
=
legend
,
separate_colorbars
=
separate_colorbars
,
rescale_colorbars
=
rescale_colorbars
,
columns
=
columns
)
else
:
block
=
self
.
block
if
block
is
None
else
block
from
pymor.discretizers.builtin.gui.qt
import
visualize_matplotlib_1d
return
visualize_matplotlib_1d
(
self
.
grid
,
U
,
codim
=
self
.
codim
,
title
=
title
,
legend
=
legend
,
block
=
block
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment