def test_boundary_fluxes( self ): import os.path as op from sfepy.linalg import rotation_matrix2d from sfepy.discrete.evaluate import BasicEvaluator from sfepy.discrete import Material problem = self.problem angles = [0, 30, 45] region_names = ['Left', 'Right', 'Gamma'] values = [5.0, -5.0, 0.0] variables = problem.get_variables() get_state = variables.get_state_part_view state = self.state.copy(deep=True) problem.time_update(ebcs={}, epbcs={}) # problem.save_ebc( 'aux.vtk' ) state.apply_ebc() ev = BasicEvaluator( problem ) aux = ev.eval_residual(state()) field = variables['t'].field conf_m = problem.conf.get_item_by_name('materials', 'm') m = Material.from_conf(conf_m, problem.functions) name = op.join( self.options.out_dir, op.split( problem.domain.mesh.name )[1] + '_%02d.mesh' ) orig_coors = problem.get_mesh_coors().copy() ok = True for ia, angle in enumerate( angles ): self.report( '%d: mesh rotation %d degrees' % (ia, angle) ) problem.domain.mesh.transform_coors( rotation_matrix2d( angle ), ref_coors = orig_coors ) problem.set_mesh_coors(problem.domain.mesh.coors, update_fields=True) problem.domain.mesh.write( name % angle, io = 'auto' ) for ii, region_name in enumerate( region_names ): flux_term = 'd_surface_flux.i.%s( m.K, t )' % region_name val1 = problem.evaluate(flux_term, t=variables['t'], m=m) rvec = get_state( aux, 't', True ) reg = problem.domain.regions[region_name] nods = field.get_dofs_in_region(reg, merge=True) val2 = rvec[nods].sum() # Assume 1 dof per node. ok = ok and ((abs( val1 - values[ii] ) < 1e-10) and (abs( val2 - values[ii] ) < 1e-10)) self.report( ' %d. %s: %e == %e == %e'\ % (ii, region_name, val1, val2, values[ii]) ) # Restore original coordinates. problem.domain.mesh.transform_coors(rotation_matrix2d(0), ref_coors=orig_coors) problem.set_mesh_coors(problem.domain.mesh.coors, update_fields=True) return ok
def test_boundary_fluxes(self): import os.path as op from sfepy.linalg import rotation_matrix2d from sfepy.fem.evaluate import BasicEvaluator from sfepy.fem import Material problem = self.problem angles = [0, 30, 45] region_names = ['Left', 'Right', 'Gamma'] values = [5.0, -5.0, 0.0] variables = problem.get_variables() get_state = variables.get_state_part_view state = self.state.copy(deep=True) problem.time_update(ebcs={}, epbcs={}) # problem.save_ebc( 'aux.vtk' ) state.apply_ebc() ev = BasicEvaluator(problem) aux = ev.eval_residual(state()) field = variables['t'].field conf_m = problem.conf.get_item_by_name('materials', 'm') m = Material.from_conf(conf_m, problem.functions) name = op.join(self.options.out_dir, op.split(problem.domain.mesh.name)[1] + '_%02d.mesh') orig_coors = problem.get_mesh_coors().copy() ok = True for ia, angle in enumerate(angles): self.report('%d: mesh rotation %d degrees' % (ia, angle)) problem.domain.mesh.transform_coors(rotation_matrix2d(angle), ref_coors=orig_coors) problem.set_mesh_coors(problem.domain.mesh.coors, update_fields=True) problem.domain.mesh.write(name % angle, io='auto') for ii, region_name in enumerate(region_names): flux_term = 'd_surface_flux.i2.%s( m.K, t )' % region_name val1 = problem.evaluate(flux_term, t=variables['t'], m=m) rvec = get_state(aux, 't', True) reg = problem.domain.regions[region_name] nods = field.get_dofs_in_region(reg, merge=True) val2 = rvec[nods].sum() # Assume 1 dof per node. ok = ok and ((abs(val1 - values[ii]) < 1e-10) and (abs(val2 - values[ii]) < 1e-10)) self.report( ' %d. %s: %e == %e == %e'\ % (ii, region_name, val1, val2, values[ii]) ) # Restore original coordinates. problem.domain.mesh.transform_coors(rotation_matrix2d(0), ref_coors=orig_coors) problem.set_mesh_coors(problem.domain.mesh.coors, update_fields=True) return ok
def move_bottom(ts, coor, **kwargs): from sfepy.linalg import rotation_matrix2d vec = coor[:, 0:2] - centre angle = 3 * ts.step print('angle:', angle) mtx = rotation_matrix2d(angle) out = nm.dot(vec, mtx) - vec return out
def rotate_yz(ts, coor, **kwargs): from sfepy.linalg import rotation_matrix2d vec = coor[:, 1:3] - centre angle = 10.0 * ts.step print 'angle:', angle mtx = rotation_matrix2d(angle) vec_rotated = nm.dot(vec, mtx) displacement = vec_rotated - vec return displacement.T.flat
def rotate_yz(ts, coor, **kwargs): from sfepy.linalg import rotation_matrix2d vec = coor[:,1:3] - centre angle = 10.0 * ts.step print 'angle:', angle mtx = rotation_matrix2d( angle ) vec_rotated = nm.dot( vec, mtx ) displacement = vec_rotated - vec return displacement.T.flat
def test_boundary_fluxes( self ): import os.path as op from sfepy.base.base import Struct from sfepy.linalg import rotation_matrix2d from sfepy.fem.evaluate import BasicEvaluator problem = self.problem vec = self.vec angles = [0, 30, 45] region_names = ['Left', 'Right', 'Gamma'] values = [5.0, -5.0, 0.0] variables = problem.get_variables() get_state = variables.get_state_part_view state = vec.copy() problem.time_update(ebcs={}, epbcs={}) # problem.save_ebc( 'aux.vtk' ) problem.apply_ebc( state ) ev = BasicEvaluator( problem ) aux = ev.eval_residual( state ) field = variables['t'].field name = op.join( self.options.out_dir, op.split( problem.domain.mesh.name )[1] + '_%02d.mesh' ) orig_coors = problem.get_mesh_coors().copy() ok = True for ia, angle in enumerate( angles ): self.report( '%d: mesh rotation %d degrees' % (ia, angle) ) problem.domain.mesh.transform_coors( rotation_matrix2d( angle ), ref_coors = orig_coors ) problem.domain.mesh.write( name % angle, io = 'auto' ) for ii, region_name in enumerate( region_names ): flux_term = 'd_hdpm_surfdvel.i2.%s( m.K, t )' % region_name val1 = problem.evaluate(flux_term, t=variables['t']) rvec = get_state( aux, 't', True ) reg = problem.domain.regions[region_name] nods = reg.get_field_nodes( field, merge = True ) val2 = rvec[nods].sum() # Assume 1 dof per node. ok = ok and ((abs( val1 - values[ii] ) < 1e-10) and (abs( val2 - values[ii] ) < 1e-10)) self.report( ' %d. %s: %e == %e == %e'\ % (ii, region_name, val1, val2, values[ii]) ) return ok