saxs restraints

[django_unres.git] / django_simple / todo / jobfiles.py

diff --git a/django_simple/todo/jobfiles.py b/django_simple/todo/jobfiles.py
index d09ab98..fd7fd7c 100644 (file)
--- a/django_simple/todo/jobfiles.py
+++ b/django_simple/todo/jobfiles.py
@@ -48,15 +48,13 @@ def write_on_task_save(sender, instance, **kwargs):
 WSCLOC=0.10554 WTOR=1.34316 WTORD=1.26571 WCORRH=0.19212 WCORR5=0.00000        &
 WCORR6=0.00000 WEL_LOC=0.37357 WTURN3=1.40323 WTURN4=0.64673 WTURN6=0.00000    &
 WVDWPP=0.23173 WHPB=1.00000 WSCCOR=0.25                                        &
-CUTOFF=7.00000 WCORR4=0.00000
-"""
+CUTOFF=7.00000 WCORR4=0.00000"""
      else:
       w="""WSC=0.82686 WSCP=0.96947 WELEC=0.79373 WBOND=1.00000 WANG=0.46542              &
 WSCLOC=0.07969 WTOR=0.81684 WTORD=0.67806 WCORRH=0.00000 WCORR5=0.00000        &
 WCORR6=0.00000 WEL_LOC=0.71100 WTURN3=2.30298 WTURN4=0.86517 WTURN6=0.00000    &
 WSCCOR=0.14577 WVDWPP=0.17781 WHPB=1.00000 WSCP14=0.00000                      &
-CUTOFF=7.00000 WCORR4=0.00000
-"""
+CUTOFF=7.00000 WCORR4=0.00000"""
      
 
      with open(instance.jobdirname+'/file.inp','w') as f:
@@ -134,7 +132,16 @@ CUTOFF=7.00000 WCORR4=0.00000
            
           if instance.md_pdbref:
              control_line = control_line+'pdbref '
-           
+          
+          cntrl_saxs=''
+          if instance.saxs_data != '':
+             lines=instance.saxs_data.split('\n')
+             nsaxs=0
+             for line in lines:
+              if len(line.split())==2:
+               nsaxs+=1 
+             cntrl_saxs=' nsaxs='+str(nsaxs)+' scal_rad='+str(instance.scal_rad) 
+             control_line += cntrl_saxs             
             
           type_line = 'reset_vel='+ str(instance.remd_nstex)\
              +' nstep='+str(instance.md_nstep)\
@@ -195,8 +202,10 @@ CUTOFF=7.00000 WCORR4=0.00000
           for element in word_list[:-1]:
               f.write('{:79}'.format(element)+'&\n')
           f.write(word_list[-1]+'\n')
-
-       f.write(w)
+       
+       if instance.saxs_data != '':
+         w+=' wsaxs='+str(instance.wsaxs)
+       f.write(w+'\n')
 
 
        if instance.type == 'min' or instance.md_start == 'pdbstart':
@@ -223,7 +232,12 @@ CUTOFF=7.00000 WCORR4=0.00000
              f.write(seq[i:i+80]+'\n')
           f.write('0\n0\n')
 
-
+       if instance.saxs_data != '':
+             lines=instance.saxs_data.split('\n')
+             for line in lines:
+              tmp=line.split()
+              if len(tmp)==2:
+               f.write(tmp[0]+' '+tmp[1]+'\n') 
        
         
      if instance.type == 'min':
@@ -309,20 +323,29 @@ CUTOFF=7.00000 WCORR4=0.00000
             f.write('{:79}'.format('SEED='+str(instance.md_seed)+' isampl='+str(isampl)+
                ' einicheck=1 rescale=2 delta=0.02 cxfile classify')+'&\n')
             f.write('BOXX='+str(instance.boxx)+' BOXY='+str(instance.boxy)+
-                    ' BOXZ='+str(instance.boxz) +'\n')
+                    ' BOXZ='+str(instance.boxz)+cntrl_saxs +'\n')
           else:    
             f.write('{:79}'.format('SEED='+str(instance.md_seed)+' isampl='+str(isampl)+
               ' einicheck=1 rescale=2 delta=0.02 cxfile')+'&\n')
             f.write('BOXX='+str(instance.boxx)+' BOXY='+str(instance.boxy)+
-                    ' BOXZ='+str(instance.boxz) +'\n')
+                    ' BOXZ='+str(instance.boxz)+cntrl_saxs +'\n')
           f.write('nres='+str(len(seq))+' one_letter\n')
           for i in range(0,len(seq),80):
              f.write(seq[i:i+80]+'\n')
 
           f.write(write_ssbond(instance.ssbond))
 
-
-          f.write(w)
+          if instance.saxs_data != '':
+             fsaxs = open(instance.jobdirname+'/saxs.data', 'w')
+             lines=instance.saxs_data.split('\n')
+             for line in lines:
+              tmp=line.split()
+              if len(tmp)==2:
+               f.write(tmp[0]+' '+tmp[1]+'\n')
+               fsaxs.write(tmp[0]+' '+tmp[1]+'\n') 
+             fsaxs.close()
+
+          f.write(w+'\n')
           f.write('\n')
           f.write('nt='+str(instance.remd_nrep)+' replica read_iset\n')
           tmp1=json.loads(instance.remd_multi_t)
@@ -344,11 +367,17 @@ CUTOFF=7.00000 WCORR4=0.00000
              +' one_letter rescale=2 PRINT_CART PDBOUT=1 iopt=1'
              +' temper='+str(instance.remd_cluter_temp))+'&\n')
           f.write('BOXX='+str(instance.boxx)+' BOXY='+str(instance.boxy)+
-                  ' BOXZ='+str(instance.boxz) +'\n')
-          f.write(w)
+                  ' BOXZ='+str(instance.boxz)+cntrl_saxs +'\n')
+          f.write(w+'\n')
           for i in range(0,len(seq),80):
              f.write(seq[i:i+80]+'\n')
           f.write(write_ssbond(instance.ssbond))
+          if instance.saxs_data != '':
+             lines=instance.saxs_data.split('\n')
+             for line in lines:
+              tmp=line.split()
+              if len(tmp)==2:
+               f.write(tmp[0]+' '+tmp[1]+'\n') 
       
       os.chdir(instance.jobdirname)  
       ret_code = subprocess.Popen(' /opt/torque/bin/qsub pbs8.csh', shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)