Skip to content

Commit ce7f533

Browse files
committed
added fallback for xyz format
1 parent b002827 commit ce7f533

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

rmsd/calculate_rmsd.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1138,11 +1138,16 @@ def get_coordinates_xyz(filename):
11381138
if lines_read == n_atoms:
11391139
break
11401140

1141-
atom = re.findall(r'[a-zA-Z]+', line)[0]
1142-
atom = atom.upper()
1141+
values = line.split()
11431142

1144-
numbers = re.findall(r'[-]?\d+\.\d*(?:[Ee][-\+]\d+)?', line)
1145-
numbers = [float(number) for number in numbers]
1143+
if len(values) < 4:
1144+
atom = re.findall(r'[a-zA-Z]+', line)[0]
1145+
atom = atom.upper()
1146+
numbers = re.findall(r'[-]?\d+\.\d*(?:[Ee][-\+]\d+)?', line)
1147+
numbers = [float(number) for number in numbers]
1148+
else:
1149+
atom = values[0].upper()
1150+
numbers = [float(number) for number in values[1:]]
11461151

11471152
# The numbers are not valid unless we obtain exacly three
11481153
if len(numbers) >= 3:

0 commit comments

Comments
 (0)