00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046 #ifndef XPETRA_CRSMATRIX_HPP
00047 #define XPETRA_CRSMATRIX_HPP
00048
00049
00050
00051 #include <Kokkos_DefaultNode.hpp>
00052 #include <Kokkos_DefaultKernels.hpp>
00053 #include "Xpetra_ConfigDefs.hpp"
00054 #include "Xpetra_RowMatrix.hpp"
00055 #include "Xpetra_DistObject.hpp"
00056 #include "Xpetra_CrsGraph.hpp"
00057 #include "Xpetra_Vector.hpp"
00058
00059 namespace Xpetra {
00060
00061 template <class Scalar, class LocalOrdinal = int, class GlobalOrdinal = LocalOrdinal, class Node = Kokkos::DefaultNode::DefaultNodeType, class LocalMatOps = typename Kokkos::DefaultKernels<Scalar,LocalOrdinal,Node>::SparseOps>
00062 class CrsMatrix
00063 : public RowMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node>, public DistObject<char, LocalOrdinal,GlobalOrdinal,Node>
00064 {
00065
00066 public:
00067
00069
00070
00072 virtual ~CrsMatrix() { }
00073
00075
00077
00078
00080 virtual void insertGlobalValues(GlobalOrdinal globalRow, const ArrayView< const GlobalOrdinal > &cols, const ArrayView< const Scalar > &vals)= 0;
00081
00083 virtual void insertLocalValues(LocalOrdinal localRow, const ArrayView< const LocalOrdinal > &cols, const ArrayView< const Scalar > &vals)= 0;
00084
00086 virtual void setAllToScalar(const Scalar &alpha)= 0;
00087
00089 virtual void scale(const Scalar &alpha)= 0;
00090
00092
00094
00095
00097 virtual void fillComplete(const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &domainMap, const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &rangeMap, const RCP< ParameterList > ¶ms=null)= 0;
00098
00100 virtual void fillComplete(const RCP< ParameterList > ¶ms=null)= 0;
00101
00103
00105
00106
00108 virtual const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getRowMap() const = 0;
00109
00111 virtual const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getColMap() const = 0;
00112
00114 virtual RCP< const CrsGraph< LocalOrdinal, GlobalOrdinal, Node, LocalMatOps > > getCrsGraph() const = 0;
00115
00117 virtual global_size_t getGlobalNumRows() const = 0;
00118
00120 virtual global_size_t getGlobalNumCols() const = 0;
00121
00123 virtual size_t getNodeNumRows() const = 0;
00124
00126 virtual global_size_t getGlobalNumEntries() const = 0;
00127
00129 virtual size_t getNodeNumEntries() const = 0;
00130
00132 virtual size_t getNumEntriesInLocalRow(LocalOrdinal localRow) const = 0;
00133
00135 virtual global_size_t getGlobalNumDiags() const = 0;
00136
00138 virtual size_t getNodeNumDiags() const = 0;
00139
00141 virtual size_t getGlobalMaxNumRowEntries() const = 0;
00142
00144 virtual size_t getNodeMaxNumRowEntries() const = 0;
00145
00147 virtual bool isLocallyIndexed() const = 0;
00148
00150 virtual bool isGloballyIndexed() const = 0;
00151
00153 virtual bool isFillComplete() const = 0;
00154
00156 virtual typename ScalarTraits< Scalar >::magnitudeType getFrobeniusNorm() const = 0;
00157
00159 virtual void getGlobalRowView(GlobalOrdinal GlobalRow, ArrayView< const GlobalOrdinal > &indices, ArrayView< const Scalar > &values) const = 0;
00160
00162 virtual void getLocalRowView(LocalOrdinal LocalRow, ArrayView< const LocalOrdinal > &indices, ArrayView< const Scalar > &values) const = 0;
00163
00165 virtual void getLocalDiagCopy(Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &diag) const = 0;
00166
00168
00170
00171
00173 virtual void apply(const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &X, MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &Y, Teuchos::ETransp mode=Teuchos::NO_TRANS, Scalar alpha=ScalarTraits< Scalar >::one(), Scalar beta=ScalarTraits< Scalar >::zero()) const = 0;
00174
00176 virtual const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getDomainMap() const = 0;
00177
00179 virtual const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getRangeMap() const = 0;
00180
00182
00184
00185
00187 virtual std::string description() const = 0;
00188
00190 virtual void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const = 0;
00191
00193
00194
00196 virtual size_t getNodeNumCols() const = 0;
00197
00199 virtual void getLocalRowCopy(LocalOrdinal LocalRow, const ArrayView< LocalOrdinal > &Indices, const ArrayView< Scalar > &Values, size_t &NumEntries) const = 0;
00200
00201
00202 };
00203
00204 }
00205
00206 #define XPETRA_CRSMATRIX_SHORT
00207 #endif // XPETRA_CRSMATRIX_HPP