Add support for hoomd.md.improper.Periodic in to_hoomd_forcefield

environment-dev.yml

@@ -2,7 +2,7 @@ name: gmso-dev
 channels:
   - conda-forge
 dependencies:
-  - python>=3.8
+  - python<3.12
   - boltons
   - numpy
   - sympy
@@ -12,7 +12,7 @@ dependencies:
   - pydantic>=2
   - networkx
   - pytest
-  - mbuild>=0.11.0
+  - mbuild>=0.17.0
   - openbabel>=3.0.0
   - foyer>=0.11.3
   - forcefield-utilities>=0.2.1

gmso/external/convert_hoomd.py

@@ -43,6 +43,8 @@
     "mass": u.g / u.mol,  # aka amu
 }
 
+hoomd_version = hoomd.version.version.split(".")
+
 
 def to_gsd_snapshot(
     top,
@@ -678,13 +680,15 @@ def _validate_compatibility(top):
     harmonic_bond_potential = templates["HarmonicBondPotential"]
     harmonic_angle_potential = templates["HarmonicAnglePotential"]
     periodic_torsion_potential = templates["PeriodicTorsionPotential"]
+    harmonic_torsion_potential = templates["HarmonicTorsionPotential"]
     opls_torsion_potential = templates["OPLSTorsionPotential"]
     rb_torsion_potential = templates["RyckaertBellemansTorsionPotential"]
     accepted_potentials = (
         lennard_jones_potential,
         harmonic_bond_potential,
         harmonic_angle_potential,
         periodic_torsion_potential,
+        harmonic_torsion_potential,
         opls_torsion_potential,
         rb_torsion_potential,
     )
@@ -1120,7 +1124,6 @@ def _parse_dihedral_forces(
         },
     }
 
-    hoomd_version = hoomd.version.version.split(".")
     if int(hoomd_version[0]) >= 4 or (
         int(hoomd_version[0]) == 3 and int(hoomd_version[1]) >= 8
     ):
@@ -1307,12 +1310,24 @@ def _parse_improper_forces(
             base_units,
         )
 
-    itype_group_map = {
-        "HarmonicImproperPotenial": {
-            "container": hoomd.md.improper.Harmonic,
-            "parser": _parse_harmonic_improper,
-        },
-    }
+    if int(hoomd_version[0]) >= 4 and int(hoomd_version[1]) >= 5:
+        itype_group_map = {
+            "HarmonicImproperPotential": {
+                "container": hoomd.md.improper.Harmonic,
+                "parser": _parse_harmonic_improper,
+            },
+            "PeriodicTorsionPotential": {
+                "container": hoomd.md.improper.Periodic,
+                "parser": _parse_periodic_improper,
+            },
+        }
+    else:
+        itype_group_map = {
+            "HarmonicImproperPotential": {
+                "container": hoomd.md.improper.Harmonic,
+                "parser": _parse_harmonic_improper,
+            },
+        }
 
     improper_forces = list()
     for group in groups:
@@ -1338,6 +1353,21 @@ def _parse_harmonic_improper(
     return container
 
 
+def _parse_periodic_improper(
+    container,
+    itypes,
+):
+    for itype in itypes:
+        member_types = sort_by_classes(itype)
+        container.params["-".join(member_types)] = {
+            "k": itype.parameters["k"],
+            "chi0": itype.parameters["phi_eq"],
+            "n": itype.parameters["n"],
+            "d": 1.0,
+        }
+    return container
+
+
 def _validate_base_units(base_units, top, auto_scale, potential_types=None):
     """Validate the provided base units, infer units (based on top's positions and masses) if none is provided."""
     if base_units and auto_scale:

gmso/tests/test_hoomd.py

@@ -410,8 +410,9 @@ def test_gaff_sim(self, gaff_forcefield):
             "energy": u.kJ / u.mol,
         }
         ethanol = mb.load("CCO", smiles=True)
-        ethanol.box = mb.Box([5, 5, 5])
-        top = ethanol.to_gmso()
+        benzene = mb.load("c1ccccc1", smiles=True)
+        benzene.box = mb.Box([5, 5, 5])
+        top = benzene.to_gmso()
         parameterized_top = apply(
             top, gaff_forcefield, identify_connections=True
         )