PetscSolverFeti.cc 45.9 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13
//
// Software License for AMDiS
//
// Copyright (c) 2010 Dresden University of Technology 
// All rights reserved.
// Authors: Simon Vey, Thomas Witkowski et al.
//
// This file is part of AMDiS
//
// See also license.opensource.txt in the distribution.


#include "parallel/PetscSolverFeti.h"
14
#include "parallel/PetscSolverFetiStructs.h"
15 16
#include "parallel/StdMpi.h"
#include "parallel/MpiHelper.h"
17
#include "parallel/SubDomainSolver.h"
18
#include "io/VtkWriter.h"
19 20 21 22 23

namespace AMDiS {

  using namespace std;

24 25 26 27 28 29 30
  // y = mat * x
  int petscMultMatSchurPrimal(Mat mat, Vec x, Vec y)
  {
    // S_PiPi = K_PiPi - K_PiB inv(K_BB) K_BPi

    void *ctx;
    MatShellGetContext(mat, &ctx);
31
    SchurPrimalData* data = static_cast<SchurPrimalData*>(ctx);
32 33

    MatMult(*(data->mat_b_primal), x, data->tmp_vec_b);
Thomas Witkowski's avatar
Thomas Witkowski committed
34
    data->fetiSolver->solveLocalProblem(data->tmp_vec_b, data->tmp_vec_b);
35 36 37 38 39 40 41 42 43 44 45
    MatMult(*(data->mat_primal_b), data->tmp_vec_b, data->tmp_vec_primal);
    MatMult(*(data->mat_primal_primal), x, y);
    VecAXPBY(y, -1.0, 1.0, data->tmp_vec_primal);

    return 0;
  }


  // y = mat * x
  int petscMultMatFeti(Mat mat, Vec x, Vec y)
  {
46 47
    //    F = L inv(K_BB) trans(L) + L inv(K_BB) K_BPi inv(S_PiPi) K_PiB inv(K_BB) trans(L)
    // => F = L [I + inv(K_BB) K_BPi inv(S_PiPi) K_PiB] inv(K_BB) trans(L)
48 49 50

    void *ctx;
    MatShellGetContext(mat, &ctx);
51
    FetiData* data = static_cast<FetiData*>(ctx);
52

Thomas Witkowski's avatar
Thomas Witkowski committed
53 54 55
    MatMultTranspose(*(data->mat_lagrange), x, data->tmp_vec_b);
    data->fetiSolver->solveLocalProblem(data->tmp_vec_b, data->tmp_vec_b);
    MatMult(*(data->mat_lagrange), data->tmp_vec_b, data->tmp_vec_lagrange);
Thomas Witkowski's avatar
Thomas Witkowski committed
56

Thomas Witkowski's avatar
Thomas Witkowski committed
57
    MatMult(*(data->mat_primal_b), data->tmp_vec_b, data->tmp_vec_primal);
58
    KSPSolve(*(data->ksp_schur_primal), data->tmp_vec_primal, data->tmp_vec_primal);
Thomas Witkowski's avatar
Thomas Witkowski committed
59 60 61
    MatMult(*(data->mat_b_primal), data->tmp_vec_primal, data->tmp_vec_b);
    data->fetiSolver->solveLocalProblem(data->tmp_vec_b, data->tmp_vec_b);
    MatMult(*(data->mat_lagrange), data->tmp_vec_b, y);
62

Thomas Witkowski's avatar
Thomas Witkowski committed
63
    VecAXPBY(y, 1.0, 1.0, data->tmp_vec_lagrange);
64 65 66 67 68

    return 0;
  }


69
  // y = PC * x
70
  PetscErrorCode petscApplyFetiDirichletPrecon(PC pc, Vec x, Vec y)
71
  {
72
    // Get data for the preconditioner
73 74
    void *ctx;
    PCShellGetContext(pc, &ctx);
75
    FetiDirichletPreconData* data = static_cast<FetiDirichletPreconData*>(ctx);
76

77
    // Multiply with scaled Lagrange constraint matrix.
78 79 80
    MatMultTranspose(*(data->mat_lagrange_scaled), x, data->tmp_vec_b);


81
    // === Restriction of the B nodes to the boundary nodes. ===
82

83 84 85 86
    int nLocalB;
    int nLocalDuals;
    VecGetLocalSize(data->tmp_vec_b, &nLocalB);
    VecGetLocalSize(data->tmp_vec_duals0, &nLocalDuals);
87

88 89 90 91
    PetscScalar *local_b, *local_duals;
    VecGetArray(data->tmp_vec_b, &local_b);
    VecGetArray(data->tmp_vec_duals0, &local_duals);

92 93 94
    for (map<int, int>::iterator it = data->localToDualMap.begin();
	 it != data->localToDualMap.end(); ++it)
      local_duals[it->second] = local_b[it->first];
95 96

    VecRestoreArray(data->tmp_vec_b, &local_b);
97
    VecRestoreArray(data->tmp_vec_duals0, &local_duals);
98 99


100
    // === K_DD - K_DI inv(K_II) K_ID ===
101

102
    MatMult(*(data->mat_duals_duals), data->tmp_vec_duals0, data->tmp_vec_duals1);
103

104
    MatMult(*(data->mat_interior_duals), data->tmp_vec_duals0, data->tmp_vec_interior);
105
    KSPSolve(*(data->ksp_interior), data->tmp_vec_interior, data->tmp_vec_interior);
106 107 108 109 110 111 112 113 114 115
    MatMult(*(data->mat_duals_interior), data->tmp_vec_interior, data->tmp_vec_duals0);

    VecAXPBY(data->tmp_vec_duals0, 1.0, -1.0, data->tmp_vec_duals1);


    // === Prolongation from local dual nodes to B nodes.

    VecGetArray(data->tmp_vec_b, &local_b);
    VecGetArray(data->tmp_vec_duals0, &local_duals);

116 117 118
    for (map<int, int>::iterator it = data->localToDualMap.begin();
	 it != data->localToDualMap.end(); ++it)
      local_b[it->first] = local_duals[it->second];
119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140

    VecRestoreArray(data->tmp_vec_b, &local_b);
    VecRestoreArray(data->tmp_vec_duals0, &local_duals);


    // Multiply with scaled Lagrange constraint matrix.
    MatMult(*(data->mat_lagrange_scaled), data->tmp_vec_b, y);

    return 0;
  }


