X-Git-Url: http://mmka.chem.univ.gda.pl/gitweb/?a=blobdiff_plain;f=files%2Fplot_saxs.py;h=834e842a093b3a60e9bee3dbd97c19d712066c67;hb=d03bf58e6dc3d690486f8eb7a8f18ad34938c71f;hp=292550bb8143d56b3bbceff2349bb30f7ecf836a;hpb=18720bfd2db199afb2ebf034b3206fc07f16c1ad;p=django_unres.git

diff --git a/files/plot_saxs.py b/files/plot_saxs.py
index 292550b..834e842 100755
--- 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')