Custom ptsets

baseclasses/solvers/pyAero_solver.py

@@ -54,8 +54,9 @@
         self.families = CaseInsensitiveDict()
         self._updateGeomInfo = False
 
-        # Initialize kwargs for addPointSet
+        # Initialize kwargs for addPointSet and customPointSetFamilies
         self.pointSetKwargs = None
+        self.customPointSetFamilies = None
 
     def setMesh(self, mesh):
         """
@@ -79,7 +80,7 @@
         pts = self.getSurfaceCoordinates(self.meshFamilyGroup)
         self.mesh.setSurfaceDefinition(pts, conn, faceSizes)
 
-    def setDVGeo(self, DVGeo, pointSetKwargs=None):
+    def setDVGeo(self, DVGeo, pointSetKwargs={}, customPointSetFamilies=None):
         """
         Set the DVGeometry object that will manipulate 'geometry' in
         this object. Note that <SOLVER> does not **strictly** need a
@@ -94,6 +95,16 @@
         pointSetKwargs : dict
             Keyword arguments to be passed to the DVGeo addPointSet call.
             Useful for DVGeometryMulti, specifying FFD projection tolerances, etc.
+            These arguments are used for all point sets added by this solver.
+
+        customPointSetFamilies : dict of dicts
+            Keyword arguments to be passed to the DVGeo addPointSet call for each surface family,
+            specified by the keys. The surface families need to be all part of the designSurfaceFamily.
+            Useful for DVGeometryMulti, specifying FFD projection tolerances, etc.
+            If this is provided together with pointSetKwargs, the regular pointSetKwargs
+            will be appended to each component's dictionary. If the same argument
+            is also provided in pointSetKwargs, the value specified in customPointSetFamilies
+            will be used.
 
         Examples
         --------
@@ -103,10 +114,11 @@
 
         self.DVGeo = DVGeo
 
-        if pointSetKwargs is None:
-            self.pointSetKwargs = {}
-        else:
-            self.pointSetKwargs = pointSetKwargs
+        # save the common kwargs dict. default is empty
+        self.pointSetKwargs = pointSetKwargs
+
+        # save if we have customPointSetFamilies
+        self.customPointSetFamilies = customPointSetFamilies
 
     def getTriangulatedMeshSurface(self, groupName=None, **kwargs):
         """