  // y = PC * x
  PetscErrorCode petscApplyFetiLumpedPrecon(PC pc, Vec x, Vec y)
  {
    // Get data for the preconditioner
    void *ctx;
    PCShellGetContext(pc, &ctx);
    FetiLumpedPreconData* data = static_cast<FetiLumpedPreconData*>(ctx);

    // Multiply with scaled Lagrange constraint matrix.
    MatMultTranspose(*(data->mat_lagrange_scaled), x, data->tmp_vec_b);
141 142


143
    // === Restriction of the B nodes to the boundary nodes. ===
144

145 146 147 148
    int nLocalB;
    int nLocalDuals;
    VecGetLocalSize(data->tmp_vec_b, &nLocalB);
    VecGetLocalSize(data->tmp_vec_duals0, &nLocalDuals);
149

150
    PetscScalar *local_b, *local_duals;
151
    VecGetArray(data->tmp_vec_b, &local_b);
152
    VecGetArray(data->tmp_vec_duals0, &local_duals);
153

Thomas Witkowski's avatar
Thomas Witkowski committed
154 155 156
    for (map<int, int>::iterator it = data->localToDualMap.begin();
	 it != data->localToDualMap.end(); ++it)
      local_duals[it->second] = local_b[it->first];
157 158

    VecRestoreArray(data->tmp_vec_b, &local_b);
159 160 161 162 163 164 165
    VecRestoreArray(data->tmp_vec_duals0, &local_duals);


    // === K_DD ===

    MatMult(*(data->mat_duals_duals), data->tmp_vec_duals0, data->tmp_vec_duals1);

166

167
    // === Prolongation from local dual nodes to B nodes.
168

169 170 171
    VecGetArray(data->tmp_vec_b, &local_b);
    VecGetArray(data->tmp_vec_duals1, &local_duals);

Thomas Witkowski's avatar
Thomas Witkowski committed
172 173 174
    for (map<int, int>::iterator it = data->localToDualMap.begin();
	 it != data->localToDualMap.end(); ++it)
      local_b[it->first] = local_duals[it->second];
175

176 177
    VecRestoreArray(data->tmp_vec_b, &local_b);
    VecRestoreArray(data->tmp_vec_duals0, &local_duals);
178

179 180

    // Multiply with scaled Lagrange constraint matrix.
181 182 183 184 185 186
    MatMult(*(data->mat_lagrange_scaled), data->tmp_vec_b, y);

    return 0;
  }


187 188
  PetscSolverFeti::PetscSolverFeti()
    : PetscSolver(),
189
      schurPrimalSolver(0),
190 191
      multiLevelTest(false),
      subDomainSolver(NULL)
192 193 194 195 196 197
  {
    FUNCNAME("PetscSolverFeti::PetscSolverFeti()");

    string preconditionerName = "";
    Parameters::get("parallel->solver->precon", preconditionerName);
    if (preconditionerName == "" || preconditionerName == "none") {
198
      MSG("Create FETI-DP solver with no preconditioner!\n");
199 200
      fetiPreconditioner = FETI_NONE;
    } else if (preconditionerName == "dirichlet") {
201
      MSG("Create FETI-DP solver with Dirichlet preconditioner!\n");
202 203
      fetiPreconditioner = FETI_DIRICHLET;
    } else if (preconditionerName == "lumped") {
204
      MSG("Create FETI-DP solver with lumped preconditioner!\n");
205 206
      fetiPreconditioner = FETI_LUMPED;
    } else {
Thomas Witkowski's avatar
Thomas Witkowski committed
207 208
      ERROR_EXIT("Preconditioner \"%s\" not available!\n", 
		 preconditionerName.c_str());
209
    }
Thomas Witkowski's avatar
Thomas Witkowski committed
210 211 212 213 214

    Parameters::get("parallel->feti->schur primal solver", schurPrimalSolver);
    TEST_EXIT(schurPrimalSolver == 0 || schurPrimalSolver == 1)
      ("Wrong solver \"%d\"for the Schur primal complement!\n", 
       schurPrimalSolver);
215 216

    Parameters::get("parallel->multi level test", multiLevelTest);
217 218 219
  }


220
  void PetscSolverFeti::updateDofData()
221 222
  {
    FUNCNAME("PetscSolverFeti::updateDofData()");
223

224 225 226 227 228 229
    primalDofMap.clear();
    dualDofMap.clear();
    lagrangeMap.clear();
    localDofMap.clear();
    interiorDofMap.clear();

230 231
    for (unsigned int i = 0; i < meshDistributor->getFeSpaces().size(); i++) {
      const FiniteElemSpace *feSpace = meshDistributor->getFeSpace(i);
232

233 234
      createPrimals(feSpace);      
      createDuals(feSpace);
235
      createLagrange(feSpace);      
236 237
      createIndexB(feSpace);
    }
Thomas Witkowski's avatar
Thomas Witkowski committed
238

239 240 241 242 243
    primalDofMap.setDofComm(meshDistributor->getSendDofs(),
			    meshDistributor->getRecvDofs());
    lagrangeMap.setDofComm(meshDistributor->getSendDofs(), 
			   meshDistributor->getRecvDofs());

Thomas Witkowski's avatar
Thomas Witkowski committed
244 245 246 247
    primalDofMap.update();
    dualDofMap.update();
    lagrangeMap.update();
    localDofMap.update();
248
    if (fetiPreconditioner != FETI_NONE)
249
      interiorDofMap.update();
Thomas Witkowski's avatar
Thomas Witkowski committed
250 251 252

    for (unsigned int i = 0; i < meshDistributor->getFeSpaces().size(); i++) {
      const FiniteElemSpace *feSpace = meshDistributor->getFeSpace(i);
253
      MSG("FETI-DP data for %d-ith FE space:\n", i);
Thomas Witkowski's avatar
Thomas Witkowski committed
254

255
      MSG("  nRankPrimals = %d   nOverallPrimals = %d\n", 
Thomas Witkowski's avatar
Thomas Witkowski committed
256 257
	  primalDofMap[feSpace].nRankDofs, primalDofMap[feSpace].nOverallDofs);
      
258
      MSG("  nRankDuals = %d  nOverallDuals = %d\n",
259
	  dualDofMap[feSpace].nRankDofs, dualDofMap[feSpace].nOverallDofs);
Thomas Witkowski's avatar
Thomas Witkowski committed
260

261
      MSG("  nRankLagrange = %d  nOverallLagrange = %d\n",
262
	  lagrangeMap[feSpace].nRankDofs, lagrangeMap[feSpace].nOverallDofs);
263 264 265 266

      TEST_EXIT_DBG(localDofMap[feSpace].size() + primalDofMap[feSpace].size() == 
		    static_cast<unsigned int>(feSpace->getAdmin()->getUsedDofs()))
	("Should not happen!\n");
Thomas Witkowski's avatar
Thomas Witkowski committed
267
    }
268 269 270


    // If multi level test, inform sub domain solver about coarse space.
271
    subDomainSolver->setDofMapping(&primalDofMap, &localDofMap);
272 273 274
  }


275
  void PetscSolverFeti::createPrimals(const FiniteElemSpace *feSpace)
Thomas Witkowski's avatar
Thomas Witkowski committed
276
  {
277
    FUNCNAME("PetscSolverFeti::createPrimals()");  
278

279 280 281
    // === Define all vertices on the interior boundaries of the macro mesh ===
    // === to be primal variables.                                          ===

282 283 284
    /// Set of DOF indices that are considered to be primal variables.
    
    DofIndexSet primals;
285
    DofContainerSet& vertices = 
286
      meshDistributor->getBoundaryDofInfo(feSpace).geoDofs[VERTEX];
287 288 289 290
    TEST_EXIT_DBG(vertices.size())("No primal vertices on this rank!\n");
    for (DofContainerSet::iterator it = vertices.begin(); 
	 it != vertices.end(); ++it)
      primals.insert(**it);
291

292 293 294 295

    // === Calculate the number of primals that are owned by the rank and ===
    // === create local indices of the primals starting at zero.          ===

296
    for (DofIndexSet::iterator it = primals.begin(); it != primals.end(); ++it)
297 298
      if (meshDistributor->getIsRankDof(feSpace, *it))
	primalDofMap[feSpace].insertRankDof(*it);
Thomas Witkowski's avatar
FETI-DP  
Thomas Witkowski committed
299 300
      else
  	primalDofMap[feSpace].insert(*it);
301 302 303
  }


304
  void PetscSolverFeti::createDuals(const FiniteElemSpace *feSpace)
305 306
  {
    FUNCNAME("PetscSolverFeti::createDuals()");
307

308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323
    // === Create global index of the dual nodes on each rank. ===

    DofContainer allBoundaryDofs;
    meshDistributor->getAllBoundaryDofs(feSpace, allBoundaryDofs);

    for (DofContainer::iterator it = allBoundaryDofs.begin();
	 it != allBoundaryDofs.end(); ++it)
      if (!isPrimal(feSpace, **it))
	dualDofMap[feSpace].insertRankDof(**it);
  }

  
  void PetscSolverFeti::createLagrange(const FiniteElemSpace *feSpace)
  {
    FUNCNAME("PetscSolverFeti::createLagrange()");

324 325
    // === Create for each dual node that is owned by the rank, the set ===
    // === of ranks that contain this node (denoted by W(x_j)).         ===
326

327
    boundaryDofRanks[feSpace].clear();
328

329 330 331
    for (DofComm::Iterator it(meshDistributor->getSendDofs(), feSpace); 
	 !it.end(); it.nextRank())
      for (; !it.endDofIter(); it.nextDof()) {
332
	if (!isPrimal(feSpace, it.getDofIndex())) {
333 334
	  boundaryDofRanks[feSpace][it.getDofIndex()].insert(mpiRank);
	  boundaryDofRanks[feSpace][it.getDofIndex()].insert(it.getRank());
335 336
	}
      }
337
	
338 339 340 341

    // === Communicate these sets for all rank owned dual nodes to other ===
    // === ranks that also have this node.                               ===

342
    StdMpi<vector<std::set<int> > > stdMpi(meshDistributor->getMpiComm());
343 344 345 346

    for (DofComm::Iterator it(meshDistributor->getSendDofs(), feSpace);
	 !it.end(); it.nextRank())
      for (; !it.endDofIter(); it.nextDof())
347
	if (!isPrimal(feSpace, it.getDofIndex()))
348
	  stdMpi.getSendData(it.getRank()).push_back(boundaryDofRanks[feSpace][it.getDofIndex()]);
349 350 351

    stdMpi.updateSendDataSize();

352 353
    for (DofComm::Iterator it(meshDistributor->getRecvDofs(), feSpace); 
	 !it.end(); it.nextRank()) {
354
      bool recvFromRank = false;
355
      for (; !it.endDofIter(); it.nextDof()) {
356
	if (!isPrimal(feSpace, it.getDofIndex())) {
357 358 359
	  recvFromRank = true;
	  break;
	}
360
      }
361 362

      if (recvFromRank)
363
	stdMpi.recv(it.getRank());
364
    }
365

366 367
    stdMpi.startCommunication();

368 369
    for (DofComm::Iterator it(meshDistributor->getRecvDofs(), feSpace); 
	 !it.end(); it.nextRank()) {
370
      int i = 0;
371
      for (; !it.endDofIter(); it.nextDof())
372
	if (!isPrimal(feSpace, it.getDofIndex()))
373
	  boundaryDofRanks[feSpace][it.getDofIndex()] = 
374
	    stdMpi.getRecvData(it.getRank())[i++];
375 376 377
    }


378 379 380
    // === Reserve for each dual node, on the rank that owns this node, the ===
    // === appropriate number of Lagrange constraints.                      ===

381
    int nRankLagrange = 0;
Thomas Witkowski's avatar
Thomas Witkowski committed
382 383
    map<DegreeOfFreedom, MultiIndex>& dualMap = dualDofMap[feSpace].getMap();
    for (map<DegreeOfFreedom, MultiIndex>::iterator it = dualMap.begin(); it != dualMap.end(); ++it) {
384
      if (meshDistributor->getIsRankDof(feSpace, it->first)) {
Thomas Witkowski's avatar
Thomas Witkowski committed
385
	lagrangeMap[feSpace].insertRankDof(it->first, nRankLagrange);
386
	int degree = boundaryDofRanks[feSpace][it->first].size();
387
	nRankLagrange += (degree * (degree - 1)) / 2;
Thomas Witkowski's avatar
Thomas Witkowski committed
388 389
      } else {
	lagrangeMap[feSpace].insert(it->first);
390 391
      }
    }
392
    lagrangeMap[feSpace].nRankDofs = nRankLagrange;
393 394 395
  }


396
  void PetscSolverFeti::createIndexB(const FiniteElemSpace *feSpace)
397
  {
398
    FUNCNAME("PetscSolverFeti::createIndexB()");
399

400
    DOFAdmin* admin = feSpace->getAdmin();
401 402 403 404

    // === To ensure that all interior node on each rank are listen first in ===
    // === the global index of all B nodes, insert all interior nodes first, ===
    // === without defining a correct index.                                 ===
405

406
    int nLocalInterior = 0;    
407
    for (int i = 0; i < admin->getUsedSize(); i++) {
408
      if (admin->isDofFree(i) == false && 
409
	  isPrimal(feSpace, i) == false &&
410
	  dualDofMap[feSpace].isSet(i) == false) {
411 412
	localDofMap[feSpace].insertRankDof(i, nLocalInterior);

413
	if (fetiPreconditioner != FETI_NONE)
414 415 416
	  interiorDofMap[feSpace].insertRankDof(i, nLocalInterior);

	nLocalInterior++;
417
      }
418
    }
Thomas Witkowski's avatar
FETI-DP  
Thomas Witkowski committed
419
    
420 421
    // === And finally, add the global indicies of all dual nodes. ===

Thomas Witkowski's avatar
Thomas Witkowski committed
422
    for (map<DegreeOfFreedom, MultiIndex>::iterator it = dualDofMap[feSpace].getMap().begin();
423
	 it != dualDofMap[feSpace].getMap().end(); ++it)
424
      localDofMap[feSpace].insertRankDof(it->first);
425 426 427
  }


428
  void PetscSolverFeti::createMatLagrange(vector<const FiniteElemSpace*> &feSpaces)
429 430 431
  {
    FUNCNAME("PetscSolverFeti::createMatLagrange()");

432 433
    // === Create distributed matrix for Lagrange constraints. ===

434
    MatCreateMPIAIJ(mpiComm,
435 436
		    lagrangeMap.getRankDofs(), localDofMap.getRankDofs(),
		    lagrangeMap.getOverallDofs(), localDofMap.getOverallDofs(),	
437 438 439
		    2, PETSC_NULL, 2, PETSC_NULL,
		    &mat_lagrange);

440 441 442 443 444 445 446
    // === Create for all duals the corresponding Lagrange constraints. On ===
    // === each rank we traverse all pairs (n, m) of ranks, with n < m,    ===
    // === that contain this node. If the current rank number is r, and    ===
    // === n == r, the rank sets 1.0 for the corresponding constraint, if  ===
    // === m == r, than the rank sets -1.0 for the corresponding           ===
    // === constraint.                                                     ===

Thomas Witkowski's avatar
Thomas Witkowski committed
447 448 449
    for (unsigned int k = 0; k < feSpaces.size(); k++) {
      map<DegreeOfFreedom, MultiIndex> &dualMap = 
	dualDofMap[feSpaces[k]].getMap();
450

Thomas Witkowski's avatar
Thomas Witkowski committed
451 452
      for (map<DegreeOfFreedom, MultiIndex>::iterator it = dualMap.begin(); 
	   it != dualMap.end(); ++it) {
453
	TEST_EXIT_DBG(boundaryDofRanks[feSpaces[k]].count(it->first))
Thomas Witkowski's avatar
Thomas Witkowski committed
454 455 456 457 458 459
	  ("Should not happen!\n");
	
	// Global index of the first Lagrange constriant for this node.
	int index = lagrangeMap.getMatIndex(k, it->first);
	
	// Copy set of all ranks that contain this dual node.
460 461
	vector<int> W(boundaryDofRanks[feSpaces[k]][it->first].begin(), 
		      boundaryDofRanks[feSpaces[k]][it->first].end());
Thomas Witkowski's avatar
Thomas Witkowski committed
462 463 464 465 466 467
	// Number of ranks that contain this dual node.
	int degree = W.size();
	
	for (int i = 0; i < degree; i++) {
	  for (int j = i + 1; j < degree; j++) {
	    if (W[i] == mpiRank || W[j] == mpiRank) {
468
	      int colIndex = localDofMap.getMatIndex(k, it->first);
469
	      double value = (W[i] == mpiRank ? 1.0 : -1.0);
470
	      MatSetValue(mat_lagrange, index, colIndex, value, INSERT_VALUES);
471
	    }
Thomas Witkowski's avatar
Thomas Witkowski committed
472
	    index++;	      
473 474 475 476 477 478 479 480 481 482
	  }
	}
      }
    }

    MatAssemblyBegin(mat_lagrange, MAT_FINAL_ASSEMBLY);
    MatAssemblyEnd(mat_lagrange, MAT_FINAL_ASSEMBLY);
  }


483
  void PetscSolverFeti::createSchurPrimalKsp(vector<const FiniteElemSpace*> &feSpaces)
484 485 486
  {
    FUNCNAME("PetscSolverFeti::createSchurPrimal()");

Thomas Witkowski's avatar
Thomas Witkowski committed
487
    if (schurPrimalSolver == 0) {
488 489
      MSG("Create iterative schur primal solver!\n");

490 491 492
      schurPrimalData.mat_primal_primal = &(subDomainSolver->getMatCoarseCoarse());
      schurPrimalData.mat_primal_b = &(subDomainSolver->getMatCoarseInt());
      schurPrimalData.mat_b_primal = &(subDomainSolver->getMatIntCoarse());
Thomas Witkowski's avatar
Thomas Witkowski committed
493 494
      schurPrimalData.fetiSolver = this;
      
495
      VecCreateMPI(mpiComm, 
496
		   localDofMap.getRankDofs(), localDofMap.getOverallDofs(),
Thomas Witkowski's avatar
Thomas Witkowski committed
497
		   &(schurPrimalData.tmp_vec_b));
498
      VecCreateMPI(mpiComm, 
499
		   primalDofMap.getRankDofs(), primalDofMap.getOverallDofs(),
Thomas Witkowski's avatar
Thomas Witkowski committed
500 501
		   &(schurPrimalData.tmp_vec_primal));

502
      MatCreateShell(mpiComm,
503 504
		     primalDofMap.getRankDofs(), primalDofMap.getRankDofs(), 
		     primalDofMap.getOverallDofs(), primalDofMap.getOverallDofs(),
Thomas Witkowski's avatar
Thomas Witkowski committed
505 506 507 508 509
		     &schurPrimalData, 
		     &mat_schur_primal);
      MatShellSetOperation(mat_schur_primal, MATOP_MULT, 
			   (void(*)(void))petscMultMatSchurPrimal);
      
510
      KSPCreate(mpiComm, &ksp_schur_primal);
Thomas Witkowski's avatar
Thomas Witkowski committed
511
      KSPSetOperators(ksp_schur_primal, mat_schur_primal, mat_schur_primal, SAME_NONZERO_PATTERN);
512 513
      KSPSetOptionsPrefix(ksp_schur_primal, "schur_primal_");
      KSPSetType(ksp_schur_primal, KSPGMRES);
Thomas Witkowski's avatar
Thomas Witkowski committed
514 515
      KSPSetFromOptions(ksp_schur_primal);
    } else {
516 517
      MSG("Create direct schur primal solver!\n");

518 519
      double wtime = MPI::Wtime();

520 521
      int nRowsRankPrimal = primalDofMap.getRankDofs();
      int nRowsOverallPrimal = primalDofMap.getOverallDofs();
522 523
      int nRowsRankB = localDofMap.getRankDofs();
      int nRowsOverallB = localDofMap.getOverallDofs();
524

Thomas Witkowski's avatar
Thomas Witkowski committed
525
      Mat matBPi;
526
      MatCreateMPIAIJ(mpiComm,
Thomas Witkowski's avatar
Thomas Witkowski committed
527 528 529 530
		      nRowsRankB, nRowsRankPrimal, 
		      nRowsOverallB, nRowsOverallPrimal,
		      30, PETSC_NULL, 30, PETSC_NULL, &matBPi);
      Mat matPrimal;
531

532 533 534
      PetscInt nCols;
      const PetscInt *cols;
      const PetscScalar *values;
Thomas Witkowski's avatar
Thomas Witkowski committed
535

536 537
      map<int, vector<pair<int, double> > > mat_b_primal_cols;

538 539
      for (int i = 0; i < nRowsRankB; i++) {
	PetscInt row = localDofMap.getStartDofs() + i;
540
	MatGetRow(subDomainSolver->getMatIntCoarse(), row, &nCols, &cols, &values);
541 542 543 544 545

	for (int j = 0; j < nCols; j++)
	  if (values[j] != 0.0)
	    mat_b_primal_cols[cols[j]].push_back(make_pair(i, values[j]));

546
	MatRestoreRow(subDomainSolver->getMatIntCoarse(), row, &nCols, &cols, &values);
547 548
      }

549 550
      TEST_EXIT(static_cast<int>(mat_b_primal_cols.size()) == 
		primalDofMap.getLocalDofs())
551
	("Should not happen!\n");
552

553 554 555
      for (map<int, vector<pair<int, double> > >::iterator it = mat_b_primal_cols.begin();
	   it != mat_b_primal_cols.end(); ++it) {
	Vec tmpVec;
556
	VecCreateSeq(PETSC_COMM_SELF, nRowsRankB, &tmpVec);
557 558 559 560 561 562 563 564

 	for (unsigned int i = 0; i < it->second.size(); i++) 
 	  VecSetValue(tmpVec, 
 		      it->second[i].first, it->second[i].second, INSERT_VALUES);

	VecAssemblyBegin(tmpVec);
	VecAssemblyEnd(tmpVec);

565
	subDomainSolver->solve(tmpVec, tmpVec);
566

Thomas Witkowski's avatar
Thomas Witkowski committed
567
	PetscScalar *tmpValues;
568
	VecGetArray(tmpVec, &tmpValues);
569
	for (int i  = 0; i < nRowsRankB; i++)
Thomas Witkowski's avatar
Thomas Witkowski committed
570
	  MatSetValue(matBPi, 
571
		      localDofMap.getStartDofs() + i,
Thomas Witkowski's avatar
Thomas Witkowski committed
572 573 574
		      it->first,
		      tmpValues[i],
		      ADD_VALUES);
575 576 577 578 579
	VecRestoreArray(tmpVec, &tmpValues);

	VecDestroy(&tmpVec);
      }

Thomas Witkowski's avatar
Thomas Witkowski committed
580 581
      MatAssemblyBegin(matBPi, MAT_FINAL_ASSEMBLY);
      MatAssemblyEnd(matBPi, MAT_FINAL_ASSEMBLY);
582 583
      MatMatMult(subDomainSolver->getMatCoarseInt(), matBPi, MAT_INITIAL_MATRIX, PETSC_DEFAULT, &matPrimal);
      MatAXPY(subDomainSolver->getMatCoarseCoarse(), -1.0, matPrimal, DIFFERENT_NONZERO_PATTERN);
584 585

      MatDestroy(&matPrimal);
Thomas Witkowski's avatar
Thomas Witkowski committed
586
      MatDestroy(&matBPi);
587 588

      MatInfo minfo;
589
      MatGetInfo(subDomainSolver->getMatCoarseCoarse(), MAT_GLOBAL_SUM, &minfo);
590 591
      MSG("Schur primal matrix nnz = %f\n", minfo.nz_used);

592
      KSPCreate(mpiComm, &ksp_schur_primal);
593 594 595 596
      KSPSetOperators(ksp_schur_primal, 
		      subDomainSolver->getMatCoarseCoarse(),
		      subDomainSolver->getMatCoarseCoarse(),
		      SAME_NONZERO_PATTERN);
597 598 599 600 601 602
      KSPSetOptionsPrefix(ksp_schur_primal, "schur_primal_");
      KSPSetType(ksp_schur_primal, KSPPREONLY);
      PC pc_schur_primal;      
      KSPGetPC(ksp_schur_primal, &pc_schur_primal);
      PCSetType(pc_schur_primal, PCLU);
      PCFactorSetMatSolverPackage(pc_schur_primal, MATSOLVERMUMPS);
603
      KSPSetFromOptions(ksp_schur_primal);
Thomas Witkowski's avatar
Thomas Witkowski committed
604

605 606
      MSG("Creating Schur primal matrix needed %.5f seconds.\n",
	  MPI::Wtime() - wtime);
Thomas Witkowski's avatar
Thomas Witkowski committed
607
    }
608 609 610 611 612 613 614
  }


