#!/usr/bin/python

import sys, os
from subprocess import call, Popen
from os import listdir
from os.path import isfile, join

print "Usage: SE_deconseq_run.py FastqFile Reference Threads"

try:
    file = sys.argv[1]
except:
    file = raw_input("Introduce a Fastq file: ")

try:
    ref = sys.argv[2]
except:
    ref = raw_input("Introduce FASTA reference: ")

try:
    thr = sys.argv[3]
except:
    thr = raw_input("Introduce number of threads")

refp = ref.split(".")
refpoints = refp[0:-1]
refname = ".".join(refpoints)

#files = open(files).readlines()

dsdir = "deconseq-standalone-0.4.3"

elements = os.listdir(".")
if dsdir not in elements:
    call("cp -r /usr/local/lib/%s ." % dsdir, shell=True)
    call("mkdir %s/db" % dsdir, shell=True)

os.chdir(dsdir+"/db")
call("ln -sf ../../%s" % ref, shell=True)
call("../bwa64 index -p %s -a is %s" % (refname,ref), shell=True)
os.chdir("../")
conf = open("DeconSeqConfig.pm").readlines()
conf_out = open("tmp_conf.txt", "w")
line1 = " "*21+"%s => {name => \047%s\047,\n" % (refname,refname)
line2 = " "*30+"db => \047%s\047},\n" % (refname)
conf.insert(20, line1+line2)
conf_out.write("".join(conf))
conf_out.close()
call("mv tmp_conf.txt DeconSeqConfig.pm", shell=True)
os.chdir("../")

file1 = file
ext1 = file1.split(".")
if ext1[-1] == "gz":
    file1_n = ext1[0:-1]
    file1_n = ".".join(file1_n)
    print "Uncompressing file %s" % file1
    call("seqtk seq %s > %s" % (file1, file1_n), shell=True)
    file1 = file1_n

filename = file1.split(".")
filename = filename[0]
call("mkdir %s/%s" % (dsdir,filename), shell=True)
os.chdir("%s/%s" % (dsdir,filename))
    
call("ln -sf ../../%s ." % file1 , shell=True)

call("FastQ.split.pl %s tmp_queries_1 %s" % (file1, thr), shell=True)

onlyfiles = [f for f in listdir(".") if isfile(join(".",f))]
splits = []
for f in onlyfiles:
    if f.startswith("tmp_queries_1") and f.endswith(".fastq"):
        splits.append(f)
splits.sort()

commands = []
for round in range(0,2):
    commands.append([])

for n in range(0,len(splits)):
    fq = splits[n]
    com = "perl deconseq.pl -f ./%s -out_dir %s.dir -dbs %s" % (fq, fq, refname)
    rr = n/int(thr)
    commands[rr].append(com)

print "Running DeconSeq"
call("ln -s ../db", shell=True)
call("ln -s ../deconseq.pl", shell=True)
call("ln -s ../bwa64", shell=True)
call("ln -s ../DeconSeqConfig.pm", shell=True)
for command in commands:
    processes = [Popen(cmd, shell=True) for cmd in command]
    for p in processes:
        p.wait()

for n in range(1,2):
    concat = ["cat"]
    for fq in splits:
        if fq.startswith("tmp_queries_%s" % (str(n))):
            concat.append("%s.dir/*clean*" % (fq))
    call("%s > %s_clean.fastq" % (" ".join(concat), filename), shell=True)

call("rm db deconseq.pl bwa64 DeconSeqConfig.pm", shell=True)
call("rm tmp_queries*.fastq", shell=True)
call("rm -r tmp_queries*.fastq.dir", shell=True)
call("rm %s" % (file1), shell=True)
os.chdir("../../")

call("mv %s/%s ." % (dsdir,filename), shell=True)

if ext1[-1] == "gz":
    call("rm %s" % (file1), shell=True)

call("rm -r %s" % dsdir, shell=True)