Update.
This commit is contained in:
parent
c489ce7d1e
commit
6599ccb39f
@ -2,9 +2,13 @@
|
||||
|
||||
# flake8: noqa
|
||||
|
||||
from .documentclass import *
|
||||
from .figure import figure
|
||||
from .formatter import formatter
|
||||
from .input_element import input_element
|
||||
from .makebox import makebox
|
||||
from .resizebox import resizebox
|
||||
from .subfloat import subfloat
|
||||
from .table import table, tabledf
|
||||
from .tabular import tabular
|
||||
from .inlines import it, bf, mt
|
||||
|
88
pyltk/documentclass.py
Normal file
88
pyltk/documentclass.py
Normal file
@ -0,0 +1,88 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from collections import Iterable
|
||||
from .element import element
|
||||
|
||||
import subprocess
|
||||
|
||||
|
||||
class documentclass(element):
|
||||
|
||||
packages = {}
|
||||
preamble = []
|
||||
options = []
|
||||
classname = 'standalone'
|
||||
|
||||
templates = {
|
||||
'package': '\\usepackage{pkgoptions}{{{pkgname}}}',
|
||||
'options': '[{content}]',
|
||||
'def': '\\def\\{name}#{nargs}{{{value}}}',
|
||||
'content': """\\documentclass{options}{{{classname}}}
|
||||
{packages}
|
||||
{preamble}
|
||||
\\begin{{document}}
|
||||
{content}
|
||||
\\end{{document}}
|
||||
"""
|
||||
}
|
||||
|
||||
def __init__(self, classname, childrens=[], options=[], packages=[]):
|
||||
super().__init__(parent=None, childrens=childrens)
|
||||
self.options = options
|
||||
self.classname = classname
|
||||
self.add_packages(packages)
|
||||
|
||||
def add_packages(self, packages):
|
||||
if type(packages) is str:
|
||||
self.packages[packages] = True
|
||||
elif type(packages) is dict:
|
||||
self.packages.update(packages)
|
||||
else:
|
||||
self.packages.update({p: True for p in packages})
|
||||
|
||||
def add2preamble(self, value):
|
||||
if isinstance(value, Iterable):
|
||||
self.preamble.extend(value)
|
||||
else:
|
||||
self.preamble.append(value)
|
||||
|
||||
def add_def(self, name, nargs, value):
|
||||
self.preamble.append(self.fmt().format('def', {
|
||||
'name': name,
|
||||
'nargs': nargs,
|
||||
'value': value
|
||||
}))
|
||||
|
||||
def format_preamble(self, defs):
|
||||
return '\n'.join(self.preamble)
|
||||
|
||||
def format_packages(self, pkgs):
|
||||
out = []
|
||||
for pkg, opts in self.packages.items():
|
||||
options = ''
|
||||
if opts is not True:
|
||||
options = self.format_options(opts)
|
||||
out.append(self.fmt().format('package', {
|
||||
'pkgname': pkg,
|
||||
'pkgoptions': options
|
||||
}))
|
||||
return '\n'.join(out)
|
||||
|
||||
def content(self):
|
||||
return self.fmt().format('content', {
|
||||
'options': self.format_options(self.options),
|
||||
'packages': self.format_packages(self.packages),
|
||||
'preamble': self.format_preamble(self.preamble),
|
||||
'classname': self.classname,
|
||||
'content': super().content()
|
||||
})
|
||||
|
||||
def compile(self, outfile, infile=None, outlog=None):
|
||||
if infile is None:
|
||||
infile = '/tmp/pyltk-{}.tex'.format(id(self))
|
||||
|
||||
with open(infile, 'w') as infd:
|
||||
infd.write(self.content())
|
||||
|
||||
call = ['pdflatex', '-halt-on-error', '-jobname', outfile, infile]
|
||||
subprocess.call(call, stdout=outlog)
|
85
pyltk/element.py
Normal file
85
pyltk/element.py
Normal file
@ -0,0 +1,85 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .formatter import formatter
|
||||
|
||||
from collections import Iterable
|
||||
|
||||
|
||||
class element:
|
||||
|
||||
""" Class representing a latex element. """
|
||||
|
||||
# Joiner for childrens
|
||||
sep = '\n'
|
||||
|
||||
# Automatically add label after element if present
|
||||
autolabel = True
|
||||
|
||||
def __init__(self, parent=None, childrens=[], label=None):
|
||||
self.parent = parent
|
||||
if childrens is None:
|
||||
childrens = []
|
||||
if not isinstance(childrens, Iterable):
|
||||
childrens = [childrens]
|
||||
self.childrens = []
|
||||
self.add(childrens)
|
||||
self.label = label
|
||||
|
||||
def add(self, childrens):
|
||||
if not isinstance(childrens, Iterable):
|
||||
childrens = [childrens]
|
||||
for child in childrens:
|
||||
if not isinstance(child, element):
|
||||
child = wrapper_element(child, self)
|
||||
try:
|
||||
child.parent.remove(child)
|
||||
except:
|
||||
pass
|
||||
child.parent = self
|
||||
self.childrens.append(child)
|
||||
|
||||
def content(self):
|
||||
return self.sep.join(map(str, self.childrens))
|
||||
|
||||
def fmt(self):
|
||||
return formatter(self.templates)
|
||||
|
||||
def format_options(self, options):
|
||||
if not options:
|
||||
return ''
|
||||
opts = []
|
||||
for k, v in options.items():
|
||||
opts.append('{key} = {{{value}}}'.format(key=k, value=v))
|
||||
return self.fmt().format('options', {
|
||||
'content': ', '.join(opts)
|
||||
})
|
||||
|
||||
def format_label(self, label):
|
||||
if not label:
|
||||
return ''
|
||||
return '\\label{{{}}}'.format(label)
|
||||
|
||||
def __str__(self):
|
||||
out = self.content().strip()
|
||||
if self.autolabel:
|
||||
out += self.format_label(self.label)
|
||||
return out
|
||||
|
||||
|
||||
class wrapper_element(element):
|
||||
|
||||
""" Wrapper for standard python types working as latex elements. """
|
||||
|
||||
escape_list = ['_', '#', '%']
|
||||
|
||||
def __init__(self, data, parent):
|
||||
super().__init__(parent=parent)
|
||||
self.data = data
|
||||
|
||||
def escape(self, s):
|
||||
for e in self.escape_list:
|
||||
s = s.replace(e, '\\' + e)
|
||||
return s
|
||||
|
||||
def content(self):
|
||||
return self.escape(str(self.data))
|
@ -1,34 +1,46 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .formatter import formatter
|
||||
from .element import element
|
||||
|
||||
|
||||
class figure:
|
||||
class figure(element):
|
||||
""" Class representing a latex figure. """
|
||||
|
||||
fmt = formatter({
|
||||
templates = {
|
||||
'caption': '\\caption{{{content}}}',
|
||||
'content': """\\begin{{figure}}[{options}]
|
||||
{centered}
|
||||
{inner}
|
||||
{caption}{label}
|
||||
\\end{{figure}}"""
|
||||
})
|
||||
}
|
||||
|
||||
def __init__(self, inner, options='!h', centered=True):
|
||||
autolabel = False
|
||||
|
||||
def __init__(self, childrens=None, parent=None,
|
||||
caption=None, options='!h',
|
||||
centered=True, label=None):
|
||||
""" Create a figure with given options.
|
||||
|
||||
Parameters:
|
||||
- options Options for the figure.
|
||||
- centered Set to true to add a centering command.
|
||||
"""
|
||||
self.inner = inner
|
||||
super().__init__(parent, childrens, label=label)
|
||||
self.options = options
|
||||
self.caption = caption
|
||||
self.centered = ''
|
||||
if centered:
|
||||
self.centered = '\\centering'
|
||||
|
||||
def __str__(self):
|
||||
return self.fmt.format('content', {
|
||||
def content(self):
|
||||
caption = ''
|
||||
if self.caption:
|
||||
caption = self.fmt().format('caption', {'content': self.caption})
|
||||
return self.fmt().format('content', {
|
||||
'options': self.options,
|
||||
'centered': self.centered,
|
||||
'inner': self.inner
|
||||
'caption': caption,
|
||||
'label': self.format_label(self.label),
|
||||
'inner': super().content()
|
||||
})
|
||||
|
33
pyltk/inlines.py
Normal file
33
pyltk/inlines.py
Normal file
@ -0,0 +1,33 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .element import element
|
||||
|
||||
|
||||
class inline_element(element):
|
||||
|
||||
""" Class representing simple latex inline elements such as bf,
|
||||
it, math. """
|
||||
|
||||
def __init__(self, content, parent=None):
|
||||
super().__init__(parent=parent)
|
||||
self.value = content
|
||||
self.templates = {
|
||||
'wrapper': self.wrapper
|
||||
}
|
||||
|
||||
def content(self):
|
||||
return self.fmt().format('wrapper', {
|
||||
'value': self.value
|
||||
})
|
||||
|
||||
|
||||
class it(inline_element):
|
||||
wrapper = '\\textit{{{value}}}'
|
||||
|
||||
|
||||
class bf(inline_element):
|
||||
wrapper = '\\textbf{{{value}}}'
|
||||
|
||||
|
||||
class mt(inline_element):
|
||||
wrapper = '${value}$'
|
19
pyltk/input_element.py
Normal file
19
pyltk/input_element.py
Normal file
@ -0,0 +1,19 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .element import element
|
||||
|
||||
|
||||
class input_element(element):
|
||||
|
||||
templates = {
|
||||
'element': '\\input{{{filename}}}'
|
||||
}
|
||||
|
||||
def __init__(self, filename, parent=None):
|
||||
super().__init__(parent)
|
||||
self.filename = filename.replace('.tex', '')
|
||||
|
||||
def content(self):
|
||||
return self.fmt().format('element', {
|
||||
'filename': self.filename
|
||||
})
|
@ -1,18 +1,18 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .formatter import formatter
|
||||
from .element import element
|
||||
|
||||
|
||||
class makebox:
|
||||
class makebox(element):
|
||||
""" Class representing an unindented makebox. """
|
||||
|
||||
fmt = formatter({
|
||||
templates = {
|
||||
'content': """{noindent}\\makebox[{width}]{{
|
||||
{inner}
|
||||
}}"""
|
||||
})
|
||||
}
|
||||
|
||||
def __init__(self, inner, width=None, noindent=True):
|
||||
def __init__(self, content=None, parent=None, width=None, noindent=True):
|
||||
""" Create a new makebox with given width.
|
||||
|
||||
Parameters:
|
||||
@ -21,7 +21,7 @@ class makebox:
|
||||
\\textwidth.
|
||||
- noindent Add a \\noindent command before the box.
|
||||
"""
|
||||
self.inner = inner
|
||||
super().__init__(parent, content)
|
||||
|
||||
self.width = width
|
||||
if self.width is None:
|
||||
@ -33,9 +33,9 @@ class makebox:
|
||||
if self.noindent:
|
||||
self.noindent = '\\noindent'
|
||||
|
||||
def __str__(self):
|
||||
return self.fmt.format('content', {
|
||||
'inner': self.inner,
|
||||
def content(self):
|
||||
return self.fmt().format('content', {
|
||||
'inner': super().content(),
|
||||
'width': self.width,
|
||||
'noindent': self.noindent
|
||||
})
|
||||
|
7
pyltk/pgfplots/__init__.py
Normal file
7
pyltk/pgfplots/__init__.py
Normal file
@ -0,0 +1,7 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
# flake8: noqa
|
||||
|
||||
from .axis import axis
|
||||
from .semilogaxis import *
|
||||
from .plots import *
|
39
pyltk/pgfplots/axis.py
Normal file
39
pyltk/pgfplots/axis.py
Normal file
@ -0,0 +1,39 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from ..element import element
|
||||
|
||||
|
||||
class axis(element):
|
||||
""" Class representing an axis. """
|
||||
|
||||
templates = {
|
||||
'options': '[{content}]',
|
||||
'legends': '\\legend{{{content}}}',
|
||||
'content': """\\begin{{axis}}{options}
|
||||
{inner}
|
||||
{legend}
|
||||
\\end{{axis}}"""
|
||||
}
|
||||
|
||||
def __init__(self, *args, parent=None, legend=True, label=None, **kargs):
|
||||
super().__init__(parent, label=label)
|
||||
self.options = kargs
|
||||
self.legend = legend
|
||||
for arg in args:
|
||||
self.add(arg)
|
||||
|
||||
def content(self):
|
||||
leg = ''
|
||||
if self.legend:
|
||||
if type(self.legend) is str:
|
||||
leg = self.legend
|
||||
else:
|
||||
leg = self.fmt().format('legends', {
|
||||
'content': ', '.join('{{{}}}'.format(p.legend)
|
||||
for p in self.childrens)
|
||||
})
|
||||
return self.fmt().format('content', {
|
||||
'options': self.format_options(self.options),
|
||||
'inner': super().content(),
|
||||
'legend': leg
|
||||
})
|
@ -5,10 +5,10 @@ from .generic_plot import generic_plot
|
||||
|
||||
class errorbars_plot(generic_plot):
|
||||
|
||||
def __init__(self, legend, data, options=[]):
|
||||
def __init__(self, legend, data, label=None, options=[]):
|
||||
options = options + [
|
||||
'error bars/.cd', ('y dir', 'both'), ('y explicit')]
|
||||
super().__init__(legend, data, options)
|
||||
super().__init__(legend, data, label=label, options=options)
|
||||
|
||||
def format_data(self, data):
|
||||
rows = []
|
@ -1,17 +1,18 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from ..formatter import formatter
|
||||
from ..element import element
|
||||
|
||||
|
||||
class generic_plot:
|
||||
class generic_plot(element):
|
||||
""" Class representing a pgfplots plot. """
|
||||
|
||||
fmt = formatter({
|
||||
templates = {
|
||||
'options': '+[{content}]',
|
||||
'content': """\\addplot{options} {data};"""
|
||||
})
|
||||
}
|
||||
|
||||
def __init__(self, legend, data, options=[]):
|
||||
def __init__(self, legend, data, label=None, options=[]):
|
||||
super().__init__(label=label)
|
||||
self.legend = legend
|
||||
self.data = data
|
||||
self.options = options
|
||||
@ -24,12 +25,12 @@ class generic_plot:
|
||||
if type(opt) is not str:
|
||||
opt = '{}={}'.format(*opt)
|
||||
opts.append(str(opt))
|
||||
return self.fmt.format('options', {
|
||||
return self.fmt().format('options', {
|
||||
'content': ', '.join(opts)
|
||||
})
|
||||
|
||||
def __str__(self):
|
||||
return self.fmt.format('content', {
|
||||
def content(self):
|
||||
return self.fmt().format('content', {
|
||||
'data': self.format_data(self.data),
|
||||
'options': self.format_options(self.options)
|
||||
})
|
35
pyltk/pgfplots/semilogaxis.py
Normal file
35
pyltk/pgfplots/semilogaxis.py
Normal file
@ -0,0 +1,35 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .axis import axis
|
||||
|
||||
|
||||
class semilogyaxis(axis):
|
||||
""" Class representing an axis. """
|
||||
|
||||
templates = {
|
||||
'options': '[{content}]',
|
||||
'legends': '\\legend{{{content}}}',
|
||||
'content': """\\begin{{semilogyaxis}}{options}
|
||||
{inner}
|
||||
{legend}
|
||||
\\end{{semilogyaxis}}"""
|
||||
}
|
||||
|
||||
def __init__(self, *args, **kargs):
|
||||
super().__init__(*args, **kargs)
|
||||
|
||||
|
||||
class semilogxaxis(axis):
|
||||
""" Class representing an axis. """
|
||||
|
||||
templates = {
|
||||
'options': '[{content}]',
|
||||
'legends': '\\legend{{{content}}}',
|
||||
'content': """\\begin{{semilogxaxis}}{options}
|
||||
{inner}
|
||||
{legend}
|
||||
\\end{{semilogxaxis}}"""
|
||||
}
|
||||
|
||||
def __init__(self, *args, **kargs):
|
||||
super().__init__(*args, **kargs)
|
@ -1,18 +1,19 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .formatter import formatter
|
||||
from .element import element
|
||||
|
||||
|
||||
class resizebox:
|
||||
class resizebox(element):
|
||||
""" Class representing a latex resizebox. """
|
||||
|
||||
fmt = formatter({
|
||||
templates = {
|
||||
'content': """\\resizebox{{{width}}}{{{height}}}{{
|
||||
{inner}
|
||||
}}"""
|
||||
})
|
||||
}
|
||||
|
||||
def __init__(self, inner, width, height="!"):
|
||||
def __init__(self, content=None, parent=None,
|
||||
width="\\linewidth", height="!"):
|
||||
""" Create a resizebox with given width and height.
|
||||
|
||||
Parameters:
|
||||
@ -24,15 +25,15 @@ class resizebox:
|
||||
- str Use as it is (e.g. "0.5\\linewidth", "!").
|
||||
- others Use as a percentage of linewidth (e.g. 0.5).
|
||||
"""
|
||||
self.inner = inner
|
||||
super().__init__(parent, content)
|
||||
self.width = width
|
||||
if type(width) != 'str':
|
||||
if type(width) is not str:
|
||||
self.width = '{}\\linewidth'.format(width)
|
||||
self.height = height
|
||||
|
||||
def __str__(self):
|
||||
return self.fmt.format('content', {
|
||||
def content(self):
|
||||
return self.fmt().format('content', {
|
||||
'width': self.width,
|
||||
'height': self.height,
|
||||
'inner': self.inner
|
||||
'inner': super().content()
|
||||
})
|
||||
|
12
pyltk/strelement.py
Normal file
12
pyltk/strelement.py
Normal file
@ -0,0 +1,12 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .element import element
|
||||
|
||||
|
||||
def strelement(element):
|
||||
|
||||
def __init__(self, content, parent=None):
|
||||
self.inner = content
|
||||
|
||||
def __str__(self):
|
||||
return str(self.inner)
|
@ -1,23 +1,23 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .formatter import formatter
|
||||
from .element import element
|
||||
|
||||
|
||||
class subfloat:
|
||||
class subfloat(element):
|
||||
""" Class representing a latex subfloat. """
|
||||
|
||||
fmt = formatter({
|
||||
templates = {
|
||||
'content': """\\subfloat[{caption}]{{
|
||||
{inner}
|
||||
}}"""
|
||||
})
|
||||
}
|
||||
|
||||
def __init__(self, inner, caption):
|
||||
def __init__(self, caption, parent=None, childrens=None, label=None):
|
||||
super().__init__(parent, childrens, label=label)
|
||||
self.caption = caption
|
||||
self.inner = inner
|
||||
|
||||
def __str__(self):
|
||||
return self.fmt.format('content', {
|
||||
def content(self):
|
||||
return self.fmt().format('content', {
|
||||
'caption': self.caption,
|
||||
'inner': self.inner
|
||||
'inner': super().content()
|
||||
})
|
||||
|
60
pyltk/table.py
Normal file
60
pyltk/table.py
Normal file
@ -0,0 +1,60 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .element import element
|
||||
from .tabular import tabular
|
||||
|
||||
|
||||
class table(element):
|
||||
""" Class representing a latex figure. """
|
||||
|
||||
templates = {
|
||||
'caption': '\\caption{{{content}}}',
|
||||
'content': """\\begin{{table}}[{options}]
|
||||
{centered}
|
||||
{inner}
|
||||
{caption}{label}
|
||||
\\end{{table}}"""
|
||||
}
|
||||
|
||||
autolabel = False
|
||||
|
||||
def __init__(self, childrens=None, parent=None,
|
||||
caption=None, options='!ht',
|
||||
centered=True, label=None):
|
||||
""" Create a table with given options.
|
||||
|
||||
Parameters:
|
||||
- options Options for the table.
|
||||
- centered Set to true to add a centering command.
|
||||
"""
|
||||
super().__init__(parent, childrens, label=label)
|
||||
self.options = options
|
||||
self.caption = caption
|
||||
self.centered = ''
|
||||
if centered:
|
||||
self.centered = '\\centering'
|
||||
|
||||
def content(self):
|
||||
caption = ''
|
||||
if self.caption:
|
||||
caption = self.fmt().format('caption', {'content': self.caption})
|
||||
return self.fmt().format('content', {
|
||||
'options': self.options,
|
||||
'centered': self.centered,
|
||||
'caption': caption,
|
||||
'label': self.format_label(self.label),
|
||||
'inner': super().content()
|
||||
})
|
||||
|
||||
|
||||
class tabledf(table):
|
||||
|
||||
def __init__(self, df, header=None, **kargs):
|
||||
super().__init__(**kargs)
|
||||
if header is None:
|
||||
header = df.columns
|
||||
tab = tabular(columns='r' * len(header))
|
||||
tab.addrow(header)
|
||||
tab.addhline()
|
||||
tab.addrows(df.as_matrix())
|
||||
self.add(tab)
|
@ -1,53 +1,94 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from .formatter import formatter
|
||||
from .element import element
|
||||
|
||||
|
||||
class tabular:
|
||||
class row_element(element):
|
||||
|
||||
""" Class representing a row of a tabular element. """
|
||||
|
||||
def __init__(self, columns, parent, label=None,
|
||||
wrapper=False, endline=False):
|
||||
super().__init__(parent, columns, label=label)
|
||||
self.wrapper = wrapper
|
||||
self.endline = endline
|
||||
|
||||
def content(self):
|
||||
wp = str
|
||||
if self.wrapper:
|
||||
cw = 1/len(self.childrens)
|
||||
wp = lambda c: str(self.wrapper(c, width=cw))
|
||||
out = ' & '.join(map(wp, self.childrens))
|
||||
if self.endline:
|
||||
out += '\\\\'
|
||||
return out
|
||||
|
||||
|
||||
class hline_element(row_element):
|
||||
|
||||
def __init__(self, parent=None):
|
||||
super().__init__([], parent)
|
||||
|
||||
def content(self):
|
||||
return '\\hline'
|
||||
|
||||
# instance of hline element
|
||||
hline = hline_element()
|
||||
|
||||
|
||||
class tabular(element):
|
||||
""" Class representing a latex tabular. """
|
||||
|
||||
fmt = formatter({
|
||||
templates = {
|
||||
'content': """\\begin{{tabular}}{{{columns}}}
|
||||
{inner}
|
||||
\\end{{tabular}}"""
|
||||
})
|
||||
}
|
||||
|
||||
def __init__(self, rows, columns=None):
|
||||
def __init__(self, parent=None, rows=[], columns=None,
|
||||
autowrap=False, label=None):
|
||||
""" Create a tabular with given rows and column specification.
|
||||
|
||||
Parameters:
|
||||
- rows Rows for the tabular (list of list or list of string).
|
||||
- columns String representing columns.
|
||||
|
||||
If `rows` is a str, an empty tabular is created and `rows` is used
|
||||
instead of `columns` for the columns specification.
|
||||
- columns String representing columns options.
|
||||
"""
|
||||
if type(rows) == str:
|
||||
columns = rows
|
||||
rows = []
|
||||
super().__init__(parent, label=label)
|
||||
self.columns = columns
|
||||
self.rows = []
|
||||
self.autowrap = autowrap
|
||||
for row in rows:
|
||||
self.addrow(row)
|
||||
|
||||
def addrow(self, row):
|
||||
def addhline(self):
|
||||
""" Add a hline to the current tabular element. """
|
||||
self.addrow(hline_element(self), False)
|
||||
|
||||
def addrow(self, row, wrap=None):
|
||||
""" Add a row to the tabular.
|
||||
|
||||
Parameters:
|
||||
- row Array or string (array will be joined with ' & ').
|
||||
- wrap Can contains a wrapper element (typically a resizebox) for
|
||||
the cell of the row.
|
||||
"""
|
||||
if type(row) != 'str':
|
||||
row = ' & '.join(map(str, row))
|
||||
self.rows.append(row)
|
||||
if not isinstance(row, row_element):
|
||||
row = row_element(row, parent=self)
|
||||
if wrap is None:
|
||||
wrap = self.autowrap
|
||||
row.wrapper = wrap
|
||||
if self.childrens:
|
||||
last = self.childrens[-1]
|
||||
if isinstance(last, row_element):
|
||||
last.endline = True
|
||||
self.add(row)
|
||||
|
||||
def addrows(self, rows):
|
||||
""" Add multiple rows to the tabular (see addrow). """
|
||||
for row in rows:
|
||||
self.addrow(row)
|
||||
|
||||
def __str__(self):
|
||||
inner = '\\\\\n'.join(self.rows)
|
||||
return self.fmt.format('content', {
|
||||
def content(self):
|
||||
return self.fmt().format('content', {
|
||||
'columns': self.columns,
|
||||
'inner': inner
|
||||
'inner': super().content()
|
||||
})
|
||||
|
@ -2,5 +2,4 @@
|
||||
|
||||
# flake8: noqa
|
||||
|
||||
from .plots import *
|
||||
from .tikzpicture import tikzpicture
|
||||
|
@ -1,45 +0,0 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
from ..formatter import formatter
|
||||
|
||||
|
||||
class axis:
|
||||
""" Class representing an axis. """
|
||||
|
||||
fmt = formatter({
|
||||
'options': '[{content}]',
|
||||
'legends': '\\legend{{{content}}}',
|
||||
'content': """\\begin{{axis}}{options}
|
||||
{inner}
|
||||
{legend}
|
||||
\\end{{axis}}"""
|
||||
})
|
||||
|
||||
def __init__(self, *args, **kargs):
|
||||
self.options = kargs
|
||||
self.plots = []
|
||||
for arg in args:
|
||||
self.addplot(arg)
|
||||
|
||||
def addplot(self, pl):
|
||||
self.plots.append(pl)
|
||||
|
||||
def format_options(self, options):
|
||||
if not options:
|
||||
return ''
|
||||
opts = []
|
||||
for k, v in options.items():
|
||||
opts.append('{key} = {{{value}}}'.format(key=k, value=v))
|
||||
return self.fmt.format('options', {
|
||||
'content': ', '.join(opts)
|
||||
})
|
||||
|
||||
def __str__(self):
|
||||
return self.fmt.format('content', {
|
||||
'options': self.format_options(self.options),
|
||||
'inner': '\n'.join(map(str, self.plots)),
|
||||
'legend': self.fmt.format('legends', {
|
||||
'content': ', '.join('{{{}}}'.format(p.legend)
|
||||
for p in self.plots)
|
||||
})
|
||||
})
|
@ -1,29 +1,21 @@
|
||||
# -*- encoding: utf-8
|
||||
|
||||
from ..formatter import formatter
|
||||
|
||||
from .axis import axis
|
||||
from ..element import element
|
||||
|
||||
|
||||
class tikzpicture:
|
||||
class tikzpicture(element):
|
||||
""" Class representing a latex tikzfigure. """
|
||||
|
||||
fmt = formatter({
|
||||
templates = {
|
||||
'content': """\\begin{{tikzpicture}}
|
||||
{inner}
|
||||
\\end{{tikzpicture}}"""
|
||||
})
|
||||
}
|
||||
|
||||
def __init__(self, axis=[]):
|
||||
self.axis = []
|
||||
def __init__(self, childrens=None, parent=None, label=None):
|
||||
super().__init__(parent, childrens, label=label)
|
||||
|
||||
def addaxis(self, *args, **kargs):
|
||||
ax = args[0]
|
||||
if not isinstance(ax, axis):
|
||||
ax = axis(*args, **kargs)
|
||||
self.axis.append(ax)
|
||||
|
||||
def __str__(self):
|
||||
return self.fmt.format('content', {
|
||||
'inner': '\n'.join(map(str, self.axis))
|
||||
def content(self):
|
||||
return self.fmt().format('content', {
|
||||
'inner': super().content()
|
||||
})
|
||||
|
@ -1,18 +1,21 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
import os
|
||||
import sys
|
||||
|
||||
sys.path += ['..']
|
||||
|
||||
from pyltk import subfloat, tabular, resizebox, makebox, figure
|
||||
from pyltk.tikz import tikzpicture, errorbars_plot, coordinates_plot
|
||||
from pyltk.tikz import tikzpicture
|
||||
from pyltk.pgfplots import axis, errorbars_plot, coordinates_plot
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
data = [[1.00000000e+00,
|
||||
4.57213083e+00],
|
||||
[2.00000000e+00,
|
||||
1.16046200e+01],
|
||||
[ 4.00000000e+00,
|
||||
1.16046200e+01],
|
||||
[4.00000000e+00,
|
||||
2.71873880e+01],
|
||||
[ 5.00000000e+00,
|
||||
3.36039380e+01],
|
||||
@ -21,30 +24,57 @@ if __name__ == "__main__":
|
||||
[ 2.00000000e+01,
|
||||
1.50422028e+02]]
|
||||
|
||||
tk = tikzpicture()
|
||||
tk.addaxis(coordinates_plot('test', data),
|
||||
xlabel='', ylabel='Computation Time [s]')
|
||||
tk1 = tikzpicture()
|
||||
tk1.add(axis(coordinates_plot('test', data),
|
||||
xlabel='', ylabel='Computation Time [s]'))
|
||||
|
||||
tk2 = tikzpicture()
|
||||
tk2.addaxis(errorbars_plot('test 2', [
|
||||
tk2.add(axis(errorbars_plot('test 2', [
|
||||
[0, 5, (2, 3)],
|
||||
[1, 8, (4, 5)]
|
||||
]))
|
||||
])))
|
||||
|
||||
tk3 = tikzpicture()
|
||||
tk3.addaxis(errorbars_plot('test 2', [
|
||||
tk3.add(axis(errorbars_plot('test 2', [
|
||||
[0, 5, 3],
|
||||
[4, 8, 4]
|
||||
]))
|
||||
])))
|
||||
|
||||
s1 = subfloat(tk, 'Caption 1')
|
||||
s2 = subfloat(tk2, 'Caption 2')
|
||||
s3 = subfloat(tk3, 'Caption 3')
|
||||
s4 = subfloat('', 'Caption 4')
|
||||
tk4 = tikzpicture()
|
||||
tk4.add(axis(coordinates_plot('test', [])))
|
||||
|
||||
tab = tabular([
|
||||
[resizebox(s1, 0.5), resizebox(s2, 0.5)],
|
||||
[resizebox(s3, 0.5), resizebox(s4, 0.5)]
|
||||
], columns='cc')
|
||||
s1 = subfloat('Caption 1', childrens=tk1)
|
||||
s2 = subfloat('Caption 2', childrens=tk2)
|
||||
s3 = subfloat('Caption 3', childrens=tk3)
|
||||
s4 = subfloat('Caption 4', childrens=tk4)
|
||||
|
||||
print(figure(makebox(tab), '!h'))
|
||||
tab = tabular(rows=[
|
||||
[s1, s2],
|
||||
[s3, s4]
|
||||
], columns='cc', autowrap=resizebox)
|
||||
|
||||
f = figure(makebox(tab), '!h')
|
||||
print(f) # f.childrens[0].childrens[0].childrens)
|
||||
|
||||
wrap = """\\documentclass{{article}}
|
||||
\\usepackage{{amsmath,amsfonts,amssymb}}
|
||||
\\usepackage{{tabularx}}
|
||||
\\usepackage{{multirow}}
|
||||
\\usepackage{{stmaryrd}}
|
||||
\\usepackage{{hhline}}
|
||||
\\usepackage{{caption}}
|
||||
\\usepackage{{tikz}}
|
||||
\\usetikzlibrary{{arrows}}
|
||||
\\usetikzlibrary{{positioning}}
|
||||
\\usepackage{{subfig}}
|
||||
\\usepackage{{pgfplots}}
|
||||
|
||||
\\begin{{document}}
|
||||
{content}
|
||||
\\end{{document}}""".format(content=f)
|
||||
|
||||
with open('/tmp/test.tex', 'w') as fd:
|
||||
fd.write(wrap)
|
||||
|
||||
os.system('pdflatex /tmp/test.tex')
|
||||
os.system('evince test.pdf')
|
||||
|
Loading…
Reference in New Issue
Block a user