Mercurial > repos > fcaramia > contra
diff Contra/scripts/split_chromosome.py @ 0:7564f3b1e675
Uploaded
author | fcaramia |
---|---|
date | Thu, 13 Sep 2012 02:31:43 -0400 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Contra/scripts/split_chromosome.py Thu Sep 13 02:31:43 2012 -0400 @@ -0,0 +1,105 @@ +# ----------------------------------------------------------------------# +# Copyright (c) 2011, Richard Lupat & Jason Li. +# +# > Source License < +# This file is part of CONTRA. +# +# CONTRA is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# CONTRA is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with CONTRA. If not, see <http://www.gnu.org/licenses/>. +# +# +#-----------------------------------------------------------------------# +# Last Updated : 03 Sep 2011 15:00PM + + +import os + +def splitByChromosome(destFolder): + + try: + os.mkdir(destFolder + "chr/") + except: + print "folder exist" + + inputfile = destFolder + "sample.BEDGRAPH" + outputfile = destFolder + "chr/chr1.txt" + file = open(inputfile,"r") + output = open(outputfile,"w") + check = "1" + + for row in file: + if row[0] == '#': + continue + + cols = row.split() + chr = cols[0].strip("chr") + if (chr != check): + output.close() + check = chr + output = open(destFolder+ "chr/chr"+check+".txt","w") + output.write(row) + + output.close() + +#JLMod +def splitByChromosome3(infile): + destFolder = os.path.dirname(infile)+"/" + + try: + os.mkdir(destFolder + "chr/") + except: + print "folder exist" + + #inputfile = destFolder + "sample.BEDGRAPH" + inputfile=infile + outputfile = destFolder + "chr/chr1.txt" + file = open(inputfile,"r") + output = open(outputfile,"w") + check = "1" + + for row in file: + cols = row.split() + chr = cols[0].strip("chr") + if (chr != check): + output.close() + check = chr + output = open(destFolder+ "chr/chr"+check+".txt","w") + output.write(row) + + output.close() + +def splitByChromosome2(folder): + + try: + os.mkdir(folder + "target/") + except: + print "folder exist" + + inputfile = folder + "geneRefCoverage.txt" + outputfile = folder + "target/chr1.txt" + file = open(inputfile,"r") + output = open(outputfile,"w") + check = "1" + + for row in file: + cols = row.split() + chr = cols[0].strip("chr") + if (chr != check): + output.close() + check = chr + output = open(folder+ "target/chr"+check+".txt","w") + output.write(row) + + output.close() + +