Add spanwise shape variables

pygeo/DVConstraints.py

@@ -4250,7 +4250,10 @@ def evalFunctions(self, funcs):
                 cons.extend(self.jac[key].dot(self.DVGeo.DV_listLocal[key].value))
             elif key in self.DVGeo.DV_listSectionLocal:
                 cons.extend(self.jac[key].dot(self.DVGeo.DV_listSectionLocal[key].value))
-
+            elif key in self.DVGeo.DV_listSpanwiseLocal:
+                cons.extend(self.jac[key].dot(self.DVGeo.DV_listSpanwiseLocal[key].value))
+            else:
+                raise Error(f"con {self.name} diffined wrt {key}, but {key} not found in DVGeo")
         funcs[self.name] = numpy.array(cons).real.astype('d')
 
     def evalFunctionsSens(self, funcsSens):
@@ -4326,6 +4329,27 @@ def _finalize(self):
                 self.ncon += len(cons)
                 self.vizConIndices[key] = cons
 
+        # Section local shape variables
+        for key in self.DVGeo.DV_listSpanwiseLocal:
+             if self.config is None or self.config in self.DVGeo.DV_listSpanwiseLocal[key].config:
+
+                # end for (indSet loop)
+                cons = self.DVGeo.DV_listSpanwiseLocal[key].mapIndexSets(self.indSetA,self.indSetB)
+                ncon = len(cons)
+                if ncon > 0:
+                    # Now form the jacobian:
+                    ndv = self.DVGeo.DV_listSpanwiseLocal[key].nVal
+                    jacobian = numpy.zeros((ncon, ndv))
+                    for i in range(ncon):
+                        jacobian[i, cons[i][0]] = self.factorA[i]
+                        jacobian[i, cons[i][1]] = self.factorB[i]
+                    self.jac[key] = jacobian
+
+                # Add to the number of constraints and store indices which
+                # we need for tecplot visualization
+                self.ncon += len(cons)
+                self.vizConIndices[key] = cons
+
         # with-respect-to are just the keys of the jacobian
         self.wrt = list(self.jac.keys())