  void PetscSolverFeti::destroySchurPrimalKsp()
  {
    FUNCNAME("PetscSolverFeti::destroySchurPrimal()");

615 616 617 618 619
    if (schurPrimalSolver == 0) {
      schurPrimalData.mat_primal_primal = PETSC_NULL;
      schurPrimalData.mat_primal_b = PETSC_NULL;
      schurPrimalData.mat_b_primal = PETSC_NULL;
      schurPrimalData.fetiSolver = NULL;
620

621 622
      VecDestroy(&schurPrimalData.tmp_vec_b);
      VecDestroy(&schurPrimalData.tmp_vec_primal);
623

624 625 626 627 628
      MatDestroy(&mat_schur_primal);
      KSPDestroy(&ksp_schur_primal);
    } else {
      KSPDestroy(&ksp_schur_primal);
    }
629 630 631
  }


632
  void PetscSolverFeti::createFetiKsp(vector<const FiniteElemSpace*> &feSpaces)
633 634 635
  {
    FUNCNAME("PetscSolverFeti::createFetiKsp()");

636 637
    // === Create FETI-DP solver object. ===

638 639
    fetiData.mat_primal_b = &(subDomainSolver->getMatCoarseInt());
    fetiData.mat_b_primal = &(subDomainSolver->getMatIntCoarse());
640
    fetiData.mat_lagrange = &mat_lagrange;
Thomas Witkowski's avatar
Thomas Witkowski committed
641
    fetiData.fetiSolver = this;
642
    fetiData.ksp_schur_primal = &ksp_schur_primal;
643

644
    VecCreateMPI(mpiComm, 
645
		 localDofMap.getRankDofs(), localDofMap.getOverallDofs(),
Thomas Witkowski's avatar
Thomas Witkowski committed
646
		 &(fetiData.tmp_vec_b));
647
    VecCreateMPI(mpiComm,
648
		 lagrangeMap.getRankDofs(), lagrangeMap.getOverallDofs(),
Thomas Witkowski's avatar
Thomas Witkowski committed
649
		 &(fetiData.tmp_vec_lagrange));
650
    VecCreateMPI(mpiComm, 
651
		 primalDofMap.getRankDofs(), primalDofMap.getOverallDofs(),
652
		 &(fetiData.tmp_vec_primal));
653

654
    MatCreateShell(mpiComm,
655 656
		   lagrangeMap.getRankDofs(), lagrangeMap.getRankDofs(),
		   lagrangeMap.getOverallDofs(), lagrangeMap.getOverallDofs(),
657
		   &fetiData, &mat_feti);
658 659 660
    MatShellSetOperation(mat_feti, MATOP_MULT, (void(*)(void))petscMultMatFeti);


661
    KSPCreate(mpiComm, &ksp_feti);
662
    KSPSetOperators(ksp_feti, mat_feti, mat_feti, SAME_NONZERO_PATTERN);
663 664 665
    KSPSetOptionsPrefix(ksp_feti, "feti_");
    KSPSetType(ksp_feti, KSPGMRES);
    KSPSetTolerances(ksp_feti, 0, 1e-8, 1e+3, 1000);
666
    KSPSetFromOptions(ksp_feti);
667 668


669
    // === Create FETI-DP preconditioner object. ===
670

671 672 673 674
    if (fetiPreconditioner != FETI_NONE) {
      MatDuplicate(mat_lagrange, MAT_COPY_VALUES, &mat_lagrange_scaled);
      MatScale(mat_lagrange_scaled, 0.5);
    }
675

676 677 678
    switch (fetiPreconditioner) {
    case FETI_DIRICHLET:           
      KSPCreate(PETSC_COMM_SELF, &ksp_interior);
Thomas Witkowski's avatar
Thomas Witkowski committed
679 680
      KSPSetOperators(ksp_interior, mat_interior_interior, mat_interior_interior, 
		      SAME_NONZERO_PATTERN);
681 682 683 684 685 686
      KSPSetOptionsPrefix(ksp_interior, "precon_interior_");
      KSPSetType(ksp_interior, KSPPREONLY);
      PC pc_interior;
      KSPGetPC(ksp_interior, &pc_interior);
      PCSetType(pc_interior, PCLU);
      PCFactorSetMatSolverPackage(pc_interior, MATSOLVERUMFPACK);
687 688 689 690 691 692 693 694 695
      KSPSetFromOptions(ksp_interior);
            
      fetiDirichletPreconData.mat_lagrange_scaled = &mat_lagrange_scaled;
      fetiDirichletPreconData.mat_interior_interior = &mat_interior_interior;
      fetiDirichletPreconData.mat_duals_duals = &mat_duals_duals;
      fetiDirichletPreconData.mat_interior_duals = &mat_interior_duals;
      fetiDirichletPreconData.mat_duals_interior = &mat_duals_interior;
      fetiDirichletPreconData.ksp_interior = &ksp_interior;
      
696
      VecCreateMPI(mpiComm, 
697
		   localDofMap.getRankDofs(),localDofMap.getOverallDofs(),
698
		   &(fetiDirichletPreconData.tmp_vec_b));      
699 700 701 702 703 704
      MatGetVecs(mat_duals_duals, PETSC_NULL, 
		 &(fetiDirichletPreconData.tmp_vec_duals0));
      MatGetVecs(mat_duals_duals, PETSC_NULL, 
		 &(fetiDirichletPreconData.tmp_vec_duals1));
      MatGetVecs(mat_interior_interior, PETSC_NULL, 
		 &(fetiDirichletPreconData.tmp_vec_interior));
705 706 707 708 709 710 711 712 713 714 715 716 717

      for (unsigned int i = 0; i < feSpaces.size(); i++) {
	map<DegreeOfFreedom, MultiIndex> &dualMap = 
	  dualDofMap[feSpaces[i]].getMap();
	for (map<DegreeOfFreedom, MultiIndex>::iterator it = dualMap.begin(); 
	     it != dualMap.end(); ++it) {
	  DegreeOfFreedom d = it->first;
	  int matIndexLocal = localDofMap.getLocalMatIndex(i, d);
	  int matIndexDual = dualDofMap.getLocalMatIndex(i, d);
	  fetiDirichletPreconData.localToDualMap[matIndexLocal] = matIndexDual;
	}
      }

718 719 720 721 722 723 724 725 726 727 728
      KSPGetPC(ksp_feti, &precon_feti);
      PCSetType(precon_feti, PCSHELL);
      PCShellSetContext(precon_feti, static_cast<void*>(&fetiDirichletPreconData));
      PCShellSetApply(precon_feti, petscApplyFetiDirichletPrecon);
      
      break;

    case FETI_LUMPED:
      fetiLumpedPreconData.mat_lagrange_scaled = &mat_lagrange_scaled;
      fetiLumpedPreconData.mat_duals_duals = &mat_duals_duals;

Thomas Witkowski's avatar
Thomas Witkowski committed
729 730 731 732 733 734 735 736 737 738 739 740
      for (unsigned int i = 0; i < feSpaces.size(); i++) {
	map<DegreeOfFreedom, MultiIndex> &dualMap = 
	  dualDofMap[feSpaces[i]].getMap();
	for (map<DegreeOfFreedom, MultiIndex>::iterator it = dualMap.begin(); 
	     it != dualMap.end(); ++it) {
	  DegreeOfFreedom d = it->first;
	  int matIndexLocal = localDofMap.getLocalMatIndex(i, d);
	  int matIndexDual = dualDofMap.getLocalMatIndex(i, d);
	  fetiLumpedPreconData.localToDualMap[matIndexLocal] = matIndexDual;
	}
      }

741
      VecCreateMPI(mpiComm, 
742 743
		   localDofMap.getRankDofs(),
		   localDofMap.getOverallDofs(),
744
		   &(fetiLumpedPreconData.tmp_vec_b));
745 746 747 748
      MatGetVecs(mat_duals_duals, PETSC_NULL, 
		 &(fetiLumpedPreconData.tmp_vec_duals0));
      MatGetVecs(mat_duals_duals, PETSC_NULL, 
		 &(fetiLumpedPreconData.tmp_vec_duals1));
749 750 751 752 753 754

      KSPGetPC(ksp_feti, &precon_feti);
      PCSetType(precon_feti, PCSHELL);
      PCShellSetContext(precon_feti, static_cast<void*>(&fetiLumpedPreconData));
      PCShellSetApply(precon_feti, petscApplyFetiLumpedPrecon);
      
755 756
      break;
    default:
757 758
      break;
    }
759 760 761 762 763 764 765
  }
  

