Actual source code: petsc_p4est_package.c
petsc-3.8.4 2018-03-24
1: #include <petscdmforest.h>
2: #include <petsc/private/petscimpl.h>
3: #include <p4est_base.h>
4: #include "petsc_p4est_package.h"
6: static const char*const SCLogTypes[] = {"DEFAULT","ALWAYS","TRACE","DEBUG","VERBOSE","INFO","STATISTICS","PRODUCTION","ESSENTIAL","ERROR","SILENT","SCLogTypes","SC_LP_",0};
8: static PetscBool PetscP4estInitialized = PETSC_FALSE;
9: static PetscBool PetscBeganSc = PETSC_FALSE;
10: static PetscClassId P4ESTLOGGING_CLASSID;
12: PetscObject P4estLoggingObject; /* Just a vehicle for its classid */
14: static void PetscScLogHandler(FILE *log_stream, const char *filename, int lineno,int package, int category,int priority, const char *msg)
15: {
16: PetscInfo_Private(filename,P4estLoggingObject,":%d{%s} %s",lineno,package == sc_package_id ? "sc" : package == p4est_package_id ? "p4est" : "",msg);
17: }
19: /* p4est tries to abort: if possible, use setjmp to enable at least a little unwinding */
20: #if defined(PETSC_HAVE_SETJMP_H) && defined(PETSC_USE_ERRORCHECKING)
21: #include <setjmp.h>
22: PETSC_VISIBILITY_INTERNAL jmp_buf PetscScJumpBuf;
23: PETSC_INTERN void PetscScAbort_longjmp(void)
24: {
25: PetscError(PETSC_COMM_SELF,-1,"p4est function","p4est file",PETSC_ERR_LIB,PETSC_ERROR_INITIAL,"Error in p4est stack call\n");
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: {
40: if (PetscBeganSc) {
41: /* We do not want libsc to abort on a mismatched allocation and prevent further Petsc unwinding */
42: PetscStackCallP4est(sc_package_set_abort_alloc_mismatch,(sc_package_id,0));
43: PetscStackCallP4est(sc_package_set_abort_alloc_mismatch,(p4est_package_id,0));
44: PetscStackCallP4est(sc_package_set_abort_alloc_mismatch,(-1,0));
45: PetscStackCallP4est(sc_finalize,());
46: }
47: PetscHeaderDestroy(&P4estLoggingObject);
48: return(0);
49: }
51: PetscErrorCode PetscP4estInitialize(void)
52: {
53: PetscBool psc_catch_signals = PETSC_FALSE;
54: PetscBool psc_print_backtrace = PETSC_TRUE;
55: int psc_log_threshold = SC_LP_DEFAULT;
56: int pp4est_log_threshold = SC_LP_DEFAULT;
57: char logList[256], *className;
58: PetscBool opt;
62: if (PetscP4estInitialized) return(0);
63: PetscP4estInitialized = PETSC_TRUE;
65: PetscClassIdRegister("p4est logging",&P4ESTLOGGING_CLASSID);
66: PetscOptionsGetString(NULL,NULL, "-info_exclude", logList, 256, &opt);
67: if (opt) {
68: PetscStrstr(logList, "p4est", &className);
69: if (className) {
70: PetscInfoDeactivateClass(P4ESTLOGGING_CLASSID);
71: }
72: }
73: PetscHeaderCreate(P4estLoggingObject,P4ESTLOGGING_CLASSID,"p4est","p4est logging","DM",PETSC_COMM_WORLD,NULL,PetscObjectView);
74: if (sc_package_id == -1) {
75: int log_threshold_shifted = psc_log_threshold + 1;
76: PetscBool set;
78: PetscBeganSc = PETSC_TRUE;
79: PetscOptionsGetBool(NULL,NULL,"-petsc_sc_catch_signals",&psc_catch_signals,NULL);
80: PetscOptionsGetBool(NULL,NULL,"-petsc_sc_print_backtrace",&psc_print_backtrace,NULL);
81: PetscOptionsGetEnum(NULL,NULL,"-petsc_sc_log_threshold",SCLogTypes,(PetscEnum*)&log_threshold_shifted,&set);
82: if (set) psc_log_threshold = log_threshold_shifted - 1;
83: sc_init(PETSC_COMM_WORLD,(int)psc_catch_signals,(int)psc_print_backtrace,PetscScLogHandler,psc_log_threshold);
84: if (sc_package_id == -1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_LIB,"Could not initialize libsc package used by p4est");
85: sc_set_abort_handler(PetscScAbort);
86: }
87: if (p4est_package_id == -1) {
88: int log_threshold_shifted = pp4est_log_threshold + 1;
89: PetscBool set;
91: PetscOptionsGetEnum(NULL,NULL,"-petsc_p4est_log_threshold",SCLogTypes,(PetscEnum*)&log_threshold_shifted,&set);
92: if (set) pp4est_log_threshold = log_threshold_shifted - 1;
93: PetscStackCallP4est(p4est_init,(PetscScLogHandler,pp4est_log_threshold));
94: if (p4est_package_id == -1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_LIB,"Could not initialize p4est");
95: }
96: DMForestRegisterType(DMP4EST);
97: DMForestRegisterType(DMP8EST);
98: PetscRegisterFinalize(PetscP4estFinalize);
99: return(0);
100: }