# -*- coding: utf-8 -*-
from pyfr.solvers.baseadvec import BaseAdvectionElements
class BaseFluidElements(object):
formulations = ['std', 'dual']
privarmap = {2: ['rho', 'u', 'v', 'p'],
3: ['rho', 'u', 'v', 'w', 'p']}
convarmap = {2: ['rho', 'rhou', 'rhov', 'E'],
3: ['rho', 'rhou', 'rhov', 'rhow', 'E']}
dualcoeffs = convarmap
visvarmap = {
2: [('density', ['rho']),
('velocity', ['u', 'v']),
('pressure', ['p'])],
3: [('density', ['rho']),
('velocity', ['u', 'v', 'w']),
('pressure', ['p'])]
}
@staticmethod
def pri_to_con(pris, cfg):
rho, p = pris[0], pris[-1]
# Multiply velocity components by rho
rhovs = [rho*c for c in pris[1:-1]]
# Compute the energy
gamma = cfg.getfloat('constants', 'gamma')
E = p/(gamma - 1) + 0.5*rho*sum(c*c for c in pris[1:-1])
return [rho] + rhovs + [E]
@staticmethod
def con_to_pri(cons, cfg):
rho, E = cons[0], cons[-1]
# Divide momentum components by rho
vs = [rhov/rho for rhov in cons[1:-1]]
# Compute the pressure
gamma = cfg.getfloat('constants', 'gamma')
p = (gamma - 1)*(E - 0.5*rho*sum(v*v for v in vs))
return [rho] + vs + [p]
[docs]class EulerElements(BaseFluidElements, BaseAdvectionElements):
[docs] def set_backend(self, *args, **kwargs):
super().set_backend(*args, **kwargs)
# Register our flux kernels
self._be.pointwise.register('pyfr.solvers.euler.kernels.tflux')
self._be.pointwise.register('pyfr.solvers.euler.kernels.tfluxlin')
# Template parameters for the flux kernels
tplargs = {
'ndims': self.ndims,
'nvars': self.nvars,
'nverts': len(self.basis.linspts),
'c': self.cfg.items_as('constants', float),
'jac_exprs': self.basis.jac_exprs
}
# Common arguments
if 'flux' in self.antialias:
u = lambda s: self._slice_mat(self._scal_qpts, s)
f = lambda s: self._slice_mat(self._vect_qpts, s)
pts, npts = 'qpts', self.nqpts
else:
u = lambda s: self._slice_mat(self.scal_upts_inb, s)
f = lambda s: self._slice_mat(self._vect_upts, s)
pts, npts = 'upts', self.nupts
# Mesh regions
regions = self._mesh_regions
if 'curved' in regions:
self.kernels['tdisf_curved'] = lambda: self._be.kernel(
'tflux', tplargs=tplargs, dims=[npts, regions['curved']],
u=u('curved'), f=f('curved'),
smats=self.smat_at(pts, 'curved')
)
if 'linear' in regions:
upts = getattr(self, pts)
self.kernels['tdisf_linear'] = lambda: self._be.kernel(
'tfluxlin', tplargs=tplargs, dims=[npts, regions['linear']],
u=u('linear'), f=f('linear'),
verts=self.ploc_at('linspts', 'linear'), upts=upts
)