  void PetscSolverFeti::destroyFetiKsp()
  {
    FUNCNAME("PetscSolverFeti::destroyFetiKsp()");

766 767
    // === Destroy FETI-DP solver object. ===

768 769 770
    fetiData.mat_primal_b = PETSC_NULL;
    fetiData.mat_b_primal = PETSC_NULL;
    fetiData.mat_lagrange = PETSC_NULL;
Thomas Witkowski's avatar
Thomas Witkowski committed
771
    fetiData.fetiSolver = NULL;
772
    fetiData.ksp_schur_primal = PETSC_NULL;
773

Thomas Witkowski's avatar
Thomas Witkowski committed
774 775
    VecDestroy(&fetiData.tmp_vec_b);
    VecDestroy(&fetiData.tmp_vec_lagrange);
776
    VecDestroy(&fetiData.tmp_vec_primal);
777 778
    MatDestroy(&mat_feti);
    KSPDestroy(&ksp_feti);
779 780


781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808
    // === Destroy FETI-DP preconditioner object. ===

    switch (fetiPreconditioner) {
    case FETI_DIRICHLET:           
      KSPDestroy(&ksp_interior);

      fetiDirichletPreconData.mat_lagrange_scaled = NULL;
      fetiDirichletPreconData.mat_interior_interior = NULL;
      fetiDirichletPreconData.mat_duals_duals = NULL;
      fetiDirichletPreconData.mat_interior_duals = NULL;
      fetiDirichletPreconData.mat_duals_interior = NULL;
      fetiDirichletPreconData.ksp_interior = NULL;
      
      VecDestroy(&fetiDirichletPreconData.tmp_vec_b);
      VecDestroy(&fetiDirichletPreconData.tmp_vec_duals0);
      VecDestroy(&fetiDirichletPreconData.tmp_vec_duals1);
      VecDestroy(&fetiDirichletPreconData.tmp_vec_interior);
      MatDestroy(&mat_lagrange_scaled);
      break;

    case FETI_LUMPED:
      fetiLumpedPreconData.mat_lagrange_scaled = NULL;
      fetiLumpedPreconData.mat_duals_duals = NULL;

      VecDestroy(&fetiLumpedPreconData.tmp_vec_b);
      VecDestroy(&fetiLumpedPreconData.tmp_vec_duals0);
      VecDestroy(&fetiLumpedPreconData.tmp_vec_duals1);
      break;
809 810
    default:
      break;
811
    }
812 813 814 815 816 817 818 819 820
  }


