1292f8084SBarry Smith 2c6db04a5SJed Brown #include <../src/vec/pf/pfimpl.h> /*I "petscpf.h" I*/ 382c86c8fSBarry Smith #include <petscmatlab.h> /*I "petscmatlab.h" I*/ 4292f8084SBarry Smith 5292f8084SBarry Smith /* 6*a5b23f4aSJose E. Roman This PF generates a MATLAB function on the fly 7292f8084SBarry Smith */ 8292f8084SBarry Smith typedef struct { 91e1716dcSBarry Smith PetscInt dimin,dimout; 10292f8084SBarry Smith PetscMatlabEngine mengine; 11292f8084SBarry Smith char *string; 12292f8084SBarry Smith } PF_Matlab; 13292f8084SBarry Smith 14292f8084SBarry Smith PetscErrorCode PFView_Matlab(void *value,PetscViewer viewer) 15292f8084SBarry Smith { 16292f8084SBarry Smith PetscErrorCode ierr; 17ace3abfcSBarry Smith PetscBool iascii; 18292f8084SBarry Smith PF_Matlab *matlab = (PF_Matlab*)value; 19292f8084SBarry Smith 20292f8084SBarry Smith PetscFunctionBegin; 21251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 22292f8084SBarry Smith if (iascii) { 23292f8084SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Matlab Matlab = %s\n",matlab->string);CHKERRQ(ierr); 24292f8084SBarry Smith } 25292f8084SBarry Smith PetscFunctionReturn(0); 26292f8084SBarry Smith } 27292f8084SBarry Smith 28292f8084SBarry Smith PetscErrorCode PFDestroy_Matlab(void *value) 29292f8084SBarry Smith { 30292f8084SBarry Smith PetscErrorCode ierr; 31292f8084SBarry Smith PF_Matlab *matlab = (PF_Matlab*)value; 32292f8084SBarry Smith 33292f8084SBarry Smith PetscFunctionBegin; 34503cfb0cSBarry Smith ierr = PetscFree(matlab->string);CHKERRQ(ierr); 35a32ab379SBarry Smith ierr = PetscMatlabEngineDestroy(&matlab->mengine);CHKERRQ(ierr); 36292f8084SBarry Smith ierr = PetscFree(matlab);CHKERRQ(ierr); 37292f8084SBarry Smith PetscFunctionReturn(0); 38292f8084SBarry Smith } 39292f8084SBarry Smith 4069b4f73cSBarry Smith PetscErrorCode PFApply_Matlab(void *value,PetscInt n,const PetscScalar *in,PetscScalar *out) 41292f8084SBarry Smith { 42292f8084SBarry Smith PF_Matlab *matlab = (PF_Matlab*)value; 43292f8084SBarry Smith PetscErrorCode ierr; 44292f8084SBarry Smith 45292f8084SBarry Smith PetscFunctionBegin; 46e3c5b3baSBarry Smith if (!value) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Need to set string for MATLAB function, via -pf_matlab string"); 47292f8084SBarry Smith ierr = PetscMatlabEnginePutArray(matlab->mengine,matlab->dimin,n,in,"x");CHKERRQ(ierr); 48292f8084SBarry Smith ierr = PetscMatlabEngineEvaluate(matlab->mengine,matlab->string);CHKERRQ(ierr); 49292f8084SBarry Smith ierr = PetscMatlabEngineGetArray(matlab->mengine,matlab->dimout,n,out,"f");CHKERRQ(ierr); 50292f8084SBarry Smith PetscFunctionReturn(0); 51292f8084SBarry Smith } 52292f8084SBarry Smith 534416b707SBarry Smith PetscErrorCode PFSetFromOptions_Matlab(PetscOptionItems *PetscOptionsObject,PF pf) 54292f8084SBarry Smith { 55292f8084SBarry Smith PetscErrorCode ierr; 56ace3abfcSBarry Smith PetscBool flag; 57292f8084SBarry Smith char value[256]; 58292f8084SBarry Smith PF_Matlab *matlab = (PF_Matlab*)pf->data; 59292f8084SBarry Smith 60292f8084SBarry Smith PetscFunctionBegin; 61e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"Matlab function options");CHKERRQ(ierr); 62589a23caSBarry Smith ierr = PetscOptionsString("-pf_matlab","Matlab function","None","",value,sizeof(value),&flag);CHKERRQ(ierr); 63292f8084SBarry Smith if (flag) { 64292f8084SBarry Smith ierr = PetscStrallocpy((char*)value,&matlab->string);CHKERRQ(ierr); 65292f8084SBarry Smith } 66292f8084SBarry Smith ierr = PetscOptionsTail();CHKERRQ(ierr); 67292f8084SBarry Smith PetscFunctionReturn(0); 68292f8084SBarry Smith } 69292f8084SBarry Smith 708cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PFCreate_Matlab(PF pf,void *value) 71292f8084SBarry Smith { 72292f8084SBarry Smith PetscErrorCode ierr; 73292f8084SBarry Smith PF_Matlab *matlab; 74292f8084SBarry Smith 75292f8084SBarry Smith PetscFunctionBegin; 76b00a9115SJed Brown ierr = PetscNewLog(pf,&matlab);CHKERRQ(ierr); 77292f8084SBarry Smith matlab->dimin = pf->dimin; 78292f8084SBarry Smith matlab->dimout = pf->dimout; 79292f8084SBarry Smith 80ce94432eSBarry Smith ierr = PetscMatlabEngineCreate(PetscObjectComm((PetscObject)pf),NULL,&matlab->mengine);CHKERRQ(ierr); 81292f8084SBarry Smith 82292f8084SBarry Smith if (value) { 83292f8084SBarry Smith ierr = PetscStrallocpy((char*)value,&matlab->string);CHKERRQ(ierr); 84292f8084SBarry Smith } 850298fd71SBarry Smith ierr = PFSet(pf,PFApply_Matlab,NULL,PFView_Matlab,PFDestroy_Matlab,matlab);CHKERRQ(ierr); 86292f8084SBarry Smith 87292f8084SBarry Smith pf->ops->setfromoptions = PFSetFromOptions_Matlab; 88292f8084SBarry Smith PetscFunctionReturn(0); 89292f8084SBarry Smith } 90292f8084SBarry Smith 91