1 #include <petsc/private/sfimpl.h> 2 3 static PetscBool PetscSFPackageInitialized = PETSC_FALSE; 4 5 PetscClassId PETSCSF_CLASSID; 6 7 PetscLogEvent PETSCSF_SetGraph; 8 PetscLogEvent PETSCSF_SetUp; 9 PetscLogEvent PETSCSF_BcastBegin; 10 PetscLogEvent PETSCSF_BcastEnd; 11 PetscLogEvent PETSCSF_ReduceBegin; 12 PetscLogEvent PETSCSF_ReduceEnd; 13 PetscLogEvent PETSCSF_FetchAndOpBegin; 14 PetscLogEvent PETSCSF_FetchAndOpEnd; 15 PetscLogEvent PETSCSF_EmbedSF; 16 PetscLogEvent PETSCSF_DistSect; 17 PetscLogEvent PETSCSF_SectSF; 18 PetscLogEvent PETSCSF_RemoteOff; 19 PetscLogEvent PETSCSF_Pack; 20 PetscLogEvent PETSCSF_Unpack; 21 22 /*@C 23 PetscSFInitializePackage - Initialize SF package 24 25 Logically Collective 26 27 Level: developer 28 29 .seealso: PetscSFFinalizePackage() 30 @*/ 31 PetscErrorCode PetscSFInitializePackage(void) 32 { 33 char logList[256]; 34 PetscBool opt,pkg; 35 PetscErrorCode ierr; 36 37 PetscFunctionBegin; 38 if (PetscSFPackageInitialized) PetscFunctionReturn(0); 39 PetscSFPackageInitialized = PETSC_TRUE; 40 /* Register Class */ 41 ierr = PetscClassIdRegister("Star Forest Graph", &PETSCSF_CLASSID);CHKERRQ(ierr); 42 /* Register Constructors */ 43 ierr = PetscSFRegisterAll();CHKERRQ(ierr); 44 /* Register Events */ 45 ierr = PetscLogEventRegister("SFSetGraph" , PETSCSF_CLASSID, &PETSCSF_SetGraph);CHKERRQ(ierr); 46 ierr = PetscLogEventRegister("SFSetUp" , PETSCSF_CLASSID, &PETSCSF_SetUp);CHKERRQ(ierr); 47 ierr = PetscLogEventRegister("SFBcastBegin" , PETSCSF_CLASSID, &PETSCSF_BcastBegin);CHKERRQ(ierr); 48 ierr = PetscLogEventRegister("SFBcastEnd" , PETSCSF_CLASSID, &PETSCSF_BcastEnd);CHKERRQ(ierr); 49 ierr = PetscLogEventRegister("SFReduceBegin" , PETSCSF_CLASSID, &PETSCSF_ReduceBegin);CHKERRQ(ierr); 50 ierr = PetscLogEventRegister("SFReduceEnd" , PETSCSF_CLASSID, &PETSCSF_ReduceEnd);CHKERRQ(ierr); 51 ierr = PetscLogEventRegister("SFFetchOpBegin" , PETSCSF_CLASSID, &PETSCSF_FetchAndOpBegin);CHKERRQ(ierr); 52 ierr = PetscLogEventRegister("SFFetchOpEnd" , PETSCSF_CLASSID, &PETSCSF_FetchAndOpEnd);CHKERRQ(ierr); 53 ierr = PetscLogEventRegister("SFCreateEmbed" , PETSCSF_CLASSID, &PETSCSF_EmbedSF);CHKERRQ(ierr); 54 ierr = PetscLogEventRegister("SFDistSection" , PETSCSF_CLASSID, &PETSCSF_DistSect);CHKERRQ(ierr); 55 ierr = PetscLogEventRegister("SFSectionSF" , PETSCSF_CLASSID, &PETSCSF_SectSF);CHKERRQ(ierr); 56 ierr = PetscLogEventRegister("SFRemoteOff" , PETSCSF_CLASSID, &PETSCSF_RemoteOff);CHKERRQ(ierr); 57 ierr = PetscLogEventRegister("SFPack" , PETSCSF_CLASSID, &PETSCSF_Pack);CHKERRQ(ierr); 58 ierr = PetscLogEventRegister("SFUnpack" , PETSCSF_CLASSID, &PETSCSF_Unpack);CHKERRQ(ierr); 59 /* Process Info */ 60 { 61 PetscClassId classids[1]; 62 63 classids[0] = PETSCSF_CLASSID; 64 ierr = PetscInfoProcessClass("sf", 1, classids);CHKERRQ(ierr); 65 } 66 /* Process summary exclusions */ 67 ierr = PetscOptionsGetString(NULL,NULL,"-log_exclude",logList,sizeof(logList),&opt);CHKERRQ(ierr); 68 if (opt) { 69 ierr = PetscStrInList("sf",logList,',',&pkg);CHKERRQ(ierr); 70 if (pkg) {ierr = PetscLogEventExcludeClass(PETSCSF_CLASSID);CHKERRQ(ierr);} 71 } 72 /* Register package finalizer */ 73 ierr = PetscRegisterFinalize(PetscSFFinalizePackage);CHKERRQ(ierr); 74 PetscFunctionReturn(0); 75 } 76 77 /*@C 78 PetscSFFinalizePackage - Finalize PetscSF package, it is called from PetscFinalize() 79 80 Logically Collective 81 82 Level: developer 83 84 .seealso: PetscSFInitializePackage() 85 @*/ 86 PetscErrorCode PetscSFFinalizePackage(void) 87 { 88 PetscErrorCode ierr; 89 90 PetscFunctionBegin; 91 ierr = PetscFunctionListDestroy(&PetscSFList);CHKERRQ(ierr); 92 PetscSFPackageInitialized = PETSC_FALSE; 93 PetscSFRegisterAllCalled = PETSC_FALSE; 94 PetscFunctionReturn(0); 95 } 96