1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
!
!
! Reads in data prepared by ex1.c in parallel, does a suitable partitioning of the
! data and arranges it in a format suitable for Pflotran.
!
!
! volumes_orig - vector in the "natural" order read from the data file
! volumes - vector in the "permuted" PETSc ordering for parallel computing
! vs - scatter from the natural ordering vector to the PETSc ordering vector
! volumesghosted - ghosted version of PETSc vector
! vslocal - scatter from PETSc order vector to ghosted PETSc order vector
!
program main
implicit none
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Include files
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
!
! The following include statements are required for Fortran programs
! that use PETSc vectors:
! petscsys.h - base PETSc routines
! petscvec.h - defines (INSERT_VALUES)
! petscmat.h - matrices
! petscmat.h90 - to allow access to Fortran 90 features of matrices
#include "finclude/petscsys.h"
#include "finclude/petscis.h"
#include "finclude/petscvec.h"
#include "finclude/petscmat.h"
#include "finclude/petscviewer.h"
PetscErrorCode ierr
call PetscInitialize(PETSC_NULL_CHARACTER,ierr)
call Loader('mesh')
call PetscFinalize(ierr)
end
subroutine Loader(filename)
implicit none
#include "finclude/petscsys.h"
#include "finclude/petscis.h"
#include "finclude/petscvec.h"
#include "finclude/petscmat.h"
#include "finclude/petscmat.h90"
#include "finclude/petscviewer.h"
character(*) filename
Mat Adj, Adjnew
PetscErrorCode ierr
PetscViewer petscviewer
PetscViewer hdf5viewer
MatPartitioning mp
IS partitioning,newnumbering
IS oldnumbering
IS all_oldnumbering
IS allnewnumbering
IS newnumberingsmall
PetscInt nlocal,nlocalnew,bs
PetscInt ntotal
PetscMPIInt size,rank
Vec volumes_orig,volumes
VecScatter vs
PetscInt, pointer :: ghosts(:)
PetscInt nghost,i
bs = 3
call PetscViewerBinaryOpen(PETSC_COMM_WORLD,filename//'.petsc', &
& FILE_MODE_READ,petscviewer,ierr)
call MatLoad(petscviewer,MATMPIBAIJ,Adj,ierr)
call MatGetLocalSize(Adj,nlocal,nlocal,ierr)
call PetscViewerDestroy(petscviewer,ierr)
call MatPartitioningCreate(PETSC_COMM_WORLD,mp,ierr)
call MatPartitioningSetAdjacency(mp,Adj,ierr)
call MatPartitioningSetFromOptions(mp,ierr)
call MatPartitioningApply(mp,partitioning,ierr)
call MatPartitioningDestroy(mp,ierr)
!
! newnumbering is the new cell number for each old cell number
!
call ISPartitioningToNumbering(partitioning,newnumbering,ierr)
call ISDestroy(partitioning,ierr)
call ISAllGather(newnumbering,allnewnumbering,ierr)
call MatPermute(Adj,newnumbering,allnewnumbering,Adjnew,ierr)
call MatGetLocalSize(Adjnew,nlocalnew,nlocalnew,ierr)
call ISDestroy(allnewnumbering,ierr)
call PetscViewerHDF5Open(PETSC_COMM_WORLD,filename//'.h5', &
& FILE_MODE_READ,hdf5viewer,ierr)
call VecCreateMPI(PETSC_COMM_WORLD,nlocal/bs,PETSC_DETERMINE, &
& volumes_orig,ierr)
call PetscObjectSetName(volumes_orig,'/Cells/Volumes',ierr)
call VecLoadIntoVector(hdf5viewer,volumes_orig,ierr)
call VecCreateMPI(PETSC_COMM_WORLD,nlocalnew/bs,PETSC_DETERMINE, &
& volumes,ierr)
call MatGetSize(Adjnew,ntotal,ntotal,ierr)
call ISCompressIndicesGeneral(ntotal,bs,1, newnumbering,
& newnumberingsmall,ierr)
call VecScatterCreate(volumes_orig,newnumberingsmall,volumes, &
& PETSC_NULL_OBJECT,vs,ierr)
call VecScatterBegin(vs,volumes_orig,volumes,INSERT_VALUES, &
& SCATTER_FORWARD,ierr)
call VecScatterEnd(vs,volumes_orig,volumes,INSERT_VALUES, &
& SCATTER_FORWARD,ierr)
call PetscViewerDestroy(hdf5viewer,ierr)
call MatGetGhostsF90(Adjnew,ghosts,ierr)
print*, ghosts
call VecDestroy(volumes_orig,ierr)
call MatDestroy(Adj,ierr)
call MatDestroy(Adjnew,ierr)
call ISDestroy(newnumbering,ierr)
return
end
|
