1*b6972d74SZach Atkins // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors. 2*b6972d74SZach Atkins // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 3*b6972d74SZach Atkins // 4*b6972d74SZach Atkins // SPDX-License-Identifier: BSD-2-Clause 5*b6972d74SZach Atkins // 6*b6972d74SZach Atkins // This file is part of CEED: http://github.com/ceed 7*b6972d74SZach Atkins 8*b6972d74SZach Atkins /// @file 9*b6972d74SZach Atkins /// Utility functions for particle-based methods with DMSwarm 10*b6972d74SZach Atkins 11*b6972d74SZach Atkins #pragma once 12*b6972d74SZach Atkins 13*b6972d74SZach Atkins #include <ceed.h> 14*b6972d74SZach Atkins #include <math.h> 15*b6972d74SZach Atkins #include <petscdmplex.h> 16*b6972d74SZach Atkins #include <petscdmswarm.h> 17*b6972d74SZach Atkins #include <petsc/private/petscfeimpl.h> /* For interpolation */ 18*b6972d74SZach Atkins 19*b6972d74SZach Atkins #include "petscutils.h" 20*b6972d74SZach Atkins 21*b6972d74SZach Atkins // libCEED context data 22*b6972d74SZach Atkins typedef struct DMSwarmCeedContext_ *DMSwarmCeedContext; 23*b6972d74SZach Atkins struct DMSwarmCeedContext_ { 24*b6972d74SZach Atkins Ceed ceed; 25*b6972d74SZach Atkins CeedVector u_mesh, v_mesh, u_points; 26*b6972d74SZach Atkins CeedOperator op_mass, op_mesh_to_points, op_points_to_mesh; 27*b6972d74SZach Atkins }; 28*b6972d74SZach Atkins 29*b6972d74SZach Atkins PetscErrorCode DMSwarmCeedContextCreate(DM dm_swarm, const char *ceed_resource, DMSwarmCeedContext *ctx); 30*b6972d74SZach Atkins PetscErrorCode DMSwarmCeedContextDestroy(DMSwarmCeedContext *ctx); 31*b6972d74SZach Atkins 32*b6972d74SZach Atkins // Swarm point distribution 33*b6972d74SZach Atkins typedef enum { SWARM_GAUSS = 0, SWARM_UNIFORM = 1, SWARM_CELL_RANDOM = 2, SWARM_SINUSOIDAL = 3 } PointSwarmType; 34*b6972d74SZach Atkins static const char *const point_swarm_types[] = {"gauss", "uniform", "cell_random", "sinusoidal", "PointSwarmType", "SWARM", 0}; 35*b6972d74SZach Atkins 36*b6972d74SZach Atkins // Memory utilities 37*b6972d74SZach Atkins PetscErrorCode DMSwarmPICFieldP2C(DM dm_swarm, const char *field, CeedVector x_ceed); 38*b6972d74SZach Atkins PetscErrorCode DMSwarmPICFieldC2P(DM dm_swarm, const char *field, CeedVector x_ceed); 39*b6972d74SZach Atkins 40*b6972d74SZach Atkins // Swarm helper function 41*b6972d74SZach Atkins PetscErrorCode DMSwarmInitalizePointLocations(DM dm_swarm, PointSwarmType point_swarm_type, PetscInt num_points, PetscInt num_points_per_cell); 42*b6972d74SZach Atkins PetscErrorCode DMSwarmCreateReferenceCoordinates(DM dm_swarm, IS *is_points, Vec *ref_coords); 43*b6972d74SZach Atkins 44*b6972d74SZach Atkins // Swarm to mesh projection 45*b6972d74SZach Atkins PetscErrorCode DMSwarmCreateProjectionRHS(DM dm_swarm, const char *field, Vec B_mesh); 46*b6972d74SZach Atkins PetscErrorCode MatMult_SwarmMass(Mat A, Vec U_mesh, Vec V_mesh); 47*b6972d74SZach Atkins PetscErrorCode DMSwarmProjectFromSwarmToCells(DM dm_swarm, const char *field, Vec U_mesh); 48