Mercurial > repos > rnateam > graphclust_preprocessing
diff splitSHAPE.py @ 7:07ad2d77f28a draft
planemo upload for repository https://github.com/eteriSokhoyan/galaxytools/tree/branchForIterations/tools/GraphClust commit 4379e712f76f2bb12ee2cc270dd8a0e806df2cd6
author | rnateam |
---|---|
date | Mon, 22 May 2017 12:45:22 -0400 |
parents | |
children | a04e93fdb40a |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/splitSHAPE.py Mon May 22 12:45:22 2017 -0400 @@ -0,0 +1,64 @@ +import os +import re +import sys + +shape_file = sys.argv[1] +win_size = int(sys.argv[2]) + +pattern = re.compile("^>.*$") +toWrite = "" + +count_for_id = 1 +seq_counter = 0 +new_id = "" + +seq_id = [] +seq_string = [] +orig_id = [] +name_file = "FASTA/data.names" +array_all_chunks = [] +with open(name_file, 'r') as f: + content = f.read() + lines = content.split('\n')[:-1] + for line in lines: + seq_id.append(int(line.split()[0])) + seq_string.append(line.split()[1]) + orig_id_srt = line.split()[3] + orig_id_srt = orig_id_srt.rsplit('_',1)[0] + orig_id.append(orig_id_srt) + + +react_dict = {} +react_arr = [] + +with open(shape_file, 'r') as shape: + content = shape.read() + lines = content.split('\n') + for line in lines: + if pattern.match(line): + line = line.replace('>','').strip() + react_arr=[] + react_dict[line] = react_arr + continue + else: + react_arr.append(line) + +toWrite = "" +chunks = [] +for i in range(len(orig_id)): + if not orig_id[i] in react_dict: + raise RuntimeError('Error key {} not found'.format(orig_id)) + + react_val = react_dict[orig_id[i]] + toWrite += '>' + str(seq_id[i]) + " " + seq_string[i] + "\n" + chunks = re.findall(r'\d+', seq_string[i]) + + for j in react_val[int(chunks[1])-1:int(chunks[2])]: + id_s = int(j.split()[0]) + + if id_s > win_size: + id_s = id_s - int(chunks[1]) + toWrite += str(id_s) + '\t' + j.split()[1] + "\n" + +with open("shape_data_split.react", 'w') as out: + out.write(toWrite)