#!/usr/bin/python
# Adds ANOVA results to the collapsed and non-collapsed tables
# Greg Donahue, 05-05-2016

import sys, string

def main(args):
    anova = dict()
    f = open("ANOVA Table with FDR.txt"); lines = f.readlines(); f.close()
    for line in lines[1:]:
        t = line[:-1].split("\t")
        anova[t[1]] = (t[2],t[3])
    f = open("Multipeak Table.MTLs.Corrected.txt")
    lines = f.readlines(); f.close()
    g = open("Multipeak Table.MTLs.Corrected.ANOVA.txt", 'w')
    g.write(lines[0][:-1]+"\tANOVA p-value\tANOVA q-value\n")
    for line in lines[1:]:
        t = line[:-1].split("\t")
        A = anova[t[0]]
        g.write(line[:-1]+"\t"+A[0]+"\t"+A[1]+"\n")
    g.close()
    f = open("Multipeak Table.MTLs.Collapsed.Corrected.txt")
    lines = f.readlines(); f.close()
    g = open("Multipeak Table.MTLs.Collapsed.Corrected.ANOVA.txt", 'w')
    g.write(lines[0][:-1]+"\tANOVA p-value\tANOVA q-value\n")
    for line in lines[1:]:
        t = line[:-1].split("\t")
        A = anova[t[0]]
        g.write(line[:-1]+"\t"+A[0]+"\t"+A[1]+"\n")
    g.close()
    
if __name__ == "__main__": main(sys.argv)