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 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129
|
//------------------------------------------------------------------------------
// gbfull: add identity values to a matrix so all entries are present
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// The input may be either a GraphBLAS matrix struct or a standard built-in
// sparse or full matrix. The output is a GraphBLAS matrix by default, with
// all entries present, of the given type. Entries are filled in with the id
// value, whose default value is zero.
// If desc.kind = 'grb', or if the descriptor is not present, the output is a
// GraphBLAS full matrix. Otherwise the output is a built-in full matrix
// (desc.kind = 'full'). The two other cases, desc.kind = 'sparse' and
// 'builtin' are treated as 'full'.
// Usage:
// C = gbfull (A)
// C = gbfull (A, type)
// C = gbfull (A, type, id)
// C = gbfull (A, type, id, desc)
#include "gb_interface.h"
#define USAGE "usage: C = gbfull (A, type, id, desc)"
void mexFunction
(
int nargout,
mxArray *pargout [ ],
int nargin,
const mxArray *pargin [ ]
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
gb_usage (nargin >= 1 && nargin <= 4 && nargout <= 2, USAGE) ;
//--------------------------------------------------------------------------
// get a shallow copy of the input matrix
//--------------------------------------------------------------------------
GrB_Matrix A = gb_get_shallow (pargin [0]) ;
GrB_Index nrows, ncols ;
OK (GrB_Matrix_nrows (&nrows, A)) ;
OK (GrB_Matrix_ncols (&ncols, A)) ;
//--------------------------------------------------------------------------
// get the type of C
//--------------------------------------------------------------------------
GrB_Type type ;
if (nargin > 1)
{
type = gb_mxstring_to_type (pargin [1]) ;
}
else
{
// the output type defaults to the same as the input type
OK (GxB_Matrix_type (&type, A)) ;
}
//--------------------------------------------------------------------------
// get the identity scalar
//--------------------------------------------------------------------------
GrB_Matrix id = NULL ;
if (nargin > 2)
{
id = gb_get_shallow (pargin [2]) ;
}
//--------------------------------------------------------------------------
// get the descriptor
//--------------------------------------------------------------------------
base_enum_t base = BASE_DEFAULT ;
kind_enum_t kind = KIND_GRB ;
GxB_Format_Value fmt = GxB_NO_FORMAT ;
int sparsity = 0 ;
GrB_Descriptor desc = NULL ;
if (nargin > 3)
{
desc = gb_mxarray_to_descriptor (pargin [nargin-1], &kind, &fmt,
&sparsity, &base) ;
}
OK (GrB_Descriptor_free (&desc)) ;
//--------------------------------------------------------------------------
// finalize the kind and format
//--------------------------------------------------------------------------
// ignore desc.kind = 'sparse' or 'builtin' and just use 'full' instead
kind = (kind == KIND_SPARSE || kind == KIND_BUILTIN) ? KIND_FULL : kind ;
if (kind == KIND_FULL)
{
// built-in matrices are always held by column
fmt = GxB_BY_COL ;
}
else
{
// A determines the format of C, unless defined by the descriptor
fmt = gb_get_format (nrows, ncols, A, NULL, fmt) ;
}
//--------------------------------------------------------------------------
// expand A to a full matrix
//--------------------------------------------------------------------------
GrB_Matrix C = gb_expand_to_full (A, type, fmt, id) ;
OK (GrB_Matrix_free (&A)) ;
OK (GrB_Matrix_free (&id)) ;
//--------------------------------------------------------------------------
// export C
//--------------------------------------------------------------------------
pargout [0] = gb_export (&C, kind) ;
pargout [1] = mxCreateDoubleScalar (kind) ;
GB_WRAPUP ;
}
|