Changeset 27 for biocomp.pscoils/branches/biopython-enabled/src/biocomp/pscoils/__init__.py

Timestamp:

06/10/08 23:46:02 (4 years ago)

Author:

mauro

Message:

 

Files:

• biocomp.pscoils/branches/biopython-enabled/src/biocomp/pscoils/__init__.py (modified) (5 diffs)

biocomp.pscoils/branches/biopython-enabled/src/biocomp/pscoils/__init__.py

@@ -35 +35 @@
 import math
 import string
+
 import Params # this must be in the sys.path. Params contains the COILS paramters
+import utils
+
+from Bio.SeqFeature import FeatureLocation, SeqFeature
 
 #
@@ -69 +73 @@
     return prof,pos
     
-#---------------------------------#
-
-
-#----------- readFasta files 
-def readFasta(fname):
-    ''' readFasta(fname) returns a dictionary 
-    '''
-    try:
-        lines=open(fname,'r').readlines()
-    except:
-        sys.stderr.write('cannot open file '+fname+'\n')
-        sys.exit()
-    seqs={}
-    name=''
-    seq=''
-    for line in lines:
-        if line[0]=='>': # assuming fasta file
-           if name != '':
-               seqs[name]=seq
-               seq=''
-           name=line[1:].strip()
-        else:
-           seq+=''.join(line.split())
-    if name != '':
-        seqs[name]=seq
-    return seqs
 #---------------------------------#
 
@@ -134 +112 @@
 
     
-#---------------------------------#
-def pred_coil(seq,seqLen,params,fScore):
+def pred_coil(seqr, params, fScore=None):
     ''' pred_coil(seq,seqLen,params,fScore) returns the coiled coil prediction of sequence seq'''
+    if fScore == None:
+        fScore = seqScore
+    seq = seqr.seq
+    seqLen = len(seqr.seq)
     hept_pos=['a','b','c','d','e','f','g']
     score=[0.0]*seqLen
-    prob=[0.0]*seqLen
-    gcc=[0.0]*seqLen
-    gg=[0.0]*seqLen
     hept_seq=['x']*seqLen
     for i in range(seqLen-params.win+1):
@@ -160 +138 @@
                 hept_seq[i+j]=hept_pos[pos]
     for i in range(seqLen):
-        gg[i],gcc[i],prob[i]= coilProb(score[i],params)
-    return gg,gcc,prob,hept_seq,score
-  
+        gg, gcc, prob = coilProb(score[i],params)
+        seqf = SeqFeature(location=FeatureLocation(i,i), type="pscoils")
+        seqf.qualifiers = {'gg':gg, 
+            'gcc': gcc, 
+            'prob': prob, 
+            'score': score[i],
+            'hept_seq': hept_seq[i]}
+        seqr.features.append(seqf)
+    return seqr  
     
 #---------------------------------#
@@ -195 +179 @@
     return gg,gcc,prob,hept_seq,score
   
-def printCoil(seq,seqLen,hept_seq,score,P,Gcc,Gg,labels='T',io=sys.stdout):
-    ''' '''
-    ccLab="C"
-    loopLab="L"
-    if labels !='T' :
-        for i in range(seqLen):
-            #print "%4d %c %c %7.3f %7.3f (%7.3f %7.3f)" % (i+1,seq[i],hept_seq[i],score[i],P[i],Gcc[i],Gg[i])
-            io.write("%4d %c %c %7.3f %7.3f %7.3f %7.3f\n" % (i+1,seq[i],hept_seq[i],score[i],P[i],Gcc[i],Gg[i]))
-    else:
-        io.write(" Pos A Hep Score   Prob    Gcc     Gg    Pred (Loop=L Coiledcoil=C)\n")
-        for i in range(seqLen):
-            if P[i] > 0.5:
-               clabel=ccLab
-            else:
-               clabel=loopLab
-            io.write("%4d %c %c %7.3f %7.3f %7.3f %7.3f %s\n" % (i+1,seq[i],hept_seq[i],score[i],P[i],Gcc[i],Gg[i],clabel))