imdclient 0.1.1__tar.gz → 0.1.3__tar.gz

imdclient-0.1.3/LICENSE

@@ -0,0 +1,5 @@
+Copyright 2024 Lawson Woods
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.

{imdclient-0.1.1 → imdclient-0.1.3}/PKG-INFO

@@ -1,31 +1,14 @@
 Metadata-Version: 2.1
 Name: imdclient
-Version: 0.1.1
+Version: 0.1.3
 Summary: Receiver for IMD v2 and v3 data from simulation engines like Gromacs, LAMMPS, and NAMD
 Author-email: Lawson <ljwoods2@asu.edu>
 Maintainer-email: Lawson <ljwoods2@asu.edu>
-License: MIT License
+License: Copyright 2024 Lawson Woods
         
-        Copyright (c) 2024 ljwoods2
-        
-        Permission is hereby granted, free of charge, to any person obtaining a copy
-        of this software and associated documentation files (the "Software"), to deal
-        in the Software without restriction, including without limitation the rights
-        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-        copies of the Software, and to permit persons to whom the Software is
-        furnished to do so, subject to the following conditions:
-        
-        The above copyright notice and this permission notice shall be included in all
-        copies or substantial portions of the Software.
-        
-        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-        SOFTWARE.
+        Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
         
+        The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
 Keywords: molecular simulations
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
@@ -37,6 +20,7 @@ Requires-Dist: pytest>=6.0; extra == "test"
 Requires-Dist: pytest-xdist>=2.5; extra == "test"
 Requires-Dist: pytest-cov>=3.0; extra == "test"
 Requires-Dist: MDAnalysisTests>=2.7.0; extra == "test"
+Requires-Dist: docker-py; extra == "test"
 Provides-Extra: doc
 Requires-Dist: sphinx; extra == "doc"
 Requires-Dist: sphinx_rtd_theme; extra == "doc"
@@ -48,7 +32,7 @@ IMDClient
 | **Latest release** | [![Last release tag][badge_release]][url_latest_release] ![GitHub commits since latest release (by date) for a branch][badge_commits_since]  [![Documentation Status][badge_docs]][url_docs]|
 | :----------------- | :------- |
 | **Status**         | [![GH Actions Status][badge_actions]][url_actions] [![codecov][badge_codecov]][url_codecov] |
-| **Community**      | [![License: GPL v2][badge_license]][url_license]  [![Powered by MDAnalysis][badge_mda]][url_mda]|
+| **Community**      | [![License: MIT][badge_license]][url_license]  [![Powered by MDAnalysis][badge_mda]][url_mda]|
 
 [badge_actions]: https://github.com/becksteinlab/imdclient/actions/workflows/gh-ci.yaml/badge.svg
 [badge_codecov]: https://codecov.io/gh/becksteinlab/imdclient/branch/main/graph/badge.svg
@@ -61,15 +45,20 @@ IMDClient
 [url_codecov]: https://codecov.io/gh/becksteinlab/imdclient/branch/main
 [url_docs]: https://imdclient.readthedocs.io/en/latest/?badge=latest
 [url_latest_release]: https://github.com/becksteinlab/imdclient/releases
-[url_license]: https://www.gnu.org/licenses/gpl-2.0
+[url_license]: https://opensource.org/license/mit
 [url_mda]: https://www.mdanalysis.org
 
