Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F85499889
numpy_engine_base.py
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Sun, Sep 29, 15:05
Size
4 KB
Mime Type
text/x-python
Expires
Tue, Oct 1, 15:05 (2 d)
Engine
blob
Format
Raw Data
Handle
21192614
Attached To
R6746 RationalROMPy
numpy_engine_base.py
View Options
# Copyright (C) 2018 by the RROMPy authors
#
# This file is part of RROMPy.
#
# RROMPy is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# RROMPy is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with RROMPy. If not, see <http://www.gnu.org/licenses/>.
#
import
numpy
as
np
from
matplotlib
import
pyplot
as
plt
from
.hfengine_base
import
HFEngineBase
from
rrompy.utilities.base.types
import
Np1D
,
strLst
,
List
from
rrompy.utilities.base
import
purgeList
,
getNewFilename
__all__
=
[
'NumpyEngineBase'
]
class
NumpyEngineBase
(
HFEngineBase
):
"""Generic solver for parametric matricial problems."""
def
plot
(
self
,
u
:
Np1D
,
warping
:
List
[
callable
]
=
None
,
name
:
str
=
"u"
,
save
:
str
=
None
,
what
:
strLst
=
'all'
,
saveFormat
:
str
=
"eps"
,
saveDPI
:
int
=
100
,
show
:
bool
=
True
,
pyplotArgs
:
dict
=
{},
**
figspecs
)
->
str
:
"""
Do some nice plots of the complex-valued function with given dofs.
Args:
u: numpy complex array with function dofs.
name(optional): Name to be shown as title of the plots. Defaults to
'u'.
save(optional): Where to save plot(s). Defaults to None, i.e. no
saving.
what(optional): Which plots to do. If list, can contain 'ABS',
'PHASE', 'REAL', 'IMAG'. If str, same plus wildcard 'ALL'.
Defaults to 'ALL'.
saveFormat(optional): Format for saved plot(s). Defaults to "eps".
saveDPI(optional): DPI for saved plot(s). Defaults to 100.
show(optional): Whether to show figure. Defaults to True.
pyplotArgs(optional): Optional arguments for pyplot.
figspecs(optional key args): Optional arguments for matplotlib
figure creation.
Returns:
Output filename.
"""
if
isinstance
(
what
,
(
str
,)):
if
what
.
upper
()
==
'ALL'
:
what
=
[
'ABS'
,
'PHASE'
,
'REAL'
,
'IMAG'
]
else
:
what
=
[
what
]
what
=
purgeList
(
what
,
[
'ABS'
,
'PHASE'
,
'REAL'
,
'IMAG'
],
listname
=
self
.
name
()
+
".what"
,
baselevel
=
1
)
if
len
(
what
)
==
0
:
return
if
'figsize'
not
in
figspecs
.
keys
():
figspecs
[
'figsize'
]
=
(
13.
*
len
(
what
)
/
4
,
3
)
subplotcode
=
100
+
len
(
what
)
*
10
idxs
=
np
.
arange
(
self
.
spacedim
)
if
warping
is
not
None
:
idxs
=
warping
[
0
](
np
.
arange
(
self
.
spacedim
))
plt
.
figure
(
**
figspecs
)
plt
.
jet
()
if
'ABS'
in
what
:
subplotcode
=
subplotcode
+
1
plt
.
subplot
(
subplotcode
)
plt
.
plot
(
idxs
,
np
.
abs
(
u
)
.
flatten
(),
**
pyplotArgs
)
plt
.
title
(
"|{0}|"
.
format
(
name
))
if
'PHASE'
in
what
:
subplotcode
=
subplotcode
+
1
plt
.
subplot
(
subplotcode
)
plt
.
plot
(
idxs
,
np
.
angle
(
u
)
.
flatten
(),
**
pyplotArgs
)
plt
.
title
(
"phase({0})"
.
format
(
name
))
if
'REAL'
in
what
:
subplotcode
=
subplotcode
+
1
plt
.
subplot
(
subplotcode
)
plt
.
plot
(
idxs
,
np
.
real
(
u
)
.
flatten
(),
**
pyplotArgs
)
plt
.
title
(
"Re({0})"
.
format
(
name
))
if
'IMAG'
in
what
:
subplotcode
=
subplotcode
+
1
plt
.
subplot
(
subplotcode
)
plt
.
plot
(
idxs
,
np
.
imag
(
u
)
.
flatten
(),
**
pyplotArgs
)
plt
.
title
(
"Im({0})"
.
format
(
name
))
if
save
is
not
None
:
save
=
save
.
strip
()
fileOut
=
getNewFilename
(
"{}_fig_"
.
format
(
save
),
saveFormat
)
plt
.
savefig
(
fileOut
,
format
=
saveFormat
,
dpi
=
saveDPI
)
else
:
fileOut
=
None
if
show
:
plt
.
show
()
plt
.
close
()
return
fileOut
Event Timeline
Log In to Comment