annotate TEisotools-1.0/commons/core/coord/PathUtils.py @ 6:20ec0d14798e draft

Uploaded
author urgi-team
date Wed, 20 Jul 2016 05:00:24 -0400
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
6
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
1 # Copyright INRA (Institut National de la Recherche Agronomique)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
2 # http://www.inra.fr
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
3 # http://urgi.versailles.inra.fr
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
4 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
5 # This software is governed by the CeCILL license under French law and
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
6 # abiding by the rules of distribution of free software. You can use,
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
7 # modify and/ or redistribute the software under the terms of the CeCILL
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
8 # license as circulated by CEA, CNRS and INRIA at the following URL
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
9 # "http://www.cecill.info".
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
10 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
11 # As a counterpart to the access to the source code and rights to copy,
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
12 # modify and redistribute granted by the license, users are provided only
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
13 # with a limited warranty and the software's author, the holder of the
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
14 # economic rights, and the successive licensors have only limited
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
15 # liability.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
16 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
17 # In this respect, the user's attention is drawn to the risks associated
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
18 # with loading, using, modifying and/or developing or reproducing the
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
19 # software by the user in light of its specific status of free software,
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
20 # that may mean that it is complicated to manipulate, and that also
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
21 # therefore means that it is reserved for developers and experienced
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
22 # professionals having in-depth computer knowledge. Users are therefore
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
23 # encouraged to load and test the software's suitability as regards their
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
24 # requirements in conditions enabling the security of their systems and/or
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
25 # data to be ensured and, more generally, to use and operate it in the
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
26 # same conditions as regards security.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
27 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
28 # The fact that you are presently reading this means that you have had
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
29 # knowledge of the CeCILL license and that you accept its terms.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
30
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
31
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
32 import os
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
33 import sys
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
34 import copy
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
35 from commons.core.coord.Map import Map
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
36 from commons.core.coord.Path import Path
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
37 from commons.core.coord.Align import Align
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
38 from commons.core.coord.Range import Range
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
39 from commons.core.coord.SetUtils import SetUtils
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
40 from commons.core.coord.AlignUtils import AlignUtils
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
41 from commons.core.checker.RepetException import RepetDataException
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
42
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
43 ## Static methods for the manipulation of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
44 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
45 class PathUtils ( object ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
46
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
47 ## Change the identifier of each Set instance in the given list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
48 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
49 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
50 # @param newId new identifier
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
51 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
52 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
53 def changeIdInList(lPaths, newId):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
54 for iPath in lPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
55 iPath.id = newId
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
56
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
57
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
58 ## Return a list of Set instances containing the query range from a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
59 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
60 # @param lPaths a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
61 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
62 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
63 def getSetListFromQueries(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
64 lSets = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
65 for iPath in lPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
66 lSets.append( iPath.getSubjectAsSetOfQuery() )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
67 return lSets
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
68
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
69
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
70 ## Return a list of Set instances containing the subject range from a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
71 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
72 # @param lPaths a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
73 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
74 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
75 def getSetListFromSubjects(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
76 lSets = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
77 for iPath in lPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
78 lSets.append( iPath.getQuerySetOfSubject() )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
79 return lSets
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
80
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
81
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
82 ## Return a sorted list of Range instances containing the subjects from a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
83 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
84 # @param lPaths a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
85 # @note meaningful only if all Path instances have same identifier
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
86 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
87 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
88 def getRangeListFromSubjects( lPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
89 lRanges = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
90 for iPath in lPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
91 lRanges.append( iPath.range_subject )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
92 if lRanges[0].isOnDirectStrand():
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
93 return sorted( lRanges, key=lambda iRange: ( iRange.getMin(), iRange.getMax() ) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
94 else:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
95 return sorted( lRanges, key=lambda iRange: ( iRange.getMax(), iRange.getMin() ) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
96
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
97
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
98 ## Return a tuple with min and max of query coordinates from Path instances in the given list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
99 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
100 # @param lPaths a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
101 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
102 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
103 def getQueryMinMaxFromPathList(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
104 qmin = -1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
105 qmax = -1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
106 for iPath in lPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
107 if qmin == -1:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
108 qmin = iPath.range_query.start
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
109 qmin = min(qmin, iPath.range_query.getMin())
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
110 qmax = max(qmax, iPath.range_query.getMax())
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
111 return (qmin, qmax)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
112
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
113
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
114 ## Return a tuple with min and max of subject coordinates from Path instances in the given list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
115 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
116 # @param lPaths lists of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
117 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
118 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
119 def getSubjectMinMaxFromPathList(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
120 smin = -1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
121 smax = -1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
122 for iPath in lPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
123 if smin == -1:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
124 smin = iPath.range_subject.start
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
125 smin = min(smin, iPath.range_subject.getMin())
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
126 smax = max(smax, iPath.range_subject.getMax())
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
127 return (smin, smax)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
128
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
129
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
130 ## Returns a Path objects list where Paths query coordinates overlapping with
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
131 # any Path in a list are removed.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
132 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
133 # WARNING: input Path lists are modified (sort)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
134 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
135 # @param lRefPaths list of paths to check overlaps
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
136 # @param lPathsToClean list of paths to remove overlapping Paths on query coordinates
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
137 # @return path list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
138 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
139 def removeOverlappingPathsOnQueriesBetweenPathLists(lRefPaths, lPathsToClean):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
140 if not lRefPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
141 print "WARNING: empty reference Paths list"
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
142 return lPathsToClean
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
143
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
144 lRefQueries = PathUtils.getListOfDistinctQueryNames(lRefPaths)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
145 lToCleanQueries = PathUtils.getListOfDistinctQueryNames(lPathsToClean)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
146
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
147 lCommonQueries = list(set(lRefQueries) & set(lToCleanQueries))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
148 lCommonQueries.sort()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
149 lSpecificToCleanQueries = list(set(lToCleanQueries) - set(lCommonQueries))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
150 lSpecificToCleanQueries.sort()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
151
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
152 lRefPaths.sort(key=lambda iPath: (iPath.getQueryName(), iPath.getIdentifier(), iPath.getQueryMin(), iPath.getQueryMax()))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
153 lPathsToClean.sort(key=lambda iPath: (iPath.getQueryName(), iPath.getIdentifier(), iPath.getQueryMin(), iPath.getQueryMax()))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
154
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
155 lCleanedPaths = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
156 lSpecificToCleanQueries = list(set(lToCleanQueries) - set(lCommonQueries))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
157 lCleanedPaths.extend(PathUtils.extractPathsFromQueryNameList(lPathsToClean, lSpecificToCleanQueries))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
158
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
159 dRefQueryToPathList = PathUtils.getDictOfListsWithQueryNameAsKey(lRefPaths)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
160 dToCleanQueryToPathList = PathUtils.getDictOfListsWithQueryNameAsKey(lPathsToClean)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
161
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
162 for queryName in lCommonQueries:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
163
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
164 refQueryHash = PathUtils.getDictOfListsWithIdAsKey(dRefQueryToPathList[queryName])
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
165 toCleanQueryHash = PathUtils.getDictOfListsWithIdAsKey(dToCleanQueryToPathList[queryName])
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
166
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
167 for lCleanPathById in toCleanQueryHash.values():
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
168 isOverlapping = False
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
169
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
170 for lRefPathById in refQueryHash.values():
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
171 if PathUtils.areQueriesOverlappingBetweenPathLists(lRefPathById, lCleanPathById, areListsAlreadySort = True):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
172 isOverlapping = True
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
173 break
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
174
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
175 if not isOverlapping:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
176 lCleanedPaths.extend(lCleanPathById)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
177
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
178 return lCleanedPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
179
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
180
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
181 ## Return True if the query range of any Path instance from the first list overlaps with the query range of any Path instance from the second list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
182 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
183 # @param lPaths1: list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
184 # @param lPaths2: list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
185 # @return boolean
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
186 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
187 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
188 def areQueriesOverlappingBetweenPathLists( lPaths1, lPaths2, areListsAlreadySort = False):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
189 if not areListsAlreadySort:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
190 lSortedPaths1 = PathUtils.getPathListSortedByIncreasingMinQueryThenMaxQuery( lPaths1 )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
191 lSortedPaths2 = PathUtils.getPathListSortedByIncreasingMinQueryThenMaxQuery( lPaths2 )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
192 else:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
193 lSortedPaths1 = lPaths1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
194 lSortedPaths2 = lPaths2
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
195 i = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
196 j = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
197 while i != len(lSortedPaths1):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
198 j = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
199 while j != len(lSortedPaths2):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
200 if not lSortedPaths1[i].range_query.isOverlapping( lSortedPaths2[j].range_query ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
201 j += 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
202 else:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
203 return True
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
204 i += 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
205 return False
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
206
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
207
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
208 ## Show Path instances contained in the given list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
209 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
210 # @param lPaths a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
211 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
212 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
213 def showList(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
214 for iPath in lPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
215 iPath.show()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
216
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
217
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
218 ## Write Path instances contained in the given list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
219 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
220 # @param lPaths a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
221 # @param fileName name of the file to write the Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
222 # @param mode the open mode of the file ""w"" or ""a""
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
223 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
224 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
225 def writeListInFile(lPaths, fileName, mode="w"):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
226 AlignUtils.writeListInFile(lPaths, fileName, mode)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
227
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
228
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
229 ## Return new list of Path instances with no duplicate
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
230 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
231 # @param lPaths a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
232 # @param useOnlyCoord boolean if True, check only coordinates and sequence names
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
233 # @return lUniqPaths a path instances list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
234 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
235 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
236 def getPathListWithoutDuplicates(lPaths, useOnlyCoord = False):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
237 if len(lPaths) < 2:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
238 return lPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
239 lSortedPaths = PathUtils.getPathListSortedByIncreasingMinQueryThenMaxQueryThenIdentifier( lPaths )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
240 lUniqPaths = [ lSortedPaths[0] ]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
241 if useOnlyCoord:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
242 for iPath in lSortedPaths[1:]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
243 if iPath.range_query.start != lUniqPaths[-1].range_query.start \
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
244 or iPath.range_query.end != lUniqPaths[-1].range_query.end \
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
245 or iPath.range_query.seqname != lUniqPaths[-1].range_query.seqname \
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
246 or iPath.range_subject.start != lUniqPaths[-1].range_subject.start \
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
247 or iPath.range_subject.end != lUniqPaths[-1].range_subject.end \
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
248 or iPath.range_subject.seqname != lUniqPaths[-1].range_subject.seqname:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
249 lUniqPaths.append( iPath )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
250 else:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
251 for iPath in lSortedPaths[1:]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
252 if iPath != lUniqPaths[-1]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
253 lUniqPaths.append( iPath )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
254 return lUniqPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
255
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
256
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
257 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
258 def getPathListWithoutDuplicatesOnQueryCoord(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
259 if len(lPaths) < 2:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
260 return lPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
261 lSortedPaths = PathUtils.getPathListSortedByIncreasingMinQueryThenMaxQueryThenIdentifier( lPaths )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
262 lUniqPaths = [ lSortedPaths[0] ]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
263 for iPath in lSortedPaths[1:]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
264 if iPath.range_query.start != lUniqPaths[-1].range_query.start \
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
265 or iPath.range_query.end != lUniqPaths[-1].range_query.end \
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
266 or iPath.range_query.seqname != lUniqPaths[-1].range_query.seqname:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
267 lUniqPaths.append( iPath )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
268 return lUniqPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
269
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
270
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
271 ## Split a Path list in several Path lists according to the identifier
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
272 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
273 # @param lPaths a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
274 # @return a dictionary which keys are identifiers and values Path lists
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
275 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
276 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
277 def getDictOfListsWithIdAsKey(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
278 dId2PathList = dict((ident, []) for ident in PathUtils.getListOfDistinctIdentifiers(lPaths))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
279 for iPath in lPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
280 dId2PathList[iPath.id].append(iPath)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
281 return dId2PathList
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
282
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
283
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
284 ## Split a Path list in several Path lists according to the query name
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
285 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
286 # @param lPaths a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
287 # @return a dictionary which keys are query_names and values Path lists
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
288 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
289 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
290 def getDictOfListsWithQueryNameAsKey(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
291 dId2PathList = dict((qn, []) for qn in PathUtils.getListOfDistinctQueryNames(lPaths))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
292 for iPath in lPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
293 dId2PathList[iPath.getQueryName()].append(iPath)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
294 return dId2PathList
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
295
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
296
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
297 ## Split a Path file in several Path lists according to the identifier
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
298 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
299 # @param pathFile name of the input Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
300 # @return a dictionary which keys are identifiers and values Path lists
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
301 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
302 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
303 def getDictOfListsWithIdAsKeyFromFile( pathFile ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
304 dId2PathList = {}
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
305 pathFileHandler = open(pathFile, "r")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
306 for line in pathFileHandler:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
307 iPath = Path()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
308 iPath.setFromString(line)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
309 if dId2PathList.has_key(iPath.id):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
310 dId2PathList[ iPath.id ].append(iPath)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
311 else:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
312 dId2PathList[ iPath.id ] = [ iPath ]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
313 pathFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
314 return dId2PathList
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
315
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
316
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
317 ## Return a list of Path list(s) obtained while splitting a list of connected Path instances according to another based on query coordinates
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
318 # Only the path instance of lToKeep between path instance of lToUnjoin are used to split lToUnjoin
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
319 # @param lToKeep: a list of Path instances to keep (reference)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
320 # @param lToUnjoin: a list of Path instances to unjoin
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
321 # @return: list of Path list(s) (can be empty if one of the input lists is empty)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
322 # @warning: all the path instances in a given list MUST be connected (i.e. same identifier)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
323 # @warning: if the path instances in a given list overlap neither within each other nor with the Path instances of the other list, these path instances are not used to split the lToUnjoin
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
324 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
325 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
326 def getPathListUnjoinedBasedOnQuery( lToKeep, lToUnjoin ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
327 lSortedToKeep = PathUtils.getPathListSortedByIncreasingMinQueryThenMaxQuery( lToKeep )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
328 length_lSortedToKeep = len(lSortedToKeep)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
329 # PathUtils.showList(lSortedToKeep)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
330 lSortedToUnjoin = PathUtils.getPathListSortedByIncreasingMinQueryThenMaxQuery( lToUnjoin )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
331 # PathUtils.showList(lSortedToUnjoin)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
332 length_lSortedToUnjoin = len(lSortedToUnjoin)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
333 if lToUnjoin == []:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
334 return []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
335 if lToKeep == []:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
336 return [ lToUnjoin ]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
337
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
338 lLists = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
339 k = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
340 while k < length_lSortedToKeep:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
341 j1 = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
342 while j1 < length_lSortedToUnjoin and lSortedToKeep[k].range_query.getMin() > lSortedToUnjoin[j1].range_query.getMax():
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
343 j1 += 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
344 if j1 == length_lSortedToUnjoin:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
345 break
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
346 if j1 != 0:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
347 lLists.append( lSortedToUnjoin[:j1] )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
348 del lSortedToUnjoin[:j1]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
349 j1 = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
350 if k+1 == len(lSortedToKeep):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
351 break
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
352 j2 = j1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
353 minQueryOf_lSortedToKeepKplus1 = lSortedToKeep[k+1].range_query.getMin()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
354 maxQueryOf_lSortedToUnjoinJ2 = lSortedToUnjoin[j2].range_query.getMax()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
355 if j2 < length_lSortedToUnjoin and minQueryOf_lSortedToKeepKplus1 > maxQueryOf_lSortedToUnjoinJ2:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
356 while j2 < len(lSortedToUnjoin) and minQueryOf_lSortedToKeepKplus1 > maxQueryOf_lSortedToUnjoinJ2:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
357 j2 += 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
358 maxQueryOf_lSortedToUnjoinJ2 = lSortedToUnjoin[j2].range_query.getMax()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
359 lLists.append( lSortedToUnjoin[j1:j2] )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
360 del lSortedToUnjoin[j1:j2]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
361 k += 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
362
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
363 if lLists != [] or k == 0:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
364 lLists.append( lSortedToUnjoin )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
365 else:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
366 lLists = lSortedToUnjoin
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
367
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
368 return lLists
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
369
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
370
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
371 ## Return the identity of the Path list, the identity of each instance being weighted by the length of each query range
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
372 # All Paths should have the same query and subject.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
373 # The Paths are merged using query coordinates only.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
374 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
375 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
376 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
377 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
378 def getIdentityFromPathList( lPaths, checkSubjects=True ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
379 if len( PathUtils.getListOfDistinctQueryNames( lPaths ) ) > 1:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
380 msg = "ERROR: try to compute identity from Paths with different queries"
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
381 sys.stderr.write( "%s\n" % msg )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
382 sys.stderr.flush()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
383 raise Exception
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
384 if checkSubjects and len( PathUtils.getListOfDistinctSubjectNames( lPaths ) ) > 1:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
385 msg = "ERROR: try to compute identity from Paths with different subjects"
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
386 sys.stderr.write( "%s\n" % msg )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
387 sys.stderr.flush()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
388 raise Exception
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
389 identity = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
390 lMergedPaths = PathUtils.mergePathsInListUsingQueryCoordsOnly( lPaths )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
391 lQuerySets = PathUtils.getSetListFromQueries( lMergedPaths )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
392 lMergedQuerySets = SetUtils.mergeSetsInList( lQuerySets )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
393 totalLengthOnQry = SetUtils.getCumulLength( lMergedQuerySets )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
394 for iPath in lMergedPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
395 identity += iPath.identity * iPath.getLengthOnQuery()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
396 weightedIdentity = identity / float(totalLengthOnQry)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
397 if weightedIdentity < 0:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
398 msg = "ERROR: weighted identity '%.2f' outside range" % weightedIdentity
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
399 sys.stderr.write("%s\n" % msg)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
400 sys.stderr.flush()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
401 raise Exception
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
402 elif weightedIdentity > 100:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
403 msg = "ERROR: weighted identity '%.2f' outside range" % weightedIdentity
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
404 sys.stderr.write("%s\n" % msg)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
405 sys.stderr.flush()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
406 raise RepetDataException(msg)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
407 return weightedIdentity
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
408
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
409
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
410 ## Return a list of Path instances sorted in increasing order according to the min of the query, then the max of the query, and finally their initial order.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
411 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
412 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
413 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
414 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
415 def getPathListSortedByIncreasingMinQueryThenMaxQuery(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
416 return sorted( lPaths, key=lambda iPath: ( iPath.getQueryMin(), iPath.getQueryMax() ) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
417
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
418
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
419 ## Return a list of Path instances sorted in increasing order according to the min of the query, then the max of the query, then their identifier, and finally their initial order.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
420 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
421 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
422 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
423 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
424 def getPathListSortedByIncreasingMinQueryThenMaxQueryThenIdentifier(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
425 return sorted( lPaths, key=lambda iPath: ( iPath.getQueryMin(), iPath.getQueryMax(), iPath.getIdentifier() ) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
426
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
427
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
428 ## Return a list of Path instances sorted in increasing order according to the min of the query, then the max of the query, then the min of the subject, then the max of the subject and finally their initial order.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
429 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
430 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
431 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
432 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
433 def getPathListSortedByIncreasingMinQueryThenMaxQueryThenMinSubjectThenMaxSubject(lPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
434 return sorted(lPaths, key=lambda iPath: (iPath.getQueryMin(), iPath.getQueryMax(), iPath.getSubjectMin(), iPath.getSubjectMax()))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
435
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
436
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
437 ## Return a list of Path instances sorted in increasing order according to the min, then the inverse of the query length, and finally their initial order
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
438 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
439 # @param lPaths: list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
440 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
441 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
442 def getPathListSortedByIncreasingQueryMinThenInvQueryLength( lPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
443 return sorted( lPaths, key=lambda iPath: ( iPath.getQueryMin(), 1 / float(iPath.getLengthOnQuery()) ) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
444
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
445
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
446 ## Return a list of the distinct identifiers
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
447 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
448 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
449 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
450 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
451 def getListOfDistinctIdentifiers( lPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
452 sDistinctIdentifiers = set([iPath.id for iPath in lPaths])
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
453 return list(sDistinctIdentifiers)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
454
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
455
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
456 ## Return a list of the distinct query names present in the collection
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
457 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
458 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
459 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
460 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
461 def getListOfDistinctQueryNames( lPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
462 sDistinctQueryNames = set([iPath.range_query.seqname for iPath in lPaths])
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
463 return list(sDistinctQueryNames)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
464
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
465
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
466 ## Return a list of the distinct subject names present in the collection
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
467 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
468 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
469 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
470 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
471 def getListOfDistinctSubjectNames( lPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
472 sDistinctSubjectNames = set([iPath.range_subject.seqname for iPath in lPaths])
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
473 return list(sDistinctSubjectNames)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
474
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
475
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
476 ## Return a list of paths with matching query names
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
477 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
478 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
479 # @param queryName query name to extract
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
480 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
481 def extractPathsFromQueryName(lPaths, queryName):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
482 return [iPath for iPath in lPaths if iPath.getQueryName() == queryName]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
483
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
484
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
485 ## Return a list of paths with matching query names
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
486 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
487 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
488 # @param lQueryName query name list to extract
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
489 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
490 def extractPathsFromQueryNameList(lPaths, lQueryNames):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
491 d = dict.fromkeys(lQueryNames)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
492 return [iPath for iPath in lPaths if iPath.getQueryName() in d]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
493
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
494
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
495 ## Return a list of paths with matching subject names
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
496 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
497 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
498 # @param subjectName subject name to extract
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
499 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
500 def extractPathsFromSubjectName(lPaths, subjectName):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
501 return [iPath for iPath in lPaths if iPath.getSubjectName() == subjectName]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
502
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
503
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
504 ## Return a list of paths with coordinates overlap a given range
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
505 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
506 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
507 # @param queryName query name to extract
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
508 # @param start starting position
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
509 # @param end ending position
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
510 # @return list of Path instance
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
511 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
512 def extractPathsFromQueryCoord(lPaths, queryName, start, end):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
513 lExtractedPaths = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
514 iAlign = Align(range_q = Range(queryName, start, end))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
515
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
516 for path in PathUtils.extractPathsFromQueryName(lPaths, queryName):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
517 if path.isQueryOverlapping(iAlign):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
518 lExtractedPaths.append(path)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
519
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
520 return lExtractedPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
521
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
522
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
523 ## Return a list of lists containing query coordinates of the connections sorted in increasing order.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
524 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
525 # @param lConnectedPaths: list of Path instances having the same identifier
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
526 # @param minLength: threshold below which connections are not reported (default= 0 bp)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
527 # @note: return only connections longer than threshold
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
528 # @note: if coordinate on query ends at 100, return 101
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
529 # @warning: Path instances MUST be sorted in increasing order according to query coordinates
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
530 # @warning: Path instances MUST be on direct query strand (and maybe on reverse subject strand)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
531 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
532 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
533 def getListOfJoinCoordinatesOnQuery(lConnectedPaths, minLength=0):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
534 lJoinCoordinates = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
535 for i in xrange(1,len(lConnectedPaths)):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
536 startJoin = lConnectedPaths[i-1].range_query.end
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
537 endJoin = lConnectedPaths[i].range_query.start
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
538 if endJoin - startJoin + 1 > minLength:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
539 lJoinCoordinates.append( [ startJoin + 1, endJoin - 1 ] )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
540 return lJoinCoordinates
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
541
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
542
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
543 ## Return the length on the query of all Path instance in the given list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
544 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
545 # @param lPaths list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
546 # @note overlapping ranges are not summed but truncated.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
547 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
548 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
549 def getLengthOnQueryFromPathList( lPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
550 lSets = PathUtils.getSetListFromQueries( lPaths )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
551 lMergedSets = SetUtils.mergeSetsInList( lSets )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
552 length = SetUtils.getCumulLength( lMergedSets )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
553 return length
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
554
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
555
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
556 ## Convert a Path file into an Align file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
557 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
558 # @param pathFile: name of the input Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
559 # @param alignFile: name of the output Align file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
560 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
561 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
562 def convertPathFileIntoAlignFile(pathFile, alignFile):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
563 pathFileHandler = open(pathFile, "r")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
564 alignFileHandler = open(alignFile, "w")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
565 iPath = Path()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
566 for line in pathFileHandler:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
567 iPath.setFromString(line)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
568 iAlign = iPath.getAlignInstance()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
569 iAlign.write(alignFileHandler)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
570 pathFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
571 alignFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
572
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
573
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
574 #TODO: duplicated method => to rename with the name of the next method (which is called) ?
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
575 ## Convert a Path File into a Map file with query coordinates only
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
576 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
577 # @param pathFile: name of the input Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
578 # @param mapFile: name of the output Map file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
579 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
580 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
581 def convertPathFileIntoMapFileWithQueryCoordsOnly( pathFile, mapFile ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
582 pathFileHandler = open(pathFile, "r")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
583 mapFileHandler = open(mapFile, "w")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
584 p = Path()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
585 for line in pathFileHandler:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
586 p.reset()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
587 p.setFromTuple(line.split("\t"))
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
588 p.writeSubjectAsMapOfQuery(mapFileHandler)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
589 pathFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
590 mapFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
591
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
592
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
593 ## for each line of a given Path file, write the coordinates of the subject on the query as one line in a Map file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
594 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
595 # @param pathFile: name of the input Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
596 # @param mapFile: name of the output Map file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
597 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
598 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
599 def convertPathFileIntoMapFileWithSubjectsOnQueries( pathFile, mapFile ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
600 PathUtils.convertPathFileIntoMapFileWithQueryCoordsOnly( pathFile, mapFile )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
601
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
602
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
603 ## Merge matches on queries
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
604 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
605 # @param inFile: name of the input Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
606 # @param outFile: name of the output Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
607 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
608 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
609 def mergeMatchesOnQueries(inFile, outFile):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
610 mapFile = "%s.map" % inFile
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
611 PathUtils.convertPathFileIntoMapFileWithQueryCoordsOnly(inFile, mapFile)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
612 cmd = "mapOp"
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
613 cmd += " -q %s" % mapFile
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
614 cmd += " -m"
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
615 cmd += " 2>&1 > /dev/null"
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
616 exitStatus = os.system(cmd)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
617 if exitStatus != 0:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
618 print "ERROR: mapOp returned %i" % exitStatus
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
619 sys.exit(1)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
620 os.remove(mapFile)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
621 mergeFile = "%s.merge" % mapFile
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
622 mergeFileHandler = open(mergeFile, "r")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
623 outFileHandler = open(outFile, "w")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
624 m = Map()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
625 for line in mergeFileHandler:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
626 m.reset()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
627 m.setFromString(line, "\t")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
628 m.writeAsQueryOfPath(outFileHandler)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
629 mergeFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
630 os.remove(mergeFile)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
631 outFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
632
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
633
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
634 ## Filter chains of Path(s) which length is below a given threshold
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
635 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
636 # @param lPaths: list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
637 # @param minLengthChain: minimum length of a chain to be kept
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
638 # @note: a chain may contain a single Path instance
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
639 # @return: a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
640 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
641 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
642 def filterPathListOnChainLength( lPaths, minLengthChain ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
643 lFilteredPaths = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
644 dPathnum2Paths = PathUtils.getDictOfListsWithIdAsKey( lPaths )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
645 for pathnum in dPathnum2Paths.keys():
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
646 length = PathUtils.getLengthOnQueryFromPathList( dPathnum2Paths[ pathnum ] )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
647 if length >= minLengthChain:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
648 lFilteredPaths += dPathnum2Paths[ pathnum ]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
649 return lFilteredPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
650
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
651
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
652 ## Return a Path list from a Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
653 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
654 # @param pathFile string name of a Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
655 # @return a list of Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
656 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
657 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
658 def getPathListFromFile( pathFile ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
659 lPaths = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
660 with open(pathFile, "r") as pathFileHandler:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
661 for line in pathFileHandler:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
662 iPath = Path()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
663 iPath.setFromString(line)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
664 lPaths.append(iPath)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
665 return lPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
666
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
667
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
668 ## Convert a chain into a 'pathrange'
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
669 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
670 # @param lPaths a list of Path instances with the same identifier
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
671 # @note: the min and max of each Path is used
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
672 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
673 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
674 def convertPathListToPathrange( lPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
675 if len(lPaths) == 0:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
676 return
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
677 if len(lPaths) == 1:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
678 return lPaths[0]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
679 iPathrange = copy.deepcopy( lPaths[0] )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
680 iPathrange.identity = lPaths[0].identity * lPaths[0].getLengthOnQuery()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
681 cumulQueryLength = iPathrange.getLengthOnQuery()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
682 for iPath in lPaths[1:]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
683 if iPath.id != iPathrange.id:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
684 msg = "ERROR: two Path instances in the chain have different identifiers"
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
685 sys.stderr.write( "%s\n" % ( msg ) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
686 sys.exit(1)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
687 if iPathrange.range_subject.isOnDirectStrand() != iPath.range_subject.isOnDirectStrand():
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
688 msg = "ERROR: two Path instances in the chain are on different strands"
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
689 sys.stderr.write( "%s\n" % ( msg ) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
690 sys.exit(1)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
691 iPathrange.range_query.start = min( iPathrange.range_query.start, iPath.range_query.start )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
692 iPathrange.range_query.end = max( iPathrange.range_query.end, iPath.range_query.end )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
693 if iPathrange.range_subject.isOnDirectStrand():
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
694 iPathrange.range_subject.start = min( iPathrange.range_subject.start, iPath.range_subject.start )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
695 iPathrange.range_subject.end = max( iPathrange.range_subject.end, iPath.range_subject.end )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
696 else:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
697 iPathrange.range_subject.start = max( iPathrange.range_subject.start, iPath.range_subject.start )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
698 iPathrange.range_subject.end = min( iPathrange.range_subject.end, iPath.range_subject.end )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
699 iPathrange.e_value = min( iPathrange.e_value, iPath.e_value )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
700 iPathrange.score += iPath.score
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
701 iPathrange.identity += iPath.identity * iPath.getLengthOnQuery()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
702 cumulQueryLength += iPath.getLengthOnQuery()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
703 iPathrange.identity = iPathrange.identity / float(cumulQueryLength)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
704 return iPathrange
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
705
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
706
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
707 ## Convert a Path file into an Align file via 'pathrange'
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
708 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
709 # @param pathFile: name of the input Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
710 # @param alignFile: name of the output Align file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
711 # @param verbose integer verbosity level
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
712 # @note: the min and max of each Path is used
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
713 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
714 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
715 def convertPathFileIntoAlignFileViaPathrange( pathFile, alignFile, verbose=0 ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
716 lPaths = PathUtils.getPathListFromFile( pathFile )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
717 dId2PathList = PathUtils.getDictOfListsWithIdAsKey( lPaths )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
718 lIds = dId2PathList.keys()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
719 lIds.sort()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
720 if verbose > 0:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
721 msg = "number of chains: %i" % ( len(lIds) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
722 sys.stdout.write( "%s\n" % ( msg ) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
723 sys.stdout.flush()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
724 alignFileHandler = open( alignFile, "w" )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
725 for identifier in lIds:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
726 iPath = PathUtils.convertPathListToPathrange( dId2PathList[ identifier ] )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
727 iAlign = iPath.getAlignInstance()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
728 iAlign.write( alignFileHandler )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
729 alignFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
730
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
731
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
732 ## Split a list of Path instances according to the name of the query
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
733 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
734 # @param lInPaths list of align instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
735 # @return lOutPathLists list of align instances lists
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
736 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
737 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
738 def splitPathListByQueryName( lInPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
739 lInSortedPaths = sorted( lInPaths, key=lambda o: o.range_query.seqname )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
740 lOutPathLists = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
741 if len(lInSortedPaths) != 0 :
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
742 lPathsForCurrentQuery = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
743 previousQuery = lInSortedPaths[0].range_query.seqname
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
744 for iPath in lInSortedPaths :
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
745 currentQuery = iPath.range_query.seqname
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
746 if previousQuery != currentQuery :
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
747 lOutPathLists.append( lPathsForCurrentQuery )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
748 previousQuery = currentQuery
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
749 lPathsForCurrentQuery = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
750 lPathsForCurrentQuery.append( iPath )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
751
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
752 lOutPathLists.append(lPathsForCurrentQuery)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
753
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
754 return lOutPathLists
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
755
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
756
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
757 ## Create an Path file from each list of Path instances in the input list
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
758 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
759 # @param lPathList list of lists with Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
760 # @param pattern string
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
761 # @param dirName string
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
762 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
763 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
764 def createPathFiles( lPathList, pattern, dirName="" ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
765 nbFiles = len(lPathList)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
766 countFile = 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
767 if dirName != "" :
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
768 if dirName[-1] != "/":
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
769 dirName = dirName + '/'
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
770 os.mkdir( dirName )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
771
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
772 for lPath in lPathList:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
773 fileName = dirName + pattern + "_%s.path" % ( str(countFile).zfill( len(str(nbFiles)) ) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
774 PathUtils.writeListInFile( lPath, fileName )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
775 countFile += 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
776
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
777
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
778 ## Merge all overlapping Path instances in a list without considering the identifiers
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
779 # Start by sorting the Path instances by their increasing min coordinate
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
780 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
781 # @return: a new list with the merged Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
782 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
783 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
784 def mergePathsInList( lPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
785 lMergedPaths = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
786 if len(lPaths)==0:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
787 return lMergedPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
788
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
789 lSortedPaths = PathUtils.getPathListSortedByIncreasingQueryMinThenInvQueryLength( lPaths )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
790
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
791 prev_count = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
792 for iPath in lSortedPaths[0:]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
793 if prev_count != len(lSortedPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
794 for i in lSortedPaths[ prev_count + 1: ]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
795 if iPath.isOverlapping( i ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
796 iPath.merge( i )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
797 isAlreadyInList = False
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
798 for newPath in lMergedPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
799 if newPath.isOverlapping( iPath ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
800 isAlreadyInList = True
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
801 newPath.merge( iPath )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
802 lMergedPaths [ lMergedPaths.index( newPath ) ] = newPath
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
803 if not isAlreadyInList:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
804 lMergedPaths.append( iPath )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
805 prev_count += 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
806 return lMergedPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
807
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
808
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
809 ## Merge all overlapping Path instances in a list without considering if subjects are overlapping.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
810 # Start by sorting the Path instances by their increasing min coordinate.
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
811 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
812 # @return: a new list with the merged Path instances
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
813 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
814 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
815 def mergePathsInListUsingQueryCoordsOnly( lPaths ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
816 lMergedPaths = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
817 if len(lPaths)==0:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
818 return lMergedPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
819
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
820 lSortedPaths = PathUtils.getPathListSortedByIncreasingQueryMinThenInvQueryLength( lPaths )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
821
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
822 prev_count = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
823 for iPath in lSortedPaths[0:]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
824 if prev_count != len(lSortedPaths):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
825 for i in lSortedPaths[ prev_count + 1: ]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
826 if iPath.isQueryOverlapping( i ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
827 iPath.merge( i )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
828 isAlreadyInList = False
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
829 for newPath in lMergedPaths:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
830 if newPath.isQueryOverlapping( iPath ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
831 isAlreadyInList = True
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
832 newPath.merge( iPath )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
833 lMergedPaths [ lMergedPaths.index( newPath ) ] = newPath
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
834 if not isAlreadyInList:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
835 lMergedPaths.append( iPath )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
836 prev_count += 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
837 return lMergedPaths
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
838
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
839
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
840 ## Convert a Path file into a GFF file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
841 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
842 # @param pathFile: name of the input Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
843 # @param gffFile: name of the output GFF file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
844 # @param source: source to write in the GFF file (column 2)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
845 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
846 # @note the 'path' query is supposed to correspond to the 'gff' first column
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
847 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
848 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
849 def convertPathFileIntoGffFile( pathFile, gffFile, source="REPET", verbose=0 ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
850 dId2PathList = PathUtils.getDictOfListsWithIdAsKeyFromFile( pathFile )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
851 if verbose > 0:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
852 msg = "number of chains: %i" % ( len(dId2PathList.keys()) )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
853 sys.stdout.write( "%s\n" % msg )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
854 sys.stdout.flush()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
855 gffFileHandler = open( gffFile, "w" )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
856 for id in dId2PathList.keys():
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
857 if len( dId2PathList[ id ] ) == 1:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
858 iPath = dId2PathList[ id ][0]
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
859 string = iPath.toStringAsGff( ID="%i" % iPath.getIdentifier(),
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
860 source=source )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
861 gffFileHandler.write( "%s\n" % string )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
862 else:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
863 iPathrange = PathUtils.convertPathListToPathrange( dId2PathList[ id ] )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
864 string = iPathrange.toStringAsGff( ID="ms%i" % iPathrange.getIdentifier(),
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
865 source=source )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
866 gffFileHandler.write( "%s\n" % string )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
867 count = 0
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
868 for iPath in dId2PathList[ id ]:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
869 count += 1
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
870 string = iPath.toStringAsGff( type="match_part",
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
871 ID="mp%i-%i" % ( iPath.getIdentifier(), count ),
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
872 Parent="ms%i" % iPathrange.getIdentifier(),
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
873 source=source )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
874 gffFileHandler.write( "%s\n" % string )
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
875 gffFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
876
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
877
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
878 ## Convert a Path file into a Set file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
879 # replace old parser.pathrange2set
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
880 # @param pathFile: name of the input Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
881 # @param setFile: name of the output Set file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
882 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
883 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
884 def convertPathFileIntoSetFile( pathFile, setFile ):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
885 pathFileHandler = open(pathFile, "r")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
886 setFileHandler = open(setFile, "w")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
887 iPath = Path()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
888 for line in pathFileHandler:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
889 iPath.setFromString(line)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
890 iSet = iPath.getSubjectAsSetOfQuery()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
891 iSet.write(setFileHandler)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
892 pathFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
893 setFileHandler.close()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
894
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
895 ## Write Path File without duplicated Path (same query, same subject and same coordinate)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
896 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
897 # @param inputFile: name of the input Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
898 # @param outputFile: name of the output Path file
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
899 #
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
900 @staticmethod
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
901 def removeInPathFileDuplicatedPathOnQueryNameQueryCoordAndSubjectName(inputFile, outputFile):
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
902 f = open(inputFile, "r")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
903 line = f.readline()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
904 previousQuery = ""
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
905 previousSubject = ""
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
906 lPaths = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
907 while line:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
908 iPath = Path()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
909 iPath.setFromString(line)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
910 query = iPath.getQueryName()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
911 subject = iPath.getSubjectName()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
912 if (query != previousQuery or subject != previousSubject) and lPaths != []:
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
913 lPathsWithoutDuplicate = PathUtils.getPathListWithoutDuplicatesOnQueryCoord(lPaths)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
914 PathUtils.writeListInFile(lPathsWithoutDuplicate, outputFile, "a")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
915 lPaths = []
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
916 lPaths.append(iPath)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
917 previousQuery = query
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
918 previousSubject = subject
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
919 line = f.readline()
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
920 lPathsWithoutDuplicate = PathUtils.getPathListWithoutDuplicatesOnQueryCoord(lPaths)
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
921 PathUtils.writeListInFile(lPathsWithoutDuplicate, outputFile, "a")
20ec0d14798e Uploaded
urgi-team
parents:
diff changeset
922 f.close()