PetscErrorCode PetscRMTree(const char dir[])
{
PetscErrorCode ierr;
struct dirent *data;
char loc[PETSC_MAX_PATH_LEN];
PetscBool flg1, flg2;
DIR *dirp;
struct stat statbuf;
PetscFunctionBegin;
dirp = opendir(dir);
if(!dirp) {
PetscBool flg;
ierr = PetscTestDirectory(dir,'r',&flg);CHKERRQ(ierr);
if (flg) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Cannot access directory to delete: %s",dir);
ierr = PetscTestFile(dir,'r',&flg);CHKERRQ(ierr);
if (flg) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Specified path is a file - not a dir: %s",dir);
PetscFunctionReturn(0); /* perhaps the dir was not yet created */
}
while((data = readdir(dirp))) {
ierr = PetscStrcmp(data->d_name, ".",&flg1);CHKERRQ(ierr);
ierr = PetscStrcmp(data->d_name, "..",&flg2);CHKERRQ(ierr);
if (flg1 || flg2) continue;
ierr = PetscPathJoin(dir,data->d_name,PETSC_MAX_PATH_LEN,loc);CHKERRQ(ierr);
if (lstat(loc,&statbuf) <0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"cannot run lstat() on: %s",loc);
if (S_ISDIR(statbuf.st_mode)) {
ierr = PetscRMTree(loc);CHKERRQ(ierr);
} else {
if (unlink(loc)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Could not delete file: %s",loc);
}
}
closedir(dirp);
if (rmdir(dir)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Could not delete dir: %s",dir);
PetscFunctionReturn(0);
}
PetscErrorCode PetscRMTree(const char dir[])
{
PetscErrorCode ierr;
struct _finddata_t data;
char loc[PETSC_MAX_PATH_LEN];
PetscBool flg1, flg2;
#if defined (PETSC_HAVE_STDINT_H)
intptr_t handle;
#else
long handle;
#endif
PetscFunctionBegin;
ierr = PetscPathJoin(dir,"*",PETSC_MAX_PATH_LEN,loc);CHKERRQ(ierr);
handle = _findfirst(loc, &data);
if(handle == -1) {
PetscBool flg;
ierr = PetscTestDirectory(loc,'r',&flg);CHKERRQ(ierr);
if (flg) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Cannot access directory to delete: %s",dir);
ierr = PetscTestFile(loc,'r',&flg);CHKERRQ(ierr);
if (flg) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Specified path is a file - not a dir: %s",dir);
PetscFunctionReturn(0); /* perhaps the dir was not yet created */
}
while(_findnext(handle, &data) != -1) {
ierr = PetscStrcmp(data.name, ".",&flg1);CHKERRQ(ierr);
ierr = PetscStrcmp(data.name, "..",&flg2);CHKERRQ(ierr);
if (flg1 || flg2) continue;
ierr = PetscPathJoin(dir,data.name,PETSC_MAX_PATH_LEN,loc);CHKERRQ(ierr);
if(data.attrib & _A_SUBDIR) {
ierr = PetscRMTree(loc);CHKERRQ(ierr);
} else{
if (remove(loc)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Could not delete file: %s",loc);
}
}
_findclose(handle);
if (_rmdir(dir)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Could not delete dir: %s",dir);
PetscFunctionReturn(0);
}
static PetscErrorCode PetscPythonFindLibrary(char pythonexe[PETSC_MAX_PATH_LEN],char pythonlib[PETSC_MAX_PATH_LEN])
{
const char cmdline[] = "-c 'import sys; print(sys.exec_prefix); print(sys.version[:3])'";
char command[PETSC_MAX_PATH_LEN+1+sizeof(cmdline)+1];
char prefix[PETSC_MAX_PATH_LEN],version[8],sep[2]={PETSC_DIR_SEPARATOR, 0},*eol;
FILE *fp = NULL;
char path[PETSC_MAX_PATH_LEN+1];
PetscBool found = PETSC_FALSE;
PetscErrorCode ierr;
PetscFunctionBegin;
#if defined(PETSC_PYTHON_LIB)
ierr = PetscStrcpy(pythonlib,PETSC_PYTHON_LIB);CHKERRQ(ierr);
PetscFunctionReturn(0);
#endif
/* call Python to find out the name of the Python dynamic library */
ierr = PetscStrncpy(command,pythonexe,PETSC_MAX_PATH_LEN);CHKERRQ(ierr);
ierr = PetscStrcat(command," ");CHKERRQ(ierr);
ierr = PetscStrcat(command,cmdline);CHKERRQ(ierr);
#if defined(PETSC_HAVE_POPEN)
ierr = PetscPOpen(PETSC_COMM_SELF,NULL,command,"r",&fp);CHKERRQ(ierr);
if (!fgets(prefix,sizeof(prefix),fp)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Python: bad output from executable: %s",pythonexe);
if (!fgets(version,sizeof(version),fp)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Python: bad output from executable: %s",pythonexe);
ierr = PetscPClose(PETSC_COMM_SELF,fp,NULL);CHKERRQ(ierr);
#else
SETERRQ(PETSC_COMM_SELF,1,"Python: Aborted due to missing popen()");
#endif
/* remove newlines */
ierr = PetscStrchr(prefix,'\n',&eol);CHKERRQ(ierr);
if (eol) eol[0] = 0;
ierr = PetscStrchr(version,'\n',&eol);CHKERRQ(ierr);
if (eol) eol[0] = 0;
/* test for $prefix/lib64/libpythonX.X[.so]*/
ierr = PetscStrcpy(pythonlib,prefix);CHKERRQ(ierr);
ierr = PetscStrcat(pythonlib,sep);CHKERRQ(ierr);
ierr = PetscStrcat(pythonlib,"lib64");CHKERRQ(ierr);
ierr = PetscTestDirectory(pythonlib,'r',&found);CHKERRQ(ierr);
if (found) {
ierr = PetscStrcat(pythonlib,sep);CHKERRQ(ierr);
ierr = PetscStrcat(pythonlib,"libpython");CHKERRQ(ierr);
ierr = PetscStrcat(pythonlib,version);CHKERRQ(ierr);
ierr = PetscDLLibraryRetrieve(PETSC_COMM_SELF,pythonlib,path,PETSC_MAX_PATH_LEN,&found);CHKERRQ(ierr);
if (found) PetscFunctionReturn(0);
}
/* test for $prefix/lib/libpythonX.X[.so]*/
ierr = PetscStrcpy(pythonlib,prefix);CHKERRQ(ierr);
ierr = PetscStrcat(pythonlib,sep);CHKERRQ(ierr);
ierr = PetscStrcat(pythonlib,"lib");CHKERRQ(ierr);
ierr = PetscTestDirectory(pythonlib,'r',&found);CHKERRQ(ierr);
if (found) {
ierr = PetscStrcat(pythonlib,sep);CHKERRQ(ierr);
ierr = PetscStrcat(pythonlib,"libpython");CHKERRQ(ierr);
ierr = PetscStrcat(pythonlib,version);CHKERRQ(ierr);
ierr = PetscDLLibraryRetrieve(PETSC_COMM_SELF,pythonlib,path,PETSC_MAX_PATH_LEN,&found);CHKERRQ(ierr);
if (found) PetscFunctionReturn(0);
}
/* nothing good found */
ierr = PetscMemzero(pythonlib,PETSC_MAX_PATH_LEN);CHKERRQ(ierr);
ierr = PetscInfo(0,"Python dynamic library not found\n");CHKERRQ(ierr);
PetscFunctionReturn(0);
}
