mirror of
https://github.com/galaxyproject/galaxy.git
synced 2026-09-24 07:12:04 +08:00
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
#!/usr/bin/env python
|
|
#Greg Von Kuster
|
|
|
|
import sys
|
|
from rpy import *
|
|
|
|
def stop_err(msg):
|
|
sys.stderr.write(msg)
|
|
sys.exit()
|
|
|
|
def main():
|
|
|
|
in_fname = sys.argv[1]
|
|
out_fname = sys.argv[2]
|
|
try:
|
|
columns = int( sys.argv[3] ) - 1, int( sys.argv[4] ) - 1
|
|
except:
|
|
stop_err( "Columns not specified, your query does not contain a column of numerical data." )
|
|
title = sys.argv[5]
|
|
xlab = sys.argv[6]
|
|
ylab = sys.argv[7]
|
|
out_type = sys.argv[8]
|
|
out_width = int(sys.argv[9])
|
|
out_height = int(sys.argv[10])
|
|
point_size = float(sys.argv[11])
|
|
|
|
|
|
xvec=[]
|
|
yvec=[]
|
|
skipped_lines = 0
|
|
first_invalid_line = 0
|
|
invalid_value = ''
|
|
invalid_column = 0
|
|
i = 0
|
|
for i, line in enumerate( file( in_fname ) ):
|
|
valid = True
|
|
vals = []
|
|
line = line.rstrip( '\r\n' )
|
|
if line and not line.startswith( '#' ):
|
|
row = []
|
|
fields = line.split( "\t" )
|
|
for c,column in enumerate(columns):
|
|
try:
|
|
val = fields[column]
|
|
if val.lower() == "na":
|
|
v = float( "nan" )
|
|
else:
|
|
v = float( fields[column] )
|
|
vals.append(val)
|
|
except:
|
|
valid = False
|
|
skipped_lines += 1
|
|
if not first_invalid_line:
|
|
first_invalid_line = i + 1
|
|
try:
|
|
invalid_value = fields[column]
|
|
except:
|
|
invalid_value = ''
|
|
invalid_column = column + 1
|
|
break
|
|
else:
|
|
valid = False
|
|
skipped_lines += 1
|
|
if not first_invalid_line:
|
|
first_invalid_line = i+1
|
|
|
|
if valid:
|
|
xvec.append(vals[0])
|
|
yvec.append(vals[1])
|
|
if skipped_lines < i:
|
|
try:
|
|
if out_type == "jpg":
|
|
r.jpeg(out_fname,width=out_width,height=out_height)
|
|
elif out_type == "png":
|
|
# type="cairo" needed to be set for headless servers
|
|
r.png(out_fname,type="cairo",width=out_width,height=out_height)
|
|
else:
|
|
r.pdf(out_fname, out_width, out_height)
|
|
|
|
r.plot(xvec,yvec, type="p", main=title, xlab=xlab, ylab=ylab, col="blue", pch=19,cex=point_size )
|
|
r.dev_off()
|
|
except Exception, exc:
|
|
stop_err( "%s" %str( exc ) )
|
|
else:
|
|
stop_err( "All values in both columns %s and %s are non-numeric or empty." % ( sys.argv[3], sys.argv[4] ) )
|
|
|
|
print "Scatter plot on columns %s, %s. " % ( sys.argv[3], sys.argv[4] )
|
|
if skipped_lines > 0:
|
|
print "Skipped %d lines starting with line #%d, value '%s' in column %d is not numeric." % ( skipped_lines, first_invalid_line, invalid_value, invalid_column )
|
|
|
|
r.quit( save="no" )
|
|
|
|
if __name__ == "__main__":
|
|
main()
|