PCBlockDiagonal.hpp

00001 /*
00002 
00003 Copyright (C) University of Oxford, 2005-2011
00004 
00005 University of Oxford means the Chancellor, Masters and Scholars of the
00006 University of Oxford, having an administrative office at Wellington
00007 Square, Oxford OX1 2JD, UK.
00008 
00009 This file is part of Chaste.
00010 
00011 Chaste is free software: you can redistribute it and/or modify it
00012 under the terms of the GNU Lesser General Public License as published
00013 by the Free Software Foundation, either version 2.1 of the License, or
00014 (at your option) any later version.
00015 
00016 Chaste is distributed in the hope that it will be useful, but WITHOUT
00017 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
00018 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
00019 License for more details. The offer of Chaste under the terms of the
00020 License is subject to the License being interpreted in accordance with
00021 English Law and subject to any action against the University of Oxford
00022 being under the jurisdiction of the English Courts.
00023 
00024 You should have received a copy of the GNU Lesser General Public License
00025 along with Chaste. If not, see <http://www.gnu.org/licenses/>.
00026 
00027 */
00028 
00029 
00030 #ifndef PCBLOCKDIAGONAL_HPP_
00031 #define PCBLOCKDIAGONAL_HPP_
00032 
00033 #include <cassert>
00034 #include <petscvec.h>
00035 #include <petscmat.h>
00036 #include <petscksp.h>
00037 #include <petscpc.h>
00038 #include "PetscTools.hpp"
00039 
00050 #if (PETSC_VERSION_MAJOR == 3 && PETSC_VERSION_MINOR == 1) //PETSc 3.1
00051 PetscErrorCode PCBlockDiagonalApply(PC pc_context, Vec x, Vec y);
00052 #else
00053 PetscErrorCode PCBlockDiagonalApply(void* pc_context, Vec x, Vec y);
00054 #endif
00055 
00079 class PCBlockDiagonal
00080 {
00081 public:
00085     typedef struct{
00086         Mat A11_matrix_subblock; 
00087         Mat A22_matrix_subblock; 
00088         PC  PC_amg_A11; 
00089         PC  PC_amg_A22; 
00090         Vec x1_subvector;
00091         Vec x2_subvector;
00092         Vec y1_subvector;
00093         Vec y2_subvector;
00094         VecScatter A11_scatter_ctx;
00095         VecScatter A22_scatter_ctx;
00096 #ifdef TRACE_KSP
00097         double mScatterTime;
00098         double mA1PreconditionerTime;
00099         double mA2PreconditionerTime;
00100         double mGatherTime;
00101 #endif
00102 
00103     } PCBlockDiagonalContext;
00104 
00105     PCBlockDiagonalContext mPCContext; 
00106     PC mPetscPCObject;
00113     PCBlockDiagonal(KSP& rKspObject);
00114 
00115     ~PCBlockDiagonal();
00116 
00117 private:
00123     void PCBlockDiagonalCreate(KSP& rKspObject);
00124 
00128     void PCBlockDiagonalSetUp();
00129 };
00130 #endif /*PCBLOCKDIAGONAL_HPP_*/

Generated on Tue May 31 14:31:42 2011 for Chaste by  doxygen 1.5.5