Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Sebastian Steinlechner
pykat
Commits
a8ac6ee3
Commit
a8ac6ee3
authored
Dec 08, 2015
by
Daniel Brown
Browse files
adding in styles argument
parent
9acf1206
Changes
2
Hide whitespace changes
Inline
Side-by-side
bin/test_plot.py
View file @
a8ac6ee3
import
pykat
pykat
.
init_pykat_plotting
()
#
pykat.init_pykat_plotting()
from
pykat
import
finesse
from
pykat.detectors
import
*
...
...
@@ -32,7 +32,7 @@ attr m1 Rc 1
kat
=
finesse
.
kat
()
kat
.
parseCommands
(
code
)
kat
.
add
(
xaxis
(
"l
og
"
,
[
1
,
360
],
kat
.
m2
.
phi
,
500
))
kat
.
add
(
xaxis
(
"l
in
"
,
[
1
,
360
],
kat
.
m2
.
phi
,
500
))
kat
.
m1
.
Rcx
=
-
1000.0
kat
.
m1
.
Rcy
=
-
1000.0
...
...
@@ -43,4 +43,4 @@ kat.maxtem = 0
out
=
kat
.
run
()
fig
=
out
.
plot
()
\ No newline at end of file
fig
=
out
.
plot
(
styles
=
{
'circ'
:
'c:'
},
yaxis
=
"log abs:deg"
)
\ No newline at end of file
pykat/finesse.py
View file @
a8ac6ee3
...
...
@@ -259,7 +259,7 @@ class katRun(object):
print
(
"One xaxis used: %s"
%
kat
.
xaxis
.
getFinesseText
())
def
plot
(
self
,
detectors
=
None
,
filename
=
None
,
show
=
True
,
yaxis
=
None
,
legend
=
True
,
loc
=
0
,
title
=
None
):
def
plot
(
self
,
detectors
=
None
,
filename
=
None
,
show
=
True
,
yaxis
=
None
,
legend
=
True
,
loc
=
0
,
title
=
None
,
styles
=
None
):
"""
This will generate a plot for the output data of this particular pykat run.
It will attempt to generate a plot that shows all the various traces and plots
...
...
@@ -275,6 +275,7 @@ class katRun(object):
legend: True | False - whether to include a legend
loc: Location value for the legend, the usual matplotlib one.
title: Provide a title for the plot if required.
styles: A dictionary which keys being the detector names and the value being a colour and linestyle of the sort 'k:'
"""
import
matplotlib.pyplot
as
pyplot
import
pykat.plotting
as
plt
...
...
@@ -349,12 +350,15 @@ class katRun(object):
for
det
in
detectors
:
if
not
getattr
(
kat
,
det
).
noplot
:
if
not
dual_plot
:
plot_cmd1
(
self
.
x
,
_func1
(
self
[
det
]),
label
=
det
)
else
:
if
dual_plot
:
ax
=
pyplot
.
subplot
(
2
,
1
,
1
)
if
styles
is
not
None
and
det
in
styles
:
l
,
=
plot_cmd1
(
self
.
x
,
_func1
(
self
[
det
]),
styles
[
det
],
label
=
det
)
else
:
l
,
=
plot_cmd1
(
self
.
x
,
_func1
(
self
[
det
]),
label
=
det
)
if
dual_plot
:
pyplot
.
subplot
(
2
,
1
,
2
)
plot_cmd2
(
self
.
x
,
_func2
(
self
[
det
]),
color
=
l
.
get_color
(),
ls
=
l
.
get_linestyle
(),
label
=
det
)
...
...
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