1: #include <petscdmforest.h>
  2: #include <petsc/private/petscimpl.h>
  3: #include "petsc_p4est_package.h"

  5: static const char *const SCLogTypes[] = {"DEFAULT", "ALWAYS", "TRACE", "DEBUG", "VERBOSE", "INFO", "STATISTICS", "PRODUCTION", "ESSENTIAL", "ERROR", "SILENT", "SCLogTypes", "SC_LP_", NULL};

  7: static PetscBool    PetscP4estInitialized = PETSC_FALSE;
  8: static PetscBool    PetscBeganSc          = PETSC_FALSE;
  9: static PetscClassId P4ESTLOGGING_CLASSID;

 11: PetscObject P4estLoggingObject; /* Just a vehicle for its classid */

 13: static void PetscScLogHandler(FILE *log_stream, const char *filename, int lineno, int package, int category, int priority, const char *msg)
 14: {
 15:   PetscCallVoid(PetscInfo_Private(filename, P4estLoggingObject, ":%d{%s} %s", lineno, package == sc_package_id ? "sc" : package == p4est_package_id ? "p4est" : "", msg));
 16: }

 18: /* p4est tries to abort: if possible, use setjmp to enable at least a little unwinding */
 19: #if PetscDefined(HAVE_SETJMP_H) && PetscDefined(USE_DEBUG)
 20:   #include <setjmp.h>
 21: PETSC_VISIBILITY_INTERNAL jmp_buf PetscScJumpBuf;
 22: PETSC_INTERN void                 PetscScAbort_longjmp(void)
 23: {
 24:   PetscErrorCode ierr = PetscError(PETSC_COMM_SELF, -1, "p4est function", "p4est file", PETSC_ERR_LIB, PETSC_ERROR_INITIAL, "Error in p4est stack call\n");
 25:   (void)ierr;
 26:   longjmp(PetscScJumpBuf, 1);
 27:   return;
 28: }

 30:   #define PetscScAbort PetscScAbort_longjmp
 31: #else
 32:   #define PetscScAbort NULL
 33: #endif

 35: static PetscErrorCode PetscP4estFinalize(void)
 36: {
 37:   PetscFunctionBegin;
 38:   if (PetscBeganSc) {
 39:     /* We do not want libsc to abort on a mismatched allocation and prevent further PETSc unwinding */
 40:     PetscCallP4est(sc_package_set_abort_alloc_mismatch, sc_package_id, 0);
 41:     PetscCallP4est(sc_package_set_abort_alloc_mismatch, p4est_package_id, 0);
 42:     PetscCallP4est(sc_package_set_abort_alloc_mismatch, -1, 0);
 43:     PetscCallP4est(sc_finalize, );
 44:   }
 45:   PetscCall(PetscHeaderDestroy(&P4estLoggingObject));
 46:   PetscFunctionReturn(PETSC_SUCCESS);
 47: }

 49: #if defined(__GNUC__) && !defined(__clang__)
 50:   #pragma GCC diagnostic push
 51:   #pragma GCC diagnostic ignored "-Wclobbered"
 52: #endif
 53: PetscErrorCode PetscP4estInitialize(void)
 54: {
 55:   PetscBool psc_catch_signals    = PETSC_FALSE;
 56:   PetscBool psc_print_backtrace  = PETSC_TRUE;
 57:   int       psc_log_threshold    = SC_LP_DEFAULT;
 58:   int       pp4est_log_threshold = SC_LP_DEFAULT;
 59:   char      logList[256];
 60:   PetscBool opt, pkg;

 62:   PetscFunctionBegin;
 63:   if (PetscP4estInitialized) PetscFunctionReturn(PETSC_SUCCESS);
 64:   PetscP4estInitialized = PETSC_TRUE;

 66:   /* Register Classes */
 67:   PetscCall(PetscClassIdRegister("p4est logging", &P4ESTLOGGING_CLASSID));
 68:   /* Process Info */
 69:   {
 70:     PetscClassId classids[1];

 72:     classids[0] = P4ESTLOGGING_CLASSID;
 73:     PetscCall(PetscInfoProcessClass("p4est", 1, classids));
 74:   }
 75:   /* Process summary exclusions */
 76:   PetscCall(PetscOptionsGetString(NULL, NULL, "-log_exclude", logList, sizeof(logList), &opt));
 77:   if (opt) {
 78:     PetscCall(PetscStrInList("p4est", logList, ',', &pkg));
 79:     if (pkg) PetscCall(PetscLogEventExcludeClass(P4ESTLOGGING_CLASSID));
 80:   }
 81:   PetscCall(PetscHeaderCreate(P4estLoggingObject, P4ESTLOGGING_CLASSID, "p4est", "p4est logging", "DM", PETSC_COMM_WORLD, NULL, PetscObjectView));
 82:   if (sc_package_id == -1) {
 83:     int       log_threshold_shifted = psc_log_threshold + 1;
 84:     PetscBool set;
 85: #if PetscDefined(HAVE_MPIUNI)
 86:     sc_MPI_Comm comm_world = sc_MPI_COMM_WORLD;
 87: #else
 88:     MPI_Comm comm_world = PETSC_COMM_WORLD;
 89: #endif

 91:     PetscBeganSc = PETSC_TRUE;
 92:     PetscCall(PetscOptionsGetBool(NULL, NULL, "-petsc_sc_catch_signals", &psc_catch_signals, NULL));
 93:     PetscCall(PetscOptionsGetBool(NULL, NULL, "-petsc_sc_print_backtrace", &psc_print_backtrace, NULL));
 94:     PetscCall(PetscOptionsGetEnum(NULL, NULL, "-petsc_sc_log_threshold", SCLogTypes, (PetscEnum *)&log_threshold_shifted, &set));
 95:     if (set) psc_log_threshold = log_threshold_shifted - 1;
 96:     sc_init(comm_world, (int)psc_catch_signals, (int)psc_print_backtrace, PetscScLogHandler, psc_log_threshold);
 97:     PetscCheck(sc_package_id != -1, PETSC_COMM_WORLD, PETSC_ERR_LIB, "Could not initialize libsc package used by p4est");
 98:     sc_set_abort_handler(PetscScAbort);
 99:   }
100:   if (p4est_package_id == -1) {
101:     int       log_threshold_shifted = pp4est_log_threshold + 1;
102:     PetscBool set;

104:     PetscCall(PetscOptionsGetEnum(NULL, NULL, "-petsc_p4est_log_threshold", SCLogTypes, (PetscEnum *)&log_threshold_shifted, &set));
105:     if (set) pp4est_log_threshold = log_threshold_shifted - 1;
106:     PetscCallP4est(p4est_init, PetscScLogHandler, pp4est_log_threshold);
107:     PetscCheck(p4est_package_id != -1, PETSC_COMM_WORLD, PETSC_ERR_LIB, "Could not initialize p4est");
108:   }
109:   PetscCall(DMForestRegisterType(DMP4EST));
110:   PetscCall(DMForestRegisterType(DMP8EST));
111:   PetscCall(PetscRegisterFinalize(PetscP4estFinalize));
112:   PetscFunctionReturn(PETSC_SUCCESS);
113: }
114: #if defined(__GNUC__) && !defined(__clang__)
115:   #pragma GCC diagnostic pop
116: #endif