def set_arg_types(self): if self.mode == 'weak': self.function = terms.dw_lin_prestress use_method_with_name(self, self.get_fargs_weak, 'get_fargs') else: self.function = self.d_lin_prestress use_method_with_name(self, self.get_fargs_eval, 'get_fargs') self.use_caches = {'cauchy_strain' : [['parameter']]}
def set_arg_types(self): """Dynamically inherits from either BiotGradTH or BiotDivTH.""" if self.mode == "grad": self.function = terms.dw_biot_grad use_method_with_name(self, self.get_fargs_grad, "get_fargs") self.use_caches = {"state_in_volume_qp": [["state", {"state": (-1, -1)}]]} elif self.mode == "div": self.function = terms.dw_biot_div use_method_with_name(self, self.get_fargs_div, "get_fargs") self.use_caches = {"cauchy_strain": [["state", {"strain": (-1, -1)}]]} else: raise NotImplementedError