unres paper list update

[django_unres.git] / files / plot_saxs.py

diff --git a/files/plot_saxs.py b/files/plot_saxs.py
index 292550b..834e842 100755 (executable)
--- a/files/plot_saxs.py
+++ b/files/plot_saxs.py
@@ -7,6 +7,10 @@ import matplotlib.pyplot as plt
 import matplotlib.cm as cm
 import numpy as np
 import sys
+import os.path
+
+matplotlib.rc('font', size=15)
+matplotlib.rc('legend', fontsize=15)
 
 x,y= np.loadtxt('saxs.data',usecols=(0,1),unpack=True)
 x1,y1= np.loadtxt('MODEL1_saxs.data',usecols=(0,2),unpack=True)
@@ -17,7 +21,7 @@ x5,y5= np.loadtxt('MODEL5_saxs.data',usecols=(0,2),unpack=True)
 
 
 plt.clf()
-plt.xlabel('r')
+plt.xlabel(r'r [$\AA$]')
 plt.ylabel('P(r)')
 plt.xlim(x[0], x[-1])
 plt.plot(x,y,'-',color='black') 
@@ -26,5 +30,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')