#pragma once

#include "../spmat.h"

alphasparse_status_t add_s_bsr_plain(const spmat_bsr_s_t *A, const float alpha, const spmat_bsr_s_t *B, spmat_bsr_s_t **C);
alphasparse_status_t add_s_bsr_trans_plain(const spmat_bsr_s_t *A, const float alpha, const spmat_bsr_s_t *B, spmat_bsr_s_t **C);


// --------------------------------------------------------------------------------------------------------------------------------

// mv
// alpha*A*x + beta*y
alphasparse_status_t gemv_s_bsr_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*A^T*x + beta*y
alphasparse_status_t gemv_s_bsr_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);

// alpha*(L+D+L')*x + beta*y
alphasparse_status_t symv_s_bsr_n_lo_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*(L+I+L')*x + beta*y
alphasparse_status_t symv_s_bsr_u_lo_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*(U'+D+U)*x + beta*y
alphasparse_status_t symv_s_bsr_n_hi_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*(U'+I+U)*x + beta*y
alphasparse_status_t symv_s_bsr_u_hi_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);

// alpha*(L+D)*x + beta*y
alphasparse_status_t trmv_s_bsr_n_lo_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*(L+I)*x + beta*y
alphasparse_status_t trmv_s_bsr_u_lo_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*(U+D)*x + beta*y
alphasparse_status_t trmv_s_bsr_n_hi_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*(U+I)*x + beta*y
alphasparse_status_t trmv_s_bsr_u_hi_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);

// alpha*(L+D)^T*x + beta*y
alphasparse_status_t trmv_s_bsr_n_lo_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*(L+I)^T*x + beta*y
alphasparse_status_t trmv_s_bsr_u_lo_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*(U+D)^T*x + beta*y
alphasparse_status_t trmv_s_bsr_n_hi_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*(U+I)^T*x + beta*y
alphasparse_status_t trmv_s_bsr_u_hi_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);

// alpha*D*x + beta*y
alphasparse_status_t diagmv_s_bsr_n_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);
// alpha*x + beta*y
alphasparse_status_t diagmv_s_bsr_u_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const float beta, float *y);



// --------------------------------------------------------------------------------------------------------------------------------

// ---------------------------------------------------------------------------------------------------------------------------

// alpha*A*B + beta*C
alphasparse_status_t gemm_s_bsr_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
alphasparse_status_t gemm_s_bsr_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*A^T*B + beta*C
alphasparse_status_t gemm_s_bsr_row_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
alphasparse_status_t gemm_s_bsr_col_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);

// alpha*（L+D+L')^T*B + beta*C
alphasparse_status_t symm_s_bsr_n_lo_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(L+I+L')*B + beta*C
alphasparse_status_t symm_s_bsr_u_lo_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U'+D+U)*B + beta*C
alphasparse_status_t symm_s_bsr_n_hi_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U'+I+U)*B + beta*C
alphasparse_status_t symm_s_bsr_u_hi_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);

// alpha*（L+D+L')^T*B + beta*C
alphasparse_status_t symm_s_bsr_n_lo_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(L+I+L')*B + beta*C
alphasparse_status_t symm_s_bsr_u_lo_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U'+D+U)*B + beta*C
alphasparse_status_t symm_s_bsr_n_hi_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U'+I+U)*B + beta*C
alphasparse_status_t symm_s_bsr_u_hi_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);

// alpha*(L+D)*B + beta*C
alphasparse_status_t trmm_s_bsr_n_lo_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(L+I)*B + beta*C
alphasparse_status_t trmm_s_bsr_u_lo_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*op(U+D)*B + beta*C
alphasparse_status_t trmm_s_bsr_n_hi_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*op(U+I)*B + beta*C
alphasparse_status_t trmm_s_bsr_u_hi_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);

// alpha*(L+D)*B + beta*C
alphasparse_status_t trmm_s_bsr_n_lo_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(L+I)*B + beta*C
alphasparse_status_t trmm_s_bsr_u_lo_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U+D)*B + beta*C
alphasparse_status_t trmm_s_bsr_n_hi_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U+I)*B + beta*C
alphasparse_status_t trmm_s_bsr_u_hi_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);

// alpha*(L+D)^T*B + beta*C
alphasparse_status_t trmm_s_bsr_n_lo_row_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(L+I)^T*B + beta*C
alphasparse_status_t trmm_s_bsr_u_lo_row_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U+D)^T*B + beta*C
alphasparse_status_t trmm_s_bsr_n_hi_row_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U+I)^T*B + beta*C
alphasparse_status_t trmm_s_bsr_u_hi_row_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(L+D)^T*B + beta*C
alphasparse_status_t trmm_s_bsr_n_lo_col_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(L+I)^T*B + beta*C
alphasparse_status_t trmm_s_bsr_u_lo_col_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U+D)^T*B + beta*C
alphasparse_status_t trmm_s_bsr_n_hi_col_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*(U+I)^T*B + beta*C
alphasparse_status_t trmm_s_bsr_u_hi_col_trans_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);

