From cfcb97a2ee48e7830ce29a6ae078cb6ce0b69d43 Mon Sep 17 00:00:00 2001
From: HoWil <HoWilgh@users.noreply.github.com>
Date: Mon, 25 May 2020 15:46:02 +0200
Subject: [PATCH] FEM electric force equation object, implement in elmer writer

---
 src/Mod/Fem/femsolver/elmer/writer.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/src/Mod/Fem/femsolver/elmer/writer.py b/src/Mod/Fem/femsolver/elmer/writer.py
index 75c1ec10e9..4254d09659 100644
--- a/src/Mod/Fem/femsolver/elmer/writer.py
+++ b/src/Mod/Fem/femsolver/elmer/writer.py
@@ -115,6 +115,7 @@ class Writer(object):
         self._handleElasticity()
         self._handleElectrostatic()
         self._handleFluxsolver()
+        self._handleElectricforce()
         self._handleFlow()
         self._addOutputSolver()
 
@@ -371,6 +372,8 @@ class Writer(object):
                         self._boundary(name, "Potential Constant", True)
                     if obj.ElectricInfinity:
                         self._boundary(name, "Electric Infinity BC", True)
+                    if obj.ElectricForcecalculation:
+                        self._boundary(name, "Calculate Electric Force", True)
                     if obj.CapacitanceBodyEnabled:
                         if hasattr(obj, "CapacitanceBody"):
                             self._boundary(name, "Capacitance Body", obj.CapacitanceBody)
@@ -397,6 +400,24 @@ class Writer(object):
         s["Calculate Grad"] = equation.CalculateGrad
         return s
 
+    def _handleElectricforce(self):
+        activeIn = []
+        for equation in self.solver.Group:
+            if femutils.is_of_type(equation, "Fem::EquationElectricforce"):
+                if equation.References:
+                    activeIn = equation.References[0][1]
+                else:
+                    activeIn = self._getAllBodies()
+                solverSection = self._getElectricforceSolver(equation)
+                for body in activeIn:
+                    self._addSolver(body, solverSection)
+
+    def _getElectricforceSolver(self, equation):
+        s = self._createEmptySolver(equation)
+        s["Equation"] = "Electric Force"  # equation.Name
+        s["Procedure"] = sifio.FileAttr("ElectricForce/StatElecForce")
+        return s
+
     def _handleElasticity(self):
         activeIn = []
         for equation in self.solver.Group:
@@ -679,6 +700,10 @@ class Writer(object):
         for b in bodies:
             self._equation(b, "Convection", "Computed")
 
+    def _createEmptySolver(self, equation):
+        s = sifio.createSection(sifio.SOLVER)
+        return s
+
     def _createLinearSolver(self, equation):
         s = sifio.createSection(sifio.SOLVER)
         s.priority = equation.Priority