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_Pending_alloc: allocate a list of pending tuples
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
#include "pending/GB_Pending.h"
bool GB_Pending_alloc // create a list of pending tuples
(
GrB_Matrix C, // matrix to create C->Pending for
bool iso, // if true, do not allocate Pending->x
GrB_Type type, // type of pending tuples
GrB_BinaryOp op, // operator for assembling pending tuples
int64_t nmax // # of pending tuples to hold
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
ASSERT (C != NULL) ;
ASSERT (C->Pending == NULL) ;
//--------------------------------------------------------------------------
// allocate the Pending header
//--------------------------------------------------------------------------
size_t header_size ;
GB_Pending Pending = GB_MALLOC_MEMORY (1, sizeof (struct GB_Pending_struct),
&header_size) ;
if (Pending == NULL)
{
// out of memory
return (false) ;
}
//--------------------------------------------------------------------------
// initialize the contents of the Pending tuples
//--------------------------------------------------------------------------
nmax = GB_IMAX (nmax, GB_PENDING_INIT) ;
Pending->header_size = header_size ;
Pending->n = 0 ; // no pending tuples yet
Pending->nmax = nmax ; // initial size of list
Pending->sorted = true ; // keep track if tuples are sorted
Pending->type = type ; // type of pending tuples
Pending->size = type->size ; // size of pending tuple type
Pending->op = (iso) ? NULL : op ; // pending operator (NULL is OK)
Pending->i_size = 0 ;
Pending->j_size = 0 ;
Pending->x_size = 0 ;
bool is_matrix = (C->vdim > 1) ;
size_t jsize = (C->j_is_32) ? sizeof (uint32_t) : sizeof (uint64_t) ;
size_t isize = (C->i_is_32) ? sizeof (uint32_t) : sizeof (uint64_t) ;
Pending->i = GB_MALLOC_MEMORY (nmax, isize, &(Pending->i_size)) ;
Pending->j = NULL ;
if (is_matrix)
{
Pending->j = GB_MALLOC_MEMORY (nmax, jsize, &(Pending->j_size)) ;
}
Pending->x = NULL ;
if (!iso)
{
Pending->x = GB_MALLOC_MEMORY (nmax, Pending->size,
&(Pending->x_size)) ;
}
if (Pending->i == NULL
|| (!iso && Pending->x == NULL)
|| (is_matrix && Pending->j == NULL))
{
// out of memory
GB_Pending_free (&Pending) ;
return (false) ;
}
//--------------------------------------------------------------------------
// return result
//--------------------------------------------------------------------------
C->Pending = Pending ;
return (true) ;
}
|