-Receiver for IMDv3 protocol from simulation engines like Gromacs, LAMMPS, and NAMD.
+Receiver for [IMDv3 protocol](https://imdclient.readthedocs.io/en/latest/protocol_v3.html) from simulation engines like Gromacs, LAMMPS, and NAMD.
 
 IMDClient is bound by a [Code of Conduct](https://github.com/becksteinlab/imdreader/blob/main/CODE_OF_CONDUCT.md).
 
 ### Installation
 
+IMDClient is available via PyPi and can be installed with pip:
+```bash
+pip install imdclient
+```
+
 To build IMDClient from source,
 we highly recommend using virtual environments.
 If possible, we strongly recommend that you use

{imdclient-0.1.1 → imdclient-0.1.3}/README.md

@@ -5,7 +5,7 @@ IMDClient
 | **Latest release** | [![Last release tag][badge_release]][url_latest_release] ![GitHub commits since latest release (by date) for a branch][badge_commits_since]  [![Documentation Status][badge_docs]][url_docs]|
 | :----------------- | :------- |
 | **Status**         | [![GH Actions Status][badge_actions]][url_actions] [![codecov][badge_codecov]][url_codecov] |
-| **Community**      | [![License: GPL v2][badge_license]][url_license]  [![Powered by MDAnalysis][badge_mda]][url_mda]|
+| **Community**      | [![License: MIT][badge_license]][url_license]  [![Powered by MDAnalysis][badge_mda]][url_mda]|
 
 [badge_actions]: https://github.com/becksteinlab/imdclient/actions/workflows/gh-ci.yaml/badge.svg
 [badge_codecov]: https://codecov.io/gh/becksteinlab/imdclient/branch/main/graph/badge.svg
@@ -18,15 +18,20 @@ IMDClient
 [url_codecov]: https://codecov.io/gh/becksteinlab/imdclient/branch/main
 [url_docs]: https://imdclient.readthedocs.io/en/latest/?badge=latest
 [url_latest_release]: https://github.com/becksteinlab/imdclient/releases
-[url_license]: https://www.gnu.org/licenses/gpl-2.0
+[url_license]: https://opensource.org/license/mit
 [url_mda]: https://www.mdanalysis.org
 
-Receiver for IMDv3 protocol from simulation engines like Gromacs, LAMMPS, and NAMD.
+Receiver for [IMDv3 protocol](https://imdclient.readthedocs.io/en/latest/protocol_v3.html) from simulation engines like Gromacs, LAMMPS, and NAMD.
 
 IMDClient is bound by a [Code of Conduct](https://github.com/becksteinlab/imdreader/blob/main/CODE_OF_CONDUCT.md).
 
 ### Installation
 
+IMDClient is available via PyPi and can be installed with pip:
+```bash
+pip install imdclient
+```
+
 To build IMDClient from source,
 we highly recommend using virtual environments.
 If possible, we strongly recommend that you use

imdclient-0.1.1/imdclient/IMDREADER.py → imdclient-0.1.3/imdclient/IMD.py

@@ -24,6 +24,17 @@ logger = logging.getLogger("imdclient.IMDClient")
 class IMDReader(StreamReaderBase):
     """
     Reader for IMD protocol packets.
+
+    Parameters
+    ----------
+    filename : a string of the form "host:port" where host is the hostname
+        or IP address of the listening GROMACS server and port
+        is the port number.
+    n_atoms : int (optional)
+        number of atoms in the system. defaults to number of atoms
+        in the topology. don't set this unless you know what you're doing.
+    kwargs : dict (optional)
+        keyword arguments passed to the constructed :class:`IMDClient`
     """
 
     format = "IMD"
@@ -37,17 +48,6 @@ class IMDReader(StreamReaderBase):
         n_atoms=None,
         **kwargs,
     ):
-        """
-        Parameters
-        ----------
-        filename : a string of the form "host:port" where host is the hostname
-            or IP address of the listening GROMACS server and port
-            is the port number.
-        n_atoms : int (optional)
-            number of atoms in the system. defaults to number of atoms
-            in the topology. don't set this unless you know what you're doing.
-        """
-
         super(IMDReader, self).__init__(filename, **kwargs)
 
         logger.debug("IMDReader initializing")
@@ -83,9 +83,8 @@ class IMDReader(StreamReaderBase):
 
         try:
             imdf = self._imdclient.get_imdframe()
-        except EOFError:
-            # Not strictly necessary, but for clarity
-            raise StopIteration
+        except EOFError as e:
+            raise e
 
         self._frame = frame
         self._load_imdframe_into_ts(imdf)
@@ -101,7 +100,9 @@ class IMDReader(StreamReaderBase):
             self.ts.data["dt"] = imdf.dt
             self.ts.data["step"] = imdf.step
         if imdf.energies is not None:
-            self.ts.data.update(imdf.energies)
+            self.ts.data.update(
+                {k: v for k, v in imdf.energies.items() if k != "step"}
+            )
         if imdf.box is not None:
             self.ts.dimensions = core.triclinic_box(*imdf.box)
         if imdf.positions is not None:

{imdclient-0.1.1 → imdclient-0.1.3}/imdclient/IMDClient.py

@@ -25,11 +25,31 @@ import time
 import numpy as np
 from typing import Union, Dict
 import signal
+import atexit
 
 logger = logging.getLogger(__name__)
 
 
 class IMDClient:
+    """
+    Parameters
+    ----------
+    host : str
+        Hostname of the server
+    port : int
+        Port number of the server
+    n_atoms : int
+        Number of atoms in the simulation
+    socket_bufsize : int, (optional)
+        Size of the socket buffer in bytes. Default is to use the system default
+    buffer_size : int (optional)
+        IMDFramebuffer will be filled with as many :class:`IMDFrame` fit in `buffer_size` bytes [``10MB``]
+    timeout : int, optional
+        Timeout for the socket in seconds [``5``]
+    **kwargs : dict (optional)
+        Additional keyword arguments to pass to the :class:`BaseIMDProducer` and :class:`IMDFrameBuffer`
+    """
+
     def __init__(
         self,
         host,
@@ -39,22 +59,7 @@ class IMDClient:
         multithreaded=True,
         **kwargs,
     ):
-        """
-        Parameters
-        ----------
-        host : str
-            Hostname of the server
-        port : int
-            Port number of the server
-        n_atoms : int
-            Number of atoms in the simulation
-        socket_bufsize : int, optional
-            Size of the socket buffer in bytes. Default is to use the system default
-        buffer_size : int, optional
-            IMDFramebuffer will be filled with as many IMDFrames fit in `buffer_size` [``10MB``]
-        **kwargs : optional
-            Additional keyword arguments to pass to the IMDProducer and IMDFrameBuffer
-        """
+
         self._stopped = False
         self._conn = self._connect_to_server(host, port, socket_bufsize)
         self._imdsinfo = self._await_IMD_handshake()
@@ -66,8 +71,10 @@ class IMDClient:
                 n_atoms,
                 **kwargs,
             )
+            self._error_queue = queue.Queue()
         else:
             self._buf = None
+            self._error_queue = None
         if self._imdsinfo.version == 2:
             self._producer = IMDProducerV2(
                 self._conn,
@@ -75,6 +82,7 @@ class IMDClient:
                 self._imdsinfo,
                 n_atoms,
                 multithreaded,
+                self._error_queue,
                 **kwargs,
             )
         elif self._imdsinfo.version == 3:
@@ -84,23 +92,60 @@ class IMDClient:
                 self._imdsinfo,
                 n_atoms,
                 multithreaded,
+                self._error_queue,
                 **kwargs,
             )
 
         self._go()
 
         if self._multithreaded:
+            # Disconnect MUST occur. This covers typical cases (Python, IPython interpreter)
             signal.signal(signal.SIGINT, self.signal_handler)
+            signal.signal(signal.SIGTERM, self.signal_handler)
+
+            # Disconnect and socket shutdown MUST occur. This covers Jupyter use
+            # since in jupyter, the signal handler is reset to the default
+            # by pre- and post- hooks
+            # https://stackoverflow.com/questions/70841648/jupyter-reverts-signal-handler-to-default-when-running-next-cell
+            try:
+                import IPython
+            except ImportError:
+                has_ipython = False
+            else:
+                has_ipython = True
+
+            if has_ipython:
+                try:
+                    from IPython import get_ipython
+
+                    if get_ipython() is not None:
+                        kernel = get_ipython().kernel
+                        kernel.pre_handler_hook = lambda: None
+                        kernel.post_handler_hook = lambda: None
+                        logger.debug("Running in Jupyter")
+                except NameError:
+                    logger.debug("Running in non-jupyter IPython environment")
+
+            # Final case: error is raised outside of IMDClient code
+            logger.debug("Registering atexit")
+            atexit.register(self.stop)
+
             self._producer.start()
 
     def signal_handler(self, sig, frame):
         """Catch SIGINT to allow clean shutdown on CTRL+C
         This also ensures that main thread execution doesn't get stuck
         waiting in buf.pop_full_imdframe()"""
+        logger.debug("Intercepted signal")
         self.stop()
+        logger.debug("Shutdown success")
 
     def get_imdframe(self):
         """
+        Returns
+        -------
+        IMDFrame
+            The next frame from the IMD server
         Raises
         ------
         EOFError
@@ -114,6 +159,9 @@ class IMDClient:
                 # and doesn't need to be notified
                 self._disconnect()
                 self._stopped = True
+
+                if self._error_queue.qsize():
+                    raise EOFError(f"{self._error_queue.get()}")
                 raise EOFError
         else:
             try:
@@ -123,14 +171,23 @@ class IMDClient:
                 raise EOFError
 
     def get_imdsessioninfo(self):
+        """
+        Returns
+        -------
+        IMDSessionInfo
+            Information about the IMD session
+        """
         return self._imdsinfo
 
     def stop(self):
+        """
+        Stop the client and close the connection
+        """
         if self._multithreaded:
             if not self._stopped:
-                self._buf.notify_consumer_finished()
-                self._disconnect()
                 self._stopped = True
+                self._disconnect()
+                self._buf.notify_consumer_finished()
         else:
             self._disconnect()
 
@@ -144,9 +201,7 @@ class IMDClient:
             # /proc/sys/net/core/rmem_default
             # Max (linux):
             # /proc/sys/net/core/rmem_max
-            conn.setsockopt(
-                socket.SOL_SOCKET, socket.SO_RCVBUF, socket_bufsize
-            )
+            conn.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, socket_bufsize)
         try:
             logger.debug(f"IMDClient: Connecting to {host}:{port}")
             conn.connect((host, port))
@@ -252,8 +307,37 @@ class IMDClient:
         finally:
             self._conn.close()
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.stop()
+        return False
+
 
 class BaseIMDProducer(threading.Thread):
+    """
+
+    Parameters
+    ----------
+    conn : socket.socket
+        Connection object to the server
+    buffer : IMDFrameBuffer
+        Buffer object to hold IMD frames. If `multithreaded` is False, this
+        argument is ignored
+    sinfo : IMDSessionInfo
+        Information about the IMD session
+    n_atoms : int
+        Number of atoms in the simulation
+    multithreaded : bool
+        If True, socket interaction will occur in a separate thread &
+        frames will be buffered. Single-threaded, blocking IMDClient
+        should only be used in testing
+    error_queue: queue.Queue
+        Queue to hold errors produced by the producer thread
+    timeout : int, optional
+        Timeout for the socket in seconds [``5``]
+    """
 
     def __init__(
         self,
@@ -261,33 +345,17 @@ class BaseIMDProducer(threading.Thread):
         buffer,
         sinfo,
         n_atoms,
-        multithreaded=True,
+        multithreaded,
+        error_queue,
         timeout=5,
         **kwargs,
     ):
-        """
-        Parameters
-        ----------
-        conn : socket.socket
-            Connection object to the server
-        buffer : IMDFrameBuffer
-            Buffer object to hold IMD frames. If `multithreaded` is False, this
-            argument is ignored
-        sinfo : IMDSessionInfo
-            Information about the IMD session
-        n_atoms : int
-            Number of atoms in the simulation
-        multithreaded : bool, optional
-            If True, socket interaction will occur in a separate thread &
-            frames will be buffered. Single-threaded, blocking IMDClient
-            should only be used in testing [[``True``]]
-
-        """
         super(BaseIMDProducer, self).__init__(daemon=True)
         self._conn = conn
         self._imdsinfo = sinfo
         self._paused = False
 
+        self.error_queue = error_queue
         # Timeout for first frame should be longer
         # than rest of frames
         self._timeout = timeout
@@ -382,6 +450,7 @@ class BaseIMDProducer(threading.Thread):
             logger.debug("IMDProducer: Simulation ended normally, cleaning up")
         except Exception as e:
             logger.debug("IMDProducer: An unexpected error occurred: %s", e)
+            self.error_queue.put(e)
         finally:
             logger.debug("IMDProducer: Stopping run loop")
             # Tell consumer not to expect more frames to be added
@@ -397,9 +466,19 @@ class BaseIMDProducer(threading.Thread):
             )
         # Sometimes we do not care what the value is
         if expected_value is not None and header.length != expected_value:
-            raise RuntimeError(
-                f"IMDProducer: Expected header value {expected_value}, got {header.length}"
-            )
+            if expected_type in [
+                IMDHeaderType.IMD_FCOORDS,
+                IMDHeaderType.IMD_VELOCITIES,
+                IMDHeaderType.IMD_FORCES,
+            ]:
+                raise RuntimeError(
+                    f"IMDProducer: Expected n_atoms value {expected_value}, got {header.length}. "
+                    + "Ensure you are using the correct topology file."
+                )
+            else:
+                raise RuntimeError(
+                    f"IMDProducer: Expected header value {expected_value}, got {header.length}"
+                )
 
     def _get_header(self):
         self._read(self._header)
@@ -419,9 +498,11 @@ class BaseIMDProducer(threading.Thread):
 
 
 class IMDProducerV2(BaseIMDProducer):
-    def __init__(self, conn, buffer, sinfo, n_atoms, multithreaded, **kwargs):
+    def __init__(
+        self, conn, buffer, sinfo, n_atoms, multithreaded, error_queue, **kwargs
+    ):
         super(IMDProducerV2, self).__init__(
-            conn, buffer, sinfo, n_atoms, multithreaded, **kwargs
+            conn, buffer, sinfo, n_atoms, multithreaded, error_queue, **kwargs
         )
 
         self._energies = bytearray(IMDENERGYPACKETLENGTH)
@@ -514,6 +595,7 @@ class IMDProducerV3(BaseIMDProducer):
         sinfo,
         n_atoms,
         multithreaded,
+        error_queue,
         **kwargs,
     ):
         super(IMDProducerV3, self).__init__(
@@ -522,6 +604,7 @@ class IMDProducerV3(BaseIMDProducer):
             sinfo,
             n_atoms,
             multithreaded,
+            error_queue,
             **kwargs,
         )
         # The body of an x/v/f packet should contain
@@ -638,6 +721,21 @@ class IMDFrameBuffer:
     """
     Acts as interface between producer (IMDProducer) and consumer (IMDClient) threads
     when IMDClient runs in multithreaded mode
+
+    Parameters
+    ----------
+    imdsinfo : IMDSessionInfo
+        Information about the IMD session
+    n_atoms : int
+        Number of atoms in the simulation
+    buffer_size : int, optional
+        Size of the buffer in bytes [``10MB``]
+    pause_empty_proportion : float, optional
+        Lower threshold proportion of the buffer's IMDFrames that are empty
+        before the simulation is paused [``0.25``]
+    unpause_empty_proportion : float, optional
+        Proportion of the buffer's IMDFrames that must be empty
+        before the simulation is unpaused [``0.5``]
     """
 
     def __init__(
@@ -649,23 +747,6 @@ class IMDFrameBuffer:
         unpause_empty_proportion=0.5,
         **kwargs,
     ):
-        """
-        Parameters
-        ----------
-        imdsinfo : IMDSessionInfo
-            Information about the IMD session
-        n_atoms : int
-            Number of atoms in the simulation
-        buffer_size : int, optional
-            Size of the buffer in bytes [``10MB``]
-        pause_empty_proportion : float, optional
-            Lower threshold proportion of the buffer's IMDFrames that are empty
-            before the simulation is paused [``0.25``]
-        unpause_empty_proportion : float, optional
-            Proportion of the buffer's IMDFrames that must be empty
-            before the simulation is unpaused [``0.5``]
-        """
-
         # Syncing reader and producer
         self._producer_finished = False
         self._consumer_finished = False
@@ -681,9 +762,7 @@ class IMDFrameBuffer:
             raise ValueError("pause_empty_proportion must be between 0 and 1")
         self._pause_empty_proportion = pause_empty_proportion
         if unpause_empty_proportion < 0 or unpause_empty_proportion > 1:
-            raise ValueError(
-                "unpause_empty_proportion must be between 0 and 1"
-            )
+            raise ValueError("unpause_empty_proportion must be between 0 and 1")
         self._unpause_empty_proportion = unpause_empty_proportion
 
         if buffer_size <= 0:
@@ -750,9 +829,7 @@ class IMDFrameBuffer:
                 logger.debug("IMDProducer: Noticing consumer finished")
                 raise EOFError
         except Exception as e:
-            logger.debug(
-                f"IMDProducer: Error waiting for space in buffer: {e}"
-            )
+            logger.debug(f"IMDProducer: Error waiting for space in buffer: {e}")
 
     def pop_empty_imdframe(self):
         logger.debug("IMDProducer: Getting empty frame")
@@ -798,9 +875,7 @@ class IMDFrameBuffer:
             imdf = self._full_q.get()
         else:
             with self._full_imdf_avail:
-                while (
-                    self._full_q.qsize() == 0 and not self._producer_finished
-                ):
+                while self._full_q.qsize() == 0 and not self._producer_finished:
                     self._full_imdf_avail.wait()
 
             if self._producer_finished and self._full_q.qsize() == 0:

imdclient-0.1.3/imdclient/data/gromacs/md/gromacs_v3_nst8.mdp

@@ -0,0 +1,58 @@
+title                   = PRODUCTION IN NPT
+ld-seed                 = 1
+; Run parameters
+integrator              = md        ; leap-frog integrator
+nsteps                  = 100    ; 1 * 1000 = 1 ps
+dt                      = 0.001     ; 1 fs
+; Output control
+nstxout                 = 8      ; save coordinates every 1 fs
+nstvout                 = 8      ; save velocities every 1 fs
+nstfout                 = 8
+nstenergy               = 8      ; save energies every 1 fs
+nstlog                  = 10      ; update log file every 1 ps
+; Center of mass (COM) motion
+nstcomm                 = 10        ; remove COM motion every 10 steps
+comm-mode               = Linear    ; remove only COM translation (liquids in PBC) 
+; Bond parameters
+continuation            = yes       ; first dynamics run
+constraint_algorithm    = lincs     ; holonomic constraints
+constraints             = all-bonds ; all bonds lengths are constrained
+lincs_iter              = 1         ; accuracy of LINCS
+lincs_order             = 4         ; also related to accuracy
+; Nonbonded settings
+cutoff-scheme           = Verlet    ; Buffered neighbor searching
+ns_type                 = grid      ; search neighboring grid cells
+nstlist                 = 10        ; 10 fs, largely irrelevant with Verlet
+rcoulomb                = 1.0       ; short-range electrostatic cutoff (in nm)
+rvdw                    = 1.0       ; short-range van der Waals cutoff (in nm)
+DispCorr                = EnerPres  ; account for cut-off vdW scheme
+; Electrostatics
+coulombtype             = PME       ; Particle Mesh Ewald for long-range electrostatics
+pme_order               = 4         ; cubic interpolation
+fourierspacing          = 0.12      ; grid spacing for FFT
+; Temperature coupling is on
+tcoupl                  = Nose-Hoover       ; good for production, after equilibration
+; we define separate thermostats for the solute and solvent (need to adapt)
+; see default groups defined by Gromacs for your system or define your own (make_ndx)
+tc-grps                 = Protein  SOL      ; the separate groups for the thermostats
+tau-t                   = 1.0      1.0      ; time constants for thermostats (ps)
+ref-t                   = 300      300      ; reference temperature for thermostats (K)
+; Pressure coupling is off
+pcoupl                  = Parrinello-Rahman ; good for production, after equilibration
+tau-p                   = 2.0               ; time constant for barostat (ps)
+compressibility         = 4.5e-5            ; compressibility (1/bar) set to water at ~300K
+ref-p                   = 1.0               ; reference pressure for barostat (bar)
+; Periodic boundary conditions
+pbc                     = xyz       ; 3-D PBC
+; Velocity generation
+gen_vel                 = no
+IMD-group               = System
+IMD-nst                 = 8
+IMD-version             = 3
+IMD-time                = yes
+IMD-box                 = yes
+IMD-coords              = yes
+IMD-unwrap              = no
+IMD-vels                = yes
+IMD-forces              = yes
+IMD-energies            = no

imdclient-0.1.1/imdclient/data/lammps/md/lammps_v3.in → imdclient-0.1.3/imdclient/data/lammps/md/lammps_v3_nst_1.in

@@ -52,8 +52,8 @@ velocity all create 300 102939 dist gaussian mom yes rot yes
 fix 1 all nve 
 
 # Create source of truth trajectory
-# dump h5md1 all h5md 1 lammps_trj.h5md position velocity force box yes
-# dump_modify h5md1 unwrap no
+dump h5md1 all h5md 1 lammps_trj.h5md position velocity force box yes
+dump_modify h5md1 unwrap no
 
 ## IMD settings
 # https://docs.lammps.org/fix_imd.html
@@ -63,7 +63,7 @@ fix 2 all imd 8888 version 3 unwrap off nowait off
 run 100
 
 # Stop dumping information to the dump file.
-# undump h5md1 
+undump h5md1 
 
 # Unfix the NVE. Additional lines if any will assume that this fix is off.
 unfix 1