/* -----------------------------------------------------------------
 * Programmer(s): Slaven Peles @ LLNL
 * -----------------------------------------------------------------
 * SUNDIALS Copyright Start
 * Copyright (c) 2002-2024, Lawrence Livermore National Security
 * and Southern Methodist University.
 * All rights reserved.
 *
 * See the top-level LICENSE and NOTICE files for details.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 * SUNDIALS Copyright End
 * -----------------------------------------------------------------*/

#ifndef _SUNDIALS_TPETRA_INTERFACE_HPP_
#define _SUNDIALS_TPETRA_INTERFACE_HPP_

#include <Tpetra_Vector.hpp>
#include <nvector/nvector_trilinos.h>

namespace sundials {
namespace trilinos {
namespace nvector_tpetra {

struct TpetraVectorInterface : public _N_VectorContent_Trilinos
{
  // Typedef of Tpetra vector class to be used with SUNDIALS
  typedef Tpetra::Vector<sunrealtype, int, sunindextype> vector_type;

  TpetraVectorInterface(Teuchos::RCP<vector_type> rcpvec) { rcpvec_ = rcpvec; }

  ~TpetraVectorInterface() = default;

  Teuchos::RCP<vector_type> rcpvec_;
};

} // namespace nvector_tpetra
} // namespace trilinos
} // namespace sundials

inline Teuchos::RCP<sundials::trilinos::nvector_tpetra::TpetraVectorInterface::vector_type> N_VGetVector_Trilinos(
  N_Vector v)
{
  sundials::trilinos::nvector_tpetra::TpetraVectorInterface* iface =
    reinterpret_cast<sundials::trilinos::nvector_tpetra::TpetraVectorInterface*>(
      v->content);

  return iface->rcpvec_;
}

/*
 * -----------------------------------------------------------------
 * Function : N_VMake_Trilinos
 * -----------------------------------------------------------------
 * This function attaches N_Vector functions to a Tpetra vector.
 * -----------------------------------------------------------------
 */

SUNDIALS_EXPORT N_Vector N_VMake_Trilinos(
  Teuchos::RCP<sundials::trilinos::nvector_tpetra::TpetraVectorInterface::vector_type> v,
  SUNContext sunctx);

#endif // _TPETRA_SUNDIALS_INTERFACE_HPP_