// alpha*D*B + beta*C
alphasparse_status_t diagmm_s_bsr_n_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*B + beta*C
alphasparse_status_t diagmm_s_bsr_u_row_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*D*B + beta*C
alphasparse_status_t diagmm_s_bsr_n_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);
// alpha*B + beta*C
alphasparse_status_t diagmm_s_bsr_u_col_plain(const float alpha, const spmat_bsr_s_t *mat, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, const float beta, float *y, const ALPHA_INT ldy);

// ---------------------------------------------------------------------------------------------------------------------------

// -----------------------------------------------------------------------------------------------------

// A*B
alphasparse_status_t spmmd_s_bsr_row_plain(const spmat_bsr_s_t *matA, const spmat_bsr_s_t *matB, float *C, const ALPHA_INT ldc);
// A*B
alphasparse_status_t spmmd_s_bsr_col_plain(const spmat_bsr_s_t *matA, const spmat_bsr_s_t *matB, float *C, const ALPHA_INT ldc);
// A^T*B
alphasparse_status_t spmmd_s_bsr_row_trans_plain(const spmat_bsr_s_t *matA, const spmat_bsr_s_t *matB, float *C, const ALPHA_INT ldc);
// A^T*B
alphasparse_status_t spmmd_s_bsr_col_trans_plain(const spmat_bsr_s_t *matA, const spmat_bsr_s_t *matB, float *C, const ALPHA_INT ldc);

alphasparse_status_t spmm_s_bsr_plain(const spmat_bsr_s_t *A, const spmat_bsr_s_t *B, spmat_bsr_s_t **C);
alphasparse_status_t spmm_s_bsr_trans_plain(const spmat_bsr_s_t *A, const spmat_bsr_s_t *B, spmat_bsr_s_t **C);

// -----------------------------------------------------------------------------------------------------

// alpha*inv(L)*x
alphasparse_status_t trsv_s_bsr_n_lo_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);
// alpha*inv(L)*x
alphasparse_status_t trsv_s_bsr_u_lo_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);
// alpha*inv(U)*x
alphasparse_status_t trsv_s_bsr_n_hi_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);
// alpha*inv(U)*x
alphasparse_status_t trsv_s_bsr_u_hi_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);
// alpha*inv(L^T)*x
alphasparse_status_t trsv_s_bsr_n_lo_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);
// alpha*inv(L^T)*x
alphasparse_status_t trsv_s_bsr_u_lo_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);
// alpha*inv(U^T)*x
alphasparse_status_t trsv_s_bsr_n_hi_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);
// alpha*inv(U^T)*x
alphasparse_status_t trsv_s_bsr_u_hi_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);

// alpha*inv(D)*x
alphasparse_status_t diagsv_s_bsr_n_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);
// alpha*x
alphasparse_status_t diagsv_s_bsr_u_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, float *y);

// alpha*inv(L)*B
alphasparse_status_t trsm_s_bsr_n_lo_row_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(L)*B
alphasparse_status_t trsm_s_bsr_u_lo_row_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(U)*B
alphasparse_status_t trsm_s_bsr_n_hi_row_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(U)*B
alphasparse_status_t trsm_s_bsr_u_hi_row_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(L)*B
alphasparse_status_t trsm_s_bsr_n_lo_col_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(L)*B
alphasparse_status_t trsm_s_bsr_u_lo_col_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(U)*B
alphasparse_status_t trsm_s_bsr_n_hi_col_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(U)*B
alphasparse_status_t trsm_s_bsr_u_hi_col_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);

// alpha*inv(L)*B
alphasparse_status_t trsm_s_bsr_n_lo_row_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(L)*B
alphasparse_status_t trsm_s_bsr_u_lo_row_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(U)*B
alphasparse_status_t trsm_s_bsr_n_hi_row_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(U)*B
alphasparse_status_t trsm_s_bsr_u_hi_row_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(L)*B
alphasparse_status_t trsm_s_bsr_n_lo_col_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(L)*B
alphasparse_status_t trsm_s_bsr_u_lo_col_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(U)*B
alphasparse_status_t trsm_s_bsr_n_hi_col_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(U)*B
alphasparse_status_t trsm_s_bsr_u_hi_col_trans_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);

// alpha*inv(D)*x
alphasparse_status_t diagsm_s_bsr_n_row_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*x
alphasparse_status_t diagsm_s_bsr_u_row_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*inv(D)*x
alphasparse_status_t diagsm_s_bsr_n_col_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);
// alpha*x
alphasparse_status_t diagsm_s_bsr_u_col_plain(const float alpha, const spmat_bsr_s_t *A, const float *x, const ALPHA_INT columns, const ALPHA_INT ldx, float *y, const ALPHA_INT ldy);

alphasparse_status_t set_value_s_bsr_plain (spmat_bsr_s_t * A, const ALPHA_INT row, const ALPHA_INT col, const float value);
alphasparse_status_t update_values_s_bsr_plain (spmat_bsr_s_t * A, const ALPHA_INT nvalues, const ALPHA_INT *indx, const ALPHA_INT *indy, float *values);

