Actual source code: reg.c
petsc-3.6.1 2015-08-06
2: /*
3: Provides a general mechanism to allow one to register new routines in
4: dynamic libraries for many of the PETSc objects (including, e.g., KSP and PC).
5: */
6: #include <petsc/private/petscimpl.h> /*I "petscsys.h" I*/
7: #include <petscviewer.h>
9: /*
10: This is the default list used by PETSc with the PetscDLLibrary register routines
11: */
12: PetscDLLibrary PetscDLLibrariesLoaded = 0;
14: #if defined(PETSC_HAVE_DYNAMIC_LIBRARIES)
18: static PetscErrorCode PetscLoadDynamicLibrary(const char *name,PetscBool *found)
19: {
20: char libs[PETSC_MAX_PATH_LEN],dlib[PETSC_MAX_PATH_LEN];
24: PetscStrcpy(libs,"${PETSC_LIB_DIR}/libpetsc");
25: PetscStrcat(libs,name);
26: PetscDLLibraryRetrieve(PETSC_COMM_WORLD,libs,dlib,1024,found);
27: if (*found) {
28: PetscDLLibraryAppend(PETSC_COMM_WORLD,&PetscDLLibrariesLoaded,dlib);
29: } else {
30: PetscStrcpy(libs,"${PETSC_DIR}/${PETSC_ARCH}/lib/libpetsc");
31: PetscStrcat(libs,name);
32: PetscDLLibraryRetrieve(PETSC_COMM_WORLD,libs,dlib,1024,found);
33: if (*found) {
34: PetscDLLibraryAppend(PETSC_COMM_WORLD,&PetscDLLibrariesLoaded,dlib);
35: }
36: }
37: return(0);
38: }
40: #endif
42: #if defined(PETSC_HAVE_THREADSAFETY)
43: extern PetscErrorCode AOInitializePackage(void);
44: extern PetscErrorCode PetscSFInitializePackage(void);
45: extern PetscErrorCode CharacteristicInitializePackage(void);
46: extern PetscErrorCode ISInitializePackage(void);
47: extern PetscErrorCode VecInitializePackage(void);
48: extern PetscErrorCode MatInitializePackage(void);
49: extern PetscErrorCode DMInitializePackage(void);
50: extern PetscErrorCode PCInitializePackage(void);
51: extern PetscErrorCode KSPInitializePackage(void);
52: extern PetscErrorCode SNESInitializePackage(void);
53: extern PetscErrorCode TSInitializePackage(void);
54: static MPI_Comm PETSC_COMM_WORLD_INNER = 0,PETSC_COMM_SELF_INNER = 0;
55: #endif
59: /*
60: PetscInitialize_DynamicLibraries - Adds the default dynamic link libraries to the
61: search path.
62: */
63: PetscErrorCode PetscInitialize_DynamicLibraries(void)
64: {
65: char *libname[32];
67: PetscInt nmax,i;
68: #if defined(PETSC_HAVE_DYNAMIC_LIBRARIES)
69: PetscBool preload;
70: #endif
73: nmax = 32;
74: PetscOptionsGetStringArray(NULL,"-dll_prepend",libname,&nmax,NULL);
75: for (i=0; i<nmax; i++) {
76: PetscDLLibraryPrepend(PETSC_COMM_WORLD,&PetscDLLibrariesLoaded,libname[i]);
77: PetscFree(libname[i]);
78: }
80: #if !defined(PETSC_HAVE_DYNAMIC_LIBRARIES)
81: /*
82: This just initializes the most basic PETSc stuff.
84: The classes, from PetscDraw to PetscTS, are initialized the first
85: time an XXCreate() is called.
86: */
87: PetscSysInitializePackage();
88: #else
89: preload = PETSC_FALSE;
90: PetscOptionsGetBool(NULL,"-dynamic_library_preload",&preload,NULL);
91: if (preload) {
92: PetscBool found;
93: #if defined(PETSC_USE_SINGLE_LIBRARY)
94: PetscLoadDynamicLibrary("",&found);
95: if (!found) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to locate PETSc dynamic library \n You cannot move the dynamic libraries!");
96: #else
97: PetscLoadDynamicLibrary("sys",&found);
98: if (!found) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to locate PETSc dynamic library \n You cannot move the dynamic libraries!");
99: PetscLoadDynamicLibrary("vec",&found);
100: if (!found) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to locate PETSc Vec dynamic library \n You cannot move the dynamic libraries!");
101: PetscLoadDynamicLibrary("mat",&found);
102: if (!found) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to locate PETSc Mat dynamic library \n You cannot move the dynamic libraries!");
103: PetscLoadDynamicLibrary("dm",&found);
104: if (!found) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to locate PETSc DM dynamic library \n You cannot move the dynamic libraries!");
105: PetscLoadDynamicLibrary("ksp",&found);
106: if (!found) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to locate PETSc KSP dynamic library \n You cannot move the dynamic libraries!");
107: PetscLoadDynamicLibrary("snes",&found);
108: if (!found) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to locate PETSc SNES dynamic library \n You cannot move the dynamic libraries!");
109: PetscLoadDynamicLibrary("ts",&found);
110: if (!found) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to locate PETSc TS dynamic library \n You cannot move the dynamic libraries!");
111: #endif
112: }
113: #endif
115: nmax = 32;
116: PetscOptionsGetStringArray(NULL,"-dll_append",libname,&nmax,NULL);
117: for (i=0; i<nmax; i++) {
118: PetscDLLibraryAppend(PETSC_COMM_WORLD,&PetscDLLibrariesLoaded,libname[i]);
119: PetscFree(libname[i]);
120: }
122: #if defined(PETSC_HAVE_THREADSAFETY)
123: /* These must be done here because it is not safe for individual threads to call these initialize routines */
124: AOInitializePackage();
125: PetscSFInitializePackage();
126: CharacteristicInitializePackage();
127: ISInitializePackage();
128: VecInitializePackage();
129: MatInitializePackage();
130: DMInitializePackage();
131: PCInitializePackage();
132: KSPInitializePackage();
133: SNESInitializePackage();
134: TSInitializePackage();
135: PetscCommDuplicate(PETSC_COMM_SELF,&PETSC_COMM_SELF_INNER,NULL);
136: PetscCommDuplicate(PETSC_COMM_WORLD,&PETSC_COMM_WORLD_INNER,NULL);
137: #endif
138: return(0);
139: }
143: /*
144: PetscFinalize_DynamicLibraries - Closes the opened dynamic libraries.
145: */
146: PetscErrorCode PetscFinalize_DynamicLibraries(void)
147: {
149: PetscBool flg = PETSC_FALSE;
152: PetscOptionsGetBool(NULL,"-dll_view",&flg,NULL);
153: if (flg) { PetscDLLibraryPrintPath(PetscDLLibrariesLoaded); }
154: PetscDLLibraryClose(PetscDLLibrariesLoaded);
156: #if defined(PETSC_HAVE_THREADSAFETY)
157: PetscCommDestroy(&PETSC_COMM_SELF_INNER);
158: PetscCommDestroy(&PETSC_COMM_WORLD_INNER);
159: #endif
161: PetscDLLibrariesLoaded = 0;
162: return(0);
163: }
167: /* ------------------------------------------------------------------------------*/
168: struct _n_PetscFunctionList {
169: void (*routine)(void); /* the routine */
170: char *name; /* string to identify routine */
171: PetscFunctionList next; /* next pointer */
172: PetscFunctionList next_list; /* used to maintain list of all lists for freeing */
173: };
175: /*
176: Keep a linked list of PetscFunctionLists so that we can destroy all the left-over ones.
177: */
178: static PetscFunctionList dlallhead = 0;
180: /*MC
181: PetscFunctionListAdd - Given a routine and a string id, saves that routine in the
182: specified registry.
184: Synopsis:
185: #include <petscsys.h>
186: PetscErrorCode PetscFunctionListAdd(PetscFunctionList flist,const char name[],void (*fptr)(void))
188: Not Collective
190: Input Parameters:
191: + flist - pointer registry
192: . name - string to identify routine
193: - fptr - function pointer
195: Notes:
196: To remove a registered routine, pass in a NULL fptr.
198: Users who wish to register new classes for use by a particular PETSc
199: component (e.g., SNES) should generally call the registration routine
200: for that particular component (e.g., SNESRegister()) instead of
201: calling PetscFunctionListAdd() directly.
203: Level: developer
205: .seealso: PetscFunctionListDestroy(), SNESRegister(), KSPRegister(),
206: PCRegister(), TSRegister(), PetscFunctionList, PetscObjectComposeFunction()
207: M*/
210: PETSC_EXTERN PetscErrorCode PetscFunctionListAdd_Private(PetscFunctionList *fl,const char name[],void (*fnc)(void))
211: {
212: PetscFunctionList entry,ne;
213: PetscErrorCode ierr;
216: if (!*fl) {
217: PetscNew(&entry);
218: PetscStrallocpy(name,&entry->name);
219: entry->routine = fnc;
220: entry->next = 0;
221: *fl = entry;
223: #if defined(PETSC_USE_LOG)
224: /* add this new list to list of all lists */
225: if (!dlallhead) {
226: dlallhead = *fl;
227: (*fl)->next_list = 0;
228: } else {
229: ne = dlallhead;
230: dlallhead = *fl;
231: (*fl)->next_list = ne;
232: }
233: #endif
235: } else {
236: /* search list to see if it is already there */
237: ne = *fl;
238: while (ne) {
239: PetscBool founddup;
241: PetscStrcmp(ne->name,name,&founddup);
242: if (founddup) { /* found duplicate */
243: ne->routine = fnc;
244: return(0);
245: }
246: if (ne->next) ne = ne->next;
247: else break;
248: }
249: /* create new entry and add to end of list */
250: PetscNew(&entry);
251: PetscStrallocpy(name,&entry->name);
252: entry->routine = fnc;
253: entry->next = 0;
254: ne->next = entry;
255: }
256: return(0);
257: }
261: /*@
262: PetscFunctionListDestroy - Destroys a list of registered routines.
264: Input Parameter:
265: . fl - pointer to list
267: Level: developer
269: .seealso: PetscFunctionListAdd(), PetscFunctionList
270: @*/
271: PetscErrorCode PetscFunctionListDestroy(PetscFunctionList *fl)
272: {
273: PetscFunctionList next,entry,tmp = dlallhead;
274: PetscErrorCode ierr;
277: if (!*fl) return(0);
279: /*
280: Remove this entry from the master DL list (if it is in it)
281: */
282: if (dlallhead == *fl) {
283: if (dlallhead->next_list) dlallhead = dlallhead->next_list;
284: else dlallhead = NULL;
285: } else if (tmp) {
286: while (tmp->next_list != *fl) {
287: tmp = tmp->next_list;
288: if (!tmp->next_list) break;
289: }
290: if (tmp->next_list) tmp->next_list = tmp->next_list->next_list;
291: }
293: /* free this list */
294: entry = *fl;
295: while (entry) {
296: next = entry->next;
297: PetscFree(entry->name);
298: PetscFree(entry);
299: entry = next;
300: }
301: *fl = 0;
302: return(0);
303: }
305: /*
306: Print any PetscFunctionLists that have not be destroyed
307: */
310: PetscErrorCode PetscFunctionListPrintAll(void)
311: {
312: PetscFunctionList tmp = dlallhead;
313: PetscErrorCode ierr;
316: if (tmp) {
317: PetscPrintf(PETSC_COMM_WORLD,"The following PetscFunctionLists were not destroyed\n");
318: }
319: while (tmp) {
320: PetscPrintf(PETSC_COMM_WORLD,"%s \n",tmp->name);
321: tmp = tmp->next_list;
322: }
323: return(0);
324: }
326: /*MC
327: PetscFunctionListFind - Find function registered under given name
329: Synopsis:
330: #include <petscsys.h>
331: PetscErrorCode PetscFunctionListFind(PetscFunctionList flist,const char name[],void (**fptr)(void))
333: Input Parameters:
334: + flist - pointer to list
335: - name - name registered for the function
337: Output Parameters:
338: . fptr - the function pointer if name was found, else NULL
340: Level: developer
342: .seealso: PetscFunctionListAdd(), PetscFunctionList, PetscObjectQueryFunction()
343: M*/
346: PETSC_EXTERN PetscErrorCode PetscFunctionListFind_Private(PetscFunctionList fl,const char name[],void (**r)(void))
347: {
348: PetscFunctionList entry = fl;
349: PetscErrorCode ierr;
350: PetscBool flg;
353: if (!name) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NULL,"Trying to find routine with null name");
355: *r = 0;
356: while (entry) {
357: PetscStrcmp(name,entry->name,&flg);
358: if (flg) {
359: *r = entry->routine;
360: return(0);
361: }
362: entry = entry->next;
363: }
364: return(0);
365: }
369: /*@
370: PetscFunctionListView - prints out contents of an PetscFunctionList
372: Collective over MPI_Comm
374: Input Parameters:
375: + list - the list of functions
376: - viewer - currently ignored
378: Level: developer
380: .seealso: PetscFunctionListAdd(), PetscFunctionListPrintTypes(), PetscFunctionList
381: @*/
382: PetscErrorCode PetscFunctionListView(PetscFunctionList list,PetscViewer viewer)
383: {
385: PetscBool iascii;
388: if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF;
392: PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);
393: if (!iascii) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ASCII viewer supported");
395: while (list) {
396: PetscViewerASCIIPrintf(viewer," %s\n",list->name);
397: list = list->next;
398: }
399: PetscViewerASCIIPrintf(viewer,"\n");
400: return(0);
401: }
405: /*@C
406: PetscFunctionListGet - Gets an array the contains the entries in PetscFunctionList, this is used
407: by help etc.
409: Not Collective
411: Input Parameter:
412: . list - list of types
414: Output Parameter:
415: + array - array of names
416: - n - length of array
418: Notes:
419: This allocates the array so that must be freed. BUT the individual entries are
420: not copied so should not be freed.
422: Level: developer
424: .seealso: PetscFunctionListAdd(), PetscFunctionList
425: @*/
426: PetscErrorCode PetscFunctionListGet(PetscFunctionList list,const char ***array,int *n)
427: {
428: PetscErrorCode ierr;
429: PetscInt count = 0;
430: PetscFunctionList klist = list;
433: while (list) {
434: list = list->next;
435: count++;
436: }
437: PetscMalloc1(count+1,array);
438: count = 0;
439: while (klist) {
440: (*array)[count] = klist->name;
441: klist = klist->next;
442: count++;
443: }
444: (*array)[count] = 0;
445: *n = count+1;
446: return(0);
447: }
452: /*@C
453: PetscFunctionListPrintTypes - Prints the methods available.
455: Collective over MPI_Comm
457: Input Parameters:
458: + comm - the communicator (usually MPI_COMM_WORLD)
459: . fd - file to print to, usually stdout
460: . prefix - prefix to prepend to name (optional)
461: . name - option string (for example, "-ksp_type")
462: . text - short description of the object (for example, "Krylov solvers")
463: . man - name of manual page that discusses the object (for example, "KSPCreate")
464: . list - list of types
465: - def - default (current) value
467: Level: developer
469: .seealso: PetscFunctionListAdd(), PetscFunctionList
470: @*/
471: PetscErrorCode PetscFunctionListPrintTypes(MPI_Comm comm,FILE *fd,const char prefix[],const char name[],const char text[],const char man[],PetscFunctionList list,const char def[])
472: {
474: char p[64];
477: if (!fd) fd = PETSC_STDOUT;
479: PetscStrcpy(p,"-");
480: if (prefix) {PetscStrcat(p,prefix);}
481: PetscFPrintf(comm,fd," %s%s <%s>: %s (one of)",p,name+1,def,text);
483: while (list) {
484: PetscFPrintf(comm,fd," %s",list->name);
485: list = list->next;
486: }
487: PetscFPrintf(comm,fd," (%s)\n",man);
488: return(0);
489: }
493: /*@
494: PetscFunctionListDuplicate - Creates a new list from a given object list.
496: Input Parameters:
497: . fl - pointer to list
499: Output Parameters:
500: . nl - the new list (should point to 0 to start, otherwise appends)
502: Level: developer
504: .seealso: PetscFunctionList, PetscFunctionListAdd(), PetscFlistDestroy()
506: @*/
507: PetscErrorCode PetscFunctionListDuplicate(PetscFunctionList fl,PetscFunctionList *nl)
508: {
512: while (fl) {
513: PetscFunctionListAdd(nl,fl->name,fl->routine);
514: fl = fl->next;
515: }
516: return(0);
517: }