Skip to content
Snippets Groups Projects
Commit a8ac6ee3 authored by Daniel Brown's avatar Daniel Brown
Browse files

adding in styles argument

parent 9acf1206
No related branches found
No related tags found
No related merge requests found
import pykat import pykat
pykat.init_pykat_plotting() #pykat.init_pykat_plotting()
from pykat import finesse from pykat import finesse
from pykat.detectors import * from pykat.detectors import *
...@@ -32,7 +32,7 @@ attr m1 Rc 1 ...@@ -32,7 +32,7 @@ attr m1 Rc 1
kat = finesse.kat() kat = finesse.kat()
kat.parseCommands(code) kat.parseCommands(code)
kat.add(xaxis("log", [1, 360], kat.m2.phi, 500)) kat.add(xaxis("lin", [1, 360], kat.m2.phi, 500))
kat.m1.Rcx = -1000.0 kat.m1.Rcx = -1000.0
kat.m1.Rcy = -1000.0 kat.m1.Rcy = -1000.0
...@@ -43,4 +43,4 @@ kat.maxtem = 0 ...@@ -43,4 +43,4 @@ kat.maxtem = 0
out = kat.run() out = kat.run()
fig = out.plot() fig = out.plot(styles={'circ':'c:'}, yaxis="log abs:deg")
\ No newline at end of file \ No newline at end of file
...@@ -259,7 +259,7 @@ class katRun(object): ...@@ -259,7 +259,7 @@ class katRun(object):
print("One xaxis used: %s" % kat.xaxis.getFinesseText()) 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. 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 It will attempt to generate a plot that shows all the various traces and plots
...@@ -275,6 +275,7 @@ class katRun(object): ...@@ -275,6 +275,7 @@ class katRun(object):
legend: True | False - whether to include a legend legend: True | False - whether to include a legend
loc: Location value for the legend, the usual matplotlib one. loc: Location value for the legend, the usual matplotlib one.
title: Provide a title for the plot if required. 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 matplotlib.pyplot as pyplot
import pykat.plotting as plt import pykat.plotting as plt
...@@ -349,12 +350,15 @@ class katRun(object): ...@@ -349,12 +350,15 @@ class katRun(object):
for det in detectors: for det in detectors:
if not getattr(kat, det).noplot: if not getattr(kat, det).noplot:
if not dual_plot: if dual_plot:
plot_cmd1(self.x, _func1(self[det]), label=det)
else:
ax = pyplot.subplot(2,1,1) 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) l, = plot_cmd1(self.x, _func1(self[det]), label=det)
if dual_plot:
pyplot.subplot(2,1,2) pyplot.subplot(2,1,2)
plot_cmd2(self.x, _func2(self[det]), color=l.get_color(), ls=l.get_linestyle(), label=det) plot_cmd2(self.x, _func2(self[det]), color=l.get_color(), ls=l.get_linestyle(), label=det)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment