1
0
mirror of https://frontier.innolan.net/rainlance/amiga-ntimed.git synced 2025-11-19 21:50:39 +00:00
Files
amiga-ntimed/plotgen.py
2014-12-21 22:31:18 +00:00

164 lines
4.3 KiB
Python

#!/usr/bin/env python
#
# Copyright (c) 2014 Poul-Henning Kamp
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.
#
# Postprocess simulation output for gnuplot
# =========================================
#
# python plotgen.py
# then
# gnuplot
# load '/tmp/_g'
#
from __future__ import print_function
fi = open("/tmp/_")
fg = open("/tmp/_g", "w")
fg.write("#\n")
plots = []
class plot(object):
def __init__(self, title = "-"):
self.s = []
self.title = title
plots.append(self)
def write(self, l):
self.s.append(l)
def commit(self, fg, title):
if title:
fg.write("set y2label '%s'\n" % self.title)
else:
fg.write("unset y2label\n")
for i in self.s:
fg.write(i + "\n")
now = 0.0
class simpll(object):
def __init__(self):
self.fo = open("/tmp/_simpll", "w")
self.pl1 = plot('p/freq')
self.pl1.write(
"plot '/tmp/_simpll' using 1:($2*1e6) with line notitle")
self.pl2 = plot('p/off')
self.pl2.write(
"plot '/tmp/_simpll' using 1:($3*1e6/($4 == 0 ? 1 : $4)) with impulse notitle")
if False:
self.pl3 = plot('p/dur')
self.pl3.write(
"plot '/tmp/_simpll' using 1:4 with imp notitle")
def data(self, ll):
self.fo.write("%f " % now + " " + " ".join(ll[1:]) + "\n")
class combine(object):
def __init__(self):
self.fo = open("/tmp/_combine", "w")
self.pl1 = plot("c/in")
self.pl1.write(
"plot '/tmp/_combine' using 1:2 notitle"
", '/tmp/_combine' using 1:4 notitle"
", '/tmp/_combine' using 1:3 notitle"
)
self.pl2 = plot('c/peak')
self.pl2.write(
"plot '/tmp/_combine' using 1:5 with line notitle"
)
self.pl3 = plot("c/weight")
self.pl3.write("set yrange[0:]")
self.pl3.write(
"plot '/tmp/_combine' using 1:6 axis x1y2 with line notitle"
)
self.pl3.write("set autoscale y")
def data(self, ll):
self.fo.write("%f " % now + " " + " ".join(ll[3:]) + "\n")
p_simpll = simpll()
p_combine = combine()
n = 0
for l in fi:
n += 1
if len(l) == 0 or l[0] == "#":
continue
ll = l.split()
if ll[0] == "Now":
now = float(ll[1]) - 1e6
continue
if ll[0] == "SIMPLL":
p_simpll.data(ll)
continue
if ll[0] == "Combine":
p_combine.data(ll)
continue
def plotset(xlo, xhi, lbl):
fg.write("set autoscale x\n")
fg.write("set xrange [%g:%g]\n" % (xlo, xhi))
fg.write("set bmargin 0\n")
fg.write("set xtics format ''\n")
fg.write("set xtics (%g, %g)\n" % (xlo, xhi))
for i in plots[:-1]:
i.commit(fg, lbl)
fg.write("set bmargin 2\n")
fg.write("set xtics format '%g'\n")
fg.write("set xtics (%g, %g)\n" % (xlo, xhi))
plots[-1].commit(fg, lbl)
fg.write("set grid\n")
fg.write("set pointsize .5\n")
fg.write("set size .8,.8\n")
fg.write("set tmargin 1\n")
fg.write("set lmargin 8\n")
fg.write("set multiplot layout %d,3 columnsfirst\n" % len(plots))
fg.write("set style data point\n")
plotset(0, 64, False)
if now > 10000:
fg.write("set pointsize .2\n")
plotset(60, 3600, False)
fg.write("set pointsize .1\n")
plotset(3600, now, True)
else:
fg.write("set pointsize .5\n")
plotset(60, 600, False)
fg.write("set pointsize .2\n")
plotset(5, now, True)
fg.write("unset multiplot\n")
fg.write("set autoscale\n")
fg.write("set xtics auto\n")
fg.write("set ytics auto\n")