PetscErrorCode TestSetup(DMLabel label, AppCtx *user)
{
PetscRandom r;
PetscInt n = (PetscInt) (user->fill*(user->pEnd - user->pStart)), i;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = PetscRandomCreate(PETSC_COMM_SELF, &r);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(r);CHKERRQ(ierr);/* -random_type <> */
ierr = PetscRandomSetInterval(r, user->pStart, user->pEnd);CHKERRQ(ierr);
ierr = PetscRandomSetSeed(r, 123456789L);CHKERRQ(ierr);
ierr = PetscRandomSeed(r);CHKERRQ(ierr);
user->size = 0;
for(i = 0; i < n; ++i) {
PetscReal p;
PetscInt val;
ierr = PetscRandomGetValueReal(r, &p);CHKERRQ(ierr);
ierr = DMLabelGetValue(label, (PetscInt) p, &val);CHKERRQ(ierr);
if (val < 0) {
++user->size;
ierr = DMLabelSetValue(label, (PetscInt) p, i % user->numStrata);CHKERRQ(ierr);
}
}
ierr = PetscRandomDestroy(&r);CHKERRQ(ierr);
ierr = DMLabelCreateIndex(label, user->pStart, user->pEnd);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_SELF, "Created label with chart [%D, %D) and set %D values\n", user->pStart, user->pEnd, user->size);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@
PetscRandomSetFromOptions - Configures the random number generator from the options database.
Collective on PetscRandom
Input Parameter:
. rnd - The random number generator context
Options Database:
. -random_seed <integer> - provide a seed to the random number generater
Notes: To see all options, run your program with the -help option.
Must be called after PetscRandomCreate() but before the rnd is used.
Level: beginner
.keywords: PetscRandom, set, options, database
.seealso: PetscRandomCreate(), PetscRandomSetType()
@*/
PetscErrorCode PetscRandomSetFromOptions(PetscRandom rnd)
{
PetscErrorCode ierr;
PetscBool set;
PetscInt seed;
PetscFunctionBegin;
PetscValidHeaderSpecific(rnd,PETSC_RANDOM_CLASSID,1);
ierr = PetscObjectOptionsBegin((PetscObject)rnd);CHKERRQ(ierr);
/* Handle PetscRandom type options */
ierr = PetscRandomSetTypeFromOptions_Private(rnd);CHKERRQ(ierr);
/* Handle specific random generator's options */
if (rnd->ops->setfromoptions) {
ierr = (*rnd->ops->setfromoptions)(rnd);CHKERRQ(ierr);
}
ierr = PetscOptionsInt("-random_seed","Seed to use to generate random numbers","PetscRandomSetSeed",0,&seed,&set);CHKERRQ(ierr);
if (set) {
ierr = PetscRandomSetSeed(rnd,(unsigned long int)seed);CHKERRQ(ierr);
ierr = PetscRandomSeed(rnd);CHKERRQ(ierr);
}
ierr = PetscOptionsEnd();CHKERRQ(ierr);
ierr = PetscRandomViewFromOptions(rnd,NULL, "-random_view");CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@
PetscRandomSetFromOptions - Configures the random number generator from the options database.
Collective on PetscRandom
Input Parameter:
. rnd - The random number generator context
Options Database:
+ -random_seed <integer> - provide a seed to the random number generater
- -random_no_imaginary_part - makes the imaginary part of the random number zero, this is useful when you want the
same code to produce the same result when run with real numbers or complex numbers for regression testing purposes
Notes: To see all options, run your program with the -help option.
Must be called after PetscRandomCreate() but before the rnd is used.
Level: beginner
.keywords: PetscRandom, set, options, database
.seealso: PetscRandomCreate(), PetscRandomSetType()
@*/
PetscErrorCode PetscRandomSetFromOptions(PetscRandom rnd)
{
PetscErrorCode ierr;
PetscBool set,noimaginary = PETSC_FALSE;
PetscInt seed;
PetscFunctionBegin;
PetscValidHeaderSpecific(rnd,PETSC_RANDOM_CLASSID,1);
ierr = PetscObjectOptionsBegin((PetscObject)rnd);
CHKERRQ(ierr);
/* Handle PetscRandom type options */
ierr = PetscRandomSetTypeFromOptions_Private(PetscOptionsObject,rnd);
CHKERRQ(ierr);
/* Handle specific random generator's options */
if (rnd->ops->setfromoptions) {
ierr = (*rnd->ops->setfromoptions)(PetscOptionsObject,rnd);
CHKERRQ(ierr);
}
ierr = PetscOptionsInt("-random_seed","Seed to use to generate random numbers","PetscRandomSetSeed",0,&seed,&set);
CHKERRQ(ierr);
if (set) {
ierr = PetscRandomSetSeed(rnd,(unsigned long int)seed);
CHKERRQ(ierr);
ierr = PetscRandomSeed(rnd);
CHKERRQ(ierr);
}
ierr = PetscOptionsBool("-random_no_imaginary_part","The imaginary part of the random number will be zero","PetscRandomSetInterval",noimaginary,&noimaginary,&set);
CHKERRQ(ierr);
#if defined(PETSC_HAVE_COMPLEX)
if (set) {
if (noimaginary) {
PetscScalar low,high;
ierr = PetscRandomGetInterval(rnd,&low,&high);
CHKERRQ(ierr);
low = low - PetscImaginaryPart(low);
high = high - PetscImaginaryPart(high);
ierr = PetscRandomSetInterval(rnd,low,high);
CHKERRQ(ierr);
}
}
#endif
ierr = PetscOptionsEnd();
CHKERRQ(ierr);
ierr = PetscRandomViewFromOptions(rnd,NULL, "-random_view");
CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@
BVSetRandomColumn - Set one column of a BV to random numbers.
Logically Collective on BV
Input Parameters:
+ bv - basis vectors
. j - column number to be set
- rctx - the random number context, formed by PetscRandomCreate(), or NULL and
it will create one internally.
Note:
This operation is analogue to VecSetRandom - the difference is that the
generated random vector is the same irrespective of the size of the
communicator (if all processes pass a PetscRandom context initialized
with the same seed).
Level: advanced
.seealso: BVSetRandom(), BVSetActiveColumns()
@*/
PetscErrorCode BVSetRandomColumn(BV bv,PetscInt j,PetscRandom rctx)
{
PetscErrorCode ierr;
PetscRandom rand=NULL;
PetscInt i,low,high;
PetscScalar *px,t;
Vec x;
PetscFunctionBegin;
PetscValidHeaderSpecific(bv,BV_CLASSID,1);
PetscValidLogicalCollectiveInt(bv,j,2);
if (rctx) PetscValidHeaderSpecific(rctx,PETSC_RANDOM_CLASSID,3);
else {
ierr = PetscRandomCreate(PetscObjectComm((PetscObject)bv),&rand);CHKERRQ(ierr);
ierr = PetscRandomSetSeed(rand,0x12345678);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(rand);CHKERRQ(ierr);
rctx = rand;
}
PetscValidType(bv,1);
BVCheckSizes(bv,1);
if (j<0 || j>=bv->m) SETERRQ2(PetscObjectComm((PetscObject)bv),PETSC_ERR_ARG_OUTOFRANGE,"Argument j has wrong value %D, the number of columns is %D",j,bv->m);
ierr = PetscLogEventBegin(BV_SetRandom,bv,rctx,0,0);CHKERRQ(ierr);
ierr = BVGetColumn(bv,j,&x);CHKERRQ(ierr);
ierr = VecGetOwnershipRange(x,&low,&high);CHKERRQ(ierr);
ierr = VecGetArray(x,&px);CHKERRQ(ierr);
for (i=0;i<bv->N;i++) {
ierr = PetscRandomGetValue(rctx,&t);CHKERRQ(ierr);
if (i>=low && i<high) px[i-low] = t;
}
ierr = VecRestoreArray(x,&px);CHKERRQ(ierr);
ierr = BVRestoreColumn(bv,j,&x);CHKERRQ(ierr);
ierr = PetscLogEventEnd(BV_SetRandom,bv,rctx,0,0);CHKERRQ(ierr);
ierr = PetscRandomDestroy(&rand);CHKERRQ(ierr);
ierr = PetscObjectStateIncrease((PetscObject)bv);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@
BVSetRandom - Set the columns of a BV to random numbers.
Logically Collective on BV
Input Parameters:
+ bv - basis vectors
- rctx - the random number context, formed by PetscRandomCreate(), or NULL and
it will create one internally.
Note:
All active columns (except the leading ones) are modified.
Level: advanced
.seealso: BVSetRandomColumn(), BVSetActiveColumns()
@*/
PetscErrorCode BVSetRandom(BV bv,PetscRandom rctx)
{
PetscErrorCode ierr;
PetscRandom rand=NULL;
PetscInt i,low,high,k;
PetscScalar *px,t;
Vec x;
PetscFunctionBegin;
PetscValidHeaderSpecific(bv,BV_CLASSID,1);
if (rctx) PetscValidHeaderSpecific(rctx,PETSC_RANDOM_CLASSID,2);
else {
ierr = PetscRandomCreate(PetscObjectComm((PetscObject)bv),&rand);CHKERRQ(ierr);
ierr = PetscRandomSetSeed(rand,0x12345678);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(rand);CHKERRQ(ierr);
rctx = rand;
}
PetscValidType(bv,1);
BVCheckSizes(bv,1);
ierr = PetscLogEventBegin(BV_SetRandom,bv,rctx,0,0);CHKERRQ(ierr);
for (k=bv->l;k<bv->k;k++) {
ierr = BVGetColumn(bv,k,&x);CHKERRQ(ierr);
ierr = VecGetOwnershipRange(x,&low,&high);CHKERRQ(ierr);
ierr = VecGetArray(x,&px);CHKERRQ(ierr);
for (i=0;i<bv->N;i++) {
ierr = PetscRandomGetValue(rctx,&t);CHKERRQ(ierr);
if (i>=low && i<high) px[i-low] = t;
}
ierr = VecRestoreArray(x,&px);CHKERRQ(ierr);
ierr = BVRestoreColumn(bv,k,&x);CHKERRQ(ierr);
}
ierr = PetscLogEventEnd(BV_SetRandom,bv,rctx,0,0);CHKERRQ(ierr);
ierr = PetscRandomDestroy(&rand);CHKERRQ(ierr);
ierr = PetscObjectStateIncrease((PetscObject)bv);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
int main(int argc, char** args){
PetscErrorCode err;
PetscViewer fd = NULL;
Mat invL1 = NULL, invU1 = NULL, invL2 = NULL, invU2 = NULL, H12 = NULL, H21 = NULL;
Vec order = NULL, r = NULL; //, or = NULL; //dimension: n: n1 + n2
Vec seeds = NULL;
// Vec r1 = NULL, q1 = NULL, t1_1 = NULL, t1_2 = NULL, t1_3 = NULL, t1_4 = NULL, t1_5 = NULL; // dimension: n1
// Vec r2 = NULL, q2 = NULL, q_tilda = NULL, t2_1 = NULL, t2_2 = NULL, t2_3 = NULL; // dimension: n2
PetscRandom rand;
PetscLogDouble tic, toc, total_time, time;
PetscInt n, i;
PetscMPIInt rank, size;
PetscInt seed;
PetscScalar c, val;
PetscInt QN = 100;
// Initialize PETSC and MPI
err = PetscInitialize(&argc, &args, (char*) 0, help); CHKERRQ(err);
err = MPI_Comm_size(PETSC_COMM_WORLD, &size); CHKERRQ(err);
err = MPI_Comm_rank(PETSC_COMM_WORLD, &rank); CHKERRQ(err);
err = PetscPrintf(PETSC_COMM_WORLD, "mpi size: %d\n", size); CHKERRQ(err);
// Read matrices and an ordering vector
err = PetscPrintf(PETSC_COMM_WORLD, "Read inputs (invL1, invU1, invL2, invU2, H12, H21, order)\n"); CHKERRQ(err);
err = loadMat("./data/invL1.dat", &invL1, PETSC_COMM_WORLD, MATMPIAIJ, &fd); CHKERRQ(err);
err = checkMat("invL1", invL1); CHKERRQ(err);
err = loadMat("./data/invU1.dat", &invU1, PETSC_COMM_WORLD, MATMPIAIJ, &fd); CHKERRQ(err);
err = checkMat("invU1", invU1); CHKERRQ(err);
err = loadMat("./data/invL2.dat", &invL2, PETSC_COMM_WORLD, MATMPIAIJ, &fd); CHKERRQ(err);
err = checkMat("invL2", invL2); CHKERRQ(err);
err = loadMat("./data/invU2.dat", &invU2, PETSC_COMM_WORLD, MATMPIAIJ, &fd); CHKERRQ(err);
err = checkMat("invU2", invU2); CHKERRQ(err);
err = loadMat("./data/H12.dat", &H12, PETSC_COMM_WORLD, MATMPIAIJ, &fd); CHKERRQ(err);
err = checkMat("H12", H12); CHKERRQ(err);
err = loadMat("./data/H21.dat", &H21, PETSC_COMM_WORLD, MATMPIAIJ, &fd); CHKERRQ(err);
err = checkMat("H21", H21); CHKERRQ(err);
err = loadVec("./data/order.dat", &order, PETSC_COMM_SELF, &fd); CHKERRQ(err); //all processes must have this vector for ordering the result vector.
err = checkVec("order", order); CHKERRQ(err);
// shift -1 for zero-based index
err = VecShift(order, -1); CHKERRQ(err);
err = VecGetSize(order, &n); CHKERRQ(err);
seed = 5;
c = 0.05;
err = PetscTime(&tic); CHKERRQ(err);
//err = BearQueryMat(seed, c, invL1, invU1, invL2, invU2, H12, H21, order); CHKERRQ(err);
//err = PetscTime(&toc); CHKERRQ(err);
//time = toc - tic;
//err = PetscPrintf(PETSC_COMM_WORLD, "running time: %f sec\n", time); CHKERRQ(err);
///* 100 times querying
err = VecCreateSeq(PETSC_COMM_SELF, QN, &seeds); CHKERRQ(err);
err = VecSetFromOptions(seeds); CHKERRQ(err);
err = PetscRandomCreate(PETSC_COMM_WORLD, &rand); CHKERRQ(err);
err = PetscRandomSetSeed(rand, 100); CHKERRQ(err);
err = PetscRandomSetInterval(rand, (PetscScalar) 0, (PetscScalar) n); CHKERRQ(err);
err = PetscRandomSetFromOptions(rand); CHKERRQ(err);
err = VecSetRandom(seeds, rand); CHKERRQ(err);
err = PetscRandomDestroy(&rand); CHKERRQ(err);
seed = 5; //seed is give by user on one-based index
c = 0.05;
i = 0;
err = VecDuplicate(order, &r); CHKERRQ(err);
for(i = 0; i < QN; i++){
err = VecGetValues(seeds, 1, &i, &val);
seed = (PetscInt) val;
//err = PetscPrintf(PETSC_COMM_SELF, "rank: %d, seed: %d\n", rank, seed);
err = PetscTime(&tic); CHKERRQ(err);
err = BearQuery(seed, c, invL1, invU1, invL2, invU2, H12, H21, order, r); CHKERRQ(err);
err = PetscTime(&toc); CHKERRQ(err);
time = toc - tic;
err = PetscPrintf(PETSC_COMM_WORLD, "running time: %f sec\n", time); CHKERRQ(err);
total_time += time;
}
err = PetscPrintf(PETSC_COMM_WORLD, "average running time: %f sec\n", total_time/QN); CHKERRQ(err);
err = VecDestroy(&r);
/* err = MatGetSize(H12, &n1, &n2); CHKERRQ(err);
n = n1 + n2;
err = PetscPrintf(PETSC_COMM_WORLD, "n1: %d, n2: %d\n", n1, n2); CHKERRQ(err);
err = VecCreateMPI(PETSC_COMM_WORLD, PETSC_DECIDE, n, &r); CHKERRQ(err);
err = VecCreateMPI(PETSC_COMM_WORLD, PETSC_DECIDE, n1, &q1); CHKERRQ(err);
err = VecCreateMPI(PETSC_COMM_WORLD, PETSC_DECIDE, n2, &q2); CHKERRQ(err);
err = VecSet(q1, 0); CHKERRQ(err);
err = VecSet(q2, 0); CHKERRQ(err);
seed = seed - 1; // shift -1 for zero-based index
err = VecGetValues(order, 1, &seed, &val); CHKERRQ(err);
oseed = (PetscInt) val;
err = PetscPrintf(PETSC_COMM_WORLD, "Given seed: %d, Reorered seed: %d (0 ~ n-1)\n", seed, oseed); CHKERRQ(err);
if(oseed < n1){
err = VecSetValues(q1, 1, &oseed, &one, INSERT_VALUES); CHKERRQ(err);
}else{
oseed = oseed - n1;
err = VecSetValues(q2, 1, &oseed, &one, INSERT_VALUES); CHKERRQ(err);
//err = printVecSum(q2);
}
err = VecAssemblyBegin(q1); CHKERRQ(err);
err = VecAssemblyBegin(q2); CHKERRQ(err);
err = VecAssemblyEnd(q1); CHKERRQ(err);
err = VecAssemblyEnd(q2); CHKERRQ(err);
err = VecDuplicate(q1, &r1); CHKERRQ(err);
err = VecDuplicate(q1, &t1_1); CHKERRQ(err);
err = VecDuplicate(q1, &t1_2); CHKERRQ(err);
err = VecDuplicate(q1, &t1_3); CHKERRQ(err);
err = VecDuplicate(q1, &t1_4); CHKERRQ(err);
err = VecDuplicate(q1, &t1_5); CHKERRQ(err);
err = VecDuplicate(q2, &r2); CHKERRQ(err);
err = VecDuplicate(q2, &q_tilda); CHKERRQ(err);
err = VecDuplicate(q2, &t2_1); CHKERRQ(err);
err = VecDuplicate(q2, &t2_2); CHKERRQ(err);
err = VecDuplicate(q2, &t2_3); CHKERRQ(err);
// Start matrix-vec multiplications
err = MatMult(invL1, q1, t1_1); CHKERRQ(err);
err = MatMult(invU1, t1_1, t1_2); CHKERRQ(err);
err = MatMult(H21, t1_2, t2_1); CHKERRQ(err);
err = VecAXPBYPCZ(q_tilda, 1.0, -1.0, 0.0, q2, t2_1); CHKERRQ(err);
err = MatMult(invL2, q_tilda, t2_2); CHKERRQ(err);
err = MatMult(invU2, t2_2, r2); CHKERRQ(err);
err = MatMult(H12, r2, t1_3); CHKERRQ(err);
err = VecAXPBYPCZ(t1_4, 1.0, -1.0, 0.0, q1, t1_3); CHKERRQ(err);
err = MatMult(invL1, t1_4, t1_5); CHKERRQ(err);
err = MatMult(invU1, t1_5, r1); CHKERRQ(err);
//err = printVecSum(r1);
//err = VecView(r2, PETSC_VIEWER_STDOUT_WORLD);
// Concatenate r1 and r2
err = VecMerge(r1, r2, r); CHKERRQ(err);
err = VecScale(r, c); CHKERRQ(err);
//err = VecView(r, PETSC_VIEWER_STDOUT_WORLD);
err = VecDuplicate(r, &or); CHKERRQ(err);
err = VecReorder(r, order, or); CHKERRQ(err);
//err = VecView(or, PETSC_VIEWER_STDOUT_WORLD);*/
// Destory matrices and vectors
err = MatDestroy(&invL1); CHKERRQ(err);
err = MatDestroy(&invU1); CHKERRQ(err);
err = MatDestroy(&invL2); CHKERRQ(err);
err = MatDestroy(&invU2); CHKERRQ(err);
err = MatDestroy(&H12); CHKERRQ(err);
err = MatDestroy(&H21); CHKERRQ(err);
err = VecDestroy(&order); CHKERRQ(err);
err = VecDestroy(&r); CHKERRQ(err);
err = VecDestroy(&seeds); CHKERRQ(err);
//err = VecDestroy(&or); CHKERRQ(err);
/* err = VecDestroy(&r1); CHKERRQ(err);
err = VecDestroy(&q1); CHKERRQ(err);
err = VecDestroy(&t1_1); CHKERRQ(err);
err = VecDestroy(&t1_2); CHKERRQ(err);
err = VecDestroy(&t1_3); CHKERRQ(err);
err = VecDestroy(&t1_4); CHKERRQ(err);
err = VecDestroy(&t1_5); CHKERRQ(err);
err = VecDestroy(&r2); CHKERRQ(err);
err = VecDestroy(&q2); CHKERRQ(err);
err = VecDestroy(&q_tilda); CHKERRQ(err);
err = VecDestroy(&t2_1); CHKERRQ(err);
err = VecDestroy(&t2_2); CHKERRQ(err);
err = VecDestroy(&t2_3); CHKERRQ(err);*/
// Finalize
err = PetscFinalize(); CHKERRQ(err);
return 0;
}
PetscErrorCode ComputeMatrix(DM da,Mat B)
{
PetscErrorCode ierr;
PetscInt i,j,k,mx,my,mz,xm,ym,zm,xs,ys,zs,dof,k1,k2,k3;
PetscScalar *v,*v_neighbor,Hx,Hy,Hz,HxHydHz,HyHzdHx,HxHzdHy,r1,r2;
MatStencil row,col;
PetscRandom rand;
PetscFunctionBegin;
ierr = PetscRandomCreate(PETSC_COMM_WORLD,&rand);CHKERRQ(ierr);
ierr = PetscRandomSetType(rand,PETSCRAND);CHKERRQ(ierr);
ierr = PetscRandomSetSeed(rand,1);CHKERRQ(ierr);
ierr = PetscRandomSetInterval(rand,-.001,.001);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(rand);CHKERRQ(ierr);
ierr = DMDAGetInfo(da,0,&mx,&my,&mz,0,0,0,&dof,0,0,0,0,0);CHKERRQ(ierr);
/* For simplicity, this example only works on mx=my=mz */
if ( mx != my || mx != mz) SETERRQ3(PETSC_COMM_SELF,1,"This example only works with mx %d = my %d = mz %d\n",mx,my,mz);
Hx = 1.0 / (PetscReal)(mx-1); Hy = 1.0 / (PetscReal)(my-1); Hz = 1.0 / (PetscReal)(mz-1);
HxHydHz = Hx*Hy/Hz; HxHzdHy = Hx*Hz/Hy; HyHzdHx = Hy*Hz/Hx;
ierr = PetscMalloc((2*dof*dof+1)*sizeof(PetscScalar),&v);CHKERRQ(ierr);
v_neighbor = v + dof*dof;
ierr = PetscMemzero(v,(2*dof*dof+1)*sizeof(PetscScalar));CHKERRQ(ierr);
k3 = 0;
for (k1=0; k1<dof; k1++){
for (k2=0; k2<dof; k2++){
if (k1 == k2){
v[k3] = 2.0*(HxHydHz + HxHzdHy + HyHzdHx);
v_neighbor[k3] = -HxHydHz;
} else {
ierr = PetscRandomGetValue(rand,&r1);CHKERRQ(ierr);
ierr = PetscRandomGetValue(rand,&r2);CHKERRQ(ierr);
v[k3] = r1;
v_neighbor[k3] = r2;
}
k3++;
}
}
ierr = DMDAGetCorners(da,&xs,&ys,&zs,&xm,&ym,&zm);CHKERRQ(ierr);
for (k=zs; k<zs+zm; k++){
for (j=ys; j<ys+ym; j++){
for (i=xs; i<xs+xm; i++){
row.i = i; row.j = j; row.k = k;
if (i==0 || j==0 || k==0 || i==mx-1 || j==my-1 || k==mz-1){ /* boudary points */
ierr = MatSetValuesBlockedStencil(B,1,&row,1,&row,v,INSERT_VALUES);CHKERRQ(ierr);
} else { /* interior points */
/* center */
col.i = i; col.j = j; col.k = k;
ierr = MatSetValuesBlockedStencil(B,1,&row,1,&col,v,INSERT_VALUES);CHKERRQ(ierr);
/* x neighbors */
col.i = i-1; col.j = j; col.k = k;
ierr = MatSetValuesBlockedStencil(B,1,&row,1,&col,v_neighbor,INSERT_VALUES);CHKERRQ(ierr);
col.i = i+1; col.j = j; col.k = k;
ierr = MatSetValuesBlockedStencil(B,1,&row,1,&col,v_neighbor,INSERT_VALUES);CHKERRQ(ierr);
/* y neighbors */
col.i = i; col.j = j-1; col.k = k;
ierr = MatSetValuesBlockedStencil(B,1,&row,1,&col,v_neighbor,INSERT_VALUES);CHKERRQ(ierr);
col.i = i; col.j = j+1; col.k = k;
ierr = MatSetValuesBlockedStencil(B,1,&row,1,&col,v_neighbor,INSERT_VALUES);CHKERRQ(ierr);
/* z neighbors */
col.i = i; col.j = j; col.k = k-1;
ierr = MatSetValuesBlockedStencil(B,1,&row,1,&col,v_neighbor,INSERT_VALUES);CHKERRQ(ierr);
col.i = i; col.j = j; col.k = k+1;
ierr = MatSetValuesBlockedStencil(B,1,&row,1,&col,v_neighbor,INSERT_VALUES);CHKERRQ(ierr);
}
}
}
}
ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = PetscFree(v);CHKERRQ(ierr);
ierr = PetscRandomDestroy(&rand);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@C
EPSCreate - Creates the default EPS context.
Collective on MPI_Comm
Input Parameter:
. comm - MPI communicator
Output Parameter:
. eps - location to put the EPS context
Note:
The default EPS type is EPSKRYLOVSCHUR
Level: beginner
.seealso: EPSSetUp(), EPSSolve(), EPSDestroy(), EPS
@*/
PetscErrorCode EPSCreate(MPI_Comm comm,EPS *outeps)
{
PetscErrorCode ierr;
EPS eps;
PetscFunctionBegin;
PetscValidPointer(outeps,2);
*outeps = 0;
ierr = EPSInitializePackage();
CHKERRQ(ierr);
ierr = SlepcHeaderCreate(eps,_p_EPS,struct _EPSOps,EPS_CLASSID,"EPS","Eigenvalue Problem Solver","EPS",comm,EPSDestroy,EPSView);
CHKERRQ(ierr);
eps->max_it = 0;
eps->nev = 1;
eps->ncv = 0;
eps->mpd = 0;
eps->nini = 0;
eps->nds = 0;
eps->target = 0.0;
eps->tol = PETSC_DEFAULT;
eps->conv = EPS_CONV_EIG;
eps->which = (EPSWhich)0;
eps->inta = 0.0;
eps->intb = 0.0;
eps->problem_type = (EPSProblemType)0;
eps->extraction = EPS_RITZ;
eps->balance = EPS_BALANCE_NONE;
eps->balance_its = 5;
eps->balance_cutoff = 1e-8;
eps->trueres = PETSC_FALSE;
eps->trackall = PETSC_FALSE;
eps->converged = EPSConvergedEigRelative;
eps->convergeddestroy= NULL;
eps->arbitrary = NULL;
eps->convergedctx = NULL;
eps->arbitraryctx = NULL;
eps->numbermonitors = 0;
eps->st = NULL;
eps->ds = NULL;
eps->V = NULL;
eps->rg = NULL;
eps->rand = NULL;
eps->D = NULL;
eps->IS = NULL;
eps->defl = NULL;
eps->eigr = NULL;
eps->eigi = NULL;
eps->errest = NULL;
eps->rr = NULL;
eps->ri = NULL;
eps->perm = NULL;
eps->nwork = 0;
eps->work = NULL;
eps->data = NULL;
eps->state = EPS_STATE_INITIAL;
eps->nconv = 0;
eps->its = 0;
eps->nloc = 0;
eps->nrma = 0.0;
eps->nrmb = 0.0;
eps->isgeneralized = PETSC_FALSE;
eps->ispositive = PETSC_FALSE;
eps->ishermitian = PETSC_FALSE;
eps->reason = EPS_CONVERGED_ITERATING;
ierr = PetscNewLog(eps,&eps->sc);
CHKERRQ(ierr);
ierr = PetscRandomCreate(comm,&eps->rand);
CHKERRQ(ierr);
ierr = PetscRandomSetSeed(eps->rand,0x12345678);
CHKERRQ(ierr);
ierr = PetscLogObjectParent((PetscObject)eps,(PetscObject)eps->rand);
CHKERRQ(ierr);
*outeps = eps;
PetscFunctionReturn(0);
}
PetscErrorCode random_network(PetscInt nvertex,PetscInt *pnbranch,Node **pnode,Branch **pbranch,PetscInt **pedgelist,PetscInt seed)
{
PetscErrorCode ierr;
PetscInt i, j, nedges = 0;
PetscInt *edgelist;
PetscInt nbat, ncurr, fr, to;
PetscReal *x, *y, value, xmax = 10.0; /* generate points in square */
PetscReal maxdist = 0.0, dist, alpha, beta, prob;
PetscRandom rnd;
Branch *branch;
Node *node;
Edge *head = NULL, *nnew= NULL, *aux= NULL;
PetscFunctionBeginUser;
ierr = PetscRandomCreate(PETSC_COMM_SELF,&rnd);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(rnd);CHKERRQ(ierr);
ierr = PetscRandomSetSeed(rnd, seed);CHKERRQ(ierr);
ierr = PetscRandomSeed(rnd);CHKERRQ(ierr);
/* These parameters might be modified for experimentation */
nbat = (PetscInt)(0.1*nvertex);
ncurr = (PetscInt)(0.1*nvertex);
alpha = 0.6;
beta = 0.2;
ierr = PetscMalloc2(nvertex,&x,nvertex,&y);CHKERRQ(ierr);
ierr = PetscRandomSetInterval(rnd,0.0,xmax);CHKERRQ(ierr);
for (i=0; i<nvertex; i++) {
ierr = PetscRandomGetValueReal(rnd,&x[i]);CHKERRQ(ierr);
ierr = PetscRandomGetValueReal(rnd,&y[i]);CHKERRQ(ierr);
}
/* find maximum distance */
for (i=0; i<nvertex; i++) {
for (j=0; j<nvertex; j++) {
dist = findDistance(x[i],x[j],y[i],y[j]);
if (dist >= maxdist) maxdist = dist;
}
}
ierr = PetscRandomSetInterval(rnd,0.0,1.0);CHKERRQ(ierr);
for (i=0; i<nvertex; i++) {
for (j=0; j<nvertex; j++) {
if (j != i) {
dist = findDistance(x[i],x[j],y[i],y[j]);
prob = beta*PetscExpScalar(-dist/(maxdist*alpha));
ierr = PetscRandomGetValueReal(rnd,&value);CHKERRQ(ierr);
if (value <= prob) {
ierr = PetscMalloc1(1,&nnew);CHKERRQ(ierr);
if (head == NULL) {
head = nnew;
head->next = NULL;
head->n = nedges;
head->i = i;
head->j = j;
} else {
aux = head;
head = nnew;
head->n = nedges;
head->next = aux;
head->i = i;
head->j = j;
}
nedges += 1;
}
}
}
}
ierr = PetscMalloc1(2*nedges,&edgelist);CHKERRQ(ierr);
for (aux = head; aux; aux = aux->next) {
edgelist[(aux->n)*2] = aux->i;
edgelist[(aux->n)*2 + 1] = aux->j;
}
aux = head;
while (aux != NULL) {
nnew = aux;
aux = aux->next;
ierr = PetscFree(nnew);CHKERRQ(ierr);
}
ierr = PetscCalloc2(nvertex,&node,nedges,&branch);CHKERRQ(ierr);
for (i = 0; i < nvertex; i++) {
node[i].id = i;
node[i].inj = 0;
node[i].gr = PETSC_FALSE;
}
for (i = 0; i < nedges; i++) {
branch[i].id = i;
branch[i].r = 1.0;
branch[i].bat = 0;
}
/* Chose random node as ground voltage */
ierr = PetscRandomSetInterval(rnd,0.0,nvertex);CHKERRQ(ierr);
ierr = PetscRandomGetValueReal(rnd,&value);CHKERRQ(ierr);
node[(int)value].gr = PETSC_TRUE;
/* Create random current and battery injectionsa */
for (i=0; i<ncurr; i++) {
ierr = PetscRandomSetInterval(rnd,0.0,nvertex);CHKERRQ(ierr);
ierr = PetscRandomGetValueReal(rnd,&value);CHKERRQ(ierr);
fr = edgelist[(int)value*2];
to = edgelist[(int)value*2 + 1];
node[fr].inj += 1.0;
node[to].inj -= 1.0;
}
for (i=0; i<nbat; i++) {
ierr = PetscRandomSetInterval(rnd,0.0,nedges);CHKERRQ(ierr);
ierr = PetscRandomGetValueReal(rnd,&value);CHKERRQ(ierr);
branch[(int)value].bat += 1.0;
}
ierr = PetscFree2(x,y);CHKERRQ(ierr);
ierr = PetscRandomDestroy(&rnd);CHKERRQ(ierr);
/* assign pointers */
*pnbranch = nedges;
*pedgelist = edgelist;
*pbranch = branch;
*pnode = node;
PetscFunctionReturn(ierr);
}
void PETSC_STDCALL petscrandomsetseed_(PetscRandom r,unsigned long *seed, int *__ierr ){
*__ierr = PetscRandomSetSeed(
(PetscRandom)PetscToPointer((r) ),*seed);
}
