s_mart: smart_toolShed/SMART/Java/Python/CompareOverlappingSmallQuery.py annotate

annotate smart_toolShed/SMART/Java/Python/CompareOverlappingSmallQuery.py @ 1:30c5431d8ce4

Uploaded

author	yufei-luo
date	Thu, 17 Jan 2013 11:01:51 -0500
parents	e0f8dcca02ed
children

rev	line source
0 e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	1 #! /usr/bin/env python
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	2 #
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	3 # Copyright INRA-URGI 2009-2011
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	4 #
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	5 # This software is governed by the CeCILL license under French law and
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	6 # abiding by the rules of distribution of free software. You can use,
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	7 # modify and/ or redistribute the software under the terms of the CeCILL
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	8 # license as circulated by CEA, CNRS and INRIA at the following URL
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	9 # "http://www.cecill.info".
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	10 #
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	11 # As a counterpart to the access to the source code and rights to copy,
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	12 # modify and redistribute granted by the license, users are provided only
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	13 # with a limited warranty and the software's author, the holder of the
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	14 # economic rights, and the successive licensors have only limited
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	15 # liability.
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	16 #
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	17 # In this respect, the user's attention is drawn to the risks associated
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	18 # with loading, using, modifying and/or developing or reproducing the
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	19 # software by the user in light of its specific status of free software,
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	20 # that may mean that it is complicated to manipulate, and that also
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	21 # therefore means that it is reserved for developers and experienced
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	22 # professionals having in-depth computer knowledge. Users are therefore
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	23 # encouraged to load and test the software's suitability as regards their
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	24 # requirements in conditions enabling the security of their systems and/or
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	25 # data to be ensured and, more generally, to use and operate it in the
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	26 # same conditions as regards security.
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	27 #
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	28 # The fact that you are presently reading this means that you have had
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	29 # knowledge of the CeCILL license and that you accept its terms.
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	30 #
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	31 from optparse import OptionParser
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	32 from commons.core.parsing.ParserChooser import ParserChooser
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	33 from commons.core.writer.TranscriptWriter import TranscriptWriter
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	34 from SMART.Java.Python.structure.Interval import Interval
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	35 from SMART.Java.Python.structure.Transcript import Transcript
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	36 from SMART.Java.Python.structure.Mapping import Mapping
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	37 from SMART.Java.Python.misc.Progress import Progress
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	38 from SMART.Java.Python.misc.UnlimitedProgress import UnlimitedProgress
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	39
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	40 MINBIN = 3
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	41 MAXBIN = 7
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	42 REFERENCE = 0
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	43 QUERY = 1
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	44
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	45 def getBin(start, end):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	46 for i in range(MINBIN, MAXBIN + 1):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	47 binLevel = 10 ** i
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	48 if int(start / binLevel) == int(end / binLevel):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	49 return int(i * 10 ** (MAXBIN + 1) + int(start / binLevel))
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	50 return int((MAXBIN + 1) * 10 ** (MAXBIN + 1))
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	51
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	52 def getOverlappingBins(start, end):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	53 array = []
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	54 bigBin = int((MAXBIN + 1) * 10 ** (MAXBIN + 1))
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	55 for i in range(MINBIN, MAXBIN + 1):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	56 binLevel = 10 ** i
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	57 array.append((int(i * 10 ** (MAXBIN + 1) + int(start / binLevel)), int(i * 10 ** (MAXBIN + 1) + int(end / binLevel))))
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	58 array.append((bigBin, bigBin))
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	59 return array
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	60
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	61
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	62 class CompareOverlappingSmallQuery(object):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	63
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	64 def __init__(self, verbosity):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	65 self.verbosity = verbosity
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	66 self.tableNames = {}
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	67 self.nbQueries = 0
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	68 self.nbRefs = 0
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	69 self.nbWritten = 0
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	70 self.nbOverlaps = 0
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	71 self.distance = None
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	72 self.invert = False
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	73 self.antisense = False
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	74 self.collinear = False
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	75 self.bins = {}
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	76 self.overlaps = {}
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	77 self.notOverlapping = False
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	78
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	79 def setReferenceFile(self, fileName, format):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	80 chooser = ParserChooser(self.verbosity)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	81 chooser.findFormat(format)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	82 self.refParser = chooser.getParser(fileName)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	83
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	84 def setQueryFile(self, fileName, format):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	85 chooser = ParserChooser(self.verbosity)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	86 chooser.findFormat(format)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	87 self.queryParser = chooser.getParser(fileName)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	88
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	89 def setOutputFile(self, fileName):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	90 self.writer = TranscriptWriter(fileName, "gff3", self.verbosity)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	91
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	92 def setDistance(self, distance):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	93 self.distance = distance
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	94
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	95 def setInvert(self, boolean):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	96 self.invert = boolean
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	97
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	98 def setCollinear(self, boolean):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	99 self.collinear = boolean
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	100
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	101 def setAntisense(self, boolean):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	102 self.antisense = boolean
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	103
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	104 def includeNotOverlapping(self, boolean):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	105 self.notOverlapping = boolean
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	106
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	107 def loadQuery(self):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	108 progress = UnlimitedProgress(10000, "Reading queries", self.verbosity)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	109 for transcript in self.queryParser.getIterator():
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	110 if transcript.__class__.__name__ == "Mapping":
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	111 transcript = transcript.getTranscript()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	112 chromosome = transcript.getChromosome()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	113 bin = getBin(transcript.getStart(), transcript.getEnd())
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	114 if chromosome not in self.bins:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	115 self.bins[chromosome] = {}
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	116 if bin not in self.bins[chromosome]:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	117 self.bins[chromosome][bin] = []
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	118 self.bins[chromosome][bin].append(transcript)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	119 if self.notOverlapping or self.invert:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	120 self.overlaps[transcript] = {}
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	121 self.nbQueries += 1
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	122 progress.inc()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	123 progress.done()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	124
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	125 def _compareTwoTranscripts(self, queryTranscript, refTranscript):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	126 if not queryTranscript.overlapWithExon(refTranscript):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	127 return False
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	128 if self.collinear and queryTranscript.getDirection() != refTranscript.getDirection():
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	129 return False
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	130 if self.antisense and queryTranscript.getDirection() == refTranscript.getDirection():
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	131 return False
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	132 return True
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	133
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	134 def _alterTranscript(self, transcript, type):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	135 if type == REFERENCE:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	136 if self.distance != None:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	137 transcript.extendExons(self.distance)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	138 return transcript
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	139
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	140 def _compareTranscript(self, refTranscript):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	141 refChromosome = refTranscript.getChromosome()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	142 if refChromosome not in self.bins:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	143 return []
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	144 refStart = refTranscript.getStart()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	145 refEnd = refTranscript.getEnd()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	146 bins = getOverlappingBins(refStart, refEnd)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	147 for binRange in bins:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	148 for bin in range(binRange[0], binRange[1]+1):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	149 if bin not in self.bins[refChromosome]:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	150 continue
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	151 for queryTranscript in self.bins[refChromosome][bin]:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	152 if self._compareTwoTranscripts(queryTranscript, refTranscript):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	153 if queryTranscript not in self.overlaps:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	154 self.overlaps[queryTranscript] = {}
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	155 nbElements = int(float(refTranscript.getTagValue("nbElements"))) if "nbElements" in refTranscript.getTagNames() else 1
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	156 self.overlaps[queryTranscript][refTranscript.getName()] = int(float(refTranscript.getTagValue("nbElements"))) if "nbElements" in refTranscript.getTagNames() else 1
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	157 self.nbOverlaps += nbElements
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	158
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	159 def _updateTranscript(self, queryTranscript):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	160 overlaps = self.overlaps[queryTranscript]
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	161 queryTranscript.setTagValue("nbOverlaps", sum(overlaps.values()))
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	162 if overlaps:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	163 queryTranscript.setTagValue("overlapsWith", "--".join(overlaps.keys())[:100])
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	164 return queryTranscript
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	165
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	166 def compare(self):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	167 progress = UnlimitedProgress(10000, "Comparing references", self.verbosity)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	168 for refTranscript in self.refParser.getIterator():
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	169 if refTranscript.__class__.__name__ == "Mapping":
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	170 refTranscript = refTranscript.getTranscript()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	171 refTranscript = self._alterTranscript(refTranscript, REFERENCE)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	172 self._compareTranscript(refTranscript)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	173 self.nbRefs += 1
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	174 progress.inc()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	175 progress.done()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	176
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	177 def printResults(self):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	178 for transcript in self.overlaps:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	179 if not self.invert or not self.overlaps[transcript]:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	180 if not self.invert:
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	181 transcript = self._updateTranscript(transcript)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	182 self.writer.addTranscript(transcript)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	183 self.nbWritten += 1
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	184 self.writer.close()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	185
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	186 def displayResults(self):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	187 print "# queries: %d" % (self.nbQueries)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	188 print "# refs: %d" % (self.nbRefs)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	189 print "# written: %d (%d overlaps)" % (self.nbWritten, self.nbOverlaps)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	190
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	191 def run(self):
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	192 self.loadQuery()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	193 self.compare()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	194 self.printResults()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	195 self.displayResults()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	196
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	197 if __name__ == "__main__":
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	198
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	199 description = "Compare Overlapping Small Query v1.0.1: Provide the queries that overlap with a reference, when the query is small. [Category: Data Comparison]"
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	200
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	201 parser = OptionParser(description = description)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	202 parser.add_option("-i", "--input1", dest="inputFileName1", action="store", type="string", help="query input file [compulsory] [format: file in transcript format given by -f]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	203 parser.add_option("-f", "--format1", dest="format1", action="store", type="string", help="format of previous file [compulsory] [format: transcript file format]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	204 parser.add_option("-j", "--input2", dest="inputFileName2", action="store", type="string", help="reference input file [compulsory] [format: file in transcript format given by -g]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	205 parser.add_option("-g", "--format2", dest="format2", action="store", type="string", help="format of previous file [compulsory] [format: transcript file format]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	206 parser.add_option("-o", "--output", dest="outputFileName", action="store", type="string", help="output file [format: output file in GFF3 format]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	207 parser.add_option("-O", "--notOverlapping", dest="notOverlapping", action="store_true", default=False, help="also output not overlapping data [format: bool] [default: false]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	208 parser.add_option("-d", "--distance", dest="distance", action="store", default=0, type="int", help="accept some distance between query and reference [format: int]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	209 parser.add_option("-c", "--collinear", dest="collinear", action="store_true", default=False, help="provide collinear features [format: bool] [default: false]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	210 parser.add_option("-a", "--antisense", dest="antisense", action="store_true", default=False, help="provide antisense features [format: bool] [default: false]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	211 parser.add_option("-x", "--exclude", dest="exclude", action="store_true", default=False, help="invert the match [format: bool] [default: false]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	212 parser.add_option("-v", "--verbosity", dest="verbosity", action="store", default=1, type="int", help="trace level [format: int]")
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	213 (options, args) = parser.parse_args()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	214
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	215 cosq = CompareOverlappingSmallQuery(options.verbosity)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	216 cosq.setQueryFile(options.inputFileName1, options.format1)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	217 cosq.setReferenceFile(options.inputFileName2, options.format2)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	218 cosq.setOutputFile(options.outputFileName)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	219 cosq.includeNotOverlapping(options.notOverlapping)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	220 cosq.setDistance(options.distance)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	221 cosq.setCollinear(options.collinear)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	222 cosq.setAntisense(options.antisense)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	223 cosq.setInvert(options.exclude)
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	224 cosq.run()
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	225
e0f8dcca02ed Uploaded S-MART tool. A toolbox manages RNA-Seq and ChIP-Seq data. yufei-luo parents: diff changeset	226

Mercurial > repos > yufei-luo > s_mart

annotate smart_toolShed/SMART/Java/Python/CompareOverlappingSmallQuery.py @ 1:30c5431d8ce4