Mercurial > repos > simon-gladman > vennt

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/vennt.py	Mon Feb 24 21:44:41 2014 -0500
@@ -0,0 +1,126 @@
+#!/usr/bin/env python
+
+import argparse
+import json
+import re
+import sys,os
+import csv, StringIO
+
+bigFC = 100
+
+def error(message):
+    sys.stderr.write("Error: %s\n" % message)
+    sys.exit(1)
+
+def embed(csv, args):
+    html="""
+<html>
+  <head>
+
+    <link rel="stylesheet" href="http://netdna.bootstrapcdn.com/bootstrap/3.0.0/css/bootstrap.min.css" />
+    <link rel="stylesheet" href="http://ajax.googleapis.com/ajax/libs/jqueryui/1.10.3/themes/ui-lightness/jquery-ui.min.css" />
+
+    <link rel="stylesheet" type="text/css" href='http://drpowell.github.io/vennt/dist/main.min.css'>
+    <script type="text/javascript" src='http://drpowell.github.io/vennt/dist/main.js'></script>
+  </head>
+
+  <body>
+    <script type="text/javascript">
+      window.venn_settings = { };
+    </script>
+
+    <div id='loading'><img src='http://drpowell.github.io/vennt/dist/images/ajax-loader.gif'></div>
+  </body>
+</html>
+
+         """
+    enc = json.dumps(csv)
+    settings = ("window.venn_settings = {key_column: %s, id_column: %s, fdr_column: %s,"
+                "logFC_column: %s, info_columns: %s, csv_data: data};")%(
+                  json.dumps(args.key), json.dumps(args.id), json.dumps(args.fdr),
+                  json.dumps(args.logFC), json.dumps(args.info))
+    s = html.replace('window.venn_settings = { };', "var data=%s;\n\n%s"%(enc,settings), 1)
+    return s
+
+def combine_csv(files,key):
+    data = []
+    sys.stderr.write("Using a separate CSV files\n")
+    for f in files:
+        sys.stderr.write("  Reading : %s\n"%f)
+        d = open(f).read()
+        # Separate header (and keep if it is the first)
+        hdr, d = d.split("\n",1)
+        if len(data)==0:
+            data.append('"%s",'%(key)+hdr+"\n")
+        d = re.sub(r'^(.{2})',r'"%s",\1'%os.path.splitext(os.path.basename(f))[0], d, 0, re.MULTILINE)   # Add a key column to all rows
+        data.append(d)
+
+    return ''.join(data)
+
+def cuffdiff_process(f):
+    with open(f, 'r') as csvfile:
+        reader = csv.reader(csvfile, delimiter="\t")
+        si = StringIO.StringIO()
+        cw = csv.writer(si, delimiter=",")
+
+        headers = reader.next()
+        cw.writerow(headers + ['key'])
+        idx1 = headers.index("sample_1")
+        idx2 = headers.index("sample_2")
+        fcIdx = headers.index("log2(fold_change)")
+        for r in reader:
+            # Replace an infinite fold-change with something vennt can handle
+            if r[fcIdx]=='inf':
+                r[fcIdx]=bigFC
+            if r[fcIdx]=='-inf':
+                r[fcIdx]=-bigFC
+            k = r[idx1] + ' vs ' + r[idx2]
+            cw.writerow(r + [k])
+
+        return si.getvalue()
+
+parser = argparse.ArgumentParser(description='Produce a standalone Vennt html file from a CSV file containing gene-lists.  You may use a single CSV file containing all the gene lists - in which case you should have a "key" column specifying the gene lists.  Alternatively, you can use separate CSV files for each gene list then a "key" column will be created based on the filenames.  With separate CSV files they are expected to be in the same format with the same column names in the same column order.')
+parser.add_argument('csvfile',
+                    nargs='*', default='-',
+                    help="CSV file to process (default stdin).  Multiple files may be specified - in which case it is assumed each file contains one gene list and the filenames will be used to create a 'key' column")
+parser.add_argument('-o','--out', type=argparse.FileType('w'),
+                    default='-',
+                    help="Output file (default stdout)")
+parser.add_argument('--key', default='key',
+                    help='Name for "key" column in CSV file (default "key").  Ignored if using multiple CSV files.')
+parser.add_argument('--id', default='Feature',
+                    help='Name for "id" column in CSV file (default "Feature")')
+parser.add_argument('--fdr', default='adj.P.Val',
+                    help='Name for "FDR" column in CSV file (default "adj.P.Val")')
+parser.add_argument('--logFC', default='logFC',
+                    help='Name for "logFC" column in CSV file (default "logFC")')
+parser.add_argument('--info', default=['Feature'], nargs='*',
+                    help='Names for info columns in CSV file - accepts multiple strings (default "Feature")')
+parser.add_argument('--cuffdiff', action='store_true', default=False,
+                    help='Input file is from cuffdiff (gene_exp.diff).  Other options will be ignored')
+
+args = parser.parse_args()
+
+#print args
+
+csv_data = None
+if args.csvfile == '-':
+    sys.stderr.write("Reading from stdin...\n")
+    csv_data = sys.stdin.read()
+elif len(args.csvfile)==1:
+    if args.cuffdiff:
+        csv_data = cuffdiff_process(args.csvfile[0])
+        args.id  = 'test_id'
+        args.fdr = 'q_value'
+        args.logFC = 'log2(fold_change)'
+        args.info = ['gene_id','gene']
+    else:
+        sys.stderr.write("Using a single CSV file with the key column '%s'\n"%(args.key))
+        csv_data = open(args.csvfile[0],'r').read()
+else:
+    if args.cuffdiff:
+        error("Only 1 file (gene_exp.diff) expected when using --cuffdiff")
+    csv_data = combine_csv(args.csvfile, args.key)
+
+args.out.write(embed(csv_data, args))
+