comparison pileup_interval.py @ 4:9c1c0b947e46 draft default tip

"planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/tools/pileup_interval commit 8b2095c59ecc2e94c58a42e2e04dbcecdc823dbf"

3:ef11139d4545

   -s, --seq_column=s: Sequence column
   -l, --loc_column=l: Base location column
   -r, --base_column=r: Reference base column
   -C, --cvrg_column=C: Coverage column
"""

from galaxy import eggs
import pkg_resources; pkg_resources.require( "bx-python" )
from bx.cookbook import doc_optparse
import sys

def stop_err( msg ):
    sys.stderr.write( msg )
    sys.exit()

def __main__():
    strout = ''
    #Parse Command Line
    options, args = doc_optparse.parse( __doc__ )
    coverage = int(options.coverage)
    fin = file(options.input, 'r')
    fout = file(options.output, 'w')
    inLine = fin.readline()
    if options.format == 'six':
        seqIndex = 0
        locIndex = 1
        baseIndex = 2

    bases = []
    while inLine.strip() != '':
        lineParts = inLine.split('\t')
        try:
            seq, loc, base, cov = lineParts[seqIndex], int(lineParts[locIndex]), lineParts[baseIndex], int(lineParts[covIndex])
        except IndexError, ei:
            if options.format == 'ten':
                stop_err( 'It appears that you have selected 10 columns while your file has 6. Make sure that the number of columns you specify matches the number in your file.\n' + str( ei ) )
            else:
                stop_err( 'There appears to be something wrong with your column index values.\n' + str( ei ) )
        except ValueError, ev:
            if options.format == 'six':
                stop_err( 'It appears that you have selected 6 columns while your file has 10. Make sure that the number of columns you specify matches the number in your file.\n' + str( ev ) )
            else:
                stop_err( 'There appears to be something wrong with your column index values.\n' + str( ev ) )
#        strout += str(startLoc) + '\n'
#        strout += str(bases) + '\n'
#        strout += '%s\t%s\t%s\t%s\n' % (seq, loc, base, cov)
        if loc == lastLoc+1 or lastLoc == -1:
            if cov >= coverage:
                if seq == lastSeq or lastSeq == '':
                    if startLoc == -1:
                        startLoc = loc
                    locs.append(loc)
                    bases.append(base)
                else:
                    if len(bases) > 0:
                        fout.write('%s\t%s\t%s\t%s\n' % (lastSeq, startLoc-1, lastLoc, ''.join(bases)))
                    startLoc = loc
                    locs = [loc]
                    bases = [base]
            else:
                if len(bases) > 0:
                    fout.write('%s\t%s\t%s\t%s\n' % (lastSeq, startLoc-1, lastLoc, ''.join(bases)))
                startLoc = -1
                locs = []
                bases = []
        else:
            if len(bases) > 0:
                fout.write('%s\t%s\t%s\t%s\n' % (lastSeq, startLoc-1, lastLoc, ''.join(bases)))
            if cov >= coverage:
                startLoc = loc
                locs = [loc]
                bases = [base]
            else:

                bases = []
        lastSeq = seq
        lastLoc = loc
        inLine = fin.readline()
    if len(bases) > 0:
        fout.write('%s\t%s\t%s\t%s\n' % (lastSeq, startLoc-1, lastLoc, ''.join(bases)))
    fout.close()
    fin.close()
    
#    import sys
#    strout += file(fout.name,'r').read()
#    sys.stderr.write(strout)

if __name__ == "__main__" : __main__()

4:9c1c0b947e46

   -s, --seq_column=s: Sequence column
   -l, --loc_column=l: Base location column
   -r, --base_column=r: Reference base column
   -C, --cvrg_column=C: Coverage column
"""
import sys

from bx.cookbook import doc_optparse


def __main__():
    # Parse Command Line
    options, args = doc_optparse.parse(__doc__)
    coverage = int(options.coverage)
    fin = open(options.input, 'r')
    fout = open(options.output, 'w')
    inLine = fin.readline()
    if options.format == 'six':
        seqIndex = 0
        locIndex = 1
        baseIndex = 2

    bases = []
    while inLine.strip() != '':
        lineParts = inLine.split('\t')
        try:
            seq, loc, base, cov = lineParts[seqIndex], int(lineParts[locIndex]), lineParts[baseIndex], int(lineParts[covIndex])
        except IndexError as ei:
            if options.format == 'ten':
                sys.exit('It appears that you have selected 10 columns while your file has 6. Make sure that the number of columns you specify matches the number in your file.\n' + str(ei))
            else:
                sys.exit('There appears to be something wrong with your column index values.\n' + str(ei))
        except ValueError as ev:
            if options.format == 'six':
                sys.exit('It appears that you have selected 6 columns while your file has 10. Make sure that the number of columns you specify matches the number in your file.\n' + str(ev))
            else:
                sys.exit('There appears to be something wrong with your column index values.\n' + str(ev))
#        strout += str(startLoc) + '\n'
#        strout += str(bases) + '\n'
#        strout += '%s\t%s\t%s\t%s\n' % (seq, loc, base, cov)
        if loc == lastLoc + 1 or lastLoc == -1:
            if cov >= coverage:
                if seq == lastSeq or lastSeq == '':
                    if startLoc == -1:
                        startLoc = loc
                    locs.append(loc)
                    bases.append(base)
                else:
                    if len(bases) > 0:
                        fout.write('%s\t%s\t%s\t%s\n' % (lastSeq, startLoc - 1, lastLoc, ''.join(bases)))
                    startLoc = loc
                    locs = [loc]
                    bases = [base]
            else:
                if len(bases) > 0:
                    fout.write('%s\t%s\t%s\t%s\n' % (lastSeq, startLoc - 1, lastLoc, ''.join(bases)))
                startLoc = -1
                locs = []
                bases = []
        else:
            if len(bases) > 0:
                fout.write('%s\t%s\t%s\t%s\n' % (lastSeq, startLoc - 1, lastLoc, ''.join(bases)))
            if cov >= coverage:
                startLoc = loc
                locs = [loc]
                bases = [base]
            else:

                bases = []
        lastSeq = seq
        lastLoc = loc
        inLine = fin.readline()
    if len(bases) > 0:
        fout.write('%s\t%s\t%s\t%s\n' % (lastSeq, startLoc - 1, lastLoc, ''.join(bases)))
    fout.close()
    fin.close()
#    import sys
#    strout += file(fout.name,'r').read()
#    sys.stderr.write(strout)


if __name__ == "__main__":
    __main__()