  void PetscSolverFeti::recoverSolution(Vec &vec_sol_b,
					Vec &vec_sol_primal,
					SystemVector &vec)
  {
    FUNCNAME("PetscSolverFeti::recoverSolution()");

821
    // === Get local part of the solution for B variables. ===
822 823 824 825 826

    PetscScalar *localSolB;
    VecGetArray(vec_sol_b, &localSolB);


827 828
    // === Create scatter to get solutions of all primal nodes that are ===
    // === contained in rank's domain.                                  ===
829
    
Thomas Witkowski's avatar
Thomas Witkowski committed
830 831
    vector<const FiniteElemSpace*> feSpaces = getFeSpaces(&vec);

832
    vector<PetscInt> globalIsIndex, localIsIndex;
Thomas Witkowski's avatar
Thomas Witkowski committed
833 834
    globalIsIndex.reserve(primalDofMap.getLocalDofs());
    localIsIndex.reserve(primalDofMap.getLocalDofs());
835 836

    {
Thomas Witkowski's avatar
Thomas Witkowski committed
837 838 839 840 841 842 843 844
      int cnt = 0;
      for (unsigned int i = 0; i < feSpaces.size(); i++) {
	map<DegreeOfFreedom, MultiIndex>& dofMap = 
	  primalDofMap[feSpaces[i]].getMap();
	for (map<DegreeOfFreedom, MultiIndex>::iterator it = dofMap.begin();
	     it != dofMap.end(); ++it) {
	  globalIsIndex.push_back(primalDofMap.getMatIndex(i, it->first));
	  localIsIndex.push_back(cnt++);
845 846
	}
      }
847
      
Thomas Witkowski's avatar
Thomas Witkowski committed
848
      TEST_EXIT_DBG(cnt == primalDofMap.getLocalDofs())("Should not happen!\n");
849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873
    }
    
    IS globalIs, localIs;
    ISCreateGeneral(PETSC_COMM_SELF, 
		    globalIsIndex.size(), 
		    &(globalIsIndex[0]),
		    PETSC_USE_POINTER,
		    &globalIs);

    ISCreateGeneral(PETSC_COMM_SELF, 
		    localIsIndex.size(), 
		    &(localIsIndex[0]),
		    PETSC_USE_POINTER,
		    &localIs);

    Vec local_sol_primal;
    VecCreateSeq(PETSC_COMM_SELF, localIsIndex.size(), &local_sol_primal);

    VecScatter primalScatter;
    VecScatterCreate(vec_sol_primal, globalIs, local_sol_primal, localIs, &primalScatter);
    VecScatterBegin(primalScatter, vec_sol_primal, local_sol_primal, 
		    INSERT_VALUES, SCATTER_FORWARD);
    VecScatterEnd(primalScatter, vec_sol_primal, local_sol_primal, 
		  INSERT_VALUES, SCATTER_FORWARD);

874 875 876
    ISDestroy(&globalIs);
    ISDestroy(&localIs);    
    VecScatterDestroy(&primalScatter);    
877 878 879 880

    PetscScalar *localSolPrimal;
    VecGetArray(local_sol_primal, &localSolPrimal);

881
    // === And copy from PETSc local vectors to the DOF vectors. ===
882

Thomas Witkowski's avatar
Thomas Witkowski committed
883 884
    int cnt = 0;
    for (int i = 0; i < vec.getSize(); i++) {
885 886
      DOFVector<double>& dofVec = *(vec.getDOFVector(i));

Thomas Witkowski's avatar
Thomas Witkowski committed
887 888
      for (map<DegreeOfFreedom, MultiIndex>::iterator it = localDofMap[feSpaces[i]].getMap().begin();
	   it != localDofMap[feSpaces[i]].getMap().end(); ++it) {
889
	int petscIndex = localDofMap.getLocalMatIndex(i, it->first);
890 891 892
	dofVec[it->first] = localSolB[petscIndex];
      }

Thomas Witkowski's avatar
Thomas Witkowski committed
893 894 895
      for (map<DegreeOfFreedom, MultiIndex>::iterator it = primalDofMap[feSpaces[i]].getMap().begin();
	   it != primalDofMap[feSpaces[i]].getMap().end(); ++it)
	dofVec[it->first] = localSolPrimal[cnt++];