PetscExport.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2016-2024 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2024 TotalEnergies
 * Copyright (c) 2018-2024 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2023-2024 Chevron
 * Copyright (c) 2019-     GEOS/GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_LINEARALGEBRA_INTERFACES_PETSCEXPORT_HPP_
#define GEOS_LINEARALGEBRA_INTERFACES_PETSCEXPORT_HPP_
 
#include "common/DataTypes.hpp"
 
extern "C" struct _p_IS;
 
extern "C" struct _p_VecScatter;
 
namespace geos
{
 
class PetscMatrix;
 
class PetscVector;
 
class PetscExport
{
public:
 
  PetscExport();
 
  PetscExport( PetscMatrix const & mat, integer targetRank );
 
  ~PetscExport();
 
  template< typename OFFSET_TYPE, typename COLUMN_TYPE >
  void exportCRS( PetscMatrix const & mat,
                  arrayView1d< OFFSET_TYPE > const & rowOffsets,
                  arrayView1d< COLUMN_TYPE > const & colIndices,
                  arrayView1d< real64 > const & values ) const;
 
  void exportVector( PetscVector const & vec,
                     arrayView1d< real64 > const & values ) const;
 
  void importVector( arrayView1d< const real64 > const & values,
                     PetscVector & vec ) const;
 
private:
 
  using IS = struct _p_IS *;
 
  using VecScatter = struct _p_VecScatter *;
 
  integer m_targetRank = -1;
 
  IS m_indexSet{};
 
  VecScatter m_scatter{};
};
 
} // namespace geos
 
#endif //GEOS_LINEARALGEBRA_INTERFACES_PETSCEXPORT_HPP_