diff SMART/Java/Python/structure/SubMapping.py @ 38:2c0c0a89fad7

Uploaded
author m-zytnicki
date Thu, 02 May 2013 09:56:47 -0400
parents 769e306b7933
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/SMART/Java/Python/structure/SubMapping.py	Thu May 02 09:56:47 2013 -0400
@@ -0,0 +1,258 @@
+#
+# Copyright INRA-URGI 2009-2010
+# 
+# This software is governed by the CeCILL license under French law and
+# abiding by the rules of distribution of free software. You can use,
+# modify and/ or redistribute the software under the terms of the CeCILL
+# license as circulated by CEA, CNRS and INRIA at the following URL
+# "http://www.cecill.info".
+# 
+# As a counterpart to the access to the source code and rights to copy,
+# modify and redistribute granted by the license, users are provided only
+# with a limited warranty and the software's author, the holder of the
+# economic rights, and the successive licensors have only limited
+# liability.
+# 
+# In this respect, the user's attention is drawn to the risks associated
+# with loading, using, modifying and/or developing or reproducing the
+# software by the user in light of its specific status of free software,
+# that may mean that it is complicated to manipulate, and that also
+# therefore means that it is reserved for developers and experienced
+# professionals having in-depth computer knowledge. Users are therefore
+# encouraged to load and test the software's suitability as regards their
+# requirements in conditions enabling the security of their systems and/or
+# data to be ensured and, more generally, to use and operate it in the
+# same conditions as regards security.
+# 
+# The fact that you are presently reading this means that you have had
+# knowledge of the CeCILL license and that you accept its terms.
+#
+from SMART.Java.Python.structure.Interval import Interval
+from commons.core.coord.Align import Align
+
+class SubMapping(Align):
+    """
+    A class that represents a part of a mapping, more precisely, a pair (target interval, query interval) that match together
+    @ivar targetInterval: the target interval
+    @type targetInterval: class L{Interval<Interval>}
+    @ivar queryInterval:  the query interval
+    @type queryInterval:  class L{Interval<Interval>}
+    @ivar size:           size of this sub-mapping
+    @type size:           int
+    @ivar tags:           various information
+    @type tags:           dict
+    """
+
+    def __init__(self, subMapping = None):
+        """
+        Constructor
+        @param subMapping: a sub-mapping to be copied
+        @type  subMapping: class L{SubMapping<SubMapping>}
+        """
+        self.targetInterval = Interval()
+        self.queryInterval  = Interval()
+        Align.__init__(self, self.queryInterval, self.targetInterval)
+        self.size = None
+        self.tags = {}
+        if subMapping != None:
+            self.copy(subMapping)
+    
+    def __eq__(self, o):
+        if o == None:
+            return False
+        areAlignAttributesEquals = Align.__eq__(self, o)
+        return areAlignAttributesEquals and (self.targetInterval == o.targetInterval) and (self.queryInterval == o.queryInterval) and self.size == o.getSize() and self.tags == o.getTags()
+    
+    def getSuperAdress(self):
+        return hex(id(super(Align, self)))
+    
+#    def setRangesAlignToRangesInterval(self):
+#        self.range_query = super(Range, self.queryInterval)
+#        self.range_subject = super(Range, self.targetInterval)
+        
+    def copy(self, subMapping):
+        """
+        Copy method
+        @param subMapping: a sub-mapping to be copied
+        @type    subMapping: class L{SubMapping<SubMapping>}
+        """
+        self.setQueryName(subMapping.getQueryName())
+        self.setQueryStart(subMapping.getQueryStart())
+        self.setQueryEnd(subMapping.getQueryEnd())
+        self.setSubjectName(subMapping.getSubjectName())
+        self.setSubjectStart(subMapping.getSubjectStart())
+        self.setSubjectEnd(subMapping.getSubjectEnd())
+        self.e_value = subMapping.getEvalue()
+        self.score = subMapping.getScore()
+        self.identity = subMapping.getIdentity()
+        
+        self.targetInterval.copy(subMapping.targetInterval)
+        self.queryInterval.copy(subMapping.queryInterval)
+        self.size = subMapping.size
+        for tag in subMapping.tags:
+            self.tags[tag] = subMapping.tags[tag]
+
+
+    def setTargetInterval(self, interval):
+        """
+        Set target interval
+        @param targetInterval: the target interval of the sub-mapping
+        @type    targetInterval: class L{Interval<Interval>}
+        """
+        self.targetInterval.copy(interval)
+
+
+    def setQueryInterval(self, interval):
+        """
+        Set query interval
+        @param queryInterval: the query interval of the sub-mapping
+        @type    queryInterval: class L{Interval<Interval>}
+        """
+        self.queryInterval.copy(interval)
+
+
+    def setSize(self, size):
+        """
+        Set the size of the sub-mapping
+        Possibly also target and query interval sizes, as well as number of mismatches
+        @param size: the size of the sub-mapping
+        @type    size: int
+        """
+        self.size = size
+        if "identity" in self.getTagNames():
+            self.setTagValue("nbMismatches", self.size - round(self.size * self.getTagValue("identity") / 100.0))
+
+
+    def getDirection(self):
+        """
+        Get the direction of the alignment
+        """
+        return self.targetInterval.getDirection()
+
+
+    def setDirection(self, direction):
+        """
+        Set the direction of the alignment
+        @param direction: the directio of the alignment
+        type   direction: int or string
+        """
+        return self.targetInterval.setDirection(direction)
+
+
+    def setTagValue(self, name, value):
+        """
+        Set the value of a tag
+        @param name:    name of the tag
+        @type    name:    string
+        @param value: value of the tag
+        @type    value: string or int
+        """
+        self.tags[name] = value
+
+
+    def getTagValue(self, name):
+        """
+        Get the value of a tag
+        @param name:    name of the tag
+        @type    name:    string
+        @return:            value of the tag
+        """
+        return self.tags[name]
+
+    
+    def getTagNames(self):
+        """
+        Get all the names of the tags
+        @return: the names of the tags
+        """
+        return self.tags.keys()
+
+    def getTargetInterval(self):
+        return self.targetInterval
+    
+    def getQueryInterval(self):
+        return self.queryInterval
+    
+    def getSize(self):
+        return self.size
+    
+    def getTags(self):
+        return self.tags
+
+    def setIdentity(self, identity):
+        """
+        Set the percentage of identity of the sub-mapping
+        Possibly also set number of mismatches
+        @param identity: the percentage of identity of the sub-mapping
+        @type  identity: float
+        """
+        self.identity = identity
+        self.setTagValue("identity", identity)
+        if self.size != None and "nbMismatches" not in self.getTagNames():
+            self.setTagValue("nbMismatches", self.size - round(self.size * self.getTagValue("identity") / 100.0))
+
+
+    def setNbMismatches(self, nbMismatches):
+        """
+        Set the number of mismatches of the sub-mapping
+        Possibly also set percentage of identity
+        @param nbMismatches: the number of mismatches of the sub-mapping
+        @type    nbMismatches: int
+        """
+        self.nbMismatches = nbMismatches
+        if self.size != None and "identity" not in self.getTagNames():
+            self.setTagValue("identity", (self.size - self.getTagValue("nbMismatches")) / float(self.size) * 100)
+
+
+    def setNbGaps(self, nbGaps):
+        """
+        Set the number of gaps of the sub-mapping
+        @param nbGaps: the number of gaps of the sub-mapping
+        @type    nbGaps: int
+        """
+        self.setTagValue("nbGaps", nbGaps)
+        
+        
+    def merge(self, subMapping):
+        """
+        Merge two subMappings
+        @param subMapping: another sub-mapping
+        @type    subMapping: class L{SubMapping<SubMapping>}
+        """
+        self.targetInterval.merge(subMapping.targetInterval)
+        self.queryInterval.merge(subMapping.queryInterval)
+
+
+    def printCoordinates(self):
+        """
+        Print the coordinates of the sub-mapping (considering the direction)
+        @return: a string
+        """
+        if self.getDirection() == 1:
+            return "%d-%d" % (self.targetInterval.getStart(), self.targetInterval.getEnd())
+        else:
+            return "%d-%d" % (self.targetInterval.getEnd(), self.targetInterval.getStart())
+
+
+    def __str__(self):
+        """
+        Return a representation of this object
+        @return: a string
+        """
+
+        if "match" in self.getTagNames() and not self.getTagValue("match"):
+            return "%s ---" % self.queryName
+
+        direction = "+"
+        if self.getDirection() == -1:
+            direction = "-"
+        string = "%s:%d-%d -- %s:%d-%d    (%s)" % (self.targetInterval.getChromosome(), self.targetInterval.getStart(), self.targetInterval.getEnd(), self.queryInterval.name, self.queryInterval.getStart(), self.queryInterval.getEnd(), direction)
+        if "nbMismatches" in self.getTagNames():
+            string += "(%i mm)" % (self.getTagValue("nbMismatches"))
+        if "identity" in self.getTagNames():
+            string += "(id: %i%%)" % (self.getTagValue("identity"))
+        if self.targetInterval.getSize() != None and self.queryInterval.getSize() != None and self.size != None:
+            string += "(sizes: %d, %d -> %d)" % (self.targetInterval.getSize(), self.queryInterval.getSize(), self.size)
+        return string
+
+