Mercurial > repos > chemteam > gmx_setup
annotate merge_top.py @ 5:35b314fe6c90 draft
"planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 10f45d117e34624bceb32442742ae0b8eb2dae30"
author | chemteam |
---|---|
date | Wed, 30 Oct 2019 13:35:25 -0400 |
parents | c124921a9e5f |
children | cac1886249a2 |
rev | line source |
---|---|
2
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
1 import re |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
2 import sys |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
3 |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
4 |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
5 def combine_tops(top_text, itp_texts): |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
6 """ |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
7 Search through parent topology top_text and replace |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
8 #include lines with the relevant child topologies |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
9 from the dictionary itp_texts |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
10 """ |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
11 for itp in itp_texts: |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
12 # split on include string, then rejoin around itp file |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
13 spl = re.split('#include ".*{}"\n'.format(itp), top_text) |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
14 top_text = itp_texts[itp].join(spl) |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
15 return top_text |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
16 |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
17 |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
18 top = sys.argv[1] # parent topology file |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
19 itps_file = sys.argv[2] # file with list of child topologies (.itp files) |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
20 |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
21 with open(itps_file) as f: |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
22 itps = f.read().split() |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
23 |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
24 with open(top, 'r') as f: |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
25 top_text = f.read() |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
26 |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
27 itp_texts = {} # create dictionary of child topologies |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
28 for itp in itps: |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
29 with open(itp, 'r') as f: |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
30 itp_texts[itp] = f.read() |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
31 |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
32 for itp in itp_texts: |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
33 # child tops may also refer to each other; we need to check this |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
34 itp_texts[itp] = combine_tops(itp_texts[itp], itp_texts) |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
35 |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
36 with open('top_output.top', 'w') as f: |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
37 # now combine all children into the parent |
c124921a9e5f
planemo upload for repository https://github.com/galaxycomputationalchemistry/galaxy-tools-compchem/tools/gromacs commit 7ba07c3130a2170e8f91acc4a81f5626563a79eb
chemteam
parents:
diff
changeset
|
38 f.write(combine_tops(top_text, itp_texts)) |