def _pEqn(rho, thermo, UEqn, nNonOrthCorr, psi, U, mesh, phi, p, DpDt,
cumulativeContErr, corr, nCorr, nOuterCorr, transonic):
rho.ext_assign(thermo.rho())
rUA = 1.0 / UEqn.A()
U.ext_assign(rUA * UEqn.H())
from Foam import fvc, fvm
from Foam.OpenFOAM import word
from Foam.finiteVolume import surfaceScalarField
if transonic:
phid = surfaceScalarField(
word("phid"),
fvc.interpolate(psi) * ((fvc.interpolate(U) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, rho, U, phi)))
for nonOrth in range(nNonOrthCorr + 1):
pEqn = fvm.ddt(psi, p) + fvm.div(phid, p) - fvm.laplacian(
rho * rUA, p)
pEqn.solve()
if nonOrth == nNonOrthCorr:
phi == pEqn.flux()
pass
pass
pass
else:
phi.ext_assign(
fvc.interpolate(rho) * ((fvc.interpolate(U) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, rho, U, phi)))
for nonOrth in range(nNonOrthCorr + 1):
pEqn = fvm.ddt(psi, p) + fvc.div(phi) - fvm.laplacian(rho * rUA, p)
pEqn.solve()
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi + pEqn.flux())
pass
pass
pass
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn(rho, phi)
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs(rho, thermo,
cumulativeContErr)
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
DpDt.ext_assign(
fvc.DDt(surfaceScalarField(word("phiU"), phi / fvc.interpolate(rho)),
p))
return cumulativeContErr
def step(self, getPISOControls):
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "Time = " << self.runTime.timeName() << nl << nl
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = getPISOControls(self.mesh)
from Foam.finiteVolume.cfdTools.incompressible import CourantNo
CoNum, meanCoNum = CourantNo(self.mesh, self.phi, self.runTime)
from Foam import fvm
UEqn = fvm.ddt(self.U) + fvm.div(self.phi, self.U) - fvm.laplacian(self.nu, self.U)
from Foam import fvc
from Foam.finiteVolume import solve
solve(UEqn == -fvc.grad(self.p))
# --- PISO loop
for corr in range(nCorr):
rUA = 1.0 / UEqn.A()
self.U.ext_assign(rUA * UEqn.H())
self.phi.ext_assign((fvc.interpolate(self.U) & self.mesh.Sf()) + fvc.ddtPhiCorr(rUA, self.U, self.phi))
from Foam.finiteVolume import adjustPhi
adjustPhi(self.phi, self.U, self.p)
for nonOrth in range(nNonOrthCorr + 1):
pEqn = fvm.laplacian(rUA, self.p) == fvc.div(self.phi)
pEqn.setReference(self.pRefCell, self.pRefValue)
pEqn.solve()
if nonOrth == nNonOrthCorr:
self.phi.ext_assign(self.phi - pEqn.flux())
pass
pass
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs(self.mesh, self.phi, self.runTime, self.cumulativeContErr)
self.U.ext_assign(self.U - rUA * fvc.grad(self.p))
self.U.correctBoundaryConditions()
pass
self.runTime.write()
ext_Info() << "ExecutionTime = " << self.runTime.elapsedCpuTime() << " s" << " ClockTime = " << self.runTime.elapsedClockTime() << " s" << nl << nl
self.runTime += self.runTime.deltaT()
return self.runTime.value()
def _pEqn(rho, thermo, UEqn, nNonOrthCorr, psi, U, mesh, phi, p,
cumulativeContErr):
from Foam.finiteVolume import volScalarField
rUA = 1.0 / UEqn.A()
U.ext_assign(rUA * UEqn.H())
from Foam import fvc
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
phid = surfaceScalarField(
word("phid"),
fvc.interpolate(thermo.psi()) *
((fvc.interpolate(U) & mesh.Sf()) + fvc.ddtPhiCorr(rUA, rho, U, phi)))
for nonOrth in range(nNonOrthCorr + 1):
from Foam import fvm
pEqn = (fvm.ddt(psi, p) + fvm.div(phid, word("div(phid,p)")) -
fvm.laplacian(rho * rUA, p))
pEqn.solve()
if (nonOrth == nNonOrthCorr):
phi.ext_assign(pEqn.flux())
pass
pass
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn(rho, phi)
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs(rho, thermo,
cumulativeContErr)
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
return cumulativeContErr
def fun_pEqn( i, mesh, p, rho, turb, thermo, thermoFluid, K, UEqn, U, phi, psi, DpDt, initialMass, p_rgh, gh, ghf, \
nNonOrthCorr, oCorr, nOuterCorr, corr, nCorr, cumulativeContErr ) :
closedVolume = p_rgh.needReference()
rho.ext_assign( thermo.rho() )
rUA = 1.0 / UEqn.A()
from Foam import fvc
from Foam.OpenFOAM import word
from Foam.finiteVolume import surfaceScalarField
rhorUAf = surfaceScalarField( word( "(rho*(1|A(U)))" ) , fvc.interpolate( rho * rUA ) )
U.ext_assign( rUA * UEqn.H() )
from Foam import fvc
phiU = ( fvc.interpolate( rho ) *
( ( fvc.interpolate( U ) & mesh.Sf() ) +
fvc.ddtPhiCorr( rUA, rho, U, phi ) ) )
phi.ext_assign( phiU - rhorUAf * ghf * fvc.snGrad( rho ) * mesh.magSf() )
from Foam import fvm
for nonOrth in range ( nNonOrthCorr + 1 ):
p_rghEqn = ( fvm.ddt( psi, p_rgh) + fvc.ddt( psi, rho ) * gh + fvc.div( phi ) - fvm.laplacian( rhorUAf, p_rgh ) )
p_rghEqn.solve( mesh.solver( p_rgh.select( ( oCorr == nOuterCorr-1 and corr == ( nCorr-1 ) and nonOrth == nNonOrthCorr ) ) ) )
if nonOrth == nNonOrthCorr :
phi.ext_assign( phi + p_rghEqn.flux() )
pass
pass
# Correct velocity field
U.ext_assign( U + rUA * fvc.reconstruct( ( phi - phiU ) / rhorUAf ) )
U.correctBoundaryConditions()
p.ext_assign( p_rgh + rho * gh )
#Update pressure substantive derivative
DpDt.ext_assign( fvc.DDt( surfaceScalarField( word( "phiU" ), phi / fvc.interpolate( rho ) ), p ) )
# Solve continuity
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn( rho, phi )
# Update continuity errors
cumulativeContErr = compressibleContinuityErrors( i, mesh, rho, thermo, cumulativeContErr )
# For closed-volume cases adjust the pressure and density levels
# to obey overall mass continuity
if closedVolume :
p.ext_assign( p + ( initialMass - fvc.domainIntegrate( psi * p ) ) / fvc.domainIntegrate( psi ) )
rho.ext_assign( thermo.rho() )
p_rgh.ext_assign( p - rho * gh )
pass
#Update thermal conductivity
K.ext_assign( thermoFluid[ i ].Cp() * turb.alphaEff() )
return cumulativeContErr
def _pEqn( runTime, mesh, UEqn, thermo, p, psi, U, rho, phi, DpDt, g, initialMass, totalVolume, corr, nCorr, nNonOrthCorr, cumulativeContErr ):
closedVolume = p.needReference()
rho.ext_assign( thermo.rho() )
# Thermodynamic density needs to be updated by psi*d(p) after the
# pressure solution - done in 2 parts. Part 1:
thermo.rho().ext_assign( thermo.rho() - psi * p )
rUA = 1.0/UEqn.A()
from Foam.OpenFOAM import word
from Foam.finiteVolume import surfaceScalarField
from Foam import fvc
rhorUAf = surfaceScalarField( word( "(rho*(1|A(U)))" ), fvc.interpolate( rho * rUA ) )
U.ext_assign( rUA * UEqn.H() )
phiU = fvc.interpolate( rho ) * ( (fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, rho, U, phi ) )
phi.ext_assign( phiU + rhorUAf * fvc.interpolate( rho ) * (g & mesh.Sf() ) )
for nonOrth in range( nNonOrthCorr+1 ):
from Foam import fvm
from Foam.finiteVolume import correction
pEqn = fvc.ddt( rho ) + psi * correction( fvm.ddt( p ) ) + fvc.div( phi ) - fvm.laplacian( rhorUAf, p )
if corr == nCorr-1 and nonOrth == nNonOrthCorr:
pEqn.solve( mesh.solver( word( str( p.name() ) + "Final" ) ) )
pass
else:
pEqn.solve( mesh.solver( p.name() ) )
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign( phi + pEqn.flux() )
pass
# Second part of thermodynamic density update
thermo.rho().ext_assign( thermo.rho() + psi * p )
U.ext_assign( U + rUA * fvc.reconstruct( ( phi - phiU ) / rhorUAf ) )
U.correctBoundaryConditions()
DpDt.ext_assign( fvc.DDt( surfaceScalarField( word( "phiU" ), phi / fvc.interpolate( rho ) ), p ) )
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn( rho, phi )
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs( rho, thermo, cumulativeContErr )
# For closed-volume cases adjust the pressure and density levels
# to obey overall mass continuity
if closedVolume:
p.ext_assign( p + ( initialMass - fvc.domainIntegrate( psi * p ) ) / fvc.domainIntegrate( psi ) )
thermo.rho().ext_assign( psi * p )
rho.ext_assign( rho + ( initialMass - fvc.domainIntegrate( rho ) ) / totalVolume )
pass
return cumulativeContErr
def fun_pEqn( runTime, mesh, UEqn, p, p_rgh, phi, U, rho, rho1, rho2, rho10, rho20, gh, ghf, dgdt, pMin, \
psi1, psi2, alpha1, alpha2, interface, transonic, oCorr, nOuterCorr, corr, nCorr, nNonOrthCorr ):
rUA = 1.0/UEqn.A()
from Foam import fvc
rUAf = fvc.interpolate( rUA )
p_rghEqnComp = None
from Foam import fvm
if transonic:
p_rghEqnComp = fvm.ddt( p_rgh ) + fvm.div( phi, p_rgh ) - fvm.Sp( fvc.div( phi ), p_rgh )
pass
else:
p_rghEqnComp = fvm.ddt( p_rgh ) + fvc.div( phi, p_rgh ) - fvc.Sp( fvc.div( phi ), p_rgh )
pass
U.ext_assign( rUA * UEqn.H() )
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
phiU = surfaceScalarField( word( "phiU" ),
( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, rho, U, phi ) )
phi.ext_assign(phiU + ( fvc.interpolate( interface.sigmaK() ) * fvc.snGrad( alpha1 ) - ghf * fvc.snGrad( rho ) ) * rUAf * mesh.magSf() )
from Foam.finiteVolume import solve
from Foam.OpenFOAM import scalar
for nonOrth in range( nNonOrthCorr +1 ):
p_rghEqnIncomp = fvc.div( phi ) - fvm.laplacian( rUAf, p_rgh )
solve( ( alpha1.ext_max( scalar( 0 ) ) * ( psi1 / rho1 ) + alpha2.ext_max( scalar( 0 ) ) * ( psi2 / rho2 ) ) *p_rghEqnComp() + p_rghEqnIncomp,
mesh.solver( p_rgh.select( oCorr == ( nOuterCorr - 1 ) and corr == ( nCorr-1 ) and nonOrth == nNonOrthCorr ) ) )
if nonOrth == nNonOrthCorr:
dgdt.ext_assign( ( alpha2.pos() * ( psi2 / rho2 ) - alpha1.pos() * ( psi1 / rho1 ) ) * ( p_rghEqnComp & p_rgh ) )
phi.ext_assign( phi + p_rghEqnIncomp.flux() )
pass
U.ext_assign( U + rUA * fvc.reconstruct( ( phi - phiU ) / rUAf ) )
U.correctBoundaryConditions()
p.ext_assign( ( ( p_rgh + gh * ( alpha1 * rho10 + alpha2 * rho20 ) ) /( 1.0 - gh * ( alpha1 * psi1 + alpha2 * psi2 ) ) ).ext_max( pMin ) )
rho1.ext_assign( rho10 + psi1 * p )
rho2.ext_assign( rho20 + psi2 * p )
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "max(U) " << U.mag().ext_max().value() << nl
ext_Info() << "min(p_rgh) " << p_rgh.ext_min().value() << nl
pass
def _pEqn(mesh, UEqn, U, p, phi, alpha1, rho, g, interface, corr, nCorr, nNonOrthCorr, pRefCell, pRefValue):
rUA = 1.0 / UEqn.A()
from Foam import fvc
rUAf = fvc.interpolate(rUA)
U.ext_assign(rUA * UEqn.H())
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
phiU = surfaceScalarField(word("phiU"), (fvc.interpolate(U) & mesh.Sf()) + fvc.ddtPhiCorr(rUA, rho, U, phi))
from Foam.finiteVolume import adjustPhi
adjustPhi(phiU, U, p)
phi.ext_assign(
phiU
+ (
fvc.interpolate(interface.sigmaK()) * fvc.snGrad(alpha1) * mesh.magSf()
+ fvc.interpolate(rho) * (g & mesh.Sf())
)
* rUAf
)
from Foam import fvm
for nonOrth in range(nNonOrthCorr + 1):
pEqn = fvm.laplacian(rUAf, p) == fvc.div(phi)
pEqn.setReference(pRefCell, pRefValue)
if corr == nCorr - 1 and nonOrth == nNonOrthCorr:
pEqn.solve(mesh.solver(word(str(p.name()) + "Final")))
pass
else:
pEqn.solve(mesh.solver(p.name()))
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi - pEqn.flux())
pass
pass
U.ext_assign(U + rUA * fvc.reconstruct((phi - phiU) / rUAf))
U.correctBoundaryConditions()
pass
def fun_pEqn( i, fluidRegions, Uf, pdf, rhof, thermof, phif, ghf, Kf, DpDtf, turb, initialMassf, UEqn, pRef, corr, nCorr, nNonOrthCorr, cumulativeContErr ) :
closedVolume = False
rhof[ i ].ext_assign( thermof[ i ].rho() )
rUA = 1.0 / UEqn.A()
Uf[ i ].ext_assign( rUA * UEqn.H() )
from Foam import fvc
phif[ i ] .ext_assign( fvc.interpolate( rhof[ i ] ) * ( ( fvc.interpolate( Uf[ i ] ) & fluidRegions[ i ].Sf() ) +
fvc.ddtPhiCorr( rUA, rhof[ i ], Uf[ i ], phif[ i ] ) ) -
fvc.interpolate( rhof[ i ] * rUA * ghf[ i ] ) * fvc.snGrad( rhof[ i ] ) * fluidRegions[ i ].magSf() )
# Solve pressure difference
pdEqn, closedVolume = fun_pdEqn( corr, nCorr, nNonOrthCorr, closedVolume, pdf[i], pRef, rhof[i], thermof[i].psi(), rUA, ghf[i], phif[i] )
# Solve continuity
rhoEqn( rhof[i], phif[i] )
# Update pressure field (including bc)
from Foam.OpenFOAM import word
thermof[i].p() == pdf[ i ] + rhof[ i ] * ghf[ i ] + pRef
from Foam.finiteVolume import surfaceScalarField
DpDtf[i].ext_assign( fvc.DDt( surfaceScalarField( word( "phiU" ), phif[ i ] / fvc.interpolate( rhof[ i ] ) ), thermof[i].p() ) )
# Update continuity errors
cumulativeContErr = compressibleContinuityErrors( cumulativeContErr, rhof[i], thermof[i] )
# Correct velocity field
Uf[ i ].ext_assign( Uf[i] - rUA * ( fvc.grad( pdf[ i ] ) + fvc.grad( rhof[ i ] ) * ghf[ i ] ) )
Uf[ i ].correctBoundaryConditions()
# For closed-volume cases adjust the pressure and density levels
# to obey overall mass continuity
if (closedVolume):
from Foam.OpenFOAM import dimensionedScalar, dimMass
thermof[i].p().ext_assign( thermof[i].p() +
( dimensionedScalar( word( "massIni" ),
dimMass,
initialMassf[ i ] ) - fvc.domainIntegrate( thermof[ i ].psi() * thermof[ i ].p() ) )
/ fvc.domainIntegrate( thermof[ i ].psi() ) )
rhof[ i ].ext_assign( thermof[ i ].rho() )
# Update thermal conductivity
Kf[i].ext_assign( rhof[ i ] * thermof[ i ].Cp() * turb[ i ].alphaEff() )
return cumulativeContErr
def pEqn(runTime, mesh, U, rUA, UEqn, phi, p, nCorr, nOuterCorr, nNonOrthCorr,
oCorr, corr, pRefCell, pRefValue, cumulativeContErr):
U.ext_assign(rUA * UEqn.H())
if (nCorr <= 1):
UEqn.clear()
pass
from Foam import fvc
phi.ext_assign((fvc.interpolate(U) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, U, phi))
from Foam.finiteVolume import adjustPhi
adjustPhi(phi, U, p)
# Non-orthogonal pressure corrector loop
for nonOrth in range(nNonOrthCorr + 1):
#Pressure corrector
from Foam import fvm
pEqn = fvm.laplacian(rUA, p) == fvc.div(phi)
pEqn.setReference(pRefCell, pRefValue)
if (oCorr == nOuterCorr - 1) and (corr == nCorr -
1) and (nonOrth == nNonOrthCorr):
from Foam.OpenFOAM import word
pEqn.solve(mesh.solver(word("pFinal")))
pass
else:
pEqn.solve()
pass
if (nonOrth == nNonOrthCorr):
phi.ext_assign(phi - pEqn.flux())
pass
pass
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs(mesh, phi, runTime, cumulativeContErr)
# Explicitly relax pressure for momentum corrector except for last corrector
if (oCorr != nOuterCorr - 1):
p.relax()
pass
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
return cumulativeContErr
def _pEqn( runTime, mesh, UEqn, U, p, p_rgh, gh, ghf, phi, alpha1, rho, g, interface, corr, nCorr, nNonOrthCorr, pRefCell, pRefValue, cumulativeContErr ):
rAU = 1.0/UEqn.A()
from Foam import fvc
rAUf = fvc.interpolate( rAU )
U.ext_assign( rAU * UEqn.H() )
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
phiU = surfaceScalarField( word( "phiU" ),
( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rAU, rho, U, phi ) )
from Foam.finiteVolume import adjustPhi
adjustPhi(phiU, U, p)
phi.ext_assign( phiU + ( fvc.interpolate( interface.sigmaK() ) * fvc.snGrad( alpha1 ) - ghf * fvc.snGrad( rho ) )*rAUf*mesh.magSf() )
from Foam import fvm
for nonOrth in range( nNonOrthCorr + 1 ):
p_rghEqn = fvm.laplacian( rAUf, p_rgh ) == fvc.div( phi )
p_rghEqn.setReference( pRefCell, pRefValue )
p_rghEqn.solve( mesh.solver( p_rgh.select(corr == nCorr-1 and nonOrth == nNonOrthCorr) ) )
if nonOrth == nNonOrthCorr:
phi.ext_assign( phi - p_rghEqn.flux() )
pass
pass
U.ext_assign( U + rAU * fvc.reconstruct( ( phi - phiU ) / rAUf ) )
U.correctBoundaryConditions()
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs( mesh, phi, runTime, cumulativeContErr )
p == p_rgh + rho * gh
if p_rgh.needReference():
from Foam.OpenFOAM import pRefValue
p.ext_assign( p + dimensionedScalar( word( "p" ),
p.dimensions(),
pRefValue - getRefCellValue(p, pRefCell) ) )
p_rgh.ext_assign( p - rho * gh )
pass
pass
def pEqn( runTime, mesh, U, rUA, UEqn, phi, p, nCorr, nOuterCorr, nNonOrthCorr, oCorr, corr, pRefCell, pRefValue, cumulativeContErr ):
U.ext_assign( rUA * UEqn.H() )
if ( nCorr <= 1 ):
UEqn.clear()
pass
from Foam import fvc
phi.ext_assign( ( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, U, phi ) )
from Foam.finiteVolume import adjustPhi
adjustPhi( phi, U, p )
# Non-orthogonal pressure corrector loop
for nonOrth in range( nNonOrthCorr + 1):
#Pressure corrector
from Foam import fvm
pEqn = fvm.laplacian( rUA, p ) == fvc.div( phi )
pEqn.setReference( pRefCell, pRefValue )
if ( oCorr == nOuterCorr-1 ) and ( corr == nCorr-1 ) and ( nonOrth == nNonOrthCorr ) :
from Foam.OpenFOAM import word
pEqn.solve( mesh.solver( word( "pFinal" ) ) )
pass
else:
pEqn.solve()
pass
if ( nonOrth == nNonOrthCorr ) :
phi.ext_assign( phi - pEqn.flux() )
pass
pass
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs( mesh, phi, runTime, cumulativeContErr )
# Explicitly relax pressure for momentum corrector except for last corrector
if ( oCorr != nOuterCorr-1 ):
p.relax()
pass
U.ext_assign( U - rUA * fvc.grad( p ) )
U.correctBoundaryConditions()
return cumulativeContErr
def _pEqn(mesh, UEqn, U, p, pd, phi, alpha1, rho, ghf, interface, corr, nCorr,
nNonOrthCorr, pdRefCell, pdRefValue):
rUA = 1.0 / UEqn.A()
from Foam import fvc
rUAf = fvc.interpolate(rUA)
U.ext_assign(rUA * UEqn.H())
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
phiU = surfaceScalarField(word("phiU"), (fvc.interpolate(U) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, rho, U, phi))
from Foam.finiteVolume import adjustPhi
adjustPhi(phiU, U, p)
phi.ext_assign(phiU +
(fvc.interpolate(interface.sigmaK()) * fvc.snGrad(alpha1) -
ghf * fvc.snGrad(rho)) * rUAf * mesh.magSf())
from Foam import fvm
for nonOrth in range(nNonOrthCorr + 1):
pdEqn = fvm.laplacian(rUAf, pd) == fvc.div(phi)
pdEqn.setReference(pdRefCell, pdRefValue)
if corr == nCorr - 1 and nonOrth == nNonOrthCorr:
pdEqn.solve(mesh.solver(word(str(pd.name()) + "Final")))
pass
else:
pdEqn.solve(mesh.solver(pd.name()))
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi - pdEqn.flux())
pass
pass
U.ext_assign(U + rUA * fvc.reconstruct((phi - phiU) / rUAf))
U.correctBoundaryConditions()
pass
def _pEqn(runTime, mesh, U, UEqn, phi, p, rhok, g, corr, nCorr, nNonOrthCorr,
cumulativeContErr):
from Foam.finiteVolume import volScalarField, surfaceScalarField
from Foam.OpenFOAM import word
from Foam import fvc
rUA = volScalarField(word("rUA"), 1.0 / UEqn.A())
rUAf = surfaceScalarField(word("(1|A(U))"), fvc.interpolate(rUA))
U.ext_assign(rUA * UEqn.H())
phiU = (fvc.interpolate(U) & mesh.Sf()) + fvc.ddtPhiCorr(rUA, U, phi)
phi.ext_assign(phiU + rUAf * fvc.interpolate(rhok) * (g & mesh.Sf()))
for nonOrth in range(nNonOrthCorr + 1):
from Foam import fvm
pEqn = fvm.laplacian(rUAf, p) == fvc.div(phi)
if (corr == nCorr - 1) and (nonOrth == nNonOrthCorr):
from Foam.OpenFOAM import word
pEqn.solve(mesh.solver(word(str(p.name()) + "Final")))
pass
else:
pEqn.solve(mesh.solver(p.name()))
pass
if (nonOrth == nNonOrthCorr):
phi.ext_assign(phi - pEqn.flux())
pass
pass
U.ext_assign(U + rUA * fvc.reconstruct((phi - phiU) / rUAf))
U.correctBoundaryConditions()
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs(mesh, phi, runTime, cumulativeContErr)
return pEqn
def _pEqn( runTime, mesh, U, UEqn, phi, p, rhok, g, corr, nCorr, nNonOrthCorr, cumulativeContErr ):
from Foam.finiteVolume import volScalarField, surfaceScalarField
from Foam.OpenFOAM import word
from Foam import fvc
rUA = volScalarField( word( "rUA" ), 1.0 / UEqn.A() )
rUAf = surfaceScalarField(word( "(1|A(U))" ), fvc.interpolate( rUA ) )
U.ext_assign( rUA * UEqn.H() )
phiU = ( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, U, phi )
phi.ext_assign( phiU + rUAf * fvc.interpolate( rhok ) * ( g & mesh.Sf() ) )
for nonOrth in range( nNonOrthCorr+1 ):
from Foam import fvm
pEqn = fvm.laplacian( rUAf, p ) == fvc.div( phi )
if ( corr == nCorr-1 ) and (nonOrth == nNonOrthCorr):
from Foam.OpenFOAM import word
pEqn.solve(mesh.solver( word( str( p.name() ) + "Final" ) ) )
pass
else:
pEqn.solve( mesh.solver( p.name() ) )
pass
if (nonOrth == nNonOrthCorr):
phi.ext_assign( phi - pEqn.flux() )
pass
pass
U.ext_assign( U + rUA * fvc.reconstruct( ( phi - phiU ) / rUAf ) )
U.correctBoundaryConditions()
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs( mesh, phi, runTime, cumulativeContErr )
return pEqn
def main_standalone( argc, argv ):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase( argc, argv )
from Foam.OpenFOAM.include import createTime
runTime = createTime( args )
from Foam.OpenFOAM.include import createMesh
mesh = createMesh( runTime )
thermodynamicProperties, R, Cv, Cp, gamma, Pr = readThermodynamicProperties( runTime, mesh )
p, T, psi, pbf, rhoBoundaryTypes, rho, U, Ubf, rhoUboundaryTypes, \
rhoU, Tbf, rhoEboundaryTypes, rhoE, phi, phiv, rhoU, fields, magRhoU, H = _createFields( runTime, mesh, R, Cv )
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
while runTime.loop():
ext_Info() << "Time = " << runTime.value() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls( mesh )
from Foam.OpenFOAM import readScalar, word
HbyAblend = readScalar( piso.lookup( word( "HbyAblend" ) ) )
from Foam.finiteVolume.cfdTools.general.include import readTimeControls
adjustTimeStep, maxCo, maxDeltaT = readTimeControls( runTime )
CoNum = ( mesh.deltaCoeffs() * phiv.mag() / mesh.magSf() ).ext_max().value() * runTime.deltaT().value()
ext_Info() << "Max Courant Number = " << CoNum << nl
from Foam.finiteVolume.cfdTools.general.include import setDeltaT
runTime = setDeltaT( runTime, adjustTimeStep, maxCo, maxDeltaT, CoNum )
for outerCorr in range( nOuterCorr):
magRhoU.ext_assign( rhoU.mag() )
H.ext_assign( ( rhoE + p ) / rho )
from Foam.fv import multivariateGaussConvectionScheme_scalar
mvConvection = multivariateGaussConvectionScheme_scalar( mesh, fields, phiv, mesh.divScheme( word( "div(phiv,rhoUH)" ) ) )
from Foam.finiteVolume import solve
from Foam import fvm
solve( fvm.ddt( rho ) + mvConvection.fvmDiv( phiv, rho ) )
tmp = mvConvection.interpolationScheme()()( magRhoU )
rhoUWeights = tmp.ext_weights( magRhoU )
from Foam.finiteVolume import weighted_vector
rhoUScheme = weighted_vector(rhoUWeights)
from Foam import fv, fvc
rhoUEqn = fvm.ddt(rhoU) + fv.gaussConvectionScheme_vector( mesh, phiv, rhoUScheme ).fvmDiv( phiv, rhoU )
solve( rhoUEqn == -fvc.grad( p ) )
solve( fvm.ddt( rhoE ) + mvConvection.fvmDiv( phiv, rhoE ) == - mvConvection.fvcDiv( phiv, p ) )
T.ext_assign( (rhoE - 0.5 * rho * ( rhoU / rho ).magSqr() ) / Cv / rho )
psi.ext_assign( 1.0 / ( R * T ) )
p.ext_assign( rho / psi )
for corr in range( nCorr ):
rrhoUA = 1.0 / rhoUEqn.A()
from Foam.finiteVolume import surfaceScalarField
rrhoUAf = surfaceScalarField( word( "rrhoUAf" ), fvc.interpolate( rrhoUA ) )
HbyA = rrhoUA * rhoUEqn.H()
from Foam.finiteVolume import LimitedScheme_vector_MUSCLLimiter_NVDTVD_limitFuncs_magSqr
from Foam.OpenFOAM import IStringStream, word
HbyAWeights = HbyAblend * mesh.weights() + ( 1.0 - HbyAblend ) * \
LimitedScheme_vector_MUSCLLimiter_NVDTVD_limitFuncs_magSqr( mesh, phi, IStringStream( "HbyA" )() ).weights( HbyA )
from Foam.finiteVolume import surfaceInterpolationScheme_vector
phi.ext_assign( ( surfaceInterpolationScheme_vector.ext_interpolate(HbyA, HbyAWeights) & mesh.Sf() ) \
+ HbyAblend * fvc.ddtPhiCorr( rrhoUA, rho, rhoU, phi ) )
p.ext_boundaryField().updateCoeffs()
phiGradp = rrhoUAf * mesh.magSf() * fvc.snGrad( p )
phi.ext_assign( phi - phiGradp )
resetPhiPatches( phi, rhoU, mesh )
rhof = mvConvection.interpolationScheme()()(rho).interpolate(rho)
phiv.ext_assign( phi/rhof )
pEqn = fvm.ddt( psi, p ) + mvConvection.fvcDiv( phiv, rho ) + fvc.div( phiGradp ) - fvm.laplacian( rrhoUAf, p )
pEqn.solve()
phi.ext_assign( phi + phiGradp + pEqn.flux() )
rho.ext_assign( psi * p )
rhof.ext_assign( mvConvection.interpolationScheme()()( rho ).interpolate(rho) )
phiv.ext_assign( phi / rhof )
rhoU.ext_assign( HbyA - rrhoUA * fvc.grad(p) )
rhoU.correctBoundaryConditions()
pass
pass
U.ext_assign( rhoU / rho )
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n"
import os
return os.EX_OK
def main_standalone( argc, argv ):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase( argc, argv )
from Foam.OpenFOAM.include import createTime
runTime = createTime( args )
from Foam.OpenFOAM.include import createMeshNoClear
mesh = createMeshNoClear( runTime )
p, U, phi, fluid, pRefCell, pRefValue = _createFields( runTime, mesh )
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
while runTime.loop() :
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls( mesh )
from Foam.finiteVolume.cfdTools.incompressible import CourantNo
CoNum, meanCoNum, velMag = CourantNo( mesh, phi, runTime )
fluid.correct()
from Foam import fvm, fvc
UEqn = fvm.ddt( U ) + fvm.div( phi, U ) - fvm.laplacian( fluid.ext_nu(), U )
from Foam.finiteVolume import solve
solve( UEqn == -fvc.grad( p ) )
# --- PISO loop
for corr in range( nCorr ):
rUA = 1.0 / UEqn.A()
U.ext_assign( rUA * UEqn.H() )
phi.ext_assign( ( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, U, phi ) )
from Foam.finiteVolume import adjustPhi
adjustPhi(phi, U, p)
for nonOrth in range( nNonOrthCorr + 1):
pEqn = ( fvm.laplacian( rUA, p ) == fvc.div( phi ) )
pEqn.setReference( pRefCell, pRefValue )
pEqn.solve()
if nonOrth == nNonOrthCorr:
phi.ext_assign( phi - pEqn.flux() )
pass
pass
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs( mesh, phi, runTime, cumulativeContErr )
U.ext_assign( U - rUA * fvc.grad( p ) )
U.correctBoundaryConditions()
pass
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n" << nl
import os
return os.EX_OK
def fun_pEqn(mesh, p, rho, psi, p_rgh, U, phi, ghf, gh, DpDt, UEqn, thermo,
nNonOrthCorr, corr, nCorr, finalIter, cumulativeContErr):
rho.ext_assign(thermo.rho())
# Thermodynamic density needs to be updated by psi*d(p) after the
# pressure solution - done in 2 parts. Part 1:
thermo.rho().ext_assign(thermo.rho() - psi * p_rgh)
rUA = 1.0 / UEqn.A()
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
from Foam import fvc
rhorUAf = surfaceScalarField(word("(rho*(1|A(U)))"),
fvc.interpolate(rho * rUA))
U.ext_assign(rUA * UEqn.H())
phi.ext_assign(
fvc.interpolate(rho) *
((fvc.interpolate(U) & mesh.Sf()) + fvc.ddtPhiCorr(rUA, rho, U, phi)))
buoyancyPhi = -rhorUAf * ghf * fvc.snGrad(rho) * mesh.magSf()
phi.ext_assign(phi + buoyancyPhi)
from Foam import fvm
from Foam.finiteVolume import correction
for nonOrth in range(nNonOrthCorr + 1):
p_rghEqn = fvc.ddt(rho) + psi * correction(
fvm.ddt(p_rgh)) + fvc.div(phi) - fvm.laplacian(rhorUAf, p_rgh)
p_rghEqn.solve(
mesh.solver(
p_rgh.select((finalIter and corr == nCorr - 1
and nonOrth == nNonOrthCorr))))
if nonOrth == nNonOrthCorr:
# Calculate the conservative fluxes
phi.ext_assign(phi + p_rghEqn.flux())
# Explicitly relax pressure for momentum corrector
p_rgh.relax()
# Correct the momentum source with the pressure gradient flux
# calculated from the relaxed pressure
U.ext_assign(U + rUA * fvc.reconstruct(
(buoyancyPhi + p_rghEqn.flux()) / rhorUAf))
U.correctBoundaryConditions()
pass
p.ext_assign(p_rgh + rho * gh)
# Second part of thermodynamic density update
thermo.rho().ext_assign(thermo.rho() + psi * p_rgh)
DpDt.ext_assign(
fvc.DDt(surfaceScalarField(word("phiU"), phi / fvc.interpolate(rho)),
p))
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn(rho, phi)
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs(rho, thermo,
cumulativeContErr)
return cumulativeContErr
def main_standalone(argc, argv):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase(argc, argv)
from Foam.OpenFOAM.include import createTime
runTime = createTime(args)
from Foam.OpenFOAM.include import createMeshNoClear
mesh = createMeshNoClear(runTime)
transportProperties, nu = readTransportProperties(runTime, mesh)
p, U, phi = _createFields(runTime, mesh)
turbulenceProperties, force, K, forceGen = readTurbulenceProperties(
runTime, mesh, U)
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
# * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * #
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n"
while runTime.loop():
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls(
mesh)
from Foam.randomProcesses import fft
from Foam.OpenFOAM import ReImSum
force.internalField().ext_assign(
ReImSum(
fft.reverseTransform(
K / (K.mag() + 1.0e-6) ^ forceGen.newField(), K.nn())))
globalProperties(runTime, U, nu, force)
from Foam import fvm
UEqn = fvm.ddt(U) + fvm.div(phi, U) - fvm.laplacian(nu, U) == force
from Foam import fvc
from Foam.finiteVolume import solve
solve(UEqn == -fvc.grad(p))
# --- PISO loop
for corr in range(1):
rUA = 1.0 / UEqn.A()
U.ext_assign(rUA * UEqn.H())
phi.ext_assign((fvc.interpolate(U) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, U, phi))
pEqn = fvm.laplacian(rUA, p) == fvc.div(phi)
pEqn.solve()
phi.ext_assign(phi - pEqn.flux())
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs(mesh, phi, runTime,
cumulativeContErr)
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
pass
runTime.write()
if runTime.outputTime():
from Foam.randomProcesses import calcEk
from Foam.OpenFOAM import word, fileName
calcEk(U, K).write(fileName(runTime.timePath() / fileName("Ek")),
runTime.graphFormat())
pass
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" \
<< " ClockTime = " << runTime.elapsedClockTime() << " s" << nl
pass
ext_Info() << "End\n" << nl
import os
return os.EX_OK
def fun_pEqn( i, mesh, p, rho, turb, thermo, thermoFluid, K, UEqn, U, phi, psi, DpDt, initialMass, p_rgh, gh, ghf, \
nNonOrthCorr, oCorr, nOuterCorr, corr, nCorr, cumulativeContErr ) :
closedVolume = p_rgh.needReference()
rho.ext_assign(thermo.rho())
rUA = 1.0 / UEqn.A()
from Foam import fvc
from Foam.OpenFOAM import word
from Foam.finiteVolume import surfaceScalarField
rhorUAf = surfaceScalarField(word("(rho*(1|A(U)))"),
fvc.interpolate(rho * rUA))
U.ext_assign(rUA * UEqn.H())
from Foam import fvc
phiU = (fvc.interpolate(rho) * (
(fvc.interpolate(U) & mesh.Sf()) + fvc.ddtPhiCorr(rUA, rho, U, phi)))
phi.ext_assign(phiU - rhorUAf * ghf * fvc.snGrad(rho) * mesh.magSf())
from Foam import fvm
for nonOrth in range(nNonOrthCorr + 1):
p_rghEqn = (fvm.ddt(psi, p_rgh) + fvc.ddt(psi, rho) * gh +
fvc.div(phi) - fvm.laplacian(rhorUAf, p_rgh))
p_rghEqn.solve(
mesh.solver(
p_rgh.select((oCorr == nOuterCorr - 1 and corr == (nCorr - 1)
and nonOrth == nNonOrthCorr))))
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi + p_rghEqn.flux())
pass
pass
# Correct velocity field
U.ext_assign(U + rUA * fvc.reconstruct((phi - phiU) / rhorUAf))
U.correctBoundaryConditions()
p.ext_assign(p_rgh + rho * gh)
#Update pressure substantive derivative
DpDt.ext_assign(
fvc.DDt(surfaceScalarField(word("phiU"), phi / fvc.interpolate(rho)),
p))
# Solve continuity
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn(rho, phi)
# Update continuity errors
cumulativeContErr = compressibleContinuityErrors(i, mesh, rho, thermo,
cumulativeContErr)
# For closed-volume cases adjust the pressure and density levels
# to obey overall mass continuity
if closedVolume:
p.ext_assign(p + (initialMass - fvc.domainIntegrate(psi * p)) /
fvc.domainIntegrate(psi))
rho.ext_assign(thermo.rho())
p_rgh.ext_assign(p - rho * gh)
pass
#Update thermal conductivity
K.ext_assign(thermoFluid[i].Cp() * turb.alphaEff())
return cumulativeContErr
def main_standalone( argc, argv ):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase( argc, argv )
from Foam.OpenFOAM.include import createTime
runTime = createTime( args )
from Foam.OpenFOAM.include import createMesh
mesh = createMesh( runTime )
thermodynamicProperties, R, Cv = readThermodynamicProperties( runTime, mesh )
transportProperties, mu = readingTransportProperties( runTime, mesh )
p, T, e, U, psi, rho, phi = _createFields( runTime, mesh, R, Cv )
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
runTime.increment()
while not runTime.end() :
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transSonic, nOuterCorr = readPISOControls( mesh )
from Foam.finiteVolume.cfdTools.compressible import compressibleCourantNo
CoNum, meanCoNum = compressibleCourantNo( mesh, phi, rho, runTime )
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn( rho, phi )
from Foam import fvm
UEqn = fvm.ddt( rho, U ) + fvm.div( phi, U ) - fvm.laplacian( mu, U )
from Foam import fvc
from Foam.finiteVolume import solve
solve( UEqn == -fvc.grad( p ) )
solve( fvm.ddt( rho, e ) + fvm.div( phi, e ) - fvm.laplacian( mu, e ) == \
- p * fvc.div( phi / fvc.interpolate( rho ) ) + mu * fvc.grad( U ).symm().magSqr() )
T.ext_assign( e / Cv )
psi.ext_assign( 1.0 / ( R * T ) )
# --- PISO loop
for corr in range( nCorr ):
rUA = 1.0/UEqn.A()
U.ext_assign( rUA * UEqn.H() )
from Foam.OpenFOAM import word
phid = ( ( fvc.interpolate( rho * U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, rho, U, phi ) ) / fvc.interpolate( p )
print "111111111111"
for nonOrth in range( nNonOrthCorr + 1 ):
pEqn = fvm.ddt( psi, p ) + fvm.div( phid, p, word( "div(phid,p)" ) ) - fvm.laplacian( rho * rUA, p )
pEqn.solve()
phi = pEqn.flux()
pass
cumulativeContErr = compressibleContinuityErrs( p, rho, phi, psi, cumulativeContErr )
U.ext_assign( U - rUA * fvc.grad( p ) )
U.correctBoundaryConditions()
pass
rho.ext_assign( psi * p )
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
runTime.increment()
pass
ext_Info() << "End\n"
import os
return os.EX_OK
def main_standalone( argc, argv ):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase( argc, argv )
from Foam.OpenFOAM.include import createTime
runTime = createTime( args )
from Foam.OpenFOAM.include import createMesh
mesh = createMesh( runTime )
p, U, phi, turbulence, pRefCell, pRefValue, laminarTransport = _createFields( runTime, mesh )
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" <<nl
while runTime.loop() :
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls( mesh )
from Foam.finiteVolume.cfdTools.incompressible import CourantNo
CoNum, meanCoNum = CourantNo( mesh, phi, runTime )
# Pressure-velocity PISO corrector
from Foam import fvm
#Momentum predictor
# The initial C++ expression does not work properly, because of
# 1. turbulence.divDevRhoReff( U ) - changes values for the U boundaries
# 2. the order of expression arguments computation differs with C++
#UEqn = fvm.ddt( U ) + fvm.div( phi, U ) + turbulence.divDevReff( U )
UEqn = turbulence.divDevReff( U ) + ( fvm.ddt( U ) + fvm.div( phi, U ) )
UEqn.relax()
from Foam.finiteVolume import solve
from Foam import fvc
if momentumPredictor :
solve( UEqn == -fvc.grad( p ) )
pass
# --- PISO loop
for corr in range( nCorr ) :
rUA = 1.0 / UEqn.A()
U.ext_assign( rUA * UEqn.H() )
phi.ext_assign( ( fvc.interpolate(U) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, U, phi ) )
from Foam.finiteVolume import adjustPhi
adjustPhi( phi, U, p )
#Non-orthogonal pressure corrector loop
for nonOrth in range( nNonOrthCorr + 1 ):
#Pressure corrector
pEqn = fvm.laplacian( rUA, p ) == fvc.div( phi )
pEqn.setReference( pRefCell, pRefValue )
if corr == ( nCorr-1 ) and nonOrth == nNonOrthCorr :
from Foam.OpenFOAM import word
pEqn.solve( mesh.solver( word( "pFinal" ) ) )
pass
else:
pEqn.solve()
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign( phi - pEqn.flux() )
pass
pass
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs( mesh, phi, runTime, cumulativeContErr )
U.ext_assign( U - rUA * fvc.grad( p ) )
U.correctBoundaryConditions()
pass
turbulence.correct()
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n" << nl
import os
return os.EX_OK
def main_standalone(argc, argv):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase(argc, argv)
from Foam.OpenFOAM.include import createTime
runTime = createTime(args)
from Foam.OpenFOAM.include import createMesh
mesh = createMesh(runTime)
transportProperties, nu, p, U, phi, pRefCell, pRefValue = createFields(runTime, mesh)
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n"
while runTime.loop():
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls(mesh)
from Foam.finiteVolume.cfdTools.incompressible import CourantNo
CoNum, meanCoNum = CourantNo(mesh, phi, runTime)
from Foam import fvm
UEqn = fvm.ddt(U) + fvm.div(phi, U) - fvm.laplacian(nu, U)
from Foam import fvc
from Foam.finiteVolume import solve
solve(UEqn == -fvc.grad(p))
# --- PISO loop
for corr in range(nCorr):
rUA = 1.0 / UEqn.A()
U.ext_assign(rUA * UEqn.H())
phi.ext_assign((fvc.interpolate(U) & mesh.Sf()) + fvc.ddtPhiCorr(rUA, U, phi))
from Foam.finiteVolume import adjustPhi
adjustPhi(phi, U, p)
for nonOrth in range(nNonOrthCorr + 1):
pEqn = fvm.laplacian(rUA, p) == fvc.div(phi)
pEqn.setReference(pRefCell, pRefValue)
pEqn.solve()
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi - pEqn.flux())
pass
pass
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs(mesh, phi, runTime, cumulativeContErr)
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
pass
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << " ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n"
import os
return os.EX_OK
def main_standalone(argc, argv):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase(argc, argv)
from Foam.OpenFOAM.include import createTime
runTime = createTime(args)
from Foam.OpenFOAM.include import createMesh
mesh = createMesh(runTime)
gravitationalProperties, g, rotating, Omega, magg, gHat = readGravitationalAcceleration(
runTime, mesh)
h, h0, U, hU, hTotal, phi, F = _createFields(runTime, mesh, Omega, gHat)
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
while runTime.loop():
ext_Info() << "\nTime = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls(
mesh)
CoNum, meanCoNum, waveCoNum = CourantNo(runTime, mesh, h, phi, magg)
for ucorr in range(nOuterCorr):
from Foam.finiteVolume import surfaceScalarField
from Foam import fvc
from Foam.OpenFOAM import word
phiv = surfaceScalarField(word("phiv"), phi / fvc.interpolate(h))
from Foam import fvm
hUEqn = fvm.ddt(hU) + fvm.div(phiv, hU)
hUEqn.relax()
if momentumPredictor:
from Foam.finiteVolume import solve
from Foam import fvc
if rotating:
solve(hUEqn + (F ^ hU) == -magg * h * fvc.grad(h + h0))
pass
else:
solve(hUEqn == -magg * h * fvc.grad(h + h0))
pass
# Constrain the momentum to be in the geometry if 3D geometry
if mesh.nGeometricD() == 3:
hU.ext_assign(hU - (gHat & hU) * gHat)
hU.correctBoundaryConditions()
pass
for corr in range(nCorr):
hf = fvc.interpolate(h)
rUA = 1.0 / hUEqn.A()
ghrUAf = magg * fvc.interpolate(h * rUA)
phih0 = ghrUAf * mesh.magSf() * fvc.snGrad(h0)
if rotating:
hU.ext_assign(rUA * (hUEqn.H() - (F ^ hU)))
pass
else:
hU = rUA * hUEqn.H()
pass
phi.ext_assign((fvc.interpolate(hU) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, h, hU, phi) - phih0)
for nonOrth in range(nNonOrthCorr + 1):
hEqn = fvm.ddt(h) + fvc.div(phi) - fvm.laplacian(ghrUAf, h)
if ucorr < nOuterCorr - 1 or corr < nCorr - 1:
hEqn.solve()
pass
else:
hEqn.solve(mesh.solver(word(str(h.name()) + "Final")))
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi + hEqn.flux())
pass
hU.ext_assign(hU - rUA * h * magg * fvc.grad(h + h0))
#Constrain the momentum to be in the geometry if 3D geometry
if mesh.nGeometricD() == 3:
hU.ext_assign(hU - (gHat & hU) * gHat)
pass
hU.correctBoundaryConditions()
pass
pass
U == hU / h
hTotal == h + h0
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n" << nl
import os
return os.EX_OK
def main_standalone( argc, argv ):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase( argc, argv )
from Foam.OpenFOAM.include import createTime
runTime = createTime( args )
from Foam.OpenFOAM.include import createMesh
mesh = createMesh( runTime )
thermodynamicProperties, rho0, p0, psi, rhoO = readThermodynamicProperties( runTime, mesh )
transportProperties, mu = readTransportProperties( runTime, mesh )
p, U, rho, phi = _createFields( runTime, mesh, rhoO, psi )
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
while runTime.loop() :
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls( mesh )
from Foam.finiteVolume.cfdTools.compressible import compressibleCourantNo
CoNum, meanCoNum = compressibleCourantNo( mesh, phi, rho, runTime )
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn( rho, phi )
from Foam import fvm, fvc
from Foam.finiteVolume import solve
UEqn = fvm.ddt(rho, U) + fvm.div(phi, U) - fvm.laplacian(mu, U)
solve( UEqn == -fvc.grad( p ) )
for corr in range( nCorr ):
rUA = 1.0 / UEqn.A()
U.ext_assign( rUA * UEqn.H() )
from Foam.OpenFOAM import word
from Foam.finiteVolume import surfaceScalarField
phid = surfaceScalarField( word( "phid" ),
psi * ( ( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, rho, U, phi ) ) )
phi.ext_assign( ( rhoO / psi ) * phid )
pEqn = fvm.ddt( psi, p ) + fvc.div( phi ) + fvm.div( phid, p ) - fvm.laplacian( rho * rUA, p )
pEqn.solve()
phi.ext_assign( phi + pEqn.flux() )
cumulativeContErr = compressibleContinuityErrs( rho, phi, psi, rho0, p, p0, cumulativeContErr )
U.ext_assign( U - rUA * fvc.grad( p ) )
U.correctBoundaryConditions()
pass
rho.ext_assign( rhoO + psi*p )
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n"
import os
return os.EX_OK
def _pEqn(
rho,
thermo,
UEqn,
nNonOrthCorr,
psi,
U,
mesh,
phi,
p,
DpDt,
pMin,
corr,
cumulativeContErr,
nCorr,
oCorr,
nOuterCorr,
transonic,
):
rho.ext_assign(thermo.rho())
rUA = 1.0 / UEqn.A()
U.ext_assign(rUA * UEqn.H())
if nCorr <= 1:
UEqn.clear()
pass
from Foam import fvc, fvm
from Foam.OpenFOAM import word
from Foam.finiteVolume import surfaceScalarField
if transonic:
phid = surfaceScalarField(
word("phid"), fvc.interpolate(psi) * ((fvc.interpolate(U) & mesh.Sf()) + fvc.ddtPhiCorr(rUA, rho, U, phi))
)
for nonOrth in range(nNonOrthCorr + 1):
pEqn = fvm.ddt(psi, p) + fvm.div(phid, p) - fvm.laplacian(rho * rUA, p)
if oCorr == nOuterCorr - 1 and corr == nCorr - 1 and nonOrth == nNonOrthCorr:
pEqn.solve(mesh.solver(word("pFinal")))
pass
else:
pEqn.solve()
pass
if nonOrth == nNonOrthCorr:
phi == pEqn.flux()
pass
pass
pass
else:
phi.ext_assign(fvc.interpolate(rho) * ((fvc.interpolate(U) & mesh.Sf())))
for nonOrth in range(nNonOrthCorr + 1):
# Pressure corrector
pEqn = fvm.ddt(psi, p) + fvc.div(phi) - fvm.laplacian(rho * rUA, p)
if oCorr == nOuterCorr - 1 and corr == nCorr - 1 and nonOrth == nNonOrthCorr:
pEqn.solve(mesh.solver(word("pFinal")))
pass
else:
pEqn.solve()
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi + pEqn.flux())
pass
pass
pass
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn(rho, phi)
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs(rho, thermo, cumulativeContErr)
p.relax()
rho.ext_assign(thermo.rho())
rho.relax()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "rho max/min : " << rho.ext_max().value() << " " << rho.ext_min().value() << nl
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
DpDt.ext_assign(fvc.DDt(surfaceScalarField(word("phiU"), phi / fvc.interpolate(rho)), p))
from Foam.finiteVolume import bound
bound(p, pMin)
return cumulativeContErr
def main_standalone( argc, argv ):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase( argc, argv )
from Foam.OpenFOAM.include import createTime
runTime = createTime( args )
from Foam.OpenFOAM.include import createMesh
mesh = createMesh( runTime )
gravitationalProperties, g, rotating, Omega, magg, gHat = readGravitationalAcceleration( runTime, mesh )
h, h0, U, hU, hTotal, phi, F = _createFields( runTime, mesh, Omega, gHat )
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
while runTime.loop() :
ext_Info() << "\nTime = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls( mesh )
CoNum, meanCoNum, waveCoNum = CourantNo( runTime, mesh, h, phi, magg )
for ucorr in range( nOuterCorr ):
from Foam.finiteVolume import surfaceScalarField
from Foam import fvc
from Foam.OpenFOAM import word
phiv = surfaceScalarField( word( "phiv" ), phi / fvc.interpolate( h ) )
from Foam import fvm
hUEqn = fvm.ddt( hU ) + fvm.div( phiv, hU )
hUEqn.relax()
if momentumPredictor:
from Foam.finiteVolume import solve
from Foam import fvc
if rotating:
solve( hUEqn + ( F ^ hU ) == -magg * h * fvc.grad( h + h0 ) )
pass
else:
solve( hUEqn == -magg * h * fvc.grad( h + h0 ) )
pass
# Constrain the momentum to be in the geometry if 3D geometry
if mesh.nGeometricD() == 3 :
hU.ext_assign( hU - ( gHat & hU ) * gHat )
hU.correctBoundaryConditions();
pass
for corr in range( nCorr ):
hf = fvc.interpolate( h )
rUA = 1.0 / hUEqn.A()
ghrUAf = magg * fvc.interpolate( h * rUA )
phih0 = ghrUAf * mesh.magSf() * fvc.snGrad( h0 )
if rotating:
hU.ext_assign( rUA * ( hUEqn .H() - ( F ^ hU ) ) )
pass
else:
hU = rUA * hUEqn.H()
pass
phi.ext_assign( ( fvc.interpolate( hU ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, h, hU, phi )- phih0 )
for nonOrth in range(nNonOrthCorr + 1):
hEqn = fvm.ddt( h ) + fvc.div( phi ) - fvm.laplacian( ghrUAf, h )
if ucorr < nOuterCorr-1 or corr < nCorr-1 :
hEqn.solve()
pass
else:
hEqn.solve( mesh.solver( word( str( h.name() ) + "Final" ) ) )
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign( phi + hEqn.flux() )
pass
hU.ext_assign( hU - rUA * h * magg * fvc.grad( h + h0 ) )
#Constrain the momentum to be in the geometry if 3D geometry
if mesh.nGeometricD() == 3:
hU.ext_assign( hU - ( gHat & hU ) * gHat )
pass
hU.correctBoundaryConditions()
pass
pass
U == hU / h
hTotal == h + h0
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n" << nl
import os
return os.EX_OK
def main_standalone(argc, argv):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase(argc, argv)
from Foam.OpenFOAM.include import createTime
runTime = createTime(args)
from Foam.OpenFOAM.include import createMesh
mesh = createMesh(runTime)
p, U, phi, turbulence, pRefCell, pRefValue, laminarTransport = _createFields(
runTime, mesh)
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
while runTime.loop():
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls(
mesh)
from Foam.finiteVolume.cfdTools.incompressible import CourantNo
CoNum, meanCoNum = CourantNo(mesh, phi, runTime)
# Pressure-velocity PISO corrector
from Foam import fvm
#Momentum predictor
# The initial C++ expression does not work properly, because of
# 1. turbulence.divDevRhoReff( U ) - changes values for the U boundaries
# 2. the order of expression arguments computation differs with C++
#UEqn = fvm.ddt( U ) + fvm.div( phi, U ) + turbulence.divDevReff( U )
UEqn = turbulence.divDevReff(U) + (fvm.ddt(U) + fvm.div(phi, U))
UEqn.relax()
from Foam.finiteVolume import solve
from Foam import fvc
if momentumPredictor:
solve(UEqn == -fvc.grad(p))
pass
# --- PISO loop
for corr in range(nCorr):
rUA = 1.0 / UEqn.A()
U.ext_assign(rUA * UEqn.H())
phi.ext_assign((fvc.interpolate(U) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, U, phi))
from Foam.finiteVolume import adjustPhi
adjustPhi(phi, U, p)
#Non-orthogonal pressure corrector loop
for nonOrth in range(nNonOrthCorr + 1):
#Pressure corrector
pEqn = fvm.laplacian(rUA, p) == fvc.div(phi)
pEqn.setReference(pRefCell, pRefValue)
if corr == (nCorr - 1) and nonOrth == nNonOrthCorr:
from Foam.OpenFOAM import word
pEqn.solve(mesh.solver(word("pFinal")))
pass
else:
pEqn.solve()
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi - pEqn.flux())
pass
pass
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs(mesh, phi, runTime,
cumulativeContErr)
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
pass
turbulence.correct()
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n" << nl
import os
return os.EX_OK
def fun_pEqn( mesh, p, rho, psi, p_rgh, U, phi, ghf, gh, DpDt, UEqn, thermo, nNonOrthCorr, corr, nCorr, finalIter, cumulativeContErr ):
rho.ext_assign( thermo.rho() )
# Thermodynamic density needs to be updated by psi*d(p) after the
# pressure solution - done in 2 parts. Part 1:
thermo.rho().ext_assign( thermo.rho() - psi * p_rgh )
rUA = 1.0 / UEqn.A()
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
from Foam import fvc
rhorUAf = surfaceScalarField( word( "(rho*(1|A(U)))" ), fvc.interpolate( rho * rUA ) )
U.ext_assign( rUA*UEqn.H() )
phi.ext_assign( fvc.interpolate( rho ) * ( ( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, rho, U, phi ) ) )
buoyancyPhi = -rhorUAf * ghf * fvc.snGrad( rho ) * mesh.magSf()
phi.ext_assign( phi + buoyancyPhi )
from Foam import fvm
from Foam.finiteVolume import correction
for nonOrth in range( nNonOrthCorr +1 ):
p_rghEqn = fvc.ddt( rho ) + psi * correction( fvm.ddt( p_rgh ) ) + fvc.div( phi ) - fvm.laplacian( rhorUAf, p_rgh )
p_rghEqn.solve( mesh.solver( p_rgh.select( ( finalIter and corr == nCorr-1 and nonOrth == nNonOrthCorr ) ) ) )
if nonOrth == nNonOrthCorr:
# Calculate the conservative fluxes
phi.ext_assign( phi + p_rghEqn.flux() )
# Explicitly relax pressure for momentum corrector
p_rgh.relax()
# Correct the momentum source with the pressure gradient flux
# calculated from the relaxed pressure
U.ext_assign( U + rUA * fvc.reconstruct( ( buoyancyPhi + p_rghEqn.flux() ) / rhorUAf ) )
U.correctBoundaryConditions()
pass
p.ext_assign( p_rgh + rho * gh )
# Second part of thermodynamic density update
thermo.rho().ext_assign( thermo.rho() + psi * p_rgh )
DpDt.ext_assign( fvc.DDt( surfaceScalarField( word( "phiU" ), phi / fvc.interpolate( rho ) ), p ) )
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn( rho, phi )
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs( rho, thermo, cumulativeContErr )
return cumulativeContErr
def main_standalone( argc, argv ):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase( argc, argv )
from Foam.OpenFOAM.include import createTime
runTime = createTime( args )
from Foam.OpenFOAM.include import createMesh
mesh = createMesh( runTime )
transportProperties, nu, Ubar, magUbar, flowDirection = readTransportProperties( runTime, mesh)
p, U, phi, laminarTransport, sgsModel, pRefCell, pRefValue = _createFields( runTime, mesh )
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
gradP, gradPFile = createGradP( runTime)
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
while runTime.loop() :
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls( mesh )
from Foam.finiteVolume.cfdTools.incompressible import CourantNo
CoNum, meanCoNum = CourantNo( mesh, phi, runTime )
sgsModel.correct()
from Foam import fvm
UEqn = fvm.ddt( U ) + fvm.div( phi, U ) + sgsModel.divDevBeff( U ) == flowDirection * gradP
if momentumPredictor:
from Foam.finiteVolume import solve
from Foam import fvc
solve( UEqn == -fvc.grad( p ) )
pass
rUA = 1.0 / UEqn.A()
for corr in range( nCorr ):
U.ext_assign( rUA * UEqn.H() )
from Foam import fvc
phi.ext_assign( ( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, U, phi ) )
from Foam.finiteVolume import adjustPhi
adjustPhi(phi, U, p)
from Foam.OpenFOAM import word
for nonOrth in range( nNonOrthCorr + 1 ):
pEqn = fvm.laplacian( rUA, p ) == fvc.div( phi )
pEqn.setReference( pRefCell, pRefValue )
if corr == nCorr-1 and nonOrth == nNonOrthCorr:
pEqn.solve( mesh.solver( word( str( p.name() ) + "Final" ) ) )
pass
else:
pEqn.solve( mesh.solver( p.name() ) )
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign( phi - pEqn.flux() )
pass
pass
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs( mesh, phi, runTime, cumulativeContErr )
U.ext_assign( U - rUA * fvc.grad( p ) )
U.correctBoundaryConditions()
pass
# Correct driving force for a constant mass flow rate
# Extract the velocity in the flow direction
magUbarStar = ( flowDirection & U ).weightedAverage( mesh.V() )
# Calculate the pressure gradient increment needed to
# adjust the average flow-rate to the correct value
gragPplus = ( magUbar - magUbarStar ) / rUA.weightedAverage( mesh.V() )
U.ext_assign( U + flowDirection * rUA * gragPplus )
gradP +=gragPplus
ext_Info() << "Uncorrected Ubar = " << magUbarStar.value() << " " << "pressure gradient = " << gradP.value() << nl
runTime.write()
writeGradP( runTime, gradP )
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n" << nl
import os
return os.EX_OK
def main_standalone(argc, argv):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase(argc, argv)
from Foam.OpenFOAM.include import createTime
runTime = createTime(args)
from Foam.OpenFOAM.include import createMesh
mesh = createMesh(runTime)
thermodynamicProperties, R, Cv, Cp, gamma, Pr = readThermodynamicProperties(
runTime, mesh)
p, T, psi, pbf, rhoBoundaryTypes, rho, U, Ubf, rhoUboundaryTypes, \
rhoU, Tbf, rhoEboundaryTypes, rhoE, phi, phiv, rhoU, fields, magRhoU, H = _createFields( runTime, mesh, R, Cv )
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
while runTime.loop():
ext_Info() << "Time = " << runTime.value() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls(
mesh)
from Foam.OpenFOAM import readScalar, word
HbyAblend = readScalar(piso.lookup(word("HbyAblend")))
from Foam.finiteVolume.cfdTools.general.include import readTimeControls
adjustTimeStep, maxCo, maxDeltaT = readTimeControls(runTime)
CoNum = (mesh.deltaCoeffs() * phiv.mag() /
mesh.magSf()).ext_max().value() * runTime.deltaT().value()
ext_Info() << "Max Courant Number = " << CoNum << nl
from Foam.finiteVolume.cfdTools.general.include import setDeltaT
runTime = setDeltaT(runTime, adjustTimeStep, maxCo, maxDeltaT, CoNum)
for outerCorr in range(nOuterCorr):
magRhoU.ext_assign(rhoU.mag())
H.ext_assign((rhoE + p) / rho)
from Foam.fv import multivariateGaussConvectionScheme_scalar
mvConvection = multivariateGaussConvectionScheme_scalar(
mesh, fields, phiv, mesh.divScheme(word("div(phiv,rhoUH)")))
from Foam.finiteVolume import solve
from Foam import fvm
solve(fvm.ddt(rho) + mvConvection.fvmDiv(phiv, rho))
tmp = mvConvection.interpolationScheme()()(magRhoU)
rhoUWeights = tmp.ext_weights(magRhoU)
from Foam.finiteVolume import weighted_vector
rhoUScheme = weighted_vector(rhoUWeights)
from Foam import fv, fvc
rhoUEqn = fvm.ddt(rhoU) + fv.gaussConvectionScheme_vector(
mesh, phiv, rhoUScheme).fvmDiv(phiv, rhoU)
solve(rhoUEqn == -fvc.grad(p))
solve(
fvm.ddt(rhoE) + mvConvection.fvmDiv(phiv, rhoE) ==
-mvConvection.fvcDiv(phiv, p))
T.ext_assign((rhoE - 0.5 * rho * (rhoU / rho).magSqr()) / Cv / rho)
psi.ext_assign(1.0 / (R * T))
p.ext_assign(rho / psi)
for corr in range(nCorr):
rrhoUA = 1.0 / rhoUEqn.A()
from Foam.finiteVolume import surfaceScalarField
rrhoUAf = surfaceScalarField(word("rrhoUAf"),
fvc.interpolate(rrhoUA))
HbyA = rrhoUA * rhoUEqn.H()
from Foam.finiteVolume import LimitedScheme_vector_MUSCLLimiter_NVDTVD_limitFuncs_magSqr
from Foam.OpenFOAM import IStringStream, word
HbyAWeights = HbyAblend * mesh.weights() + ( 1.0 - HbyAblend ) * \
LimitedScheme_vector_MUSCLLimiter_NVDTVD_limitFuncs_magSqr( mesh, phi, IStringStream( "HbyA" )() ).weights( HbyA )
from Foam.finiteVolume import surfaceInterpolationScheme_vector
phi.ext_assign( ( surfaceInterpolationScheme_vector.ext_interpolate(HbyA, HbyAWeights) & mesh.Sf() ) \
+ HbyAblend * fvc.ddtPhiCorr( rrhoUA, rho, rhoU, phi ) )
p.ext_boundaryField().updateCoeffs()
phiGradp = rrhoUAf * mesh.magSf() * fvc.snGrad(p)
phi.ext_assign(phi - phiGradp)
resetPhiPatches(phi, rhoU, mesh)
rhof = mvConvection.interpolationScheme()()(rho).interpolate(
rho)
phiv.ext_assign(phi / rhof)
pEqn = fvm.ddt(psi, p) + mvConvection.fvcDiv(
phiv, rho) + fvc.div(phiGradp) - fvm.laplacian(rrhoUAf, p)
pEqn.solve()
phi.ext_assign(phi + phiGradp + pEqn.flux())
rho.ext_assign(psi * p)
rhof.ext_assign(
mvConvection.interpolationScheme()()(rho).interpolate(rho))
phiv.ext_assign(phi / rhof)
rhoU.ext_assign(HbyA - rrhoUA * fvc.grad(p))
rhoU.correctBoundaryConditions()
pass
pass
U.ext_assign(rhoU / rho)
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n"
import os
return os.EX_OK
def main_standalone(argc, argv):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase(argc, argv)
from Foam.OpenFOAM.include import createTime
runTime = createTime(args)
from Foam.OpenFOAM.include import createMesh
mesh = createMesh(runTime)
thermodynamicProperties, rho0, p0, psi, rhoO = readThermodynamicProperties(
runTime, mesh)
transportProperties, mu = readTransportProperties(runTime, mesh)
p, U, rho, phi = _createFields(runTime, mesh, rhoO, psi)
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
while runTime.loop():
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls(
mesh)
from Foam.finiteVolume.cfdTools.compressible import compressibleCourantNo
CoNum, meanCoNum, velMag = compressibleCourantNo(
mesh, phi, rho, runTime)
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn(rho, phi)
from Foam import fvm, fvc
from Foam.finiteVolume import solve
UEqn = fvm.ddt(rho, U) + fvm.div(phi, U) - fvm.laplacian(mu, U)
solve(UEqn == -fvc.grad(p))
for corr in range(nCorr):
rUA = 1.0 / UEqn.A()
U.ext_assign(rUA * UEqn.H())
from Foam.OpenFOAM import word
from Foam.finiteVolume import surfaceScalarField
phid = surfaceScalarField(
word("phid"),
psi * ((fvc.interpolate(U) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, rho, U, phi)))
phi.ext_assign((rhoO / psi) * phid)
pEqn = fvm.ddt(psi, p) + fvc.div(phi) + fvm.div(
phid, p) - fvm.laplacian(rho * rUA, p)
pEqn.solve()
phi.ext_assign(phi + pEqn.flux())
cumulativeContErr = compressibleContinuityErrs(
rho, phi, psi, rho0, p, p0, cumulativeContErr)
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
pass
rho.ext_assign(rhoO + psi * p)
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
pass
ext_Info() << "End\n"
import os
return os.EX_OK
def _pEqn( rho, thermo, UEqn, nNonOrthCorr, psi, U, mesh, phi, p, cumulativeContErr ):
from Foam.finiteVolume import volScalarField
rUA = 1.0/UEqn.A()
U.ext_assign( rUA*UEqn.H() )
from Foam import fvc
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
phid = surfaceScalarField( word( "phid" ),
fvc.interpolate( thermo.psi() ) * ( (fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, rho, U, phi ) ) )
for nonOrth in range( nNonOrthCorr + 1 ) :
from Foam import fvm
pEqn = ( fvm.ddt(psi, p) + fvm.div(phid, word( "div(phid,p)" ) ) - fvm.laplacian(rho*rUA, p) )
pEqn.solve()
if (nonOrth == nNonOrthCorr) :
phi.ext_assign( pEqn.flux() )
pass
pass
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn( rho, phi )
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs( rho, thermo, cumulativeContErr )
U.ext_assign( U - rUA * fvc.grad(p) )
U.correctBoundaryConditions()
return cumulativeContErr
def _pEqn(runTime, mesh, UEqn, thermo, p, psi, U, rho, phi, DpDt, g,
initialMass, totalVolume, corr, nCorr, nNonOrthCorr,
cumulativeContErr):
closedVolume = p.needReference()
rho.ext_assign(thermo.rho())
# Thermodynamic density needs to be updated by psi*d(p) after the
# pressure solution - done in 2 parts. Part 1:
thermo.rho().ext_assign(thermo.rho() - psi * p)
rUA = 1.0 / UEqn.A()
from Foam.OpenFOAM import word
from Foam.finiteVolume import surfaceScalarField
from Foam import fvc
rhorUAf = surfaceScalarField(word("(rho*(1|A(U)))"),
fvc.interpolate(rho * rUA))
U.ext_assign(rUA * UEqn.H())
phiU = fvc.interpolate(rho) * (
(fvc.interpolate(U) & mesh.Sf()) + fvc.ddtPhiCorr(rUA, rho, U, phi))
phi.ext_assign(phiU + rhorUAf * fvc.interpolate(rho) * (g & mesh.Sf()))
for nonOrth in range(nNonOrthCorr + 1):
from Foam import fvm
from Foam.finiteVolume import correction
pEqn = fvc.ddt(rho) + psi * correction(
fvm.ddt(p)) + fvc.div(phi) - fvm.laplacian(rhorUAf, p)
if corr == nCorr - 1 and nonOrth == nNonOrthCorr:
pEqn.solve(mesh.solver(word(str(p.name()) + "Final")))
pass
else:
pEqn.solve(mesh.solver(p.name()))
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi + pEqn.flux())
pass
# Second part of thermodynamic density update
thermo.rho().ext_assign(thermo.rho() + psi * p)
U.ext_assign(U + rUA * fvc.reconstruct((phi - phiU) / rhorUAf))
U.correctBoundaryConditions()
DpDt.ext_assign(
fvc.DDt(surfaceScalarField(word("phiU"), phi / fvc.interpolate(rho)),
p))
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn(rho, phi)
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs(rho, thermo,
cumulativeContErr)
# For closed-volume cases adjust the pressure and density levels
# to obey overall mass continuity
if closedVolume:
p.ext_assign(p + (initialMass - fvc.domainIntegrate(psi * p)) /
fvc.domainIntegrate(psi))
thermo.rho().ext_assign(psi * p)
rho.ext_assign(rho +
(initialMass - fvc.domainIntegrate(rho)) / totalVolume)
pass
return cumulativeContErr
def main_standalone(argc, argv):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase(argc, argv)
from Foam.OpenFOAM.include import createTime
runTime = createTime(args)
from Foam.OpenFOAM.include import createMesh
mesh = createMesh(runTime)
thermodynamicProperties, R, Cv = readThermodynamicProperties(runTime, mesh)
transportProperties, mu = readingTransportProperties(runTime, mesh)
p, T, e, U, psi, rho, phi = _createFields(runTime, mesh, R, Cv)
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n" << nl
runTime.increment()
while not runTime.end():
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transSonic, nOuterCorr = readPISOControls(
mesh)
from Foam.finiteVolume.cfdTools.compressible import compressibleCourantNo
CoNum, meanCoNum = compressibleCourantNo(mesh, phi, rho, runTime)
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn(rho, phi)
from Foam import fvm
UEqn = fvm.ddt(rho, U) + fvm.div(phi, U) - fvm.laplacian(mu, U)
from Foam import fvc
from Foam.finiteVolume import solve
solve(UEqn == -fvc.grad(p))
solve( fvm.ddt( rho, e ) + fvm.div( phi, e ) - fvm.laplacian( mu, e ) == \
- p * fvc.div( phi / fvc.interpolate( rho ) ) + mu * fvc.grad( U ).symm().magSqr() )
T.ext_assign(e / Cv)
psi.ext_assign(1.0 / (R * T))
# --- PISO loop
for corr in range(nCorr):
rUA = 1.0 / UEqn.A()
U.ext_assign(rUA * UEqn.H())
from Foam.OpenFOAM import word
phid = ((fvc.interpolate(rho * U) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, rho, U, phi)) / fvc.interpolate(p)
print "111111111111"
for nonOrth in range(nNonOrthCorr + 1):
pEqn = fvm.ddt(psi, p) + fvm.div(
phid, p, word("div(phid,p)")) - fvm.laplacian(
rho * rUA, p)
pEqn.solve()
phi = pEqn.flux()
pass
cumulativeContErr = compressibleContinuityErrs(
p, rho, phi, psi, cumulativeContErr)
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
pass
rho.ext_assign(psi * p)
runTime.write()
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" << \
" ClockTime = " << runTime.elapsedClockTime() << " s" << nl << nl
runTime.increment()
pass
ext_Info() << "End\n"
import os
return os.EX_OK
def _pEqn( rho, thermo, UEqn, nNonOrthCorr, psi, U, mesh, phi, p, DpDt, cumulativeContErr, corr, nCorr, nOuterCorr, transonic ):
rho.ext_assign( thermo.rho() )
rUA = 1.0/UEqn.A()
U.ext_assign( rUA * UEqn.H() )
from Foam import fvc, fvm
from Foam.OpenFOAM import word
from Foam.finiteVolume import surfaceScalarField
if transonic:
phid = surfaceScalarField( word( "phid" ),
fvc.interpolate( psi ) * ( ( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, rho, U, phi ) ) )
for nonOrth in range( nNonOrthCorr + 1 ) :
pEqn = fvm.ddt( psi, p ) + fvm.div( phid, p ) - fvm.laplacian( rho * rUA, p )
pEqn.solve()
if nonOrth == nNonOrthCorr:
phi == pEqn.flux()
pass
pass
pass
else:
phi.ext_assign( fvc.interpolate( rho ) * ( ( fvc.interpolate(U) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, rho, U, phi ) ) )
for nonOrth in range( nNonOrthCorr + 1 ) :
pEqn = fvm.ddt( psi, p ) + fvc.div( phi ) - fvm.laplacian( rho * rUA, p )
pEqn.solve()
if nonOrth == nNonOrthCorr:
phi.ext_assign( phi + pEqn.flux() )
pass
pass
pass
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn( rho, phi )
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs( rho, thermo, cumulativeContErr )
U.ext_assign( U - rUA * fvc.grad( p ) )
U.correctBoundaryConditions()
DpDt.ext_assign( fvc.DDt( surfaceScalarField( word( "phiU" ), phi / fvc.interpolate( rho ) ), p ) )
return cumulativeContErr
def fun_pEqn(mesh, thermo, p, rho, psi, U, phi, DpDt, pMin, UEqn, mrfZones,
nNonOrthCorr, nCorr, oCorr, nOuterCorr, corr, transonic,
cumulativeContErr):
rho.ext_assign(thermo.rho())
rUA = 1.0 / UEqn.A()
U.ext_assign(rUA * UEqn.H())
if nCorr <= 1:
UEqn.clear()
pass
if transonic:
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
phid = surfaceScalarField(
word("phid"),
fvc.interpolate(psi) * ((fvc.interpolate(U) & mesh.Sf()) +
fvc.ddtPhiCorr(rUA, rho, U, phi)))
mrfZones.relativeFlux(fvc.interpolate(psi), phid)
from Foam import fvm
for nonOrth in range(nNonOrthCorr + 1):
pEqn = fvm.ddt(psi, p) + fvm.div(phid, p) - fvm.laplacian(
rho * rUA, p)
if oCorr == (nOuterCorr - 1) and (corr == nCorr -
1) and (nonOrth == nNonOrthCorr):
from Foam.OpenFOAM import word
pEqn.solve(mesh.solver(word("pFinal")))
pass
else:
pEqn.solve()
pass
if nonOrth == nNonOrthCorr:
phi == pEqn.flux()
pass
else:
from Foam import fvc
phi.ext_assign(
fvc.interpolate(rho) * ((fvc.interpolate(U) & mesh.Sf())))
mrfZones.relativeFlux(fvc.interpolate(rho), phi)
from Foam import fvm
for nonOrth in range(nNonOrthCorr + 1):
# Pressure corrector
pEqn = fvm.ddt(psi, p) + fvc.div(phi) - fvm.laplacian(rho * rUA, p)
if oCorr == (nOuterCorr - 1) and corr == (
nCorr - 1) and nonOrth == nNonOrthCorr:
from Foam.OpenFOAM import word
pEqn.solve(mesh.solver(word("pFinal")))
pass
else:
pEqn.solve()
pass
if nonOrth == nNonOrthCorr:
phi.ext_assign(phi + pEqn.flux())
pass
pass
from Foam.finiteVolume.cfdTools.compressible import rhoEqn
rhoEqn(rho, phi)
from Foam.finiteVolume.cfdTools.compressible import compressibleContinuityErrs
cumulativeContErr = compressibleContinuityErrs(rho, thermo,
cumulativeContErr)
# Explicitly relax pressure for momentum corrector
p.relax()
rho.ext_assign(thermo.rho())
rho.relax()
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "rho max/min : " << rho.ext_max().value(
) << " " << rho.ext_min().value() << nl
U.ext_assign(U - rUA * fvc.grad(p))
U.correctBoundaryConditions()
from Foam.finiteVolume import surfaceScalarField
from Foam.OpenFOAM import word
DpDt.ext_assign(
fvc.DDt(surfaceScalarField(word("phiU"), phi / fvc.interpolate(rho)),
p))
from Foam.finiteVolume import bound
bound(p, pMin)
pass
def main_standalone( argc, argv ):
from Foam.OpenFOAM.include import setRootCase
args = setRootCase( argc, argv )
from Foam.OpenFOAM.include import createTime
runTime = createTime( args )
from Foam.OpenFOAM.include import createMeshNoClear
mesh = createMeshNoClear( runTime )
transportProperties, nu = readTransportProperties( runTime, mesh )
p, U, phi = _createFields( runTime, mesh )
turbulenceProperties, force, K, forceGen = readTurbulenceProperties( runTime, mesh, U )
from Foam.finiteVolume.cfdTools.general.include import initContinuityErrs
cumulativeContErr = initContinuityErrs()
# * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * #
from Foam.OpenFOAM import ext_Info, nl
ext_Info() << "\nStarting time loop\n"
while runTime.loop():
ext_Info() << "Time = " << runTime.timeName() << nl << nl
from Foam.finiteVolume.cfdTools.general.include import readPISOControls
piso, nCorr, nNonOrthCorr, momentumPredictor, transonic, nOuterCorr = readPISOControls( mesh )
from Foam.randomProcesses import fft
from Foam.OpenFOAM import ReImSum
force.internalField().ext_assign( ReImSum( fft.reverseTransform( K / ( K.mag() + 1.0e-6 ) ^ forceGen.newField(), K.nn() ) ) )
globalProperties( runTime, U, nu, force )
from Foam import fvm
UEqn = fvm.ddt( U ) + fvm.div( phi, U ) - fvm.laplacian( nu, U ) == force
from Foam import fvc
from Foam.finiteVolume import solve
solve( UEqn == -fvc.grad( p ) )
# --- PISO loop
for corr in range( 1 ):
rUA = 1.0 / UEqn.A()
U.ext_assign( rUA*UEqn.H() )
phi.ext_assign( ( fvc.interpolate( U ) & mesh.Sf() ) + fvc.ddtPhiCorr( rUA, U, phi ) )
pEqn = fvm.laplacian( rUA, p ) == fvc.div( phi )
pEqn.solve()
phi.ext_assign( phi - pEqn.flux() )
from Foam.finiteVolume.cfdTools.incompressible import continuityErrs
cumulativeContErr = continuityErrs( mesh, phi, runTime, cumulativeContErr )
U.ext_assign( U - rUA * fvc.grad( p ) )
U.correctBoundaryConditions()
pass
runTime.write()
if runTime.outputTime():
from Foam.randomProcesses import calcEk
from Foam.OpenFOAM import word, fileName
calcEk( U, K ).write( fileName( runTime.timePath() / fileName( "Ek" ) ), runTime.graphFormat() )
pass
ext_Info() << "ExecutionTime = " << runTime.elapsedCpuTime() << " s" \
<< " ClockTime = " << runTime.elapsedClockTime() << " s" << nl
pass
ext_Info() << "End\n" << nl
import os
return os.EX_OK
