xref: /petsc/src/vec/is/sf/interface/dlregissf.c (revision 8e81d0682c925ac65a16095f60c998f9f3b7b642)
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;
1129046d53SLisandro Dalcin PetscLogEvent PETSCSF_ReduceBegin;
1229046d53SLisandro Dalcin PetscLogEvent PETSCSF_ReduceEnd;
1329046d53SLisandro Dalcin PetscLogEvent PETSCSF_FetchAndOpBegin;
1429046d53SLisandro Dalcin PetscLogEvent PETSCSF_FetchAndOpEnd;
1595fce210SBarry Smith 
1695fce210SBarry Smith /*@C
1795fce210SBarry Smith    PetscSFInitializePackage - Initialize SF package
1895fce210SBarry Smith 
1995fce210SBarry Smith    Logically Collective
2095fce210SBarry Smith 
2195fce210SBarry Smith    Level: developer
2295fce210SBarry Smith 
2395fce210SBarry Smith .seealso: PetscSFFinalizePackage()
2495fce210SBarry Smith @*/
25607a6623SBarry Smith PetscErrorCode PetscSFInitializePackage(void)
2695fce210SBarry Smith {
2729046d53SLisandro Dalcin   char           logList[256];
28*8e81d068SLisandro Dalcin   PetscBool      opt,pkg;
2995fce210SBarry Smith   PetscErrorCode ierr;
3095fce210SBarry Smith 
3195fce210SBarry Smith   PetscFunctionBegin;
3295fce210SBarry Smith   if (PetscSFPackageInitialized) PetscFunctionReturn(0);
3395fce210SBarry Smith   PetscSFPackageInitialized = PETSC_TRUE;
34*8e81d068SLisandro Dalcin   /* Register Class */
353c2f6d5dSBarry Smith   ierr = PetscClassIdRegister("Star Forest Graph", &PETSCSF_CLASSID);CHKERRQ(ierr);
36*8e81d068SLisandro Dalcin   /* Register Constructors */
37607a6623SBarry Smith   ierr = PetscSFRegisterAll();CHKERRQ(ierr);
38*8e81d068SLisandro Dalcin   /* Register Events */
39acd23793SJed Brown   ierr = PetscLogEventRegister("SFSetGraph"     , PETSCSF_CLASSID, &PETSCSF_SetGraph);CHKERRQ(ierr);
4029046d53SLisandro Dalcin   ierr = PetscLogEventRegister("SFSetUp"        , PETSCSF_CLASSID, &PETSCSF_SetUp);CHKERRQ(ierr);
41acd23793SJed Brown   ierr = PetscLogEventRegister("SFBcastBegin"   , PETSCSF_CLASSID, &PETSCSF_BcastBegin);CHKERRQ(ierr);
42acd23793SJed Brown   ierr = PetscLogEventRegister("SFBcastEnd"     , PETSCSF_CLASSID, &PETSCSF_BcastEnd);CHKERRQ(ierr);
43acd23793SJed Brown   ierr = PetscLogEventRegister("SFReduceBegin"  , PETSCSF_CLASSID, &PETSCSF_ReduceBegin);CHKERRQ(ierr);
44acd23793SJed Brown   ierr = PetscLogEventRegister("SFReduceEnd"    , PETSCSF_CLASSID, &PETSCSF_ReduceEnd);CHKERRQ(ierr);
45acd23793SJed Brown   ierr = PetscLogEventRegister("SFFetchOpBegin" , PETSCSF_CLASSID, &PETSCSF_FetchAndOpBegin);CHKERRQ(ierr);
46acd23793SJed Brown   ierr = PetscLogEventRegister("SFFetchOpEnd"   , PETSCSF_CLASSID, &PETSCSF_FetchAndOpEnd);CHKERRQ(ierr);
4729046d53SLisandro Dalcin   /* Process info exclusions */
4829046d53SLisandro Dalcin   ierr = PetscOptionsGetString(NULL,NULL,"-info_exclude",logList,sizeof(logList),&opt);CHKERRQ(ierr);
4929046d53SLisandro Dalcin   if (opt) {
50*8e81d068SLisandro Dalcin     ierr = PetscStrInList("sf",logList,',',&pkg);CHKERRQ(ierr);
51*8e81d068SLisandro Dalcin     if (pkg) {ierr = PetscInfoDeactivateClass(PETSCSF_CLASSID);CHKERRQ(ierr);}
5229046d53SLisandro Dalcin   }
5329046d53SLisandro Dalcin   /* Process summary exclusions */
5429046d53SLisandro Dalcin   ierr = PetscOptionsGetString(NULL,NULL,"-log_exclude",logList,sizeof(logList),&opt);CHKERRQ(ierr);
5529046d53SLisandro Dalcin   if (opt) {
56*8e81d068SLisandro Dalcin     ierr = PetscStrInList("sf",logList,',',&pkg);CHKERRQ(ierr);
57*8e81d068SLisandro Dalcin     if (pkg) {ierr = PetscLogEventDeactivateClass(PETSCSF_CLASSID);CHKERRQ(ierr);}
5829046d53SLisandro Dalcin   }
59*8e81d068SLisandro Dalcin   /* Register package finalizer */
6095fce210SBarry Smith   ierr = PetscRegisterFinalize(PetscSFFinalizePackage);CHKERRQ(ierr);
6195fce210SBarry Smith   PetscFunctionReturn(0);
6295fce210SBarry Smith }
6395fce210SBarry Smith 
6495fce210SBarry Smith /*@C
6595fce210SBarry Smith    PetscSFFinalizePackage - Finalize PetscSF package, it is called from PetscFinalize()
6695fce210SBarry Smith 
6795fce210SBarry Smith    Logically Collective
6895fce210SBarry Smith 
6995fce210SBarry Smith    Level: developer
7095fce210SBarry Smith 
7195fce210SBarry Smith .seealso: PetscSFInitializePackage()
7295fce210SBarry Smith @*/
7395fce210SBarry Smith PetscErrorCode PetscSFFinalizePackage(void)
7495fce210SBarry Smith {
7537e93019SBarry Smith   PetscErrorCode ierr;
7695fce210SBarry Smith 
7795fce210SBarry Smith   PetscFunctionBegin;
7837e93019SBarry Smith   ierr = PetscFunctionListDestroy(&PetscSFList);CHKERRQ(ierr);
7995fce210SBarry Smith   PetscSFPackageInitialized = PETSC_FALSE;
8093ef888dSBarry Smith   PetscSFRegisterAllCalled  = PETSC_FALSE;
8195fce210SBarry Smith   PetscFunctionReturn(0);
8295fce210SBarry Smith }
83