1c4762a1bSJed Brown static char help[] = "VecView() with a DMDA1d vector and draw viewer.\n\n";
2c4762a1bSJed Brown
3c4762a1bSJed Brown #include <petscdm.h>
4c4762a1bSJed Brown #include <petscdmda.h>
5c4762a1bSJed Brown #include <petscao.h>
6c4762a1bSJed Brown
apply(PetscCtx ctx,PetscInt n,const PetscScalar * x,PetscScalar * y)7*2a8381b2SBarry Smith PetscErrorCode apply(PetscCtx ctx, PetscInt n, const PetscScalar *x, PetscScalar *y)
8d71ae5a4SJacob Faibussowitsch {
9c4762a1bSJed Brown PetscInt i;
10c4762a1bSJed Brown
119371c9d4SSatish Balay for (i = 0; i < n; i++) {
129371c9d4SSatish Balay y[3 * i] = x[i];
139371c9d4SSatish Balay y[3 * i + 1] = x[i] * x[i];
149371c9d4SSatish Balay y[3 * i + 2] = x[i] * x[i] * x[i];
159371c9d4SSatish Balay }
163ba16761SJacob Faibussowitsch return PETSC_SUCCESS;
17c4762a1bSJed Brown }
18c4762a1bSJed Brown
main(int argc,char ** argv)19d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv)
20d71ae5a4SJacob Faibussowitsch {
21c4762a1bSJed Brown DM da;
22c4762a1bSJed Brown Vec global;
23c4762a1bSJed Brown PF pf;
24c4762a1bSJed Brown
25327415f7SBarry Smith PetscFunctionBeginUser;
26c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &argv, NULL, help));
279566063dSJacob Faibussowitsch PetscCall(DMDACreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, 10, 3, 1, NULL, &da));
289566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(da));
299566063dSJacob Faibussowitsch PetscCall(DMSetUp(da));
309566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(da, &global));
319566063dSJacob Faibussowitsch PetscCall(PFCreate(PETSC_COMM_WORLD, 1, 3, &pf));
329566063dSJacob Faibussowitsch PetscCall(PFSet(pf, apply, NULL, NULL, NULL, NULL));
339566063dSJacob Faibussowitsch PetscCall(PFApplyVec(pf, NULL, global));
349566063dSJacob Faibussowitsch PetscCall(PFDestroy(&pf));
359566063dSJacob Faibussowitsch PetscCall(VecView(global, PETSC_VIEWER_DRAW_WORLD));
369566063dSJacob Faibussowitsch PetscCall(VecDestroy(&global));
379566063dSJacob Faibussowitsch PetscCall(DMDestroy(&da));
389566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
39b122ec5aSJacob Faibussowitsch return 0;
40c4762a1bSJed Brown }
41c4762a1bSJed Brown
42c4762a1bSJed Brown /*TEST
43c4762a1bSJed Brown
44c4762a1bSJed Brown test:
45c4762a1bSJed Brown nsize: 2
46c4762a1bSJed Brown requires: x
473886731fSPierre Jolivet output_file: output/empty.out
48c4762a1bSJed Brown
49c4762a1bSJed Brown TEST*/
50