/*! \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_random.hpp"

template <typename T, typename I, typename J>
rocsparse_matrix_factory_random<T, I, J>::rocsparse_matrix_factory_random(
    bool fullrank, bool to_int, rocsparse_matrix_init_kind matrix_init_kind)
    : m_fullrank(fullrank)
    , m_to_int(to_int)
    , m_matrix_init_kind(matrix_init_kind){};

template <typename T, typename I, typename J>
void rocsparse_matrix_factory_random<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_random(ptr,
                                  ind,
                                  val,
                                  M,
                                  N,
                                  nnz,
                                  base,
                                  this->m_matrix_init_kind,
                                  this->m_fullrank,
                                  this->m_to_int);

        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_random(csr_row_ptr,
                                  csr_col_ind,
                                  csr_val,
                                  M,
                                  N,
                                  nnz,
                                  base,
                                  this->m_matrix_init_kind,
                                  this->m_fullrank,
                                  this->m_to_int);
        break;
    }
    }
}

template <typename T, typename I, typename J>
void rocsparse_matrix_factory_random<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_random(bsr_row_ptr,
                                bsr_col_ind,
                                bsr_val,
                                Mb,
                                Nb,
                                nnzb,
                                row_block_dim,
                                col_block_dim,
                                base,
                                this->m_matrix_init_kind,
                                this->m_fullrank,
                                this->m_to_int);
}

template <typename T, typename I, typename J>
void rocsparse_matrix_factory_random<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_random(
        coo_row_ind, coo_col_ind, coo_val, M, N, nnz, base, this->m_fullrank, this->m_to_int);
};

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

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

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

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