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
|
//------------------------------------------------------------------------------
// GB_convert_any_to_iso: convert a matrix from non-iso to iso
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// If the matrix is not iso, it is changed to iso with the scalar value
// provided or with A->x [0] if the scalar is NULL on input. The A->x array is
// optionally compacted to be exactly large enough to hold the single scalar.
#include "GB.h"
GrB_Info GB_convert_any_to_iso // convert non-iso matrix to iso
(
GrB_Matrix A, // input/output matrix
GB_void *scalar, // scalar value, of size A->type->size, or NULL
GB_Context Context
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
ASSERT_MATRIX_OK (A, "A to convert to iso", GB0) ;
//--------------------------------------------------------------------------
// get the iso entry of A, or use the scalar provided on input
//--------------------------------------------------------------------------
size_t asize = A->type->size ;
GB_void ascalar [GB_VLA(asize)] ;
memset (ascalar, 0, asize) ;
if (scalar == NULL && A->iso)
{
memcpy (ascalar, A->x, asize) ;
}
//--------------------------------------------------------------------------
// compact the matrix
//--------------------------------------------------------------------------
if (A->x_size != asize || A->x_shallow || A->x == NULL)
{
//----------------------------------------------------------------------
// free the old A->x and allocate the new A->x
//----------------------------------------------------------------------
if (!A->x_shallow)
{
// free the old space
GB_FREE (&(A->x), A->x_size) ;
}
// allocate the new space
A->x = GB_MALLOC (asize, GB_void, &(A->x_size)) ; // x:OK
A->x_shallow = false ;
if (A->x == NULL)
{
// out of memory
GB_phybix_free (A) ;
return (GrB_OUT_OF_MEMORY) ;
}
}
//--------------------------------------------------------------------------
// copy the iso scalar into A->x [0]
//--------------------------------------------------------------------------
if (scalar == NULL)
{
memcpy (A->x, ascalar, asize) ;
}
else
{
memcpy (A->x, scalar, asize) ;
}
//--------------------------------------------------------------------------
// finalize the matrix and return result
//--------------------------------------------------------------------------
A->iso = true ; // OK: convert_any_to_iso
ASSERT_MATRIX_OK (A, "A converted to iso", GB0) ;
return (GrB_SUCCESS) ;
}
|