xref: /petsc/src/vec/is/ao/tests/ex7.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254)
1c4762a1bSJed Brown 
2c4762a1bSJed Brown static char help[] = "Demonstrates constructing an application ordering.\n\n";
3c4762a1bSJed Brown 
4c4762a1bSJed Brown #include <petscao.h>
5c4762a1bSJed Brown #include <petscviewer.h>
6c4762a1bSJed Brown 
7c4762a1bSJed Brown int main(int argc,char **argv)
8c4762a1bSJed Brown {
9c4762a1bSJed Brown   PetscInt       n = 5;
10c4762a1bSJed Brown   PetscMPIInt    rank,size;
11c4762a1bSJed Brown   IS             ispetsc,isapp;
12c4762a1bSJed Brown   AO             ao;
13c4762a1bSJed Brown 
14*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscInitialize(&argc,&argv,(char*)0,help));
155f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL));
165f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank));
175f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size));
18c4762a1bSJed Brown 
19c4762a1bSJed Brown   /* create the index sets */
205f80ce2aSJacob Faibussowitsch   CHKERRQ(ISCreateStride(PETSC_COMM_WORLD,n,rank,size,&ispetsc));
215f80ce2aSJacob Faibussowitsch   CHKERRQ(ISCreateStride(PETSC_COMM_WORLD,n,n*rank,1,&isapp));
22c4762a1bSJed Brown 
23c4762a1bSJed Brown   /* create the application ordering */
245f80ce2aSJacob Faibussowitsch   CHKERRQ(AOCreateBasicIS(isapp,ispetsc,&ao));
25c4762a1bSJed Brown 
265f80ce2aSJacob Faibussowitsch   CHKERRQ(AOView(ao,PETSC_VIEWER_STDOUT_WORLD));
27c4762a1bSJed Brown 
285f80ce2aSJacob Faibussowitsch   CHKERRQ(ISView(ispetsc,PETSC_VIEWER_STDOUT_WORLD));
295f80ce2aSJacob Faibussowitsch   CHKERRQ(ISView(isapp,PETSC_VIEWER_STDOUT_WORLD));
305f80ce2aSJacob Faibussowitsch   CHKERRQ(AOPetscToApplicationIS(ao,ispetsc));
315f80ce2aSJacob Faibussowitsch   CHKERRQ(ISView(isapp,PETSC_VIEWER_STDOUT_WORLD));
325f80ce2aSJacob Faibussowitsch   CHKERRQ(ISView(ispetsc,PETSC_VIEWER_STDOUT_WORLD));
33c4762a1bSJed Brown 
345f80ce2aSJacob Faibussowitsch   CHKERRQ(ISDestroy(&ispetsc));
355f80ce2aSJacob Faibussowitsch   CHKERRQ(ISDestroy(&isapp));
36c4762a1bSJed Brown 
375f80ce2aSJacob Faibussowitsch   CHKERRQ(AODestroy(&ao));
38*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscFinalize());
39*b122ec5aSJacob Faibussowitsch   return 0;
40c4762a1bSJed Brown }
41c4762a1bSJed Brown 
42c4762a1bSJed Brown /*TEST
43c4762a1bSJed Brown 
44c4762a1bSJed Brown    test:
45c4762a1bSJed Brown       nsize: 2
46c4762a1bSJed Brown 
47c4762a1bSJed Brown TEST*/
48