1 // Copyright (c) 2017-2026, Lawrence Livermore National Security, LLC and other CEED contributors.
2 // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3 //
4 // SPDX-License-Identifier: BSD-2-Clause
5 //
6 // This file is part of CEED: http://github.com/ceed
7
8 /// @file
9 /// Utility functions for using PETSc with libCEED
10 #pragma once
11
12 #include <ceed.h>
13 #include <petsc.h>
14
15 #include "structs.h"
16
17 CeedMemType MemTypeP2C(PetscMemType mtype);
18 PetscErrorCode VecP2C(Vec X_petsc, PetscMemType *mem_type, CeedVector x_ceed);
19 PetscErrorCode VecC2P(CeedVector x_ceed, PetscMemType mem_type, Vec X_petsc);
20 PetscErrorCode VecReadP2C(Vec X_petsc, PetscMemType *mem_type, CeedVector x_ceed);
21 PetscErrorCode VecReadC2P(CeedVector x_ceed, PetscMemType mem_type, Vec X_petsc);
22 PetscErrorCode Kershaw(DM dm_orig, PetscScalar eps);
23 PetscErrorCode SetupDMByDegree(DM dm, PetscInt p_degree, PetscInt q_extra, PetscInt num_comp_u, PetscInt topo_dim, bool enforce_bc);
24 PetscErrorCode CreateRestrictionFromPlex(Ceed ceed, DM dm, CeedInt height, DMLabel domain_label, CeedInt value, CeedElemRestriction *elem_restr);
25 CeedElemTopology ElemTopologyP2C(DMPolytopeType cell_type);
26 PetscErrorCode DMFieldToDSField(DM dm, DMLabel domain_label, PetscInt dm_field, PetscInt *ds_field);
27 PetscErrorCode BasisCreateFromTabulation(Ceed ceed, DM dm, DMLabel domain_label, PetscInt label_value, PetscInt height, PetscInt face, PetscFE fe,
28 PetscTabulation basis_tabulation, PetscQuadrature quadrature, CeedBasis *basis);
29 PetscErrorCode CreateBasisFromPlex(Ceed ceed, DM dm, DMLabel domain_label, CeedInt label_value, CeedInt height, CeedInt dm_field, BPData bp_data,
30 CeedBasis *basis);
31 PetscErrorCode CreateDistributedDM(RunParams rp, DM *dm);
32
33 /**
34 @brief Translate array of `PetscInt` to `CeedInt`.
35 If the types differ, `array_petsc` is freed with `PetscFree()` and `array_ceed` is allocated with `PetscMalloc1()`.
36 Caller is responsible for freeing `array_ceed` with `PetscFree()`.
37
38 Not collective across MPI processes.
39
40 @param[in] num_entries Number of array entries
41 @param[in,out] array_petsc Array of `PetscInt`
42 @param[out] array_ceed Array of `CeedInt`
43
44 @return An error code: 0 - success, otherwise - failure
45 **/
IntArrayCeedToPetsc(PetscInt num_entries,CeedInt ** array_ceed,PetscInt ** array_petsc)46 static inline PetscErrorCode IntArrayCeedToPetsc(PetscInt num_entries, CeedInt **array_ceed, PetscInt **array_petsc) {
47 const CeedInt int_c = 0;
48 const PetscInt int_p = 0;
49
50 PetscFunctionBeginUser;
51 if (sizeof(int_c) == sizeof(int_p)) {
52 *array_petsc = (PetscInt *)*array_ceed;
53 } else {
54 *array_petsc = malloc(num_entries * sizeof(PetscInt));
55 for (PetscInt i = 0; i < num_entries; i++) (*array_petsc)[i] = (*array_ceed)[i];
56 free(*array_ceed);
57 }
58 *array_ceed = NULL;
59 PetscFunctionReturn(PETSC_SUCCESS);
60 }
61
62 /**
63 @brief Translate array of `PetscInt` to `CeedInt`.
64 If the types differ, `array_petsc` is freed with `PetscFree()` and `array_ceed` is allocated with `PetscMalloc1()`.
65 Caller is responsible for freeing `array_ceed` with `PetscFree()`.
66
67 Not collective across MPI processes.
68
69 @param[in] num_entries Number of array entries
70 @param[in,out] array_petsc Array of `PetscInt`
71 @param[out] array_ceed Array of `CeedInt`
72
73 @return An error code: 0 - success, otherwise - failure
74 **/
IntArrayPetscToCeed(PetscInt num_entries,PetscInt ** array_petsc,CeedInt ** array_ceed)75 static inline PetscErrorCode IntArrayPetscToCeed(PetscInt num_entries, PetscInt **array_petsc, CeedInt **array_ceed) {
76 const CeedInt int_c = 0;
77 const PetscInt int_p = 0;
78
79 PetscFunctionBeginUser;
80 if (sizeof(int_c) == sizeof(int_p)) {
81 *array_ceed = (CeedInt *)*array_petsc;
82 } else {
83 PetscCall(PetscMalloc1(num_entries, array_ceed));
84 for (PetscInt i = 0; i < num_entries; i++) (*array_ceed)[i] = (*array_petsc)[i];
85 PetscCall(PetscFree(*array_petsc));
86 }
87 *array_petsc = NULL;
88 PetscFunctionReturn(PETSC_SUCCESS);
89 }
90