def get_fargs(self,
mat,
kappa,
virtual,
state,
mode=None,
term_mode=None,
diff_var=None,
**kwargs):
from sfepy.discrete.variables import create_adof_conn, expand_basis
geo, _ = self.get_mapping(state)
n_el, n_qp, dim, n_en, n_c = self.get_data_shape(virtual)
ebf = expand_basis(geo.bf, dim)
gmat = _build_wave_strain_op(kappa, ebf)
if diff_var is None:
econn = state.field.get_econn('volume', self.region)
adc = create_adof_conn(nm.arange(state.n_dof, dtype=nm.int32),
econn, n_c, 0)
vals = state()[adc]
# Same as nm.einsum('qij,cj->cqi', gmat[0], vals)[..., None]
aux = dot_sequences(gmat, vals[:, None, :, None])
out_qp = dot_sequences(gmat, dot_sequences(mat, aux), 'ATB')
fmode = 0
else:
out_qp = dot_sequences(gmat, dot_sequences(mat, gmat), 'ATB')
fmode = 1
return out_qp, geo, fmode
def get_fargs(self,
kappa,
virtual,
state,
mode=None,
term_mode=None,
diff_var=None,
**kwargs):
from sfepy.discrete.variables import create_adof_conn, expand_basis
geo, _ = self.get_mapping(state)
n_el, n_qp, dim, n_en, n_c = self.get_data_shape(virtual)
ebf = expand_basis(geo.bf, dim)
aux = nm.einsum('i,...ij->...j', kappa, ebf)[0, :, None, :]
kebf = insert_strided_axis(aux, 0, n_el)
if diff_var is None:
econn = state.field.get_econn('volume', self.region)
adc = create_adof_conn(nm.arange(state.n_dof, dtype=nm.int32),
econn, n_c, 0)
vals = state()[adc]
aux = dot_sequences(kebf, vals[:, None, :, None])
out_qp = dot_sequences(kebf, aux, 'ATB')
fmode = 0
else:
out_qp = dot_sequences(kebf, kebf, 'ATB')
fmode = 1
return out_qp, geo, fmode
def get_fargs(self, kappa, virtual, state,
mode=None, term_mode=None, diff_var=None, **kwargs):
from sfepy.discrete.variables import create_adof_conn, expand_basis
geo, _ = self.get_mapping(state)
n_el, n_qp, dim, n_en, n_c = self.get_data_shape(virtual)
ebf = expand_basis(geo.bf, dim)
aux = nm.einsum('i,...ij->...j', kappa, ebf)[0, :, None, :]
kebf = insert_strided_axis(aux, 0, n_el)
if diff_var is None:
econn = state.field.get_econn('volume', self.region)
adc = create_adof_conn(nm.arange(state.n_dof, dtype=nm.int32),
econn, n_c, 0)
vals = state()[adc]
aux = dot_sequences(kebf, vals[:, None, :, None])
out_qp = dot_sequences(kebf, aux, 'ATB')
fmode = 0
else:
out_qp = dot_sequences(kebf, kebf, 'ATB')
fmode = 1
return out_qp, geo, fmode
def get_fargs(self, mat, kappa, virtual, state,
mode=None, term_mode=None, diff_var=None, **kwargs):
from sfepy.discrete.variables import create_adof_conn, expand_basis
geo, _ = self.get_mapping(state)
n_el, n_qp, dim, n_en, n_c = self.get_data_shape(virtual)
ebf = expand_basis(geo.bf, dim)
gmat = _build_wave_strain_op(kappa, ebf)
if diff_var is None:
econn = state.field.get_econn('volume', self.region)
adc = create_adof_conn(nm.arange(state.n_dof, dtype=nm.int32),
econn, n_c, 0)
vals = state()[adc]
# Same as nm.einsum('qij,cj->cqi', gmat[0], vals)[..., None]
aux = dot_sequences(gmat, vals[:, None, :, None])
out_qp = dot_sequences(gmat, dot_sequences(mat, aux), 'ATB')
fmode = 0
else:
out_qp = dot_sequences(gmat, dot_sequences(mat, gmat), 'ATB')
fmode = 1
return out_qp, geo, fmode
def get_fargs(self,
kappa,
kvar,
dvar,
mode=None,
term_mode=None,
diff_var=None,
**kwargs):
from sfepy.discrete.variables import create_adof_conn, expand_basis
geo, _ = self.get_mapping(dvar)
n_el, n_qp, dim, n_en, n_c = self.get_data_shape(kvar)
ebf = expand_basis(geo.bf, dim)
aux = nm.einsum('i,...ij->...j', kappa, ebf)[0, :, None, :]
kebf = insert_strided_axis(aux, 0, n_el)
div_bf = geo.bfg
div_bf = div_bf.reshape((n_el, n_qp, 1, dim * n_en))
div_bf = nm.ascontiguousarray(div_bf)
if diff_var is None:
avar = dvar if self.mode == 'kd' else kvar
econn = avar.field.get_econn('volume', self.region)
adc = create_adof_conn(nm.arange(avar.n_dof, dtype=nm.int32),
econn, n_c, 0)
vals = avar()[adc]
if self.mode == 'kd':
aux = dot_sequences(div_bf, vals[:, None, :, None])
out_qp = dot_sequences(kebf, aux, 'ATB')
else:
aux = dot_sequences(kebf, vals[:, None, :, None])
out_qp = dot_sequences(div_bf, aux, 'ATB')
fmode = 0
else:
if self.mode == 'kd':
out_qp = dot_sequences(kebf, div_bf, 'ATB')
else:
out_qp = dot_sequences(div_bf, kebf, 'ATB')
fmode = 1
return out_qp, geo, fmode
def get_fargs(self,
mat,
kappa,
gvar,
evar,
mode=None,
term_mode=None,
diff_var=None,
**kwargs):
from sfepy.discrete.variables import create_adof_conn, expand_basis
geo, _ = self.get_mapping(evar)
n_el, n_qp, dim, n_en, n_c = self.get_data_shape(gvar)
ebf = expand_basis(geo.bf, dim)
mat = Term.tile_mat(mat, n_el)
gmat = _build_wave_strain_op(kappa, ebf)
emat = _build_cauchy_strain_op(geo.bfg)
if diff_var is None:
avar = evar if self.mode == 'ge' else gvar
econn = avar.field.get_econn('volume', self.region)
adc = create_adof_conn(nm.arange(avar.n_dof, dtype=nm.int32),
econn, n_c, 0)
vals = avar()[adc]
if self.mode == 'ge':
# Same as aux = self.get(avar, 'cauchy_strain'),
aux = dot_sequences(emat, vals[:, None, :, None])
out_qp = dot_sequences(gmat, dot_sequences(mat, aux), 'ATB')
else:
aux = dot_sequences(gmat, vals[:, None, :, None])
out_qp = dot_sequences(emat, dot_sequences(mat, aux), 'ATB')
fmode = 0
else:
if self.mode == 'ge':
out_qp = dot_sequences(gmat, dot_sequences(mat, emat), 'ATB')
else:
out_qp = dot_sequences(emat, dot_sequences(mat, gmat), 'ATB')
fmode = 1
return out_qp, geo, fmode
def get_fargs(self, mat, vvar, svar,
mode=None, term_mode=None, diff_var=None, **kwargs):
from sfepy.discrete.variables import expand_basis
if diff_var is None:
val_qp = self.get(svar, 'val')
else:
val_qp = nm.array([0], ndmin=4, dtype=nm.float64)
sg, _ = self.get_mapping(vvar)
n_fa, n_qp, dim, n_fn, n_c = self.get_data_shape(vvar)
ebf = expand_basis(sg.bf, dim)
return val_qp, ebf, mat, sg, diff_var
def get_fargs(self, kappa, kvar, dvar,
mode=None, term_mode=None, diff_var=None, **kwargs):
from sfepy.discrete.variables import create_adof_conn, expand_basis
geo, _ = self.get_mapping(dvar)
n_el, n_qp, dim, n_en, n_c = self.get_data_shape(kvar)
ebf = expand_basis(geo.bf, dim)
aux = nm.einsum('i,...ij->...j', kappa, ebf)[0, :, None, :]
kebf = insert_strided_axis(aux, 0, n_el)
div_bf = geo.bfg
div_bf = div_bf.reshape((n_el, n_qp, 1, dim * n_en))
div_bf = nm.ascontiguousarray(div_bf)
if diff_var is None:
avar = dvar if self.mode == 'kd' else kvar
econn = avar.field.get_econn('volume', self.region)
adc = create_adof_conn(nm.arange(avar.n_dof, dtype=nm.int32),
econn, n_c, 0)
vals = avar()[adc]
if self.mode == 'kd':
aux = dot_sequences(div_bf, vals[:, None, :, None])
out_qp = dot_sequences(kebf, aux, 'ATB')
else:
aux = dot_sequences(kebf, vals[:, None, :, None])
out_qp = dot_sequences(div_bf, aux, 'ATB')
fmode = 0
else:
if self.mode == 'kd':
out_qp = dot_sequences(kebf, div_bf, 'ATB')
else:
out_qp = dot_sequences(div_bf, kebf, 'ATB')
fmode = 1
return out_qp, geo, fmode
def get_fargs(self, mat, kappa, gvar, evar,
mode=None, term_mode=None, diff_var=None, **kwargs):
from sfepy.discrete.variables import create_adof_conn, expand_basis
geo, _ = self.get_mapping(evar)
n_el, n_qp, dim, n_en, n_c = self.get_data_shape(gvar)
ebf = expand_basis(geo.bf, dim)
gmat = _build_wave_strain_op(kappa, ebf)
emat = _build_cauchy_strain_op(geo.bfg)
if diff_var is None:
avar = evar if self.mode == 'ge' else gvar
econn = avar.field.get_econn('volume', self.region)
adc = create_adof_conn(nm.arange(avar.n_dof, dtype=nm.int32),
econn, n_c, 0)
vals = avar()[adc]
if self.mode == 'ge':
# Same as aux = self.get(avar, 'cauchy_strain'),
aux = dot_sequences(emat, vals[:, None, :, None])
out_qp = dot_sequences(gmat, dot_sequences(mat, aux), 'ATB')
else:
aux = dot_sequences(gmat, vals[:, None, :, None])
out_qp = dot_sequences(emat, dot_sequences(mat, aux), 'ATB')
fmode = 0
else:
if self.mode == 'ge':
out_qp = dot_sequences(gmat, dot_sequences(mat, emat), 'ATB')
else:
out_qp = dot_sequences(emat, dot_sequences(mat, gmat), 'ATB')
fmode = 1
return out_qp, geo, fmode
def get_fargs(self, mat, vvar, svar,
mode=None, term_mode=None, diff_var=None, **kwargs):
from sfepy.discrete.variables import expand_basis
if self.mode == 'grad':
qp_var = svar
else:
qp_var = vvar
val_qp = self.get(qp_var, 'val')
vsg, _ = self.get_mapping(vvar)
ssg, _ = self.get_mapping(svar)
n_fa, n_qp, dim, n_fn, n_c = self.get_data_shape(vvar)
if mat is None:
mat = nm.ones((1, n_qp, 1, 1), dtype=nm.float64)
ebf = expand_basis(vsg.bf, dim)
return val_qp, ebf, ssg.bf, mat, vsg, diff_var, self.mode
