Actual source code: cmatlab.c
2: #include <../src/vec/pf/pfimpl.h> /*I "petscpf.h" I*/
4: /*
5: Ths PF generates a MATLAB function on the fly
6: */
7: typedef struct {
8: PetscInt dimin,dimout;
9: PetscMatlabEngine mengine;
10: char *string;
11: } PF_Matlab;
12:
15: PetscErrorCode PFView_Matlab(void *value,PetscViewer viewer)
16: {
18: PetscBool iascii;
19: PF_Matlab *matlab = (PF_Matlab*)value;
22: PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);
23: if (iascii) {
24: PetscViewerASCIIPrintf(viewer,"Matlab Matlab = %s\n",matlab->string);
25: }
26: return(0);
27: }
31: PetscErrorCode PFDestroy_Matlab(void *value)
32: {
34: PF_Matlab *matlab = (PF_Matlab*)value;
37: PetscFree(matlab->string);
38: PetscMatlabEngineDestroy(&matlab->mengine);
39: PetscFree(matlab);
40: return(0);
41: }
45: PetscErrorCode PFApply_Matlab(void *value,PetscInt n,const PetscScalar *in,PetscScalar *out)
46: {
47: PF_Matlab *matlab = (PF_Matlab*)value;
51: if (!value) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Need to set string for MATLAB function, via -pf_matlab string");
52: PetscMatlabEnginePutArray(matlab->mengine,matlab->dimin,n,in,"x");
53: PetscMatlabEngineEvaluate(matlab->mengine,matlab->string);
54: PetscMatlabEngineGetArray(matlab->mengine,matlab->dimout,n,out,"f");
55: return(0);
56: }
60: PetscErrorCode PFSetFromOptions_Matlab(PF pf)
61: {
63: PetscBool flag;
64: char value[256];
65: PF_Matlab *matlab = (PF_Matlab*)pf->data;
68: PetscOptionsHead("Matlab function options");
69: PetscOptionsString("-pf_matlab","Matlab function","None","",value,256,&flag);
70: if (flag) {
71: PetscStrallocpy((char*)value,&matlab->string);
72: }
73: PetscOptionsTail();
74: return(0);
75: }
81: PetscErrorCode PFCreate_Matlab(PF pf,void *value)
82: {
84: PF_Matlab *matlab;
87: PetscNewLog(pf,PF_Matlab,&matlab);
88: matlab->dimin = pf->dimin;
89: matlab->dimout = pf->dimout;
91: PetscMatlabEngineCreate(((PetscObject)pf)->comm,PETSC_NULL,&matlab->mengine);
92:
93: if (value) {
94: PetscStrallocpy((char*)value,&matlab->string);
95: }
96: PFSet(pf,PFApply_Matlab,PETSC_NULL,PFView_Matlab,PFDestroy_Matlab,matlab);
98: pf->ops->setfromoptions = PFSetFromOptions_Matlab;
99: return(0);
100: }