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_TPETRAEXPORT_HPP
00047 #define XPETRA_TPETRAEXPORT_HPP
00048
00049
00050
00051 #include "Xpetra_TpetraConfigDefs.hpp"
00052
00053 #include "Xpetra_Export.hpp"
00054 #include "Xpetra_Exceptions.hpp"
00055
00056 #include "Xpetra_TpetraMap.hpp"
00057 #include "Tpetra_Export.hpp"
00058
00059
00060
00061 namespace Xpetra {
00062
00063
00064 template <class LocalOrdinal, class GlobalOrdinal, class Node>
00065 const Tpetra::Export<LocalOrdinal,GlobalOrdinal,Node> & toTpetra(const Export<LocalOrdinal,GlobalOrdinal,Node> &);
00066 template <class LocalOrdinal, class GlobalOrdinal, class Node>
00067 RCP<const Export<LocalOrdinal,GlobalOrdinal,Node> > toXpetra(const RCP<const Tpetra::Export<LocalOrdinal,GlobalOrdinal,Node> >& exp);
00068
00069 template <class LocalOrdinal = Export<>::local_ordinal_type,
00070 class GlobalOrdinal = typename Export<LocalOrdinal>::global_ordinal_type,
00071 class Node = typename Export<LocalOrdinal, GlobalOrdinal>::node_type>
00072 class TpetraExport
00073 : public Export<LocalOrdinal, GlobalOrdinal, Node>
00074 {
00075
00076 public:
00077
00079 typedef Map<LocalOrdinal,GlobalOrdinal,Node> map_type;
00080
00082
00083
00085 TpetraExport(const Teuchos::RCP< const map_type > &source, const Teuchos::RCP< const map_type > &target)
00086 : export_(Teuchos::rcp(new Tpetra::Export< LocalOrdinal, GlobalOrdinal, Node >(toTpetra(source), toTpetra(target)))) { }
00087
00089 TpetraExport(const Teuchos::RCP< const map_type > &source, const Teuchos::RCP< const map_type > &target, const Teuchos::RCP< Teuchos::ParameterList > &plist)
00090 : export_(Teuchos::rcp(new Tpetra::Export< LocalOrdinal, GlobalOrdinal, Node >(toTpetra(source), toTpetra(target), plist))) { }
00091
00093 TpetraExport(const Export< LocalOrdinal, GlobalOrdinal, Node > &rhs)
00094 : export_(Teuchos::rcp(new Tpetra::Export< LocalOrdinal, GlobalOrdinal, Node >(toTpetra(rhs)))) { }
00095
00097 ~TpetraExport() { }
00098
00100
00102
00103
00105 size_t getNumSameIDs() const { XPETRA_MONITOR("TpetraExport::getNumSameIDs"); return export_->getNumSameIDs(); }
00106
00108 size_t getNumPermuteIDs() const { XPETRA_MONITOR("TpetraExport::getNumPermuteIDs"); return export_->getNumPermuteIDs(); }
00109
00111 ArrayView< const LocalOrdinal > getPermuteFromLIDs() const { XPETRA_MONITOR("TpetraExport::getPermuteFromLIDs"); return export_->getPermuteFromLIDs(); }
00112
00114 ArrayView< const LocalOrdinal > getPermuteToLIDs() const { XPETRA_MONITOR("TpetraExport::getPermuteToLIDs"); return export_->getPermuteToLIDs(); }
00115
00117 size_t getNumRemoteIDs() const { XPETRA_MONITOR("TpetraExport::getNumRemoteIDs"); return export_->getNumRemoteIDs(); }
00118
00120 ArrayView< const LocalOrdinal > getRemoteLIDs() const { XPETRA_MONITOR("TpetraExport::getRemoteLIDs"); return export_->getRemoteLIDs(); }
00121
00123 size_t getNumExportIDs() const { XPETRA_MONITOR("TpetraExport::getNumExportIDs"); return export_->getNumExportIDs(); }
00124
00126 ArrayView< const LocalOrdinal > getExportLIDs() const { XPETRA_MONITOR("TpetraExport::getExportLIDs"); return export_->getExportLIDs(); }
00127
00129 ArrayView< const int > getExportPIDs() const { XPETRA_MONITOR("TpetraExport::getExportPIDs"); return export_->getExportPIDs(); }
00130
00132 Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getSourceMap() const { XPETRA_MONITOR("TpetraExport::getSourceMap"); return toXpetra(export_->getSourceMap()); }
00133
00135 Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getTargetMap() const { XPETRA_MONITOR("TpetraExport::getTargetMap"); return toXpetra(export_->getTargetMap()); }
00136
00138
00140
00141
00143 void print(std::ostream &os) const { XPETRA_MONITOR("TpetraExport::print"); export_->print(os); }
00144
00146
00148
00149
00151 TpetraExport(const RCP<const Tpetra::Export< LocalOrdinal, GlobalOrdinal, Node > > &exp) : export_(exp) { }
00152
00153 RCP< const Tpetra::Export< LocalOrdinal, GlobalOrdinal, Node > > getTpetra_Export() const { return export_; }
00154
00156
00157 private:
00158
00159 RCP<const Tpetra::Export< LocalOrdinal, GlobalOrdinal, Node > > export_;
00160
00161 };
00162
00163
00164 template <class LocalOrdinal, class GlobalOrdinal, class Node>
00165 const Tpetra::Export<LocalOrdinal,GlobalOrdinal,Node> & toTpetra(const Export<LocalOrdinal,GlobalOrdinal,Node> &exp) {
00166
00167 const TpetraExport<LocalOrdinal,GlobalOrdinal,Node> & tpetraExport = dynamic_cast<const TpetraExport<LocalOrdinal,GlobalOrdinal,Node> &>(exp);
00168 return *tpetraExport.getTpetra_Export();
00169 }
00170
00171 template<class LocalOrdinal, class GlobalOrdinal, class Node>
00172 RCP<const Export<LocalOrdinal,GlobalOrdinal,Node> > toXpetra(const RCP<const Tpetra::Export<LocalOrdinal,GlobalOrdinal,Node> >& exp) {
00173 if (!exp.is_null())
00174 return rcp(new TpetraExport<LocalOrdinal,GlobalOrdinal,Node>(exp));
00175
00176 return Teuchos::null;
00177 }
00178
00179 }
00180
00181 #define XPETRA_TPETRAEXPORT_SHORT
00182 #endif // XPETRA_TPETRAEXPORT_HPP