xref: /petsc/src/tao/constrained/tutorials/ex1.c (revision dfd57a172ac9fa6c7b5fe6de6ab5df85cefc2996)
1aad13602SShrirang Abhyankar /* Program usage: mpiexec -n 2 ./ex1 [-help] [all TAO options] */
2aad13602SShrirang Abhyankar 
3aad13602SShrirang Abhyankar /* ----------------------------------------------------------------------
4628da978Sresundermann min f(x) = (x0 - 2)^2 + (x1 - 2)^2 - 2*(x0 + x1)
5aad13602SShrirang Abhyankar s.t.  x0^2 + x1 - 2 = 0
6aad13602SShrirang Abhyankar       0  <= x0^2 - x1 <= 1
7aad13602SShrirang Abhyankar       -1 <= x0, x1 <= 2
8628da978Sresundermann -->
9628da978Sresundermann       g(x)  = 0
10628da978Sresundermann       h(x) >= 0
11628da978Sresundermann       -1 <= x0, x1 <= 2
12628da978Sresundermann where
13628da978Sresundermann       g(x) = x0^2 + x1 - 2
14628da978Sresundermann       h(x) = [x0^2 - x1
15628da978Sresundermann               1 -(x0^2 - x1)]
16aad13602SShrirang Abhyankar ---------------------------------------------------------------------- */
17aad13602SShrirang Abhyankar 
18aad13602SShrirang Abhyankar #include <petsctao.h>
19aad13602SShrirang Abhyankar 
20aad13602SShrirang Abhyankar static  char help[]= "Solves constrained optimiztion problem using pdipm.\n\
21aad13602SShrirang Abhyankar Input parameters include:\n\
22aad13602SShrirang Abhyankar   -tao_type pdipm    : sets Tao solver\n\
238e58fa1dSresundermann   -no_eq             : removes the equaility constraints from the problem\n\
24aad13602SShrirang Abhyankar   -snes_fd           : snes with finite difference Jacobian (needed for pdipm)\n\
25628da978Sresundermann   -snes_compare_explicit : compare user Jacobian with finite difference Jacobian \n\
26aad13602SShrirang Abhyankar   -tao_cmonitor      : convergence monitor with constraint norm \n\
27aad13602SShrirang Abhyankar   -tao_view_solution : view exact solution at each itteration\n\
28628da978Sresundermann   Note: external package MUMPS is required to run pdipm. This is designed for a maximum of 2 processors, the code will error if size > 2.\n";
29aad13602SShrirang Abhyankar 
30aad13602SShrirang Abhyankar /*
31628da978Sresundermann    User-defined application context - contains data needed by the application
32aad13602SShrirang Abhyankar */
33aad13602SShrirang Abhyankar typedef struct {
34aad13602SShrirang Abhyankar   PetscInt   n;  /* Global length of x */
35aad13602SShrirang Abhyankar   PetscInt   ne; /* Global number of equality constraints */
36aad13602SShrirang Abhyankar   PetscInt   ni; /* Global number of inequality constraints */
378e58fa1dSresundermann   PetscBool  noeqflag;
38aad13602SShrirang Abhyankar   Vec        x,xl,xu;
39aad13602SShrirang Abhyankar   Vec        ce,ci,bl,bu,Xseq;
40aad13602SShrirang Abhyankar   Mat        Ae,Ai,H;
41aad13602SShrirang Abhyankar   VecScatter scat;
42aad13602SShrirang Abhyankar } AppCtx;
43aad13602SShrirang Abhyankar 
44aad13602SShrirang Abhyankar /* -------- User-defined Routines --------- */
45aad13602SShrirang Abhyankar PetscErrorCode InitializeProblem(AppCtx *);
46aad13602SShrirang Abhyankar PetscErrorCode DestroyProblem(AppCtx *);
47aad13602SShrirang Abhyankar PetscErrorCode FormFunctionGradient(Tao,Vec,PetscReal *,Vec,void *);
48aad13602SShrirang Abhyankar PetscErrorCode FormHessian(Tao,Vec,Mat,Mat, void*);
49aad13602SShrirang Abhyankar PetscErrorCode FormInequalityConstraints(Tao,Vec,Vec,void*);
50aad13602SShrirang Abhyankar PetscErrorCode FormEqualityConstraints(Tao,Vec,Vec,void*);
51aad13602SShrirang Abhyankar PetscErrorCode FormInequalityJacobian(Tao,Vec,Mat,Mat, void*);
52aad13602SShrirang Abhyankar PetscErrorCode FormEqualityJacobian(Tao,Vec,Mat,Mat, void*);
53aad13602SShrirang Abhyankar 
54aad13602SShrirang Abhyankar PetscErrorCode main(int argc,char **argv)
55aad13602SShrirang Abhyankar {
56aad13602SShrirang Abhyankar   PetscErrorCode ierr;  /* used to check for functions returning nonzeros */
57aad13602SShrirang Abhyankar   Tao            tao;
58aad13602SShrirang Abhyankar   KSP            ksp;
59aad13602SShrirang Abhyankar   PC             pc;
60aad13602SShrirang Abhyankar   AppCtx         user;  /* application context */
61aad13602SShrirang Abhyankar   Vec            x;
62aad13602SShrirang Abhyankar   PetscMPIInt    rank;
63661095bbSAlp Dener   TaoType        type;
64661095bbSAlp Dener   PetscBool      pdipm;
65aad13602SShrirang Abhyankar 
66aad13602SShrirang Abhyankar   ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr;
67ffc4695bSBarry Smith   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr);
68d8185827SBarry Smith   if (rank>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_WRONG_MPI_SIZE,"More than 2 processors detected. Example written to use max of 2 processors.\n");
69aad13602SShrirang Abhyankar 
70aad13602SShrirang Abhyankar   ierr = PetscPrintf(PETSC_COMM_WORLD,"---- Constrained Problem -----\n");CHKERRQ(ierr);
71aad13602SShrirang Abhyankar   ierr = InitializeProblem(&user);CHKERRQ(ierr); /* sets up problem, function below */
72aad13602SShrirang Abhyankar   ierr = TaoCreate(PETSC_COMM_WORLD,&tao);CHKERRQ(ierr);
73aad13602SShrirang Abhyankar   ierr = TaoSetType(tao,TAOPDIPM);CHKERRQ(ierr);
74aad13602SShrirang Abhyankar   ierr = TaoSetInitialVector(tao,user.x);CHKERRQ(ierr); /* gets solution vector from problem */
75aad13602SShrirang Abhyankar   ierr = TaoSetVariableBounds(tao,user.xl,user.xu);CHKERRQ(ierr); /* sets lower upper bounds from given solution */
76aad13602SShrirang Abhyankar   ierr = TaoSetObjectiveAndGradientRoutine(tao,FormFunctionGradient,(void*)&user);CHKERRQ(ierr);
77aad13602SShrirang Abhyankar 
788e58fa1dSresundermann   if (!user.noeqflag) {
79aad13602SShrirang Abhyankar     ierr = TaoSetEqualityConstraintsRoutine(tao,user.ce,FormEqualityConstraints,(void*)&user);CHKERRQ(ierr);
808e58fa1dSresundermann   }
81aad13602SShrirang Abhyankar   ierr = TaoSetInequalityConstraintsRoutine(tao,user.ci,FormInequalityConstraints,(void*)&user);CHKERRQ(ierr);
828e58fa1dSresundermann   if (!user.noeqflag) {
83aad13602SShrirang Abhyankar     ierr = TaoSetJacobianEqualityRoutine(tao,user.Ae,user.Ae,FormEqualityJacobian,(void*)&user);CHKERRQ(ierr); /* equality jacobian */
848e58fa1dSresundermann   }
85aad13602SShrirang Abhyankar   ierr = TaoSetJacobianInequalityRoutine(tao,user.Ai,user.Ai,FormInequalityJacobian,(void*)&user);CHKERRQ(ierr); /* inequality jacobian */
86aad13602SShrirang Abhyankar   ierr = TaoSetTolerances(tao,1.e-6,1.e-6,1.e-6);CHKERRQ(ierr);
87aad13602SShrirang Abhyankar   ierr = TaoSetConstraintTolerances(tao,1.e-6,1.e-6);CHKERRQ(ierr);
88aad13602SShrirang Abhyankar 
89aad13602SShrirang Abhyankar   ierr = TaoGetKSP(tao,&ksp);CHKERRQ(ierr);
90aad13602SShrirang Abhyankar   ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
9112d688e0SRylee Sundermann   ierr = PCSetType(pc,PCCHOLESKY);CHKERRQ(ierr);
92aad13602SShrirang Abhyankar   /*
93aad13602SShrirang Abhyankar       This algorithm produces matrices with zeros along the diagonal therefore we use
9412d688e0SRylee Sundermann     MUMPS which provides solver for indefinite matrices
95aad13602SShrirang Abhyankar   */
9612d688e0SRylee Sundermann   ierr = PCFactorSetMatSolverType(pc,MATSOLVERMUMPS);CHKERRQ(ierr);  /* requires mumps to solve pdipm */
97aad13602SShrirang Abhyankar   ierr = KSPSetType(ksp,KSPPREONLY);CHKERRQ(ierr);
98aad13602SShrirang Abhyankar   ierr = KSPSetFromOptions(ksp);CHKERRQ(ierr);
99aad13602SShrirang Abhyankar 
100aad13602SShrirang Abhyankar   ierr = TaoSetFromOptions(tao);CHKERRQ(ierr);
101661095bbSAlp Dener   ierr = TaoGetType(tao,&type);CHKERRQ(ierr);
102661095bbSAlp Dener   ierr = PetscObjectTypeCompare((PetscObject)tao,TAOPDIPM,&pdipm);CHKERRQ(ierr);
103661095bbSAlp Dener   if (pdipm) {
104661095bbSAlp Dener     ierr = TaoSetHessianRoutine(tao,user.H,user.H,FormHessian,(void*)&user);CHKERRQ(ierr);
105661095bbSAlp Dener   }
106aad13602SShrirang Abhyankar 
107aad13602SShrirang Abhyankar   ierr = TaoSolve(tao);CHKERRQ(ierr);
108aad13602SShrirang Abhyankar   ierr = TaoGetSolutionVector(tao,&x);CHKERRQ(ierr);
109aad13602SShrirang Abhyankar   ierr = VecView(x,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
110aad13602SShrirang Abhyankar 
111aad13602SShrirang Abhyankar   /* Free objects */
112aad13602SShrirang Abhyankar   ierr = DestroyProblem(&user);CHKERRQ(ierr);
113aad13602SShrirang Abhyankar   ierr = TaoDestroy(&tao);CHKERRQ(ierr);
114aad13602SShrirang Abhyankar   ierr = PetscFinalize();
115aad13602SShrirang Abhyankar   return ierr;
116aad13602SShrirang Abhyankar }
117aad13602SShrirang Abhyankar 
118aad13602SShrirang Abhyankar PetscErrorCode InitializeProblem(AppCtx *user)
119aad13602SShrirang Abhyankar {
120aad13602SShrirang Abhyankar   PetscErrorCode ierr;
121aad13602SShrirang Abhyankar   PetscMPIInt    size;
122aad13602SShrirang Abhyankar   PetscMPIInt    rank;
123aad13602SShrirang Abhyankar   PetscInt       nloc,neloc,niloc;
124aad13602SShrirang Abhyankar 
125aad13602SShrirang Abhyankar   PetscFunctionBegin;
126ffc4695bSBarry Smith   ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr);
127ffc4695bSBarry Smith   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr);
1288e58fa1dSresundermann   user->noeqflag = PETSC_FALSE;
1298e58fa1dSresundermann   ierr = PetscOptionsGetBool(NULL,NULL,"-no_eq",&user->noeqflag,NULL);CHKERRQ(ierr);
130628da978Sresundermann 
1318e58fa1dSresundermann   if (!user->noeqflag) {
1328e58fa1dSresundermann     ierr = PetscPrintf(PETSC_COMM_WORLD,"Solution should be f(1,1)=-2\n");CHKERRQ(ierr);
1338e58fa1dSresundermann   }
134aad13602SShrirang Abhyankar 
1352d4ee042Sprj-   /* create vector x and set initial values */
136aad13602SShrirang Abhyankar   user->n = 2; /* global length */
137aad13602SShrirang Abhyankar   nloc = (rank==0)?user->n:0;
138661095bbSAlp Dener   ierr = VecCreate(PETSC_COMM_WORLD,&user->x);CHKERRQ(ierr);
139661095bbSAlp Dener   ierr = VecSetSizes(user->x,nloc,user->n);CHKERRQ(ierr);
140661095bbSAlp Dener   ierr = VecSetFromOptions(user->x);CHKERRQ(ierr);
141aad13602SShrirang Abhyankar   ierr = VecSet(user->x,0);CHKERRQ(ierr);
142aad13602SShrirang Abhyankar 
143aad13602SShrirang Abhyankar   /* create and set lower and upper bound vectors */
144aad13602SShrirang Abhyankar   ierr = VecDuplicate(user->x,&user->xl);CHKERRQ(ierr);
145aad13602SShrirang Abhyankar   ierr = VecDuplicate(user->x,&user->xu);CHKERRQ(ierr);
146aad13602SShrirang Abhyankar   ierr = VecSet(user->xl,-1.0);CHKERRQ(ierr);
147aad13602SShrirang Abhyankar   ierr = VecSet(user->xu,2.0);CHKERRQ(ierr);
148aad13602SShrirang Abhyankar 
149aad13602SShrirang Abhyankar   /* create scater to zero */
150aad13602SShrirang Abhyankar   ierr = VecScatterCreateToZero(user->x,&user->scat,&user->Xseq);CHKERRQ(ierr);
151aad13602SShrirang Abhyankar 
152aad13602SShrirang Abhyankar     user->ne = 1;
153628da978Sresundermann     user->ni = 2;
154aad13602SShrirang Abhyankar     neloc = (rank==0)?user->ne:0;
155628da978Sresundermann     niloc = (rank==0)?user->ni:0;
156aad13602SShrirang Abhyankar 
1578e58fa1dSresundermann   if (!user->noeqflag) {
158661095bbSAlp Dener     ierr = VecCreate(PETSC_COMM_WORLD,&user->ce);CHKERRQ(ierr); /* a 1x1 vec for equality constraints */
159661095bbSAlp Dener     ierr = VecSetSizes(user->ce,neloc,user->ne);CHKERRQ(ierr);
160661095bbSAlp Dener     ierr = VecSetFromOptions(user->ce);CHKERRQ(ierr);
161661095bbSAlp Dener     ierr = VecSetUp(user->ce);CHKERRQ(ierr);
1628e58fa1dSresundermann   }
163628da978Sresundermann 
164661095bbSAlp Dener   ierr = VecCreate(PETSC_COMM_WORLD,&user->ci);CHKERRQ(ierr); /* a 2x1 vec for inequality constraints */
165661095bbSAlp Dener   ierr = VecSetSizes(user->ci,niloc,user->ni);CHKERRQ(ierr);
166661095bbSAlp Dener   ierr = VecSetFromOptions(user->ci);CHKERRQ(ierr);
167661095bbSAlp Dener   ierr = VecSetUp(user->ci);CHKERRQ(ierr);
168aad13602SShrirang Abhyankar 
169aad13602SShrirang Abhyankar   /* nexn & nixn matricies for equaly and inequalty constriants */
1708e58fa1dSresundermann   if (!user->noeqflag) {
171aad13602SShrirang Abhyankar     ierr = MatCreate(PETSC_COMM_WORLD,&user->Ae);CHKERRQ(ierr);
1728e58fa1dSresundermann     ierr = MatSetSizes(user->Ae,neloc,nloc,user->ne,user->n);CHKERRQ(ierr);
1738e58fa1dSresundermann     ierr = MatSetFromOptions(user->Ae);CHKERRQ(ierr);
174661095bbSAlp Dener     ierr = MatSetUp(user->Ae);CHKERRQ(ierr);
1758e58fa1dSresundermann   }
1768e58fa1dSresundermann 
177aad13602SShrirang Abhyankar   ierr = MatCreate(PETSC_COMM_WORLD,&user->Ai);CHKERRQ(ierr);
178aad13602SShrirang Abhyankar   ierr = MatSetSizes(user->Ai,niloc,nloc,user->ni,user->n);CHKERRQ(ierr);
179aad13602SShrirang Abhyankar   ierr = MatSetFromOptions(user->Ai);CHKERRQ(ierr);
180661095bbSAlp Dener   ierr = MatSetUp(user->Ai);CHKERRQ(ierr);
181628da978Sresundermann 
182628da978Sresundermann   ierr = MatCreate(PETSC_COMM_WORLD,&user->H);CHKERRQ(ierr);
183628da978Sresundermann   ierr = MatSetSizes(user->H,nloc,nloc,user->n,user->n);CHKERRQ(ierr);
184628da978Sresundermann   ierr = MatSetFromOptions(user->H);CHKERRQ(ierr);
185661095bbSAlp Dener   ierr = MatSetUp(user->H);CHKERRQ(ierr);
186aad13602SShrirang Abhyankar   PetscFunctionReturn(0);
187aad13602SShrirang Abhyankar }
188aad13602SShrirang Abhyankar 
189aad13602SShrirang Abhyankar PetscErrorCode DestroyProblem(AppCtx *user)
190aad13602SShrirang Abhyankar {
191aad13602SShrirang Abhyankar   PetscErrorCode ierr;
192aad13602SShrirang Abhyankar 
193aad13602SShrirang Abhyankar   PetscFunctionBegin;
1948e58fa1dSresundermann   if (!user->noeqflag) {
195aad13602SShrirang Abhyankar    ierr = MatDestroy(&user->Ae);CHKERRQ(ierr);
1968e58fa1dSresundermann   }
197aad13602SShrirang Abhyankar   ierr = MatDestroy(&user->Ai);CHKERRQ(ierr);
198aad13602SShrirang Abhyankar   ierr = MatDestroy(&user->H);CHKERRQ(ierr);
199aad13602SShrirang Abhyankar 
200aad13602SShrirang Abhyankar   ierr = VecDestroy(&user->x);CHKERRQ(ierr);
2018e58fa1dSresundermann   if (!user->noeqflag) {
202aad13602SShrirang Abhyankar     ierr = VecDestroy(&user->ce);CHKERRQ(ierr);
2038e58fa1dSresundermann   }
204aad13602SShrirang Abhyankar   ierr = VecDestroy(&user->ci);CHKERRQ(ierr);
205aad13602SShrirang Abhyankar   ierr = VecDestroy(&user->xl);CHKERRQ(ierr);
206aad13602SShrirang Abhyankar   ierr = VecDestroy(&user->xu);CHKERRQ(ierr);
207aad13602SShrirang Abhyankar   ierr = VecDestroy(&user->Xseq);CHKERRQ(ierr);
208aad13602SShrirang Abhyankar   ierr = VecScatterDestroy(&user->scat);CHKERRQ(ierr);
209aad13602SShrirang Abhyankar   PetscFunctionReturn(0);
210aad13602SShrirang Abhyankar }
211aad13602SShrirang Abhyankar 
212628da978Sresundermann /* Evaluate
213628da978Sresundermann    f(x) = (x0 - 2)^2 + (x1 - 2)^2 - 2*(x0 + x1)
214628da978Sresundermann    G = grad f = [2*(x0 - 2) - 2;
215628da978Sresundermann                  2*(x1 - 2) - 2]
216aad13602SShrirang Abhyankar */
217aad13602SShrirang Abhyankar PetscErrorCode FormFunctionGradient(Tao tao, Vec X, PetscReal *f, Vec G, void *ctx)
218aad13602SShrirang Abhyankar {
219aad13602SShrirang Abhyankar   PetscScalar       g;
220aad13602SShrirang Abhyankar   const PetscScalar *x;
221aad13602SShrirang Abhyankar   MPI_Comm          comm;
222aad13602SShrirang Abhyankar   PetscMPIInt       rank;
223aad13602SShrirang Abhyankar   PetscErrorCode    ierr;
224aad13602SShrirang Abhyankar   PetscReal         fin;
225aad13602SShrirang Abhyankar   AppCtx            *user=(AppCtx*)ctx;
226aad13602SShrirang Abhyankar   Vec               Xseq=user->Xseq;
227aad13602SShrirang Abhyankar   VecScatter        scat=user->scat;
228aad13602SShrirang Abhyankar 
229aad13602SShrirang Abhyankar   PetscFunctionBegin;
230aad13602SShrirang Abhyankar   ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr);
231ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr);
232aad13602SShrirang Abhyankar 
233aad13602SShrirang Abhyankar   ierr = VecScatterBegin(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
234aad13602SShrirang Abhyankar   ierr = VecScatterEnd(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
235aad13602SShrirang Abhyankar 
236aad13602SShrirang Abhyankar   fin = 0.0;
237aad13602SShrirang Abhyankar   if (!rank) {
238aad13602SShrirang Abhyankar     ierr = VecGetArrayRead(Xseq,&x);CHKERRQ(ierr);
239aad13602SShrirang Abhyankar     fin = (x[0]-2.0)*(x[0]-2.0) + (x[1]-2.0)*(x[1]-2.0) - 2.0*(x[0]+x[1]);
240aad13602SShrirang Abhyankar     g = 2.0*(x[0]-2.0) - 2.0;
241aad13602SShrirang Abhyankar     ierr = VecSetValue(G,0,g,INSERT_VALUES);CHKERRQ(ierr);
242aad13602SShrirang Abhyankar     g = 2.0*(x[1]-2.0) - 2.0;
243aad13602SShrirang Abhyankar     ierr = VecSetValue(G,1,g,INSERT_VALUES);CHKERRQ(ierr);
244aad13602SShrirang Abhyankar     ierr = VecRestoreArrayRead(Xseq,&x);CHKERRQ(ierr);
245aad13602SShrirang Abhyankar   }
246ffc4695bSBarry Smith   ierr = MPI_Allreduce(&fin,f,1,MPIU_REAL,MPIU_SUM,comm);CHKERRMPI(ierr);
247aad13602SShrirang Abhyankar   ierr = VecAssemblyBegin(G);CHKERRQ(ierr);
248aad13602SShrirang Abhyankar   ierr = VecAssemblyEnd(G);CHKERRQ(ierr);
249aad13602SShrirang Abhyankar   PetscFunctionReturn(0);
250aad13602SShrirang Abhyankar }
251aad13602SShrirang Abhyankar 
252628da978Sresundermann /* Evaluate
253628da978Sresundermann    H = fxx + grad (grad g^T*DI) - grad (grad h^T*DE)]
254628da978Sresundermann      = [ 2*(1+de[0]-di[0]+di[1]), 0;
255628da978Sresundermann                    0,             2]
256628da978Sresundermann */
257aad13602SShrirang Abhyankar PetscErrorCode FormHessian(Tao tao, Vec x,Mat H, Mat Hpre, void *ctx)
258aad13602SShrirang Abhyankar {
259aad13602SShrirang Abhyankar   AppCtx            *user=(AppCtx*)ctx;
260aad13602SShrirang Abhyankar   Vec               DE,DI;
261aad13602SShrirang Abhyankar   const PetscScalar *de,*di;
262aad13602SShrirang Abhyankar   PetscInt          zero=0,one=1;
263aad13602SShrirang Abhyankar   PetscScalar       two=2.0;
264aad13602SShrirang Abhyankar   PetscScalar       val=0.0;
265aad13602SShrirang Abhyankar   Vec               Deseq,Diseq=user->Xseq;
266aad13602SShrirang Abhyankar   VecScatter        Descat,Discat=user->scat;
267aad13602SShrirang Abhyankar   PetscMPIInt       rank;
268aad13602SShrirang Abhyankar   MPI_Comm          comm;
269aad13602SShrirang Abhyankar   PetscErrorCode    ierr;
270aad13602SShrirang Abhyankar 
271aad13602SShrirang Abhyankar   PetscFunctionBegin;
272aad13602SShrirang Abhyankar   ierr = TaoGetDualVariables(tao,&DE,&DI);CHKERRQ(ierr);
273aad13602SShrirang Abhyankar 
274aad13602SShrirang Abhyankar   ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr);
275ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr);
276aad13602SShrirang Abhyankar 
2778e58fa1dSresundermann   if (!user->noeqflag) {
278aad13602SShrirang Abhyankar    ierr = VecScatterCreateToZero(DE,&Descat,&Deseq);CHKERRQ(ierr);
279aad13602SShrirang Abhyankar    ierr = VecScatterBegin(Descat,DE,Deseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
280aad13602SShrirang Abhyankar    ierr = VecScatterEnd(Descat,DE,Deseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2818e58fa1dSresundermann   }
2828e58fa1dSresundermann   ierr = VecScatterBegin(Discat,DI,Diseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
283aad13602SShrirang Abhyankar   ierr = VecScatterEnd(Discat,DI,Diseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
284aad13602SShrirang Abhyankar 
285aad13602SShrirang Abhyankar   if (!rank) {
2868e58fa1dSresundermann     if (!user->noeqflag) {
287aad13602SShrirang Abhyankar       ierr = VecGetArrayRead(Deseq,&de);CHKERRQ(ierr);  /* places equality constraint dual into array */
2888e58fa1dSresundermann     }
289aad13602SShrirang Abhyankar     ierr = VecGetArrayRead(Diseq,&di);CHKERRQ(ierr);  /* places inequality constraint dual into array */
290628da978Sresundermann 
2918e58fa1dSresundermann     if (!user->noeqflag) {
292628da978Sresundermann       val = 2.0 * (1 + de[0] - di[0] + di[1]);
293aad13602SShrirang Abhyankar       ierr = VecRestoreArrayRead(Deseq,&de);CHKERRQ(ierr);
294628da978Sresundermann       ierr = VecRestoreArrayRead(Diseq,&di);CHKERRQ(ierr);
2958e58fa1dSresundermann     } else {
296628da978Sresundermann       val = 2.0 * (1 - di[0] + di[1]);
2978e58fa1dSresundermann     }
298aad13602SShrirang Abhyankar     ierr = VecRestoreArrayRead(Diseq,&di);CHKERRQ(ierr);
299aad13602SShrirang Abhyankar     ierr = MatSetValues(H,1,&zero,1,&zero,&val,INSERT_VALUES);CHKERRQ(ierr);
300aad13602SShrirang Abhyankar     ierr = MatSetValues(H,1,&one,1,&one,&two,INSERT_VALUES);CHKERRQ(ierr);
301aad13602SShrirang Abhyankar   }
302aad13602SShrirang Abhyankar   ierr = MatAssemblyBegin(H,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
303aad13602SShrirang Abhyankar   ierr = MatAssemblyEnd(H,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
3048e58fa1dSresundermann   if (!user->noeqflag) {
305aad13602SShrirang Abhyankar     ierr = VecScatterDestroy(&Descat);CHKERRQ(ierr);
306aad13602SShrirang Abhyankar     ierr = VecDestroy(&Deseq);CHKERRQ(ierr);
3078e58fa1dSresundermann   }
308aad13602SShrirang Abhyankar   PetscFunctionReturn(0);
309aad13602SShrirang Abhyankar }
310aad13602SShrirang Abhyankar 
311628da978Sresundermann /* Evaluate
312628da978Sresundermann    h = [ x0^2 - x1;
313628da978Sresundermann          1 -(x0^2 - x1)]
314628da978Sresundermann */
315aad13602SShrirang Abhyankar PetscErrorCode FormInequalityConstraints(Tao tao,Vec X,Vec CI,void *ctx)
316aad13602SShrirang Abhyankar {
317aad13602SShrirang Abhyankar   const PetscScalar *x;
318aad13602SShrirang Abhyankar   PetscScalar       ci;
319aad13602SShrirang Abhyankar   PetscErrorCode    ierr;
320aad13602SShrirang Abhyankar   MPI_Comm          comm;
321aad13602SShrirang Abhyankar   PetscMPIInt       rank;
322aad13602SShrirang Abhyankar   AppCtx            *user=(AppCtx*)ctx;
323aad13602SShrirang Abhyankar   Vec               Xseq=user->Xseq;
324aad13602SShrirang Abhyankar   VecScatter        scat=user->scat;
325aad13602SShrirang Abhyankar 
326aad13602SShrirang Abhyankar   PetscFunctionBegin;
327aad13602SShrirang Abhyankar   ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr);
328ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr);
329aad13602SShrirang Abhyankar 
330aad13602SShrirang Abhyankar   ierr = VecScatterBegin(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
331aad13602SShrirang Abhyankar   ierr = VecScatterEnd(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
332aad13602SShrirang Abhyankar 
333aad13602SShrirang Abhyankar   if (!rank) {
334aad13602SShrirang Abhyankar     ierr = VecGetArrayRead(Xseq,&x);CHKERRQ(ierr);
335aad13602SShrirang Abhyankar     ci = x[0]*x[0] - x[1];
336aad13602SShrirang Abhyankar     ierr = VecSetValue(CI,0,ci,INSERT_VALUES);CHKERRQ(ierr);
337aad13602SShrirang Abhyankar     ci = -x[0]*x[0] + x[1] + 1.0;
338aad13602SShrirang Abhyankar     ierr = VecSetValue(CI,1,ci,INSERT_VALUES);CHKERRQ(ierr);
339aad13602SShrirang Abhyankar     ierr = VecRestoreArrayRead(Xseq,&x);CHKERRQ(ierr);
340aad13602SShrirang Abhyankar   }
341aad13602SShrirang Abhyankar   ierr = VecAssemblyBegin(CI);CHKERRQ(ierr);
342aad13602SShrirang Abhyankar   ierr = VecAssemblyEnd(CI);CHKERRQ(ierr);
343aad13602SShrirang Abhyankar   PetscFunctionReturn(0);
344aad13602SShrirang Abhyankar }
345aad13602SShrirang Abhyankar 
346628da978Sresundermann /* Evaluate
347628da978Sresundermann    g = [ x0^2 + x1 - 2]
348628da978Sresundermann */
349aad13602SShrirang Abhyankar PetscErrorCode FormEqualityConstraints(Tao tao,Vec X,Vec CE,void *ctx)
350aad13602SShrirang Abhyankar {
351aad13602SShrirang Abhyankar   const PetscScalar *x;
352aad13602SShrirang Abhyankar   PetscScalar       ce;
353aad13602SShrirang Abhyankar   PetscErrorCode    ierr;
354aad13602SShrirang Abhyankar   MPI_Comm          comm;
355aad13602SShrirang Abhyankar   PetscMPIInt       rank;
356aad13602SShrirang Abhyankar   AppCtx            *user=(AppCtx*)ctx;
357aad13602SShrirang Abhyankar   Vec               Xseq=user->Xseq;
358aad13602SShrirang Abhyankar   VecScatter        scat=user->scat;
359aad13602SShrirang Abhyankar 
360aad13602SShrirang Abhyankar   PetscFunctionBegin;
361aad13602SShrirang Abhyankar   ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr);
362ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr);
363aad13602SShrirang Abhyankar 
364aad13602SShrirang Abhyankar   ierr = VecScatterBegin(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
365aad13602SShrirang Abhyankar   ierr = VecScatterEnd(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
366aad13602SShrirang Abhyankar 
367aad13602SShrirang Abhyankar   if (!rank) {
368aad13602SShrirang Abhyankar     ierr = VecGetArrayRead(Xseq,&x);CHKERRQ(ierr);
369aad13602SShrirang Abhyankar     ce = x[0]*x[0] + x[1] - 2.0;
370aad13602SShrirang Abhyankar     ierr = VecSetValue(CE,0,ce,INSERT_VALUES);CHKERRQ(ierr);
371aad13602SShrirang Abhyankar     ierr = VecRestoreArrayRead(Xseq,&x);CHKERRQ(ierr);
372aad13602SShrirang Abhyankar   }
373aad13602SShrirang Abhyankar   ierr = VecAssemblyBegin(CE);CHKERRQ(ierr);
374aad13602SShrirang Abhyankar   ierr = VecAssemblyEnd(CE);CHKERRQ(ierr);
375aad13602SShrirang Abhyankar   PetscFunctionReturn(0);
376aad13602SShrirang Abhyankar }
377aad13602SShrirang Abhyankar 
378628da978Sresundermann /*
379628da978Sresundermann   grad h = [  2*x0, -1;
380628da978Sresundermann              -2*x0,  1]
381628da978Sresundermann */
382aad13602SShrirang Abhyankar PetscErrorCode FormInequalityJacobian(Tao tao, Vec X, Mat JI, Mat JIpre,  void *ctx)
383aad13602SShrirang Abhyankar {
384aad13602SShrirang Abhyankar   AppCtx            *user=(AppCtx*)ctx;
385aad13602SShrirang Abhyankar   PetscInt          cols[2],min,max,i;
386aad13602SShrirang Abhyankar   PetscScalar       vals[2];
387aad13602SShrirang Abhyankar   const PetscScalar *x;
388aad13602SShrirang Abhyankar   PetscErrorCode    ierr;
389aad13602SShrirang Abhyankar   Vec               Xseq=user->Xseq;
390aad13602SShrirang Abhyankar   VecScatter        scat=user->scat;
391aad13602SShrirang Abhyankar   MPI_Comm          comm;
392aad13602SShrirang Abhyankar   PetscMPIInt       rank;
393aad13602SShrirang Abhyankar 
394aad13602SShrirang Abhyankar   PetscFunctionBegin;
395aad13602SShrirang Abhyankar   ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr);
396ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr);
397aad13602SShrirang Abhyankar   ierr = VecScatterBegin(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
398aad13602SShrirang Abhyankar   ierr = VecScatterEnd(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
399aad13602SShrirang Abhyankar 
400aad13602SShrirang Abhyankar   ierr = VecGetArrayRead(Xseq,&x);CHKERRQ(ierr);
401aad13602SShrirang Abhyankar   ierr = MatGetOwnershipRange(JI,&min,&max);CHKERRQ(ierr);
402aad13602SShrirang Abhyankar 
403aad13602SShrirang Abhyankar   cols[0] = 0; cols[1] = 1;
404aad13602SShrirang Abhyankar   for (i=min;i<max;i++) {
405aad13602SShrirang Abhyankar     if (i==0) {
406628da978Sresundermann       vals[0] = 2*x[0]; vals[1] = -1.0;
407aad13602SShrirang Abhyankar       ierr = MatSetValues(JI,1,&i,2,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
408aad13602SShrirang Abhyankar     }
409aad13602SShrirang Abhyankar     if (i==1) {
410628da978Sresundermann       vals[0] = -2*x[0]; vals[1] = 1.0;
411aad13602SShrirang Abhyankar       ierr = MatSetValues(JI,1,&i,2,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
412aad13602SShrirang Abhyankar     }
413aad13602SShrirang Abhyankar   }
414aad13602SShrirang Abhyankar   ierr = VecRestoreArrayRead(Xseq,&x);CHKERRQ(ierr);
415aad13602SShrirang Abhyankar 
416aad13602SShrirang Abhyankar   ierr = MatAssemblyBegin(JI,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
417aad13602SShrirang Abhyankar   ierr = MatAssemblyEnd(JI,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
418aad13602SShrirang Abhyankar   PetscFunctionReturn(0);
419aad13602SShrirang Abhyankar }
420aad13602SShrirang Abhyankar 
421628da978Sresundermann /*
422628da978Sresundermann   grad g = [2*x0
423628da978Sresundermann              1.0 ]
424628da978Sresundermann */
425aad13602SShrirang Abhyankar PetscErrorCode FormEqualityJacobian(Tao tao,Vec X,Mat JE,Mat JEpre,void *ctx)
426aad13602SShrirang Abhyankar {
427aad13602SShrirang Abhyankar   PetscInt          rows[2];
428aad13602SShrirang Abhyankar   PetscScalar       vals[2];
429aad13602SShrirang Abhyankar   const PetscScalar *x;
430aad13602SShrirang Abhyankar   PetscMPIInt       rank;
431aad13602SShrirang Abhyankar   MPI_Comm          comm;
432aad13602SShrirang Abhyankar   PetscErrorCode    ierr;
433aad13602SShrirang Abhyankar 
434aad13602SShrirang Abhyankar   PetscFunctionBegin;
435aad13602SShrirang Abhyankar   ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr);
436ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr);
437aad13602SShrirang Abhyankar 
438aad13602SShrirang Abhyankar   if (!rank) {
439aad13602SShrirang Abhyankar     ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr);
440aad13602SShrirang Abhyankar     rows[0] = 0;       rows[1] = 1;
441aad13602SShrirang Abhyankar     vals[0] = 2*x[0];  vals[1] = 1.0;
442aad13602SShrirang Abhyankar     ierr = MatSetValues(JE,1,rows,2,rows,vals,INSERT_VALUES);CHKERRQ(ierr);
443aad13602SShrirang Abhyankar     ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr);
444aad13602SShrirang Abhyankar   }
445aad13602SShrirang Abhyankar   ierr = MatAssemblyBegin(JE,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
446aad13602SShrirang Abhyankar   ierr = MatAssemblyEnd(JE,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
447aad13602SShrirang Abhyankar   PetscFunctionReturn(0);
448aad13602SShrirang Abhyankar }
449aad13602SShrirang Abhyankar 
450aad13602SShrirang Abhyankar /*TEST
451aad13602SShrirang Abhyankar 
452aad13602SShrirang Abhyankar    build:
453*dfd57a17SPierre Jolivet       requires: !complex !defined(PETSC_USE_CXX) mumps
454aad13602SShrirang Abhyankar 
455aad13602SShrirang Abhyankar    test:
45609ee8bb0SRylee Sundermann       args: -tao_converged_reason -tao_pdipm_kkt_shift_pd
457aad13602SShrirang Abhyankar 
458aad13602SShrirang Abhyankar    test:
459aad13602SShrirang Abhyankar       suffix: 2
460aad13602SShrirang Abhyankar       nsize: 2
46109ee8bb0SRylee Sundermann       args: -tao_converged_reason -tao_pdipm_kkt_shift_pd
462aad13602SShrirang Abhyankar 
4638e58fa1dSresundermann    test:
4648e58fa1dSresundermann       suffix: 3
4658e58fa1dSresundermann       args: -tao_converged_reason -no_eq
4668e58fa1dSresundermann 
4678e58fa1dSresundermann    test:
4688e58fa1dSresundermann       suffix: 4
4698e58fa1dSresundermann       nsize: 2
4708e58fa1dSresundermann       args: -tao_converged_reason -no_eq
4718e58fa1dSresundermann 
472661095bbSAlp Dener    test:
473661095bbSAlp Dener       suffix: 5
474661095bbSAlp Dener       args: -tao_cmonitor -tao_type almm
475661095bbSAlp Dener 
476661095bbSAlp Dener    test:
477661095bbSAlp Dener       suffix: 6
478661095bbSAlp Dener       args: -tao_cmonitor -tao_type almm -tao_almm_type phr
479661095bbSAlp Dener 
480661095bbSAlp Dener    test:
481661095bbSAlp Dener       suffix: 7
482661095bbSAlp Dener       nsize: 2
483661095bbSAlp Dener       args: -tao_cmonitor -tao_type almm
484661095bbSAlp Dener 
485661095bbSAlp Dener    test:
486661095bbSAlp Dener       suffix: 8
487661095bbSAlp Dener       nsize: 2
488661095bbSAlp Dener       requires: cuda
489661095bbSAlp Dener       args: -tao_cmonitor -tao_type almm -vec_type cuda -mat_type aijcusparse
490661095bbSAlp Dener 
491661095bbSAlp Dener    test:
492661095bbSAlp Dener       suffix: 9
493661095bbSAlp Dener       nsize: 2
494661095bbSAlp Dener       args: -tao_cmonitor -tao_type almm -no_eq
495661095bbSAlp Dener 
496661095bbSAlp Dener    test:
497661095bbSAlp Dener       suffix: 10
498661095bbSAlp Dener       nsize: 2
499661095bbSAlp Dener       args: -tao_cmonitor -tao_type almm -tao_almm_type phr -no_eq
500661095bbSAlp Dener 
501aad13602SShrirang Abhyankar TEST*/
502