Tpetra parallel linear algebra  Version of the Day
Tpetra_Import_decl.hpp
1 // @HEADER
2 // ***********************************************************************
3 //
4 // Tpetra: Templated Linear Algebra Services Package
5 // Copyright (2008) Sandia Corporation
6 //
7 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
8 // the U.S. Government retains certain rights in this software.
9 //
10 // Redistribution and use in source and binary forms, with or without
11 // modification, are permitted provided that the following conditions are
12 // met:
13 //
14 // 1. Redistributions of source code must retain the above copyright
15 // notice, this list of conditions and the following disclaimer.
16 //
17 // 2. Redistributions in binary form must reproduce the above copyright
18 // notice, this list of conditions and the following disclaimer in the
19 // documentation and/or other materials provided with the distribution.
20 //
21 // 3. Neither the name of the Corporation nor the names of the
22 // contributors may be used to endorse or promote products derived from
23 // this software without specific prior written permission.
24 //
25 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36 //
37 // Questions? Contact Michael A. Heroux (maherou@sandia.gov)
38 //
39 // ************************************************************************
40 // @HEADER
41 
42 #ifndef TPETRA_IMPORT_DECL_HPP
43 #define TPETRA_IMPORT_DECL_HPP
44 
45 #include <Tpetra_Details_Transfer.hpp>
46 
47 namespace Tpetra {
48  //
49  // Forward declarations. The "doxygen" bit simply tells Doxygen
50  // (our automatic documentation generation system) to skip forward
51  // declarations.
52  //
53 #ifndef DOXYGEN_SHOULD_SKIP_THIS
54  template<class LocalOrdinal, class GlobalOrdinal, class Node>
55  class ImportExportData;
56 
57  template<class LocalOrdinal, class GlobalOrdinal, class Node>
58  class Export;
59 #endif // DOXYGEN_SHOULD_SKIP_THIS
60 
121  template<class LocalOrdinal = Details::DefaultTypes::local_ordinal_type,
122  class GlobalOrdinal = Details::DefaultTypes::global_ordinal_type,
124  class Import:
125  public ::Tpetra::Details::Transfer<LocalOrdinal, GlobalOrdinal, Node>
126  {
127  private:
128  friend class Export<LocalOrdinal,GlobalOrdinal,Node>;
129  public:
131  //typedef typename ::Tpetra::Details::Transfer<LocalOrdinal, GlobalOrdinal, Node>::map_type map_type;
132  typedef ::Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node> map_type;
133 
135 
136 
144  Import (const Teuchos::RCP<const map_type>& source,
145  const Teuchos::RCP<const map_type>& target);
146 
157  Import (const Teuchos::RCP<const map_type>& source,
158  const Teuchos::RCP<const map_type>& target,
159  const RCP<Teuchos::FancyOStream>& out);
160 
173  Import (const Teuchos::RCP<const map_type>& source,
174  const Teuchos::RCP<const map_type>& target,
175  const Teuchos::RCP<Teuchos::ParameterList>& plist);
176 
192  Import (const Teuchos::RCP<const map_type>& source,
193  const Teuchos::RCP<const map_type>& target,
194  const RCP<Teuchos::FancyOStream>& out,
195  const Teuchos::RCP<Teuchos::ParameterList>& plist);
196 
208  Import (const Teuchos::RCP<const map_type>& source,
209  const Teuchos::RCP<const map_type>& target,
210  Teuchos::Array<int> & remotePIDs);
211 
217 
225 
227  virtual ~Import ();
228 
233  void setParameterList (const Teuchos::RCP<Teuchos::ParameterList>& plist);
234 
236 
238 
243  size_t getNumSameIDs() const;
244 
251  size_t getNumPermuteIDs() const;
252 
254  ArrayView<const LocalOrdinal> getPermuteFromLIDs() const;
255 
257  ArrayView<const LocalOrdinal> getPermuteToLIDs() const;
258 
260  size_t getNumRemoteIDs() const;
261 
263  ArrayView<const LocalOrdinal> getRemoteLIDs() const;
264 
266  size_t getNumExportIDs() const;
267 
269  ArrayView<const LocalOrdinal> getExportLIDs() const;
270 
275  ArrayView<const int> getExportPIDs() const;
276 
278  Teuchos::RCP<const map_type> getSourceMap () const;
279 
281  Teuchos::RCP<const map_type> getTargetMap () const;
282 
284  Distributor & getDistributor() const;
285 
289 
323  Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node> >
325 
342  Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node> >
343  setUnion () const;
344 
345 
350  Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node> >
351  createRemoteOnlyImport (const Teuchos::RCP<const map_type>& remoteTarget) const;
352 
353 
355 
357 
373  virtual void print (std::ostream& os) const;
374 
376 
377  private:
379  RCP<ImportExportData<LocalOrdinal,GlobalOrdinal,Node> > ImportData_;
381  RCP<Teuchos::FancyOStream> out_;
383  bool debug_;
384 
386 
387 
404  void
405  init (const Teuchos::RCP<const map_type>& source,
406  const Teuchos::RCP<const map_type>& target,
407  bool useRemotePIDs,
408  Teuchos::Array<int> & remotePIDs,
409  const Teuchos::RCP<Teuchos::ParameterList>& plist);
410 
442  void setupSamePermuteRemote (Teuchos::Array<GlobalOrdinal>& remoteGIDs);
443 
472  void
473  setupExport (Teuchos::Array<GlobalOrdinal>& remoteGIDs, bool useRemotePIDs, Teuchos::Array<int> & remotePIDs);
475 
476 
484  Import (const Teuchos::RCP<const map_type>& source,
485  const Teuchos::RCP<const map_type>& target,
486  const size_t numSameIDs,
487  Teuchos::Array<LocalOrdinal>& permuteToLIDs,
488  Teuchos::Array<LocalOrdinal>& permuteFromLIDs,
489  Teuchos::Array<LocalOrdinal>& remoteLIDs,
490  Teuchos::Array<LocalOrdinal>& exportLIDs,
491  Teuchos::Array<int>& exportPIDs,
492  Distributor& distributor,
493  const Teuchos::RCP<Teuchos::FancyOStream>& out = Teuchos::null,
494  const Teuchos::RCP<Teuchos::ParameterList>& plist = Teuchos::null);
495 
496 
497  }; // class Import
498 
509  template<class LocalOrdinal, class GlobalOrdinal, class Node>
510  Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node> >
511  createImport (const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> >& src,
512  const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> >& tgt)
513  {
514  if (src == tgt) {
515  return Teuchos::null;
516  }
517 #ifdef HAVE_TPETRA_DEBUG
518  TEUCHOS_TEST_FOR_EXCEPTION(
519  src == null || tgt == null, std::runtime_error,
520  "Tpetra::createImport(): neither source nor target map may be null:"
521  << std::endl << "source: " << src << std::endl << "target: " << tgt
522  << std::endl);
523 #endif // HAVE_TPETRA_DEBUG
524  typedef Import<LocalOrdinal, GlobalOrdinal, Node> import_type;
525  return Teuchos::rcp (new import_type (src, tgt));
526  }
527 
539  template<class LocalOrdinal, class GlobalOrdinal, class Node>
540  Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node> >
541  createImport (const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> >& src,
542  const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> >& tgt,
543  const Teuchos::RCP<Teuchos::ParameterList>& plist)
544  {
545  if (src == tgt) {
546  return Teuchos::null;
547  }
548 #ifdef HAVE_TPETRA_DEBUG
549  TEUCHOS_TEST_FOR_EXCEPTION(
550  src == null || tgt == null, std::runtime_error,
551  "Tpetra::createImport(): neither source nor target map may be null:"
552  << std::endl << "source: " << src << std::endl << "target: " << tgt
553  << std::endl);
554 #endif // HAVE_TPETRA_DEBUG
555  typedef Import<LocalOrdinal, GlobalOrdinal, Node> import_type;
556  return Teuchos::rcp (new import_type (src, tgt, plist));
557  }
558 
559 } // namespace Tpetra
560 
561 #endif // TPETRA_IMPORT_DECL_HPP
Communication plan for data redistribution from a uniquely-owned to a (possibly) multiply-owned distr...
::Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > map_type
The specialization of Map used by this class.
Namespace Tpetra contains the class and methods constituting the Tpetra library.
ArrayView< const int > getExportPIDs() const
List of processes to which entries will be sent.
Common base class of Import and Export.
ArrayView< const LocalOrdinal > getRemoteLIDs() const
List of entries in the target Map to receive from other processes.
KokkosClassic::DefaultNode::DefaultNodeType node_type
Default value of Node template parameter.
Teuchos::RCP< const Import< LocalOrdinal, GlobalOrdinal, Node > > createRemoteOnlyImport(const Teuchos::RCP< const map_type > &remoteTarget) const
Returns an importer that contains only the remote entries of this.
size_t getNumExportIDs() const
Number of entries that must be sent by the calling process to other processes.
virtual ~Import()
Destructor.
int local_ordinal_type
Default value of LocalOrdinal template parameter.
Teuchos::RCP< const Import< LocalOrdinal, GlobalOrdinal, Node > > setUnion() const
Return the union of this Import this->getSourceMap()
Import(const Teuchos::RCP< const map_type > &source, const Teuchos::RCP< const map_type > &target)
Construct an Import from the source and target Maps.
ArrayView< const LocalOrdinal > getPermuteFromLIDs() const
List of local IDs in the source Map that are permuted.
Communication plan for data redistribution from a (possibly) multiply-owned to a uniquely-owned distr...
Import< LocalOrdinal, GlobalOrdinal, Node > & operator=(const Import< LocalOrdinal, GlobalOrdinal, Node > &Source)
Assignment operator.
Sets up and executes a communication plan for a Tpetra DistObject.
ArrayView< const LocalOrdinal > getPermuteToLIDs() const
List of local IDs in the target Map that are permuted.
void setParameterList(const Teuchos::RCP< Teuchos::ParameterList > &plist)
Set parameters.
ArrayView< const LocalOrdinal > getExportLIDs() const
List of entries in the source Map that will be sent to other processes.
Describes a parallel distribution of objects over processes.
size_t getNumPermuteIDs() const
Number of IDs to permute but not to communicate.
size_t getNumSameIDs() const
Number of initial identical IDs.
Teuchos::RCP< const map_type > getSourceMap() const
The Source Map used to construct this Import object.
Distributor & getDistributor() const
The Distributor that this Import object uses to move data.
virtual void print(std::ostream &os) const
Print the Import&#39;s data to the given output stream.
size_t getNumRemoteIDs() const
Number of entries not on the calling process.
Teuchos::RCP< const Import< LocalOrdinal, GlobalOrdinal, Node > > createImport(const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &src, const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &tgt)
Nonmember constructor for Import.
Teuchos::RCP< const map_type > getTargetMap() const
The Target Map used to construct this Import object.