| 1 | /*
 | 
|---|
| 2 |  *    vmg - a versatile multigrid solver
 | 
|---|
| 3 |  *    Copyright (C) 2012 Institute for Numerical Simulation, University of Bonn
 | 
|---|
| 4 |  *
 | 
|---|
| 5 |  *  vmg is free software: you can redistribute it and/or modify
 | 
|---|
| 6 |  *  it under the terms of the GNU General Public License as published by
 | 
|---|
| 7 |  *  the Free Software Foundation, either version 3 of the License, or
 | 
|---|
| 8 |  *  (at your option) any later version.
 | 
|---|
| 9 |  *
 | 
|---|
| 10 |  *  vmg is distributed in the hope that it will be useful,
 | 
|---|
| 11 |  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
|---|
| 12 |  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
|---|
| 13 |  *  GNU General Public License for more details.
 | 
|---|
| 14 |  *
 | 
|---|
| 15 |  *  You should have received a copy of the GNU General Public License
 | 
|---|
| 16 |  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 | 
|---|
| 17 |  */
 | 
|---|
| 18 | 
 | 
|---|
| 19 | /**
 | 
|---|
| 20 |  * @file   solver.cpp
 | 
|---|
| 21 |  * @author Julian Iseringhausen <isering@ins.uni-bonn.de>
 | 
|---|
| 22 |  * @date   Mon Apr 18 13:10:55 2011
 | 
|---|
| 23 |  *
 | 
|---|
| 24 |  * @brief  VMG::Solver
 | 
|---|
| 25 |  *
 | 
|---|
| 26 |  */
 | 
|---|
| 27 | 
 | 
|---|
| 28 | #ifdef HAVE_CONFIG_H
 | 
|---|
| 29 | #include <libvmg_config.h>
 | 
|---|
| 30 | #endif
 | 
|---|
| 31 | 
 | 
|---|
| 32 | #include "solver/solver.hpp"
 | 
|---|
| 33 | 
 | 
|---|
| 34 | using namespace VMG;
 | 
|---|
| 35 | 
 | 
|---|
| 36 | void Solver::Run(Grid& sol, Grid& rhs)
 | 
|---|
| 37 | {
 | 
|---|
| 38 | #ifdef DEBUG_MATRIX_CHECKS
 | 
|---|
| 39 |   sol.IsConsistent();
 | 
|---|
| 40 |   rhs.IsConsistent();
 | 
|---|
| 41 | #endif
 | 
|---|
| 42 | 
 | 
|---|
| 43 | #ifdef DEBUG
 | 
|---|
| 44 |   sol.IsCompatible(rhs);
 | 
|---|
| 45 | #endif
 | 
|---|
| 46 | 
 | 
|---|
| 47 |   if (rhs.Global().LocalSize().Product() > 0) {
 | 
|---|
| 48 |     this->Realloc(rhs);
 | 
|---|
| 49 |     this->AssembleMatrix(rhs);
 | 
|---|
| 50 |     this->Compute();
 | 
|---|
| 51 |     this->ExportSol(sol, rhs);
 | 
|---|
| 52 |   }
 | 
|---|
| 53 | }
 | 
|---|
| 54 | 
 | 
|---|
| 55 | void Solver::Realloc(int n)
 | 
|---|
| 56 | {
 | 
|---|
| 57 |   //Reallocate memory if necessary
 | 
|---|
| 58 |   this->size = n;
 | 
|---|
| 59 | 
 | 
|---|
| 60 |   if (static_cast<int>(A.size()) < n*n)
 | 
|---|
| 61 |     A.resize(n*n);
 | 
|---|
| 62 | 
 | 
|---|
| 63 |   if (static_cast<int>(b.size()) < n)
 | 
|---|
| 64 |     b.resize(n);
 | 
|---|
| 65 | 
 | 
|---|
| 66 |   if (static_cast<int>(x.size()) < n)
 | 
|---|
| 67 |     x.resize(n);
 | 
|---|
| 68 | }
 | 
|---|
| 69 | 
 | 
|---|
| 70 | void Solver::Realloc(Grid& sol)
 | 
|---|
| 71 | {
 | 
|---|
| 72 |   this->Realloc(sol.Global().GlobalSize().Product());
 | 
|---|
| 73 | }
 | 
|---|