def createChart(title, xaxis, datay, firstweekday):
# Find the max y value
max_y = None
for dd in datay.values():
max_y = max(max_y, max(dd))
# Chart size of 400x250 pixels and specifying the range for the Y axis
if title:
chart = SimpleLineChart(1000, 250, title=title, y_range=[0, max_y])
else:
chart = SimpleLineChart(1000, 250, y_range=[0, max_y])
# add the data
for dd in datay.values():
chart.add_data(dd)
# Set the line colours
chart.set_colours(['0000FF', 'FF0000', '00FF00', 'F0F000', '0F0F00'])
# Set the horizontal dotted lines
chart.set_grid(0, 25, 5, 5)
# vertical stripes, the width of each stripe is calculated so that it covers one month
ndatapoints = len(datay.values()[0])
datapermonth = [ ]
for year in xaxis:
for month in xaxis[year]:
datapermonth.append(xaxis[year][month])
# mark months using range markers
stripes = [ ]
stripcols = ('FFFFFF', 'CCCCCC')
wlast = 0
for k in datapermonth:
w = k * 1.0 / ndatapoints
icol = len(stripes)/2 % 2
stripes.append(stripcols[icol])
stripes.append(w)
wlast += w
chart.fill_linear_stripes(Chart.CHART, 0, *stripes)
# y axis labels
if max_y > 30:
left_axis = range(0, int(max_y) + 1, 5)
elif max_y > 10:
left_axis = range(0, int(max_y) + 1, 1)
elif max_y > 5:
left_axis = [ ]
v = 0.
while v < max_y + 0.5:
left_axis.append(v)
v += 0.5
else:
left_axis = [ ]
v = 0.
while v < max_y + 0.1:
left_axis.append(v)
v += 0.1
left_axis[0] = '' # no label at 0
chart.set_axis_labels(Axis.LEFT, left_axis)
# X axis labels
monthlabels = [ ]
for year in xaxis:
for imonth in xaxis[year]:
monthlabels.append(months[imonth])
chart.set_axis_labels(Axis.BOTTOM, monthlabels)
chart.set_axis_labels(Axis.BOTTOM, xaxis.keys()) # years
# the axis is 100 positions long, position month labels in the centre for each month
positions = [ ]
p = 0
for y in xaxis:
datax = xaxis[y]
for k in datax:
w = datax[k] * 100.0 / ndatapoints
positions.append(p + w/2)
p += w
chart.set_axis_positions(1, positions)
# position year labels at the centre of the year
positions = [ ]
p = 0
for y in xaxis:
datax = xaxis[y]
w = sum(datax.values()) * 100.0 / ndatapoints
positions.append(p + w/2)
p += w
chart.set_axis_positions(2, positions)
chart.set_legend([k[0] for k in datay.keys()])
# vertical stripes for marking weeks
#weeks = [ ]
nsundays = 0
daycol = genColourRange(7)
for p in range(ndatapoints):
d = firstweekday + p
if (d % 7) == 0:
assertSunday(p)
chart.add_marker(0, p, 'V', 'FF0000', 1)
#weeks.append(daycol[d % 7])
#weeks.append(1./ndatapoints) # this does not work if the width is less than 0.01
#if len(weeks)/2 == 7: # from now it's repeats
# break
#chart.fill_linear_stripes(Chart.CHART, 0, *weeks)
#chart.add_marker(0, 100, 'V', 'FF0000', 1)
return chart
def render(self,data={}, context={}):
assert self.series is not None, "'chart.series' must be set"
converter = wfcommon.units.Converter(context["units"])
# merge builtin defaults, context and renderer config
config = ChartConfig()
if context.has_key('chart'):
config.__dict__.update(context['chart'])
config.__dict__.update(self.__dict__)
# create the chart
chart = SimpleLineChart(config.width, config.height)
colors = []
legend_set = False
legend = []
chart_min = sys.maxint
chart_max = -sys.maxint
# Prepare series config
ordered_series = []
for key, serie in self.series.iteritems():
serie_config = ChartConfig()
serie_config.__dict__.update(config.__dict__)
serie_config.__dict__.update(serie)
ordered_series.append( (serie_config.order, key, serie) )
ordered_series.sort( cmp=lambda x,y: cmp(x[0],y[0]) )
ordered_keys = []
for order, key, serie in ordered_series:
ordered_keys.append(key)
# Draws for each serie
index=0
for order, key, serie in ordered_series:
serie_config = ChartConfig()
serie_config.__dict__.update(config.__dict__)
serie_config.__dict__.update(serie)
serie_data = copy.copy(data[key.split('.')[0]]['series'][key.split('.')[1]])
measure = key.split('.')[0]
if flat(serie_data): # Series with all data = None
continue
if serie_config.accumulate:
serie_data = accumulate(serie_data)
elif serie_config.interpolate:
serie_data = interpolate(serie_data)
# Compute min and max value for the serie and the whole chart
min_data = amin(serie_data)
chart_min = rmin(chart_min, min_data)
if serie_data.__contains__(min_data):
min_index = serie_data.index(min_data)
else:
min_index = None
max_data = max(serie_data)
chart_max = max(chart_max, max_data)
if serie_data.__contains__(max_data):
max_index = serie_data.index(max_data)
else:
max_index = None
(serie_data, min_index, max_index) = compress_to(serie_data, config.nval, min_index, max_index)
chart.add_data(serie_data)
colors.append(_valid_color(serie_config.color))
if serie_config.max and not max_index == None :
max_config = ChartConfig()
max_config.__dict__.update(serie_config.__dict__)
max_config.__dict__.update(serie_config.max)
str_max_data = str(round(converter.convert(measure, max_data), 1))
chart.add_marker(index, max_index, 't'+str_max_data, _valid_color(max_config.text), max_config.size)
chart.add_marker(index, max_index, max_config.style, _valid_color(max_config.color), max_config.thickness)
if serie_config.min and not min_index == None:
min_config = ChartConfig()
min_config.__dict__.update(serie_config.__dict__)
min_config.__dict__.update(serie_config.min)
str_min_data = str(round(converter.convert(measure, min_data), 1))
chart.add_marker(index, min_index, 't'+str_min_data, _valid_color(min_config.text), min_config.size)
chart.add_marker(index, min_index, min_config.style, _valid_color(min_config.color), min_config.thickness)
if serie_config.last:
last_config = ChartConfig()
last_config.__dict__.update(serie_config.__dict__)
last_config.__dict__.update(serie_config.last)
last_index=len(serie_data)-1
last_data = serie_data[last_index]
if last_data:
str_last_data = str(round(converter.convert(measure, last_data), 1))
chart.add_marker(index, last_index, 't'+str(last_data), _valid_color(last_config.text), last_config.size)
chart.add_marker(index, last_index, last_config.style, _valid_color(last_config.color), last_config.thickness)
if serie_config.area:
fill_config = ChartConfig()
fill_config.__dict__.update(serie_config.__dict__)
fill_config.__dict__.update(serie_config.area)
to = ordered_keys.index(fill_config.to)
chart.add_fill_range(_valid_color(fill_config.color), index, to)
if serie_config.dash:
chart.set_line_style(index, serie_config.thickness, serie_config.dash, serie_config.dash)
else:
chart.set_line_style(index, serie_config.thickness)
if serie_config.legend:
legend.append(serie_config.legend)
legend_set = True
else:
legend.append('')
if serie_config.marks:
mark_config = ChartConfig()
mark_config.__dict__.update(serie_config.__dict__)
mark_config.__dict__.update(serie_config.marks)
mark_data = copy.copy(data[mark_config.serie.split('.')[0]]['series'][mark_config.serie.split('.')[1]])
mark_data = compress_to(mark_data, config.nval, min_index, max_index)[0]
for i, m in enumerate(mark_data):
if not m:
mark_data[i] = " "
density = max(1.0, 1.0 * mark_config.space * len("".join(mark_data))*mark_config.size / config.width)
for i, v in enumerate(mark_data):
if (i +1) % round(density) == 0:
if serie_data[i] != 0:
text = str(mark_data[i])
else:
text = " "
chart.add_marker(index, i, 't'+text, _valid_color(mark_config.color), mark_config.size)
index = index + 1
# Compute vertical range
if config.axes:
range_min_ref_units = 0
if not chart_min == sys.maxint and not chart_max == -sys.maxint:
range_min = chart_min-config.ymargin[0]
range_max = chart_max+config.ymargin[1]
range_min_target_units = math.floor(converter.convert(measure, range_min))
range_max_target_units = math.ceil(converter.convert(measure, range_max))
range_min_ref_units = converter.convert_back(measure, range_min_target_units)
range_max_ref_units = converter.convert_back(measure, range_max_target_units)
self.logger.debug("Y range: "+str(range_min_target_units) +" "+str(range_max_target_units))
chart.set_axis_range(Axis.LEFT, range_min_target_units, range_max_target_units+1)
chart.add_data([range_min_ref_units, range_max_ref_units])
colors.append("00000000")
else:
chart.set_axis_range(Axis.LEFT, 0, 100)
chart.set_axis_style(0, _valid_color(config.text), config.size, 0, Axis.BOTH if config.ticks else Axis.AXIS_LINES)
else:
chart.set_axis_labels(Axis.LEFT, [])
chart.set_axis_style(0, _valid_color(config.text), config.size, 0, Axis.TICK_MARKS, _valid_color(config.bgcolor))
if config.zero and config.axes and range_min_ref_units < 0 and range_max_ref_units > 0:
zero_config = ChartConfig()
zero_config.__dict__.update(config.__dict__)
zero_config.__dict__.update(config.zero)
chart.add_data([0]*2)
colors.append(_valid_color(zero_config.color))
chart.set_line_style(index, zero_config.thickness)
chart.set_colours(colors)
chart.fill_solid(Chart.BACKGROUND, _valid_color(config.bgcolor))
if legend_set:
chart.set_legend(legend)
chart.set_legend_position(config.legend_pos)
if self.labels:
labels_data = copy.copy(data[self.labels.split('.')[0]]['series'][self.labels.split('.')[1]])
labels_data = compress_to(labels_data, config.nval, None, None)[0]
if config.axes:
density = 1.0 * len("".join(labels_data))*config.size / config.width
if density > LABEL_DENSITY_THRESHOLD:
for i, v in enumerate(labels_data):
if i % round(density) != 0:
labels_data[i] = ' '
chart.set_axis_labels(Axis.BOTTOM, labels_data)
chart.set_axis_style(1, _valid_color(config.text), config.size, 0, Axis.BOTH if config.ticks else Axis.AXIS_LINES)
else:
chart.set_axis_labels(Axis.BOTTOM, [])
chart.set_axis_style(1, _valid_color(config.text), config.size, 0, Axis.TICK_MARKS, _valid_color(config.color))
try:
return chart.get_url()+"&chma=10,10,10,10" # add a margin
except:
self.logger.exception("Could not render chart")
return "http://chart.apis.google.com/chart?cht=lc&chs="+str(config.width)+"x"+str(config.height)
def createChart(title, xaxis, datay, firstweekday):
# Find the max y value
max_y = None
for dd in datay.values():
max_y = max(max_y, max(dd))
# Chart size of 400x250 pixels and specifying the range for the Y axis
if title:
chart = SimpleLineChart(1000, 250, title=title, y_range=[0, max_y])
else:
chart = SimpleLineChart(1000, 250, y_range=[0, max_y])
# add the data
for dd in datay.values():
chart.add_data(dd)
# Set the line colours
chart.set_colours(['0000FF', 'FF0000', '00FF00', 'F0F000', '0F0F00'])
# Set the horizontal dotted lines
chart.set_grid(0, 25, 5, 5)
# vertical stripes, the width of each stripe is calculated so that it covers one month
ndatapoints = len(datay.values()[0])
datapermonth = []
for year in xaxis:
for month in xaxis[year]:
datapermonth.append(xaxis[year][month])
# mark months using range markers
stripes = []
stripcols = ('FFFFFF', 'CCCCCC')
wlast = 0
for k in datapermonth:
w = k * 1.0 / ndatapoints
icol = len(stripes) / 2 % 2
stripes.append(stripcols[icol])
stripes.append(w)
wlast += w
chart.fill_linear_stripes(Chart.CHART, 0, *stripes)
# y axis labels
if max_y > 30:
left_axis = range(0, int(max_y) + 1, 5)
elif max_y > 10:
left_axis = range(0, int(max_y) + 1, 1)
elif max_y > 5:
left_axis = []
v = 0.
while v < max_y + 0.5:
left_axis.append(v)
v += 0.5
else:
left_axis = []
v = 0.
while v < max_y + 0.1:
left_axis.append(v)
v += 0.1
left_axis[0] = '' # no label at 0
chart.set_axis_labels(Axis.LEFT, left_axis)
# X axis labels
monthlabels = []
for year in xaxis:
for imonth in xaxis[year]:
monthlabels.append(months[imonth])
chart.set_axis_labels(Axis.BOTTOM, monthlabels)
chart.set_axis_labels(Axis.BOTTOM, xaxis.keys()) # years
# the axis is 100 positions long, position month labels in the centre for each month
positions = []
p = 0
for y in xaxis:
datax = xaxis[y]
for k in datax:
w = datax[k] * 100.0 / ndatapoints
positions.append(p + w / 2)
p += w
chart.set_axis_positions(1, positions)
# position year labels at the centre of the year
positions = []
p = 0
for y in xaxis:
datax = xaxis[y]
w = sum(datax.values()) * 100.0 / ndatapoints
positions.append(p + w / 2)
p += w
chart.set_axis_positions(2, positions)
chart.set_legend([k[0] for k in datay.keys()])
# vertical stripes for marking weeks
#weeks = [ ]
nsundays = 0
daycol = genColourRange(7)
for p in range(ndatapoints):
d = firstweekday + p
if (d % 7) == 0:
assertSunday(p)
chart.add_marker(0, p, 'V', 'FF0000', 1)
#weeks.append(daycol[d % 7])
#weeks.append(1./ndatapoints) # this does not work if the width is less than 0.01
#if len(weeks)/2 == 7: # from now it's repeats
# break
#chart.fill_linear_stripes(Chart.CHART, 0, *weeks)
#chart.add_marker(0, 100, 'V', 'FF0000', 1)
return chart
def render(self, data={}, context={}):
assert self.series is not None, "'chart.series' must be set"
converter = wfcommon.units.Converter(context["units"])
# merge builtin defaults, context and renderer config
config = ChartConfig()
if context.has_key('chart'):
config.__dict__.update(context['chart'])
config.__dict__.update(self.__dict__)
# create the chart
chart = SimpleLineChart(config.width, config.height)
colors = []
legend_set = False
legend = []
chart_min = sys.maxint
chart_max = -sys.maxint
# Prepare series config
ordered_series = []
for key, serie in self.series.iteritems():
serie_config = ChartConfig()
serie_config.__dict__.update(config.__dict__)
serie_config.__dict__.update(serie)
ordered_series.append((serie_config.order, key, serie))
ordered_series.sort(cmp=lambda x, y: cmp(x[0], y[0]))
ordered_keys = []
for order, key, serie in ordered_series:
ordered_keys.append(key)
# Draws for each serie
index = 0
for order, key, serie in ordered_series:
serie_config = ChartConfig()
serie_config.__dict__.update(config.__dict__)
serie_config.__dict__.update(serie)
serie_data = copy.copy(
data[key.split('.')[0]]['series'][key.split('.')[1]])
measure = key.split('.')[0]
if flat(serie_data): # Series with all data = None
continue
if serie_config.accumulate:
serie_data = accumulate(serie_data)
elif serie_config.interpolate:
serie_data = interpolate(serie_data)
# Compute min and max value for the serie and the whole chart
min_data = amin(serie_data)
chart_min = rmin(chart_min, min_data)
if serie_data.__contains__(min_data):
min_index = serie_data.index(min_data)
else:
min_index = None
max_data = max(serie_data)
chart_max = max(chart_max, max_data)
if serie_data.__contains__(max_data):
max_index = serie_data.index(max_data)
else:
max_index = None
(serie_data, min_index,
max_index) = compress_to(serie_data, config.nval, min_index,
max_index)
chart.add_data(serie_data)
colors.append(_valid_color(serie_config.color))
if serie_config.max and not max_index == None:
max_config = ChartConfig()
max_config.__dict__.update(serie_config.__dict__)
max_config.__dict__.update(serie_config.max)
str_max_data = str(
round(converter.convert(measure, max_data), 1))
chart.add_marker(index, max_index, 't' + str_max_data,
_valid_color(max_config.text),
max_config.size)
chart.add_marker(index, max_index, max_config.style,
_valid_color(max_config.color),
max_config.thickness)
if serie_config.min and not min_index == None:
min_config = ChartConfig()
min_config.__dict__.update(serie_config.__dict__)
min_config.__dict__.update(serie_config.min)
str_min_data = str(
round(converter.convert(measure, min_data), 1))
chart.add_marker(index, min_index, 't' + str_min_data,
_valid_color(min_config.text),
min_config.size)
chart.add_marker(index, min_index, min_config.style,
_valid_color(min_config.color),
min_config.thickness)
if serie_config.last:
last_config = ChartConfig()
last_config.__dict__.update(serie_config.__dict__)
last_config.__dict__.update(serie_config.last)
last_index = len(serie_data) - 1
last_data = serie_data[last_index]
if last_data:
str_last_data = str(
round(converter.convert(measure, last_data), 1))
chart.add_marker(index, last_index, 't' + str(last_data),
_valid_color(last_config.text),
last_config.size)
chart.add_marker(index, last_index, last_config.style,
_valid_color(last_config.color),
last_config.thickness)
if serie_config.area:
fill_config = ChartConfig()
fill_config.__dict__.update(serie_config.__dict__)
fill_config.__dict__.update(serie_config.area)
to = ordered_keys.index(fill_config.to)
chart.add_fill_range(_valid_color(fill_config.color), index,
to)
if serie_config.dash:
chart.set_line_style(index, serie_config.thickness,
serie_config.dash, serie_config.dash)
else:
chart.set_line_style(index, serie_config.thickness)
if serie_config.legend:
legend.append(serie_config.legend)
legend_set = True
else:
legend.append('')
if serie_config.marks:
mark_config = ChartConfig()
mark_config.__dict__.update(serie_config.__dict__)
mark_config.__dict__.update(serie_config.marks)
mark_data = copy.copy(data[mark_config.serie.split(
'.')[0]]['series'][mark_config.serie.split('.')[1]])
mark_data = compress_to(mark_data, config.nval, min_index,
max_index)[0]
for i, m in enumerate(mark_data):
if not m:
mark_data[i] = " "
density = max(
1.0, 1.0 * mark_config.space * len("".join(mark_data)) *
mark_config.size / config.width)
for i, v in enumerate(mark_data):
if (i + 1) % round(density) == 0:
if serie_data[i] != 0:
text = str(mark_data[i])
else:
text = " "
chart.add_marker(index, i, 't' + text,
_valid_color(mark_config.color),
mark_config.size)
index = index + 1
# Compute vertical range
if config.axes:
range_min_ref_units = 0
if not chart_min == sys.maxint and not chart_max == -sys.maxint:
range_min = chart_min - config.ymargin[0]
range_max = chart_max + config.ymargin[1]
range_min_target_units = math.floor(
converter.convert(measure, range_min))
range_max_target_units = math.ceil(
converter.convert(measure, range_max))
range_min_ref_units = converter.convert_back(
measure, range_min_target_units)
range_max_ref_units = converter.convert_back(
measure, range_max_target_units)
self.logger.debug("Y range: " + str(range_min_target_units) +
" " + str(range_max_target_units))
chart.set_axis_range(Axis.LEFT, range_min_target_units,
range_max_target_units + 1)
chart.add_data([range_min_ref_units, range_max_ref_units])
colors.append("00000000")
else:
chart.set_axis_range(Axis.LEFT, 0, 100)
chart.set_axis_style(
0, _valid_color(config.text), config.size, 0,
Axis.BOTH if config.ticks else Axis.AXIS_LINES)
else:
chart.set_axis_labels(Axis.LEFT, [])
chart.set_axis_style(0, _valid_color(config.text), config.size, 0,
Axis.TICK_MARKS, _valid_color(config.bgcolor))
if config.zero and config.axes and range_min_ref_units < 0 and range_max_ref_units > 0:
zero_config = ChartConfig()
zero_config.__dict__.update(config.__dict__)
zero_config.__dict__.update(config.zero)
chart.add_data([0] * 2)
colors.append(_valid_color(zero_config.color))
chart.set_line_style(index, zero_config.thickness)
chart.set_colours(colors)
chart.fill_solid(Chart.BACKGROUND, _valid_color(config.bgcolor))
if legend_set:
chart.set_legend(legend)
chart.set_legend_position(config.legend_pos)
if self.labels:
labels_data = copy.copy(data[self.labels.split('.')[0]]['series'][
self.labels.split('.')[1]])
labels_data = compress_to(labels_data, config.nval, None, None)[0]
if config.axes:
density = 1.0 * len(
"".join(labels_data)) * config.size / config.width
if density > LABEL_DENSITY_THRESHOLD:
for i, v in enumerate(labels_data):
if i % round(density) != 0:
labels_data[i] = ' '
chart.set_axis_labels(Axis.BOTTOM, labels_data)
chart.set_axis_style(
1, _valid_color(config.text), config.size, 0,
Axis.BOTH if config.ticks else Axis.AXIS_LINES)
else:
chart.set_axis_labels(Axis.BOTTOM, [])
chart.set_axis_style(1, _valid_color(config.text), config.size,
0, Axis.TICK_MARKS,
_valid_color(config.color))
try:
return chart.get_url() + "&chma=10,10,10,10" # add a margin
except:
self.logger.exception("Could not render chart")
return "http://chart.apis.google.com/chart?cht=lc&chs=" + str(
config.width) + "x" + str(config.height)
