#!/usr/bin/env python
# -*- coding: utf-8 -*-
# pywws - Python software for USB Wireless Weather Stations
# http://github.com/jim-easterbrook/pywws
# Copyright (C) 2008-13 Jim Easterbrook jim@jim-easterbrook.me.uk
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
"""Plot graphs of weather data according to an XML recipe
::
%s
Introduction
------------
Like Template.py this is one of the more difficult to use modules in
the weather station software collection. It plots a graph (or set of
graphs) of weather data. Almost everything about the graph is
controlled by an XML file. I refer to these files as templates, but
they aren't templates in the same sense as Template.py uses to create
text files.
Before writing your own graph template files, it might be useful to
look at some of the examples in the example_graph_templates directory.
If (like I was) you are unfamiliar with XML, I suggest reading the W3
Schools XML tutorial.
XML graph file syntax
^^^^^^^^^^^^^^^^^^^^^
Here is the simplest useful graph template. It plots the external
temperature for the last 24 hours. ::
<?xml version="1.0" encoding="ISO-8859-1"?>
<graph>
<plot>
<subplot>
<title>Temperature (°C)</title>
<ycalc>data['temp_out']</ycalc>
</subplot>
</plot>
</graph>
In this example, the root element graph has one plot element, which
has one subplot element. The subplot element contains a title element
and a ycalc element. To plot more data on the same set of axes (for
example dew point and temperature), we can add more subplot elements.
To plot more than one set of axes (for example wind speed is measured
in different units from temperature) in the same file we can add more
plot elements.
The complete element hierarchy is shown below. ::
graph
plot
subplot
xcalc
ycalc
axes
style
colour
title
bmargin
yrange
y2range
ytics
y2tics
ylabel
ylabelangle
y2label
y2labelangle
grid
source
boxwidth
title
command
start
stop
duration
layout
size
fileformat
terminal
lmargin
rmargin
xformat
xlabel
dateformat
xtics
title
graph
^^^^^
This is the root element of the graph XML file. It does not have to be
called "graph", but there must be exactly one root element.
plot
^^^^
Every graph element should contain at least one plot element. A
separate graph is drawn for each plot element, but all share the same
X axis.
start
^^^^^
This element sets the date & time of the start of the X axis. It is
used in the constructor of a Python datetime object. For example, to
start the graph at noon (local time) on Christmas day 2008:
``<start>year=2008, month=12, day=25, hour=12</start>``. The default
value is (stop - duration).
stop
^^^^
This element sets the date & time of the end of the X axis. It is used
in the constructor of a Python datetime object. For example, to end
the graph at 10 am (local time) on new year's day: ``<stop>year=2009,
month=1, day=1, hour=10</stop>``. The default value is (start +
duration), unless start is not defined in which case the timestamp of
the latest weather station hourly reading is used.
duration
^^^^^^^^
This element sets the extent of the X axis of the graph, unless both
start and stop are defined. It is used in the constructor of a Python
timedelta object. For example, to plot one week:
``<duration>weeks=1</duration>``. The default value is hours=24.
layout
^^^^^^
Controls the layout of the plots. Default is a single column. The
layout element specifies rows and columns. For example: ``<layout>4,
2</layout>`` will use a grid of 4 rows and 2 columns.
size
^^^^
Sets the overall dimensions of the image file containing the graph.
Default (in a single column layout) is a width of 600 pixels and
height of 200 pixels for each plot, so a graph with four plot elements
would be 600 x 800 pixels. Any size element must include both width
and height. For example: ``<size>800, 600</size>`` will produce an
image 800 pixels wide and 600 pixels high.
fileformat
^^^^^^^^^^
Sets the image format of the file containing the graph. Default is
png. Any string recognised by your installation of gnuplot should do.
For example: ``<fileformat>gif</fileformat>`` will produce a GIF
image.
terminal
^^^^^^^^
Allows complete control of gnuplot's 'terminal' settings. You may want
to use this if you are plotting to an unusual image format. Any string
recognised by your installation of gnuplot's 'set terminal' command
should do. For example: ``<terminal>svg enhanced font "arial,9" size
600,800 dynamic rounded</terminal>``. This setting overwrites both
size and fileformat.
lmargin
^^^^^^^
Sets the left margin of the plots, i.e. the distance from the left
hand axis to the left hand edge of the image area. According to the
gnuplot documentation the units of lmargin are character widths. The
default value is 5, which should look OK in most circumstances.
rmargin
^^^^^^^
Sets the right margin of the plots, i.e. the distance from the right
hand axis to the right hand edge of the image area. According to the
gnuplot documentation the units of rmargin are character widths. The
default value is -1, which sets automatic adjustment.
xformat
^^^^^^^
Sets the format of the time / date xtic labels. The value is a
strftime style format string. Default depends on the graph duration:
24 hours or less is "%%H%%M", 24 hours to 7 days is "%%a %%d" and 7
days or more is "%%Y/%%m/%%d".
xlabel
^^^^^^
Sets the X axis label. The value is a strftime style format string.
Default depends on the graph duration: 24 hours or less is "Time
(%%Z)", 24 hours to 7 days is "Day" and 7 days or more is "Date". The
datetime used to compute this is start, which may produce unexpected
results when a graph spans DST start or end.
dateformat
^^^^^^^^^^
Sets the format of the date labels at each end of X axis. The value is
a strftime style format string. Default is "%%Y/%%m/%%d". The right
hand label is only drawn if it differs from the left. To have no
labels, set an empty format: ``<dateformat></dateformat>``
xtics
^^^^^
Sets the spacing of the "tic" marks on the X axis. The value is an
integer number of hours. The default is to allow gnuplot to set an
appropriate interval.
title
^^^^^
Sets the title of the graph. A single line of text, for example:
``<title>Today's weather</title>``. This title appears at the very top
of the graph, outside any plot area.
subplot
^^^^^^^
Every plot element should contain at least one subplot element. A
separate trace is drawn for each subplot element, but all share the
same X and Y axes.
bmargin
^^^^^^^
Sets the bottom margin, i.e. the spacing between the lower X axis and
the edge of the graph (or the next plot). The default is to let
gnuplot adjust this automatically, which works OK most of the time but
you may wish to fine tune the value to suit your installation.
The permitted value is any non-negative real number. On my setup 0.9
is a good value, set as follows: ``<bmargin>0.9</bmargin>``.
yrange
^^^^^^
Sets the lower and upper limits of the (left hand) Y axis. The value
is anything understood by gnuplot, typically a pair of numbers. The
default is to allow gnuplot to set appropriate values, which is
unlikely to be what you want. For example, to plot typical UK
temperatures with no value going off the graph: ``<yrange>-10,
30</yrange>``. Note that commas are converted to colons, so
``<yrange>-10:30</yrange>`` would be equivalent.
You can use an asterisk to have gnuplot choose a suitable value. For
example, to have the upper value auto scale whilst fixing the lower
value at zero, use ``<yrange>0:*</yrange>``.
y2range
^^^^^^^
Sets the lower and upper limits of the right hand Y axis. Default is
for the right hand Y axis to be the same as the left, but setting a
different range is useful in dual axis plotting.
ytics
^^^^^
Controls the "tic" marks on the left hand Y axis. The value can be
anything that's understood by gnuplot. For example, to set the tic
spacing to 45 use ``<ytics>45</ytics>``. More complex things are also
possible, e.g. to label a wind direction graph with compass points,
use ``<y2tics>('N' 0, 'E' 90, 'S' 180, 'W' 270, 'N' 360)</y2tics>``.
y2tics
^^^^^^
Controls the "tic" marks on the right hand axis. The format is the
same as that for ytics. Default behaviour is to copy the left hand tic
marks, but without labels.
ylabel
^^^^^^
Adds a label to the (left hand) Y axis. For example, when plotting
temperature: ``<ylabel>°C</ylabel>``. If you use ylabel you will
probably want to adjust lmargin.
ylabelangle
^^^^^^^^^^^
Adjust the angle of the (left hand) Y axis label, if your version of
gnuplot supports it. For example, to write the label horizontally:
``<ylabelangle>90</ylabelangle>``.
y2label
^^^^^^^
Adds a label to the right hand Y axis. For example, when plotting
humidity: ``<y2label>%%</y2label>``. This is mostly used when plotting
dual axis graphs. If you use y2label you will probably want to adjust
rmargin.
y2labelangle
^^^^^^^^^^^^
Adjust the angle of the right hand Y axis label, if your version of
gnuplot supports it. For example, to write the label horizontally:
``<y2labelangle>90</y2labelangle>``.
grid
^^^^
Adds a grid to the plot. In most situations gnuplot's default grid is
suitable, so no value is needed: ``<grid></grid>``. More control is
possible using any of the options understood by gnuplot's set grid
command. For example, to have horizontal grid lines only:
``<grid>ytics</grid>``.
source
^^^^^^
Select the weather data to be plotted. Permitted values are
``<source>raw</source>``, ``<source>hourly</source>``,
``<source>daily</source>`` and ``<source>monthly</source>``. Default
is raw. Note that the different sources have different data
dictionaries, so this choice affects ycalc.
boxwidth
^^^^^^^^
Sets the width of the "boxes" used when drawing bar graphs. The value
is an integer expression yielding a number of seconds. Default depends
on source: raw is 240, hourly is 2800 and daily is 2800 * 24.
title
^^^^^
Sets the title of the plot. A single line of text, for example:
``<title>Temperature (°C)</title>``. This title appears within the
plot area, above any subplot titles.
command
^^^^^^^
Execute any gnuplot command, just before the main "plot" command. This
option allows advanced users to have greater control over the graph
appearance. The value is any valid gnuplot command, typically
beginning with the word set. For example: ``<command>set key tmargin
center horizontal width 1 noreverse enhanced autotitles box linetype
-1 linewidth 1</command>``. (Don't ask me what this example does — I'm
not an advanced user).
xcalc
^^^^^
Controls the X axis positioning of plotted data values. The default
value of data['idx'] is correct for most data, but there are some
exceptions. For example, when plotting bar charts of hourly rainfall,
it's nice to centre the bars on 30 minutes past the hour:
``<xcalc>data['idx'].replace(minute=30, second=0)</xcalc>``.
ycalc
^^^^^
Selects the data to be plotted. Any one line Python expression that
returns a single float value can be used. At its simplest this just
selects one value from the "data" dictionary, for example:
``<ycalc>data['temp_out']</ycalc>`` plots the external temperature.
More complex expressions are possible, and some helper functions are
provided. For example: ``<ycalc>dew_point(data['temp_out'],
data['hum_out'])</ycalc>`` plots the external dew point, and
``<ycalc>wind_mph(data['wind_ave'])</ycalc>`` plots the average wind
speed in miles per hour.
Cumulative plots are also possible. The result of each ycalc
computation is stored and made available to the next computation in
the variable last_ycalc. This can be used with any data, but is most
useful with rainfall: ``<ycalc>data['rain'] + last_ycalc</ycalc>``.
axes
^^^^
Selects which Y axis the data is plotted against. Default is the left
hand axis, but the right hand axis can be chosen with:
``<axes>x1y2</axes>``. This can be used in conjunction with y2range to
plot two unrelated quantities on one graph, for example temperature
and humidity.
style
^^^^^
Sets the line style for the graph. Default is a smooth continuous
line, thickness 1. To select a bar graph use: ``<style>box</style>``.
To select points without a connecting line use: ``<style>+</style>``
or ``<style>x</style>``. To select a line thickness 3 (for example)
use: ``<style>line 3</style>``. The thickness of points can be set in
a similar fashion. For complete control (for advanced users) a full
gnuplot style can be set: ``<style>smooth unique lc 5 lw 3</style>``.
colour
^^^^^^
Sets the colour of the subplot line or boxes. Any integer value is
accepted. The mapping of colours to numbers is set by gnuplot. Default
value is the previous colour plus one.
title
^^^^^
Sets the title of the subplot. A single line of text, for example:
``<title>Temperature (°C)</title>``. This title appears within the
plot area, next to a short segment of the line colour used for the
subplot.
Detailed API
------------
"""
__docformat__ = "restructuredtext en"
__usage__ = """
usage: python -m pywws.Plot [options] data_dir temp_dir xml_file output_file
options are:
-h or --help display this help
data_dir is the root directory of the weather data
temp_dir is a workspace for temporary files e.g. /tmp
xml_file is the name of the source file that describes the plot
output_file is the name of the image file to be created e.g. 24hrs.png
"""
__doc__ %= __usage__
__usage__ = __doc__.split('\n')[0] + __usage__
import codecs
from datetime import datetime, timedelta
import getopt
import logging
import os
import subprocess
import sys
import xml.dom.minidom
from pywws.conversions import *
from pywws import DataStore
from pywws import Localisation
from pywws.Logger import ApplicationLogger
from pywws.TimeZone import Local
[docs]class BasePlotter(object):
def __init__(self, params, status, raw_data, hourly_data,
daily_data, monthly_data, work_dir):
self.logger = logging.getLogger('pywws.%s' % self.__class__.__name__)
self.raw_data = raw_data
self.hourly_data = hourly_data
self.daily_data = daily_data
self.monthly_data = monthly_data
self.work_dir = work_dir
self.pressure_offset = eval(status.get('fixed', 'pressure offset'))
# set language related stuff
self.encoding = params.get('config', 'gnuplot encoding', 'iso_8859_1')
# create work directory
if not os.path.isdir(self.work_dir):
os.makedirs(self.work_dir)
[docs] def DoPlot(self, input_file, output_file):
# read XML graph description
self.doc = xml.dom.minidom.parse(input_file)
self.graph = self.GetChildren(self.doc, 'graph')
if not self.graph:
self.logger.error('%s has no graph node' % input_file)
self.doc.unlink()
return 1
self.graph = self.graph[0]
# get list of plots
plot_list = self.GetPlotList()
self.plot_count = len(plot_list)
if self.plot_count < 1:
# nothing to plot
self.logger.info('%s has no plot nodes' % input_file)
self.doc.unlink()
return 1
# get start and end datetimes
self.x_lo = self.GetValue(self.graph, 'start', None)
self.x_hi = self.GetValue(self.graph, 'stop', None)
self.duration = self.GetValue(self.graph, 'duration', None)
if self.duration == None:
self.duration = timedelta(hours=24)
else:
self.duration = eval('timedelta(%s)' % self.duration)
if self.x_lo != None:
self.x_lo = eval('datetime(%s)' % self.x_lo)
if self.x_hi != None:
self.x_hi = eval('datetime(%s)' % self.x_hi)
self.duration = self.x_hi - self.x_lo
else:
self.x_hi = self.x_lo + self.duration
elif self.x_hi != None:
self.x_hi = eval('datetime(%s)' % self.x_hi)
self.x_lo = self.x_hi - self.duration
else:
self.x_hi = self.hourly_data.before(datetime.max)
if self.x_hi == None:
self.x_hi = datetime.utcnow() # only if no hourly data
# set end of graph to start of the next hour after last item
self.x_hi = self.x_hi + timedelta(minutes=55)
self.x_hi = self.x_hi.replace(minute=0, second=0)
self.x_lo = self.x_hi - self.duration
self.x_hi = self.x_hi + Local.utcoffset(self.x_lo)
self.x_lo = self.x_hi - self.duration
self.utcoffset = Local.utcoffset(self.x_lo)
# open gnuplot command file
self.tmp_files = []
cmd_file = os.path.join(self.work_dir, 'plot.cmd')
self.tmp_files.append(cmd_file)
if sys.version_info[0] >= 3:
of = open(cmd_file, 'w', encoding=self.encoding)
else:
of = open(cmd_file, 'w')
# write gnuplot set up
self.rows = self.GetDefaultRows()
self.cols = (self.plot_count + self.rows - 1) // self.rows
self.rows, self.cols = eval(self.GetValue(
self.graph, 'layout', '%d, %d' % (self.rows, self.cols)))
w, h = self.GetDefaultPlotSize()
w = w * self.cols
h = h * self.rows
w, h = eval(self.GetValue(self.graph, 'size', '(%d, %d)' % (w, h)))
fileformat = self.GetValue(self.graph, 'fileformat', 'png')
if fileformat == 'svg':
terminal = '%s enhanced font "arial,9" size %d,%d dynamic rounded' % (
fileformat, w, h)
else:
terminal = '%s large size %d,%d' % (fileformat, w, h)
terminal = self.GetValue(self.graph, 'terminal', terminal)
of.write('set encoding %s\n' % (self.encoding))
of.write('set terminal %s\n' % (terminal))
of.write('set output "%s"\n' % (output_file))
# set overall title
title = self.GetValue(self.graph, 'title', '')
if title:
if sys.version_info[0] < 3:
title = title.encode(self.encoding)
title = 'title "%s"' % title
of.write('set multiplot layout %d, %d %s\n' % (self.rows, self.cols, title))
# do actual plots
of.write(self.GetPreamble())
for plot_no in range(self.plot_count):
plot = plot_list[plot_no]
# set key / title location
title = self.GetValue(plot, 'title', '')
if sys.version_info[0] < 3:
title = title.encode(self.encoding)
of.write('set key horizontal title "%s"\n' % title)
# optional yaxis labels
ylabel = self.GetValue(plot, 'ylabel', '')
if ylabel:
if sys.version_info[0] < 3:
ylabel = ylabel.encode(self.encoding)
ylabelangle = self.GetValue(plot, 'ylabelangle', '')
if ylabelangle:
ylabelangle = ' rotate by %s' % (ylabelangle)
of.write('set ylabel "%s"%s\n' % (ylabel, ylabelangle))
else:
of.write('set ylabel\n')
y2label = self.GetValue(plot, 'y2label', '')
if y2label:
if sys.version_info[0] < 3:
y2label = y2label.encode(self.encoding)
y2labelangle = self.GetValue(plot, 'y2labelangle', '')
if y2labelangle:
y2labelangle = ' rotate by %s' % (y2labelangle)
of.write('set y2label "%s"%s\n' % (y2label, y2labelangle))
else:
of.write('set y2label\n')
# set data source
source = self.GetValue(plot, 'source', 'raw')
if source == 'raw':
source = self.raw_data
elif source == 'hourly':
source = self.hourly_data
elif source == 'monthly':
source = self.monthly_data
else:
source = self.daily_data
# do the plot
of.write(self.PlotData(plot_no, plot, source))
of.close()
self.doc.unlink()
# run gnuplot on file
subprocess.check_call(['gnuplot', cmd_file])
for file in self.tmp_files:
os.unlink(file)
return 0
[docs] def GetChildren(self, node, name):
result = []
for child in node.childNodes:
if child.localName == name:
result.append(child)
return result
[docs] def GetValue(self, node, name, default):
for child in node.childNodes:
if child.localName == name:
if child.childNodes:
return child.childNodes[0].data.strip()
else:
return ''
return default
[docs]class Record(object):
pass
[docs]class GraphPlotter(BasePlotter):
[docs] def GetPlotList(self):
return self.GetChildren(self.graph, 'plot')
[docs] def GetDefaultRows(self):
return self.plot_count
[docs] def GetDefaultPlotSize(self):
return 200 // self.cols, 600 // self.cols
[docs] def GetPreamble(self):
result = """set style fill solid
set xdata time
set timefmt "%Y-%m-%dT%H:%M:%S"
"""
result += 'set xrange ["%s":"%s"]\n' % (
self.x_lo.isoformat(), self.x_hi.isoformat())
lmargin = eval(self.GetValue(self.graph, 'lmargin', '5'))
result += 'set lmargin %g\n' % (lmargin)
rmargin = eval(self.GetValue(self.graph, 'rmargin', '-1'))
result += 'set rmargin %g\n' % (rmargin)
if self.duration <= timedelta(hours=24):
xformat = '%H%M'
elif self.duration <= timedelta(days=7):
xformat = '%a %d'
else:
xformat = '%Y/%m/%d'
xformat = self.GetValue(self.graph, 'xformat', xformat)
result += 'set format x "%s"\n' % xformat
xtics = self.GetValue(self.graph, 'xtics', None)
if xtics:
result += 'set xtics %d\n' % (eval(xtics) * 3600)
if sys.version_info[0] < 3:
result = result.encode(self.encoding)
return result
[docs] def PlotData(self, plot_no, plot, source):
_ = Localisation.translation.ugettext
subplot_list = self.GetChildren(plot, 'subplot')
subplot_count = len(subplot_list)
if subplot_count < 1:
return ''
result = ''
pressure_offset = self.pressure_offset
# label x axis of last plot
if plot_no == self.plot_count - 1:
if self.duration <= timedelta(hours=24):
xlabel = _('Time (%Z)')
elif self.duration <= timedelta(days=7):
xlabel = _('Day')
else:
xlabel = _('Date')
xlabel = self.GetValue(self.graph, 'xlabel', xlabel)
if sys.version_info[0] < 3:
xlabel = xlabel.encode(self.encoding)
result += 'set xlabel "%s"\n' % (
self.x_lo.replace(tzinfo=Local).strftime(xlabel))
dateformat = '%Y/%m/%d'
dateformat = self.GetValue(self.graph, 'dateformat', dateformat)
if sys.version_info[0] < 3:
dateformat = dateformat.encode(self.encoding)
ldat = self.x_lo.replace(tzinfo=Local).strftime(dateformat)
rdat = self.x_hi.replace(tzinfo=Local).strftime(dateformat)
if ldat != '':
result += 'set label "%s" at "%s", graph -0.3 left\n' % (
ldat, self.x_lo.isoformat())
if rdat != ldat:
result += 'set label "%s" at "%s", graph -0.3 right\n' % (
rdat, self.x_hi.isoformat())
# set bottom margin
bmargin = eval(self.GetValue(plot, 'bmargin', '-1'))
result += 'set bmargin %g\n' % (bmargin)
# set y ranges and tics
yrange = self.GetValue(plot, 'yrange', None)
y2range = self.GetValue(plot, 'y2range', None)
ytics = self.GetValue(plot, 'ytics', 'autofreq')
y2tics = self.GetValue(plot, 'y2tics', '')
if y2tics and not y2range:
y2range = yrange
elif y2range and not y2tics:
y2tics = 'autofreq'
if yrange:
result += 'set yrange [%s]\n' % (yrange.replace(',', ':'))
else:
result += 'set yrange [*:*]\n'
if y2range:
result += 'set y2range [%s]\n' % (y2range.replace(',', ':'))
if y2tics:
result += 'set ytics nomirror %s; set y2tics %s\n' % (ytics, y2tics)
else:
result += 'unset y2tics; set ytics mirror %s\n' % (ytics)
# set grid
result += 'unset grid\n'
grid = self.GetValue(plot, 'grid', None)
if grid != None:
result += 'set grid %s\n' % grid
# x_lo & x_hi are in local time, data is indexed in UTC
start = self.x_lo - self.utcoffset
stop = self.x_hi - self.utcoffset
cumu_start = start
if source == self.raw_data:
boxwidth = 240 # assume 5 minute data interval
start = source.before(start)
elif source == self.hourly_data:
boxwidth = 2800
start = source.before(start)
interval = timedelta(minutes=90)
elif source == self.monthly_data:
boxwidth = 2800 * 24 * 30
interval = timedelta(days=46)
else:
interval = timedelta(hours=36)
boxwidth = 2800 * 24
boxwidth = eval(self.GetValue(plot, 'boxwidth', str(boxwidth)))
result += 'set boxwidth %d\n' % boxwidth
command = self.GetValue(plot, 'command', None)
if command:
result += '%s\n' % command
stop = source.after(stop)
if stop:
stop = stop + timedelta(minutes=1)
# write data files
subplots = []
for subplot_no in range(subplot_count):
subplot = Record()
subplot.subplot = subplot_list[subplot_no]
subplot.dat_file = os.path.join(self.work_dir, 'plot_%d_%d.dat' % (
plot_no, subplot_no))
self.tmp_files.append(subplot.dat_file)
subplot.dat = open(subplot.dat_file, 'w')
subplot.xcalc = self.GetValue(subplot.subplot, 'xcalc', None)
subplot.ycalc = self.GetValue(subplot.subplot, 'ycalc', None)
subplot.cummulative = 'last_ycalc' in subplot.ycalc
if subplot.xcalc:
subplot.xcalc = compile(subplot.xcalc, '<string>', 'eval')
subplot.ycalc = compile(subplot.ycalc, '<string>', 'eval')
subplot.last_ycalcs = 0.0
subplot.last_idx = None
subplots.append(subplot)
for data in source[start:stop]:
for subplot in subplots:
if subplot.xcalc:
idx = eval(subplot.xcalc)
if idx is None:
continue
else:
idx = data['idx']
idx += self.utcoffset
if not subplot.cummulative and subplot.last_idx:
if source == self.raw_data:
interval = timedelta(minutes=((data['delay']*3)+1)//2)
if idx - subplot.last_idx > interval:
# missing data
subplot.dat.write('%s ?\n' % (idx.isoformat()))
subplot.last_idx = idx
try:
if subplot.cummulative and data['idx'] <= cumu_start:
value = 0.0
else:
last_ycalc = subplot.last_ycalcs
value = eval(subplot.ycalc)
subplot.dat.write('%s %g\n' % (idx.isoformat(), value))
subplot.last_ycalcs = value
except TypeError:
if not subplot.cummulative:
subplot.dat.write('%s ?\n' % (idx.isoformat()))
subplot.last_ycalcs = 0.0
for subplot in subplots:
# ensure the data file isn't empty
idx = self.x_hi + self.duration
subplot.dat.write('%s ?\n' % (idx.isoformat()))
subplot.dat.close()
# plot data
result += 'plot '
colour = 0
for subplot_no in range(subplot_count):
subplot = subplots[subplot_no]
colour = eval(self.GetValue(subplot.subplot, 'colour', str(colour+1)))
style = self.GetValue(
subplot.subplot, 'style', 'smooth unique lc %d lw 1' % (colour))
words = style.split()
if len(words) > 1 and words[0] in ('+', 'x', 'line'):
width = int(words[1])
else:
width = 1
if style == 'box':
style = 'lc %d lw 0 with boxes' % (colour)
elif words[0] == '+':
style = 'lc %d lw %d pt 1 with points' % (colour, width)
elif words[0] == 'x':
style = 'lc %d lw %d pt 2 with points' % (colour, width)
elif words[0] == 'line':
style = 'smooth unique lc %d lw %d' % (colour, width)
axes = self.GetValue(subplot.subplot, 'axes', 'x1y1')
title = self.GetValue(subplot.subplot, 'title', '')
result += ' "%s" using 1:($2) axes %s title "%s" %s' % (
subplot.dat_file, axes, title, style)
if subplot_no != subplot_count - 1:
result += ', \\'
result += '\n'
if sys.version_info[0] < 3:
result = result.encode(self.encoding)
return result
[docs]def main(argv=None):
if argv is None:
argv = sys.argv
try:
opts, args = getopt.getopt(argv[1:], "h", ['help'])
except getopt.error, msg:
print >>sys.stderr, 'Error: %s\n' % msg
print >>sys.stderr, __usage__.strip()
return 1
# process options
for o, a in opts:
if o == '-h' or o == '--help':
print __usage__.strip()
return 0
# check arguments
if len(args) != 4:
print >>sys.stderr, 'Error: 4 arguments required\n'
print >>sys.stderr, __usage__.strip()
return 2
logger = ApplicationLogger(1)
params = DataStore.params(args[0])
status = DataStore.status(args[0])
Localisation.SetApplicationLanguage(params)
return GraphPlotter(
params, status,
DataStore.calib_store(args[0]), DataStore.hourly_store(args[0]),
DataStore.daily_store(args[0]), DataStore.monthly_store(args[0]),
args[1]
).DoPlot(args[2], args[3])
if __name__ == "__main__":
sys.exit(main())