1af0996ceSBarry Smith #include <petsc/private/sfimpl.h> 295fce210SBarry Smith 395fce210SBarry Smith static PetscBool PetscSFPackageInitialized = PETSC_FALSE; 495fce210SBarry Smith 529046d53SLisandro Dalcin PetscClassId PETSCSF_CLASSID; 629046d53SLisandro Dalcin 729046d53SLisandro Dalcin PetscLogEvent PETSCSF_SetGraph; 829046d53SLisandro Dalcin PetscLogEvent PETSCSF_SetUp; 929046d53SLisandro Dalcin PetscLogEvent PETSCSF_BcastBegin; 1029046d53SLisandro Dalcin PetscLogEvent PETSCSF_BcastEnd; 11*3482bfa8SJunchao Zhang PetscLogEvent PETSCSF_BcastAndOpBegin; 12*3482bfa8SJunchao Zhang PetscLogEvent PETSCSF_BcastAndOpEnd; 1329046d53SLisandro Dalcin PetscLogEvent PETSCSF_ReduceBegin; 1429046d53SLisandro Dalcin PetscLogEvent PETSCSF_ReduceEnd; 1529046d53SLisandro Dalcin PetscLogEvent PETSCSF_FetchAndOpBegin; 1629046d53SLisandro Dalcin PetscLogEvent PETSCSF_FetchAndOpEnd; 1795fce210SBarry Smith 1895fce210SBarry Smith /*@C 1995fce210SBarry Smith PetscSFInitializePackage - Initialize SF package 2095fce210SBarry Smith 2195fce210SBarry Smith Logically Collective 2295fce210SBarry Smith 2395fce210SBarry Smith Level: developer 2495fce210SBarry Smith 2595fce210SBarry Smith .seealso: PetscSFFinalizePackage() 2695fce210SBarry Smith @*/ 27607a6623SBarry Smith PetscErrorCode PetscSFInitializePackage(void) 2895fce210SBarry Smith { 2929046d53SLisandro Dalcin char logList[256]; 308e81d068SLisandro Dalcin PetscBool opt,pkg; 3195fce210SBarry Smith PetscErrorCode ierr; 3295fce210SBarry Smith 3395fce210SBarry Smith PetscFunctionBegin; 3495fce210SBarry Smith if (PetscSFPackageInitialized) PetscFunctionReturn(0); 3595fce210SBarry Smith PetscSFPackageInitialized = PETSC_TRUE; 368e81d068SLisandro Dalcin /* Register Class */ 373c2f6d5dSBarry Smith ierr = PetscClassIdRegister("Star Forest Graph", &PETSCSF_CLASSID);CHKERRQ(ierr); 388e81d068SLisandro Dalcin /* Register Constructors */ 39607a6623SBarry Smith ierr = PetscSFRegisterAll();CHKERRQ(ierr); 408e81d068SLisandro Dalcin /* Register Events */ 41acd23793SJed Brown ierr = PetscLogEventRegister("SFSetGraph" , PETSCSF_CLASSID, &PETSCSF_SetGraph);CHKERRQ(ierr); 4229046d53SLisandro Dalcin ierr = PetscLogEventRegister("SFSetUp" , PETSCSF_CLASSID, &PETSCSF_SetUp);CHKERRQ(ierr); 43acd23793SJed Brown ierr = PetscLogEventRegister("SFBcastBegin" , PETSCSF_CLASSID, &PETSCSF_BcastBegin);CHKERRQ(ierr); 44acd23793SJed Brown ierr = PetscLogEventRegister("SFBcastEnd" , PETSCSF_CLASSID, &PETSCSF_BcastEnd);CHKERRQ(ierr); 45*3482bfa8SJunchao Zhang ierr = PetscLogEventRegister("SFBcastOpBegin" , PETSCSF_CLASSID, &PETSCSF_BcastAndOpBegin);CHKERRQ(ierr); 46*3482bfa8SJunchao Zhang ierr = PetscLogEventRegister("SFBcastOpEnd" , PETSCSF_CLASSID, &PETSCSF_BcastAndOpEnd);CHKERRQ(ierr); 47acd23793SJed Brown ierr = PetscLogEventRegister("SFReduceBegin" , PETSCSF_CLASSID, &PETSCSF_ReduceBegin);CHKERRQ(ierr); 48acd23793SJed Brown ierr = PetscLogEventRegister("SFReduceEnd" , PETSCSF_CLASSID, &PETSCSF_ReduceEnd);CHKERRQ(ierr); 49acd23793SJed Brown ierr = PetscLogEventRegister("SFFetchOpBegin" , PETSCSF_CLASSID, &PETSCSF_FetchAndOpBegin);CHKERRQ(ierr); 50acd23793SJed Brown ierr = PetscLogEventRegister("SFFetchOpEnd" , PETSCSF_CLASSID, &PETSCSF_FetchAndOpEnd);CHKERRQ(ierr); 5129046d53SLisandro Dalcin /* Process info exclusions */ 5229046d53SLisandro Dalcin ierr = PetscOptionsGetString(NULL,NULL,"-info_exclude",logList,sizeof(logList),&opt);CHKERRQ(ierr); 5329046d53SLisandro Dalcin if (opt) { 548e81d068SLisandro Dalcin ierr = PetscStrInList("sf",logList,',',&pkg);CHKERRQ(ierr); 558e81d068SLisandro Dalcin if (pkg) {ierr = PetscInfoDeactivateClass(PETSCSF_CLASSID);CHKERRQ(ierr);} 5629046d53SLisandro Dalcin } 5729046d53SLisandro Dalcin /* Process summary exclusions */ 5829046d53SLisandro Dalcin ierr = PetscOptionsGetString(NULL,NULL,"-log_exclude",logList,sizeof(logList),&opt);CHKERRQ(ierr); 5929046d53SLisandro Dalcin if (opt) { 608e81d068SLisandro Dalcin ierr = PetscStrInList("sf",logList,',',&pkg);CHKERRQ(ierr); 61fa2bb9feSLisandro Dalcin if (pkg) {ierr = PetscLogEventExcludeClass(PETSCSF_CLASSID);CHKERRQ(ierr);} 6229046d53SLisandro Dalcin } 638e81d068SLisandro Dalcin /* Register package finalizer */ 6495fce210SBarry Smith ierr = PetscRegisterFinalize(PetscSFFinalizePackage);CHKERRQ(ierr); 6595fce210SBarry Smith PetscFunctionReturn(0); 6695fce210SBarry Smith } 6795fce210SBarry Smith 6895fce210SBarry Smith /*@C 6995fce210SBarry Smith PetscSFFinalizePackage - Finalize PetscSF package, it is called from PetscFinalize() 7095fce210SBarry Smith 7195fce210SBarry Smith Logically Collective 7295fce210SBarry Smith 7395fce210SBarry Smith Level: developer 7495fce210SBarry Smith 7595fce210SBarry Smith .seealso: PetscSFInitializePackage() 7695fce210SBarry Smith @*/ 7795fce210SBarry Smith PetscErrorCode PetscSFFinalizePackage(void) 7895fce210SBarry Smith { 7937e93019SBarry Smith PetscErrorCode ierr; 8095fce210SBarry Smith 8195fce210SBarry Smith PetscFunctionBegin; 8237e93019SBarry Smith ierr = PetscFunctionListDestroy(&PetscSFList);CHKERRQ(ierr); 8395fce210SBarry Smith PetscSFPackageInitialized = PETSC_FALSE; 8493ef888dSBarry Smith PetscSFRegisterAllCalled = PETSC_FALSE; 8595fce210SBarry Smith PetscFunctionReturn(0); 8695fce210SBarry Smith } 87