PetscErrorCode DMNetworkMonitorView(DMNetworkMonitor monitor,Vec x)
{
PetscErrorCode ierr;
PetscInt varoffset,i,start;
const PetscScalar *xx;
PetscScalar *vv;
DMNetworkMonitorList node;
PetscFunctionBegin;
ierr = VecGetArrayRead(x, &xx);CHKERRQ(ierr);
for (node = monitor->firstnode; node; node = node->next) {
ierr = DMNetworkGetVariableGlobalOffset(monitor->network, node->element, &varoffset);CHKERRQ(ierr);
ierr = VecGetArray(node->v, &vv);CHKERRQ(ierr);
start = varoffset + node->start;
for (i = 0; i < node->nodes; i++) {
vv[i] = xx[start+i*node->blocksize];
}
ierr = VecRestoreArray(node->v, &vv);CHKERRQ(ierr);
ierr = VecView(node->v, node->viewer);CHKERRQ(ierr);
}
ierr = VecRestoreArrayRead(x, &xx);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode FormJacobian(SNES snes,Vec X, Mat J,Mat Jpre,void *appctx)
{
PetscErrorCode ierr;
DM networkdm;
UserCtx *User=(UserCtx*)appctx;
Vec localX;
PetscInt e;
PetscInt v,vStart,vEnd,vfrom,vto;
const PetscScalar *xarr;
PetscInt offsetfrom,offsetto,goffsetfrom,goffsetto;
DMNetworkComponentGenericDataType *arr;
PetscInt row[2],col[8];
PetscScalar values[8];
PetscFunctionBegin;
ierr = MatZeroEntries(J);CHKERRQ(ierr);
ierr = SNESGetDM(snes,&networkdm);CHKERRQ(ierr);
ierr = DMGetLocalVector(networkdm,&localX);CHKERRQ(ierr);
ierr = DMGlobalToLocalBegin(networkdm,X,INSERT_VALUES,localX);CHKERRQ(ierr);
ierr = DMGlobalToLocalEnd(networkdm,X,INSERT_VALUES,localX);CHKERRQ(ierr);
ierr = VecGetArrayRead(localX,&xarr);CHKERRQ(ierr);
ierr = DMNetworkGetVertexRange(networkdm,&vStart,&vEnd);CHKERRQ(ierr);
ierr = DMNetworkGetComponentDataArray(networkdm,&arr);CHKERRQ(ierr);
for (v=vStart; v < vEnd; v++) {
PetscInt i,j,offsetd,key;
PetscInt offset,goffset;
PetscScalar Vm;
PetscScalar Sbase=User->Sbase;
VERTEXDATA bus;
PetscBool ghostvtex;
PetscInt numComps;
ierr = DMNetworkIsGhostVertex(networkdm,v,&ghostvtex);CHKERRQ(ierr);
ierr = DMNetworkGetNumComponents(networkdm,v,&numComps);CHKERRQ(ierr);
for (j = 0; j < numComps; j++) {
ierr = DMNetworkGetVariableOffset(networkdm,v,&offset);CHKERRQ(ierr);
ierr = DMNetworkGetVariableGlobalOffset(networkdm,v,&goffset);CHKERRQ(ierr);
ierr = DMNetworkGetComponentTypeOffset(networkdm,v,j,&key,&offsetd);CHKERRQ(ierr);
if (key == 1) {
PetscInt nconnedges;
const PetscInt *connedges;
bus = (VERTEXDATA)(arr+offsetd);
if (!ghostvtex) {
/* Handle reference bus constrained dofs */
if (bus->ide == REF_BUS || bus->ide == ISOLATED_BUS) {
row[0] = goffset; row[1] = goffset+1;
col[0] = goffset; col[1] = goffset+1; col[2] = goffset; col[3] = goffset+1;
values[0] = 1.0; values[1] = 0.0; values[2] = 0.0; values[3] = 1.0;
ierr = MatSetValues(J,2,row,2,col,values,ADD_VALUES);CHKERRQ(ierr);
break;
}
Vm = xarr[offset+1];
/* Shunt injections */
row[0] = goffset; row[1] = goffset+1;
col[0] = goffset; col[1] = goffset+1;
values[0] = values[1] = values[2] = values[3] = 0.0;
if (bus->ide != PV_BUS) {
values[1] = 2.0*Vm*bus->gl/Sbase;
values[3] = -2.0*Vm*bus->bl/Sbase;
}
ierr = MatSetValues(J,2,row,2,col,values,ADD_VALUES);CHKERRQ(ierr);
}
ierr = DMNetworkGetSupportingEdges(networkdm,v,&nconnedges,&connedges);CHKERRQ(ierr);
for (i=0; i < nconnedges; i++) {
EDGEDATA branch;
VERTEXDATA busf,bust;
PetscInt offsetfd,offsettd,keyf,keyt;
PetscScalar Gff,Bff,Gft,Bft,Gtf,Btf,Gtt,Btt;
const PetscInt *cone;
PetscScalar Vmf,Vmt,thetaf,thetat,thetaft,thetatf;
e = connedges[i];
ierr = DMNetworkGetComponentTypeOffset(networkdm,e,0,&key,&offsetd);CHKERRQ(ierr);
branch = (EDGEDATA)(arr+offsetd);
if (!branch->status) continue;
Gff = branch->yff[0];
Bff = branch->yff[1];
Gft = branch->yft[0];
Bft = branch->yft[1];
Gtf = branch->ytf[0];
Btf = branch->ytf[1];
Gtt = branch->ytt[0];
Btt = branch->ytt[1];
ierr = DMNetworkGetConnectedNodes(networkdm,e,&cone);CHKERRQ(ierr);
vfrom = cone[0];
vto = cone[1];
ierr = DMNetworkGetVariableOffset(networkdm,vfrom,&offsetfrom);CHKERRQ(ierr);
ierr = DMNetworkGetVariableOffset(networkdm,vto,&offsetto);CHKERRQ(ierr);
ierr = DMNetworkGetVariableGlobalOffset(networkdm,vfrom,&goffsetfrom);CHKERRQ(ierr);
ierr = DMNetworkGetVariableGlobalOffset(networkdm,vto,&goffsetto);CHKERRQ(ierr);
if (goffsetto < 0) goffsetto = -goffsetto - 1;
thetaf = xarr[offsetfrom];
Vmf = xarr[offsetfrom+1];
thetat = xarr[offsetto];
Vmt = xarr[offsetto+1];
thetaft = thetaf - thetat;
thetatf = thetat - thetaf;
ierr = DMNetworkGetComponentTypeOffset(networkdm,vfrom,0,&keyf,&offsetfd);CHKERRQ(ierr);
ierr = DMNetworkGetComponentTypeOffset(networkdm,vto,0,&keyt,&offsettd);CHKERRQ(ierr);
busf = (VERTEXDATA)(arr+offsetfd);
bust = (VERTEXDATA)(arr+offsettd);
if (vfrom == v) {
if (busf->ide != REF_BUS) {
/* farr[offsetfrom] += Gff*Vmf*Vmf + Vmf*Vmt*(Gft*PetscCosScalar(thetaft) + Bft*PetscSinScalar(thetaft)); */
row[0] = goffsetfrom;
col[0] = goffsetfrom; col[1] = goffsetfrom+1; col[2] = goffsetto; col[3] = goffsetto+1;
values[0] = Vmf*Vmt*(Gft*-PetscSinScalar(thetaft) + Bft*PetscCosScalar(thetaft)); /* df_dthetaf */
values[1] = 2.0*Gff*Vmf + Vmt*(Gft*PetscCosScalar(thetaft) + Bft*PetscSinScalar(thetaft)); /* df_dVmf */
values[2] = Vmf*Vmt*(Gft*PetscSinScalar(thetaft) + Bft*-PetscCosScalar(thetaft)); /* df_dthetat */
values[3] = Vmf*(Gft*PetscCosScalar(thetaft) + Bft*PetscSinScalar(thetaft)); /* df_dVmt */
ierr = MatSetValues(J,1,row,4,col,values,ADD_VALUES);CHKERRQ(ierr);
}
if (busf->ide != PV_BUS && busf->ide != REF_BUS) {
row[0] = goffsetfrom+1;
col[0] = goffsetfrom; col[1] = goffsetfrom+1; col[2] = goffsetto; col[3] = goffsetto+1;
/* farr[offsetfrom+1] += -Bff*Vmf*Vmf + Vmf*Vmt*(-Bft*PetscCosScalar(thetaft) + Gft*PetscSinScalar(thetaft)); */
values[0] = Vmf*Vmt*(Bft*PetscSinScalar(thetaft) + Gft*PetscCosScalar(thetaft));
values[1] = -2.0*Bff*Vmf + Vmt*(-Bft*PetscCosScalar(thetaft) + Gft*PetscSinScalar(thetaft));
values[2] = Vmf*Vmt*(-Bft*PetscSinScalar(thetaft) + Gft*-PetscCosScalar(thetaft));
values[3] = Vmf*(-Bft*PetscCosScalar(thetaft) + Gft*PetscSinScalar(thetaft));
ierr = MatSetValues(J,1,row,4,col,values,ADD_VALUES);CHKERRQ(ierr);
}
} else {
if (bust->ide != REF_BUS) {
row[0] = goffsetto;
col[0] = goffsetto; col[1] = goffsetto+1; col[2] = goffsetfrom; col[3] = goffsetfrom+1;
/* farr[offsetto] += Gtt*Vmt*Vmt + Vmt*Vmf*(Gtf*PetscCosScalar(thetatf) + Btf*PetscSinScalar(thetatf)); */
values[0] = Vmt*Vmf*(Gtf*-PetscSinScalar(thetatf) + Btf*PetscCosScalar(thetaft)); /* df_dthetat */
values[1] = 2.0*Gtt*Vmt + Vmf*(Gtf*PetscCosScalar(thetatf) + Btf*PetscSinScalar(thetatf)); /* df_dVmt */
values[2] = Vmt*Vmf*(Gtf*PetscSinScalar(thetatf) + Btf*-PetscCosScalar(thetatf)); /* df_dthetaf */
values[3] = Vmt*(Gtf*PetscCosScalar(thetatf) + Btf*PetscSinScalar(thetatf)); /* df_dVmf */
ierr = MatSetValues(J,1,row,4,col,values,ADD_VALUES);CHKERRQ(ierr);
}
if (bust->ide != PV_BUS && bust->ide != REF_BUS) {
row[0] = goffsetto+1;
col[0] = goffsetto; col[1] = goffsetto+1; col[2] = goffsetfrom; col[3] = goffsetfrom+1;
/* farr[offsetto+1] += -Btt*Vmt*Vmt + Vmt*Vmf*(-Btf*PetscCosScalar(thetatf) + Gtf*PetscSinScalar(thetatf)); */
values[0] = Vmt*Vmf*(Btf*PetscSinScalar(thetatf) + Gtf*PetscCosScalar(thetatf));
values[1] = -2.0*Btt*Vmt + Vmf*(-Btf*PetscCosScalar(thetatf) + Gtf*PetscSinScalar(thetatf));
values[2] = Vmt*Vmf*(-Btf*PetscSinScalar(thetatf) + Gtf*-PetscCosScalar(thetatf));
values[3] = Vmt*(-Btf*PetscCosScalar(thetatf) + Gtf*PetscSinScalar(thetatf));
ierr = MatSetValues(J,1,row,4,col,values,ADD_VALUES);CHKERRQ(ierr);
}
}
}
if (!ghostvtex && bus->ide == PV_BUS) {
row[0] = goffset+1; col[0] = goffset+1;
values[0] = 1.0;
ierr = MatSetValues(J,1,row,1,col,values,ADD_VALUES);CHKERRQ(ierr);
}
}
}
}
ierr = VecRestoreArrayRead(localX,&xarr);CHKERRQ(ierr);
ierr = DMRestoreLocalVector(networkdm,&localX);CHKERRQ(ierr);
ierr = MatAssemblyBegin(J,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(J,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PETSC_EXTERN void PETSC_STDCALL dmnetworkgetvariableglobaloffset_(DM dm,PetscInt *p,PetscInt *offsetg, int *__ierr ){
*__ierr = DMNetworkGetVariableGlobalOffset(
(DM)PetscToPointer((dm) ),*p,offsetg);
}
