annotate hub-archive-creator/BedSimpleRepeats.py @ 6:9193fe3ee73f draft default tip

Uploaded
author yating-l
date Thu, 22 Dec 2016 15:59:24 -0500
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
6
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
1 #!/usr/bin/python
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
2
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
3 import os
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
4 import tempfile
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
5
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
6 from Datatype import Datatype
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
7 from Track import Track
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
8 from TrackDb import TrackDb
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
9 from util import subtools
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
10
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
11
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
12 class BedSimpleRepeats( Datatype ):
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
13 def __init__(self, input_bed_simple_repeats_false_path, data_bed_simple_repeats):
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
14
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
15 super(BedSimpleRepeats, self).__init__()
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
16
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
17 self.input_bed_simple_repeats_false_path = input_bed_simple_repeats_false_path
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
18 self.name_bed_simple_repeats = data_bed_simple_repeats["name"]
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
19 self.priority = data_bed_simple_repeats["order_index"]
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
20 self.track_color = data_bed_simple_repeats["track_color"]
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
21 # TODO: Think about how to avoid repetition of the group_name everywhere
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
22 self.group_name = data_bed_simple_repeats["group_name"]
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
23
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
24 sortedBedFile = tempfile.NamedTemporaryFile(suffix=".sortedBed")
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
25
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
26 # Sort processing
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
27 subtools.sort(self.input_bed_simple_repeats_false_path, sortedBedFile.name)
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
28
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
29 # bedToBigBed processing
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
30 # TODO: Change the name of the bb, to tool + genome + .bb
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
31 trackName = "".join( ( self.name_bed_simple_repeats, '.bb' ) )
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
32 myBigBedFilePath = os.path.join(self.myTrackFolderPath, trackName)
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
33
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
34 auto_sql_option = os.path.join(self.tool_directory, 'trf_simpleRepeat.as')
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
35
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
36 with open(myBigBedFilePath, 'w') as bigBedFile:
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
37 subtools.bedToBigBed(sortedBedFile.name,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
38 self.chromSizesFile.name,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
39 bigBedFile.name,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
40 typeOption='bed4+12',
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
41 autoSql=auto_sql_option)
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
42
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
43 # Create the Track Object
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
44 self.createTrack(file_path=trackName,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
45 track_name=trackName,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
46 long_label=self.name_bed_simple_repeats, track_type='bigBed 4 +', visibility='dense',
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
47 priority=self.priority,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
48 track_file=myBigBedFilePath,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
49 track_color=self.track_color,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
50 group_name=self.group_name)
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
51
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
52 # dataURL = "tracks/%s" % trackName
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
53 #
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
54 # trackDb = TrackDb(
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
55 # trackName=trackName,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
56 # longLabel=self.name_bed_simple_repeats,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
57 # shortLabel=self.getShortName( self.name_bed_simple_repeats ),
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
58 # trackDataURL=dataURL,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
59 # trackType='bigBed 4 +',
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
60 # visibility='dense',
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
61 # priority=self.priority,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
62 # )
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
63 #
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
64 # self.track = Track(
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
65 # trackFile=myBigBedFilePath,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
66 # trackDb=trackDb,
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
67 # )
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
68
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
69 print("- Bed simple repeats %s created" % self.name_bed_simple_repeats)
9193fe3ee73f Uploaded
yating-l
parents:
diff changeset
70 #print("- %s created in %s" % (trackName, myBigBedFilePath))