Actual source code: ex8.c
1: static char help[] = "Tests ISLocalToGlobalMappingView() and ISLocalToGlobalMappingLoad()\n\n";
3: #include <petscis.h>
5: static PetscErrorCode TestEqual(MPI_Comm comm, ISLocalToGlobalMapping m1, ISLocalToGlobalMapping m2, const char *tname)
6: {
7: PetscInt n1, n2, b1, b2;
8: const PetscInt *idx1, *idx2;
9: PetscBool flg;
11: PetscFunctionBegin;
12: PetscCall(ISLocalToGlobalMappingGetSize(m1, &n1));
13: PetscCall(ISLocalToGlobalMappingGetBlockSize(m1, &b1));
14: PetscCall(ISLocalToGlobalMappingGetIndices(m1, &idx1));
15: PetscCall(ISLocalToGlobalMappingGetSize(m2, &n2));
16: PetscCall(ISLocalToGlobalMappingGetBlockSize(m2, &b2));
17: PetscCall(ISLocalToGlobalMappingGetIndices(m2, &idx2));
18: flg = (PetscBool)(b1 == b2);
19: if (!flg) PetscCall(PetscPrintf(PETSC_COMM_SELF, "%s: different block sizes %" PetscInt_FMT " %" PetscInt_FMT "\n", tname, b1, b2));
20: flg = (PetscBool)(n1 == n2);
21: if (!flg) PetscCall(PetscPrintf(PETSC_COMM_SELF, "%s: different sizes %" PetscInt_FMT " %" PetscInt_FMT "\n", tname, n1, n2));
22: if (flg) {
23: PetscCall(PetscArraycmp(idx1, idx2, n1, &flg));
24: if (!flg) PetscCall(PetscPrintf(PETSC_COMM_SELF, "%s: different indices\n", tname));
25: }
26: PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &flg, 1, MPI_C_BOOL, MPI_LAND, comm));
27: if (!flg) {
28: PetscCall(ISLocalToGlobalMappingView(m1, PETSC_VIEWER_STDOUT_(comm)));
29: PetscCall(ISLocalToGlobalMappingView(m2, PETSC_VIEWER_STDOUT_(comm)));
30: }
31: PetscCall(ISLocalToGlobalMappingRestoreIndices(m1, &idx1));
32: PetscCall(ISLocalToGlobalMappingRestoreIndices(m2, &idx2));
33: PetscFunctionReturn(PETSC_SUCCESS);
34: }
36: int main(int argc, char **argv)
37: {
38: ISLocalToGlobalMapping lg1l, lg1v, lg1lh, lg2l, lg2v, lg2lh;
39: IS is1, is2;
40: PetscInt n, n1, n2, b1, b2;
41: PetscInt *idx;
42: PetscMPIInt size, rank;
43: PetscViewer vx;
44: MPI_Comm comm;
45: char fname[PETSC_MAX_PATH_LEN], fnameh[PETSC_MAX_PATH_LEN];
47: PetscFunctionBeginUser;
48: PetscCall(PetscInitialize(&argc, &argv, NULL, help));
49: PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
50: PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
51: PetscCall(ISCreateStride(PETSC_COMM_WORLD, size - rank, -size - 1, rank + 1, &is1));
52: PetscCall(ISGetLocalSize(is1, &n));
53: PetscCall(ISGetIndices(is1, (const PetscInt **)&idx));
54: PetscCall(ISCreateBlock(PETSC_COMM_WORLD, 3, n, idx, PETSC_COPY_VALUES, &is2));
55: PetscCall(ISRestoreIndices(is1, (const PetscInt **)&idx));
56: PetscCall(ISLocalToGlobalMappingCreateIS(is1, &lg1v));
57: PetscCall(ISLocalToGlobalMappingCreateIS(is2, &lg2v));
58: PetscCall(ISDestroy(&is1));
59: PetscCall(ISDestroy(&is2));
61: /* Test MATLAB ASCII viewer */
62: PetscCall(PetscObjectSetName((PetscObject)lg1v, "map1"));
63: PetscCall(PetscViewerPushFormat(PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_ASCII_MATLAB));
64: PetscCall(ISLocalToGlobalMappingView(lg1v, PETSC_VIEWER_STDOUT_WORLD));
65: PetscCall(ISLocalToGlobalMappingView(lg2v, PETSC_VIEWER_STDOUT_WORLD));
66: PetscCall(PetscViewerPopFormat(PETSC_VIEWER_STDOUT_WORLD));
67: PetscCall(PetscObjectSetName((PetscObject)lg2v, "map2"));
69: /* Now test view/load of type binary */
70: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "testfile", FILE_MODE_WRITE, &vx));
71: PetscCall(ISLocalToGlobalMappingView(lg1v, vx));
72: PetscCall(ISLocalToGlobalMappingView(lg2v, vx));
73: PetscCall(PetscViewerDestroy(&vx));
75: PetscCall(PetscSNPrintf(fname, PETSC_STATIC_ARRAY_LENGTH(fname), "testfile_seq_%d", rank));
76: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_SELF, fname, FILE_MODE_WRITE, &vx));
77: PetscCall(ISLocalToGlobalMappingView(lg1v, vx));
78: PetscCall(ISLocalToGlobalMappingView(lg2v, vx));
79: PetscCall(PetscViewerDestroy(&vx));
81: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "testfile_noheader", FILE_MODE_WRITE, &vx));
82: PetscCall(PetscViewerBinarySetSkipHeader(vx, PETSC_TRUE));
83: PetscCall(ISLocalToGlobalMappingView(lg1v, vx));
84: PetscCall(ISLocalToGlobalMappingView(lg2v, vx));
85: PetscCall(PetscViewerDestroy(&vx));
87: PetscCall(PetscSNPrintf(fnameh, PETSC_STATIC_ARRAY_LENGTH(fname), "testfile_noheader_seq_%d", rank));
88: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_SELF, fnameh, FILE_MODE_WRITE, &vx));
89: PetscCall(PetscViewerBinarySetSkipHeader(vx, PETSC_TRUE));
90: PetscCall(ISLocalToGlobalMappingView(lg1v, vx));
91: PetscCall(ISLocalToGlobalMappingView(lg2v, vx));
92: PetscCall(PetscViewerDestroy(&vx));
94: PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD, 77, 0, NULL, PETSC_USE_POINTER, &lg1l));
95: PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD, 99, 0, NULL, PETSC_OWN_POINTER, &lg2l));
96: PetscCall(ISLocalToGlobalMappingGetSize(lg1v, &n1));
97: PetscCall(ISLocalToGlobalMappingGetBlockSize(lg1v, &b1));
98: n1 /= b1;
99: PetscCall(ISLocalToGlobalMappingGetSize(lg2v, &n2));
100: PetscCall(ISLocalToGlobalMappingGetBlockSize(lg2v, &b2));
101: n2 /= b2;
102: PetscCall(PetscMalloc1(n1, &idx));
103: PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_SELF, b1, n1, idx, PETSC_OWN_POINTER, &lg1lh));
104: PetscCall(PetscMalloc1(n2, &idx));
105: PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_SELF, b2, n2, idx, PETSC_OWN_POINTER, &lg2lh));
107: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "testfile", FILE_MODE_READ, &vx));
108: PetscCall(ISLocalToGlobalMappingLoad(lg1l, vx));
109: PetscCall(ISLocalToGlobalMappingLoad(lg2l, vx));
110: PetscCall(PetscViewerDestroy(&vx));
111: PetscCall(TestEqual(PETSC_COMM_WORLD, lg1v, lg1l, "load_world_map_world 1"));
112: PetscCall(TestEqual(PETSC_COMM_WORLD, lg2v, lg2l, "load_world_map_world 2"));
114: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "testfile_noheader", FILE_MODE_READ, &vx));
115: PetscCall(PetscViewerBinarySetSkipHeader(vx, PETSC_TRUE));
116: PetscCall(ISLocalToGlobalMappingLoad(lg1l, vx));
117: PetscCall(ISLocalToGlobalMappingLoad(lg2l, vx));
118: PetscCall(PetscViewerDestroy(&vx));
119: PetscCall(TestEqual(PETSC_COMM_WORLD, lg1v, lg1l, "load_world_map_world_noheader 1"));
120: PetscCall(TestEqual(PETSC_COMM_WORLD, lg2v, lg2l, "load_world_map_world_noheader 2"));
122: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "testfile_noheader", FILE_MODE_READ, &vx));
123: PetscCall(PetscViewerBinarySetSkipHeader(vx, PETSC_TRUE));
124: PetscCall(ISLocalToGlobalMappingLoad(lg1lh, vx));
125: PetscCall(ISLocalToGlobalMappingLoad(lg2lh, vx));
126: PetscCall(PetscViewerDestroy(&vx));
127: PetscCall(TestEqual(PETSC_COMM_WORLD, lg1v, lg1lh, "load_world_map_self_noheader 1"));
128: PetscCall(TestEqual(PETSC_COMM_WORLD, lg2v, lg2lh, "load_world_map_self_noheader 2"));
130: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_SELF, fname, FILE_MODE_READ, &vx));
131: PetscCall(ISLocalToGlobalMappingLoad(lg1l, vx));
132: PetscCall(ISLocalToGlobalMappingLoad(lg2l, vx));
133: PetscCall(PetscViewerDestroy(&vx));
134: PetscCall(TestEqual(PETSC_COMM_WORLD, lg1v, lg1l, "load_self_map_world 1"));
135: PetscCall(TestEqual(PETSC_COMM_WORLD, lg2v, lg2l, "load_self_map_world 2"));
137: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_SELF, fnameh, FILE_MODE_READ, &vx));
138: PetscCall(PetscViewerBinarySetSkipHeader(vx, PETSC_TRUE));
139: PetscCall(ISLocalToGlobalMappingLoad(lg1l, vx));
140: PetscCall(ISLocalToGlobalMappingLoad(lg2l, vx));
141: PetscCall(PetscViewerDestroy(&vx));
142: PetscCall(TestEqual(PETSC_COMM_WORLD, lg1v, lg1l, "load_self_map_world_noheader 1"));
143: PetscCall(TestEqual(PETSC_COMM_WORLD, lg2v, lg2l, "load_self_map_world_noheader 2"));
145: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_SELF, fnameh, FILE_MODE_READ, &vx));
146: PetscCall(PetscViewerBinarySetSkipHeader(vx, PETSC_TRUE));
147: PetscCall(ISLocalToGlobalMappingLoad(lg1lh, vx));
148: PetscCall(ISLocalToGlobalMappingLoad(lg2lh, vx));
149: PetscCall(PetscViewerDestroy(&vx));
150: PetscCall(TestEqual(PETSC_COMM_WORLD, lg1v, lg1lh, "load_self_map_self_noheader 1"));
151: PetscCall(TestEqual(PETSC_COMM_WORLD, lg2v, lg2lh, "load_self_map_self_noheader 2"));
153: PetscCall(PetscPrintf(PETSC_COMM_WORLD, "View world maps\n"));
154: PetscCall(ISLocalToGlobalMappingView(lg1v, PETSC_VIEWER_STDOUT_WORLD));
155: PetscCall(ISLocalToGlobalMappingView(lg2v, PETSC_VIEWER_STDOUT_WORLD));
157: PetscCallMPI(MPI_Comm_split(PETSC_COMM_WORLD, rank < 2, rank, &comm));
158: if (rank < 2) {
159: PetscCall(ISLocalToGlobalMappingDestroy(&lg1l));
160: PetscCall(ISLocalToGlobalMappingDestroy(&lg2l));
161: PetscCall(ISLocalToGlobalMappingCreate(comm, 77, 1, &n, PETSC_USE_POINTER, &lg1l));
162: PetscCall(ISLocalToGlobalMappingCreate(comm, 99, 0, NULL, PETSC_OWN_POINTER, &lg2l));
164: PetscCall(PetscViewerBinaryOpen(comm, "testfile", FILE_MODE_READ, &vx));
165: PetscCall(PetscPrintf(comm, "View world maps loaded from subcomm\n"));
166: PetscCall(ISLocalToGlobalMappingLoad(lg1l, vx));
167: PetscCall(ISLocalToGlobalMappingLoad(lg2l, vx));
168: PetscCall(ISLocalToGlobalMappingView(lg1l, PETSC_VIEWER_STDOUT_(comm)));
169: PetscCall(ISLocalToGlobalMappingView(lg2l, PETSC_VIEWER_STDOUT_(comm)));
170: PetscCall(PetscViewerDestroy(&vx));
172: PetscCall(ISLocalToGlobalMappingDestroy(&lg1l));
173: PetscCall(ISLocalToGlobalMappingDestroy(&lg2l));
174: PetscCall(ISLocalToGlobalMappingCreate(comm, 77, 1, &n, PETSC_USE_POINTER, &lg1l));
175: PetscCall(ISLocalToGlobalMappingCreate(comm, 99, 0, NULL, PETSC_OWN_POINTER, &lg2l));
176: PetscCall(PetscViewerBinaryOpen(comm, "testfile_seq_0", FILE_MODE_READ, &vx));
177: PetscCall(PetscPrintf(comm, "View sequential maps from rank 0 loaded from subcomm\n"));
178: PetscCall(ISLocalToGlobalMappingLoad(lg1l, vx));
179: PetscCall(ISLocalToGlobalMappingLoad(lg2l, vx));
180: PetscCall(ISLocalToGlobalMappingView(lg1l, PETSC_VIEWER_STDOUT_(comm)));
181: PetscCall(ISLocalToGlobalMappingView(lg2l, PETSC_VIEWER_STDOUT_(comm)));
182: PetscCall(PetscViewerDestroy(&vx));
183: }
184: PetscCallMPI(MPI_Comm_free(&comm));
186: PetscCall(ISLocalToGlobalMappingDestroy(&lg1lh));
187: PetscCall(ISLocalToGlobalMappingDestroy(&lg2lh));
188: PetscCall(ISLocalToGlobalMappingDestroy(&lg1l));
189: PetscCall(ISLocalToGlobalMappingDestroy(&lg2l));
190: PetscCall(ISLocalToGlobalMappingDestroy(&lg1v));
191: PetscCall(ISLocalToGlobalMappingDestroy(&lg2v));
192: PetscCall(PetscFinalize());
193: return 0;
194: }
196: /*TEST
198: testset:
199: args: -viewer_binary_mpiio 0
200: test:
201: output_file: output/ex8_1_1.out
202: suffix: 1_stdio_1
203: nsize: 1
204: test:
205: output_file: output/ex8_1_2.out
206: suffix: 1_stdio_2
207: nsize: 2
208: test:
209: output_file: output/ex8_1_3.out
210: suffix: 1_stdio_3
211: nsize: 3
213: testset:
214: requires: mpiio
215: args: -viewer_binary_mpiio 1
216: test:
217: output_file: output/ex8_1_1.out
218: suffix: 1_mpiio_1
219: nsize: 1
220: test:
221: output_file: output/ex8_1_2.out
222: suffix: 1_mpiio_2
223: nsize: 2
224: test:
225: output_file: output/ex8_1_3.out
226: suffix: 1_mpiio_3
227: nsize: 3
229: TEST*/