static PetscErrorCode IPMInitializeBounds(Tao tao)
{
TAO_IPM *ipmP = (TAO_IPM*)tao->data;
Vec xtmp;
PetscInt xstart,xend;
PetscInt ucstart,ucend; /* user ci */
PetscInt ucestart,uceend; /* user ce */
PetscInt sstart,send;
PetscInt bigsize;
PetscInt i,counter,nloc;
PetscInt *cind,*xind,*ucind,*uceind,*stepind;
VecType vtype;
const PetscInt *xli,*xui;
PetscInt xl_offset,xu_offset;
IS bigxl,bigxu,isuc,isc,isx,sis,is1;
PetscErrorCode ierr;
MPI_Comm comm;
PetscFunctionBegin;
cind=xind=ucind=uceind=stepind=0;
ipmP->mi=0;
ipmP->nxlb=0;
ipmP->nxub=0;
ipmP->nb=0;
ipmP->nslack=0;
ierr = VecDuplicate(tao->solution,&xtmp);CHKERRQ(ierr);
if (!tao->XL && !tao->XU && tao->ops->computebounds) {
ierr = TaoComputeVariableBounds(tao);CHKERRQ(ierr);
}
if (tao->XL) {
ierr = VecSet(xtmp,PETSC_NINFINITY);CHKERRQ(ierr);
ierr = VecWhichGreaterThan(tao->XL,xtmp,&ipmP->isxl);CHKERRQ(ierr);
ierr = ISGetSize(ipmP->isxl,&ipmP->nxlb);CHKERRQ(ierr);
} else {
ipmP->nxlb=0;
}
if (tao->XU) {
ierr = VecSet(xtmp,PETSC_INFINITY);CHKERRQ(ierr);
ierr = VecWhichLessThan(tao->XU,xtmp,&ipmP->isxu);CHKERRQ(ierr);
ierr = ISGetSize(ipmP->isxu,&ipmP->nxub);CHKERRQ(ierr);
} else {
ipmP->nxub=0;
}
ierr = VecDestroy(&xtmp);CHKERRQ(ierr);
if (tao->constraints_inequality) {
ierr = VecGetSize(tao->constraints_inequality,&ipmP->mi);CHKERRQ(ierr);
} else {
ipmP->mi = 0;
}
ipmP->nb = ipmP->nxlb + ipmP->nxub + ipmP->mi;
comm = ((PetscObject)(tao->solution))->comm;
bigsize = ipmP->n+2*ipmP->nb+ipmP->me;
ierr = PetscMalloc1(bigsize,&stepind);CHKERRQ(ierr);
ierr = PetscMalloc1(ipmP->n,&xind);CHKERRQ(ierr);
ierr = PetscMalloc1(ipmP->me,&uceind);CHKERRQ(ierr);
ierr = VecGetOwnershipRange(tao->solution,&xstart,&xend);CHKERRQ(ierr);
if (ipmP->nb > 0) {
ierr = VecCreate(comm,&ipmP->s);CHKERRQ(ierr);
ierr = VecSetSizes(ipmP->s,PETSC_DECIDE,ipmP->nb);CHKERRQ(ierr);
ierr = VecSetFromOptions(ipmP->s);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->ds);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->rhs_s);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->complementarity);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->ci);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->lamdai);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->dlamdai);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->rhs_lamdai);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->save_lamdai);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->save_s);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->rpi);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->Zero_nb);CHKERRQ(ierr);
ierr = VecSet(ipmP->Zero_nb,0.0);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->One_nb);CHKERRQ(ierr);
ierr = VecSet(ipmP->One_nb,1.0);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->s,&ipmP->Inf_nb);CHKERRQ(ierr);
ierr = VecSet(ipmP->Inf_nb,PETSC_INFINITY);CHKERRQ(ierr);
ierr = PetscMalloc1(ipmP->nb,&cind);CHKERRQ(ierr);
ierr = PetscMalloc1(ipmP->mi,&ucind);CHKERRQ(ierr);
ierr = VecGetOwnershipRange(ipmP->s,&sstart,&send);CHKERRQ(ierr);
if (ipmP->mi > 0) {
ierr = VecGetOwnershipRange(tao->constraints_inequality,&ucstart,&ucend);CHKERRQ(ierr);
counter=0;
for (i=ucstart;i<ucend;i++) {
cind[counter++] = i;
}
ierr = ISCreateGeneral(comm,counter,cind,PETSC_COPY_VALUES,&isuc);CHKERRQ(ierr);
ierr = ISCreateGeneral(comm,counter,cind,PETSC_COPY_VALUES,&isc);CHKERRQ(ierr);
ierr = VecScatterCreate(tao->constraints_inequality,isuc,ipmP->ci,isc,&ipmP->ci_scat);CHKERRQ(ierr);
ierr = ISDestroy(&isuc);CHKERRQ(ierr);
ierr = ISDestroy(&isc);CHKERRQ(ierr);
}
/* need to know how may xbound indices are on each process */
/* TODO better way */
if (ipmP->nxlb) {
ierr = ISAllGather(ipmP->isxl,&bigxl);CHKERRQ(ierr);
ierr = ISGetIndices(bigxl,&xli);CHKERRQ(ierr);
/* find offsets for this processor */
xl_offset = ipmP->mi;
for (i=0;i<ipmP->nxlb;i++) {
if (xli[i] < xstart) {
xl_offset++;
} else break;
}
ierr = ISRestoreIndices(bigxl,&xli);CHKERRQ(ierr);
ierr = ISGetIndices(ipmP->isxl,&xli);CHKERRQ(ierr);
ierr = ISGetLocalSize(ipmP->isxl,&nloc);CHKERRQ(ierr);
for (i=0;i<nloc;i++) {
xind[i] = xli[i];
cind[i] = xl_offset+i;
}
ierr = ISCreateGeneral(comm,nloc,xind,PETSC_COPY_VALUES,&isx);CHKERRQ(ierr);
ierr = ISCreateGeneral(comm,nloc,cind,PETSC_COPY_VALUES,&isc);CHKERRQ(ierr);
ierr = VecScatterCreate(tao->XL,isx,ipmP->ci,isc,&ipmP->xl_scat);CHKERRQ(ierr);
ierr = ISDestroy(&isx);CHKERRQ(ierr);
ierr = ISDestroy(&isc);CHKERRQ(ierr);
ierr = ISDestroy(&bigxl);CHKERRQ(ierr);
}
if (ipmP->nxub) {
ierr = ISAllGather(ipmP->isxu,&bigxu);CHKERRQ(ierr);
ierr = ISGetIndices(bigxu,&xui);CHKERRQ(ierr);
/* find offsets for this processor */
xu_offset = ipmP->mi + ipmP->nxlb;
for (i=0;i<ipmP->nxub;i++) {
if (xui[i] < xstart) {
xu_offset++;
} else break;
}
ierr = ISRestoreIndices(bigxu,&xui);CHKERRQ(ierr);
ierr = ISGetIndices(ipmP->isxu,&xui);CHKERRQ(ierr);
ierr = ISGetLocalSize(ipmP->isxu,&nloc);CHKERRQ(ierr);
for (i=0;i<nloc;i++) {
xind[i] = xui[i];
cind[i] = xu_offset+i;
}
ierr = ISCreateGeneral(comm,nloc,xind,PETSC_COPY_VALUES,&isx);CHKERRQ(ierr);
ierr = ISCreateGeneral(comm,nloc,cind,PETSC_COPY_VALUES,&isc);CHKERRQ(ierr);
ierr = VecScatterCreate(tao->XU,isx,ipmP->ci,isc,&ipmP->xu_scat);CHKERRQ(ierr);
ierr = ISDestroy(&isx);CHKERRQ(ierr);
ierr = ISDestroy(&isc);CHKERRQ(ierr);
ierr = ISDestroy(&bigxu);CHKERRQ(ierr);
}
}
ierr = VecCreate(comm,&ipmP->bigrhs);CHKERRQ(ierr);
ierr = VecGetType(tao->solution,&vtype);CHKERRQ(ierr);
ierr = VecSetType(ipmP->bigrhs,vtype);CHKERRQ(ierr);
ierr = VecSetSizes(ipmP->bigrhs,PETSC_DECIDE,bigsize);CHKERRQ(ierr);
ierr = VecSetFromOptions(ipmP->bigrhs);CHKERRQ(ierr);
ierr = VecDuplicate(ipmP->bigrhs,&ipmP->bigstep);CHKERRQ(ierr);
/* create scatters for step->x and x->rhs */
for (i=xstart;i<xend;i++) {
stepind[i-xstart] = i;
xind[i-xstart] = i;
}
ierr = ISCreateGeneral(comm,xend-xstart,stepind,PETSC_COPY_VALUES,&sis);CHKERRQ(ierr);
ierr = ISCreateGeneral(comm,xend-xstart,xind,PETSC_COPY_VALUES,&is1);CHKERRQ(ierr);
ierr = VecScatterCreate(ipmP->bigstep,sis,tao->solution,is1,&ipmP->step1);CHKERRQ(ierr);
ierr = VecScatterCreate(tao->solution,is1,ipmP->bigrhs,sis,&ipmP->rhs1);CHKERRQ(ierr);
ierr = ISDestroy(&sis);CHKERRQ(ierr);
ierr = ISDestroy(&is1);CHKERRQ(ierr);
if (ipmP->nb > 0) {
for (i=sstart;i<send;i++) {
stepind[i-sstart] = i+ipmP->n;
cind[i-sstart] = i;
}
ierr = ISCreateGeneral(comm,send-sstart,stepind,PETSC_COPY_VALUES,&sis);CHKERRQ(ierr);
ierr = ISCreateGeneral(comm,send-sstart,cind,PETSC_COPY_VALUES,&is1);CHKERRQ(ierr);
ierr = VecScatterCreate(ipmP->bigstep,sis,ipmP->s,is1,&ipmP->step2);CHKERRQ(ierr);
ierr = ISDestroy(&sis);CHKERRQ(ierr);
for (i=sstart;i<send;i++) {
stepind[i-sstart] = i+ipmP->n+ipmP->me;
cind[i-sstart] = i;
}
ierr = ISCreateGeneral(comm,send-sstart,stepind,PETSC_COPY_VALUES,&sis);CHKERRQ(ierr);
ierr = VecScatterCreate(ipmP->s,is1,ipmP->bigrhs,sis,&ipmP->rhs3);CHKERRQ(ierr);
ierr = ISDestroy(&sis);CHKERRQ(ierr);
ierr = ISDestroy(&is1);CHKERRQ(ierr);
}
if (ipmP->me > 0) {
ierr = VecGetOwnershipRange(tao->constraints_equality,&ucestart,&uceend);CHKERRQ(ierr);
for (i=ucestart;i<uceend;i++) {
stepind[i-ucestart] = i + ipmP->n+ipmP->nb;
uceind[i-ucestart] = i;
}
ierr = ISCreateGeneral(comm,uceend-ucestart,stepind,PETSC_COPY_VALUES,&sis);CHKERRQ(ierr);
ierr = ISCreateGeneral(comm,uceend-ucestart,uceind,PETSC_COPY_VALUES,&is1);CHKERRQ(ierr);
ierr = VecScatterCreate(ipmP->bigstep,sis,tao->constraints_equality,is1,&ipmP->step3);CHKERRQ(ierr);
ierr = ISDestroy(&sis);CHKERRQ(ierr);
for (i=ucestart;i<uceend;i++) {
stepind[i-ucestart] = i + ipmP->n;
}
ierr = ISCreateGeneral(comm,uceend-ucestart,stepind,PETSC_COPY_VALUES,&sis);CHKERRQ(ierr);
ierr = VecScatterCreate(tao->constraints_equality,is1,ipmP->bigrhs,sis,&ipmP->rhs2);CHKERRQ(ierr);
ierr = ISDestroy(&sis);CHKERRQ(ierr);
ierr = ISDestroy(&is1);CHKERRQ(ierr);
}
if (ipmP->nb > 0) {
for (i=sstart;i<send;i++) {
stepind[i-sstart] = i + ipmP->n + ipmP->nb + ipmP->me;
cind[i-sstart] = i;
}
ierr = ISCreateGeneral(comm,send-sstart,cind,PETSC_COPY_VALUES,&is1);CHKERRQ(ierr);
ierr = ISCreateGeneral(comm,send-sstart,stepind,PETSC_COPY_VALUES,&sis);CHKERRQ(ierr);
ierr = VecScatterCreate(ipmP->bigstep,sis,ipmP->s,is1,&ipmP->step4);CHKERRQ(ierr);
ierr = VecScatterCreate(ipmP->s,is1,ipmP->bigrhs,sis,&ipmP->rhs4);CHKERRQ(ierr);
ierr = ISDestroy(&sis);CHKERRQ(ierr);
ierr = ISDestroy(&is1);CHKERRQ(ierr);
}
ierr = PetscFree(stepind);CHKERRQ(ierr);
ierr = PetscFree(cind);CHKERRQ(ierr);
ierr = PetscFree(ucind);CHKERRQ(ierr);
ierr = PetscFree(uceind);CHKERRQ(ierr);
ierr = PetscFree(xind);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PETSC_EXTERN void PETSC_STDCALL vecwhichlessthan_(Vec Vec1,Vec Vec2,IS * S, int *__ierr ){
*__ierr = VecWhichLessThan(
(Vec)PetscToPointer((Vec1) ),
(Vec)PetscToPointer((Vec2) ),S);
}
