Actual source code: ex1.c
1: static char help[] = "Check if DMClone for DMNetwork Correctly Shallow Clones Topology Only \n\n";
3: #include <petscdmnetwork.h>
5: /*
6: CreateStarGraphEdgeList - Create a k-Star Graph Edgelist on current processor
7: Not Collective
9: Input Parameters:
10: . k - order of the star graph (number of edges)
11: . directin - if true direction of edges is towards the center vertex, otherwise they are directed out of the center vertex.
13: Output Parameters:
14: . ne - number of edges of this star graph
15: . edgelist - list of edges for this star graph, this is a one dimensional array with pairs of entries being the two vertices (in global numbering of the vertices) of each edge,
16: [first vertex of first edge, second vertex of first edge, first vertex of second edge, second vertex of second edge, etc].
18: User is responsible for deallocating this memory.
19: */
20: PetscErrorCode CreateStarGraphEdgeList(PetscInt k, PetscBool directin, PetscInt *ne, PetscInt *edgelist[])
21: {
22: PetscFunctionBegin;
23: *ne = k;
24: PetscCall(PetscCalloc1(2 * k, edgelist));
26: if (directin) {
27: for (PetscInt i = 0; i < k; i++) {
28: (*edgelist)[2 * i] = i + 1;
29: (*edgelist)[2 * i + 1] = 0;
30: }
31: } else {
32: for (PetscInt i = 0; i < k; i++) {
33: (*edgelist)[2 * i] = 0;
34: (*edgelist)[2 * i + 1] = i + 1;
35: }
36: }
37: PetscFunctionReturn(PETSC_SUCCESS);
38: }
40: /*
41: CreateSimpleStarGraph - Create a Distributed k-Star Graph DMNetwork with a single PetscInt component on
42: all edges and vertices, aselectable number of dofs on vertices and edges. Intended mostly to be used for testing purposes.
44: Input Parameters:
45: . comm - the communicator of the dm
46: . numdofvert - number of degrees of freedom (dofs) on vertices
47: . numdofedge - number of degrees of freedom (dofs) on edges
48: . k - order of the star graph (number of edges)
49: . directin - if true direction of edges is towards the center vertex, otherwise they are directed out of the center vertex
51: Output Parameter:
52: . newdm - The created and distributed simple Star Graph
53: */
54: PetscErrorCode CreateSimpleStarGraph(MPI_Comm comm, PetscInt numdofvert, PetscInt numdofedge, PetscInt k, PetscBool directin, DM *newdm)
55: {
56: DM dm;
57: PetscMPIInt rank;
58: PetscInt ne = 0, compkey, eStart, eEnd, vStart, vEnd, e, v;
59: PetscInt *edgelist = NULL, *compedge, *compvert;
61: PetscFunctionBegin;
62: PetscCall(DMNetworkCreate(comm, &dm));
63: PetscCall(DMNetworkSetNumSubNetworks(dm, PETSC_DECIDE, 1));
64: PetscCallMPI(MPI_Comm_rank(comm, &rank));
65: if (rank == 0) PetscCall(CreateStarGraphEdgeList(k, directin, &ne, &edgelist));
66: PetscCall(DMNetworkAddSubnetwork(dm, "Main", ne, edgelist, NULL));
67: PetscCall(DMNetworkRegisterComponent(dm, "unused", sizeof(PetscInt), &compkey));
68: PetscCall(DMNetworkLayoutSetUp(dm));
69: PetscCall(PetscFree(edgelist));
70: PetscCall(DMNetworkGetEdgeRange(dm, &eStart, &eEnd));
71: PetscCall(DMNetworkGetVertexRange(dm, &vStart, &vEnd));
72: PetscCall(PetscMalloc2(eEnd - eStart, &compedge, vEnd - vStart, &compvert));
73: for (e = eStart; e < eEnd; e++) {
74: compedge[e - eStart] = e;
75: PetscCall(DMNetworkAddComponent(dm, e, compkey, &compedge[e - eStart], numdofedge));
76: }
77: for (v = vStart; v < vEnd; v++) {
78: compvert[v - vStart] = v;
79: PetscCall(DMNetworkAddComponent(dm, v, compkey, &compvert[v - vStart], numdofvert));
80: }
81: PetscCall(DMSetFromOptions(dm));
82: PetscCall(DMViewFromOptions(dm, NULL, "-dm_view"));
83: PetscCall(DMSetUp(dm));
84: PetscCall(PetscFree2(compedge, compvert));
85: PetscCall(DMNetworkDistribute(&dm, 0));
86: *newdm = dm;
87: PetscFunctionReturn(PETSC_SUCCESS);
88: }
90: int main(int argc, char **argv)
91: {
92: DM dm, dmclone, plex;
93: PetscInt e, eStart, eEnd, ndofs, ndofsprev;
94: PetscInt *compprev, *comp, compkey;
95: PetscInt dofv = 1, dofe = 1, ne = 1;
96: PetscSection sec;
97: Vec vec;
99: PetscFunctionBeginUser;
100: PetscCall(PetscInitialize(&argc, &argv, NULL, help));
101: /* create a distributed k-Star graph DMNetwork */
102: PetscCall(PetscOptionsGetInt(NULL, NULL, "-dofv", &dofv, NULL));
103: PetscCall(PetscOptionsGetInt(NULL, NULL, "-dofe", &dofe, NULL));
104: PetscCall(PetscOptionsGetInt(NULL, NULL, "-ne", &ne, NULL));
105: PetscCall(CreateSimpleStarGraph(PETSC_COMM_WORLD, dofv, dofe, ne, PETSC_TRUE, &dm));
106: PetscCall(DMNetworkGetEdgeRange(dm, &eStart, &eEnd));
108: /* check if cloning changed any component */
109: if (eStart < eEnd) PetscCall(DMNetworkGetComponent(dm, eStart, 0, NULL, (void **)&compprev, &ndofsprev));
110: PetscCall(DMClone(dm, &dmclone));
111: if (eStart < eEnd) {
112: PetscCall(DMNetworkGetComponent(dm, eStart, 0, NULL, (void **)&comp, &ndofs));
113: PetscCheck(*comp == *compprev, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Cloning changed the Original, comp (previous) : %" PetscInt_FMT " comp (now) : %" PetscInt_FMT, *compprev, *comp);
114: PetscCheck(ndofsprev == ndofs, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Cloning changed the Original, ndofs (previous) : %" PetscInt_FMT " ndofs (now) : %" PetscInt_FMT, ndofsprev, ndofs);
115: }
117: /* register new components to the clone and add a unused component to every point */
118: PetscCall(DMNetworkRegisterComponent(dmclone, "unusedclone", sizeof(PetscInt), &compkey));
119: PetscCall(DMNetworkGetEdgeRange(dmclone, &eStart, &eEnd));
120: PetscCall(PetscMalloc1(eEnd - eStart, &comp));
121: for (e = eStart; e < eEnd; e++) {
122: comp[e - eStart] = e;
123: PetscCall(DMNetworkAddComponent(dmclone, e, compkey, &comp[e - eStart], 2));
124: }
125: PetscCall(DMNetworkFinalizeComponents(dmclone));
126: PetscCall(PetscFree(comp));
128: PetscCall(PetscPrintf(PETSC_COMM_WORLD, " dm: \n"));
129: PetscCall(DMView(dm, PETSC_VIEWER_STDOUT_WORLD));
130: PetscCall(DMNetworkGetPlex(dm, &plex));
131: PetscCall(DMGetLocalSection(plex, &sec));
132: PetscCall(PetscSectionView(sec, PETSC_VIEWER_STDOUT_WORLD));
134: PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\n dmclone: \n"));
135: PetscCall(DMView(dmclone, PETSC_VIEWER_STDOUT_WORLD));
136: PetscCall(DMNetworkGetPlex(dmclone, &plex));
137: PetscCall(DMGetLocalSection(plex, &sec));
138: PetscCall(PetscSectionView(sec, PETSC_VIEWER_STDOUT_WORLD));
140: /* create Vectors */
141: PetscCall(DMCreateGlobalVector(dm, &vec));
142: PetscCall(VecSet(vec, 1.0));
143: PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\n dm vec:\n"));
144: PetscCall(VecView(vec, PETSC_VIEWER_STDOUT_WORLD));
145: PetscCall(VecDestroy(&vec));
147: PetscCall(DMCreateGlobalVector(dmclone, &vec));
148: PetscCall(VecSet(vec, 2.0));
149: PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\n dmclone vec:\n"));
150: PetscCall(VecView(vec, PETSC_VIEWER_STDOUT_WORLD));
151: PetscCall(VecDestroy(&vec));
153: PetscCall(DMDestroy(&dm));
154: PetscCall(DMDestroy(&dmclone));
155: PetscCall(PetscFinalize());
156: }
158: /*TEST
160: test:
161: suffix: 0
162: args:
164: test:
165: suffix: 1
166: nsize: 2
167: args: -dofv 2 -dofe 2 -ne 2
169: TEST*/