Actual source code: petsc_p4est_package.c
petsc-3.7.1 2016-05-15
Report Typos and Errors

      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 */

     16: static void PetscScLogHandler (FILE *log_stream, const char *filename, int lineno,
     17:                                int package, int category,
     18:                                int priority, const char *msg)
     19: {
     20:   PetscInfo_Private(filename,P4estLoggingObject,":%d{%s} %s",lineno,package == sc_package_id ? "sc" : package == p4est_package_id ? "p4est" : "",msg);
     21: }

     23: /* p4est tries to abort: if possible, use setjmp to enable at least a little unwinding */
     24: #if defined(PETSC_HAVE_SETJMP_H) && defined(PETSC_USE_ERRORCHECKING)
     25: #include <setjmp.h>
     26: PETSC_VISIBILITY_INTERNAL jmp_buf PetscScJumpBuf;
     27: PETSC_INTERN void PetscScAbort_longjmp(void)
     28: {
     29:   PetscError(PETSC_COMM_SELF,-1,"p4est function","p4est file",PETSC_ERR_LIB,PETSC_ERROR_INITIAL,"Error in p4est stack call\n");
     30:   longjmp(PetscScJumpBuf,1);
     31:   return;
     32: }

     34: #define PetscScAbort PetscScAbort_longjmp
     35: #else
     36: #define PetscScAbort NULL
     37: #endif

     41: static PetscErrorCode PetscP4estFinalize(void)
     42: {

     46:   if (PetscBeganSc) {
     47:     /* We do not want libsc to abort on a mismatched allocation and prevent further Petsc unwinding */
     48:     PetscStackCallP4est(sc_package_set_abort_alloc_mismatch,(sc_package_id,0));
     49:     PetscStackCallP4est(sc_package_set_abort_alloc_mismatch,(p4est_package_id,0));
     50:     PetscStackCallP4est(sc_package_set_abort_alloc_mismatch,(-1,0));
     51:     PetscStackCallP4est(sc_finalize,());
     52:   }
     53:   PetscHeaderDestroy(&P4estLoggingObject);
     54:   return(0);
     55: }

     59: PetscErrorCode PetscP4estInitialize(void)
     60: {
     61:   PetscBool      psc_catch_signals    = PETSC_FALSE;
     62:   PetscBool      psc_print_backtrace  = PETSC_TRUE;
     63:   int            psc_log_threshold    = SC_LP_DEFAULT;
     64:   int            pp4est_log_threshold = SC_LP_DEFAULT;
     65:   char           logList[256], *className;
     66:   PetscBool      opt;

     70:   if (PetscP4estInitialized) return(0);
     71:   PetscP4estInitialized = PETSC_TRUE;
     72:   PetscClassIdRegister("p4est logging",&P4ESTLOGGING_CLASSID);
     73:   PetscOptionsGetString(NULL,NULL, "-info_exclude", logList, 256, &opt);
     74:   if (opt) {
     75:     PetscStrstr(logList, "p4est", &className);
     76:     if (className) {
     77:       PetscInfoDeactivateClass(P4ESTLOGGING_CLASSID);
     78:     }
     79:   }
     80:   PetscHeaderCreate(P4estLoggingObject,P4ESTLOGGING_CLASSID,"p4est","p4est logging","DM",PETSC_COMM_WORLD,NULL,PetscObjectView);
     81:   if (sc_package_id == -1) {
     82:     int log_threshold_shifted = psc_log_threshold + 1;
     83:     PetscBool set;

     85:     PetscBeganSc = PETSC_TRUE;
     86:     PetscOptionsGetBool(NULL,NULL,"-petsc_sc_catch_signals",&psc_catch_signals,NULL);
     87:     PetscOptionsGetBool(NULL,NULL,"-petsc_sc_print_backtrace",&psc_print_backtrace,NULL);
     88:     PetscOptionsGetEnum(NULL,NULL,"-petsc_sc_log_threshold",SCLogTypes,(PetscEnum*)&log_threshold_shifted,&set);
     89:     if (set) {
     90:       psc_log_threshold = log_threshold_shifted - 1;
     91:     }
     92:     sc_init(PETSC_COMM_WORLD,(int)psc_catch_signals,(int)psc_print_backtrace,PetscScLogHandler,psc_log_threshold);
     93:     if (sc_package_id == -1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_LIB,"Could not initialize libsc package used by p4est");
     94:     sc_set_abort_handler(PetscScAbort);
     95:   }
     96:   if (p4est_package_id == -1) {
     97:     int log_threshold_shifted = pp4est_log_threshold + 1;
     98:     PetscBool set;

    100:     PetscOptionsGetEnum(NULL,NULL,"-petsc_p4est_log_threshold",SCLogTypes,(PetscEnum*)&log_threshold_shifted,&set);
    101:     if (set) {
    102:       pp4est_log_threshold = log_threshold_shifted - 1;
    103:     }
    104:     PetscStackCallP4est(p4est_init,(PetscScLogHandler,pp4est_log_threshold));
    105:     if (p4est_package_id == -1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_LIB,"Could not initialize p4est");
    106:   }
    107:   DMForestRegisterType(DMP4EST);
    108:   DMForestRegisterType(DMP8EST);
    109:   PetscRegisterFinalize(PetscP4estFinalize);
    110:   return(0);
    111: }