X-Git-Url: http://mmka.chem.univ.gda.pl/gitweb/?a=blobdiff_plain;f=files%2Ffluct.py;fp=files%2Ffluct.py;h=f12f8c95124ed9ba450a8209062b8e0d99e10477;hb=bc399332f4fd88053d20cbb3708878fe5f577592;hp=6865492b9c49776e6c53669320b9fd8858c68fe7;hpb=49226cc4e01dd935d1fd063e5215a668ac07f7e5;p=django_unres.git

diff --git a/files/fluct.py b/files/fluct.py
index 6865492..f12f8c9 100644
--- a/files/fluct.py
+++ b/files/fluct.py
@@ -1,8 +1,20 @@
 from pymol import cmd, stored, math
 import time
 
-cmd.load('plik.pdb')
-mol='plik'
+with open('plik.pdb') as f, open('plik1.pdb', 'w') as f1:
+   prev_ires=None
+   ires = 0
+   for line in f:
+     if line[0:6] == 'ATOM  ':
+       if line[22:26] != prev_ires:
+         prev_ires=line[22:26]
+         ires += 1
+       f1.write(line[:22]+str(ires).rjust(4)+line[26:])
+     elif line[0:3] == 'TER':
+       f1.write(line)
+
+cmd.load('plik1.pdb')
+mol='plik1'
 obj=cmd.get_object_list(mol)[0]
 inFile = open('fluctuations.txt', 'r')