summaryrefslogtreecommitdiff
path: root/Master/texmf-dist/scripts
diff options
context:
space:
mode:
authorKarl Berry <karl@freefriends.org>2010-01-03 01:14:11 +0000
committerKarl Berry <karl@freefriends.org>2010-01-03 01:14:11 +0000
commit98837e31bcd7413ffa40c85ebbce9aa3ff3fdbed (patch)
tree1ed7f18cb8a81bb3f209bf158a38f5b624c2d607 /Master/texmf-dist/scripts
parent080069f58e2325db5ab0ac5f5ea584f3021bc383 (diff)
pgfplots 1.3 (2jan10)
git-svn-id: svn://tug.org/texlive/trunk@16576 c570f23f-e606-0410-a88d-b1316a301751
Diffstat (limited to 'Master/texmf-dist/scripts')
-rw-r--r--Master/texmf-dist/scripts/pgfplots/pgfplots.py89
1 files changed, 89 insertions, 0 deletions
diff --git a/Master/texmf-dist/scripts/pgfplots/pgfplots.py b/Master/texmf-dist/scripts/pgfplots/pgfplots.py
new file mode 100644
index 00000000000..8c00600fc77
--- /dev/null
+++ b/Master/texmf-dist/scripts/pgfplots/pgfplots.py
@@ -0,0 +1,89 @@
+"""Module to plot using Pgfplots.
+
+This module provides a means to create and display a graph very quickly.
+
+In this code, the program used to display the created PDF is 'xpdf'. Change
+it to your favorite PDF reader, such as Acrobat Reader (called acroread or
+something similar)
+
+The code used to generate the graph is printed in the command line. Edit
+your graph iteratively, and when you are satisfied with the graph, copy and
+paste the relevant part to your TEX file.
+
+This module requires the numpy module.
+
+For example of usage, see the executable part at the bottom.
+
+"""
+import numpy as np
+import subprocess
+import os
+GRAPH_N = 0
+
+class Pgf:
+ def __init__(z, xlabel='', ylabel=''):
+ """Initialize and provide axis labels."""
+ z.buf = []
+ z.options = []
+ z.opt('xlabel={{{0}}}'.format(xlabel))
+ z.opt('ylabel={{{0}}}'.format(ylabel))
+ z.legend = []
+ def opt(z, *args):
+ """Write arguments to the AXIS environment."""
+ for arg in args:
+ z.options.append(arg)
+ def plot(z, x, y, legend=None, *args):
+ """Plot the data contained in the vectors x and y.
+
+ Options to the \addplot command can be provided in *args.
+ """
+ coor = ''.join(['({0}, {1})'. format(u, v) for u, v in zip(x,y)])
+ z.buf.append('\\addplot{0} coordinates {{{1}}};\n'.format(
+ ('[' + ', '.join(args) + ']') if len(args) else '' ,coor))
+ if legend is not None:
+ z.legend.append(legend)
+ def save(z, graph_n=None):
+ """Generate graph.
+
+ If graph_n is None or a number, the graph in a file beginning with
+ zzz. This file is meant to be temporary. If graph_n is a string,
+ that string is used as the file name.
+ """
+ if type(graph_n) is str:
+ file_name = graph_n
+ else:
+ if graph_n is None:
+ global GRAPH_N
+ graph_n = GRAPH_N
+ GRAPH_N += 1
+ elif type(graph_n) is not int:
+ raise Error('graph_n should be a string or an integer')
+ file_name = 'zzz{0}'.format(graph_n)
+ with open(file_name + '.tex', 'w') as f:
+ b = []
+ b.append('\\documentclass{article}\n')
+ b.append('\\usepackage{pgfplots}\n')
+ b.append('\\begin{document}\n')
+ b.append('\\begin{tikzpicture}')
+ b.append('\\begin{axis}[\n')
+ b.append('{0}]'.format(',\n'.join(z.options)))
+ b.extend(z.buf)
+ if z.legend:
+ b.append('\\legend{{' + '}, {'.join(z.legend) + '}}\n')
+ b.append('\\end{axis}\n')
+ b.append('\\end{tikzpicture}\n')
+ b.append('\\end{document}')
+ f.writelines(b)
+ print(''.join(b))
+ os.system('pdflatex {0}.tex'.format(file_name))
+ os.remove(file_name + '.aux')
+ os.remove(file_name + '.log')
+ subprocess.Popen(['xpdf', '{0}.pdf'.format(file_name)])
+if __name__ == '__main__':
+ """Example of usage."""
+ x = np.linspace(0, 2*np.pi)
+ p = Pgf('time', 'Voltage')
+ p.opt('ybar')
+ p.plot(x, np.sin(x), 'sin')
+ p.plot(x, np.cos(x), 'cos')
+ p.save()#'graph_test_pgf_1')