1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
|
# -*- Mode: python; tab-width: 4; indent-tabs-mode:nil; coding:utf-8 -*-
# vim: tabstop=4 expandtab shiftwidth=4 softtabstop=4
#
# MDAnalysis --- https://www.mdanalysis.org
# Copyright (c) 2006-2017 The MDAnalysis Development Team and contributors
# (see the file AUTHORS for the full list of names)
#
# Released under the Lesser GNU Public Licence, v2.1 or any higher version
#
# Please cite your use of MDAnalysis in published work:
#
# R. J. Gowers, M. Linke, J. Barnoud, T. J. E. Reddy, M. N. Melo, S. L. Seyler,
# D. L. Dotson, J. Domanski, S. Buchoux, I. M. Kenney, and O. Beckstein.
# MDAnalysis: A Python package for the rapid analysis of molecular dynamics
# simulations. In S. Benthall and S. Rostrup editors, Proceedings of the 15th
# Python in Science Conference, pages 102-109, Austin, TX, 2016. SciPy.
# doi: 10.25080/majora-629e541a-00e
#
# N. Michaud-Agrawal, E. J. Denning, T. B. Woolf, and O. Beckstein.
# MDAnalysis: A Toolkit for the Analysis of Molecular Dynamics Simulations.
# J. Comput. Chem. 32 (2011), 2319--2327, doi:10.1002/jcc.21787
#
"""DL_Poly format reader :mod:`MDAnalysis.coordinates.DLPoly`
=============================================================
Read DL Poly_ format coordinate files
.. _Poly: https://www.sc.stfc.ac.uk/software/type/computational-materials-and-molecular-science/?searchquery=dl_poly
"""
import numpy as np
from . import base
from . import core
from ..lib import util
from ..lib.util import cached, store_init_arguments
_DLPOLY_UNITS = {'length': 'Angstrom', 'velocity': 'Angstrom/ps', 'time': 'ps'}
class ConfigReader(base.SingleFrameReaderBase):
"""DLPoly Config file Reader
.. versionadded:: 0.11.0
.. versionchanged:: 2.0.0
coordinates, velocities, and forces are no longer stored in 'F' memory
layout, instead now using the numpy default of 'C'.
"""
format = 'CONFIG'
units = _DLPOLY_UNITS
def _read_first_frame(self):
unitcell = np.zeros((3, 3), dtype=np.float32)
with open(self.filename, 'r') as inf:
self.title = inf.readline().strip()
levcfg, imcon, megatm = np.int64(inf.readline().split()[:3])
if not imcon == 0:
unitcell[0][:] = inf.readline().split()
unitcell[1][:] = inf.readline().split()
unitcell[2][:] = inf.readline().split()
ids = []
coords = []
if levcfg > 0:
has_vels = True
velocities = []
else:
has_vels = False
if levcfg == 2:
has_forces = True
forces = []
else:
has_forces = False
line = inf.readline().strip()
# Read records infinitely
while line:
try:
idx = int(line[8:])
except ValueError: # dl_poly classic doesn't have this
pass
else:
ids.append(idx)
xyz = np.float32(inf.readline().split())
coords.append(xyz)
if has_vels:
vxyz = np.float32(inf.readline().split())
velocities.append(vxyz)
if has_forces:
fxyz = np.float32(inf.readline().split())
forces.append(fxyz)
line = inf.readline().strip()
coords = np.array(coords, dtype=np.float32)
if has_vels:
velocities = np.array(velocities, dtype=np.float32)
if has_forces:
forces = np.array(forces, dtype=np.float32)
self.n_atoms = len(coords)
if ids:
# If we have indices then sort based on them
ids = np.array(ids)
order = np.argsort(ids)
coords = coords[order]
if has_vels:
velocities = velocities[order]
if has_forces:
forces = forces[order]
ts = self.ts = self._Timestep(self.n_atoms,
velocities=has_vels,
forces=has_forces,
**self._ts_kwargs)
ts._pos = coords
if has_vels:
ts._velocities = velocities
if has_forces:
ts._forces = forces
if not imcon == 0:
ts.dimensions = core.triclinic_box(*unitcell)
ts.frame = 0
class HistoryReader(base.ReaderBase):
"""Reads DLPoly format HISTORY files
.. versionadded:: 0.11.0
"""
format = 'HISTORY'
units = _DLPOLY_UNITS
@store_init_arguments
def __init__(self, filename, **kwargs):
super(HistoryReader, self).__init__(filename, **kwargs)
self._cache = {}
# "private" file handle
self._file = util.anyopen(self.filename, 'r')
self.title = self._file.readline().strip()
header = np.int64(self._file.readline().split()[:3])
self._levcfg, self._imcon, self.n_atoms = header
self._has_vels = True if self._levcfg > 0 else False
self._has_forces = True if self._levcfg == 2 else False
rwnd = self._file.tell()
self._file.readline()
if (len(self._file.readline().split())) == 3:
self._has_cell = True
else:
self._has_cell = False
self._file.seek(rwnd)
self.ts = self._Timestep(self.n_atoms,
velocities=self._has_vels,
forces=self._has_forces,
**self._ts_kwargs)
self._read_next_timestep()
def _read_next_timestep(self, ts=None):
if ts is None:
ts = self.ts
line = self._file.readline() # timestep line
if not line.startswith('timestep'):
raise IOError
if self._has_cell:
unitcell = np.zeros((3, 3))
unitcell[0] = self._file.readline().split()
unitcell[1] = self._file.readline().split()
unitcell[2] = self._file.readline().split()
ts.dimensions = core.triclinic_box(*unitcell)
# If ids are given, put them in here
# and later sort by them
ids = []
for i in range(self.n_atoms):
line = self._file.readline().strip() # atom info line
try:
idx = int(line.split()[1])
except IndexError:
pass
else:
ids.append(idx)
# Read in this order for now, then later reorder in place
ts._pos[i] = self._file.readline().split()
if self._has_vels:
ts._velocities[i] = self._file.readline().split()
if self._has_forces:
ts._forces[i] = self._file.readline().split()
i += 1
if ids:
ids = np.array(ids)
# if ids aren't strictly sequential
if not np.all(ids == (np.arange(self.n_atoms) + 1)):
order = np.argsort(ids)
ts._pos[:] = ts._pos[order]
if self._has_vels:
ts._velocities[:] = ts._velocities[order]
if self._has_forces:
ts._forces[:] = ts._forces[order]
ts.frame += 1
return ts
def _read_frame(self, frame):
"""frame is 0 based, error checking is done in base.getitem"""
self._file.seek(self._offsets[frame])
self.ts.frame = frame - 1 # gets +1'd in read_next_frame
return self._read_next_timestep()
@property
@cached('n_frames')
def n_frames(self):
# Second line is traj_key, imcom, n_atoms, n_frames, n_records
offsets = []
with open(self.filename, 'r') as f:
f.readline()
f.readline()
position = f.tell()
line = f.readline()
while line.startswith('timestep'):
offsets.append(position)
if self._has_cell:
f.readline()
f.readline()
f.readline()
for _ in range(self.n_atoms):
f.readline()
f.readline()
if self._has_vels:
f.readline()
if self._has_forces:
f.readline()
position = f.tell()
line = f.readline()
self._offsets = offsets
return len(self._offsets)
def _reopen(self):
self.close()
self._file = open(self.filename, 'r')
self._file.readline() # header is 2 lines
self._file.readline()
self.ts.frame = -1
def close(self):
self._file.close()
|
