X-Git-Url: http://mmka.chem.univ.gda.pl/gitweb/?a=blobdiff_plain;f=files%2Ffluct_plot.py;h=7719b2f3dc4b3b2438377a475a0a491efa583dc2;hb=32eab17b5d32751f900c80ffc18e6661bfe90572;hp=609ca4db6ec64fc88e6040a9a99577c8cef5b3dc;hpb=161801e60a4b6ac75ee8704dfb2d508e513cfa66;p=django_unres.git

diff --git a/files/fluct_plot.py b/files/fluct_plot.py
index 609ca4d..7719b2f 100755
--- a/files/fluct_plot.py
+++ b/files/fluct_plot.py
@@ -49,15 +49,21 @@ for i in range(1,natom+1):
    y.append(fluct)
 
 b=[]
-ires=''
+newchain=True
 if os.path.exists('plik.pdb'):
  with open('plik.pdb') as f:
    for line in f:
       if line[0:4]=='ATOM' and line[13:15]=='CA':
 #         print line
-        if ires=='' or int(line[22:26])>ires:
+        if newchain or int(line[22:26])>ires:
          b.append(math.sqrt(float(line[60:66])*3/8/math.pi/math.pi))
          ires=int(line[22:26])         
+         newchain=False
+      if line[0:3]=='TER':
+        newchain=True
+      if line[0:3]=='END':
+        break
+                        
 
 
 plt.xlabel('residue')