From b3e785579aa28791577135372007018a68fb6e31 Mon Sep 17 00:00:00 2001
From: Henry Weller <http://cfd.direct>
Date: Tue, 4 Jan 2022 23:48:36 +0000
Subject: [PATCH] codedFvModelTemplate: Added distribute(const
 mapDistributePolyMesh&) for mesh redistribution

---
 .../dynamicCode/codedFvModelTemplate.C              | 13 +++++++++++--
 .../dynamicCode/codedFvModelTemplate.H              |  3 +++
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/etc/codeTemplates/dynamicCode/codedFvModelTemplate.C b/etc/codeTemplates/dynamicCode/codedFvModelTemplate.C
index 135377778b..7f60b15ab4 100644
--- a/etc/codeTemplates/dynamicCode/codedFvModelTemplate.C
+++ b/etc/codeTemplates/dynamicCode/codedFvModelTemplate.C
@@ -179,9 +179,18 @@ void ${typeName}FvModel${SourceType}::addSup
 }
 
 
-void ${typeName}FvModel${SourceType}::updateMesh(const mapPolyMesh& mpm)
+void ${typeName}FvModel${SourceType}::updateMesh(const mapPolyMesh& map)
 {
-    set_.updateMesh(mpm);
+    set_.updateMesh(map);
+}
+
+
+void ${typeName}FvModel${SourceType}::distribute
+(
+    const mapDistributePolyMesh& map
+)
+{
+    set_.distribute(map);
 }
 
 
diff --git a/etc/codeTemplates/dynamicCode/codedFvModelTemplate.H b/etc/codeTemplates/dynamicCode/codedFvModelTemplate.H
index db75f4fe92..51d8681295 100644
--- a/etc/codeTemplates/dynamicCode/codedFvModelTemplate.H
+++ b/etc/codeTemplates/dynamicCode/codedFvModelTemplate.H
@@ -117,6 +117,9 @@ public:
             //- Update for mesh changes
             virtual void updateMesh(const mapPolyMesh&);
 
+            //- Update mesh corresponding to the given distribution map
+            virtual void distribute(const mapDistributePolyMesh&);
+
             //- Update for mesh motion
             virtual bool movePoints();
 };