saxs input data normalization and plot for input pdb

[django_unres.git] / files / plot_saxs.py

diff --git a/files/plot_saxs.py b/files/plot_saxs.py
index 292550b..d1ff7f6 100755 (executable)
--- a/files/plot_saxs.py
+++ b/files/plot_saxs.py
@@ -7,6 +7,7 @@ import matplotlib.pyplot as plt
 import matplotlib.cm as cm
 import numpy as np
 import sys
+import os.path
 
 x,y= np.loadtxt('saxs.data',usecols=(0,1),unpack=True)
 x1,y1= np.loadtxt('MODEL1_saxs.data',usecols=(0,2),unpack=True)
@@ -26,5 +27,10 @@ plt.plot(x2,y2,'-',color='lime')
 plt.plot(x3,y3,'-',color='blue')
 plt.plot(x4,y4,'-',color='magenta')
 plt.plot(x5,y5,'-',color='cyan')
-plt.legend(['reference','MODEL1','MODEL2','MODEL3','MODEL4','MODEL5'])
+if os.path.isfile('plik_saxs.data'):
+  x6,y6= np.loadtxt('plik_saxs.data',usecols=(0,2),unpack=True)
+  plt.plot(x6,y6,'--',color='black')
+  plt.legend(['reference','MODEL1','MODEL2','MODEL3','MODEL4','MODEL5','input pdb'])
+else:
+  plt.legend(['reference','MODEL1','MODEL2','MODEL3','MODEL4','MODEL5'])
 plt.savefig('saxs.png')