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);
}
PetscErrorCode PetscRandomSetType(PetscRandom rnd, PetscRandomType type)
{
PetscErrorCode (*r)(PetscRandom);
PetscBool match;
PetscErrorCode ierr;
PetscFunctionBegin;
PetscValidHeaderSpecific(rnd, PETSC_RANDOM_CLASSID,1);
ierr = PetscObjectTypeCompare((PetscObject)rnd, type, &match);CHKERRQ(ierr);
if (match) PetscFunctionReturn(0);
ierr = PetscFunctionListFind(PetscRandomList,type,&r);CHKERRQ(ierr);
if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE, "Unknown random type: %s", type);
if (rnd->ops->destroy) {
ierr = (*rnd->ops->destroy)(rnd);CHKERRQ(ierr);
rnd->ops->destroy = NULL;
}
ierr = (*r)(rnd);CHKERRQ(ierr);
ierr = PetscRandomSeed(rnd);CHKERRQ(ierr);
ierr = PetscObjectChangeTypeName((PetscObject)rnd, type);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
PetscInt i,n = 1000,*values;
PetscRandom rnd;
PetscScalar value,avg = 0.0;
PetscErrorCode ierr;
PetscMPIInt rank;
PetscInt view_rank=-1;
#if defined(PETSC_USE_LOG)
PetscLogEvent event;
#endif
ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr;
ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-view_randomvalues",&view_rank,NULL);CHKERRQ(ierr);
ierr = PetscRandomCreate(PETSC_COMM_WORLD,&rnd);CHKERRQ(ierr);
/* force imaginary part of random number to always be zero; thus obtain reproducible results with real and complex numbers */
ierr = PetscRandomSetInterval(rnd,0.0,1.0);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(rnd);CHKERRQ(ierr);
ierr = PetscMalloc1(n,&values);CHKERRQ(ierr);
for (i=0; i<n; i++) {
ierr = PetscRandomGetValue(rnd,&value);CHKERRQ(ierr);
avg += value;
if (view_rank == (PetscInt)rank) {
ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] value[%D] = %6.4e\n",rank,i,(double)PetscRealPart(value));CHKERRQ(ierr);
}
values[i] = (PetscInt)(n*PetscRealPart(value) + 2.0);
}
avg = avg/((PetscReal)n);
if (view_rank == (PetscInt)rank) {
ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] Average value %6.4e\n",rank,(double)PetscRealPart(avg));CHKERRQ(ierr);
}
ierr = PetscSortInt(n,values);CHKERRQ(ierr);
ierr = PetscLogEventRegister("Sort",0,&event);CHKERRQ(ierr);
ierr = PetscLogEventBegin(event,0,0,0,0);CHKERRQ(ierr);
ierr = PetscRandomSeed(rnd);CHKERRQ(ierr);
for (i=0; i<n; i++) {
ierr = PetscRandomGetValue(rnd,&value);CHKERRQ(ierr);
values[i] = (PetscInt)(n*PetscRealPart(value) + 2.0);
/* printf("value[%d] = %g\n",i,value); */
}
ierr = PetscSortInt(n,values);CHKERRQ(ierr);
ierr = PetscLogEventEnd(event,0,0,0,0);CHKERRQ(ierr);
for (i=1; i<n; i++) {
if (values[i] < values[i-1]) SETERRQ(PETSC_COMM_SELF,1,"Values not sorted");
}
ierr = PetscFree(values);CHKERRQ(ierr);
ierr = PetscRandomDestroy(&rnd);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
/*@
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);
}
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 petscrandomseed_(PetscRandom r, int *__ierr ){
*__ierr = PetscRandomSeed(
(PetscRandom)PetscToPointer((r) ));
}
