1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
|
//------------------------------------------------------------------------------
// GxB_Matrix_export_CSC: export a matrix in CSC format
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
#include "GB_export.h"
#define GB_FREE_ALL ;
GrB_Info GxB_Matrix_export_CSC // export and free a CSC matrix
(
GrB_Matrix *A, // handle of matrix to export and free
GrB_Type *type, // type of matrix exported
GrB_Index *nrows, // number of rows of the matrix
GrB_Index *ncols, // number of columns of the matrix
GrB_Index **Ap, // column "pointers"
GrB_Index **Ai, // row indices
void **Ax, // values
GrB_Index *Ap_size, // size of Ap in bytes
GrB_Index *Ai_size, // size of Ai in bytes
GrB_Index *Ax_size, // size of Ax in bytes
bool *iso, // if true, A is iso
bool *jumbled, // if true, indices in each column may be unsorted
const GrB_Descriptor desc
)
{
//--------------------------------------------------------------------------
// check inputs and get the descriptor
//--------------------------------------------------------------------------
GB_WHERE1 ("GxB_Matrix_export_CSC (&A, &type, &nrows, &ncols, "
"&Ap, &Ai, &Ax, &Ap_size, &Ai_size, &Ax_size, &iso, "
"&jumbled, desc)") ;
GB_BURBLE_START ("GxB_Matrix_export_CSC") ;
GB_RETURN_IF_NULL (A) ;
GB_RETURN_IF_NULL_OR_FAULTY (*A) ;
GB_GET_DESCRIPTOR (info, desc, xx1, xx2, xx3, xx4, xx5, xx6, xx7) ;
ASSERT_MATRIX_OK (*A, "A to export as by-col", GB0) ;
//--------------------------------------------------------------------------
// ensure the matrix is in by-col format
//--------------------------------------------------------------------------
if (!((*A)->is_csc))
{
// A = A', done in-place, to put A in by-col format
GBURBLE ("(transpose) ") ;
GB_OK (GB_transpose_in_place (*A, true, Context)) ;
}
//--------------------------------------------------------------------------
// finish any pending work
//--------------------------------------------------------------------------
if (jumbled == NULL)
{
// the exported matrix cannot be jumbled
GB_MATRIX_WAIT (*A) ;
}
else
{
// the exported matrix is allowed to be jumbled
GB_MATRIX_WAIT_IF_PENDING_OR_ZOMBIES (*A) ;
}
//--------------------------------------------------------------------------
// ensure the matrix is sparse
//--------------------------------------------------------------------------
GB_OK (GB_convert_any_to_sparse (*A, Context)) ;
//--------------------------------------------------------------------------
// export the matrix
//--------------------------------------------------------------------------
ASSERT (GB_IS_SPARSE (*A)) ;
ASSERT (((*A)->is_csc)) ;
ASSERT (!GB_ZOMBIES (*A)) ;
ASSERT (GB_IMPLIES (jumbled == NULL, !GB_JUMBLED (*A))) ;
ASSERT (!GB_PENDING (*A)) ;
int sparsity ;
bool is_csc ;
info = GB_export (false, A, type, nrows, ncols, false,
Ap, Ap_size, // Ap
NULL, NULL, // Ah
NULL, NULL, // Ab
Ai, Ai_size, // Ai
Ax, Ax_size, // Ax
NULL, jumbled, NULL, // jumbled or not
&sparsity, &is_csc, // sparse by col
iso, Context) ;
if (info == GrB_SUCCESS)
{
ASSERT (sparsity == GxB_SPARSE) ;
ASSERT (is_csc) ;
}
GB_BURBLE_END ;
return (info) ;
}
|