| 6 | 1 # | 
|  | 2 # Copyright INRA-URGI 2009-2010 | 
|  | 3 # | 
|  | 4 # This software is governed by the CeCILL license under French law and | 
|  | 5 # abiding by the rules of distribution of free software. You can use, | 
|  | 6 # modify and/ or redistribute the software under the terms of the CeCILL | 
|  | 7 # license as circulated by CEA, CNRS and INRIA at the following URL | 
|  | 8 # "http://www.cecill.info". | 
|  | 9 # | 
|  | 10 # As a counterpart to the access to the source code and rights to copy, | 
|  | 11 # modify and redistribute granted by the license, users are provided only | 
|  | 12 # with a limited warranty and the software's author, the holder of the | 
|  | 13 # economic rights, and the successive licensors have only limited | 
|  | 14 # liability. | 
|  | 15 # | 
|  | 16 # In this respect, the user's attention is drawn to the risks associated | 
|  | 17 # with loading, using, modifying and/or developing or reproducing the | 
|  | 18 # software by the user in light of its specific status of free software, | 
|  | 19 # that may mean that it is complicated to manipulate, and that also | 
|  | 20 # therefore means that it is reserved for developers and experienced | 
|  | 21 # professionals having in-depth computer knowledge. Users are therefore | 
|  | 22 # encouraged to load and test the software's suitability as regards their | 
|  | 23 # requirements in conditions enabling the security of their systems and/or | 
|  | 24 # data to be ensured and, more generally, to use and operate it in the | 
|  | 25 # same conditions as regards security. | 
|  | 26 # | 
|  | 27 # The fact that you are presently reading this means that you have had | 
|  | 28 # knowledge of the CeCILL license and that you accept its terms. | 
|  | 29 # | 
|  | 30 from commons.core.writer.TranscriptListWriter import TranscriptListWriter | 
|  | 31 | 
|  | 32 | 
|  | 33 class MapWriter(TranscriptListWriter): | 
|  | 34     """ | 
|  | 35     A class that writes a transcript list into a file with GFF3 format | 
|  | 36     @ivar fileName: name of the file | 
|  | 37     @type fileName: string | 
|  | 38     @ivar handle: handle to the file | 
|  | 39     @type handle: file handle | 
|  | 40     """ | 
|  | 41 | 
|  | 42 | 
|  | 43     def __init__(self, fileName, verbosity = 0, title="S-MART"): | 
|  | 44         """ | 
|  | 45         Constructor | 
|  | 46         @param fileName: name of the file | 
|  | 47         @type fileName: string | 
|  | 48         @param verbosity: verbosity | 
|  | 49         @type verbosity: int | 
|  | 50         """ | 
|  | 51         self.header = "" | 
|  | 52         self.title    = title | 
|  | 53         TranscriptListWriter.__init__(self, fileName, verbosity) | 
|  | 54 | 
|  | 55 | 
|  | 56     @staticmethod | 
|  | 57     def getFileFormats(): | 
|  | 58         """ | 
|  | 59         Get the format of the file | 
|  | 60         """ | 
|  | 61         return ["map"] | 
|  | 62 | 
|  | 63 | 
|  | 64     @staticmethod | 
|  | 65     def getExtension(): | 
|  | 66         """ | 
|  | 67         Get the usual extension for the file | 
|  | 68         """ | 
|  | 69         return "map" | 
|  | 70 | 
|  | 71 | 
|  | 72     def setTitle(self, title): | 
|  | 73         """ | 
|  | 74         Set the title of the transcripts | 
|  | 75         @param title: the title of the transcripts | 
|  | 76         @type    title: string | 
|  | 77         """ | 
|  | 78         self.title = title | 
|  | 79 | 
|  | 80 | 
|  | 81     def printTranscript(self, transcript): | 
|  | 82         """ | 
|  | 83         Export the given transcript to map format | 
|  | 84         @param transcript: transcript to be printed | 
|  | 85         @type transcript: class L{Transcript<Transcript>} | 
|  | 86         @return: a string | 
|  | 87         """ | 
|  | 88         name = transcript.name | 
|  | 89         if "nbOccurrences" in transcript.getTagNames() and transcript.getTagValue("nbOccurrences") != 1 and transcript.getTagValue("occurrences"): | 
|  | 90             name = "%s-%d" % (name, transcript.getTagValue("occurrence")) | 
|  | 91         sizes   = [] | 
|  | 92         starts  = [] | 
|  | 93         transcript.sortExonsIncreasing() | 
|  | 94         for exon in transcript.getExons(): | 
|  | 95             sizes.append("%d" % (exon.getSize())) | 
|  | 96             starts.append("%d" % (exon.getStart() - transcript.getStart())) | 
|  | 97         return "%s\t%s\t%d\t%d\n" % (name, transcript.getChromosome(), transcript.getStart(), transcript.getEnd()+1) | 
|  | 98 | 
|  | 99 | 
|  | 100 |