/*@C
DMNetworkMonitorAdd - Adds a new viewer to monitor
Collective on DMNetworkMonitor
Input Parameters:
+ monitor - the monitor
. name - name of viewer
. element - vertex / edge number
. nodes - number of nodes
. start - variable starting offset
. blocksize - variable blocksize
. xmin - xmin (or PETSC_DECIDE) for viewer
. xmax - xmax (or PETSC_DECIDE) for viewer
. ymin - ymin for viewer
. ymax - ymax for viewer
- hold - determines if plot limits should be held
Level: intermediate
Notes:
This is written to be independent of the semantics associated to the variables
at a given network vertex / edge.
Precisely, the parameters nodes, start and blocksize allow you to select a general
strided subarray of the variables to monitor.
.seealso: DMNetworkMonitorCreate(), DMNetworkMonitorDestroy()
@*/
PetscErrorCode DMNetworkMonitorAdd(DMNetworkMonitor monitor,const char *name,PetscInt element,PetscInt nodes,PetscInt start,PetscInt blocksize,PetscReal xmin,PetscReal xmax,PetscReal ymin,PetscReal ymax,PetscBool hold)
{
PetscErrorCode ierr;
PetscDrawLG drawlg;
PetscDrawAxis axis;
PetscMPIInt rank, size;
DMNetworkMonitorList node;
char titleBuffer[64];
PetscInt vStart,vEnd,eStart,eEnd;
PetscFunctionBegin;
ierr = MPI_Comm_rank(monitor->comm, &rank);CHKERRQ(ierr);
ierr = MPI_Comm_size(monitor->comm, &size);CHKERRQ(ierr);
ierr = DMNetworkGetVertexRange(monitor->network, &vStart, &vEnd);CHKERRQ(ierr);
ierr = DMNetworkGetEdgeRange(monitor->network, &eStart, &eEnd);CHKERRQ(ierr);
/* Make window title */
if (vStart <= element && element < vEnd) {
ierr = PetscSNPrintf(titleBuffer, 64, "%s @ vertex %d [%d / %d]", name, element - vStart, rank, size-1);CHKERRQ(ierr);
} else if (eStart <= element && element < eEnd) {
ierr = PetscSNPrintf(titleBuffer, 64, "%s @ edge %d [%d / %d]", name, element - eStart, rank, size-1);CHKERRQ(ierr);
} else {
/* vertex / edge is not on local machine, so skip! */
PetscFunctionReturn(0);
}
ierr = PetscMalloc1(1, &node);CHKERRQ(ierr);
/* Setup viewer. */
ierr = PetscViewerDrawOpen(monitor->comm, NULL, titleBuffer, PETSC_DECIDE, PETSC_DECIDE, PETSC_DRAW_QUARTER_SIZE, PETSC_DRAW_QUARTER_SIZE, &(node->viewer));CHKERRQ(ierr);
ierr = PetscViewerPushFormat(node->viewer, PETSC_VIEWER_DRAW_LG_XRANGE);CHKERRQ(ierr);
ierr = PetscViewerDrawGetDrawLG(node->viewer, 0, &drawlg);CHKERRQ(ierr);
ierr = PetscDrawLGGetAxis(drawlg, &axis);CHKERRQ(ierr);
if (xmin != PETSC_DECIDE && xmax != PETSC_DECIDE) {
ierr = PetscDrawAxisSetLimits(axis, xmin, xmax, ymin, ymax);CHKERRQ(ierr);
} else {
ierr = PetscDrawAxisSetLimits(axis, 0, nodes-1, ymin, ymax);CHKERRQ(ierr);
}
ierr = PetscDrawAxisSetHoldLimits(axis, hold);CHKERRQ(ierr);
/* Setup vector storage for drawing. */
ierr = VecCreateSeq(PETSC_COMM_SELF, nodes, &(node->v));CHKERRQ(ierr);
node->element = element;
node->nodes = nodes;
node->start = start;
node->blocksize = blocksize;
node->next = monitor->firstnode;
monitor->firstnode = node;
PetscFunctionReturn(0);
}
int main(int argc,char ** argv)
{
PetscErrorCode ierr;
char pfdata_file[PETSC_MAX_PATH_LEN]="datafiles/case9.m";
PFDATA *pfdata;
PetscInt numEdges=0,numVertices=0;
int *edges = NULL;
PetscInt i;
DM networkdm;
PetscInt componentkey[4];
UserCtx User;
PetscLogStage stage1,stage2;
PetscMPIInt size,rank;
PetscInt eStart, eEnd, vStart, vEnd,j;
PetscInt genj,loadj;
Vec X,F;
Mat J;
SNES snes;
ierr = PetscInitialize(&argc,&argv,"pfoptions",help);CHKERRQ(ierr);
ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
{
/* introduce the const crank so the clang static analyzer realizes that if it enters any of the if (crank) then it must have entered the first */
/* this is an experiment to see how the analyzer reacts */
const PetscMPIInt crank = rank;
/* Create an empty network object */
ierr = DMNetworkCreate(PETSC_COMM_WORLD,&networkdm);CHKERRQ(ierr);
/* Register the components in the network */
ierr = DMNetworkRegisterComponent(networkdm,"branchstruct",sizeof(struct _p_EDGEDATA),&componentkey[0]);CHKERRQ(ierr);
ierr = DMNetworkRegisterComponent(networkdm,"busstruct",sizeof(struct _p_VERTEXDATA),&componentkey[1]);CHKERRQ(ierr);
ierr = DMNetworkRegisterComponent(networkdm,"genstruct",sizeof(struct _p_GEN),&componentkey[2]);CHKERRQ(ierr);
ierr = DMNetworkRegisterComponent(networkdm,"loadstruct",sizeof(struct _p_LOAD),&componentkey[3]);CHKERRQ(ierr);
ierr = PetscLogStageRegister("Read Data",&stage1);CHKERRQ(ierr);
PetscLogStagePush(stage1);
/* READ THE DATA */
if (!crank) {
/* READ DATA */
/* Only rank 0 reads the data */
ierr = PetscOptionsGetString(NULL,NULL,"-pfdata",pfdata_file,PETSC_MAX_PATH_LEN-1,NULL);CHKERRQ(ierr);
ierr = PetscNew(&pfdata);CHKERRQ(ierr);
ierr = PFReadMatPowerData(pfdata,pfdata_file);CHKERRQ(ierr);
User.Sbase = pfdata->sbase;
numEdges = pfdata->nbranch;
numVertices = pfdata->nbus;
ierr = PetscMalloc(2*numEdges*sizeof(int),&edges);CHKERRQ(ierr);
ierr = GetListofEdges(pfdata->nbranch,pfdata->branch,edges);CHKERRQ(ierr);
}
PetscLogStagePop();
ierr = MPI_Barrier(PETSC_COMM_WORLD);CHKERRQ(ierr);
ierr = PetscLogStageRegister("Create network",&stage2);CHKERRQ(ierr);
PetscLogStagePush(stage2);
/* Set number of nodes/edges */
ierr = DMNetworkSetSizes(networkdm,numVertices,numEdges,PETSC_DETERMINE,PETSC_DETERMINE);CHKERRQ(ierr);
/* Add edge connectivity */
ierr = DMNetworkSetEdgeList(networkdm,edges);CHKERRQ(ierr);
/* Set up the network layout */
ierr = DMNetworkLayoutSetUp(networkdm);CHKERRQ(ierr);
if (!crank) {
ierr = PetscFree(edges);CHKERRQ(ierr);
}
/* Add network components only process 0 has any data to add*/
if (!crank) {
genj=0; loadj=0;
ierr = DMNetworkGetEdgeRange(networkdm,&eStart,&eEnd);CHKERRQ(ierr);
for (i = eStart; i < eEnd; i++) {
ierr = DMNetworkAddComponent(networkdm,i,componentkey[0],&pfdata->branch[i-eStart]);CHKERRQ(ierr);
}
ierr = DMNetworkGetVertexRange(networkdm,&vStart,&vEnd);CHKERRQ(ierr);
for (i = vStart; i < vEnd; i++) {
ierr = DMNetworkAddComponent(networkdm,i,componentkey[1],&pfdata->bus[i-vStart]);CHKERRQ(ierr);
if (pfdata->bus[i-vStart].ngen) {
for (j = 0; j < pfdata->bus[i-vStart].ngen; j++) {
ierr = DMNetworkAddComponent(networkdm,i,componentkey[2],&pfdata->gen[genj++]);CHKERRQ(ierr);
}
}
if (pfdata->bus[i-vStart].nload) {
for (j=0; j < pfdata->bus[i-vStart].nload; j++) {
ierr = DMNetworkAddComponent(networkdm,i,componentkey[3],&pfdata->load[loadj++]);CHKERRQ(ierr);
}
}
/* Add number of variables */
ierr = DMNetworkAddNumVariables(networkdm,i,2);CHKERRQ(ierr);
}
}
/* Set up DM for use */
ierr = DMSetUp(networkdm);CHKERRQ(ierr);
if (!crank) {
ierr = PetscFree(pfdata->bus);CHKERRQ(ierr);
ierr = PetscFree(pfdata->gen);CHKERRQ(ierr);
ierr = PetscFree(pfdata->branch);CHKERRQ(ierr);
ierr = PetscFree(pfdata->load);CHKERRQ(ierr);
ierr = PetscFree(pfdata);CHKERRQ(ierr);
}
ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
if (size > 1) {
DM distnetworkdm;
/* Network partitioning and distribution of data */
ierr = DMNetworkDistribute(networkdm,0,&distnetworkdm);CHKERRQ(ierr);
ierr = DMDestroy(&networkdm);CHKERRQ(ierr);
networkdm = distnetworkdm;
}
PetscLogStagePop();
ierr = DMNetworkGetEdgeRange(networkdm,&eStart,&eEnd);CHKERRQ(ierr);
ierr = DMNetworkGetVertexRange(networkdm,&vStart,&vEnd);CHKERRQ(ierr);
#if 0
PetscInt numComponents;
EDGEDATA edge;
PetscInt offset,key,kk;
DMNetworkComponentGenericDataType *arr;
VERTEXDATA bus;
GEN gen;
LOAD load;
for (i = eStart; i < eEnd; i++) {
ierr = DMNetworkGetComponentDataArray(networkdm,&arr);CHKERRQ(ierr);
ierr = DMNetworkGetComponentTypeOffset(networkdm,i,0,&key,&offset);CHKERRQ(ierr);
edge = (EDGEDATA)(arr+offset);
ierr = DMNetworkGetNumComponents(networkdm,i,&numComponents);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_SELF,"Rank %d ncomps = %d Line %d ---- %d\n",crank,numComponents,edge->internal_i,edge->internal_j);CHKERRQ(ierr);
}
for (i = vStart; i < vEnd; i++) {
ierr = DMNetworkGetComponentDataArray(networkdm,&arr);CHKERRQ(ierr);
ierr = DMNetworkGetNumComponents(networkdm,i,&numComponents);CHKERRQ(ierr);
for (kk=0; kk < numComponents; kk++) {
ierr = DMNetworkGetComponentTypeOffset(networkdm,i,kk,&key,&offset);CHKERRQ(ierr);
if (key == 1) {
bus = (VERTEXDATA)(arr+offset);
ierr = PetscPrintf(PETSC_COMM_SELF,"Rank %d ncomps = %d Bus %d\n",crank,numComponents,bus->internal_i);CHKERRQ(ierr);
} else if (key == 2) {
gen = (GEN)(arr+offset);
ierr = PetscPrintf(PETSC_COMM_SELF,"Rank %d Gen pg = %f qg = %f\n",crank,gen->pg,gen->qg);CHKERRQ(ierr);
} else if (key == 3) {
load = (LOAD)(arr+offset);
ierr = PetscPrintf(PETSC_COMM_SELF,"Rank %d Load pl = %f ql = %f\n",crank,load->pl,load->ql);CHKERRQ(ierr);
}
}
}
#endif
/* Broadcast Sbase to all processors */
ierr = MPI_Bcast(&User.Sbase,1,MPIU_SCALAR,0,PETSC_COMM_WORLD);CHKERRQ(ierr);
ierr = DMCreateGlobalVector(networkdm,&X);CHKERRQ(ierr);
ierr = VecDuplicate(X,&F);CHKERRQ(ierr);
ierr = DMCreateMatrix(networkdm,&J);CHKERRQ(ierr);
ierr = MatSetOption(J,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr);
ierr = SetInitialValues(networkdm,X,&User);CHKERRQ(ierr);
/* HOOK UP SOLVER */
ierr = SNESCreate(PETSC_COMM_WORLD,&snes);CHKERRQ(ierr);
ierr = SNESSetDM(snes,networkdm);CHKERRQ(ierr);
ierr = SNESSetFunction(snes,F,FormFunction,&User);CHKERRQ(ierr);
ierr = SNESSetJacobian(snes,J,J,FormJacobian,&User);CHKERRQ(ierr);
ierr = SNESSetFromOptions(snes);CHKERRQ(ierr);
ierr = SNESSolve(snes,NULL,X);CHKERRQ(ierr);
ierr = VecDestroy(&X);CHKERRQ(ierr);
ierr = VecDestroy(&F);CHKERRQ(ierr);
ierr = MatDestroy(&J);CHKERRQ(ierr);
ierr = SNESDestroy(&snes);CHKERRQ(ierr);
ierr = DMDestroy(&networkdm);CHKERRQ(ierr);
}
ierr = PetscFinalize();
return ierr;
}
PetscErrorCode FormOperator(DM networkdm,Mat A,Vec b)
{
PetscErrorCode ierr;
Vec localb;
Branch *branch;
Node *node;
PetscInt e,v,vStart,vEnd,eStart, eEnd;
PetscInt lofst,lofst_to,lofst_fr,row[2],col[6];
PetscBool ghost;
const PetscInt *cone;
PetscScalar *barr,val[6];
PetscFunctionBegin;
ierr = DMGetLocalVector(networkdm,&localb);CHKERRQ(ierr);
ierr = VecSet(b,0.0);CHKERRQ(ierr);
ierr = VecSet(localb,0.0);CHKERRQ(ierr);
ierr = MatZeroEntries(A);CHKERRQ(ierr);
ierr = VecGetArray(localb,&barr);CHKERRQ(ierr);
/*
We can define the current as a "edge characteristic" and the voltage
and the voltage as a "vertex characteristic". With that, we can iterate
the list of edges and vertices, query the associated voltages and currents
and use them to write the Kirchoff equations.
*/
/* Branch equations: i/r + uj - ui = battery */
ierr = DMNetworkGetEdgeRange(networkdm,&eStart,&eEnd);CHKERRQ(ierr);
for (e = 0; e < eEnd; e++) {
ierr = DMNetworkGetComponent(networkdm,e,0,NULL,(void**)&branch);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,e,&lofst);CHKERRQ(ierr);
ierr = DMNetworkGetConnectedVertices(networkdm,e,&cone);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,cone[0],&lofst_fr);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,cone[1],&lofst_to);CHKERRQ(ierr);
barr[lofst] = branch->bat;
row[0] = lofst;
col[0] = lofst; val[0] = 1;
col[1] = lofst_to; val[1] = 1;
col[2] = lofst_fr; val[2] = -1;
ierr = MatSetValuesLocal(A,1,row,3,col,val,ADD_VALUES);CHKERRQ(ierr);
/* from node */
ierr = DMNetworkGetComponent(networkdm,cone[0],0,NULL,(void**)&node);CHKERRQ(ierr);
if (!node->gr) {
row[0] = lofst_fr;
col[0] = lofst; val[0] = 1;
ierr = MatSetValuesLocal(A,1,row,1,col,val,ADD_VALUES);CHKERRQ(ierr);
}
/* to node */
ierr = DMNetworkGetComponent(networkdm,cone[1],0,NULL,(void**)&node);CHKERRQ(ierr);
if (!node->gr) {
row[0] = lofst_to;
col[0] = lofst; val[0] = -1;
ierr = MatSetValuesLocal(A,1,row,1,col,val,ADD_VALUES);CHKERRQ(ierr);
}
}
ierr = DMNetworkGetVertexRange(networkdm,&vStart,&vEnd);CHKERRQ(ierr);
for (v = vStart; v < vEnd; v++) {
ierr = DMNetworkIsGhostVertex(networkdm,v,&ghost);CHKERRQ(ierr);
if (!ghost) {
ierr = DMNetworkGetComponent(networkdm,v,0,NULL,(void**)&node);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,v,&lofst);CHKERRQ(ierr);
if (node->gr) {
row[0] = lofst;
col[0] = lofst; val[0] = 1;
ierr = MatSetValuesLocal(A,1,row,1,col,val,ADD_VALUES);CHKERRQ(ierr);
} else {
barr[lofst] -= node->inj;
}
}
}
ierr = VecRestoreArray(localb,&barr);CHKERRQ(ierr);
ierr = DMLocalToGlobalBegin(networkdm,localb,ADD_VALUES,b);CHKERRQ(ierr);
ierr = DMLocalToGlobalEnd(networkdm,localb,ADD_VALUES,b);CHKERRQ(ierr);
ierr = DMRestoreLocalVector(networkdm,&localb);CHKERRQ(ierr);
ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
int main(int argc,char ** argv)
{
PetscErrorCode ierr;
PetscInt i, nbranch = 0, eStart, eEnd, vStart, vEnd;
PetscInt seed = 0, nnode = 0;
PetscMPIInt size, rank;
DM networkdm;
Vec x, b;
Mat A;
KSP ksp;
PetscInt *edgelist = NULL;
PetscInt componentkey[2];
Node *node;
Branch *branch;
#if defined(PETSC_USE_LOG)
PetscLogStage stage[3];
#endif
ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr;
ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-seed",&seed,NULL);CHKERRQ(ierr);
ierr = PetscLogStageRegister("Network Creation", &stage[0]);CHKERRQ(ierr);
ierr = PetscLogStageRegister("DMNetwork data structures", &stage[1]);CHKERRQ(ierr);
ierr = PetscLogStageRegister("KSP", &stage[2]);CHKERRQ(ierr);
ierr = PetscLogStagePush(stage[0]);CHKERRQ(ierr);
/* "read" data only for processor 0 */
if (!rank) {
nnode = 100;
ierr = PetscOptionsGetInt(NULL,NULL,"-n",&nnode,NULL);CHKERRQ(ierr);
ierr = random_network(nnode, &nbranch, &node, &branch, &edgelist, seed);CHKERRQ(ierr);
}
ierr = PetscLogStagePop();CHKERRQ(ierr);
ierr = PetscLogStagePush(stage[1]);CHKERRQ(ierr);
ierr = DMNetworkCreate(PETSC_COMM_WORLD,&networkdm);CHKERRQ(ierr);
ierr = DMNetworkRegisterComponent(networkdm,"nstr",sizeof(Node),&componentkey[0]);CHKERRQ(ierr);
ierr = DMNetworkRegisterComponent(networkdm,"bsrt",sizeof(Branch),&componentkey[1]);CHKERRQ(ierr);
/* Set number of nodes/edges */
ierr = DMNetworkSetSizes(networkdm,1,0,&nnode,&nbranch,NULL,NULL);CHKERRQ(ierr);
/* Add edge connectivity */
ierr = DMNetworkSetEdgeList(networkdm,&edgelist,NULL);CHKERRQ(ierr);
/* Set up the network layout */
ierr = DMNetworkLayoutSetUp(networkdm);CHKERRQ(ierr);
/* Add network components: physical parameters of nodes and branches*/
if (!rank) {
ierr = DMNetworkGetEdgeRange(networkdm,&eStart,&eEnd);CHKERRQ(ierr);
for (i = eStart; i < eEnd; i++) {
ierr = DMNetworkAddComponent(networkdm,i,componentkey[1],&branch[i-eStart]);CHKERRQ(ierr);
ierr = DMNetworkAddNumVariables(networkdm,i,1);CHKERRQ(ierr);
}
ierr = DMNetworkGetVertexRange(networkdm,&vStart,&vEnd);CHKERRQ(ierr);
for (i = vStart; i < vEnd; i++) {
ierr = DMNetworkAddComponent(networkdm,i,componentkey[0],&node[i-vStart]);CHKERRQ(ierr);
/* Add number of variables */
ierr = DMNetworkAddNumVariables(networkdm,i,1);CHKERRQ(ierr);
}
}
/* Network partitioning and distribution of data */
ierr = DMSetUp(networkdm);CHKERRQ(ierr);
ierr = DMNetworkDistribute(&networkdm,0);CHKERRQ(ierr);
ierr = DMNetworkAssembleGraphStructures(networkdm);CHKERRQ(ierr);
/* We don't use these data structures anymore since they have been copied to networkdm */
if (!rank) {
ierr = PetscFree(edgelist);CHKERRQ(ierr);
ierr = PetscFree2(node,branch);CHKERRQ(ierr);
}
/* Create vectors and matrix */
ierr = DMCreateGlobalVector(networkdm,&x);CHKERRQ(ierr);
ierr = VecDuplicate(x,&b);CHKERRQ(ierr);
ierr = DMCreateMatrix(networkdm,&A);CHKERRQ(ierr);
ierr = PetscLogStagePop();CHKERRQ(ierr);
ierr = PetscLogStagePush(stage[2]);CHKERRQ(ierr);
/* Assembly system of equations */
ierr = FormOperator(networkdm,A,b);CHKERRQ(ierr);
/* Solve linear system: A x = b */
ierr = KSPCreate(PETSC_COMM_WORLD, &ksp);CHKERRQ(ierr);
ierr = KSPSetOperators(ksp, A, A);CHKERRQ(ierr);
ierr = KSPSetFromOptions(ksp);CHKERRQ(ierr);
ierr = KSPSolve(ksp, b, x);CHKERRQ(ierr);
ierr = PetscLogStagePop();CHKERRQ(ierr);
/* Free work space */
ierr = VecDestroy(&x);CHKERRQ(ierr);
ierr = VecDestroy(&b);CHKERRQ(ierr);
ierr = MatDestroy(&A);CHKERRQ(ierr);
ierr = KSPDestroy(&ksp);CHKERRQ(ierr);
ierr = DMDestroy(&networkdm);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
PetscErrorCode WASHIFunction(TS ts,PetscReal t,Vec X,Vec Xdot,Vec F,void* ctx)
{
PetscErrorCode ierr;
Wash wash=(Wash)ctx;
DM networkdm;
Vec localX,localXdot,localF;
const PetscInt *cone;
PetscInt vfrom,vto,offsetfrom,offsetto,type,varoffset,voffset;
PetscInt v,vStart,vEnd,e,eStart,eEnd,pipeoffset;
PetscBool ghost;
PetscScalar *farr,*vf,*juncx,*juncf;
Pipe pipe;
PipeField *pipex,*pipexdot,*pipef;
DMDALocalInfo info;
Junction junction;
MPI_Comm comm;
PetscMPIInt rank,size;
const PetscScalar *xarr,*xdotarr;
DMNetworkComponentGenericDataType *nwarr;
PetscFunctionBegin;
ierr = PetscObjectGetComm((PetscObject)ts,&comm);CHKERRQ(ierr);
ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
ierr = VecSet(F,0.0);CHKERRQ(ierr);
localX = wash->localX;
localXdot = wash->localXdot;
ierr = TSGetDM(ts,&networkdm);CHKERRQ(ierr);
ierr = DMGetLocalVector(networkdm,&localF);CHKERRQ(ierr);
ierr = VecSet(localF,0.0);CHKERRQ(ierr);
/* update ghost values of locaX and locaXdot */
ierr = DMGlobalToLocalBegin(networkdm,X,INSERT_VALUES,localX);CHKERRQ(ierr);
ierr = DMGlobalToLocalEnd(networkdm,X,INSERT_VALUES,localX);CHKERRQ(ierr);
ierr = DMGlobalToLocalBegin(networkdm,Xdot,INSERT_VALUES,localXdot);CHKERRQ(ierr);
ierr = DMGlobalToLocalEnd(networkdm,Xdot,INSERT_VALUES,localXdot);CHKERRQ(ierr);
ierr = VecGetArrayRead(localX,&xarr);CHKERRQ(ierr);
ierr = VecGetArrayRead(localXdot,&xdotarr);CHKERRQ(ierr);
ierr = VecGetArray(localF,&farr);CHKERRQ(ierr);
/* Initialize localF = localX at non-ghost vertices */
ierr = DMNetworkGetVertexRange(networkdm,&vStart,&vEnd);CHKERRQ(ierr);
for (v=vStart; v<vEnd; v++) {
ierr = DMNetworkIsGhostVertex(networkdm,v,&ghost);CHKERRQ(ierr);
if (!ghost) {
ierr = DMNetworkGetVariableOffset(networkdm,v,&varoffset);CHKERRQ(ierr);
juncx = (PetscScalar*)(xarr+varoffset);
juncf = (PetscScalar*)(farr+varoffset);
juncf[0] = juncx[0];
juncf[1] = juncx[1];
}
}
/* Get component(application) data array */
ierr = DMNetworkGetComponentDataArray(networkdm,&nwarr);CHKERRQ(ierr);
/* Edge */
ierr = DMNetworkGetEdgeRange(networkdm,&eStart,&eEnd);CHKERRQ(ierr);
for (e=eStart; e<eEnd; e++) {
ierr = DMNetworkGetComponentTypeOffset(networkdm,e,0,&type,&pipeoffset);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,e,&varoffset);CHKERRQ(ierr);
pipe = (Pipe)(nwarr + pipeoffset);
pipex = (PipeField*)(xarr + varoffset);
pipexdot = (PipeField*)(xdotarr + varoffset);
pipef = (PipeField*)(farr + varoffset);
/* Get boundary values H0 and QL from connected vertices */
ierr = DMNetworkGetConnectedNodes(networkdm,e,&cone);CHKERRQ(ierr);
vfrom = cone[0]; /* local ordering */
vto = cone[1];
ierr = DMNetworkGetVariableOffset(networkdm,vfrom,&offsetfrom);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,vto,&offsetto);CHKERRQ(ierr);
if (pipe->boundary.Q0 == PIPE_CHARACTERISTIC) {
pipe->boundary.H0 = (xarr+offsetfrom)[1]; /* h_from */
} else {
pipe->boundary.Q0 = (xarr+offsetfrom)[0]; /* q_from */
}
if (pipe->boundary.HL == PIPE_CHARACTERISTIC) {
pipe->boundary.QL = (xarr+offsetto)[0]; /* q_to */
} else {
pipe->boundary.HL = (xarr+offsetto)[1]; /* h_to */
}
/* Evaluate PipeIFunctionLocal() */
ierr = DMDAGetLocalInfo(pipe->da,&info);CHKERRQ(ierr);
ierr = PipeIFunctionLocal(&info, t, pipex, pipexdot, pipef, pipe);CHKERRQ(ierr);
/* Set F at vfrom */
vf = (PetscScalar*)(farr+offsetfrom);
if (pipe->boundary.Q0 == PIPE_CHARACTERISTIC) {
vf[0] -= pipex[0].q; /* q_vfrom - q[0] */
} else {
vf[1] -= pipex[0].h; /* h_vfrom - h[0] */
}
/* Set F at vto */
vf = (PetscScalar*)(farr+offsetto);
if (pipe->boundary.HL == PIPE_CHARACTERISTIC) {
vf[1] -= pipex[pipe->nnodes-1].h; /* h_vto - h[last] */
} else {
vf[0] -= pipex[pipe->nnodes-1].q; /* q_vto - q[last] */
}
}
/* Set F at boundary vertices */
for (v=vStart; v<vEnd; v++) {
ierr = DMNetworkGetComponentTypeOffset(networkdm,v,0,&type,&voffset);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,v,&varoffset);CHKERRQ(ierr);
junction = (Junction)(nwarr + voffset);
juncf = (PetscScalar *)(farr + varoffset);
if (junction->isEnd == -1) {
juncf[1] -= wash->H0;
} else if (junction->isEnd == 1) {
juncf[0] -= wash->QL;
}
}
ierr = VecRestoreArrayRead(localX,&xarr);CHKERRQ(ierr);
ierr = VecRestoreArrayRead(localXdot,&xdotarr);CHKERRQ(ierr);
ierr = VecRestoreArray(localF,&farr);CHKERRQ(ierr);
ierr = DMLocalToGlobalBegin(networkdm,localF,ADD_VALUES,F);CHKERRQ(ierr);
ierr = DMLocalToGlobalEnd(networkdm,localF,ADD_VALUES,F);CHKERRQ(ierr);
ierr = DMRestoreLocalVector(networkdm,&localF);CHKERRQ(ierr);
/* ierr = VecView(F,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); */
PetscFunctionReturn(0);
}
/* ------------------------------------------------------- */
int main(int argc,char ** argv)
{
PetscErrorCode ierr;
Wash wash;
Junction junctions,junction;
Pipe pipe,pipes;
PetscInt numEdges,numVertices,KeyPipe,KeyJunction;
int *edgelist = NULL;
PetscInt i,e,v,eStart,eEnd,vStart,vEnd,pipeOffset,key,frombType,tobType;
PetscInt vfrom,vto,vkey,fromOffset,toOffset,type,varoffset,pipeoffset;
PetscInt from_nedge_in,from_nedge_out,to_nedge_in;
const PetscInt *cone;
DM networkdm;
PetscMPIInt size,rank;
PetscReal ftime = 2500.0;
Vec X;
TS ts;
PetscInt maxsteps=-1,steps;
TSConvergedReason reason;
PetscBool viewpipes;
PetscInt pipesCase;
DMNetworkMonitor monitor;
DMNetworkComponentGenericDataType *nwarr;
ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr;
ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
/* Create and setup network */
/*--------------------------*/
ierr = DMNetworkCreate(PETSC_COMM_WORLD,&networkdm);CHKERRQ(ierr);
if (size == 1) {
ierr = DMNetworkMonitorCreate(networkdm,&monitor);CHKERRQ(ierr);
}
/* Register the components in the network */
ierr = DMNetworkRegisterComponent(networkdm,"junctionstruct",sizeof(struct _p_Junction),&KeyJunction);CHKERRQ(ierr);
ierr = DMNetworkRegisterComponent(networkdm,"pipestruct",sizeof(struct _p_Pipe),&KeyPipe);CHKERRQ(ierr);
/* Set global number of pipes, edges, and vertices */
pipesCase = 2;
ierr = PetscOptionsGetInt(NULL,NULL, "-case", &pipesCase, NULL);CHKERRQ(ierr);
ierr = WashNetworkCreate(PETSC_COMM_WORLD,pipesCase,&wash,&edgelist);CHKERRQ(ierr);
numEdges = wash->nedge;
numVertices = wash->nvertex;
junctions = wash->junction;
pipes = wash->pipe;
/* Set number of vertices and edges */
ierr = DMNetworkSetSizes(networkdm,numVertices,numEdges,PETSC_DETERMINE,PETSC_DETERMINE);CHKERRQ(ierr);
/* Add edge connectivity */
ierr = DMNetworkSetEdgeList(networkdm,edgelist);CHKERRQ(ierr);
/* Set up the network layout */
ierr = DMNetworkLayoutSetUp(networkdm);CHKERRQ(ierr);
/* Add EDGEDATA component to all edges -- currently networkdm is a sequential network */
ierr = DMNetworkGetEdgeRange(networkdm,&eStart,&eEnd);CHKERRQ(ierr);
ierr = DMNetworkGetVertexRange(networkdm,&vStart,&vEnd);CHKERRQ(ierr);
for (e = eStart; e < eEnd; e++) {
/* Add Pipe component to all edges -- create pipe here */
ierr = DMNetworkAddComponent(networkdm,e,KeyPipe,&pipes[e-eStart]);CHKERRQ(ierr);
/* Add number of variables to each edge */
ierr = DMNetworkAddNumVariables(networkdm,e,2*pipes[e-eStart].nnodes);CHKERRQ(ierr);
if (size == 1) { /* Add monitor -- show Q_{pipes[e-eStart].id}? */
ierr = DMNetworkMonitorAdd(monitor, "Pipe Q", e, pipes[e-eStart].nnodes, 0, 2, -0.8, 0.8, PETSC_TRUE);CHKERRQ(ierr);
ierr = DMNetworkMonitorAdd(monitor, "Pipe H", e, pipes[e-eStart].nnodes, 1, 2, -400.0, 800.0, PETSC_TRUE);CHKERRQ(ierr);
}
}
/* Add Junction component to all vertices */
for (v = vStart; v < vEnd; v++) {
ierr = DMNetworkAddComponent(networkdm,v,KeyJunction,&junctions[v-vStart]);CHKERRQ(ierr);
/* Add number of variables to vertex */
ierr = DMNetworkAddNumVariables(networkdm,v,2);CHKERRQ(ierr);
}
/* Set up DM for use */
ierr = DMSetUp(networkdm);CHKERRQ(ierr);
ierr = WashNetworkCleanUp(wash,edgelist);CHKERRQ(ierr);
/* Network partitioning and distribution of data */
if (size > 1) {
DM distnetworkdm;
ierr = DMNetworkDistribute(networkdm,0,&distnetworkdm);CHKERRQ(ierr);
ierr = DMDestroy(&networkdm);CHKERRQ(ierr);
networkdm = distnetworkdm;
}
/* PipeSetUp -- each process only sets its own pipes */
/*---------------------------------------------------*/
ierr = DMNetworkGetComponentDataArray(networkdm,&nwarr);CHKERRQ(ierr);
ierr = DMNetworkGetEdgeRange(networkdm,&eStart,&eEnd);CHKERRQ(ierr);
for (e=eStart; e<eEnd; e++) { /* each edge has only one component, pipe */
ierr = DMNetworkGetComponentTypeOffset(networkdm,e,0,&type,&pipeoffset);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,e,&varoffset);CHKERRQ(ierr);
pipe = (Pipe)(nwarr + pipeoffset);
/* Setup conntected vertices */
ierr = DMNetworkGetConnectedNodes(networkdm,e,&cone);CHKERRQ(ierr);
vfrom = cone[0]; /* local ordering */
vto = cone[1];
/* vfrom */
ierr = DMNetworkGetComponentTypeOffset(networkdm,vfrom,0,&vkey,&fromOffset);CHKERRQ(ierr);
junction = (Junction)(nwarr+fromOffset);
from_nedge_in = junction->nedges_in;
from_nedge_out = junction->nedges_out;
/* vto */
ierr = DMNetworkGetComponentTypeOffset(networkdm,vto,0,&vkey,&toOffset);CHKERRQ(ierr);
junction = (Junction)(nwarr+toOffset);
to_nedge_in = junction->nedges_in;
pipe->comm = PETSC_COMM_SELF; /* must be set here, otherwise crashes in my mac??? */
wash->nnodes_loc += pipe->nnodes; /* local total num of nodes, will be used by PipesView() */
ierr = PipeSetParameters(pipe,
600.0, /* length */
pipe->nnodes, /* nnodes -- rm from PipeSetParameters */
0.5, /* diameter */
1200.0, /* a */
0.018);CHKERRQ(ierr); /* friction */
/* set boundary conditions for this pipe */
if (from_nedge_in <= 1 && from_nedge_out > 0) {
frombType = 0;
} else {
frombType = 1;
}
if (to_nedge_in == 1) {
tobType = 0;
} else {
tobType = 1;
}
if (frombType == 0) {
pipe->boundary.Q0 = PIPE_CHARACTERISTIC; /* will be obtained from characteristic */
pipe->boundary.H0 = wash->H0;
} else {
pipe->boundary.Q0 = wash->Q0;
pipe->boundary.H0 = PIPE_CHARACTERISTIC; /* will be obtained from characteristic */
}
if (tobType == 0) {
pipe->boundary.QL = wash->QL;
pipe->boundary.HL = PIPE_CHARACTERISTIC; /* will be obtained from characteristic */
} else {
pipe->boundary.QL = PIPE_CHARACTERISTIC; /* will be obtained from characteristic */
pipe->boundary.HL = wash->HL;
}
ierr = PipeSetUp(pipe);CHKERRQ(ierr);
}
/* create vectors */
ierr = DMCreateGlobalVector(networkdm,&X);CHKERRQ(ierr);
ierr = DMCreateLocalVector(networkdm,&wash->localX);CHKERRQ(ierr);
ierr = DMCreateLocalVector(networkdm,&wash->localXdot);CHKERRQ(ierr);
/* Setup solver */
/*--------------------------------------------------------*/
ierr = TSCreate(PETSC_COMM_WORLD,&ts);CHKERRQ(ierr);
ierr = TSSetDM(ts,(DM)networkdm);CHKERRQ(ierr);
ierr = TSSetIFunction(ts,NULL,WASHIFunction,wash);CHKERRQ(ierr);
ierr = TSSetDuration(ts,maxsteps,ftime);CHKERRQ(ierr);
ierr = TSSetExactFinalTime(ts,TS_EXACTFINALTIME_STEPOVER);CHKERRQ(ierr);
ierr = TSSetInitialTimeStep(ts,0.0,0.1);CHKERRQ(ierr);
ierr = TSSetType(ts,TSBEULER);CHKERRQ(ierr);
if (size == 1) {
ierr = TSMonitorSet(ts, TSDMNetworkMonitor, monitor, NULL);CHKERRQ(ierr);
}
ierr = TSSetFromOptions(ts);CHKERRQ(ierr);
ierr = WASHSetInitialSolution(networkdm,X,wash);CHKERRQ(ierr);
ierr = TSSolve(ts,X);CHKERRQ(ierr);
ierr = TSGetSolveTime(ts,&ftime);CHKERRQ(ierr);
ierr = TSGetTimeStepNumber(ts,&steps);CHKERRQ(ierr);
ierr = TSGetConvergedReason(ts,&reason);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD,"%s at time %g after %D steps\n",TSConvergedReasons[reason],(double)ftime,steps);CHKERRQ(ierr);
/* ierr = VecView(X,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); */
/* View solution q and h */
/* --------------------- */
viewpipes = PETSC_FALSE;
ierr = PetscOptionsGetBool(NULL,NULL, "-pipe_view", &viewpipes,NULL);CHKERRQ(ierr);
if (viewpipes) {
ierr = PipesView(X,networkdm,wash);CHKERRQ(ierr);
}
/* Free spaces */
/* ----------- */
ierr = TSDestroy(&ts);CHKERRQ(ierr);
ierr = VecDestroy(&X);CHKERRQ(ierr);
ierr = VecDestroy(&wash->localX);CHKERRQ(ierr);
ierr = VecDestroy(&wash->localXdot);CHKERRQ(ierr);
/* Destroy objects from each pipe that are created in PipeSetUp() */
ierr = DMNetworkGetEdgeRange(networkdm,&eStart, &eEnd);CHKERRQ(ierr);
for (i = eStart; i < eEnd; i++) {
ierr = DMNetworkGetComponentTypeOffset(networkdm,i,0,&key,&pipeOffset);CHKERRQ(ierr);
pipe = (Pipe)(nwarr+pipeOffset);
ierr = DMDestroy(&(pipe->da));CHKERRQ(ierr);
ierr = VecDestroy(&pipe->x);CHKERRQ(ierr);
}
if (size == 1) {
ierr = DMNetworkMonitorDestroy(&monitor);CHKERRQ(ierr);
}
ierr = DMDestroy(&networkdm);CHKERRQ(ierr);
ierr = PetscFree(wash);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
PetscErrorCode PipesView(Vec X,DM networkdm,Wash wash)
{
PetscErrorCode ierr;
Pipe pipe;
DMNetworkComponentGenericDataType *nwarr;
PetscInt pipeOffset,key,Start,End;
PetscMPIInt rank;
PetscInt nx,nnodes,nidx,*idx1,*idx2,*idx1_h,*idx2_h,idx_start,i,k,k1,xstart,j1;
Vec Xq,Xh,localX;
IS is1_q,is2_q,is1_h,is2_h;
VecScatter ctx_q,ctx_h;
PetscFunctionBegin;
ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
/* get num of local and global total nnodes */
nidx = wash->nnodes_loc;
ierr = MPIU_Allreduce(&nidx,&nx,1,MPIU_INT,MPI_SUM,PETSC_COMM_WORLD);CHKERRQ(ierr);
ierr = VecCreate(PETSC_COMM_WORLD,&Xq);CHKERRQ(ierr);
if (rank == 0) { /* all entries of Xq are in proc[0] */
ierr = VecSetSizes(Xq,nx,PETSC_DECIDE);CHKERRQ(ierr);
} else {
ierr = VecSetSizes(Xq,0,PETSC_DECIDE);CHKERRQ(ierr);
}
ierr = VecSetFromOptions(Xq);CHKERRQ(ierr);
ierr = VecSet(Xq,0.0);CHKERRQ(ierr);
ierr = VecDuplicate(Xq,&Xh);CHKERRQ(ierr);
ierr = DMGetLocalVector(networkdm,&localX);CHKERRQ(ierr);
/* set idx1 and idx2 */
ierr = PetscCalloc4(nidx,&idx1,nidx,&idx2,nidx,&idx1_h,nidx,&idx2_h);CHKERRQ(ierr);
ierr = DMNetworkGetComponentDataArray(networkdm,&nwarr);CHKERRQ(ierr);
ierr = DMNetworkGetEdgeRange(networkdm,&Start, &End);CHKERRQ(ierr);
ierr = VecGetOwnershipRange(X,&xstart,NULL);CHKERRQ(ierr);
k1 = 0;
j1 = 0;
for (i = Start; i < End; i++) {
ierr = DMNetworkGetComponentTypeOffset(networkdm,i,0,&key,&pipeOffset);CHKERRQ(ierr);
pipe = (Pipe)(nwarr+pipeOffset);
nnodes = pipe->nnodes;
idx_start = pipe->id*nnodes;
for (k=0; k<nnodes; k++) {
idx1[k1] = xstart + j1*2*nnodes + 2*k;
idx2[k1] = idx_start + k;
idx1_h[k1] = xstart + j1*2*nnodes + 2*k + 1;
idx2_h[k1] = idx_start + k;
k1++;
}
j1++;
}
ierr = ISCreateGeneral(PETSC_COMM_SELF,nidx,idx1,PETSC_COPY_VALUES,&is1_q);CHKERRQ(ierr);
ierr = ISCreateGeneral(PETSC_COMM_SELF,nidx,idx2,PETSC_COPY_VALUES,&is2_q);CHKERRQ(ierr);
ierr = VecScatterCreate(X,is1_q,Xq,is2_q,&ctx_q);CHKERRQ(ierr);
ierr = VecScatterBegin(ctx_q,X,Xq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
ierr = VecScatterEnd(ctx_q,X,Xq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
ierr = ISCreateGeneral(PETSC_COMM_SELF,nidx,idx1_h,PETSC_COPY_VALUES,&is1_h);CHKERRQ(ierr);
ierr = ISCreateGeneral(PETSC_COMM_SELF,nidx,idx2_h,PETSC_COPY_VALUES,&is2_h);CHKERRQ(ierr);
ierr = VecScatterCreate(X,is1_h,Xh,is2_h,&ctx_h);CHKERRQ(ierr);
ierr = VecScatterBegin(ctx_h,X,Xh,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
ierr = VecScatterEnd(ctx_h,X,Xh,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
if (!rank) printf("Xq: \n");
ierr = VecView(Xq,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
if (!rank) printf("Xh: \n");
ierr = VecView(Xh,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
ierr = VecScatterDestroy(&ctx_q);CHKERRQ(ierr);
ierr = PetscFree4(idx1,idx2,idx1_h,idx2_h);CHKERRQ(ierr);
ierr = ISDestroy(&is1_q);CHKERRQ(ierr);
ierr = ISDestroy(&is2_q);CHKERRQ(ierr);
ierr = VecScatterDestroy(&ctx_h);CHKERRQ(ierr);
ierr = ISDestroy(&is1_h);CHKERRQ(ierr);
ierr = ISDestroy(&is2_h);CHKERRQ(ierr);
ierr = VecDestroy(&Xq);CHKERRQ(ierr);
ierr = VecDestroy(&Xh);CHKERRQ(ierr);
ierr = DMRestoreLocalVector(networkdm,&localX);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode WASHSetInitialSolution(DM networkdm,Vec X,Wash wash)
{
PetscErrorCode ierr;
PetscInt k,nx,vkey,vOffset,vfrom,vto,offsetfrom,offsetto;
PetscInt type,varoffset;
PetscInt e,eStart,eEnd,pipeoffset;
Vec localX;
PetscScalar *xarr;
Pipe pipe;
Junction junction;
const PetscInt *cone;
const PetscScalar *xarray;
DMNetworkComponentGenericDataType *nwarr;
PetscFunctionBegin;
ierr = VecSet(X,0.0);CHKERRQ(ierr);
ierr = DMGetLocalVector(networkdm,&localX);CHKERRQ(ierr);
ierr = VecGetArray(localX,&xarr);CHKERRQ(ierr);
/* Get component(application) data array */
ierr = DMNetworkGetComponentDataArray(networkdm,&nwarr);CHKERRQ(ierr);
/* Edge */
ierr = DMNetworkGetEdgeRange(networkdm,&eStart,&eEnd);CHKERRQ(ierr);
for (e=eStart; e<eEnd; e++) {
ierr = DMNetworkGetVariableOffset(networkdm,e,&varoffset);CHKERRQ(ierr);
ierr = DMNetworkGetComponentTypeOffset(networkdm,e,0,&type,&pipeoffset);CHKERRQ(ierr);
/* get from and to vertices */
ierr = DMNetworkGetConnectedNodes(networkdm,e,&cone);CHKERRQ(ierr);
vfrom = cone[0]; /* local ordering */
vto = cone[1];
ierr = DMNetworkGetVariableOffset(networkdm,vfrom,&offsetfrom);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,vto,&offsetto);CHKERRQ(ierr);
/* set initial values for this pipe */
/* Q0=0.477432; H0=150.0; needs to be updated by its succeeding pipe. Use SNESSolve()? */
pipe = (Pipe)(nwarr + pipeoffset);
ierr = PipeComputeSteadyState(pipe, 0.477432, wash->H0);CHKERRQ(ierr);
ierr = VecGetSize(pipe->x,&nx);CHKERRQ(ierr);
ierr = VecGetArrayRead(pipe->x,&xarray);CHKERRQ(ierr);
/* copy pipe->x to xarray */
for (k=0; k<nx; k++) {
(xarr+varoffset)[k] = xarray[k];
}
/* set boundary values into vfrom and vto */
if (pipe->boundary.Q0 == PIPE_CHARACTERISTIC) {
(xarr+offsetfrom)[0] += xarray[0]; /* Q0 -> vfrom[0] */
} else {
(xarr+offsetfrom)[1] += xarray[1]; /* H0 -> vfrom[1] */
}
if (pipe->boundary.HL == PIPE_CHARACTERISTIC) {
(xarr+offsetto)[1] += xarray[nx-1]; /* HL -> vto[1] */
} else {
(xarr+offsetto)[0] += xarray[nx-2]; /* QL -> vto[0] */
}
/* if vform is a head vertex: */
ierr = DMNetworkGetComponentTypeOffset(networkdm,vfrom,0,&vkey,&vOffset);CHKERRQ(ierr);
junction = (Junction)(nwarr+vOffset);
if (junction->isEnd == -1) { /* head junction */
(xarr+offsetfrom)[0] = 0.0; /* 1st Q -- not used */
(xarr+offsetfrom)[1] = wash->H0; /* 1st H */
}
/* if vto is an end vertex: */
ierr = DMNetworkGetComponentTypeOffset(networkdm,vto,0,&vkey,&vOffset);CHKERRQ(ierr);
junction = (Junction)(nwarr+vOffset);
if (junction->isEnd == 1) { /* end junction */
(xarr+offsetto)[0] = wash->QL; /* last Q */
(xarr+offsetto)[1] = 0.0; /* last H -- not used */
}
ierr = VecRestoreArrayRead(pipe->x,&xarray);CHKERRQ(ierr);
}
ierr = VecRestoreArray(localX,&xarr);CHKERRQ(ierr);
ierr = DMLocalToGlobalBegin(networkdm,localX,ADD_VALUES,X);CHKERRQ(ierr);
ierr = DMLocalToGlobalEnd(networkdm,localX,ADD_VALUES,X);CHKERRQ(ierr);
ierr = DMRestoreLocalVector(networkdm,&localX);CHKERRQ(ierr);
#if 0
PetscInt N;
ierr = VecGetSize(X,&N);CHKERRQ(ierr);
printf("initial solution %d:\n",N);
ierr = VecView(X,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
#endif
PetscFunctionReturn(0);
}
PETSC_EXTERN void PETSC_STDCALL dmnetworkgetedgerange_(DM dm,PetscInt *eStart,PetscInt *eEnd, int *__ierr ){
*__ierr = DMNetworkGetEdgeRange(
(DM)PetscToPointer((dm) ),eStart,eEnd);
}
