xref: /petsc/src/dm/impls/forest/p4est/petsc_p4est_package.c (revision d02705aa8a2222aec616378ef3d98c912b847a94)
1 #include <petsc/private/petscimpl.h>
2 #include <p4est_base.h>
3 #include "petsc_p4est_package.h"
4 
5 static const char *const SCLogTypes[] = {"DEFAULT","ALWAYS","TRACE","DEBUG","VERBOSE","INFO","STATISTICS","PRODUCTION","ESSENTIAL","ERROR","SILENT","SCLogTypes","SC_LP_",0};
6 
7 static PetscBool PetscP4estInitialized = PETSC_FALSE;
8 static PetscBool PetscBeganSc          = PETSC_FALSE;
9 static PetscClassId P4ESTLOGGING_CLASSID;
10 
11 PetscObject P4estLoggingObject; /* Just a vehicle for its classid */
12 
13 #undef __FUNCT__
14 #define __FUNCT__ "PetscScLogHandler"
15 static void PetscScLogHandler (FILE *log_stream, const char *filename, int lineno,
16                                int package, int category,
17                                int priority, const char *msg)
18 {
19   PetscInfo_Private(filename,P4estLoggingObject,":%d{%s} %s",lineno,package == sc_package_id ? "sc" : package == p4est_package_id ? "p4est" : "",msg);
20 }
21 
22 /* p4est tries to abort: if possible, use setjmp to enable at least a little unwinding */
23 #if defined(PETSC_HAVE_SETJMP_H) && defined(PETSC_USE_ERRORCHECKING)
24 #include <setjmp.h>
25 PETSC_VISIBILITY_INTERNAL jmp_buf PetscScJumpBuf;
26 PETSC_INTERN void PetscScAbort_longjmp(void)
27 {
28   longjmp(PetscScJumpBuf,1);
29   return;
30 }
31 
32 #define PetscScAbort PetscScAbort_longjmp
33 #else
34 #define PetscScAbort NULL
35 #endif
36 
37 #undef __FUNCT__
38 #define __FUNCT__ "PetscP4estFinalize"
39 static PetscErrorCode PetscP4estFinalize(void)
40 {
41   PetscErrorCode ierr;
42 
43   PetscFunctionBegin;
44   if (PetscBeganSc) {
45     PetscStackCallP4est(sc_finalize,());
46   }
47   ierr = PetscHeaderDestroy(&P4estLoggingObject);CHKERRQ(ierr);
48   PetscFunctionReturn(0);
49 }
50 
51 #undef __FUNCT__
52 #define __FUNCT__ "PetscP4estInitialize"
53 PetscErrorCode PetscP4estInitialize(void)
54 {
55   PetscBool      psc_catch_signals    = PETSC_TRUE;
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], *className;
60   PetscBool      opt;
61   PetscErrorCode ierr;
62 
63   PetscFunctionBegin;
64   if (PetscP4estInitialized) PetscFunctionReturn(0);
65   PetscP4estInitialized = PETSC_TRUE;
66   ierr = PetscClassIdRegister("p4est logging",&P4ESTLOGGING_CLASSID);CHKERRQ(ierr);
67   ierr = PetscOptionsGetString(NULL, "-info_exclude", logList, 256, &opt);CHKERRQ(ierr);
68   if (opt) {
69     ierr = PetscStrstr(logList, "p4est", &className);CHKERRQ(ierr);
70     if (className) {
71       ierr = PetscInfoDeactivateClass(P4ESTLOGGING_CLASSID);CHKERRQ(ierr);
72     }
73   }
74   ierr = PetscHeaderCreate(P4estLoggingObject,P4ESTLOGGING_CLASSID,"p4est","p4est logging","DM",PETSC_COMM_WORLD,NULL,PetscObjectView);CHKERRQ(ierr);
75   if (sc_package_id == -1) {
76     int log_threshold_shifted = psc_log_threshold + 1;
77     PetscBool set;
78 
79     PetscBeganSc = PETSC_TRUE;
80     ierr = PetscOptionsGetBool(NULL,"-petsc_sc_catch_signals",&psc_catch_signals,NULL);CHKERRQ(ierr);
81     ierr = PetscOptionsGetBool(NULL,"-petsc_sc_print_backtrace",&psc_print_backtrace,NULL);CHKERRQ(ierr);
82     ierr = PetscOptionsGetEnum(NULL,"-petsc_sc_log_threshold",SCLogTypes,(PetscEnum*)&log_threshold_shifted,&set);CHKERRQ(ierr);
83     if (set) {
84       psc_log_threshold = log_threshold_shifted - 1;
85     }
86     sc_init(PETSC_COMM_WORLD,(int)psc_catch_signals,(int)psc_print_backtrace,PetscScLogHandler,psc_log_threshold);
87     if (sc_package_id == -1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_LIB,"Could not initialize libsc package used by p4est");
88     sc_set_abort_handler(PetscScAbort);
89   }
90   if (p4est_package_id == -1) {
91     int log_threshold_shifted = pp4est_log_threshold + 1;
92     PetscBool set;
93 
94     ierr = PetscOptionsGetEnum(NULL,"-petsc_p4est_log_threshold",SCLogTypes,(PetscEnum*)&log_threshold_shifted,&set);CHKERRQ(ierr);
95     if (set) {
96       pp4est_log_threshold = log_threshold_shifted - 1;
97     }
98     PetscStackCallP4est(p4est_init,(PetscScLogHandler,pp4est_log_threshold));
99     if (p4est_package_id == -1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_LIB,"Could not initialize p4est");
100   }
101   ierr = PetscRegisterFinalize(PetscP4estFinalize);CHKERRQ(ierr);
102   PetscFunctionReturn(0);
103 }
104