Nektar++
GlobalLinSysPETSc.h
Go to the documentation of this file.
1 ///////////////////////////////////////////////////////////////////////////////
2 //
3 // File GlobalLinSys.h
4 //
5 // For more information, please see: http://www.nektar.info
6 //
7 // The MIT License
8 //
9 // Copyright (c) 2006 Division of Applied Mathematics, Brown University (USA),
10 // Department of Aeronautics, Imperial College London (UK), and Scientific
11 // Computing and Imaging Institute, University of Utah (USA).
12 //
13 // License for the specific language governing rights and limitations under
14 // Permission is hereby granted, free of charge, to any person obtaining a
15 // copy of this software and associated documentation files (the "Software"),
16 // to deal in the Software without restriction, including without limitation
17 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
18 // and/or sell copies of the Software, and to permit persons to whom the
19 // Software is furnished to do so, subject to the following conditions:
20 //
21 // The above copyright notice and this permission notice shall be included
22 // in all copies or substantial portions of the Software.
23 //
24 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
25 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
26 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
27 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
28 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
29 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
30 // DEALINGS IN THE SOFTWARE.
31 //
32 // Description: GlobalLinSysPETSc header
33 //
34 ///////////////////////////////////////////////////////////////////////////////
35 #ifndef NEKTAR_LIB_MULTIREGIONS_GLOBALLINSYSPETSC_H
36 #define NEKTAR_LIB_MULTIREGIONS_GLOBALLINSYSPETSC_H
37 
40 
41 #include <petscmat.h>
42 #include <petscksp.h>
43 
44 namespace Nektar
45 {
46  namespace MultiRegions
47  {
48  // Forward declarations
49  class ExpList;
50 
51  /// A global linear system.
52  class GlobalLinSysPETSc : virtual public GlobalLinSys
53  {
54  public:
55  /// Constructor for full direct matrix solve.
57  const GlobalLinSysKey &pKey,
58  const boost::weak_ptr<ExpList> &pExp,
59  const boost::shared_ptr<AssemblyMap> &pLocToGloMap);
60 
62 
63  virtual void v_SolveLinearSystem(
64  const int pNumRows,
65  const Array<OneD,const NekDouble> &pInput,
66  Array<OneD, NekDouble> &pOutput,
67  const AssemblyMapSharedPtr &locToGloMap,
68  const int pNumDir);
69 
70  protected:
71  Mat m_matrix;
72  Vec m_x, m_b, m_locVec;
73  KSP m_ksp;
74  vector<int> m_reorderedMap;
75  VecScatter m_ctx;
76  int m_nLocal;
77 
78  void SetUpScatter();
79  void SetUpMatVec();
80  void SetUpSolver(NekDouble tolerance);
82  const Array<OneD, const int> &glo2uniMap,
83  const Array<OneD, const int> &glo2unique,
84  const AssemblyMapSharedPtr &pLocToGloMap);
85  };
86  }
87 }
88 
89 #endif
void SetUpMatVec()
Construct PETSc matrix and vector handles.
boost::shared_ptr< AssemblyMap > AssemblyMapSharedPtr
Definition: AssemblyMap.h:53
void CalculateReordering(const Array< OneD, const int > &glo2uniMap, const Array< OneD, const int > &glo2unique, const AssemblyMapSharedPtr &pLocToGloMap)
Calculate a reordering of universal IDs for PETSc.
#define MULTI_REGIONS_EXPORT
void SetUpSolver(NekDouble tolerance)
Set up KSP solver object.
GlobalLinSysPETSc(const GlobalLinSysKey &pKey, const boost::weak_ptr< ExpList > &pExp, const boost::shared_ptr< AssemblyMap > &pLocToGloMap)
Constructor for full direct matrix solve.
double NekDouble
Describe a linear system.
void SetUpScatter()
Set up PETSc local (equivalent to Nektar++ global) and global (equivalent to universal) scatter maps...
A global linear system.
Definition: GlobalLinSys.h:70
virtual void v_SolveLinearSystem(const int pNumRows, const Array< OneD, const NekDouble > &pInput, Array< OneD, NekDouble > &pOutput, const AssemblyMapSharedPtr &locToGloMap, const int pNumDir)
Solve linear system using PETSc.