/*! \file */
/* ************************************************************************
 * Copyright (c) 2021-2022 Advanced Micro Devices, Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 * ************************************************************************ */
#include "rocsparse_init.hpp"
#include "rocsparse_matrix_utils.hpp"

#include "rocsparse_matrix_factory_laplace2d.hpp"

template <typename T, typename I, typename J>
rocsparse_matrix_factory_laplace2d<T, I, J>::rocsparse_matrix_factory_laplace2d(J dimx, J dimy)
    : m_dimx(dimx)
    , m_dimy(dimy){};

template <typename T, typename I, typename J>
void rocsparse_matrix_factory_laplace2d<T, I, J>::init_csr(std::vector<I>&       csr_row_ptr,
                                                           std::vector<J>&       csr_col_ind,
                                                           std::vector<T>&       csr_val,
                                                           J&                    M,
                                                           J&                    N,
                                                           I&                    nnz,
                                                           rocsparse_index_base  base,
                                                           rocsparse_matrix_type matrix_type,
                                                           rocsparse_fill_mode   uplo)
{
    switch(matrix_type)
    {
    case rocsparse_matrix_type_symmetric:
    case rocsparse_matrix_type_hermitian:
    case rocsparse_matrix_type_triangular:
    {
        std::vector<I> ptr;
        std::vector<J> ind;
        std::vector<T> val;

        rocsparse_init_csr_laplace2d(ptr, ind, val, this->m_dimx, this->m_dimy, M, N, nnz, base);

        rocsparse_matrix_utils::host_csrtri(ptr.data(),
                                            ind.data(),
                                            val.data(),
                                            csr_row_ptr,
                                            csr_col_ind,
                                            csr_val,
                                            M,
                                            N,
                                            nnz,
                                            base,
                                            uplo);
        break;
    }
    case rocsparse_matrix_type_general:
    {
        rocsparse_init_csr_laplace2d(
            csr_row_ptr, csr_col_ind, csr_val, this->m_dimx, this->m_dimy, M, N, nnz, base);
        break;
    }
    }
}

template <typename T, typename I, typename J>
void rocsparse_matrix_factory_laplace2d<T, I, J>::init_coo(std::vector<I>&      coo_row_ind,
                                                           std::vector<I>&      coo_col_ind,
                                                           std::vector<T>&      coo_val,
                                                           I&                   M,
                                                           I&                   N,
                                                           I&                   nnz,
                                                           rocsparse_index_base base)
{
    rocsparse_init_coo_laplace2d(
        coo_row_ind, coo_col_ind, coo_val, this->m_dimx, this->m_dimy, M, N, nnz, base);
}

template <typename T, typename I, typename J>
void rocsparse_matrix_factory_laplace2d<T, I, J>::init_gebsr(std::vector<I>&      bsr_row_ptr,
                                                             std::vector<J>&      bsr_col_ind,
                                                             std::vector<T>&      bsr_val,
                                                             rocsparse_direction  dirb,
                                                             J&                   Mb,
                                                             J&                   Nb,
                                                             I&                   nnzb,
                                                             J&                   row_block_dim,
                                                             J&                   col_block_dim,
                                                             rocsparse_index_base base)
{
    rocsparse_init_gebsr_laplace2d(bsr_row_ptr,
                                   bsr_col_ind,
                                   bsr_val,
                                   this->m_dimx,
                                   this->m_dimy,
                                   Mb,
                                   Nb,
                                   nnzb,
                                   row_block_dim,
                                   col_block_dim,
                                   base);
}

template struct rocsparse_matrix_factory_laplace2d<float, int32_t, int32_t>;
template struct rocsparse_matrix_factory_laplace2d<float, int64_t, int32_t>;
template struct rocsparse_matrix_factory_laplace2d<float, int64_t, int64_t>;

template struct rocsparse_matrix_factory_laplace2d<double, int32_t, int32_t>;
template struct rocsparse_matrix_factory_laplace2d<double, int64_t, int32_t>;
template struct rocsparse_matrix_factory_laplace2d<double, int64_t, int64_t>;

template struct rocsparse_matrix_factory_laplace2d<rocsparse_float_complex, int32_t, int32_t>;
template struct rocsparse_matrix_factory_laplace2d<rocsparse_float_complex, int64_t, int32_t>;
template struct rocsparse_matrix_factory_laplace2d<rocsparse_float_complex, int64_t, int64_t>;

template struct rocsparse_matrix_factory_laplace2d<rocsparse_double_complex, int32_t, int32_t>;
template struct rocsparse_matrix_factory_laplace2d<rocsparse_double_complex, int64_t, int32_t>;
template struct rocsparse_matrix_factory_laplace2d<rocsparse_double_complex, int64_t, int64_t>;