# Copyright (C) University of Tennessee Health Science Center, Memphis, TN.
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License
# as published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
# This program is available from Source Forge: at GeneNetwork Project
# (sourceforge.net/projects/genenetwork/).
#
# Contact Drs. Robert W. Williams and Xiaodong Zhou (2010)
# at rwilliams@uthsc.edu and xzhou15@uthsc.edu
#
#
#
# This module is used by GeneNetwork project (www.genenetwork.org)
#
# Created by GeneNetwork Core Team 2010/08/10
#
# Last updated by GeneNetwork Core Team 2010/10/20
from PIL import ImageColor
from PIL import ImageDraw
from PIL import ImageFont
from math import *
import utility.corestats as corestats
from base import webqtlConfig
from utility.pillow_utils import draw_rotated_text
import utility.logger
logger = utility.logger.getLogger(__name__)
# ---- Define common colours ---- #
BLUE = ImageColor.getrgb("blue")
BLACK = ImageColor.getrgb("black")
# ---- END: Define common colours ---- #
# ---- FONT FILES ---- #
VERDANA_FILE = "./wqflask/static/fonts/verdana.ttf"
COUR_FILE = "./wqflask/static/fonts/courbd.ttf"
TAHOMA_FILE = "./wqflask/static/fonts/tahoma.ttf"
# ---- END: FONT FILES ---- #
def cformat(d, rank=0):
'custom string format'
strD = "%2.6f" % d
if rank == 0:
while strD[-1] in ('0', '.'):
if strD[-1] == '0' and strD[-2] == '.' and len(strD) <= 4:
break
elif strD[-1] == '.':
strD = strD[:-1]
break
else:
strD = strD[:-1]
else:
strD = strD.split(".")[0]
if strD == '-0.0':
strD = '0.0'
return strD
def frange(start, end=None, inc=1.0):
"A faster range-like function that does accept float increments..."
if end == None:
end = start + 0.0
start = 0.0
else:
start += 0.0 # force it to be a float
count = int((end - start) / inc)
if start + count * inc != end:
# Need to adjust the count. AFAICT, it always comes up one short.
count += 1
L = [start] * count
for i in range(1, count):
L[i] = start + i * inc
return L
def find_outliers(vals):
"""Calculates the upper and lower bounds of a set of sample/case values
>>> find_outliers([3.504, 5.234, 6.123, 7.234, 3.542, 5.341, 7.852, 4.555, 12.537])
(11.252500000000001, 0.5364999999999993)
>>> find_outliers([9,12,15,17,31,50,7,5,6,8])
(32.0, -8.0)
If there are no vals, returns None for the upper and lower bounds,
which code that calls it will have to deal with.
>>> find_outliers([])
(None, None)
"""
if vals:
#logger.debug("vals is:", pf(vals))
stats = corestats.Stats(vals)
low_hinge = stats.percentile(25)
up_hinge = stats.percentile(75)
hstep = 1.5 * (up_hinge - low_hinge)
upper_bound = up_hinge + hstep
lower_bound = low_hinge - hstep
else:
upper_bound = None
lower_bound = None
return upper_bound, lower_bound
# parameter: data is either object returned by reaper permutation function (called by MarkerRegressionPage.py)
# or the first object returned by direct (pair-scan) permu function (called by DirectPlotPage.py)
def plotBar(canvas, data, barColor=BLUE, axesColor=BLACK, labelColor=BLACK, XLabel=None, YLabel=None, title=None, offset=(60, 20, 40, 40), zoom=1):
im_drawer = ImageDraw.Draw(canvas)
xLeftOffset, xRightOffset, yTopOffset, yBottomOffset = offset
plotWidth = canvas.size[0] - xLeftOffset - xRightOffset
plotHeight = canvas.size[1] - yTopOffset - yBottomOffset
if plotHeight <= 0 or plotWidth <= 0:
return
if len(data) < 2:
return
max_D = max(data)
min_D = min(data)
# add by NL 06-20-2011: fix the error: when max_D is infinite, log function in detScale will go wrong
if (max_D == float('inf') or max_D > webqtlConfig.MAXLRS) and min_D < webqtlConfig.MAXLRS:
max_D = webqtlConfig.MAXLRS # maximum LRS value
xLow, xTop, stepX = detScale(min_D, max_D)
# reduce data
# ZS: Used to determine number of bins for permutation output
step = ceil((xTop - xLow) / 50.0)
j = xLow
dataXY = []
Count = []
while j <= xTop:
dataXY.append(j)
Count.append(0)
j += step
for i, item in enumerate(data):
if (item == float('inf') or item > webqtlConfig.MAXLRS) and min_D < webqtlConfig.MAXLRS:
item = webqtlConfig.MAXLRS # maximum LRS value
j = int((item - xLow) / step)
Count[j] += 1
yLow, yTop, stepY = detScale(0, max(Count))
# draw data
xScale = plotWidth / (xTop - xLow)
yScale = plotHeight / (yTop - yLow)
barWidth = xScale * step
for i, count in enumerate(Count):
if count:
xc = (dataXY[i] - xLow) * xScale + xLeftOffset
yc = -(count - yLow) * yScale + yTopOffset + plotHeight
im_drawer.rectangle(
xy=((xc + 2, yc), (xc + barWidth - 2, yTopOffset + plotHeight)),
outline=barColor, fill=barColor)
# draw drawing region
im_drawer.rectangle(
xy=((xLeftOffset, yTopOffset),
(xLeftOffset + plotWidth, yTopOffset + plotHeight))
)
# draw scale
scaleFont = ImageFont.truetype(font=COUR_FILE, size=11)
x = xLow
for i in range(int(stepX) + 1):
xc = xLeftOffset + (x - xLow) * xScale
im_drawer.line(
xy=((xc, yTopOffset + plotHeight),
(xc, yTopOffset + plotHeight + 5)),
fill=axesColor)
strX = cformat(d=x, rank=0)
im_drawer.text(
text=strX,
xy=(xc - im_drawer.textsize(strX, font=scaleFont)[0] / 2,
yTopOffset + plotHeight + 14), font=scaleFont)
x += (xTop - xLow) / stepX
y = yLow
for i in range(int(stepY) + 1):
yc = yTopOffset + plotHeight - (y - yLow) * yScale
im_drawer.line(
xy=((xLeftOffset, yc), (xLeftOffset - 5, yc)), fill=axesColor)
strY = "%d" % y
im_drawer.text(
text=strY,
xy=(xLeftOffset - im_drawer.textsize(strY,
font=scaleFont)[0] - 6, yc + 5),
font=scaleFont)
y += (yTop - yLow) / stepY
# draw label
labelFont = ImageFont.truetype(font=TAHOMA_FILE, size=17)
if XLabel:
im_drawer.text(
text=XLabel,
xy=(xLeftOffset + (
plotWidth - im_drawer.textsize(XLabel, font=labelFont)[0]) / 2.0,
yTopOffset + plotHeight + yBottomOffset - 10),
font=labelFont, fill=labelColor)
if YLabel:
draw_rotated_text(canvas, text=YLabel,
xy=(19,
yTopOffset + plotHeight - (
plotHeight - im_drawer.textsize(
YLabel, font=labelFont)[0]) / 2.0),
font=labelFont, fill=labelColor, angle=90)
labelFont = ImageFont.truetype(font=VERDANA_FILE, size=16)
if title:
im_drawer.text(
text=title,
xy=(xLeftOffset + (plotWidth - im_drawer.textsize(
title, font=labelFont)[0]) / 2.0,
20),
font=labelFont, fill=labelColor)
# This function determines the scale of the plot
def detScaleOld(min, max):
if min >= max:
return None
elif min == -1.0 and max == 1.0:
return [-1.2, 1.2, 12]
else:
a = max - min
b = floor(log10(a))
c = pow(10.0, b)
if a < c * 5.0:
c /= 2.0
# print a,b,c
low = c * floor(min / c)
high = c * ceil(max / c)
return [low, high, round((high - low) / c)]
def detScale(min=0, max=0):
if min >= max:
return None
elif min == -1.0 and max == 1.0:
return [-1.2, 1.2, 12]
else:
a = max - min
if max != 0:
max += 0.1 * a
if min != 0:
if min > 0 and min < 0.1 * a:
min = 0.0
else:
min -= 0.1 * a
a = max - min
b = floor(log10(a))
c = pow(10.0, b)
low = c * floor(min / c)
high = c * ceil(max / c)
n = round((high - low) / c)
div = 2.0
while n < 5 or n > 15:
if n < 5:
c /= div
else:
c *= div
if div == 2.0:
div = 5.0
else:
div = 2.0
low = c * floor(min / c)
high = c * ceil(max / c)
n = round((high - low) / c)
return [low, high, n]
def bluefunc(x):
return 1.0 / (1.0 + exp(-10 * (x - 0.6)))
def redfunc(x):
return 1.0 / (1.0 + exp(10 * (x - 0.5)))
def greenfunc(x):
return 1 - pow(redfunc(x + 0.2), 2) - bluefunc(x - 0.3)
def colorSpectrum(n=100):
multiple = 10
if n == 1:
return [ImageColor.getrgb("rgb(100%,0%,0%)")]
elif n == 2:
return [ImageColor.getrgb("100%,0%,0%)"),
ImageColor.getrgb("rgb(0%,0%,100%)")]
elif n == 3:
return [ImageColor.getrgb("rgb(100%,0%,0%)"),
ImageColor.getrgb("rgb(0%,100%,0%)"),
ImageColor.getrgb("rgb(0%,0%,100%)")]
N = n * multiple
out = [None] * N
for i in range(N):
x = float(i) / N
out[i] = ImageColor.getrgb("rgb({}%,{}%,{}%".format(
*[int(i * 100) for i in (
redfunc(x), greenfunc(x), bluefunc(x))]))
out2 = [out[0]]
step = N / float(n - 1)
j = 0
for i in range(n - 2):
j += step
out2.append(out[int(j)])
out2.append(out[-1])
return out2
def _test():
import doctest
doctest.testmod()
if __name__ == "__main__":
_test()
