mikael.capelle/pyltk
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add possibility to specify custom options for pgfplots elements.

Browse Source
This commit is contained in:
Mikael Capelle 2018-03-16 12:29:15 +01:00
parent 758e99d5a7
commit 8a8cb4d177
4 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pyltk/pgfplots/axis.py
View File

@ -15,9 +15,9 @@ class axis(element):
\\end{{axis}}""" \\end{{axis}}"""
} }
def __init__(self, *args, parent=None, legend=True, label=None, **kargs): def __init__(self, *args, parent=None, legend=True, label=None, **kwargs):
super().__init__(parent, label=label) super().__init__(parent, label=label)
self.options = kargs self.options = kwargs
self.legend = legend self.legend = legend
for arg in args: for arg in args:
self.add(arg) self.add(arg)

3
pyltk/pgfplots/coordinates_plot.py
View File

@ -11,5 +11,6 @@ class coordinates_plot(generic_plot):
def format_data(self, data): def format_data(self, data):
pts = [] pts = []
for d in data: for d in data:
pts.append('({}, {})'.format(*d)) pts.append('({}, {})'.format(
*self.format_values(d)))
return 'coordinates {{{}}}'.format(''.join(pts)) return 'coordinates {{{}}}'.format(''.join(pts))

5
pyltk/pgfplots/errorbars_plot.py
View File

@ -5,7 +5,7 @@ from .generic_plot import generic_plot
class errorbars_plot(generic_plot): class errorbars_plot(generic_plot):
def __init__(self, legend, data, label=None, options=[]): def __init__(self, legend, data, precision=5, label=None, options=[]):
options = options + [ options = options + [
'error bars/.cd', ('y dir', 'both'), ('y explicit')] 'error bars/.cd', ('y dir', 'both'), ('y explicit')]
super().__init__(legend, data, label=label, options=options) super().__init__(legend, data, label=label, options=options)
@ -20,5 +20,6 @@ class errorbars_plot(generic_plot):
p = d[2] p = d[2]
n = p n = p
rows.append('({x}, {y}) += (0, {p}) -= (0, {n})'.format( rows.append('({x}, {y}) += (0, {p}) -= (0, {n})'.format(
x=x, y=y, p=p, n=n)) x=self.format_value(x), y=self.format_value(y),
p=self.format_value(p), n=self.format_value(n)))
return 'coordinates {{{}}}'.format('\n'.join(rows)) return 'coordinates {{{}}}'.format('\n'.join(rows))

9
pyltk/pgfplots/generic_plot.py
View File

@ -11,11 +11,12 @@ class generic_plot(element):
'content': """\\addplot{options} {data};""" 'content': """\\addplot{options} {data};"""
} }
def __init__(self, legend, data, label=None, options=[]): def __init__(self, legend, data, precision=6, label=None, options=[]):
super().__init__(label=label) super().__init__(label=label)
self.legend = legend self.legend = legend
self.data = data self.data = data
self.options = options self.options = options
self.precision = precision
def format_options(self, options): def format_options(self, options):
if not options: if not options:
@ -29,6 +30,12 @@ class generic_plot(element):
'content': ', '.join(opts) 'content': ', '.join(opts)
}) })
def format_value(self, value):
return '{:.{prec}f}'.format(value, prec=self.precision)
def format_values(self, values):
return [self.format_value(value) for value in values]
def content(self): def content(self):
return self.fmt().format('content', { return self.fmt().format('content', {
'data': self.format_data(self.data), 'data': self.format_data(self.data),