/*@C
PetscBinarySynchronizedRead - Reads from a binary file.
Collective on MPI_Comm
Input Parameters:
+ comm - the MPI communicator
. fd - the file
. n - the number of items to read
- type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
Output Parameters:
. p - the buffer
Level: developer
Notes:
Does a PetscBinaryRead() followed by an MPI_Bcast()
PetscBinarySynchronizedRead() uses byte swapping to work on all machines.
Integers are stored on the file as 32 long, regardless of whether
they are stored in the machine as 32 or 64, this means the same
binary file may be read on any machine.
Concepts: files^synchronized reading of binary files
Concepts: binary files^reading, synchronized
.seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(),
PetscBinarySynchronizedSeek()
@*/
PetscErrorCode PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type)
{
PetscErrorCode ierr;
PetscMPIInt rank;
MPI_Datatype mtype;
char *fname = NULL;
void *ptmp = NULL;
PetscFunctionBegin;
if (type == PETSC_FUNCTION) {
n = 64;
type = PETSC_CHAR;
ptmp = p;
fname = (char*)malloc(n*sizeof(char));
if (!fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name");
p = (void*)fname;
}
ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
if (!rank) {
ierr = PetscBinaryRead(fd,p,n,type);CHKERRQ(ierr);
}
ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr);
ierr = MPI_Bcast(p,n,mtype,0,comm);CHKERRQ(ierr);
if (type == PETSC_FUNCTION) {
#if defined(PETSC_SERIALIZE_FUNCTIONS)
ierr = PetscDLLibrarySym(PETSC_COMM_SELF,&PetscDLLibrariesLoaded,NULL,fname,(void**)ptmp);CHKERRQ(ierr);
#else
*(void**)ptmp = NULL;
#endif
free(fname);
}
PetscFunctionReturn(0);
}
/*
PFStringCreateFunction - Creates a function from a string
Collective over PF
Input Parameters:
+ pf - the function object
- string - the string that defines the function
Output Parameter:
. f - the function pointer.
.seealso: PFSetFromOptions()
*/
PetscErrorCode PETSCVEC_DLLEXPORT PFStringCreateFunction(PF pf,char *string,void **f)
{
#if defined(PETSC_USE_DYNAMIC_LIBRARIES)
PetscErrorCode ierr;
char task[1024],tmp[256],lib[PETSC_MAX_PATH_LEN],username[64];
FILE *fd;
PetscTruth tmpshared,wdshared,keeptmpfiles = PETSC_FALSE;
MPI_Comm comm;
#endif
PetscFunctionBegin;
#if defined(PETSC_USE_DYNAMIC_LIBRARIES)
ierr = PetscStrfree(pf->data);CHKERRQ(ierr);
ierr = PetscStrallocpy(string,(char**)&pf->data);CHKERRQ(ierr);
/* create the new C function and compile it */
ierr = PetscSharedTmp(((PetscObject)pf)->comm,&tmpshared);CHKERRQ(ierr);
ierr = PetscSharedWorkingDirectory(((PetscObject)pf)->comm,&wdshared);CHKERRQ(ierr);
if (tmpshared) { /* do it in /tmp since everyone has one */
ierr = PetscGetTmp(((PetscObject)pf)->comm,tmp,PETSC_MAX_PATH_LEN);CHKERRQ(ierr);
comm = ((PetscObject)pf)->comm;
} else if (!wdshared) { /* each one does in private /tmp */
ierr = PetscGetTmp(((PetscObject)pf)->comm,tmp,PETSC_MAX_PATH_LEN);CHKERRQ(ierr);
comm = PETSC_COMM_SELF;
} else { /* do it in current directory */
ierr = PetscStrcpy(tmp,".");CHKERRQ(ierr);
comm = ((PetscObject)pf)->comm;
}
ierr = PetscOptionsGetTruth(((PetscObject)pf)->prefix,"-pf_string_keep_files",&keeptmpfiles,PETSC_NULL);CHKERRQ(ierr);
if (keeptmpfiles) {
sprintf(task,"cd %s ; mkdir ${USERNAME} ; cd ${USERNAME} ; \\cp -f ${PETSC_DIR}/src/pf/impls/string/makefile ./makefile ; ke MIN=%d NOUT=%d petscdlib STRINGFUNCTION=\"%s\" ; sync\n",tmp,(int)pf->dimin,(int)pf->dimout,string);
} else {
sprintf(task,"cd %s ; mkdir ${USERNAME} ;cd ${USERNAME} ; \\cp -f ${PETSC_DIR}/src/pf/impls/string/makefile ./makefile ; make MIN=%d NOUT=%d -f makefile petscdlib STRINGFUNCTION=\"%s\" ; \\rm -f makefile petscdlib.c libpetscdlib.a ; sync\n",tmp,(int)pf->dimin,(int)pf->dimout,string);
}
#if defined(PETSC_HAVE_POPEN)
ierr = PetscPOpen(comm,PETSC_NULL,task,"r",&fd);CHKERRQ(ierr);
ierr = PetscPClose(comm,fd);CHKERRQ(ierr);
#else
SETERRQ(PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine");
#endif
ierr = MPI_Barrier(comm);CHKERRQ(ierr);
/* load the apply function from the dynamic library */
ierr = PetscGetUserName(username,64);CHKERRQ(ierr);
sprintf(lib,"%s/%s/libpetscdlib",tmp,username);
ierr = PetscDLLibrarySym(comm,PETSC_NULL,lib,"PFApply_String",f);CHKERRQ(ierr);
#endif
PetscFunctionReturn(0);
}
/*@C
PetscBinarySynchronizedRead - Reads from a binary file.
Collective on MPI_Comm
Input Parameters:
+ comm - the MPI communicator
. fd - the file descriptor
. num - the maximum number of items to read
- type - the type of items to read (PETSC_INT, PETSC_REAL, PETSC_SCALAR, etc.)
Output Parameters:
+ data - the buffer
- count - the number of items read, optional
Level: developer
Notes:
Does a PetscBinaryRead() followed by an MPI_Bcast()
If count is not provided and the number of items read is less than
the maximum number of items to read, then this routine errors.
PetscBinarySynchronizedRead() uses byte swapping to work on all machines.
Integers are stored on the file as 32 long, regardless of whether
they are stored in the machine as 32 or 64, this means the same
binary file may be read on any machine.
Concepts: files^synchronized reading of binary files
Concepts: binary files^reading, synchronized
.seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(),
PetscBinarySynchronizedSeek()
@*/
PetscErrorCode PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *data,PetscInt num,PetscInt *count,PetscDataType type)
{
PetscErrorCode ierr;
PetscMPIInt rank;
MPI_Datatype mtype;
PetscInt ibuf[2] = {0, 0};
char *fname = NULL;
void *fptr = NULL;
PetscFunctionBegin;
if (type == PETSC_FUNCTION) {
num = 64;
type = PETSC_CHAR;
fname = (char*)malloc(num*sizeof(char));
fptr = data;
data = (void*)fname;
if (!fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name");
}
ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
if (!rank) {
ibuf[0] = PetscBinaryRead(fd,data,num,count?&ibuf[1]:NULL,type);
}
ierr = MPI_Bcast(ibuf,2,MPIU_INT,0,comm);CHKERRQ(ierr);
ierr = (PetscErrorCode)ibuf[0];CHKERRQ(ierr);
ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr);
ierr = MPI_Bcast(data,count?ibuf[1]:num,mtype,0,comm);CHKERRQ(ierr);
if (count) *count = ibuf[1];
if (type == PETSC_FUNCTION) {
#if defined(PETSC_SERIALIZE_FUNCTIONS)
ierr = PetscDLLibrarySym(PETSC_COMM_SELF,&PetscDLLibrariesLoaded,NULL,fname,(void**)fptr);CHKERRQ(ierr);
#else
*(void**)fptr = NULL;
#endif
free(fname);
}
PetscFunctionReturn(0);
}
/*@C
PetscBinarySynchronizedRead - Reads from a binary file.
Collective on MPI_Comm
Input Parameters:
+ comm - the MPI communicator
. fd - the file
. n - the number of items to read
- type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
Output Parameters:
. p - the buffer
Options Database Key:
. -binary_longints - indicates the file was generated on a Cray vector
machine (not the T3E/D) and the ints are stored as 64 bit
quantities, otherwise they are stored as 32 bit
Level: developer
Notes:
Does a PetscBinaryRead() followed by an MPI_Bcast()
PetscBinarySynchronizedRead() uses byte swapping to work on all machines.
Integers are stored on the file as 32 long, regardless of whether
they are stored in the machine as 32 or 64, this means the same
binary file may be read on any machine.
Concepts: files^synchronized reading of binary files
Concepts: binary files^reading, synchronized
.seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(),
PetscBinarySynchronizedSeek()
@*/
PetscErrorCode PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type)
{
PetscErrorCode ierr;
PetscMPIInt rank;
MPI_Datatype mtype;
char *fname;
PetscBool functionload = PETSC_FALSE;
void *ptmp = NULL;
PetscFunctionBegin;
if (type == PETSC_FUNCTION) {
functionload = PETSC_TRUE;
n = 64;
type = PETSC_CHAR;
ptmp = p;
/* warning memory leak */
fname = (char*)malloc(64*sizeof(char));
p = (void*)fname;
}
ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
if (!rank) {
ierr = PetscBinaryRead(fd,p,n,type);CHKERRQ(ierr);
}
ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr);
ierr = MPI_Bcast(p,n,mtype,0,comm);CHKERRQ(ierr);
if (functionload) {
#if defined(PETSC_SERIALIZE_FUNCTIONS)
ierr = PetscDLLibrarySym(PETSC_COMM_SELF,&PetscDLLibrariesLoaded,NULL,fname,(void**)ptmp);CHKERRQ(ierr);
#else
*(void**)ptmp = NULL;
#endif
}
PetscFunctionReturn(0);
}
