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.8, <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

environment.yml

@@ -2,7 +2,7 @@ name: gmso
 channels:
   - conda-forge
 dependencies:
-  - python>=3.8
+  - python>=3.8, <3.12
   - boltons
   - numpy
   - sympy

gmso/external/convert_hoomd.py

@@ -22,7 +22,11 @@
 )
 from gmso.utils.geometry import coord_shift
 from gmso.utils.io import has_gsd, has_hoomd
-from gmso.utils.sorting import sort_by_classes, sort_connection_members
+from gmso.utils.sorting import (
+    sort_by_classes,
+    sort_by_types,
+    sort_connection_members,
+)
 from gmso.utils.units import convert_params_units
 
 if has_gsd:
@@ -43,6 +47,8 @@
     "mass": u.g / u.mol,  # aka amu
 }
 
+hoomd_version = hoomd.version.version.split(".")
+
 
 def to_gsd_snapshot(
     top,
@@ -678,13 +684,15 @@
     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,
     )
@@ -993,8 +1001,11 @@
 ):
     for btype in btypes:
         # TODO: Unit conversion
-        member_classes = sort_by_classes(btype)
-        container.params["-".join(member_classes)] = {
+        members = sort_by_classes(btype)
+        # If wild card in class, sort by types instead
+        if "*" in members:
+            members = sort_by_types(btype)
+        container.params["-".join(members)] = {
             "k": btype.parameters["k"],
             "r0": btype.parameters["r_eq"],
         }
@@ -1063,8 +1074,11 @@
     agtypes,
 ):
     for agtype in agtypes:
-        member_classes = sort_by_classes(agtype)
-        container.params["-".join(member_classes)] = {
+        members = sort_by_classes(agtype)
+        # If wild card in class, sort by types instead
+        if "*" in members:
+            members = sort_by_types(atype)
+        container.params["-".join(members)] = {
             "k": agtype.parameters["k"],
             "t0": agtype.parameters["theta_eq"],
         }
@@ -1120,7 +1134,6 @@
         },
     }
 
-    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 +1320,24 @@
             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:
@@ -1330,14 +1355,35 @@
     itypes,
 ):
     for itype in itypes:
-        member_types = sort_by_classes(itype)
+        members = sort_by_classes(itype)
+        # If wild card in class, sort by types instead
+        if "*" in members:
+            members = sort_by_types(itype)
         container.params["-".join(member_types)] = {
             "k": itype.parameters["k"],
             "chi0": itype.parameters["phi_eq"],  # diff nomenclature?
         }
     return container
 
 
+def _parse_periodic_improper(
+    container,
+    itypes,
+):
+    for itype in itypes:
+        members = sort_by_classes(itype)
+        # If wild card in class, sort by types instead
+        if "*" in members:
+            members = sort_by_types(itype)
+        container.params["-".join(members)] = {
+            "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

@@ -409,9 +409,9 @@ def test_gaff_sim(self, gaff_forcefield):
             "length": u.nm,
             "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
         )