1*c4762a1bSJed Brown static char help[] = "Demonstrates named colormaps\n"; 2*c4762a1bSJed Brown 3*c4762a1bSJed Brown #include <petscsys.h> 4*c4762a1bSJed Brown #include <petscdraw.h> 5*c4762a1bSJed Brown 6*c4762a1bSJed Brown 7*c4762a1bSJed Brown typedef PetscReal (*Function)(PetscReal,PetscReal); 8*c4762a1bSJed Brown 9*c4762a1bSJed Brown typedef struct { 10*c4762a1bSJed Brown Function function; 11*c4762a1bSJed Brown } FunctionCtx; 12*c4762a1bSJed Brown 13*c4762a1bSJed Brown #define Exp PetscExpReal 14*c4762a1bSJed Brown #define Pow PetscPowReal 15*c4762a1bSJed Brown static PetscReal Peaks(PetscReal x,PetscReal y) 16*c4762a1bSJed Brown { 17*c4762a1bSJed Brown return 3 * Pow(1-x,2) * Exp(-Pow(x,2) - Pow(y+1,2)) 18*c4762a1bSJed Brown - 10 * (x/5 - Pow(x,3) - Pow(y,5)) * Exp(-Pow(x,2) - Pow(y,2)) 19*c4762a1bSJed Brown - 1./3 * Exp(-Pow(x+1,2) - Pow(y,2)); 20*c4762a1bSJed Brown } 21*c4762a1bSJed Brown 22*c4762a1bSJed Brown static PetscErrorCode DrawFunction(PetscDraw draw,void *ctx) 23*c4762a1bSJed Brown { 24*c4762a1bSJed Brown int i,j,w,h; 25*c4762a1bSJed Brown Function function = ((FunctionCtx*)ctx)->function; 26*c4762a1bSJed Brown PetscReal min = PETSC_MAX_REAL, max = PETSC_MIN_REAL; 27*c4762a1bSJed Brown MPI_Comm comm = PetscObjectComm((PetscObject)draw); 28*c4762a1bSJed Brown PetscMPIInt size,rank; 29*c4762a1bSJed Brown PetscDraw popup; 30*c4762a1bSJed Brown PetscErrorCode ierr; 31*c4762a1bSJed Brown 32*c4762a1bSJed Brown PetscFunctionBegin; 33*c4762a1bSJed Brown ierr = PetscDrawGetWindowSize(draw,&w,&h);CHKERRQ(ierr); 34*c4762a1bSJed Brown ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 35*c4762a1bSJed Brown ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 36*c4762a1bSJed Brown 37*c4762a1bSJed Brown ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 38*c4762a1bSJed Brown for (j=rank; j<h; j+=size) { 39*c4762a1bSJed Brown for (i=0; i<w; i++) { 40*c4762a1bSJed Brown PetscReal x,y,f; int color; 41*c4762a1bSJed Brown ierr = PetscDrawPixelToCoordinate(draw,i,j,&x,&y);CHKERRQ(ierr); 42*c4762a1bSJed Brown f = function(x,y); color = PetscDrawRealToColor(f,-8,+8); 43*c4762a1bSJed Brown ierr = PetscDrawPointPixel(draw,i,j,color);CHKERRQ(ierr); 44*c4762a1bSJed Brown min = PetscMin(f,min); max = PetscMax(f,max); 45*c4762a1bSJed Brown } 46*c4762a1bSJed Brown } 47*c4762a1bSJed Brown ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 48*c4762a1bSJed Brown 49*c4762a1bSJed Brown ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr); 50*c4762a1bSJed Brown ierr = PetscDrawScalePopup(popup,-8,+8);CHKERRQ(ierr); 51*c4762a1bSJed Brown PetscFunctionReturn(0); 52*c4762a1bSJed Brown } 53*c4762a1bSJed Brown 54*c4762a1bSJed Brown int main(int argc,char **argv) 55*c4762a1bSJed Brown { 56*c4762a1bSJed Brown char title[64],cmap[32] = ""; 57*c4762a1bSJed Brown PetscDraw draw; 58*c4762a1bSJed Brown FunctionCtx ctx; 59*c4762a1bSJed Brown PetscErrorCode ierr; 60*c4762a1bSJed Brown 61*c4762a1bSJed Brown ctx.function = Peaks; 62*c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,NULL,help);if (ierr) return ierr; 63*c4762a1bSJed Brown ierr = PetscOptionsGetString(NULL,NULL,"-draw_cmap",cmap,sizeof(cmap),NULL);CHKERRQ(ierr); 64*c4762a1bSJed Brown ierr = PetscSNPrintf(title,sizeof(title),"Colormap: %s",cmap);CHKERRQ(ierr); 65*c4762a1bSJed Brown 66*c4762a1bSJed Brown ierr = PetscDrawCreate(PETSC_COMM_WORLD,NULL,title,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,&draw);CHKERRQ(ierr); 67*c4762a1bSJed Brown ierr = PetscObjectSetName((PetscObject)draw,"Peaks");CHKERRQ(ierr); 68*c4762a1bSJed Brown ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr); 69*c4762a1bSJed Brown ierr = PetscDrawSetCoordinates(draw,-3,-3,+3,+3);CHKERRQ(ierr); 70*c4762a1bSJed Brown ierr = PetscDrawZoom(draw,DrawFunction,&ctx);CHKERRQ(ierr); 71*c4762a1bSJed Brown ierr = PetscDrawSave(draw);CHKERRQ(ierr); 72*c4762a1bSJed Brown 73*c4762a1bSJed Brown ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr); 74*c4762a1bSJed Brown ierr = PetscFinalize(); 75*c4762a1bSJed Brown return ierr; 76*c4762a1bSJed Brown } 77*c4762a1bSJed Brown 78*c4762a1bSJed Brown 79*c4762a1bSJed Brown /*TEST 80*c4762a1bSJed Brown 81*c4762a1bSJed Brown build: 82*c4762a1bSJed Brown requires: x 83*c4762a1bSJed Brown 84*c4762a1bSJed Brown test: 85*c4762a1bSJed Brown args: -draw_cmap hue 86*c4762a1bSJed Brown output_file: output/ex1_1.out 87*c4762a1bSJed Brown 88*c4762a1bSJed Brown test: 89*c4762a1bSJed Brown suffix: 2 90*c4762a1bSJed Brown args: -draw_cmap gray 91*c4762a1bSJed Brown output_file: output/ex1_1.out 92*c4762a1bSJed Brown 93*c4762a1bSJed Brown test: 94*c4762a1bSJed Brown suffix: 3 95*c4762a1bSJed Brown args: -draw_cmap bone 96*c4762a1bSJed Brown output_file: output/ex1_1.out 97*c4762a1bSJed Brown 98*c4762a1bSJed Brown test: 99*c4762a1bSJed Brown suffix: 4 100*c4762a1bSJed Brown args: -draw_cmap jet 101*c4762a1bSJed Brown output_file: output/ex1_1.out 102*c4762a1bSJed Brown 103*c4762a1bSJed Brown test: 104*c4762a1bSJed Brown suffix: 5 105*c4762a1bSJed Brown args: -draw_cmap coolwarm 106*c4762a1bSJed Brown output_file: output/ex1_1.out 107*c4762a1bSJed Brown 108*c4762a1bSJed Brown test: 109*c4762a1bSJed Brown suffix: 6 110*c4762a1bSJed Brown args: -draw_cmap parula 111*c4762a1bSJed Brown output_file: output/ex1_1.out 112*c4762a1bSJed Brown 113*c4762a1bSJed Brown test: 114*c4762a1bSJed Brown suffix: 7 115*c4762a1bSJed Brown args: -draw_cmap viridis 116*c4762a1bSJed Brown output_file: output/ex1_1.out 117*c4762a1bSJed Brown 118*c4762a1bSJed Brown test: 119*c4762a1bSJed Brown suffix: 8 120*c4762a1bSJed Brown args: -draw_cmap plasma 121*c4762a1bSJed Brown output_file: output/ex1_1.out 122*c4762a1bSJed Brown 123*c4762a1bSJed Brown TEST*/ 124