Lines Matching refs:networkdm
21 PetscErrorCode FormFunction_Water(DM networkdm, Vec localX, Vec localF, PetscInt nv, PetscInt ne, c… in FormFunction_Water() argument
42 PetscCall(DMNetworkGetComponent(networkdm, edges[i], 0, &key, (void **)&edge, NULL)); in FormFunction_Water()
45 PetscCall(DMNetworkGetConnectedVertices(networkdm, edges[i], &cone)); in FormFunction_Water()
50 PetscCall(DMNetworkGetNumComponents(networkdm, vnode1, &ncomp)); in FormFunction_Water()
51 … PetscCall(DMNetworkGetComponent(networkdm, vnode1, ncomp - 1, &key, (void **)&vertexnode1, NULL)); in FormFunction_Water()
52 PetscCall(DMNetworkGetLocalVecOffset(networkdm, vnode1, ncomp - 1, &offsetnode1)); in FormFunction_Water()
54 PetscCall(DMNetworkGetNumComponents(networkdm, vnode2, &ncomp)); in FormFunction_Water()
55 … PetscCall(DMNetworkGetComponent(networkdm, vnode2, ncomp - 1, &key, (void **)&vertexnode2, NULL)); in FormFunction_Water()
56 PetscCall(DMNetworkGetLocalVecOffset(networkdm, vnode2, ncomp - 1, &offsetnode2)); in FormFunction_Water()
77 PetscCall(DMNetworkIsGhostVertex(networkdm, vtx[i], &ghostvtex)); in FormFunction_Water()
80 PetscCall(DMNetworkGetLocalVecOffset(networkdm, vtx[i], ALL_COMPONENTS, &offset)); in FormFunction_Water()
81 PetscCall(DMNetworkGetNumComponents(networkdm, vtx[i], &ncomp)); in FormFunction_Water()
82 PetscCall(DMNetworkGetComponent(networkdm, vtx[i], ncomp - 1, &key, (void **)&vertex, NULL)); in FormFunction_Water()
102 DM networkdm; in WaterFormFunction() local
109 PetscCall(SNESGetDM(snes, &networkdm)); in WaterFormFunction()
112 PetscCall(DMNetworkGetSubnetwork(networkdm, 0, &nv, &ne, &v, &e)); in WaterFormFunction()
115 PetscCall(DMGetLocalVector(networkdm, &localX)); in WaterFormFunction()
116 PetscCall(DMGetLocalVector(networkdm, &localF)); in WaterFormFunction()
119 PetscCall(DMGlobalToLocalBegin(networkdm, X, INSERT_VALUES, localX)); in WaterFormFunction()
120 PetscCall(DMGlobalToLocalEnd(networkdm, X, INSERT_VALUES, localX)); in WaterFormFunction()
126 PetscCall(FormFunction_Water(networkdm, localX, localF, nv, ne, v, e, NULL)); in WaterFormFunction()
128 PetscCall(DMRestoreLocalVector(networkdm, &localX)); in WaterFormFunction()
129 PetscCall(DMLocalToGlobalBegin(networkdm, localF, ADD_VALUES, F)); in WaterFormFunction()
130 PetscCall(DMLocalToGlobalEnd(networkdm, localF, ADD_VALUES, F)); in WaterFormFunction()
132 PetscCall(DMRestoreLocalVector(networkdm, &localF)); in WaterFormFunction()
136 PetscErrorCode WaterSetInitialGuess(DM networkdm, Vec X) in WaterSetInitialGuess() argument
143 PetscCall(DMGetLocalVector(networkdm, &localX)); in WaterSetInitialGuess()
146 PetscCall(DMGlobalToLocalBegin(networkdm, X, INSERT_VALUES, localX)); in WaterSetInitialGuess()
147 PetscCall(DMGlobalToLocalEnd(networkdm, X, INSERT_VALUES, localX)); in WaterSetInitialGuess()
150 PetscCall(DMNetworkGetSubnetwork(networkdm, 0, &nv, &ne, &vtx, &edges)); in WaterSetInitialGuess()
151 PetscCall(SetInitialGuess_Water(networkdm, localX, nv, ne, vtx, edges, NULL)); in WaterSetInitialGuess()
153 PetscCall(DMLocalToGlobalBegin(networkdm, localX, ADD_VALUES, X)); in WaterSetInitialGuess()
154 PetscCall(DMLocalToGlobalEnd(networkdm, localX, ADD_VALUES, X)); in WaterSetInitialGuess()
155 PetscCall(DMRestoreLocalVector(networkdm, &localX)); in WaterSetInitialGuess()
198 PetscErrorCode SetInitialGuess_Water(DM networkdm, Vec localX, PetscInt nv, PetscInt ne, const Pets… in SetInitialGuess_Water() argument
208 PetscCall(DMNetworkIsGhostVertex(networkdm, vtx[i], &ghostvtex)); in SetInitialGuess_Water()
209 PetscCall(DMNetworkIsSharedVertex(networkdm, vtx[i], &sharedv)); in SetInitialGuess_Water()
212 PetscCall(DMNetworkGetComponent(networkdm, vtx[i], 0, &key, (void **)&vertex, NULL)); in SetInitialGuess_Water()
213 PetscCall(DMNetworkGetLocalVecOffset(networkdm, vtx[i], 0, &offset)); in SetInitialGuess_Water()