Internal parameter storage for IMDModules

[alexxy/gromacs.git] / src / gromacs / mdrun / mdmodules.h

diff --git a/src/gromacs/mdrun/mdmodules.h b/src/gromacs/mdrun/mdmodules.h
index e3c816f8abf9693b95d4b2b2d9039596391d78a3..21ca0ba5a77f53c100c421894a08898dcb75154a 100644 (file)
--- a/src/gromacs/mdrun/mdmodules.h
+++ b/src/gromacs/mdrun/mdmodules.h
@@ -59,6 +59,7 @@ class IKeyValueTreeErrorHandler;
 class IKeyValueTreeTransformRules;
 class IMDOutputProvider;
 class KeyValueTreeObject;
+class KeyValueTreeBuilder;
 class IMDModule;
 class LocalAtomSetManager;
 
@@ -110,6 +111,8 @@ class MDModules
         //! Register callback function types for MDModules
         using notifier_type = registerMdModuleNotification<
                     CommunicationIsSetup,
+                    KeyValueTreeBuilder*,
+                    const KeyValueTreeObject &,
                     LocalAtomSetManager *
                     >::type;