Add capability to write volume slices

adflow/pyADflow.py

@@ -18,21 +18,23 @@
 v. 1.0  - Original pyAero Framework Implementation (RP,SM 2008)
 """
 
+import copy
+import hashlib
 # =============================================================================
 # Imports
 # =============================================================================
 import os
+import sys
 import time
-import copy
 import types
+from collections import OrderedDict
+
 import numpy
-import sys
+from baseclasses import AeroProblem, AeroSolver, getPy3SafeString
+from baseclasses.utils import CaseInsensitiveDict, Error
 from mpi4py import MPI
-from baseclasses import AeroSolver, AeroProblem, getPy3SafeString
-from baseclasses.utils import Error, CaseInsensitiveDict
+
 from . import MExt
-import hashlib
-from collections import OrderedDict
 
 
 class ADFLOWWarning(object):
@@ -3063,6 +3065,35 @@
             f.close()
         # end if (root proc )
 
+    def writeFamilySolution(self, familyName, outputDir=None, baseName=None, number=None):
+        if outputDir is None:
+            outputDir = self.getOption("outputDirectory")
+
+        if baseName is None:
+            baseName = self.curAP.name
+
+        if not familyName.lower() in self.families:
+            raise Error(f"Family {familyName} not found in the solver.")
+
+        famList = self._getFamilyList(familyName)
+        numDigits = self.getOption("writeSolutionDigits")
+        famID = self.families[familyName.lower()][0]
+        isBC = True if famID in self._getFamilyList("allSurfaces") else False
+
+        if number is not None:
+            baseName = f"{baseName}_{familyName.lower()}_{number:0{numDigits}d}"
+        else:
+            if self.getOption("numberSolutions"):
+                baseName = f"{baseName}_{familyName.lower()}_{self.curAP.adflowData.callCounter:0{numDigits}d}"
+
+        fileName = os.path.join(outputDir, baseName)
+        fileName += ".dat"
+
+        if isBC:
+            self.adflow.tecplotio.writebcsurfacesascii(fileName, famList)
+        else:
+            self.adflow.tecplotio.writeuserintsurf(familyName, fileName, famID)
+
     def resetAdjoint(self, obj):
         """
         Reset an adjoint 'obj' that is stored in the current

src/f2py/adflow.pyf

@@ -654,8 +654,41 @@ python module libadflow
            integer(kind=inttype), optional,intent(in),check(len(famlist)>=nfamlist),depend(famlist) :: nfamlist=len(famlist)
          end subroutine writetecplot
 
+         subroutine writebcsurfacesascii(filename,famlist) ! in output/tecplotIO.F90:tecplotio
+            use constants
+            use communication, only: myid,adflow_comm_world,nproc
+            use inputtimespectral, only: ntimeintervalsspectral
+            use inputphysics, only: equationmode
+            use inputiteration, only: printiterations
+            use inputio, only: precisionsurfgrid,precisionsurfsol
+            use outputmod, only: surfsolnames,numberofsurfsolvariables
+            use surfacefamilies, only: bcfamexchange,famnames,familyexchange
+            use utils, only: echk,setpointers,setbcpointers
+            use bcpointers, only: xx
+            use sorting, only: faminlist
+            use extraoutput, only: surfwriteblank
+            use oversetdata, only: zippermesh,zippermeshes
+            use surfaceutils
+            use petsc
+            character*(*) intent(in) :: filename
+            integer(kind=inttype) dimension(:),intent(in) :: famlist
+         end subroutine writebcsurfacesascii
+
          subroutine initializeliftdistributiondata
          end subroutine initializeliftdistributiondata
+
+         subroutine writeuserintsurf(slicename,filename,famid) ! in :test:tecplotIO.F90:tecplotio
+          use constants
+          use inputio
+          use commonformats, only: int5,sci6
+          use usersurfaceintegrationdata, only: userintsurfs,nuserintsurfs,userintsurf
+          use usersurfaceintegrations, only: getusersurfacedata
+          use sorting, only: faminlist
+          character*(*) intent(in) :: slicename
+          character*(*) intent(in) :: filename
+          integer(kind=inttype) intent(in) :: famid
+        end subroutine writeuserintsurf
+
        end subroutine tecplotio
 
        module surfaceintegrations
@@ -681,7 +714,6 @@ python module libadflow
 
          subroutine interpolateintegrationsurfaces
          end subroutine interpolateintegrationsurfaces
-
        end module usersurfaceintegrations
 
        module actuatorregion ! in :test:actuatorDiskRegion.F90

src/modules/commonFormats.F90

@@ -35,4 +35,7 @@ module commonFormats
     ! Integers written with 5 characters
     character(len=maxStringLen) :: int5 = '(*(I5))'
 
+    ! Integers written with 7 characters
+    character(len=maxStringLen) :: int7 = '(*(I7))'
+
 end module commonFormats