HypreParVector.hpp

Go to the documentation of this file.

// Copyright (c) 2010, Lawrence Livermore National Security, LLC. Produced at
// the Lawrence Livermore National Laboratory. LLNL-CODE-443211. All Rights
// reserved. See file COPYRIGHT for details.
//
// This file is part of the MFEM library. For more information and source code
// availability see http://mfem.org.
//
// MFEM is free software; you can redistribute it and/or modify it under the
// terms of the GNU Lesser General Public License (as published by the Free
// Software Foundation) version 2.1 dated February 1999.
 
/// This HYPRE library interface has been taken originally from MFEM and modified
/// to suit the needs for the MOAB library.
/// Modified by: Vijay Mahadevan
 
#ifndef MOAB_HYPREPARVECTOR
#define MOAB_HYPREPARVECTOR
 
#include "moab/MOABConfig.h"
#include "moab/Core.hpp"
 
#ifdef MOAB_HAVE_EIGEN3
#include <Eigen/Core>
#include <Eigen/Sparse>
#else
#error Configure with Eigen3 enabled
#endif
 
#ifdef MOAB_HAVE_MPI
#include "moab/ParallelComm.hpp"
 
// hypre header files
#include "HYPRE.h"
#include "HYPRE_IJ_mv.h"
#include "_hypre_IJ_mv.h"
#include "HYPRE_parcsr_ls.h"
// #include "seq_mv.h"
#include "temp_multivector.h"
 
#ifdef HYPRE_COMPLEX
#error "MOAB does not work with HYPRE's complex numbers support"
#endif
 
#include "hypre_parcsr.hpp"
 
namespace moab
{
 
class HypreParMatrix;
class HypreSolver;
 
/// Wrapper for hypre's parallel vector class
class HypreParVector
{
  private:
    int own_ParVector, rstart, rend, size, gsize;
 
    /// The actual object
    hypre_ParVector* x_par;
    HYPRE_IJVector x;
 
    friend class HypreParMatrix;
    friend class HypreSolver;
 
    moab::ParallelComm* pcomm;
    char initialized;
 
  public:
    /** Creates an empty vector with given global comm. */
    HypreParVector( moab::ParallelComm* p_comm );
    /** Creates vector with given global size and partitioning of the columns.
        Processor P owns columns [col[P],col[P+1]) */
    HypreParVector( moab::ParallelComm* p_comm, HYPRE_Int glob_size, HYPRE_Int rstart, HYPRE_Int rend );
    /// Creates vector compatible with y
    HypreParVector( const HypreParVector& y );
    /// Creates vector compatible with (i.e. in the domain of) A or A^T
    HypreParVector( HypreParMatrix& A, int tr = 0 );
 
    int resize( HYPRE_Int glob_size, HYPRE_Int rstart, HYPRE_Int rend );
 
    /// MPI communicator
    moab::ParallelComm* GetParallelCommunicator()
    {
        return pcomm;
    }
 
    /// Returns the row partitioning
    inline HYPRE_Int* Partitioning()
    {
        return x->partitioning;
    }
 
    /// Returns the global number of rows
    inline HYPRE_Int GlobalSize() const
    {
        return x->global_num_rows;
    }
 
    /// Typecasting to hypre's HYPRE_IJVector*
    operator HYPRE_IJVector() const;
    // #ifndef HYPRE_PAR_VECTOR_STRUCT
    /// Typecasting to hypre's HYPRE_ParVector, a.k.a. void *
    operator HYPRE_ParVector() const;
    // #endif
    /// Changes the ownership of the the vector
    HYPRE_IJVector StealParVector()
    {
        own_ParVector = 0;
        return x;
    }
 
    /// Sets ownership of the internal HYPRE_IJVector
    void SetOwnership( int own )
    {
        own_ParVector = own;
    }
 
    /// Gets ownership of the internal HYPRE_IJVector
    int GetOwnership() const
    {
        return own_ParVector;
    }
 
    /// Define '=' for hypre vectors.
    HypreParVector& operator=( double d );
    HypreParVector& operator=( const HypreParVector& y );
 
    HYPRE_Int GetValues( const int ndata, const HYPRE_Int* indices, HYPRE_Complex* const _data ) const;
 
    HYPRE_Int SetValues( const int ndata, const HYPRE_Int* indices, const HYPRE_Complex* const _data );
 
    HYPRE_Int AddValues( const int ndata, const HYPRE_Int* indices, const HYPRE_Complex* const _data );
 
    HYPRE_Int GetValue( const HYPRE_Int index, HYPRE_Complex* const _data ) const;
 
    HYPRE_Int SetValue( const HYPRE_Int index, const HYPRE_Complex _data );
 
    HYPRE_Int AddValue( const HYPRE_Int index, const HYPRE_Complex _data );
 
    /** Sets the data of the HYPRE_IJVector to _data.
        Must be the same length as the current local size of the vector.
        If not, this can lead to an inconsistent vector setup. */
    HYPRE_Int SetData( double* p_data, HYPRE_Int* p_col = NULL );
 
    /** Add the data of the HYPRE_IJVector to _data.
        Must be the same length as the current local size of the vector.
        If not, this can lead to an inconsistent vector setup. */
    HYPRE_Int AddData( double* p_data, HYPRE_Int* p_col = NULL );
 
    HYPRE_Int verbosity( const HYPRE_Int level );
 
    HYPRE_Int FinalizeAssembly();
 
    // const HYPRE_Complex& operator()(int index) const;
 
    // HYPRE_Complex& operator()(int index);
 
    /// Prints the locally owned rows in parallel
    void Print( const char* fname ) const;
 
    /// Calls hypre's destroy function
    ~HypreParVector();
 
    static double InnerProduct( HypreParVector& x, HypreParVector& y );
};
 
/// Returns the inner product of x and y
double InnerProduct( HypreParVector& x, HypreParVector& y );
double InnerProduct( HypreParVector* x, HypreParVector* y );
 
}  // namespace moab
 
#endif  // MOAB_HAVE_MPI
 
#endif