1dc936754SJeremy L Thompson // Copyright (c) 2017-2024, Lawrence Livermore National Security, LLC and other CEED contributors. 218fb7758SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 318fb7758SJeremy L Thompson // 418fb7758SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 518fb7758SJeremy L Thompson // 618fb7758SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 718fb7758SJeremy L Thompson #pragma once 858600ac3SJames Wright 958600ac3SJames Wright #include <ceed.h> 10*40d80af1SJames Wright #include <petsc-ceed.h> 1158600ac3SJames Wright #include <petscdm.h> 1258600ac3SJames Wright #include <petscmat.h> 1358600ac3SJames Wright 1458600ac3SJames Wright #define MATCEED "ceed" 1558600ac3SJames Wright 16*40d80af1SJames Wright // Core functionality 17*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedCreate(DM dm_x, DM dm_y, CeedOperator op_mult, CeedOperator op_mult_transpose, Mat *mat); 18*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedCopy(Mat mat_ceed, Mat mat_other); 19*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedCreateMatCOO(Mat mat_ceed, Mat *mat_coo); 20*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedSetPreallocationCOO(Mat mat_ceed, Mat mat_coo); 21*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedAssembleCOO(Mat mat_ceed, Mat mat_coo); 2258600ac3SJames Wright 23*40d80af1SJames Wright PETSC_CEED_INTERN PetscErrorCode MatCeedSetContextDouble(Mat mat, const char *name, double value); 24*40d80af1SJames Wright PETSC_CEED_INTERN PetscErrorCode MatCeedGetContextDouble(Mat mat, const char *name, double *value); 2558600ac3SJames Wright 26*40d80af1SJames Wright // Advanced functionality 27*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedSetContext(Mat mat, PetscErrorCode (*f)(void *), void *ctx); 28*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedGetContext(Mat mat, void *ctx); 29*40d80af1SJames Wright 30*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedSetOperation(Mat mat, MatOperation op, void (*g)(void)); 31*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedSetCOOMatType(Mat mat, MatType type); 32*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedGetCOOMatType(Mat mat, MatType *type); 33*40d80af1SJames Wright 34*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedSetLocalVectors(Mat mat, Vec X_loc, Vec Y_loc_transpose); 35*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedGetLocalVectors(Mat mat, Vec *X_loc, Vec *Y_loc_transpose); 36*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedRestoreLocalVectors(Mat mat, Vec *X_loc, Vec *Y_loc_transpose); 37*40d80af1SJames Wright 38*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedGetCeedOperators(Mat mat, CeedOperator *op_mult, CeedOperator *op_mult_transpose); 39*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedRestoreCeedOperators(Mat mat, CeedOperator *op_mult, CeedOperator *op_mult_transpose); 40*40d80af1SJames Wright 41*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedSetLogEvents(Mat mat, PetscLogEvent log_event_mult, PetscLogEvent log_event_mult_transpose); 42*40d80af1SJames Wright PETSC_CEED_EXTERN PetscErrorCode MatCeedGetLogEvents(Mat mat, PetscLogEvent *log_event_mult, PetscLogEvent *log_event_mult_transpose); 43