Tpetra_Details_Transfer_decl.hpp

// @HEADER
// ***********************************************************************
//
//          Tpetra: Templated Linear Algebra Services Package
//                 Copyright (2008) Sandia Corporation
//
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
// the U.S. Government retains certain rights in this software.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// 1. Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//
// 3. Neither the name of the Corporation nor the names of the
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
//
// ************************************************************************
// @HEADER

#ifndef TPETRA_DETAILS_TRANSFER_DECL_HPP
#define TPETRA_DETAILS_TRANSFER_DECL_HPP

#include "Tpetra_Details_Transfer_fwd.hpp"
#include "Tpetra_ImportExportData_fwd.hpp"
#include "Tpetra_Map_decl.hpp"
#include "Teuchos_Describable.hpp"

namespace Tpetra {

// Forward declaration.  The macro tells Doxygen (our automatic
// documentation generation system) to skip forward declarations.
#ifndef DOXYGEN_SHOULD_SKIP_THIS
class Distributor;
#endif // DOXYGEN_SHOULD_SKIP_THIS

namespace Details {

template<class LO,
         class GO,
         class NT>
class Transfer : public Teuchos::Describable {
public:
  using map_type = ::Tpetra::Map<LO, GO, NT>;

private:
  using execution_space = typename NT::device_type::execution_space;

  // See #1088 for why this is not just device_type::memory_space.
#ifdef KOKKOS_ENABLE_CUDA
  static constexpr bool is_cuda =
    std::is_same<execution_space, Kokkos::Cuda>::value;
  using memory_space = typename std::conditional<is_cuda, Kokkos::CudaSpace,
    typename NT::device_type::memory_space>::type;
#else
  using memory_space = typename NT::device_type::memory_space;
#endif // KOKKOS_ENABLE_CUDA
  using device_type = Kokkos::Device<execution_space, memory_space>;

public:
  // Don't be tempted to comment this out if code doesn't build.
  // The point is to ensure correct initialization of TransferData_.
  Transfer () = delete;

  Transfer (const Teuchos::RCP<const map_type>& source,
            const Teuchos::RCP<const map_type>& target,
            const Teuchos::RCP<Teuchos::FancyOStream>& out,
            const Teuchos::RCP<Teuchos::ParameterList>& plist,
            const std::string& className);

  Transfer (const Transfer<LO, GO, NT>& rhs) = default;

  struct reverse_tag {};
  Transfer (const Transfer<LO, GO, NT>& rhs, reverse_tag tag);

  Transfer<LO, GO, NT>& operator= (const Transfer<LO, GO, NT>&) = default;

  virtual ~Transfer () = default;

  size_t getNumSameIDs() const;

  size_t getNumPermuteIDs () const;

  Kokkos::DualView<const LO*, device_type> getPermuteFromLIDs_dv () const;

  Teuchos::ArrayView<const LO> getPermuteFromLIDs () const;

  Kokkos::DualView<const LO*, device_type> getPermuteToLIDs_dv () const;

  Teuchos::ArrayView<const LO> getPermuteToLIDs () const;

  size_t getNumRemoteIDs () const;

  Kokkos::DualView<const LO*, device_type> getRemoteLIDs_dv () const;

  Teuchos::ArrayView<const LO> getRemoteLIDs () const;

  size_t getNumExportIDs () const;

  Kokkos::DualView<const LO*, device_type> getExportLIDs_dv () const;

  Teuchos::ArrayView<const LO> getExportLIDs () const;

  Teuchos::ArrayView<const int> getExportPIDs () const;

  Teuchos::RCP<const map_type> getSourceMap () const;

  Teuchos::RCP<const map_type> getTargetMap () const;

  ::Tpetra::Distributor& getDistributor () const;

  bool isLocallyComplete () const;

  virtual void
  describe (Teuchos::FancyOStream& out,
            const Teuchos::EVerbosityLevel verbLevel =
              Teuchos::Describable::verbLevel_default) const;

protected:
  Teuchos::RCP<ImportExportData<LO, GO, NT> > TransferData_;

  Teuchos::FancyOStream& verboseOutputStream () const;

  bool verbose () const;

  void
  describeImpl (Teuchos::FancyOStream& out,
                const std::string& className,
                const Teuchos::EVerbosityLevel verbLevel =
                  Teuchos::Describable::verbLevel_default) const;

private:
  void
  setParameterList (const Teuchos::RCP<Teuchos::ParameterList>& plist,
                    const std::string& className);

  void
  globalDescribe (Teuchos::FancyOStream& out,
                  const Teuchos::EVerbosityLevel vl) const;

  std::string
  localDescribeToString (const Teuchos::EVerbosityLevel vl) const;
};

} // namespace Details
} // namespace Tpetra

// Explicit instantiation macro.
// Only invoke this when in the Tpetra::Details namespace.
// Most users do not need to use this.
//
// LO: The local ordinal type.
// GO: The global ordinal type.
// NODE: The Kokkos Node type.
#define TPETRA_DETAILS_TRANSFER_INSTANT(LO, GO, NODE) \
  template class Transfer< LO , GO , NODE >;

#endif // TPETRA_DETAILS_TRANSFER_DECL_HPP