Add number of bonds, angles, and dihedrals

chrisjonesBSU · Mar 11, 2020 · 50c11d0 · 50c11d0
1 parent 459ac4e
commit 50c11d0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gmso/formats/lammpsdata.py b/gmso/formats/lammpsdata.py
@@ -62,7 +62,10 @@ def write_lammpsdata(topology, filename, atom_style='full'):
             else:
                 data.write('0 dihedrals\n')
 
-        data.write('{:d} atom types\n'.format(len(topology.atom_types)))
+        data.write('\n{:d} atom types\n'.format(len(topology.atom_types)))
+        data.write('{:d} bond types\n'.format(len(topology.bond_types)))
+        data.write('{:d} angle types\n'.format(len(topology.angle_types)))
+        data.write('{:d} dihedral types\n'.format(len(topology.dihedral_types)))
 
         data.write('\n')