create_figure_scheduler = fk.schedules(times=[0., 1., 2., 3., 4., 5.]) triangular_mesh = fk.TriangularMesh.from_msh_file( '../simulations/inputs/square2.mesh') layer = fk.Layer(1, triangular_mesh, topography=0.) def H_0(x, y): h = 2.4 * (1.0 + np.exp(-0.25 * ((x - 10.05)**2 + (y - 10.05)**2))) return h primitives = fk.Primitive(triangular_mesh, layer, height_funct=H_0) tracers = [fk.Tracer(triangular_mesh, layer, primitives, Tinit=1.0)] slides = [fk.Slide(ref=r) for r in [1, 2, 3, 4]] vtk_writer = fk.VTKWriter(triangular_mesh, scheduler=fk.schedules(count=10), scale_h=1.) problem = fk.Problem(simutime, triangular_mesh, layer, primitives, slides=slides, numerical_parameters={'space_second_order': True}, tracers=tracers, vtk_writer=vtk_writer, restart_scheduler=restart_scheduler, custom_funct={'plot': fk_plt.plot_freesurface_3d}, custom_funct_scheduler=create_figure_scheduler) ################################################################################
angular_velocity=angular_velocity, phase=-np.pi) wheels = [ fk.Wheel(A, B, radius=RADIUS, magnitude=TORQUE, rotating=rotating), ] paddle_wheel_effect = fk.WheelEffect(wheels, triangular_mesh.vertices, layer) external_effects = {"wheel": paddle_wheel_effect} ################################################################################ # # Writter: #---------------------- # vtk_writer = fk.VTKWriter(triangular_mesh, scheduler=output_scheduler, scale_h=SCALE_H) ################################################################################ # # Problem definition: #-------------------- raceway = fk.Problem(simutime, triangular_mesh, layer, primitives, slides=slides, numerical_parameters=NUM_PARAMS, physical_parameters=PHY_PARAMS, external_effects=external_effects,
################################################################################ # #.. note:: # # * Once again, tracer labels are used to apply source terms on each one of the tracers. # * Note that tracer sources can be set in kg instead of [T]=kg/m3 in the case of species. The corresponding concentration source term is computed based on cell volume and mass source. ################################################################################ # # Writter: #---------------------- # NB_VTK = 10 vtk_writer = fk.VTKWriter(triangular_mesh, scheduler=fk.schedules(count=NB_VTK), scale_h=5.) ################################################################################ # # Problem definition: #-------------------- problem = fk.Problem(simutime, triangular_mesh, layer, primitives, slides=slides, fluvial_heights=fluvial_heights, fluvial_flowrates=fluvial_flowrates, tracers=tracers,